datacipen commited on
Commit
3ea6f79
·
verified ·
1 Parent(s): b5b1652

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -151,7 +151,7 @@ async def setup_conversationalChain():
151
  retriever = await Retriever(cl.user_session.get("selectRequest"))
152
  qa = ConversationalRetrievalChain.from_llm(
153
  model,
154
- memory=memory,
155
  chain_type="stuff",
156
  return_source_documents=True,
157
  verbose=False,
@@ -228,7 +228,7 @@ async def on_chat_resume(thread: ThreadDict):
228
  memory = ConversationBufferMemory(return_messages=True)
229
  root_messages = [m for m in thread["steps"] if m["parentId"] == None]
230
  for message in root_messages:
231
- if message["type"] == "assistant_message" and message["output"][29:38] == "thématique":
232
  resName = await switch(message["output"][42:])
233
  cl.user_session.set("selectRequest", resName)
234
  print(message["output"])
 
151
  retriever = await Retriever(cl.user_session.get("selectRequest"))
152
  qa = ConversationalRetrievalChain.from_llm(
153
  model,
154
+ memory=cl.user_session.get("memory"),
155
  chain_type="stuff",
156
  return_source_documents=True,
157
  verbose=False,
 
228
  memory = ConversationBufferMemory(return_messages=True)
229
  root_messages = [m for m in thread["steps"] if m["parentId"] == None]
230
  for message in root_messages:
231
+ if message["type"] == "assistant_message" and message["output"][28:37] == "thématique":
232
  resName = await switch(message["output"][42:])
233
  cl.user_session.set("selectRequest", resName)
234
  print(message["output"])