formatter

Author: Federaffo

datapizza-ai-clients/datapizza-ai-clients-google/datapizza/clients/google/google_client.py

@@ -592,7 +592,11 @@ def _response_to_client_response(
                     )
                 )
         else:
-            if hasattr(response, "candidates") and response.candidates and response.candidates[0].content.parts:
+            if (
+                hasattr(response, "candidates")
+                and response.candidates
+                and response.candidates[0].content.parts
+            ):
                 for part in response.candidates[0].content.parts:
                     # Handle inline_data (images from generation or code execution)
                     if hasattr(part, "inline_data") and part.inline_data is not None:
@@ -602,8 +606,8 @@ def _response_to_client_response(
                             source=base64.b64encode(part.inline_data.data).decode(
                                 "utf-8"
                             ),
-                            extension=(part.inline_data.mime_type.split("/")[-1]) 
-                            if part.inline_data.mime_type 
+                            extension=(part.inline_data.mime_type.split("/")[-1])
+                            if part.inline_data.mime_type
                             else "png",
                         )
                         blocks.append(MediaBlock(media=media))

datapizza-ai-core/datapizza/core/embedder/base.py

@@ -1,12 +1,12 @@
 from abc import abstractmethod
+from typing import Generic, TypeVar
 
 from datapizza.core.models import PipelineComponent
-from datapizza.type import SparseEmbedding
-from typing import TypeVar, Generic
 
-T = TypeVar('T')
+T = TypeVar("T")
 
-class BaseEmbedder(PipelineComponent,Generic[T]):
+
+class BaseEmbedder(PipelineComponent, Generic[T]):
     client: object
     a_client: object