UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-17 22:12:51 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy J. Baek 2024-06-11 00:24:25 -07:00
parent 3d6f5f418d
commit 049b3136e8
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/main.py
View File

@ -284,6 +284,7 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
context = ""
for tool_id in data["tool_ids"]:
print(tool_id)
response = await get_function_call_response(
prompt=prompt,
tool_id=tool_id,
@ -291,10 +292,9 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
task_model_id=task_model_id,
user=user,
)
print(response)
if response:
context = ("\n" if context != "" else "") + response
context += ("\n" if context != "" else "") + response
if context != "":
system_prompt = rag_template(
@ -304,7 +304,7 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
print(system_prompt)
data["messages"] = add_or_update_system_message(
f"\n{system_prompt}", data["messages"]
f"\n{system_prompt}", data["messages"]
)
del data["tool_ids"]