Fix: AzureOpenAIInferenceEngine fails if api_version is not set (#1680)

* Fix: AzureOpenAIInferenceEngine fails if api_version is not set

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

* Add HFOptionSelectingInferenceEngine get_engine_id method

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

---------

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

Author: martinscooper

src/unitxt/inference.py

@@ -1774,7 +1774,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
@@ -1783,6 +1783,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(),
         )
 
@@ -3294,14 +3295,18 @@ class HFOptionSelectingInferenceEngine(InferenceEngine, TorchDeviceMixin):
 
     This class uses models from the HuggingFace Transformers library to calculate log probabilities for text inputs.
     """
-
+    label = "hf_option_selection"
     model_name: str
     batch_size: int
 
     _requirements_list = {
         "transformers": "Install huggingface package using 'pip install --upgrade transformers"
     }
 
+    def get_engine_id(self):
+        return get_model_and_label_id(self.model, self.label)
+
+
     def prepare_engine(self):
         from transformers import AutoModelForCausalLM, AutoTokenizer