fix: add websocket in handlerToken

main.py

@@ -70,8 +70,6 @@ class MyCustomSyncHandler(BaseCallbackHandler):
         self.end = True
 
     def on_llm_new_token(self, token: str, **kwargs) -> Any:
-        print(token)
-
         if self.callback :
             self.callback(token)
 
@@ -262,7 +260,7 @@ async def websocket_endpoint(websocket: WebSocket,  client_id: int):
             if (oldReceiveText != prompt):
                 handlerToken.callback = websocket.send_text
                 oldReceiveText = prompt
-                asyncio.run(QA(prompt))
+                await QA(prompt)
 
     except WebSocketDisconnect:
         print('disconnect')