fix: add handle

main.py

@@ -36,11 +36,13 @@ class Delete(BaseModel):
 
 websocketClient = contextvars.ContextVar("websocketClient")
 
-class MyCustomSyncHandler(BaseCallbackHandler):
-    def on_llm_new_token(self, token: str, **kwargs) -> None:
+class MyCustomSyncHandler(AsyncCallbackHandler):
+    async def on_llm_new_token(self, token: str, **kwargs) -> None:
         ws = websocketClient.get()
 
-        ws.send_text(f"token: {token}")
+        print(f"{ws}")
+
+        await ws.send_text(f"token: {token}")
 
         print(f"token: {token}")
 
@@ -251,4 +253,6 @@ async def websocket_endpoint(websocket: WebSocket):
 
         res = QA(data)
 
+        print(f"${res}")
+
         await websocket.send_text(f"Message text was:")