Browse Source

fix: azure embedding model credentials include base_model_name is invalid for openai sdk (#917)

takatost 1 year ago
parent
commit
1c552ff23a

+ 2 - 1
api/core/model_providers/models/embedding/azure_openai_embedding.py

@@ -26,7 +26,8 @@ class AzureOpenAIEmbedding(BaseEmbedding):
             openai_api_version=AZURE_OPENAI_API_VERSION,
             chunk_size=16,
             max_retries=1,
-            **self.credentials
+            openai_api_key=self.credentials.get('openai_api_key'),
+            openai_api_base=self.credentials.get('openai_api_base')
         )
 
         super().__init__(model_provider, client, name)