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():
|
||||
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)
|
||||
|
||||
|
|
|
|||
|
|
@ -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."
|
||||
|
|
|
|||
|
|
@ -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."
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue