diff --git a/tests/test_agents_template.py b/tests/test_agents_template.py index 12cc278ec..750050c25 100644 --- a/tests/test_agents_template.py +++ b/tests/test_agents_template.py @@ -36,7 +36,7 @@ def test_zero_shot_agent(client: TestClient): "name": "llm_chain", "type": "LLMChain", "list": False, - "advanced": True, + "advanced": False, } assert template["allowed_tools"] == { "required": False, @@ -47,7 +47,7 @@ def test_zero_shot_agent(client: TestClient): "name": "allowed_tools", "type": "Tool", "list": True, - "advanced": True, + "advanced": False, } @@ -70,7 +70,7 @@ def test_json_agent(client: TestClient): "name": "toolkit", "type": "BaseToolkit", "list": False, - "advanced": True, + "advanced": False, } assert template["llm"] == { "required": True, @@ -81,7 +81,7 @@ def test_json_agent(client: TestClient): "name": "llm", "type": "BaseLanguageModel", "list": False, - "advanced": True, + "advanced": False, } @@ -108,7 +108,7 @@ def test_csv_agent(client: TestClient): "type": "file", "list": False, "content": None, - "advanced": True, + "advanced": False, } assert template["llm"] == { "required": True, @@ -119,7 +119,7 @@ def test_csv_agent(client: TestClient): "name": "llm", "type": "BaseLanguageModel", "list": False, - "advanced": True, + "advanced": False, } diff --git a/tests/test_llms_template.py b/tests/test_llms_template.py index 45b3b032d..9f5fdfb4d 100644 --- a/tests/test_llms_template.py +++ b/tests/test_llms_template.py @@ -28,7 +28,7 @@ def test_hugging_face_hub(client: TestClient): "name": "cache", "type": "bool", "list": False, - "advanced": True, + "advanced": False, } assert template["verbose"] == { "required": False, @@ -40,7 +40,7 @@ def test_hugging_face_hub(client: TestClient): "name": "verbose", "type": "bool", "list": False, - "advanced": True, + "advanced": False, } assert template["client"] == { "required": False, @@ -51,7 +51,7 @@ def test_hugging_face_hub(client: TestClient): "name": "client", "type": "Any", "list": False, - "advanced": True, + "advanced": False, } assert template["repo_id"] == { "required": False, @@ -63,7 +63,7 @@ def test_hugging_face_hub(client: TestClient): "name": "repo_id", "type": "str", "list": False, - "advanced": True, + "advanced": False, } assert template["task"] == { "required": True, @@ -120,7 +120,7 @@ def test_openai(client: TestClient): "name": "cache", "type": "bool", "list": False, - "advanced": True, + "advanced": False, } assert template["verbose"] == { "required": False, @@ -131,7 +131,7 @@ def test_openai(client: TestClient): "name": "verbose", "type": "bool", "list": False, - "advanced": True, + "advanced": False, } assert template["client"] == { "required": False, @@ -142,7 +142,7 @@ def test_openai(client: TestClient): "name": "client", "type": "Any", "list": False, - "advanced": True, + "advanced": False, } assert template["model_name"] == { "required": False, @@ -186,7 +186,7 @@ def test_openai(client: TestClient): "name": "max_tokens", "type": "int", "list": False, - "advanced": True, + "advanced": False, } assert template["top_p"] == { "required": False, @@ -198,7 +198,7 @@ def test_openai(client: TestClient): "name": "top_p", "type": "float", "list": False, - "advanced": True, + "advanced": False, } assert template["frequency_penalty"] == { "required": False, @@ -210,7 +210,7 @@ def test_openai(client: TestClient): "name": "frequency_penalty", "type": "float", "list": False, - "advanced": True, + "advanced": False, } assert template["presence_penalty"] == { "required": False, @@ -222,7 +222,7 @@ def test_openai(client: TestClient): "name": "presence_penalty", "type": "float", "list": False, - "advanced": True, + "advanced": False, } assert template["n"] == { "required": False, @@ -234,7 +234,7 @@ def test_openai(client: TestClient): "name": "n", "type": "int", "list": False, - "advanced": True, + "advanced": False, } assert template["best_of"] == { "required": False, @@ -246,7 +246,7 @@ def test_openai(client: TestClient): "name": "best_of", "type": "int", "list": False, - "advanced": True, + "advanced": False, } assert template["model_kwargs"] == { "required": False, @@ -282,7 +282,7 @@ def test_openai(client: TestClient): "name": "batch_size", "type": "int", "list": False, - "advanced": True, + "advanced": False, } assert template["request_timeout"] == { "required": False, @@ -293,7 +293,7 @@ def test_openai(client: TestClient): "name": "request_timeout", "type": "Union[float, Tuple[float, float], NoneType]", "list": False, - "advanced": True, + "advanced": False, } assert template["logit_bias"] == { "required": False, @@ -304,7 +304,7 @@ def test_openai(client: TestClient): "name": "logit_bias", "type": "code", "list": False, - "advanced": True, + "advanced": False, } assert template["max_retries"] == { "required": False, @@ -316,7 +316,7 @@ def test_openai(client: TestClient): "name": "max_retries", "type": "int", "list": False, - "advanced": True, + "advanced": False, } assert template["streaming"] == { "required": False, @@ -328,7 +328,7 @@ def test_openai(client: TestClient): "name": "streaming", "type": "bool", "list": False, - "advanced": True, + "advanced": False, } @@ -351,7 +351,7 @@ def test_chat_open_ai(client: TestClient): "name": "verbose", "type": "bool", "list": False, - "advanced": True, + "advanced": False, } assert template["client"] == { "required": False, @@ -362,7 +362,7 @@ def test_chat_open_ai(client: TestClient): "name": "client", "type": "Any", "list": False, - "advanced": True, + "advanced": False, } assert template["model_name"] == { "required": False, @@ -423,7 +423,7 @@ def test_chat_open_ai(client: TestClient): "name": "request_timeout", "type": "int", "list": False, - "advanced": True, + "advanced": False, } assert template["max_retries"] == { "required": False, @@ -435,7 +435,7 @@ def test_chat_open_ai(client: TestClient): "name": "max_retries", "type": "int", "list": False, - "advanced": True, + "advanced": False, } assert template["streaming"] == { "required": False, @@ -447,7 +447,7 @@ def test_chat_open_ai(client: TestClient): "name": "streaming", "type": "bool", "list": False, - "advanced": True, + "advanced": False, } assert template["n"] == { "required": False, @@ -459,7 +459,7 @@ def test_chat_open_ai(client: TestClient): "name": "n", "type": "int", "list": False, - "advanced": True, + "advanced": False, } assert template["max_tokens"] == { @@ -471,7 +471,7 @@ def test_chat_open_ai(client: TestClient): "name": "max_tokens", "type": "int", "list": False, - "advanced": True, + "advanced": False, } assert template["_type"] == "ChatOpenAI" assert ( diff --git a/tests/test_prompts_template.py b/tests/test_prompts_template.py index ffb53c2a8..1fe6f3775 100644 --- a/tests/test_prompts_template.py +++ b/tests/test_prompts_template.py @@ -27,7 +27,7 @@ def test_prompt_template(client: TestClient): "name": "input_variables", "type": "str", "list": True, - "advanced": True, + "advanced": False, } assert template["output_parser"] == { "required": False, @@ -38,7 +38,7 @@ def test_prompt_template(client: TestClient): "name": "output_parser", "type": "BaseOutputParser", "list": False, - "advanced": True, + "advanced": False, } assert template["partial_variables"] == { "required": False, @@ -49,7 +49,7 @@ def test_prompt_template(client: TestClient): "name": "partial_variables", "type": "code", "list": False, - "advanced": True, + "advanced": False, } assert template["template"] == { "required": True, @@ -72,7 +72,7 @@ def test_prompt_template(client: TestClient): "name": "template_format", "type": "str", "list": False, - "advanced": True, + "advanced": False, } assert template["validate_template"] == { "required": False, @@ -84,7 +84,7 @@ def test_prompt_template(client: TestClient): "name": "validate_template", "type": "bool", "list": False, - "advanced": True, + "advanced": False, } @@ -117,7 +117,7 @@ def test_few_shot_prompt_template(client: TestClient): "name": "example_selector", "type": "BaseExampleSelector", "list": False, - "advanced": True, + "advanced": False, } assert template["example_prompt"] == { "required": True, @@ -128,7 +128,7 @@ def test_few_shot_prompt_template(client: TestClient): "name": "example_prompt", "type": "PromptTemplate", "list": False, - "advanced": True, + "advanced": False, } assert template["suffix"] == { "required": True, @@ -151,7 +151,7 @@ def test_few_shot_prompt_template(client: TestClient): "name": "example_separator", "type": "str", "list": False, - "advanced": True, + "advanced": False, } assert template["prefix"] == { "required": False, @@ -184,7 +184,7 @@ def test_zero_shot_prompt(client: TestClient): "name": "prefix", "type": "str", "list": False, - "advanced": True, + "advanced": False, } assert template["suffix"] == { "required": True, @@ -196,7 +196,7 @@ def test_zero_shot_prompt(client: TestClient): "name": "suffix", "type": "str", "list": False, - "advanced": True, + "advanced": False, } assert template["format_instructions"] == { "required": False, @@ -208,5 +208,5 @@ def test_zero_shot_prompt(client: TestClient): "name": "format_instructions", "type": "str", "list": False, - "advanced": True, + "advanced": False, }