Fix range_spec key in test_llms_template.py

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-10 16:51:00 -03:00
commit 1a348bb220

View file

@ -88,7 +88,7 @@ def test_openai(client: TestClient, logged_in_headers):
"list": False, "list": False,
"advanced": False, "advanced": False,
"info": "", "info": "",
"range_spec": {"max": 1.0, "min": -1.0, "step": 0.1}, "rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1},
"fileTypes": [], "fileTypes": [],
} }
assert template["max_tokens"] == { assert template["max_tokens"] == {
@ -119,7 +119,7 @@ def test_openai(client: TestClient, logged_in_headers):
"list": False, "list": False,
"advanced": False, "advanced": False,
"info": "", "info": "",
"range_spec": {"max": 1.0, "min": -1.0, "step": 0.1}, "rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1},
"fileTypes": [], "fileTypes": [],
} }
assert template["frequency_penalty"] == { assert template["frequency_penalty"] == {
@ -135,7 +135,7 @@ def test_openai(client: TestClient, logged_in_headers):
"list": False, "list": False,
"advanced": False, "advanced": False,
"info": "", "info": "",
"range_spec": {"max": 1.0, "min": -1.0, "step": 0.1}, "rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1},
"fileTypes": [], "fileTypes": [],
} }
assert template["presence_penalty"] == { assert template["presence_penalty"] == {
@ -151,7 +151,7 @@ def test_openai(client: TestClient, logged_in_headers):
"list": False, "list": False,
"advanced": False, "advanced": False,
"info": "", "info": "",
"range_spec": {"max": 1.0, "min": -1.0, "step": 0.1}, "rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1},
"fileTypes": [], "fileTypes": [],
} }
assert template["n"] == { assert template["n"] == {
@ -241,7 +241,7 @@ def test_openai(client: TestClient, logged_in_headers):
"list": False, "list": False,
"advanced": False, "advanced": False,
"info": "", "info": "",
"range_spec": {"max": 1.0, "min": -1.0, "step": 0.1}, "rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1},
"fileTypes": [], "fileTypes": [],
} }
assert template["logit_bias"] == { assert template["logit_bias"] == {
@ -364,7 +364,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers):
"list": False, "list": False,
"advanced": False, "advanced": False,
"info": "", "info": "",
"range_spec": {"max": 1.0, "min": -1.0, "step": 0.1}, "rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1},
"fileTypes": [], "fileTypes": [],
} }
assert template["model_kwargs"] == { assert template["model_kwargs"] == {
@ -409,7 +409,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers):
"list": False, "list": False,
"advanced": False, "advanced": False,
"info": "", "info": "",
"range_spec": {"max": 1.0, "min": -1.0, "step": 0.1}, "rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1},
"fileTypes": [], "fileTypes": [],
} }
assert template["max_retries"] == { assert template["max_retries"] == {