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"])
|