This commit is contained in:
Timothy Jaeryang Baek
2026-02-04 21:09:05 -06:00
parent 8b75d34a8a
commit 020cb0d4bf

View File

@@ -2400,7 +2400,7 @@ async def process_chat_payload(request, form_data, user, metadata, model):
return form_data, metadata, events
async def background_tasks_handler(request, form_data, user, metadata, tasks):
def get_event_emitter_and_caller(metadata):
event_emitter = None
event_caller = None
if (
@@ -2413,6 +2413,11 @@ async def background_tasks_handler(request, form_data, user, metadata, tasks):
):
event_emitter = get_event_emitter(metadata)
event_caller = get_event_call(metadata)
return event_emitter, event_caller
async def background_tasks_handler(request, form_data, user, metadata, tasks):
event_emitter, event_caller = get_event_emitter_and_caller(metadata)
message = None
messages = []
@@ -2631,18 +2636,7 @@ async def background_tasks_handler(request, form_data, user, metadata, tasks):
async def process_chat_response(
request, response, form_data, user, metadata, model, events, tasks
):
event_emitter = None
event_caller = None
if (
"session_id" in metadata
and metadata["session_id"]
and "chat_id" in metadata
and metadata["chat_id"]
and "message_id" in metadata
and metadata["message_id"]
):
event_emitter = get_event_emitter(metadata)
event_caller = get_event_call(metadata)
event_emitter, event_caller = get_event_emitter_and_caller(metadata)
# Non-streaming response
if not isinstance(response, StreamingResponse):