refac/fix: direct connection floating action buttons

This commit is contained in:
Timothy Jaeryang Baek committed 2025-10-02 02:21:21 -05:00
1 parent b6538b2cdd
commit 6c4deed37a
2 files changed
+16 -6

No files matched your search

+5 -1
View File
@@ -80,6 +80,7 @@ async def generate_direct_chat_completion(
event_caller = get_event_call(metadata)
channel = f"{user_id}:{session_id}:{request_id}"
logging.info(f"WebSocket channel: {channel}")
if form_data.get("stream"):
q = asyncio.Queue()
@@ -121,7 +122,10 @@ async def generate_direct_chat_completion(
yield f"data: {json.dumps(data)}\n\n"
elif isinstance(data, str):
yield data
if "data:" in data:
yield f"{data}\n\n"
else:
yield f"data: {data}\n\n"
except Exception as e:
log.debug(f"Error in event generator: {e}")
pass