Skip to content
Snippets Groups Projects
Commit 1153a6d1 authored by dwmorris11's avatar dwmorris11
Browse files

changed score_threshold to be float only

parent 4a2bbb7c
No related branches found
No related tags found
No related merge requests found
...@@ -20,7 +20,7 @@ class MistralEncoder(BaseEncoder): ...@@ -20,7 +20,7 @@ class MistralEncoder(BaseEncoder):
self, self,
name: Optional[str] = None, name: Optional[str] = None,
mistral_api_key: Optional[str] = None, mistral_api_key: Optional[str] = None,
score_threshold: Optional[float] = 0.82, score_threshold: float = 0.82,
): ):
if name is None: if name is None:
name = os.getenv("MISTRAL_MODEL_NAME", "mistral-embed") name = os.getenv("MISTRAL_MODEL_NAME", "mistral-embed")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment