fix: azure embedding params error

f2095e0e · John Wang · bfe00db8 · f2095e0e
Commit f2095e0e authored Jun 23, 2023 by John Wang
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

azure_provider.py api/core/llm/provider/azure_provider.py +4 -1

No files found.
--- a/api/core/llm/provider/azure_provider.py
+++ b/api/core/llm/provider/azure_provider.py
@@ -42,7 +42,10 @@ class AzureProvider(BaseProvider):
        """
        config = self.get_provider_api_key(model_id=model_id)
        config['openai_api_type'] = 'azure'
-        config['deployment_name'] = model_id.replace('.', '') if model_id else None
+        if model_id == 'text-embedding-ada-002':
+            config['deployment'] = model_id.replace('.', '') if model_id else None
+        else:
+            config['deployment_name'] = model_id.replace('.', '') if model_id else None
        return config

    def get_provider_name(self):