fix tests to comply with updates

This commit is contained in:
Gabriel Almeida 2023-04-29 19:48:59 -03:00
commit 2b6f70fdb4
2 changed files with 8 additions and 8 deletions

View file

@ -149,7 +149,7 @@ def test_initialize_agent(client: TestClient):
"name": "agent",
"type": "str",
"list": True,
"advanced": True,
"advanced": False,
}
assert template["memory"] == {
"required": False,
@ -160,7 +160,7 @@ def test_initialize_agent(client: TestClient):
"name": "memory",
"type": "BaseChatMemory",
"list": False,
"advanced": True,
"advanced": False,
}
assert template["tools"] == {
"required": False,
@ -171,7 +171,7 @@ def test_initialize_agent(client: TestClient):
"name": "tools",
"type": "Tool",
"list": True,
"advanced": True,
"advanced": False,
}
assert template["llm"] == {
"required": True,
@ -182,5 +182,5 @@ def test_initialize_agent(client: TestClient):
"name": "llm",
"type": "BaseLanguageModel",
"list": False,
"advanced": True,
"advanced": False,
}