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,
"advanced": False,
"info": "",
"range_spec": {"max": 1.0, "min": -1.0, "step": 0.1},
"rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1},
"fileTypes": [],
}
assert template["max_tokens"] == {
@ -119,7 +119,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},
"rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1},
"fileTypes": [],
}
assert template["frequency_penalty"] == {
@ -135,7 +135,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},
"rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1},
"fileTypes": [],
}
assert template["presence_penalty"] == {
@ -151,7 +151,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},
"rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1},
"fileTypes": [],
}
assert template["n"] == {
@ -241,7 +241,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},
"rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1},
"fileTypes": [],
}
assert template["logit_bias"] == {
@ -364,7 +364,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},
"rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1},
"fileTypes": [],
}
assert template["model_kwargs"] == {
@ -409,7 +409,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},
"rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1},
"fileTypes": [],
}
assert template["max_retries"] == {