Skip to content
Snippets Groups Projects
Commit d06d95ed authored by James McKeown's avatar James McKeown
Browse files

add score_threshold to init for AzureOpenAIEncoder (newly required to extend BaseEncoder)

parent 22612718
No related branches found
No related tags found
No related merge requests found
...@@ -25,11 +25,12 @@ class AzureOpenAIEncoder(BaseEncoder): ...@@ -25,11 +25,12 @@ class AzureOpenAIEncoder(BaseEncoder):
azure_endpoint: str | None = None, azure_endpoint: str | None = None,
api_version: str | None = None, api_version: str | None = None,
model: str | None = None, model: str | None = None,
score_threshold: float = 0.82,
): ):
name = deployment_name name = deployment_name
if name is None: if name is None:
name = os.getenv("AZURE_OPENAI_DEPLOYMENT_NAME", "text-embedding-ada-002") name = os.getenv("AZURE_OPENAI_DEPLOYMENT_NAME", "text-embedding-ada-002")
super().__init__(name=name) super().__init__(name=name, score_threshold=score_threshold)
self.api_key = api_key self.api_key = api_key
self.deployment_name = deployment_name self.deployment_name = deployment_name
self.azure_endpoint = azure_endpoint self.azure_endpoint = azure_endpoint
......
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