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

Add range_spec to llms_template.py

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-10 11:41:01 -03:00
commit 6ea7454fe3
1 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tests/test_llms_template.py
View file

@ -84,6 +84,7 @@ def test_openai(client: TestClient, logged_in_headers):
"list": False,
"advanced": False,
"info": "",
"range_spec": {"max": 1.0, "min": -1.0, "step": 0.1},
}
assert template["max_tokens"] == {
"required": False,
@ -112,6 +113,7 @@ def test_openai(client: TestClient, logged_in_headers):
"list": False,
"advanced": False,
"info": "",
"range_spec": {"max": 1.0, "min": -1.0, "step": 0.1},
}
assert template["frequency_penalty"] == {
"required": False,
@ -126,6 +128,7 @@ def test_openai(client: TestClient, logged_in_headers):
"list": False,
"advanced": False,
"info": "",
"range_spec": {"max": 1.0, "min": -1.0, "step": 0.1},
}
assert template["presence_penalty"] == {
"required": False,
@ -140,6 +143,7 @@ def test_openai(client: TestClient, logged_in_headers):
"list": False,
"advanced": False,
"info": "",
"range_spec": {"max": 1.0, "min": -1.0, "step": 0.1},
}
assert template["n"] == {
"required": False,
@ -223,6 +227,7 @@ def test_openai(client: TestClient, logged_in_headers):
"list": False,
"advanced": False,
"info": "",
"range_spec": {"max": 1.0, "min": -1.0, "step": 0.1},
}
assert template["logit_bias"] == {
"required": False,
@ -338,6 +343,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers):
"list": False,
"advanced": False,
"info": "",
"range_spec": {"max": 1.0, "min": -1.0, "step": 0.1},
}
assert template["model_kwargs"] == {
"required": False,
@ -379,6 +385,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers):
"list": False,
"advanced": False,
"info": "",
"range_spec": {"max": 1.0, "min": -1.0, "step": 0.1},
}
assert template["max_retries"] == {
"required": False,