UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-04-25 08:48:21 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-02-13 01:57:02 -08:00
parent 6acda2e6ec
commit acb3eef619
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
backend/open_webui/routers/tasks.py
View File

@ -203,7 +203,7 @@ async def generate_title(
} }
), ),
"metadata": { "metadata": {
**(request.state.metadata if request.state.metadata else {}), **(request.state.metadata if hasattr(request.state, "metadata") else {}),
"task": str(TASKS.TITLE_GENERATION), "task": str(TASKS.TITLE_GENERATION),
"task_body": form_data, "task_body": form_data,
"chat_id": form_data.get("chat_id", None), "chat_id": form_data.get("chat_id", None),
@ -272,7 +272,7 @@ async def generate_chat_tags(
"messages": [{"role": "user", "content": content}], "messages": [{"role": "user", "content": content}],
"stream": False, "stream": False,
"metadata": { "metadata": {
**(request.state.metadata if request.state.metadata else {}), **(request.state.metadata if hasattr(request.state, "metadata") else {}),
"task": str(TASKS.TAGS_GENERATION), "task": str(TASKS.TAGS_GENERATION),
"task_body": form_data, "task_body": form_data,
"chat_id": form_data.get("chat_id", None), "chat_id": form_data.get("chat_id", None),
@ -338,7 +338,7 @@ async def generate_image_prompt(
"messages": [{"role": "user", "content": content}], "messages": [{"role": "user", "content": content}],
"stream": False, "stream": False,
"metadata": { "metadata": {
**(request.state.metadata if request.state.metadata else {}), **(request.state.metadata if hasattr(request.state, "metadata") else {}),
"task": str(TASKS.IMAGE_PROMPT_GENERATION), "task": str(TASKS.IMAGE_PROMPT_GENERATION),
"task_body": form_data, "task_body": form_data,
"chat_id": form_data.get("chat_id", None), "chat_id": form_data.get("chat_id", None),
@ -415,7 +415,7 @@ async def generate_queries(
"messages": [{"role": "user", "content": content}], "messages": [{"role": "user", "content": content}],
"stream": False, "stream": False,
"metadata": { "metadata": {
**(request.state.metadata if request.state.metadata else {}), **(request.state.metadata if hasattr(request.state, "metadata") else {}),
"task": str(TASKS.QUERY_GENERATION), "task": str(TASKS.QUERY_GENERATION),
"task_body": form_data, "task_body": form_data,
"chat_id": form_data.get("chat_id", None), "chat_id": form_data.get("chat_id", None),
@ -496,7 +496,7 @@ async def generate_autocompletion(
"messages": [{"role": "user", "content": content}], "messages": [{"role": "user", "content": content}],
"stream": False, "stream": False,
"metadata": { "metadata": {
**(request.state.metadata if request.state.metadata else {}), **(request.state.metadata if hasattr(request.state, "metadata") else {}),
"task": str(TASKS.AUTOCOMPLETE_GENERATION), "task": str(TASKS.AUTOCOMPLETE_GENERATION),
"task_body": form_data, "task_body": form_data,
"chat_id": form_data.get("chat_id", None), "chat_id": form_data.get("chat_id", None),
@ -567,7 +567,7 @@ async def generate_emoji(
), ),
"chat_id": form_data.get("chat_id", None), "chat_id": form_data.get("chat_id", None),
"metadata": { "metadata": {
**(request.state.metadata if request.state.metadata else {}), **(request.state.metadata if hasattr(request.state, "metadata") else {}),
"task": str(TASKS.EMOJI_GENERATION), "task": str(TASKS.EMOJI_GENERATION),
"task_body": form_data, "task_body": form_data,
}, },
@ -626,7 +626,7 @@ async def generate_moa_response(
"messages": [{"role": "user", "content": content}], "messages": [{"role": "user", "content": content}],
"stream": form_data.get("stream", False), "stream": form_data.get("stream", False),
"metadata": { "metadata": {
**(request.state.metadata if request.state.metadata else {}), **(request.state.metadata if hasattr(request.state, "metadata") else {}),
"chat_id": form_data.get("chat_id", None), "chat_id": form_data.get("chat_id", None),
"task": str(TASKS.MOA_RESPONSE_GENERATION), "task": str(TASKS.MOA_RESPONSE_GENERATION),
"task_body": form_data, "task_body": form_data,