More linting fixes

integrations/cohere/src/haystack_integrations/components/embedders/document_embedder.py

@@ -3,10 +3,11 @@
 # SPDX-License-Identifier: Apache-2.0
 import asyncio
 import os
+from typing import Any, Dict, List, Optional
+
 from cohere import COHERE_API_URL, AsyncClient, Client
 from haystack import Document, component, default_to_dict
 from haystack_integrations.components.embedders.utils import get_async_response, get_response
-from typing import Any, Dict, List, Optional
 
 
 @component

integrations/cohere/src/haystack_integrations/components/embedders/text_embedder.py

@@ -3,10 +3,11 @@
 # SPDX-License-Identifier: Apache-2.0
 import asyncio
 import os
+from typing import Any, Dict, List, Optional
+
 from cohere import COHERE_API_URL, AsyncClient, Client
 from haystack import component, default_to_dict
 from haystack_integrations.components.embedders.utils import get_async_response, get_response
-from typing import Any, Dict, List, Optional
 
 
 @component

integrations/cohere/src/haystack_integrations/components/generators/chat/chat_generator.py

@@ -1,10 +1,11 @@
 import logging
 import os
+from typing import Any, Callable, Dict, List, Optional
+
 from haystack import component, default_from_dict, default_to_dict
 from haystack.components.generators.utils import deserialize_callback_handler, serialize_callback_handler
 from haystack.dataclasses import ChatMessage, ChatRole, StreamingChunk
 from haystack.lazy_imports import LazyImport
-from typing import Any, Callable, Dict, List, Optional
 
 with LazyImport(message="Run 'pip install cohere'") as cohere_import:
     import cohere

integrations/cohere/src/haystack_integrations/components/generators/generator.py

@@ -4,10 +4,11 @@
 import logging
 import os
 import sys
+from typing import Any, Callable, Dict, List, Optional, cast
+
 from cohere import COHERE_API_URL, Client
 from cohere.responses import Generations
 from haystack import DeserializationError, component, default_from_dict, default_to_dict
-from typing import Any, Callable, Dict, List, Optional, cast
 
 logger = logging.getLogger(__name__)
 

integrations/cohere/tests/test_cohere_chat_generator.py

@@ -5,7 +5,6 @@
 import pytest
 from haystack.components.generators.utils import default_streaming_callback
 from haystack.dataclasses import ChatMessage, ChatRole, StreamingChunk
-
 from haystack_integrations.components.generators.chat import CohereChatGenerator
 
 pytestmark = pytest.mark.chat_generators

integrations/cohere/tests/test_cohere_generators.py

@@ -5,7 +5,6 @@
 
 import pytest
 from cohere import COHERE_API_URL
-
 from haystack_integrations.components.generators import CohereGenerator
 
 pytestmark = pytest.mark.generators

integrations/cohere/tests/test_document_embedder.py

@@ -6,7 +6,6 @@
 import pytest
 from cohere import COHERE_API_URL
 from haystack import Document
-
 from haystack_integrations.components.embedders import CohereDocumentEmbedder
 
 pytestmark = pytest.mark.embedders

integrations/cohere/tests/test_text_embedder.py

@@ -5,7 +5,6 @@
 
 import pytest
 from cohere import COHERE_API_URL
-
 from haystack_integrations.components.embedders import CohereTextEmbedder
 
 pytestmark = pytest.mark.embedders