Formatted files to pass Black checks

Author: Mohamed Shahin

weaviate_cli/commands/batch.py

@@ -5,13 +5,15 @@
 from weaviate_cli.utils import get_client_from_context
 from weaviate.exceptions import WeaviateConnectionError
 from weaviate_cli.defaults import CreateCollectionDefaults
-from weaviate_cli.managers.batch_manager import BatchManager 
+from weaviate_cli.managers.batch_manager import BatchManager
+
 
 @click.group()
 def batch() -> None:
     """Batch operations in Weaviate."""
     pass
 
+
 @batch.command("insert")
 @click.option(
     "--collection",
@@ -65,7 +67,9 @@ def batch_insert_cli(ctx, collection, path, vectorizer, shards, replication_fact
 
     # Validate JSON structure
     if not isinstance(data, list) or not all(isinstance(obj, dict) for obj in data):
-        click.echo("Error: The JSON file must contain a list of objects (e.g., [{...}, {...}]).")
+        click.echo(
+            "Error: The JSON file must contain a list of objects (e.g., [{...}, {...}])."
+        )
         sys.exit(1)
 
     # Initialize the Weaviate client
@@ -96,4 +100,4 @@ def batch_insert_cli(ctx, collection, path, vectorizer, shards, replication_fact
         sys.exit(1)
     finally:
         if client:
-            client.close()
+            client.close()

weaviate_cli/managers/batch_manager.py

@@ -62,7 +62,9 @@ def create_collection(
                 vectorizer_config=vectorizer_map[vectorizer],
                 properties=None if force_auto_schema else [],
             )
-            click.echo(f"Collection '{collection}' created successfully with vectorizer '{vectorizer}'.")
+            click.echo(
+                f"Collection '{collection}' created successfully with vectorizer '{vectorizer}'."
+            )
         except Exception as e:
             raise Exception(f"Error creating collection '{collection}': {e}")
 
@@ -79,7 +81,9 @@ def batch_insert(
             data (List[Dict]): Data to be inserted.
         """
         if not self.client.collections.exists(collection):
-            raise Exception(f"Collection '{collection}' does not exist. Cannot insert data.")
+            raise Exception(
+                f"Collection '{collection}' does not exist. Cannot insert data."
+            )
 
         try:
             # Perform batch insertion using Weaviate's dynamic batch
@@ -94,7 +98,9 @@ def batch_insert(
                         collection=collection,
                         properties=record,
                     )
-                    click.echo(f"Processed record") # add '{record}' <- if you would like to see the record being processed
+                    click.echo(
+                        f"Processed record"
+                    )  # add '{record}' <- if you would like to see the record being processed
         except Exception as e:
             raise Exception(f"Batch insertion failed: {e}")
 
@@ -105,4 +111,4 @@ def batch_insert(
             for i, failed_obj in enumerate(failed_objects, 1):
                 click.echo(f"Failed object {i}: {failed_obj}")
         else:
-            click.echo(f"All objects successfully inserted into '{collection}'.")
+            click.echo(f"All objects successfully inserted into '{collection}'.")