UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-12-28 06:42:47 +00:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2024-12-24 19:27:17 -07:00
parent 346856b578
commit e51722348a
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
backend/open_webui/utils/chat.py
View File

@ -114,7 +114,7 @@ async def generate_chat_completion(
yield chunk yield chunk
response = await generate_chat_completion( response = await generate_chat_completion(
form_data, user, bypass_filter=True request, form_data, user, bypass_filter=True
) )
return StreamingResponse( return StreamingResponse(
stream_wrapper(response.body_iterator), stream_wrapper(response.body_iterator),
@ -123,14 +123,18 @@ async def generate_chat_completion(
) )
else: else:
return { return {
**(await generate_chat_completion(form_data, user, bypass_filter=True)), **(
await generate_chat_completion(
request, form_data, user, bypass_filter=True
)
),
"selected_model_id": selected_model_id, "selected_model_id": selected_model_id,
} }
if model.get("pipe"): if model.get("pipe"):
# Below does not require bypass_filter because this is the only route the uses this function and it is already bypassing the filter # Below does not require bypass_filter because this is the only route the uses this function and it is already bypassing the filter
return await generate_function_chat_completion( return await generate_function_chat_completion(
form_data, user=user, models=models request, form_data, user=user, models=models
) )
if model["owned_by"] == "ollama": if model["owned_by"] == "ollama":
# Using /ollama/api/chat endpoint # Using /ollama/api/chat endpoint