This commit is contained in:
Timothy Jaeryang Baek
2026-03-31 23:39:54 -05:00
parent e6f38f52c8
commit 378673408e
2 changed files with 7 additions and 5 deletions
+4
View File
@@ -1882,6 +1882,10 @@ async def chat_completion(
generate_chat_completions = chat_completion
generate_chat_completion = chat_completion
# Expose as app.state so internal callers (e.g. automations) can
# use the full pipeline without importing from main.py (avoids circular deps).
app.state.CHAT_COMPLETION_HANDLER = chat_completion
##################################
#
+3 -5
View File
@@ -292,12 +292,10 @@ async def execute_automation(app, automation: AutomationModel) -> None:
if filter_ids:
form_data["filter_ids"] = filter_ids
# Call chat_completion — returns {'status': True, 'task_id': '...'}
# The background task handles everything: LLM, tools, DB, webhooks.
from open_webui.main import chat_completion as main_chat_completion
# Call the full chat completion pipeline (same as POST /api/chat/completions).
# The handler reference is stored on app.state to avoid circular imports.
request = _build_request(app)
await main_chat_completion(request, form_data, user=user)
await app.state.CHAT_COMPLETION_HANDLER(request, form_data, user=user)
# Notify user
from open_webui.socket.main import sio