ahncs commited on
Commit
f98c553
·
1 Parent(s): 0f11ca2
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -46,13 +46,13 @@ async def start_chat():
46
  cl.user_session.set("chain", chain)
47
 
48
  @cl.on_message
49
- async def main(message):
50
  chain = cl.user_session.get("chain")
51
  cb = cl.AsyncLangchainCallbackHandler(
52
  stream_final_answer=True, answer_prefix_tokens=["FINAL", "ANSWER"]
53
  )
54
  cb.answer_reached = True
55
- res = await chain.acall(message, callbacks=[cb])
56
 
57
  answer = res["result"]
58
 
 
46
  cl.user_session.set("chain", chain)
47
 
48
  @cl.on_message
49
+ async def main(message: cl.Message):
50
  chain = cl.user_session.get("chain")
51
  cb = cl.AsyncLangchainCallbackHandler(
52
  stream_final_answer=True, answer_prefix_tokens=["FINAL", "ANSWER"]
53
  )
54
  cb.answer_reached = True
55
+ res = await chain.acall(message.content, callbacks=[cb])
56
 
57
  answer = res["result"]
58