ソースを参照

fix: filter compatibility issue

Timothy J. Baek 8 ヶ月 前
コミット
6fcd40d4d8
1 ファイル変更9 行追加3 行削除
  1. 9 3
      backend/main.py

+ 9 - 3
backend/main.py

@@ -525,9 +525,8 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
             "chat_id": body.pop("chat_id", None),
             "message_id": body.pop("id", None),
             "session_id": body.pop("session_id", None),
-            "valves": body.pop("valves", None),
-            "tool_ids": body.pop("tool_ids", None),
-            "files": body.pop("files", None),
+            "tool_ids": body.get("tool_ids", None),
+            "files": body.get("files", None),
         }
         body["metadata"] = metadata
 
@@ -560,6 +559,13 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
                 content={"detail": str(e)},
             )
 
+        metadata = {
+            **metadata,
+            "tool_ids": body.pop("tool_ids", None),
+            "files": body.pop("files", None),
+        }
+        body["metadata"] = metadata
+
         try:
             body, flags = await chat_completion_tools_handler(body, user, extra_params)
             contexts.extend(flags.get("contexts", []))