fix: add websocket in handlerToken

main.py

@@ -73,8 +73,7 @@ class MyCustomSyncHandler(BaseCallbackHandler):
     def on_llm_new_token(self, token: str, **kwargs) -> Any:
         print(token)
 
-        if self.websocket != None:
-            asyncio.run(self.websocket.send_text(token))
+        self.state(token)
 
         print(token)