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

3 years ago · 1c552ff23a
parent 5163dd38e5
commit 1c552ff23a
1 changed files with 2 additions and 1 deletions
--- a/api/core/model_providers/models/embedding/azure_openai_embedding.py
+++ b/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)