Fix_vecstore_memory (#582)

src/backend/langflow/interface/initialize/loading.py

@@ -100,6 +100,8 @@ def instantiate_llm(node_type, class_object, params: Dict):
 
 def instantiate_memory(node_type, class_object, params):
     try:
+        if "retriever" in params and hasattr(params["retriever"], "as_retriever"):
+            params["retriever"] = params["retriever"].as_retriever()
         return class_object(**params)
     # I want to catch a specific attribute error that happens
     # when the object does not have a cursor attribute

src/backend/langflow/interface/run.py

@@ -62,6 +62,10 @@ def update_memory_keys(langchain_object, possible_new_mem_key):
         if key not in [langchain_object.memory.memory_key, possible_new_mem_key]
     ][0]
 
-    langchain_object.memory.input_key = input_key
+    keys = [input_key, output_key, possible_new_mem_key]
-    langchain_object.memory.output_key = output_key
+    attrs = ["input_key", "output_key", "memory_key"]
-    langchain_object.memory.memory_key = possible_new_mem_key
+    for key, attr in zip(keys, attrs):
+        try:
+            setattr(langchain_object.memory, attr, key)
+        except ValueError as exc:
+            logger.debug(f"{langchain_object.memory} has no attribute {attr} ({exc})")

src/backend/langflow/processing/base.py

@@ -51,5 +51,6 @@ async def get_result_and_steps(langchain_object, message: str, **kwargs):
         )
         thought = format_actions(intermediate_steps) if intermediate_steps else ""
     except Exception as exc:
+        logger.exception(exc)
         raise ValueError(f"Error: {str(exc)}") from exc
     return result, thought

src/backend/langflow/template/frontend_node/memories.py

@@ -37,6 +37,7 @@ class MemoryFrontendNode(FrontendNode):
                 value="",
             )
         )
+        if self.template.type_name not in {"VectorStoreRetrieverMemory"}:
             self.template.add_field(
                 TemplateField(
                     field_type="str",

src/backend/langflow/template/frontend_node/vectorstores.py

@@ -51,7 +51,7 @@ class VectorStoreFrontendNode(FrontendNode):
                 required=False,
                 show=True,
                 advanced=False,
-                value=True,
+                value=False,
                 display_name="Persist",
             )
             extra_fields.append(extra_field)