From 1c552ff23aa7ba738b31cb5186689cd039d13a5d Mon Sep 17 00:00:00 2001 From: takatost Date: Sat, 19 Aug 2023 16:24:18 +0800 Subject: [PATCH] fix: azure embedding model credentials include base_model_name is invalid for openai sdk (#917) --- .../model_providers/models/embedding/azure_openai_embedding.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/api/core/model_providers/models/embedding/azure_openai_embedding.py b/api/core/model_providers/models/embedding/azure_openai_embedding.py index 127bf640ee..1b9149392b 100644 --- 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)