test

app/main.py

@@ -72,6 +72,12 @@ async def websocket_endpoint(websocket: WebSocket, client_id: int):
         await manager.broadcast(json.dumps(message))
 
 
+async def fake_data_streamer():
+    for i in range(10):
+        yield b'some fake data\n\n'
+        await asyncio.sleep(0.5)
+
+
 async def astreamer(generator):
     try:
         print("streaming........")
@@ -88,7 +94,8 @@ async def process_input(text: str):
     if text and len(text.strip()) > 0:
         text = text.strip()
         streaming_response = session_assistant.ask(text)
-        return StreamingResponse(astreamer(streaming_response.response_gen), media_type='text/event-stream')
+        return StreamingResponse(fake_data_streamer(), media_type='text/event-stream')
+        # return StreamingResponse(astreamer(streaming_response.response_gen), media_type='text/event-stream')
 
 
 @app.post("/upload")