Browse Source

fix: files

Timothy J. Baek 10 tháng trước cách đây
mục cha
commit
eb21750466
1 tập tin đã thay đổi với 3 bổ sung5 xóa
  1. 3 5
      backend/main.py

+ 3 - 5
backend/main.py

@@ -354,10 +354,10 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
                 print(f"tool_context: {context}")
 
             # If docs field is present, generate RAG completions
-            if "docs" in data:
+            if "files" in data:
                 data = {**data}
                 rag_context, citations = get_rag_context(
-                    docs=data["docs"],
+                    docs=data["files"],
                     messages=data["messages"],
                     embedding_function=rag_app.state.EMBEDDING_FUNCTION,
                     k=rag_app.state.config.TOP_K,
@@ -369,8 +369,7 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
                 if rag_context:
                     context += ("\n" if context != "" else "") + rag_context
 
-                del data["docs"]
-
+                del data["files"]
                 log.debug(f"rag_context: {rag_context}, citations: {citations}")
 
             if context != "":
@@ -1380,7 +1379,6 @@ async def update_pipeline_valves(
         )
 
 
-
 @app.get("/api/config")
 async def get_app_config():
     # Checking and Handling the Absence of 'ui' in CONFIG_DATA