bsmith3715 commited on
Commit
dc601f5
·
verified ·
1 Parent(s): ce04382

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -129,8 +129,9 @@ async def handle_message(message: cl.Message):
129
  msg = cl.Message(content="")
130
  await msg.send()
131
 
 
132
  # Stream the response
133
- async for chunk in rag_processor.generate_response(message.content):
134
  await msg.stream_token(chunk)
135
  except Exception as e:
136
  await cl.Message(content=f"Error processing your message: {e}").send()
 
129
  msg = cl.Message(content="")
130
  await msg.send()
131
 
132
+ qa_chain = cl.user_session.get("qa_chain")
133
  # Stream the response
134
+ async for chunk in qa_chain.astream(message.content):
135
  await msg.stream_token(chunk)
136
  except Exception as e:
137
  await cl.Message(content=f"Error processing your message: {e}").send()