add azure_deployment

Canner · Jul 9, 2024 · 81a64b6 · 81a64b6
1 parent ae657bb
commit 81a64b6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/wren-ai-service/src/providers/embedder/azure_openai.py b/wren-ai-service/src/providers/embedder/azure_openai.py
@@ -49,6 +49,7 @@ def __init__(
 
         self.client = AsyncAzureOpenAI(
             azure_endpoint=api_base_url,
+            azure_deployment=model,
             api_version=api_version,
             api_key=api_key.resolve_value(),
         )