opt

omagent_core/models/encoders/azure_encoder.py

@@ -21,8 +21,8 @@ class AzureTextEmbeddingV3(EncoderBase):
 
     def __init__(self, /, **data: Any) -> None:
         super().__init__(**data)
-        self.client = AzureOpenAI(base_url=self.endpoint, api_key=self.api_key, api_version=self.api_version)
-        self.aclient = AsyncAzureOpenAI(base_url=self.endpoint, api_key=self.api_key, api_version=self.api_version)
+        self.client = AzureOpenAI(azure_endpoint=self.endpoint, api_key=self.api_key, api_version=self.api_version)
+        self.aclient = AsyncAzureOpenAI(azure_endpoint=self.endpoint, api_key=self.api_key, api_version=self.api_version)
 
     def _infer(self, data: List[str], **kwargs) -> List[List[float]]:
         res = self.client.embeddings.create(input=data, model=self.model_id)