From 3bf5c22ccc4b7b8582f386fe052a78ea24c48da9 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sat, 13 Jan 2024 01:50:53 +0000
Subject: [PATCH] adapted old typing style
---
semantic_router/encoders/zure.py | 25 +++++++++++++------------
1 file changed, 13 insertions(+), 12 deletions(-)
diff --git a/semantic_router/encoders/zure.py b/semantic_router/encoders/zure.py
index 1ac78392..8a558d0d 100644
--- a/semantic_router/encoders/zure.py
+++ b/semantic_router/encoders/zure.py
@@ -1,5 +1,6 @@
import os
from time import sleep
+from typing import List, Optional
import openai
from openai import OpenAIError
@@ -10,21 +11,21 @@ from semantic_router.utils.logger import logger
class AzureOpenAIEncoder(BaseEncoder):
- client: openai.AzureOpenAI | None = None
+ client: Optional[openai.AzureOpenAI] = None
type: str = "azure"
- api_key: str | None = None
- deployment_name: str | None = None
- azure_endpoint: str | None = None
- api_version: str | None = None
- model: str | None = None
+ api_key: Optional[str] = None
+ deployment_name: Optional[str] = None
+ azure_endpoint: Optional[str] = None
+ api_version: Optional[str] = None
+ model: Optional[str] = None
def __init__(
self,
- api_key: str | None = None,
- deployment_name: str | None = None,
- azure_endpoint: str | None = None,
- api_version: str | None = None,
- model: str | None = None,
+ api_key: Optional[str] = None,
+ deployment_name: Optional[str] = None,
+ azure_endpoint: Optional[str] = None,
+ api_version: Optional[str] = None,
+ model: Optional[str] = None,
score_threshold: float = 0.82,
):
name = deployment_name
@@ -75,7 +76,7 @@ class AzureOpenAIEncoder(BaseEncoder):
except Exception as e:
raise ValueError(f"OpenAI API client failed to initialize. Error: {e}")
- def __call__(self, docs: list[str]) -> list[list[float]]:
+ def __call__(self, docs: List[str]) -> List[List[float]]:
if self.client is None:
raise ValueError("OpenAI client is not initialized.")
embeds = None
--
GitLab