Skip to content
Snippets Groups Projects
Commit fb789551 authored by Ismail Ashraq's avatar Ismail Ashraq
Browse files

fix test error

parent 8016fa11
No related branches found
No related tags found
No related merge requests found
...@@ -52,7 +52,7 @@ class OpenAIEncoder(BaseEncoder): ...@@ -52,7 +52,7 @@ class OpenAIEncoder(BaseEncoder):
openai_org_id: Optional[str] = None, openai_org_id: Optional[str] = None,
score_threshold: Optional[float] = None, score_threshold: Optional[float] = None,
dimensions: Union[int, NotGiven] = NotGiven(), dimensions: Union[int, NotGiven] = NotGiven(),
max_retries: int | None = None, max_retries: int = 3,
): ):
if name is None: if name is None:
name = EncoderDefault.OPENAI.value["embedding_model"] name = EncoderDefault.OPENAI.value["embedding_model"]
......
...@@ -34,7 +34,7 @@ class AzureOpenAIEncoder(BaseEncoder): ...@@ -34,7 +34,7 @@ class AzureOpenAIEncoder(BaseEncoder):
model: Optional[str] = None, # TODO we should change to `name` JB model: Optional[str] = None, # TODO we should change to `name` JB
score_threshold: float = 0.82, score_threshold: float = 0.82,
dimensions: Union[int, NotGiven] = NotGiven(), dimensions: Union[int, NotGiven] = NotGiven(),
max_retries: int | None = None, max_retries: int = 3,
): ):
name = deployment_name name = deployment_name
if name is None: if name is None:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment