try fixing azure chat generator

deepset-ai · Dec 20, 2024 · 298062d · 298062d
1 parent 110eefd
commit 298062d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/haystack/components/generators/chat/azure.py b/haystack/components/generators/chat/azure.py
@@ -142,6 +142,11 @@ def __init__(  # pylint: disable=too-many-positional-arguments
         self.max_retries = max_retries or int(os.environ.get("OPENAI_MAX_RETRIES", 5))
         self.default_headers = default_headers or {}
 
+        # This ChatGenerator does not yet supports tools. The following workaround ensures that we do not
+        # get an error when invoking the run method of the parent class (OpenAIChatGenerator).
+        self.tools = None
+        self.tools_strict = False
+
         self.client = AzureOpenAI(
             api_version=api_version,
             azure_endpoint=azure_endpoint,

diff --git a/haystack/components/generators/chat/openai.py b/haystack/components/generators/chat/openai.py
@@ -289,10 +289,7 @@ def _prepare_api_call(  # noqa: PLR0913
         openai_tools = None
         if tools:
             openai_tools = [
-                {
-                    "type": "function",
-                    "function": {**t.tool_spec, **({"strict": tools_strict} if tools_strict is not None else {})},
-                }
+                {"type": "function", "function": {**t.tool_spec, **({"strict": tools_strict} if tools_strict else {})}}
                 for t in tools
             ]