From 6ea7454fe314d4d99216e14ac2c7fab8073a7011 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Sun, 10 Dec 2023 11:41:01 -0300
Subject: [PATCH] Add range_spec to llms_template.py

---
 tests/test_llms_template.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/tests/test_llms_template.py b/tests/test_llms_template.py
index 494c1cc90..90e50dacd 100644
--- a/tests/test_llms_template.py
+++ b/tests/test_llms_template.py
@@ -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,