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

fix) Fix combinedMemory

Browse source
This commit is contained in:
Johngdae 2023-09-19 12:02:18 +00:00
commit 9b22030a73
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",
"history": "chat_history",
}
memory_key = langchain_object.memory.memory_key
return mem_key_dict.get(memory_key)
# Check if memory_key attribute exists
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):
@ -86,4 +90,5 @@ def update_memory_keys(langchain_object, possible_new_mem_key):
try:
setattr(langchain_object.memory, attr, key)
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})")