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

fix tests to comply with updates

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

8
tests/test_agents_template.py
View file

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

8
tests/test_prompts_template.py
View file

@ -60,7 +60,7 @@ def test_prompt_template(client: TestClient):
"name": "template", "name": "template",
"type": "prompt", "type": "prompt",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["template_format"] == { assert template["template_format"] == {
"required": False, "required": False,
@ -106,7 +106,7 @@ def test_few_shot_prompt_template(client: TestClient):
"name": "examples", "name": "examples",
"type": "prompt", "type": "prompt",
"list": True, "list": True,
"advanced": True, "advanced": False,
} }
assert template["example_selector"] == { assert template["example_selector"] == {
"required": False, "required": False,
@ -139,7 +139,7 @@ def test_few_shot_prompt_template(client: TestClient):
"name": "suffix", "name": "suffix",
"type": "prompt", "type": "prompt",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["example_separator"] == { assert template["example_separator"] == {
"required": False, "required": False,
@ -163,7 +163,7 @@ def test_few_shot_prompt_template(client: TestClient):
"name": "prefix", "name": "prefix",
"type": "prompt", "type": "prompt",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }