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:
Gabriel Almeida 2023-05-04 18:24:12 -03:00
commit 64bf97ff84
3 changed files with 21 additions and 2 deletions

View file

@ -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)

View file

@ -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."

View file

@ -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."