dont delete files and tool_ids

This commit is contained in:
Michael Poluektov 2024-08-19 16:57:29 +01:00
parent c89df923c5
commit 9652c8f8af

View File

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