From c3715f0d36a45ba64d62612aec50b34efcd12b5d Mon Sep 17 00:00:00 2001 From: LETREMBLE Emmanuel Date: Tue, 4 Mar 2025 12:58:00 +0100 Subject: [PATCH] fix: update Azure Provider file to consider both openai and non-openai models --- aisuite/providers/azure_provider.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/aisuite/providers/azure_provider.py b/aisuite/providers/azure_provider.py index bd390e83..bfea78dd 100644 --- a/aisuite/providers/azure_provider.py +++ b/aisuite/providers/azure_provider.py @@ -92,7 +92,13 @@ def __init__(self, **config): self.transformer = AzureMessageConverter() def chat_completions_create(self, model, messages, **kwargs): - url = f"{self.base_url}/chat/completions" + + if ".openai.azure.com" in self.base_url: + url = f"{self.base_url}/openai/deployments/{model}/chat/completions" + headers = {"Content-Type": "application/json", "api-key": self.api_key} + else: + url = f"{self.base_url}/chat/completions" + headers = {"Content-Type": "application/json", "Authorization": self.api_key} if self.api_version: url = f"{url}?api-version={self.api_version}" @@ -120,7 +126,6 @@ def chat_completions_create(self, model, messages, **kwargs): data.update(kwargs) body = json.dumps(data).encode("utf-8") - headers = {"Content-Type": "application/json", "Authorization": self.api_key} try: req = urllib.request.Request(url, body, headers)