chore(base.py): add type ignore comment to clear_cache function
fix(base.py): raise ValueError if content is None in save_binary_file function feat(nodes.py): add memory field to TimeTravelGuideChainNode test(test_chains_template.py): add test for memory field in TimeTravelGuideChainNode
This commit is contained in:
parent
db951cde20
commit
64bf97ff84
3 changed files with 21 additions and 2 deletions
5
src/backend/langflow/cache/base.py
vendored
5
src/backend/langflow/cache/base.py
vendored
|
|
@ -47,7 +47,7 @@ def memoize_dict(maxsize=128):
|
||||||
def clear_cache():
|
def clear_cache():
|
||||||
cache.clear()
|
cache.clear()
|
||||||
|
|
||||||
wrapper.clear_cache = clear_cache
|
wrapper.clear_cache = clear_cache # type: ignore
|
||||||
return wrapper
|
return wrapper
|
||||||
|
|
||||||
return decorator
|
return decorator
|
||||||
|
|
@ -119,7 +119,8 @@ def save_binary_file(content: str, file_name: str, accepted_types: list[str]) ->
|
||||||
|
|
||||||
# Get the destination folder
|
# Get the destination folder
|
||||||
cache_path = Path(tempfile.gettempdir()) / PREFIX
|
cache_path = Path(tempfile.gettempdir()) / PREFIX
|
||||||
|
if content is None:
|
||||||
|
raise ValueError("Please, reload the file in the loader.")
|
||||||
data = content.split(",")[1]
|
data = content.split(",")[1]
|
||||||
decoded_bytes = base64.b64decode(data)
|
decoded_bytes = base64.b64decode(data)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -150,6 +150,13 @@ class TimeTravelGuideChainNode(FrontendNode):
|
||||||
multiline=False,
|
multiline=False,
|
||||||
name="llm",
|
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."
|
description: str = "Time travel guide chain to be used in the flow."
|
||||||
|
|
|
||||||
|
|
@ -435,5 +435,16 @@ def test_time_travel_guide_chain(client: TestClient):
|
||||||
"list": False,
|
"list": False,
|
||||||
"advanced": 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."
|
assert chain["description"] == "Time travel guide chain to be used in the flow."
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue