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

fix: loading agent correctly and added correct chat models loading

Browse source
This commit is contained in:
Gabriel Almeida 2023-04-02 10:41:40 -03:00
commit 58652f7c2b
9 changed files with 35 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/test_template.py
View file

@ -210,7 +210,7 @@ def test_format_dict():
}
assert format_dict(input_dict) == expected_output
# Test 7: Check class name-specific cases (OpenAI, OpenAIChat)
# Test 7: Check class name-specific cases (OpenAI, ChatOpenAI)
input_dict = {
"model_name": {"type": "str", "required": False},
}
@ -237,7 +237,7 @@ def test_format_dict():
},
}
assert format_dict(input_dict, "OpenAI") == expected_output_openai
assert format_dict(input_dict, "OpenAIChat") == expected_output_openai_chat
assert format_dict(input_dict, "ChatOpenAI") == expected_output_openai_chat
# Test 8: Replace dict type with str
input_dict = {