Daniel Marques commited on
Commit
e04f8da
1 Parent(s): 13026dd

fix: add websocket in handlerToken

Browse files
Files changed (1) hide show
  1. main.py +4 -3
main.py CHANGED
@@ -271,9 +271,10 @@ async def websocket_endpoint(websocket: WebSocket):
271
  oldReceiveText = prompt
272
  res = QA(prompt)
273
 
274
- if statusProcess == True :
275
- tokenState = get()
276
- await websocket.send_text(f"token: {tokenState}")
 
277
 
278
  except WebSocketDisconnect:
279
  print('disconnect')
 
271
  oldReceiveText = prompt
272
  res = QA(prompt)
273
 
274
+ print(statusProcess);
275
+
276
+ tokenState = get()
277
+ await websocket.send_text(f"token: {tokenState}")
278
 
279
  except WebSocketDisconnect:
280
  print('disconnect')