From 1153a6d13dd292d6138adb5a66b77bdfdade3c64 Mon Sep 17 00:00:00 2001
From: dwmorris11 <dustin.morris@outlook.com>
Date: Sun, 11 Feb 2024 22:02:13 -0800
Subject: [PATCH] changed score_threshold to be float only
---
semantic_router/encoders/mistral.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/semantic_router/encoders/mistral.py b/semantic_router/encoders/mistral.py
index b0314284..20f7f339 100644
--- a/semantic_router/encoders/mistral.py
+++ b/semantic_router/encoders/mistral.py
@@ -20,7 +20,7 @@ class MistralEncoder(BaseEncoder):
self,
name: Optional[str] = None,
mistral_api_key: Optional[str] = None,
- score_threshold: Optional[float] = 0.82,
+ score_threshold: float = 0.82,
):
if name is None:
name = os.getenv("MISTRAL_MODEL_NAME", "mistral-embed")
--
GitLab