Skip to content

Commit 1045c2e

Browse files
committed
bugfix, semantic caching failed when conversation memory not enabled
1 parent fd5527f commit 1045c2e

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/core/RedisRetrievalChain.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -151,12 +151,12 @@ def get_chat_history(inputs) -> str:
151151
try:
152152
result = chatbot.invoke({"question": q, "chat_history": redis_history})
153153

154+
references = {}
155+
for doc in result['source_documents']:
156+
references[doc.metadata['id'].split('idx:')[-1]] = doc.metadata
157+
154158
# decide if conversation history should be saved
155159
if self.cfg.is_memory():
156-
references = {}
157-
for doc in result['source_documents']:
158-
references[doc.metadata['id'].split('idx:')[-1]] = doc.metadata
159-
160160
redis_history.add_user_message(result["question"])
161161
redis_history.add_message(BaseMessage(content=result["answer"], type="ai", additional_kwargs=references))
162162

0 commit comments

Comments
 (0)