Update main.py
Browse files
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"][
|
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"])
|