fix the threshold env to float conversion

OWASP · Sep 28, 2023 · 42e6963 · 42e6963
1 parent bfbf8c5
commit 42e6963
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/application/prompt_client/prompt_client.py b/application/prompt_client/prompt_client.py
@@ -20,7 +20,7 @@
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.INFO)
 
-SIMILARITY_THRESHOLD = os.environ.get("CHATBOT_SIMILARITY_THRESHOLD", None) or 0.7
+SIMILARITY_THRESHOLD = float(os.environ.get("CHATBOT_SIMILARITY_THRESHOLD", "0.7"))
 
 
 def is_valid_url(url):