jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: update type to silence warning and memory chatbot json (#3265)

Browse source 
* fix: update value type in DefaultPromptField from str to Any for increased flexibility in input handling

* fix: replace _instantiate_input with instantiate_input for consistency in input instantiation across the codebase

* chore: update test durations

* update memory chatbot
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-08-09 13:01:25 -03:00 • committed by GitHub
commit 9da2c34551
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 369 additions and 361 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/backend/tests/unit/inputs/test_inputs.py
View file

@ -20,7 +20,7 @@ from langflow.inputs.inputs import (
SecretStrInput,
StrInput,
TableInput,
_instantiate_input,
instantiate_input,
)
from langflow.schema.message import Message
@ -69,14 +69,14 @@ def test_message_text_input_invalid():
def test_instantiate_input_valid():
data = {"name": "valid_input", "value": "This is a string"}
input_instance = _instantiate_input("StrInput", data)
input_instance = instantiate_input("StrInput", data)
assert isinstance(input_instance, StrInput)
assert input_instance.value == "This is a string"
def test_instantiate_input_invalid():
with pytest.raises(ValueError):
_instantiate_input("InvalidInput", {"name": "invalid_input", "value": "This is a string"})
instantiate_input("InvalidInput", {"name": "invalid_input", "value": "This is a string"})
def test_handle_input_valid():
@ -218,8 +218,8 @@ def test_instantiate_input_comprehensive():
}
for input_type, data in valid_data.items():
input_instance = _instantiate_input(input_type, data)
input_instance = instantiate_input(input_type, data)
assert isinstance(input_instance, InputTypesMap[input_type])
with pytest.raises(ValueError):
_instantiate_input("InvalidInput", {"name": "invalid_input", "value": "Invalid"})
instantiate_input("InvalidInput", {"name": "invalid_input", "value": "Invalid"})