Pārlūkot izejas kodu

dont delete files and tool_ids

Michael Poluektov 8 mēneši atpakaļ
vecāks
revīzija
9652c8f8af
1 mainītis faili ar 2 papildinājumiem un 2 dzēšanām
  1. 2 2
      backend/main.py

+ 2 - 2
backend/main.py

@@ -371,7 +371,7 @@ async def chat_completion_tools_handler(
     body: dict, user: UserModel, extra_params: dict
 ) -> tuple[dict, dict]:
     # If tool_ids field is present, call the functions
-    tool_ids = body.pop("tool_ids", None)
+    tool_ids = body.get("tool_ids", None)
     if not tool_ids:
         return body, {}
 
@@ -464,7 +464,7 @@ async def chat_completion_files_handler(body) -> tuple[dict, dict[str, list]]:
     contexts = []
     citations = []
 
-    if files := body.pop("files", None):
+    if files := body.get("files", None):
         contexts, citations = get_rag_context(
             files=files,
             messages=body["messages"],