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

🐛 fix(chains.py): temporarily fix field_type for BaseLanguageModel to improve compatibility

Browse source 
🐛 fix(test_prompts_template.py): fix value of validate_template to False to align with expected behavior
🐛 fix(test_template.py): update value of options to "gpt-4-1106-preview" to reflect the latest version
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-06 21:02:22 -03:00
commit f4e642ec47
3 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/langflow/template/frontend_node/chains.py
View file

@ -87,6 +87,8 @@ class ChainFrontendNode(FrontendNode):
field.required = True field.required = True
field.show = True field.show = True
field.advanced = False field.advanced = False
field.field_type = "BaseLanguageModel" # temporary fix
field.is_list = False
if field.name == "return_source_documents": if field.name == "return_source_documents":
field.required = False field.required = False

2
tests/test_prompts_template.py
View file

@ -96,7 +96,7 @@ def test_prompt_template(client: TestClient, logged_in_headers):
"placeholder": "", "placeholder": "",
"show": False, "show": False,
"multiline": False, "multiline": False,
"value": True, "value": False,
"password": False, "password": False,
"name": "validate_template", "name": "validate_template",
"type": "bool", "type": "bool",

2
tests/test_template.py
View file

@ -237,7 +237,7 @@ def test_format_dict():
"password": False, "password": False,
"multiline": False, "multiline": False,
"options": CHAT_OPENAI_MODELS, "options": CHAT_OPENAI_MODELS,
"value": "gpt-3.5-turbo-0613", "value": "gpt-4-1106-preview",
}, },
} }
assert format_dict(input_dict, "OpenAI") == expected_output_openai assert format_dict(input_dict, "OpenAI") == expected_output_openai