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",
"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,
}

50
tests/test_llms_template.py
View file

@ -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 (

22
tests/test_prompts_template.py
View file

@ -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,
}