fixes to tests

This commit is contained in:
Gabriel Almeida 2023-04-29 07:38:06 -03:00
commit ffb2aad45d
5 changed files with 12 additions and 9 deletions

View file

@ -130,7 +130,7 @@ def test_initialize_agent(client: TestClient):
agents = json_response["agents"]
initialize_agent = agents["initialize_agent"]
assert initialize_agent["base_classes"] == ["AgentExecutor"]
assert initialize_agent["base_classes"] == ["AgentExecutor", "function"]
template = initialize_agent["template"]
assert template["agent"] == {