jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix) Fix combinedMemory (#935)

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-09-19 13:41:51 -03:00 • committed by GitHub
commit 2901fb0e94
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/backend/langflow/interface/run.py
View file

@ -58,8 +58,12 @@ def get_memory_key(langchain_object):
"chat_history": "history", "chat_history": "history",
"history": "chat_history", "history": "chat_history",
} }
memory_key = langchain_object.memory.memory_key # Check if memory_key attribute exists
return mem_key_dict.get(memory_key) if hasattr(langchain_object.memory, 'memory_key'):
memory_key = langchain_object.memory.memory_key
return mem_key_dict.get(memory_key)
else:
return None # or some other default value or action
def update_memory_keys(langchain_object, possible_new_mem_key): def update_memory_keys(langchain_object, possible_new_mem_key):
@ -86,4 +90,5 @@ def update_memory_keys(langchain_object, possible_new_mem_key):
try: try:
setattr(langchain_object.memory, attr, key) setattr(langchain_object.memory, attr, key)
except ValueError as exc: except ValueError as exc:
logger.debug(f"{langchain_object.memory} has no attribute {attr} ({exc})") logger.debug(
f"{langchain_object.memory} has no attribute {attr} ({exc})")