diff --git a/tests/test_agents_template.py b/tests/test_agents_template.py index 4ed9d1153..12cc278ec 100644 --- a/tests/test_agents_template.py +++ b/tests/test_agents_template.py @@ -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, } diff --git a/tests/test_prompts_template.py b/tests/test_prompts_template.py index e5506d4c7..ffb53c2a8 100644 --- a/tests/test_prompts_template.py +++ b/tests/test_prompts_template.py @@ -60,7 +60,7 @@ def test_prompt_template(client: TestClient): "name": "template", "type": "prompt", "list": False, - "advanced": True, + "advanced": False, } assert template["template_format"] == { "required": False, @@ -106,7 +106,7 @@ def test_few_shot_prompt_template(client: TestClient): "name": "examples", "type": "prompt", "list": True, - "advanced": True, + "advanced": False, } assert template["example_selector"] == { "required": False, @@ -139,7 +139,7 @@ def test_few_shot_prompt_template(client: TestClient): "name": "suffix", "type": "prompt", "list": False, - "advanced": True, + "advanced": False, } assert template["example_separator"] == { "required": False, @@ -163,7 +163,7 @@ def test_few_shot_prompt_template(client: TestClient): "name": "prefix", "type": "prompt", "list": False, - "advanced": True, + "advanced": False, }