fixing imports

deepset-ai · Mar 4, 2025 · 1fc0d8f · 1fc0d8f
1 parent 25b1113
commit 1fc0d8f
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/...nthropic/src/haystack_integrations/components/generators/anthropic/chat/chat_generator.py b/...nthropic/src/haystack_integrations/components/generators/anthropic/chat/chat_generator.py
@@ -6,7 +6,7 @@
 from haystack.tools import Tool, _check_duplicate_tool_names, deserialize_tools_inplace
 from haystack.utils import Secret, deserialize_callable, deserialize_secrets_inplace, serialize_callable
 
-from anthropic import Anthropic, AsyncAnthropic, Stream, AsyncStream
+from anthropic import Anthropic, AsyncAnthropic, AsyncStream, Stream
 
 logger = logging.getLogger(__name__)
 

diff --git a/integrations/anthropic/tests/test_chat_generator.py b/integrations/anthropic/tests/test_chat_generator.py
@@ -13,9 +13,7 @@
 from anthropic.types import (
     ContentBlockDeltaEvent,
     ContentBlockStartEvent,
-    ContentBlockStopEvent,
     Message,
-    MessageDeltaEvent,
     MessageStartEvent,
     TextBlockParam,
     TextDelta,
@@ -1173,8 +1171,6 @@ async def test_run_async_with_params(self, chat_messages, mock_anthropic_complet
     #     assert len(final_message.text) > 0
     #     assert "paris" in final_message.text.lower()
 
-
-
     @pytest.mark.asyncio
     @pytest.mark.skipif(
         not os.environ.get("ANTHROPIC_API_KEY", None),
@@ -1317,4 +1313,3 @@ async def test_live_run_async_with_tools(self, tools):
     #     assert "22°" in message.text
     #     assert "12°" in message.text
     #     assert message.meta["finish_reason"] == "end_turn"
-