fix: add type filter message

main.py

@@ -230,12 +230,15 @@ async def websocket_endpoint(websocket: WebSocket,  client_id: int):
             with concurrent.futures.ThreadPoolExecutor() as executor:
                executor.submit(QA(inputs=prompt, return_only_outputs=True, callbacks=[MyCustomSyncHandler()], tags=f'{client_id}', include_run_info=True))
 
+            i  = 0
             for item in pubsub.listen():
+                i+=1
                 itemJson = str(item)
-                print(itemJson)
-                # await websocket.send_text(f'{item}')
+                print(item["type"] == "message")
 
+                if(i > 10 ): pubsub.unsubscribe({client_id})
 
+                # await websocket.send_text(f'{item}')
 
     except WebSocketDisconnect:
         print('disconnect')