From 0a08a4d2fbb5c6d04a7e2b8ca9516dfe8d5c300f Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Mon, 8 Jul 2024 21:40:22 -0700 Subject: [PATCH] refac --- backend/main.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/backend/main.py b/backend/main.py index 79915e8f1..7355a2159 100644 --- a/backend/main.py +++ b/backend/main.py @@ -1188,6 +1188,14 @@ async def chat_completed(form_data: dict, user=Depends(get_verified_user)): to=data["session_id"], ) + async def __event_call__(data): + response = await sio.call( + "chat-events", + {"chat_id": data["chat_id"], "message_id": data["id"], "data": data}, + to=data["session_id"], + ) + return response + def get_priority(function_id): function = Functions.get_function_by_id(function_id) if function is not None and hasattr(function, "valves"): @@ -1275,6 +1283,12 @@ async def chat_completed(form_data: dict, user=Depends(get_verified_user)): "__event_emitter__": __event_emitter__, } + if "__event_call__" in sig.parameters: + params = { + **params, + "__event_call__": __event_call__, + } + if inspect.iscoroutinefunction(outlet): data = await outlet(**params) else: