weaviate-cli 3.0.1__tar.gz → 3.0.2__tar.gz

{weaviate_cli-3.0.1 → weaviate_cli-3.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: weaviate-cli
-Version: 3.0.1
+Version: 3.0.2
 Summary: Command line interface to interact with weaviate
 Home-page: https://github.com/weaviate/weaviate-cli
 Download-URL: https://github.com/weaviate/weaviate-cli
@@ -125,7 +125,12 @@ Here you can see an example on how the configuration file should look like if yo
      "auth": {
          "type": "api_key",
          "api_key": "jfeRFsdfRfSasgsDoNOtTrYToUsErRQwqqdZfghasd"
-     }
+     },
+    "headers":{
+        "X-OpenAI-Api-Key":"OPEN_AI_KEY",
+        "X-Cohere-Api-Key":"Cohere_AI_KEY",
+        "X-JinaAI-Api-Key":"JINA_AI_KEY"
+        }
  }
 ```
 
@@ -138,6 +143,15 @@ Here you can see an example on how the configuration file should look like if yo
 
 Detailed documentation will be added soon.
 
+## Supported Model Provider
+
+- Contextionary
+- Transformers
+- OpenAI
+- Ollama
+- Cohere
+- JinaAI
+
 ## Community & Support
 
 - [Slack Community](https://weaviate.io/slack) - Join our active community

{weaviate_cli-3.0.1 → weaviate_cli-3.0.2}/README.md

@@ -89,7 +89,12 @@ Here you can see an example on how the configuration file should look like if yo
      "auth": {
          "type": "api_key",
          "api_key": "jfeRFsdfRfSasgsDoNOtTrYToUsErRQwqqdZfghasd"
-     }
+     },
+    "headers":{
+        "X-OpenAI-Api-Key":"OPEN_AI_KEY",
+        "X-Cohere-Api-Key":"Cohere_AI_KEY",
+        "X-JinaAI-Api-Key":"JINA_AI_KEY"
+        }
  }
 ```
 
@@ -102,6 +107,15 @@ Here you can see an example on how the configuration file should look like if yo
 
 Detailed documentation will be added soon.
 
+## Supported Model Provider
+
+- Contextionary
+- Transformers
+- OpenAI
+- Ollama
+- Cohere
+- JinaAI
+
 ## Community & Support
 
 - [Slack Community](https://weaviate.io/slack) - Join our active community

{weaviate_cli-3.0.1 → weaviate_cli-3.0.2}/weaviate_cli/commands/create.py

@@ -91,13 +91,17 @@ def create() -> None:
 @click.option(
     "--vectorizer",
     default=CreateCollectionDefaults.vectorizer,
-    type=click.Choice(["contextionary", "transformers", "openai", "ollama"]),
+    type=click.Choice(
+        ["contextionary", "transformers", "openai", "ollama", "cohere", "jinaai"]
+    ),
     help="Vectorizer to use.",
 )
 @click.option(
     "--replication_deletion_strategy",
     default=CreateCollectionDefaults.replication_deletion_strategy,
-    type=click.Choice(["delete_on_conflict", "no_automated_resolution"]),
+    type=click.Choice(
+        ["delete_on_conflict", "no_automated_resolution", "time_based_resolution"]
+    ),
     help="Replication deletion strategy (default: 'delete_on_conflict').",
 )
 @click.pass_context
@@ -281,6 +285,11 @@ def create_backup_cli(ctx, backend, backup_id, include, exclude, wait, cpu_for_b
     default=CreateDataDefaults.vector_dimensions,
     help="Number of vector dimensions to be used when the data is randomized.",
 )
+@click.option(
+    "--uuid",
+    default=None,
+    help="UUID of the object to be used when the data is randomized. It requires --limit=1 and --randomize to be enabled.",
+)
 @click.pass_context
 def create_data_cli(
     ctx,
@@ -290,6 +299,7 @@ def create_data_cli(
     randomize,
     auto_tenants,
     vector_dimensions,
+    uuid,
 ):
     """Ingest data into a collection in Weaviate."""
 
@@ -299,6 +309,14 @@ def create_data_cli(
         )
         sys.exit(1)
 
+    if uuid is not None and not randomize:
+        click.echo("Error: --uuid has no effect unless --randomize is enabled.")
+        sys.exit(1)
+
+    if uuid is not None and limit != 1:
+        click.echo("Error: --uuid has no effect unless --limit=1 is enabled.")
+        sys.exit(1)
+
     client = None
     try:
         client = get_client_from_context(ctx)
@@ -311,6 +329,7 @@ def create_data_cli(
             randomize=randomize,
             auto_tenants=auto_tenants,
             vector_dimensions=vector_dimensions,
+            uuid=uuid,
         )
     except Exception as e:
         click.echo(f"Error: {e}")

{weaviate_cli-3.0.1 → weaviate_cli-3.0.2}/weaviate_cli/commands/get.py

@@ -51,9 +51,13 @@ def get_collection_cli(ctx, collection):
     default=GetTenantsDefaults.collection,
     help="The name of the collection to get tenants from.",
 )
+@click.option(
+    "--tenant_id",
+    help="The tenant ID to get.",
+)
 @click.option("--verbose", is_flag=True, help="Print verbose output.")
 @click.pass_context
-def get_tenants_cli(ctx, collection, verbose):
+def get_tenants_cli(ctx, collection, tenant_id, verbose):
     """Get tenants from a collection in Weaviate."""
 
     client = None
@@ -62,6 +66,7 @@ def get_tenants_cli(ctx, collection, verbose):
         tenant_manager = TenantManager(client)
         tenant_manager.get_tenants(
             collection=collection,
+            tenant_id=tenant_id,
             verbose=verbose,
         )
     except Exception as e:

{weaviate_cli-3.0.1 → weaviate_cli-3.0.2}/weaviate_cli/commands/update.py

@@ -65,7 +65,9 @@ def update() -> None:
 @click.option(
     "--replication_deletion_strategy",
     default=UpdateCollectionDefaults.replication_deletion_strategy,
-    type=click.Choice(["delete_on_conflict", "no_automated_resolution"]),
+    type=click.Choice(
+        ["delete_on_conflict", "no_automated_resolution", "time_based_resolution"]
+    ),
     help="Replication deletion strategy.",
 )
 @click.pass_context

{weaviate_cli-3.0.1 → weaviate_cli-3.0.2}/weaviate_cli/defaults.py

@@ -82,6 +82,7 @@ class GetCollectionDefaults:
 @dataclass
 class GetTenantsDefaults:
     collection: str = "Movies"
+    tenant_id: Optional[str] = None
     verbose: bool = False
 
 

{weaviate_cli-3.0.1 → weaviate_cli-3.0.2}/weaviate_cli/managers/collection_manager.py

@@ -124,6 +124,8 @@ class CollectionManager:
             "ollama": wvc.Configure.Vectorizer.text2vec_ollama(
                 model="snowflake-arctic-embed:33m"
             ),
+            "cohere": wvc.Configure.Vectorizer.text2vec_cohere(),
+            "jinaai": wvc.Configure.Vectorizer.text2vec_jinaai(),
         }
 
         inverted_index_map: Dict[str, wvc.InvertedIndexConfig] = {
@@ -168,6 +170,7 @@ class CollectionManager:
         rds_map = {
             "delete_on_conflict": wvc.ReplicationDeletionStrategy.DELETE_ON_CONFLICT,
             "no_automated_resolution": wvc.ReplicationDeletionStrategy.NO_AUTOMATED_RESOLUTION,
+            "time_based_resolution": wvc.ReplicationDeletionStrategy.TIME_BASED_RESOLUTION,
         }
 
         try:
@@ -258,6 +261,7 @@ class CollectionManager:
         rds_map = {
             "delete_on_conflict": wvc.ReplicationDeletionStrategy.DELETE_ON_CONFLICT,
             "no_automated_resolution": wvc.ReplicationDeletionStrategy.NO_AUTOMATED_RESOLUTION,
+            "time_based_resolution": wvc.ReplicationDeletionStrategy.TIME_BASED_RESOLUTION,
         }
         mt = col_obj.config.get().multi_tenancy_config.enabled
         auto_tenant_creation = (

{weaviate_cli-3.0.1 → weaviate_cli-3.0.2}/weaviate_cli/managers/config_manager.py

@@ -98,8 +98,11 @@ class ConfigManager:
                 port=self.config["http_port"],
                 grpc_port=self.config["grpc_port"],
                 auth_credentials=auth_config,
+                headers=self.config["headers"] if "headers" in self.config else None,
             )
         else:
             return weaviate.connect_to_wcs(
-                cluster_url=self.config["host"], auth_credentials=auth_config
+                cluster_url=self.config["host"],
+                auth_credentials=auth_config,
+                headers=self.config["headers"] if "headers" in self.config else None,
             )

{weaviate_cli-3.0.1 → weaviate_cli-3.0.2}/weaviate_cli/managers/data_manager.py

@@ -145,6 +145,7 @@ class DataManager:
         cl: wvc.ConsistencyLevel,
         randomize: bool,
         vector_dimensions: Optional[int] = 1536,
+        uuid: Optional[str] = None,
     ) -> int:
         if randomize:
             counter = 0
@@ -160,15 +161,16 @@ class DataManager:
                 vector_dimensions = 300
             elif vectorizer == "text2vec-transformers":
                 (
-                    print("Warning: Using vector dimensions: 768")
+                    print("Warning: Using vector dimensions: 384")
                     if vector_dimensions != 1536
                     else None
                 )
-                vector_dimensions = 768
+                vector_dimensions = 384
             with cl_collection.batch.dynamic() as batch:
                 for obj in data_objects:
                     batch.add_object(
                         properties=obj,
+                        uuid=uuid,
                         vector=(
                             2 * np.random.rand(1, vector_dimensions)[0] - 1
                         ).tolist(),
@@ -200,6 +202,7 @@ class DataManager:
         randomize: bool = CreateDataDefaults.randomize,
         auto_tenants: int = CreateDataDefaults.auto_tenants,
         vector_dimensions: Optional[int] = CreateDataDefaults.vector_dimensions,
+        uuid: Optional[str] = None,
     ) -> None:
 
         if not self.client.collections.exists(collection):
@@ -252,6 +255,7 @@ class DataManager:
                     cl_map[consistency_level],
                     randomize,
                     vector_dimensions,
+                    uuid,
                 )
             else:
                 click.echo(f"Processing tenant '{tenant}'")
@@ -261,6 +265,7 @@ class DataManager:
                     cl_map[consistency_level],
                     randomize,
                     vector_dimensions,
+                    uuid,
                 )
 
             if ret == -1:

{weaviate_cli-3.0.1 → weaviate_cli-3.0.2}/weaviate_cli/managers/tenant_manager.py

@@ -184,6 +184,7 @@ class TenantManager:
     def get_tenants(
         self,
         collection: str = GetTenantsDefaults.collection,
+        tenant_id: str = GetTenantsDefaults.tenant_id,
         verbose: bool = GetTenantsDefaults.verbose,
     ) -> dict:
         """
@@ -191,6 +192,7 @@ class TenantManager:
 
         Args:
             collection (str): The name of the collection to retrieve tenants from.
+            tenant_id (str): The ID of the tenant to retrieve. If None, retrieves all tenants.
             verbose (bool): If True, prints detailed tenant information. If False, prints summary.
 
         Returns:
@@ -199,7 +201,14 @@ class TenantManager:
         Raises:
             Exception: If the collection does not exist or multi-tenancy is not enabled.
         """
-        tenants = self.client.collections.get(collection).tenants.get()
+        if tenant_id:
+            tenants = self.client.collections.get(collection).tenants.get_by_names(
+                [tenant_id]
+            )
+            if not tenants:
+                raise Exception(f"Tenant '{tenant_id}' not found in class {collection}")
+        else:
+            tenants = self.client.collections.get(collection).tenants.get()
 
         if verbose:
             click.echo(f"{'Tenant Name':<20}{'Activity Status':<20}")

{weaviate_cli-3.0.1 → weaviate_cli-3.0.2}/weaviate_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: weaviate-cli
-Version: 3.0.1
+Version: 3.0.2
 Summary: Command line interface to interact with weaviate
 Home-page: https://github.com/weaviate/weaviate-cli
 Download-URL: https://github.com/weaviate/weaviate-cli
@@ -125,7 +125,12 @@ Here you can see an example on how the configuration file should look like if yo
      "auth": {
          "type": "api_key",
          "api_key": "jfeRFsdfRfSasgsDoNOtTrYToUsErRQwqqdZfghasd"
-     }
+     },
+    "headers":{
+        "X-OpenAI-Api-Key":"OPEN_AI_KEY",
+        "X-Cohere-Api-Key":"Cohere_AI_KEY",
+        "X-JinaAI-Api-Key":"JINA_AI_KEY"
+        }
  }
 ```
 
@@ -138,6 +143,15 @@ Here you can see an example on how the configuration file should look like if yo
 
 Detailed documentation will be added soon.
 
+## Supported Model Provider
+
+- Contextionary
+- Transformers
+- OpenAI
+- Ollama
+- Cohere
+- JinaAI
+
 ## Community & Support
 
 - [Slack Community](https://weaviate.io/slack) - Join our active community