Fix: AzureOpenAIInferenceEngine fails if api_version is not set

Signed-off-by: Martín Santillán Cooper <[email protected]>

Author: martinscooper

src/unitxt/inference.py

@@ -1771,7 +1771,7 @@ def _prepare_credentials(self) -> CredentialsOpenAi:
         ), "Error while trying to run AzureOpenAIInferenceEngine: Missing environment variable param AZURE_OPENAI_HOST or OPENAI_API_VERSION"
         api_url = f"{azure_openapi_host}/openai/deployments/{self.model_name}/chat/completions?api-version={api_version}"
 
-        return {"api_key": api_key, "api_url": api_url}
+        return {"api_key": api_key, "api_url": api_url, "api_version": api_version}
 
     def create_client(self):
         from openai import AzureOpenAI
@@ -1780,6 +1780,7 @@ def create_client(self):
         return AzureOpenAI(
             api_key=self.credentials["api_key"],
             base_url=self.credentials["api_url"],
+            api_version=self.credentials["api_version"],
             default_headers=self.get_default_headers(),
         )