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

refactor(tests): change 'advanced' field to False in all tests templates

Browse source
This commit is contained in:
Gabriel Almeida 2023-05-02 11:48:00 -03:00
commit 13e58b55b3
3 changed files with 42 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/test_agents_template.py
View file

@ -36,7 +36,7 @@ def test_zero_shot_agent(client: TestClient):
"name": "llm_chain", "name": "llm_chain",
"type": "LLMChain", "type": "LLMChain",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["allowed_tools"] == { assert template["allowed_tools"] == {
"required": False, "required": False,
@ -47,7 +47,7 @@ def test_zero_shot_agent(client: TestClient):
"name": "allowed_tools", "name": "allowed_tools",
"type": "Tool", "type": "Tool",
"list": True, "list": True,
"advanced": True, "advanced": False,
} }
@ -70,7 +70,7 @@ def test_json_agent(client: TestClient):
"name": "toolkit", "name": "toolkit",
"type": "BaseToolkit", "type": "BaseToolkit",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["llm"] == { assert template["llm"] == {
"required": True, "required": True,
@ -81,7 +81,7 @@ def test_json_agent(client: TestClient):
"name": "llm", "name": "llm",
"type": "BaseLanguageModel", "type": "BaseLanguageModel",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
@ -108,7 +108,7 @@ def test_csv_agent(client: TestClient):
"type": "file", "type": "file",
"list": False, "list": False,
"content": None, "content": None,
"advanced": True, "advanced": False,
} }
assert template["llm"] == { assert template["llm"] == {
"required": True, "required": True,
@ -119,7 +119,7 @@ def test_csv_agent(client: TestClient):
"name": "llm", "name": "llm",
"type": "BaseLanguageModel", "type": "BaseLanguageModel",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }

50
tests/test_llms_template.py
View file

@ -28,7 +28,7 @@ def test_hugging_face_hub(client: TestClient):
"name": "cache", "name": "cache",
"type": "bool", "type": "bool",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["verbose"] == { assert template["verbose"] == {
"required": False, "required": False,
@ -40,7 +40,7 @@ def test_hugging_face_hub(client: TestClient):
"name": "verbose", "name": "verbose",
"type": "bool", "type": "bool",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["client"] == { assert template["client"] == {
"required": False, "required": False,
@ -51,7 +51,7 @@ def test_hugging_face_hub(client: TestClient):
"name": "client", "name": "client",
"type": "Any", "type": "Any",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["repo_id"] == { assert template["repo_id"] == {
"required": False, "required": False,
@ -63,7 +63,7 @@ def test_hugging_face_hub(client: TestClient):
"name": "repo_id", "name": "repo_id",
"type": "str", "type": "str",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["task"] == { assert template["task"] == {
"required": True, "required": True,
@ -120,7 +120,7 @@ def test_openai(client: TestClient):
"name": "cache", "name": "cache",
"type": "bool", "type": "bool",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["verbose"] == { assert template["verbose"] == {
"required": False, "required": False,
@ -131,7 +131,7 @@ def test_openai(client: TestClient):
"name": "verbose", "name": "verbose",
"type": "bool", "type": "bool",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["client"] == { assert template["client"] == {
"required": False, "required": False,
@ -142,7 +142,7 @@ def test_openai(client: TestClient):
"name": "client", "name": "client",
"type": "Any", "type": "Any",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["model_name"] == { assert template["model_name"] == {
"required": False, "required": False,
@ -186,7 +186,7 @@ def test_openai(client: TestClient):
"name": "max_tokens", "name": "max_tokens",
"type": "int", "type": "int",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["top_p"] == { assert template["top_p"] == {
"required": False, "required": False,
@ -198,7 +198,7 @@ def test_openai(client: TestClient):
"name": "top_p", "name": "top_p",
"type": "float", "type": "float",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["frequency_penalty"] == { assert template["frequency_penalty"] == {
"required": False, "required": False,
@ -210,7 +210,7 @@ def test_openai(client: TestClient):
"name": "frequency_penalty", "name": "frequency_penalty",
"type": "float", "type": "float",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["presence_penalty"] == { assert template["presence_penalty"] == {
"required": False, "required": False,
@ -222,7 +222,7 @@ def test_openai(client: TestClient):
"name": "presence_penalty", "name": "presence_penalty",
"type": "float", "type": "float",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["n"] == { assert template["n"] == {
"required": False, "required": False,
@ -234,7 +234,7 @@ def test_openai(client: TestClient):
"name": "n", "name": "n",
"type": "int", "type": "int",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["best_of"] == { assert template["best_of"] == {
"required": False, "required": False,
@ -246,7 +246,7 @@ def test_openai(client: TestClient):
"name": "best_of", "name": "best_of",
"type": "int", "type": "int",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["model_kwargs"] == { assert template["model_kwargs"] == {
"required": False, "required": False,
@ -282,7 +282,7 @@ def test_openai(client: TestClient):
"name": "batch_size", "name": "batch_size",
"type": "int", "type": "int",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["request_timeout"] == { assert template["request_timeout"] == {
"required": False, "required": False,
@ -293,7 +293,7 @@ def test_openai(client: TestClient):
"name": "request_timeout", "name": "request_timeout",
"type": "Union[float, Tuple[float, float], NoneType]", "type": "Union[float, Tuple[float, float], NoneType]",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["logit_bias"] == { assert template["logit_bias"] == {
"required": False, "required": False,
@ -304,7 +304,7 @@ def test_openai(client: TestClient):
"name": "logit_bias", "name": "logit_bias",
"type": "code", "type": "code",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["max_retries"] == { assert template["max_retries"] == {
"required": False, "required": False,
@ -316,7 +316,7 @@ def test_openai(client: TestClient):
"name": "max_retries", "name": "max_retries",
"type": "int", "type": "int",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["streaming"] == { assert template["streaming"] == {
"required": False, "required": False,
@ -328,7 +328,7 @@ def test_openai(client: TestClient):
"name": "streaming", "name": "streaming",
"type": "bool", "type": "bool",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
@ -351,7 +351,7 @@ def test_chat_open_ai(client: TestClient):
"name": "verbose", "name": "verbose",
"type": "bool", "type": "bool",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["client"] == { assert template["client"] == {
"required": False, "required": False,
@ -362,7 +362,7 @@ def test_chat_open_ai(client: TestClient):
"name": "client", "name": "client",
"type": "Any", "type": "Any",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["model_name"] == { assert template["model_name"] == {
"required": False, "required": False,
@ -423,7 +423,7 @@ def test_chat_open_ai(client: TestClient):
"name": "request_timeout", "name": "request_timeout",
"type": "int", "type": "int",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["max_retries"] == { assert template["max_retries"] == {
"required": False, "required": False,
@ -435,7 +435,7 @@ def test_chat_open_ai(client: TestClient):
"name": "max_retries", "name": "max_retries",
"type": "int", "type": "int",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["streaming"] == { assert template["streaming"] == {
"required": False, "required": False,
@ -447,7 +447,7 @@ def test_chat_open_ai(client: TestClient):
"name": "streaming", "name": "streaming",
"type": "bool", "type": "bool",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["n"] == { assert template["n"] == {
"required": False, "required": False,
@ -459,7 +459,7 @@ def test_chat_open_ai(client: TestClient):
"name": "n", "name": "n",
"type": "int", "type": "int",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["max_tokens"] == { assert template["max_tokens"] == {
@ -471,7 +471,7 @@ def test_chat_open_ai(client: TestClient):
"name": "max_tokens", "name": "max_tokens",
"type": "int", "type": "int",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["_type"] == "ChatOpenAI" assert template["_type"] == "ChatOpenAI"
assert ( assert (

22
tests/test_prompts_template.py
View file

@ -27,7 +27,7 @@ def test_prompt_template(client: TestClient):
"name": "input_variables", "name": "input_variables",
"type": "str", "type": "str",
"list": True, "list": True,
"advanced": True, "advanced": False,
} }
assert template["output_parser"] == { assert template["output_parser"] == {
"required": False, "required": False,
@ -38,7 +38,7 @@ def test_prompt_template(client: TestClient):
"name": "output_parser", "name": "output_parser",
"type": "BaseOutputParser", "type": "BaseOutputParser",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["partial_variables"] == { assert template["partial_variables"] == {
"required": False, "required": False,
@ -49,7 +49,7 @@ def test_prompt_template(client: TestClient):
"name": "partial_variables", "name": "partial_variables",
"type": "code", "type": "code",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["template"] == { assert template["template"] == {
"required": True, "required": True,
@ -72,7 +72,7 @@ def test_prompt_template(client: TestClient):
"name": "template_format", "name": "template_format",
"type": "str", "type": "str",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["validate_template"] == { assert template["validate_template"] == {
"required": False, "required": False,
@ -84,7 +84,7 @@ def test_prompt_template(client: TestClient):
"name": "validate_template", "name": "validate_template",
"type": "bool", "type": "bool",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
@ -117,7 +117,7 @@ def test_few_shot_prompt_template(client: TestClient):
"name": "example_selector", "name": "example_selector",
"type": "BaseExampleSelector", "type": "BaseExampleSelector",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["example_prompt"] == { assert template["example_prompt"] == {
"required": True, "required": True,
@ -128,7 +128,7 @@ def test_few_shot_prompt_template(client: TestClient):
"name": "example_prompt", "name": "example_prompt",
"type": "PromptTemplate", "type": "PromptTemplate",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["suffix"] == { assert template["suffix"] == {
"required": True, "required": True,
@ -151,7 +151,7 @@ def test_few_shot_prompt_template(client: TestClient):
"name": "example_separator", "name": "example_separator",
"type": "str", "type": "str",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["prefix"] == { assert template["prefix"] == {
"required": False, "required": False,
@ -184,7 +184,7 @@ def test_zero_shot_prompt(client: TestClient):
"name": "prefix", "name": "prefix",
"type": "str", "type": "str",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["suffix"] == { assert template["suffix"] == {
"required": True, "required": True,
@ -196,7 +196,7 @@ def test_zero_shot_prompt(client: TestClient):
"name": "suffix", "name": "suffix",
"type": "str", "type": "str",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }
assert template["format_instructions"] == { assert template["format_instructions"] == {
"required": False, "required": False,
@ -208,5 +208,5 @@ def test_zero_shot_prompt(client: TestClient):
"name": "format_instructions", "name": "format_instructions",
"type": "str", "type": "str",
"list": False, "list": False,
"advanced": True, "advanced": False,
} }