fix: add websocket in handlerToken

main.py

@@ -51,10 +51,10 @@ class MyCustomSyncHandler(BaseCallbackHandler):
         self.message = ''
 
     def on_llm_new_token(self, token: str, **kwargs) -> Any:
-        message += token
-        redisClient.publish(f'{kwargs["tags"][0]}',  message)
+        self.message += token
+        redisClient.publish(f'{kwargs["tags"][0]}',  self.message)
 
-        print(message)
+        print(self.message)
 
 LLM = load_model(device_type=DEVICE_TYPE, model_id=MODEL_ID, model_basename=MODEL_BASENAME, stream=True)
 
@@ -246,6 +246,8 @@ async def websocket_endpoint(websocket: WebSocket,  client_id: int):
 
 
 
+
+
     except WebSocketDisconnect:
         print('disconnect')
     except RuntimeError as error: