Daniel Marques commited on
Commit
194045e
1 Parent(s): 47df29d

feat: add backend

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -40,7 +40,7 @@ class MyCustomSyncHandler(AsyncCallbackHandler):
40
  async def on_llm_new_token(self, token: str, **kwargs) -> None:
41
  print(f"{websocket_state}")
42
 
43
- await websocket_state(f"token: {token}")
44
 
45
  print(f"token: {token}")
46
 
 
40
  async def on_llm_new_token(self, token: str, **kwargs) -> None:
41
  print(f"{websocket_state}")
42
 
43
+ await websocket_state.send_text(f"token: {token}")
44
 
45
  print(f"token: {token}")
46