Merge pull request #4760 from michaelpoluektov/tools-refac-2.1

Fix: tools filter
This commit is contained in:
Timothy Jaeryang Baek 2024-08-20 18:51:28 +02:00 committed by GitHub
commit 20dadf9b5a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -529,6 +529,7 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
"tool_ids": body.pop("tool_ids", None), "tool_ids": body.pop("tool_ids", None),
"files": body.pop("files", None), "files": body.pop("files", None),
} }
body["metadata"] = metadata
__user__ = { __user__ = {
"id": user.id, "id": user.id,
@ -600,7 +601,6 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
if len(citations) > 0: if len(citations) > 0:
data_items.append({"citations": citations}) data_items.append({"citations": citations})
body["metadata"] = metadata
modified_body_bytes = json.dumps(body).encode("utf-8") modified_body_bytes = json.dumps(body).encode("utf-8")
# Replace the request body with the modified one # Replace the request body with the modified one
request._body = modified_body_bytes request._body = modified_body_bytes