Update main.py

app/main.py

@@ -74,11 +74,13 @@ async def websocket_endpoint(websocket: WebSocket, client_id: int):
 
 async def astreamer(generator):
     try:
+        print("streaming........")
         for i in generator:
+            print(i)
             yield (i)
             await asyncio.sleep(.1)
     except asyncio.CancelledError as e:
-        print('cancelled')
+        yield ('cancelled')
 
 
 @app.get("/query")