From fb78955154fd04425992810b616e82c5266ddb87 Mon Sep 17 00:00:00 2001
From: Ismail Ashraq <ismailashraq@Ismails-MacBook-Pro.local>
Date: Mon, 19 Aug 2024 15:43:28 +0500
Subject: [PATCH] fix test error
---
semantic_router/encoders/openai.py | 2 +-
semantic_router/encoders/zure.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/semantic_router/encoders/openai.py b/semantic_router/encoders/openai.py
index ec39e828..4bc86ac2 100644
--- a/semantic_router/encoders/openai.py
+++ b/semantic_router/encoders/openai.py
@@ -52,7 +52,7 @@ class OpenAIEncoder(BaseEncoder):
openai_org_id: Optional[str] = None,
score_threshold: Optional[float] = None,
dimensions: Union[int, NotGiven] = NotGiven(),
- max_retries: int | None = None,
+ max_retries: int = 3,
):
if name is None:
name = EncoderDefault.OPENAI.value["embedding_model"]
diff --git a/semantic_router/encoders/zure.py b/semantic_router/encoders/zure.py
index 4caded0f..d6f65660 100644
--- a/semantic_router/encoders/zure.py
+++ b/semantic_router/encoders/zure.py
@@ -34,7 +34,7 @@ class AzureOpenAIEncoder(BaseEncoder):
model: Optional[str] = None, # TODO we should change to `name` JB
score_threshold: float = 0.82,
dimensions: Union[int, NotGiven] = NotGiven(),
- max_retries: int | None = None,
+ max_retries: int = 3,
):
name = deployment_name
if name is None:
--
GitLab