opt

configs/llms/text_encoder.yml

@@ -1,3 +1,4 @@
 name: AzureTextEmbeddingV3
 endpoint: ${env| custom_openai_endpoint, https://api.openai.com/v1}
-api_key: ${env| custom_openai_key, openai_api_key}
+api_key: ${env| custom_openai_key, openai_api_key}
+api_version: ${env| custom_openai_api_version, 2024-02-15-preview} # this is mandatory

omagent_core/models/encoders/azure_encoder.py

@@ -11,7 +11,8 @@ class AzureTextEmbeddingV3(EncoderBase):
     model_id: str = "text-embedding-3-large"
     api_key: str
     dim: int = 3072
-
+    api_version: str
+    
     class Config:
         """Configuration for this pydantic object."""
 
@@ -20,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)
-        self.aclient = AsyncAzureOpenAI(base_url=self.endpoint, api_key=self.api_key)
+        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)
 
     def _infer(self, data: List[str], **kwargs) -> List[List[float]]:
         res = self.client.embeddings.create(input=data, model=self.model_id)

webpage_configs/llms/text_encoder.yml

@@ -1,3 +1,4 @@
 name: AzureTextEmbeddingV3
 endpoint: ${env| custom_openai_endpoint, https://api.openai.com/v1}
-api_key: ${env| custom_openai_key, openai_api_key}
+api_key: ${env| custom_openai_key, openai_api_key}
+api_version: ${env| custom_openai_api_version, 2024-02-15-preview} # this is mandatory