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):
azure_endpoint: str | None = None,
api_version: str | None = None,
model: str | None = None,
score_threshold: float = 0.82,
):
name = deployment_name
if name is None:
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.deployment_name = deployment_name
self.azure_endpoint = azure_endpoint
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment