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

🎨 style(tests): add display_name field to LLM in test_agents_template.py and test_chains_template.py

Browse source 
The display_name field was added to the LLM object in the test_agents_template.py and test_chains_template.py files. This field is used to display the name of the LLM in the UI.
This commit is contained in:
Gabriel Almeida 2023-05-23 11:51:31 -03:00
commit 5dfaa9a81a
2 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/test_agents_template.py
View file

@ -82,6 +82,7 @@ def test_json_agent(client: TestClient):
"type": "BaseLanguageModel",
"list": False,
"advanced": False,
"display_name": "LLM",
}
@ -120,6 +121,7 @@ def test_csv_agent(client: TestClient):
"type": "BaseLanguageModel",
"list": False,
"advanced": False,
"display_name": "LLM",
}
@ -183,4 +185,5 @@ def test_initialize_agent(client: TestClient):
"type": "BaseLanguageModel",
"list": False,
"advanced": False,
"display_name": "LLM",
}

3
tests/test_chains_template.py
View file

@ -326,6 +326,7 @@ def test_series_character_chain(client: TestClient):
assert template["llm"] == {
"required": True,
"display_name": "LLM",
"placeholder": "",
"show": True,
"multiline": False,
@ -388,6 +389,7 @@ def test_mid_journey_prompt_chain(client: TestClient):
assert template["llm"] == {
"required": True,
"display_name": "LLM",
"placeholder": "",
"show": True,
"multiline": False,
@ -427,6 +429,7 @@ def test_time_travel_guide_chain(client: TestClient):
assert template["llm"] == {
"required": True,
"placeholder": "",
"display_name": "LLM",
"show": True,
"multiline": False,
"password": False,