huggingface-api-haystack 2.0.0__tar.gz → 2.1.0__tar.gz

{huggingface_api_haystack-2.0.0 → huggingface_api_haystack-2.1.0}/CHANGELOG.md

@@ -1,5 +1,16 @@
 # Changelog
 
+## [integrations/huggingface_api-v2.0.0] - 2026-10-01
+
+### 🚀 Features
+
+- [**breaking**] Add grpc support for the huggingface TEI components; rename HuggingFace API Sparse Embedders -> Hugging Face TEI Sparse Embedders (#3870)
+
+### 🧪 Testing
+
+- HF API - use Groq for gpt-oss-20b (#4001)
+
+
 ## [integrations/huggingface_api-v1.0.0] - 2026-09-07
 
 ### 🚀 Features

{huggingface_api_haystack-2.0.0 → huggingface_api_haystack-2.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: huggingface-api-haystack
-Version: 2.0.0
+Version: 2.1.0
 Summary: Haystack integration for Hugging Face API
 Project-URL: Documentation, https://github.com/deepset-ai/haystack-core-integrations/tree/main/integrations/huggingface_api#readme
 Project-URL: Issues, https://github.com/deepset-ai/haystack-core-integrations/issues

{huggingface_api_haystack-2.0.0 → huggingface_api_haystack-2.1.0}/src/haystack_integrations/components/generators/huggingface_api/chat/chat_generator.py

@@ -27,7 +27,6 @@ from haystack.tools import (
     deserialize_tools_or_toolset_inplace,
     flatten_tools_or_toolsets,
     serialize_tools_or_toolset,
-    warm_up_tools,
 )
 from haystack.utils import Secret, deserialize_callable, serialize_callable
 from haystack.utils.hf import convert_message_to_hf_format
@@ -442,7 +441,6 @@ class HuggingFaceAPIChatGenerator:
         self._model_or_url = model_or_url
         self._client: InferenceClient | None = None
         self._async_client: AsyncInferenceClient | None = None
-        self._tools_warmed_up = False
 
     def _client_kwargs(self) -> dict[str, Any]:
         """Build the keyword arguments used to create Hugging Face clients."""
@@ -452,26 +450,15 @@ class HuggingFaceAPIChatGenerator:
             **{k: v for k, v in self.api_params.items() if k not in ("model", "url")},
         }
 
-    def _warm_up_tools(self) -> None:
-        if not self._tools_warmed_up:
-            warm_up_tools(self.tools)
-            self._tools_warmed_up = True
-
     def warm_up(self) -> None:
-        """
-        Warm up the Hugging Face API chat generator.
-
-        This creates the synchronous client and warms up the configured tools.
-        """
-        self._warm_up_tools()
+        """Create the synchronous Hugging Face client."""
         if self._client is None:
             if self.api_type == HFGenerationAPIType.SERVERLESS_INFERENCE_API:
                 _check_valid_model(self._model_or_url, HFModelType.GENERATION, self.token)
             self._client = InferenceClient(**self._client_kwargs())
 
     async def warm_up_async(self) -> None:
-        """Create the asynchronous Hugging Face client and warm up the configured tools."""
-        self._warm_up_tools()
+        """Create the asynchronous Hugging Face client."""
         if self._async_client is None:
             if self.api_type == HFGenerationAPIType.SERVERLESS_INFERENCE_API:
                 await _check_valid_model_async(self._model_or_url, HFModelType.GENERATION, self.token)

{huggingface_api_haystack-2.0.0 → huggingface_api_haystack-2.1.0}/tests/test_chat_generator.py

@@ -1334,115 +1334,6 @@ class TestTools:
         assert hf_tools[0].function.description == "useful to determine the weather in a given location"
         assert hf_tools[0].function.parameters == {"city": {"type": "string"}}
 
-    def test_warm_up_with_tools(self, mock_check_valid_model):
-        """Test that warm_up() calls warm_up on tools and is idempotent."""
-
-        # Create a mock tool that tracks if warm_up() was called
-        class MockTool(Tool):
-            warm_up_call_count = 0  # Class variable to track calls
-
-            def __init__(self):
-                super().__init__(
-                    name="mock_tool",
-                    description="A mock tool for testing",
-                    parameters={"x": {"type": "string"}},
-                    function=lambda x: x,
-                )
-
-            def warm_up(self):
-                MockTool.warm_up_call_count += 1
-
-        # Reset the class variable before test
-        MockTool.warm_up_call_count = 0
-        mock_tool = MockTool()
-
-        # Create HuggingFaceAPIChatGenerator with the mock tool
-        component = HuggingFaceAPIChatGenerator(
-            api_type=HFGenerationAPIType.SERVERLESS_INFERENCE_API,
-            api_params={"model": "HuggingFaceH4/zephyr-7b-alpha"},
-            tools=[mock_tool],
-        )
-
-        # Verify initial state - warm_up not called yet
-        assert MockTool.warm_up_call_count == 0
-        assert not component._tools_warmed_up
-
-        # Call warm_up() on the generator
-        component.warm_up()
-
-        # Assert that the tool's warm_up() was called
-        assert MockTool.warm_up_call_count == 1
-        assert component._tools_warmed_up
-
-        # Call warm_up() again and verify it's idempotent (only warms up once)
-        component.warm_up()
-
-        # The tool's warm_up should still only have been called once
-        assert MockTool.warm_up_call_count == 1
-        assert component._tools_warmed_up
-
-    def test_warm_up_with_no_tools(self, mock_check_valid_model):
-        """Test that warm_up() works when no tools are provided."""
-        component = HuggingFaceAPIChatGenerator(
-            api_type=HFGenerationAPIType.SERVERLESS_INFERENCE_API, api_params={"model": "HuggingFaceH4/zephyr-7b-alpha"}
-        )
-
-        # Verify initial state
-        assert not component._tools_warmed_up
-        assert component.tools is None
-
-        # Call warm_up() - should not raise an error
-        component.warm_up()
-
-        # Verify the component is warmed up
-        assert component._tools_warmed_up
-
-        # Call warm_up() again - should be idempotent
-        component.warm_up()
-        assert component._tools_warmed_up
-
-    def test_warm_up_with_multiple_tools(self, mock_check_valid_model):
-        """Test that warm_up() works with multiple tools."""
-        # Track warm_up calls
-        warm_up_calls = []
-
-        class MockTool(Tool):
-            def __init__(self, tool_name):
-                super().__init__(
-                    name=tool_name,
-                    description=f"Mock tool {tool_name}",
-                    parameters={"type": "object", "properties": {"x": {"type": "string"}}, "required": ["x"]},
-                    function=lambda x: f"{tool_name} result: {x}",
-                )
-
-            def warm_up(self):
-                warm_up_calls.append(self.name)
-
-        mock_tool1 = MockTool("tool1")
-        mock_tool2 = MockTool("tool2")
-
-        # Use a LIST of tools, not a Toolset
-        component = HuggingFaceAPIChatGenerator(
-            api_type=HFGenerationAPIType.SERVERLESS_INFERENCE_API,
-            api_params={"model": "HuggingFaceH4/zephyr-7b-alpha"},
-            tools=[mock_tool1, mock_tool2],
-        )
-
-        # Call warm_up()
-        component.warm_up()
-
-        # Assert that both tools' warm_up() were called
-        assert "tool1" in warm_up_calls
-        assert "tool2" in warm_up_calls
-        assert component._tools_warmed_up
-
-        # Track count
-        call_count = len(warm_up_calls)
-
-        # Verify idempotency
-        component.warm_up()
-        assert len(warm_up_calls) == call_count
-
 
 class TestReasoning:
     def test_run_with_reasoning_non_streaming(self, mock_check_valid_model, chat_messages):