Spaces:
Paused
Paused
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 |
|