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

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

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",
}