diff --git a/src/backend/langflow/cache/base.py b/src/backend/langflow/cache/base.py index 73439e9dd..3d667b8b4 100644 --- a/src/backend/langflow/cache/base.py +++ b/src/backend/langflow/cache/base.py @@ -47,7 +47,7 @@ def memoize_dict(maxsize=128): def clear_cache(): cache.clear() - wrapper.clear_cache = clear_cache + wrapper.clear_cache = clear_cache # type: ignore return wrapper return decorator @@ -119,7 +119,8 @@ def save_binary_file(content: str, file_name: str, accepted_types: list[str]) -> # Get the destination folder cache_path = Path(tempfile.gettempdir()) / PREFIX - + if content is None: + raise ValueError("Please, reload the file in the loader.") data = content.split(",")[1] decoded_bytes = base64.b64decode(data) diff --git a/src/backend/langflow/template/nodes.py b/src/backend/langflow/template/nodes.py index 15e3d2d85..61a41589b 100644 --- a/src/backend/langflow/template/nodes.py +++ b/src/backend/langflow/template/nodes.py @@ -150,6 +150,13 @@ class TimeTravelGuideChainNode(FrontendNode): multiline=False, name="llm", ), + TemplateField( + field_type="BaseChatMemory", + required=False, + show=True, + name="memory", + advanced=False, + ), ], ) description: str = "Time travel guide chain to be used in the flow." diff --git a/tests/test_chains_template.py b/tests/test_chains_template.py index 13bef2e9c..33af32e57 100644 --- a/tests/test_chains_template.py +++ b/tests/test_chains_template.py @@ -435,5 +435,16 @@ def test_time_travel_guide_chain(client: TestClient): "list": False, "advanced": False, } + assert template["memory"] == { + "required": False, + "placeholder": "", + "show": True, + "multiline": False, + "password": False, + "name": "memory", + "type": "BaseChatMemory", + "list": False, + "advanced": False, + } assert chain["description"] == "Time travel guide chain to be used in the flow."