clarifai 10.9.2__py3-none-any.whl → 10.9.4__py3-none-any.whl

clarifai/__init__.py

@@ -1 +1 @@
-__version__ = "10.9.2"
+__version__ = "10.9.4"

clarifai/cli/base.py

@@ -0,0 +1,74 @@
+import os
+
+import click
+
+from ..utils.cli import dump_yaml, from_yaml, load_command_modules, set_base_url
+
+
+@click.group()
+@click.pass_context
+def cli(ctx):
+  """Clarifai CLI"""
+  ctx.ensure_object(dict)
+  config_path = 'config.yaml'
+  if os.path.exists(config_path):
+    ctx.obj = from_yaml(config_path)
+  else:
+    ctx.obj = {}
+
+
+@cli.command()
+@click.option('--config', type=click.Path(), required=False, help='Path to the config file')
+@click.option(
+    '-e',
+    '--env',
+    required=False,
+    help='Environment to use, choose from prod, staging and dev',
+    type=click.Choice(['prod', 'staging', 'dev']))
+@click.option('--user_id', required=False, help='User ID')
+@click.pass_context
+def login(ctx, config, env, user_id):
+  """Login command to set PAT and other configurations."""
+
+  if config and os.path.exists(config):
+    ctx.obj = from_yaml(config)
+
+  if 'pat' in ctx.obj:
+    os.environ["CLARIFAI_PAT"] = ctx.obj['pat']
+    click.echo("Loaded PAT from config file.")
+  elif 'CLARIFAI_PAT' in os.environ:
+    ctx.obj['pat'] = os.environ["CLARIFAI_PAT"]
+    click.echo("Loaded PAT from environment variable.")
+  else:
+    _pat = click.prompt(
+        "Get your PAT from https://clarifai.com/settings/security and pass it here", type=str)
+    os.environ["CLARIFAI_PAT"] = _pat
+    ctx.obj['pat'] = _pat
+    click.echo("PAT saved successfully.")
+
+  if user_id:
+    ctx.obj['user_id'] = user_id
+    os.environ["CLARIFAI_USER_ID"] = ctx.obj['user_id']
+  elif 'user_id' in ctx.obj or 'CLARIFAI_USER_ID' in os.environ:
+    ctx.obj['user_id'] = ctx.obj.get('user_id', os.environ["CLARIFAI_USER_ID"])
+    os.environ["CLARIFAI_USER_ID"] = ctx.obj['user_id']
+
+  if env:
+    ctx.obj['env'] = env
+    ctx.obj['base_url'] = set_base_url(env)
+    os.environ["CLARIFAI_API_BASE"] = ctx.obj['base_url']
+  elif 'env' in ctx.obj:
+    ctx.obj['env'] = ctx.obj.get('env', "prod")
+    ctx.obj['base_url'] = set_base_url(ctx.obj['env'])
+    os.environ["CLARIFAI_API_BASE"] = ctx.obj['base_url']
+  elif 'CLARIFAI_API_BASE' in os.environ:
+    ctx.obj['base_url'] = os.environ["CLARIFAI_API_BASE"]
+
+  dump_yaml(ctx.obj, 'config.yaml')
+
+
+# Import the CLI commands to register them
+load_command_modules()
+
+if __name__ == '__main__':
+  cli()

clarifai/cli/model.py

@@ -0,0 +1,65 @@
+import click
+from clarifai.cli.base import cli
+
+
+@cli.group()
+def model():
+  """Manage models: upload, test locally"""
+  pass
+
+
+@model.command()
+@click.option(
+    '--model_path',
+    type=click.Path(exists=True),
+    required=True,
+    help='Path to the model directory.')
+@click.option(
+    '--download_checkpoints',
+    is_flag=True,
+    help=
+    'Flag to download checkpoints before uploading and including them in the tar file that is uploaded. Defaults to False, which will attempt to download them at docker build time.',
+)
+@click.option(
+    '--skip_dockerfile',
+    is_flag=True,
+    help=
+    'Flag to skip generating a dockerfile so that you can manually edit an already created dockerfile.',
+)
+def upload(model_path, download_checkpoints, skip_dockerfile):
+  """Upload a model to Clarifai."""
+  from clarifai.runners.models import model_upload
+
+  model_upload.main(model_path, download_checkpoints, skip_dockerfile)
+
+
+@model.command()
+@click.option(
+    '--model_path',
+    type=click.Path(exists=True),
+    required=True,
+    help='Path to the model directory.')
+def test_locally(model_path):
+  """Test model locally."""
+  try:
+    from clarifai.runners.models import model_run_locally
+    model_run_locally.main(model_path)
+    click.echo(f"Model tested locally from {model_path}.")
+  except Exception as e:
+    click.echo(f"Failed to test model locally: {e}", err=True)
+
+
+@model.command()
+@click.option(
+    '--model_path',
+    type=click.Path(exists=True),
+    required=True,
+    help='Path to the model directory.')
+def run_locally(model_path):
+  """Run model locally and starts a GRPC server to serve the model."""
+  try:
+    from clarifai.runners.models import model_run_locally
+    model_run_locally.main(model_path, run_model_server=True)
+    click.echo(f"Model server started locally from {model_path}.")
+  except Exception as e:
+    click.echo(f"Failed to starts model server locally: {e}", err=True)

clarifai/client/auth/helper.py

@@ -5,11 +5,16 @@ from urllib.parse import urlparse
 
 from clarifai_grpc.channel.clarifai_channel import ClarifaiChannel
 from clarifai_grpc.grpc.api import resources_pb2, service_pb2_grpc
+
+from clarifai import __version__
 from clarifai.utils.constants import CLARIFAI_PAT_ENV_VAR, CLARIFAI_SESSION_TOKEN_ENV_VAR
 
 DEFAULT_BASE = "https://api.clarifai.com"
 DEFAULT_UI = "https://clarifai.com"
 
+REQUEST_ID_PREFIX_HEADER = "x-clarifai-request-id-prefix"
+REQUEST_ID_PREFIX = f"sdk-python-{__version__}"
+
 # Map from base domain to True / False for whether the base has https or http.
 # This is filled in get_stub() if it's not in there already.
 base_https_cache = {}
@@ -136,7 +141,7 @@ class ClarifaiAuthHelper:
 
     # Then add in the query params.
     try:
-      auth.add_streamlit_query_params(st.experimental_get_query_params())
+      auth.add_streamlit_query_params(dict(st.query_params))
     except Exception as e:
       st.error(e)
       st.stop()
@@ -269,9 +274,11 @@ Additionally, these optional params are supported:
       metadata: the metadata need to send with all grpc API calls in the API client.
     """
     if self._pat != "":
-      return (("authorization", "Key %s" % self._pat),)
+      return (("authorization", "Key %s" % self._pat), (REQUEST_ID_PREFIX_HEADER,
+                                                        REQUEST_ID_PREFIX))
     elif self._token != "":
-      return (("x-clarifai-session-token", self._token),)
+      return (("x-clarifai-session-token", self._token), (REQUEST_ID_PREFIX_HEADER,
+                                                          REQUEST_ID_PREFIX))
     else:
       raise Exception("'token' or 'pat' needed to be provided in the query params or env vars.")
 

clarifai/client/input.py

@@ -100,7 +100,7 @@ class Inputs(Lister, BaseClient):
       if not label_ids:
         concepts=[
             resources_pb2.Concept(
-            id=f"id-{''.join(_label.split(' '))}", name=_label, value=1.)\
+            id=_label, name=_label, value=1.)\
             for _label in labels
         ]
       else:
@@ -516,7 +516,7 @@ class Inputs(Lister, BaseClient):
                 right_col=bbox[2]  #x_max
             )),
             data=resources_pb2.Data(concepts=[
-                resources_pb2.Concept(id=f"id-{''.join(label.split(' '))}", name=label, value=1.)
+                resources_pb2.Concept(id=label, name=label, value=1.)
                 if not label_id else resources_pb2.Concept(id=label_id, name=label, value=1.)
             ]))
     ])
@@ -561,7 +561,7 @@ class Inputs(Lister, BaseClient):
                         visibility="VISIBLE") for _point in polygons
                 ])),
             data=resources_pb2.Data(concepts=[
-                resources_pb2.Concept(id=f"id-{''.join(label.split(' '))}", name=label, value=1.)
+                resources_pb2.Concept(id=label, name=label, value=1.)
                 if not label_id else resources_pb2.Concept(id=label_id, name=label, value=1.)
             ]))
     ])

clarifai/client/model.py

@@ -72,6 +72,7 @@ class Model(Lister, BaseClient):
     self.model_info = resources_pb2.Model(**self.kwargs)
     self.logger = logger
     self.training_params = {}
+    self.input_types = None
     BaseClient.__init__(
         self,
         user_id=self.user_id,
@@ -450,9 +451,55 @@ class Model(Lister, BaseClient):
 
     return response
 
+  def _check_predict_input_type(self, input_type: str) -> None:
+    """Checks if the input type is valid for the model.
+
+    Args:
+        input_type (str): The input type to check.
+    Returns:
+        None
+    """
+    if not input_type:
+      self.load_input_types()
+      if len(self.input_types) > 1:
+        raise UserError(
+            "Model has multiple input types. Please use model.predict() for this multi-modal model."
+        )
+    else:
+      self.input_types = [input_type]
+      if self.input_types[0] not in {'image', 'text', 'video', 'audio'}:
+        raise UserError(
+            f"Got input type {input_type} but expected one of image, text, video, audio.")
+
+  def load_input_types(self) -> None:
+    """Loads the input types for the model.
+
+    Returns:
+        None
+
+    Example:
+        >>> from clarifai.client.model import Model
+        >>> model = Model("url") # Example URL: https://clarifai.com/clarifai/main/models/general-image-recognition
+                    or
+        >>> model = Model(model_id='model_id', user_id='user_id', app_id='app_id')
+        >>> model.load_input_types()
+    """
+    if self.input_types:
+      return self.input_types
+    if self.model_info.model_type_id == "":
+      self.load_info()
+    request = service_pb2.GetModelTypeRequest(
+        user_app_id=self.user_app_id,
+        model_type_id=self.model_info.model_type_id,
+    )
+    response = self._grpc_request(self.STUB.GetModelType, request)
+    if response.status.code != status_code_pb2.SUCCESS:
+      raise Exception(response.status)
+    self.input_types = response.model_type.input_fields
+
   def predict_by_filepath(self,
                           filepath: str,
-                          input_type: str,
+                          input_type: str = None,
                           compute_cluster_id: str = None,
                           nodepool_id: str = None,
                           deployment_id: str = None,
@@ -462,7 +509,7 @@ class Model(Lister, BaseClient):
 
     Args:
         filepath (str): The filepath to predict.
-        input_type (str): The type of input. Can be 'image', 'text', 'video' or 'audio.
+        input_type (str, optional): The type of input. Can be 'image', 'text', 'video' or 'audio.
         compute_cluster_id (str): The compute cluster ID to use for the model.
         nodepool_id (str): The nodepool ID to use for the model.
         deployment_id (str): The deployment ID to use for the model.
@@ -477,8 +524,8 @@ class Model(Lister, BaseClient):
         >>> model = Model("url") # Example URL: https://clarifai.com/clarifai/main/models/general-image-recognition
                     or
         >>> model = Model(model_id='model_id', user_id='user_id', app_id='app_id')
-        >>> model_prediction = model.predict_by_filepath('/path/to/image.jpg', 'image')
-        >>> model_prediction = model.predict_by_filepath('/path/to/text.txt', 'text')
+        >>> model_prediction = model.predict_by_filepath('/path/to/image.jpg')
+        >>> model_prediction = model.predict_by_filepath('/path/to/text.txt')
     """
     if not os.path.isfile(filepath):
       raise UserError('Invalid filepath.')
@@ -491,7 +538,7 @@ class Model(Lister, BaseClient):
 
   def predict_by_bytes(self,
                        input_bytes: bytes,
-                       input_type: str,
+                       input_type: str = None,
                        compute_cluster_id: str = None,
                        nodepool_id: str = None,
                        deployment_id: str = None,
@@ -501,7 +548,7 @@ class Model(Lister, BaseClient):
 
     Args:
         input_bytes (bytes): File Bytes to predict on.
-        input_type (str): The type of input. Can be 'image', 'text', 'video' or 'audio.
+        input_type (str, optional): The type of input. Can be 'image', 'text', 'video' or 'audio.
         compute_cluster_id (str): The compute cluster ID to use for the model.
         nodepool_id (str): The nodepool ID to use for the model.
         deployment_id (str): The deployment ID to use for the model.
@@ -515,22 +562,17 @@ class Model(Lister, BaseClient):
         >>> from clarifai.client.model import Model
         >>> model = Model("https://clarifai.com/openai/chat-completion/models/GPT-4")
         >>> model_prediction = model.predict_by_bytes(b'Write a tweet on future of AI',
-                                                      input_type='text',
                                                       inference_params=dict(temperature=str(0.7), max_tokens=30)))
     """
-    if input_type not in {'image', 'text', 'video', 'audio'}:
-      raise UserError(
-          f"Got input type {input_type} but expected one of image, text, video, audio.")
-    if not isinstance(input_bytes, bytes):
-      raise UserError('Invalid bytes.')
+    self._check_predict_input_type(input_type)
 
-    if input_type == "image":
+    if self.input_types[0] == "image":
       input_proto = Inputs.get_input_from_bytes("", image_bytes=input_bytes)
-    elif input_type == "text":
+    elif self.input_types[0] == "text":
       input_proto = Inputs.get_input_from_bytes("", text_bytes=input_bytes)
-    elif input_type == "video":
+    elif self.input_types[0] == "video":
       input_proto = Inputs.get_input_from_bytes("", video_bytes=input_bytes)
-    elif input_type == "audio":
+    elif self.input_types[0] == "audio":
       input_proto = Inputs.get_input_from_bytes("", audio_bytes=input_bytes)
 
     if deployment_id and (compute_cluster_id or nodepool_id):
@@ -553,7 +595,7 @@ class Model(Lister, BaseClient):
 
   def predict_by_url(self,
                      url: str,
-                     input_type: str,
+                     input_type: str = None,
                      compute_cluster_id: str = None,
                      nodepool_id: str = None,
                      deployment_id: str = None,
@@ -563,7 +605,7 @@ class Model(Lister, BaseClient):
 
     Args:
         url (str): The URL to predict.
-        input_type (str): The type of input. Can be 'image', 'text', 'video' or 'audio.
+        input_type (str, optional): The type of input. Can be 'image', 'text', 'video' or 'audio'.
         compute_cluster_id (str): The compute cluster ID to use for the model.
         nodepool_id (str): The nodepool ID to use for the model.
         deployment_id (str): The deployment ID to use for the model.
@@ -578,19 +620,17 @@ class Model(Lister, BaseClient):
         >>> model = Model("url") # Example URL: https://clarifai.com/clarifai/main/models/general-image-recognition
                     or
         >>> model = Model(model_id='model_id', user_id='user_id', app_id='app_id')
-        >>> model_prediction = model.predict_by_url('url', 'image')
+        >>> model_prediction = model.predict_by_url('url')
     """
-    if input_type not in {'image', 'text', 'video', 'audio'}:
-      raise UserError(
-          f"Got input type {input_type} but expected one of image, text, video, audio.")
+    self._check_predict_input_type(input_type)
 
-    if input_type == "image":
+    if self.input_types[0] == "image":
       input_proto = Inputs.get_input_from_url("", image_url=url)
-    elif input_type == "text":
+    elif self.input_types[0] == "text":
       input_proto = Inputs.get_input_from_url("", text_url=url)
-    elif input_type == "video":
+    elif self.input_types[0] == "video":
       input_proto = Inputs.get_input_from_url("", video_url=url)
-    elif input_type == "audio":
+    elif self.input_types[0] == "audio":
       input_proto = Inputs.get_input_from_url("", audio_url=url)
 
     if deployment_id and (compute_cluster_id or nodepool_id):
@@ -668,7 +708,7 @@ class Model(Lister, BaseClient):
 
   def generate_by_filepath(self,
                            filepath: str,
-                           input_type: str,
+                           input_type: str = None,
                            compute_cluster_id: str = None,
                            nodepool_id: str = None,
                            deployment_id: str = None,
@@ -678,7 +718,7 @@ class Model(Lister, BaseClient):
 
     Args:
         filepath (str): The filepath to predict.
-        input_type (str): The type of input. Can be 'image', 'text', 'video' or 'audio.
+        input_type (str, optional): The type of input. Can be 'image', 'text', 'video' or 'audio.
         compute_cluster_id (str): The compute cluster ID to use for the model.
         nodepool_id (str): The nodepool ID to use for the model.
         deployment_id (str): The deployment ID to use for the model.
@@ -713,7 +753,7 @@ class Model(Lister, BaseClient):
 
   def generate_by_bytes(self,
                         input_bytes: bytes,
-                        input_type: str,
+                        input_type: str = None,
                         compute_cluster_id: str = None,
                         nodepool_id: str = None,
                         deployment_id: str = None,
@@ -723,7 +763,7 @@ class Model(Lister, BaseClient):
 
     Args:
         input_bytes (bytes): File Bytes to predict on.
-        input_type (str): The type of input. Can be 'image', 'text', 'video' or 'audio.
+        input_type (str, optional): The type of input. Can be 'image', 'text', 'video' or 'audio.
         compute_cluster_id (str): The compute cluster ID to use for the model.
         nodepool_id (str): The nodepool ID to use for the model.
         deployment_id (str): The deployment ID to use for the model.
@@ -737,24 +777,19 @@ class Model(Lister, BaseClient):
         >>> from clarifai.client.model import Model
         >>> model = Model("https://clarifai.com/openai/chat-completion/models/GPT-4")
         >>> stream_response = model.generate_by_bytes(b'Write a tweet on future of AI',
-                                                      input_type='text',
                                                       deployment_id='deployment_id',
                                                       inference_params=dict(temperature=str(0.7), max_tokens=30)))
         >>> list_stream_response = [response for response in stream_response]
     """
-    if input_type not in {'image', 'text', 'video', 'audio'}:
-      raise UserError(
-          f"Got input type {input_type} but expected one of image, text, video, audio.")
-    if not isinstance(input_bytes, bytes):
-      raise UserError('Invalid bytes.')
+    self._check_predict_input_type(input_type)
 
-    if input_type == "image":
+    if self.input_types[0] == "image":
       input_proto = Inputs.get_input_from_bytes("", image_bytes=input_bytes)
-    elif input_type == "text":
+    elif self.input_types[0] == "text":
       input_proto = Inputs.get_input_from_bytes("", text_bytes=input_bytes)
-    elif input_type == "video":
+    elif self.input_types[0] == "video":
       input_proto = Inputs.get_input_from_bytes("", video_bytes=input_bytes)
-    elif input_type == "audio":
+    elif self.input_types[0] == "audio":
       input_proto = Inputs.get_input_from_bytes("", audio_bytes=input_bytes)
 
     if deployment_id and (compute_cluster_id or nodepool_id):
@@ -777,7 +812,7 @@ class Model(Lister, BaseClient):
 
   def generate_by_url(self,
                       url: str,
-                      input_type: str,
+                      input_type: str = None,
                       compute_cluster_id: str = None,
                       nodepool_id: str = None,
                       deployment_id: str = None,
@@ -787,7 +822,7 @@ class Model(Lister, BaseClient):
 
     Args:
         url (str): The URL to predict.
-        input_type (str): The type of input. Can be 'image', 'text', 'video' or 'audio.
+        input_type (str, optional): The type of input. Can be 'image', 'text', 'video' or 'audio.
         compute_cluster_id (str): The compute cluster ID to use for the model.
         nodepool_id (str): The nodepool ID to use for the model.
         deployment_id (str): The deployment ID to use for the model.
@@ -802,20 +837,18 @@ class Model(Lister, BaseClient):
         >>> model = Model("url") # Example URL: https://clarifai.com/clarifai/main/models/general-image-recognition
                     or
         >>> model = Model(model_id='model_id', user_id='user_id', app_id='app_id')
-        >>> stream_response = model.generate_by_url('url', 'image', deployment_id='deployment_id')
+        >>> stream_response = model.generate_by_url('url', deployment_id='deployment_id')
         >>> list_stream_response = [response for response in stream_response]
     """
-    if input_type not in {'image', 'text', 'video', 'audio'}:
-      raise UserError(
-          f"Got input type {input_type} but expected one of image, text, video, audio.")
+    self._check_predict_input_type(input_type)
 
-    if input_type == "image":
+    if self.input_types[0] == "image":
       input_proto = Inputs.get_input_from_url("", image_url=url)
-    elif input_type == "text":
+    elif self.input_types[0] == "text":
       input_proto = Inputs.get_input_from_url("", text_url=url)
-    elif input_type == "video":
+    elif self.input_types[0] == "video":
       input_proto = Inputs.get_input_from_url("", video_url=url)
-    elif input_type == "audio":
+    elif self.input_types[0] == "audio":
       input_proto = Inputs.get_input_from_url("", audio_url=url)
 
     if deployment_id and (compute_cluster_id or nodepool_id):
@@ -893,7 +926,7 @@ class Model(Lister, BaseClient):
 
   def stream_by_filepath(self,
                          filepath: str,
-                         input_type: str,
+                         input_type: str = None,
                          compute_cluster_id: str = None,
                          nodepool_id: str = None,
                          deployment_id: str = None,
@@ -903,7 +936,7 @@ class Model(Lister, BaseClient):
 
     Args:
         filepath (str): The filepath to predict.
-        input_type (str): The type of input. Can be 'image', 'text', 'video' or 'audio.
+        input_type (str, optional): The type of input. Can be 'image', 'text', 'video' or 'audio.
         compute_cluster_id (str): The compute cluster ID to use for the model.
         nodepool_id (str): The nodepool ID to use for the model.
         deployment_id (str): The deployment ID to use for the model.
@@ -916,7 +949,7 @@ class Model(Lister, BaseClient):
     Example:
         >>> from clarifai.client.model import Model
         >>> model = Model("url")
-        >>> stream_response = model.stream_by_filepath('/path/to/image.jpg', 'image', deployment_id='deployment_id')
+        >>> stream_response = model.stream_by_filepath('/path/to/image.jpg', deployment_id='deployment_id')
         >>> list_stream_response = [response for response in stream_response]
     """
     if not os.path.isfile(filepath):
@@ -936,7 +969,7 @@ class Model(Lister, BaseClient):
 
   def stream_by_bytes(self,
                       input_bytes_iterator: Iterator[bytes],
-                      input_type: str,
+                      input_type: str = None,
                       compute_cluster_id: str = None,
                       nodepool_id: str = None,
                       deployment_id: str = None,
@@ -946,7 +979,7 @@ class Model(Lister, BaseClient):
 
     Args:
         input_bytes_iterator (Iterator[bytes]): Iterator of file bytes to predict on.
-        input_type (str): The type of input. Can be 'image', 'text', 'video' or 'audio.
+        input_type (str, optional): The type of input. Can be 'image', 'text', 'video' or 'audio.
         compute_cluster_id (str): The compute cluster ID to use for the model.
         nodepool_id (str): The nodepool ID to use for the model.
         deployment_id (str): The deployment ID to use for the model.
@@ -960,24 +993,21 @@ class Model(Lister, BaseClient):
         >>> from clarifai.client.model import Model
         >>> model = Model("https://clarifai.com/openai/chat-completion/models/GPT-4")
         >>> stream_response = model.stream_by_bytes(iter([b'Write a tweet on future of AI']),
-                                                    input_type='text',
                                                     deployment_id='deployment_id',
                                                     inference_params=dict(temperature=str(0.7), max_tokens=30)))
         >>> list_stream_response = [response for response in stream_response]
     """
-    if input_type not in {'image', 'text', 'video', 'audio'}:
-      raise UserError(
-          f"Got input type {input_type} but expected one of image, text, video, audio.")
+    self._check_predict_input_type(input_type)
 
     def input_generator():
       for input_bytes in input_bytes_iterator:
-        if input_type == "image":
+        if self.input_types[0] == "image":
           yield [Inputs.get_input_from_bytes("", image_bytes=input_bytes)]
-        elif input_type == "text":
+        elif self.input_types[0] == "text":
           yield [Inputs.get_input_from_bytes("", text_bytes=input_bytes)]
-        elif input_type == "video":
+        elif self.input_types[0] == "video":
           yield [Inputs.get_input_from_bytes("", video_bytes=input_bytes)]
-        elif input_type == "audio":
+        elif self.input_types[0] == "audio":
           yield [Inputs.get_input_from_bytes("", audio_bytes=input_bytes)]
 
     if deployment_id and (compute_cluster_id or nodepool_id):
@@ -1000,7 +1030,7 @@ class Model(Lister, BaseClient):
 
   def stream_by_url(self,
                     url_iterator: Iterator[str],
-                    input_type: str,
+                    input_type: str = None,
                     compute_cluster_id: str = None,
                     nodepool_id: str = None,
                     deployment_id: str = None,
@@ -1010,7 +1040,7 @@ class Model(Lister, BaseClient):
 
     Args:
         url_iterator (Iterator[str]): Iterator of URLs to predict.
-        input_type (str): The type of input. Can be 'image', 'text', 'video' or 'audio.
+        input_type (str, optional): The type of input. Can be 'image', 'text', 'video' or 'audio.
         compute_cluster_id (str): The compute cluster ID to use for the model.
         nodepool_id (str): The nodepool ID to use for the model.
         deployment_id (str): The deployment ID to use for the model.
@@ -1023,22 +1053,20 @@ class Model(Lister, BaseClient):
     Example:
         >>> from clarifai.client.model import Model
         >>> model = Model("url")
-        >>> stream_response = model.stream_by_url(iter(['url']), 'image', deployment_id='deployment_id')
+        >>> stream_response = model.stream_by_url(iter(['url']), deployment_id='deployment_id')
         >>> list_stream_response = [response for response in stream_response]
     """
-    if input_type not in {'image', 'text', 'video', 'audio'}:
-      raise UserError(
-          f"Got input type {input_type} but expected one of image, text, video, audio.")
+    self._check_predict_input_type(input_type)
 
     def input_generator():
       for url in url_iterator:
-        if input_type == "image":
+        if self.input_types[0] == "image":
           yield [Inputs.get_input_from_url("", image_url=url)]
-        elif input_type == "text":
+        elif self.input_types[0] == "text":
           yield [Inputs.get_input_from_url("", text_url=url)]
-        elif input_type == "video":
+        elif self.input_types[0] == "video":
           yield [Inputs.get_input_from_url("", video_url=url)]
-        elif input_type == "audio":
+        elif self.input_types[0] == "audio":
           yield [Inputs.get_input_from_url("", audio_url=url)]
 
     if deployment_id and (compute_cluster_id or nodepool_id):

clarifai/client/workflow.py

@@ -5,10 +5,12 @@ from typing import Dict, Generator, List
 from clarifai_grpc.grpc.api import resources_pb2, service_pb2
 from clarifai_grpc.grpc.api.resources_pb2 import Input
 from clarifai_grpc.grpc.api.status import status_code_pb2
+from google.protobuf.json_format import MessageToDict
 
 from clarifai.client.base import BaseClient
 from clarifai.client.input import Inputs
 from clarifai.client.lister import Lister
+from clarifai.client.model import Model
 from clarifai.constants.workflow import MAX_WORKFLOW_PREDICT_INPUTS
 from clarifai.errors import UserError
 from clarifai.urls.helper import ClarifaiUrlHelper
@@ -60,6 +62,7 @@ class Workflow(Lister, BaseClient):
     self.output_config = output_config
     self.workflow_info = resources_pb2.Workflow(**self.kwargs)
     self.logger = logger
+    self.input_types = None
     BaseClient.__init__(
         self,
         user_id=self.user_id,
@@ -109,20 +112,26 @@ class Workflow(Lister, BaseClient):
 
     return response
 
-  def predict_by_filepath(self, filepath: str, input_type: str):
+  def predict_by_filepath(self, filepath: str, input_type: str = None):
     """Predicts the workflow based on the given filepath.
 
     Args:
         filepath (str): The filepath to predict.
-        input_type (str): The type of input. Can be 'image', 'text', 'video' or 'audio.
+        input_type (str, optional): The type of input. Can be 'image', 'text', 'video' or 'audio.
 
     Example:
         >>> from clarifai.client.workflow import Workflow
         >>> workflow = Workflow("url") # Example: https://clarifai.com/clarifai/main/workflows/Face-Sentiment
                       or
         >>> workflow = Workflow(user_id='user_id', app_id='app_id', workflow_id='workflow_id')
-        >>> workflow_prediction = workflow.predict_by_filepath('filepath', 'image')
+        >>> workflow_prediction = workflow.predict_by_filepath('filepath')
     """
+    if not input_type:
+      self.load_info()
+      if len(self.input_types) > 1:
+        raise UserError("Workflow has multiple input types. Please use workflow.predict().")
+      input_type = self.input_types[0]
+
     if input_type not in {'image', 'text', 'video', 'audio'}:
       raise UserError('Invalid input type it should be image, text, video or audio.')
     if not os.path.isfile(filepath):
@@ -133,13 +142,19 @@ class Workflow(Lister, BaseClient):
 
     return self.predict_by_bytes(file_bytes, input_type)
 
-  def predict_by_bytes(self, input_bytes: bytes, input_type: str):
+  def predict_by_bytes(self, input_bytes: bytes, input_type: str = None):
     """Predicts the workflow based on the given bytes.
 
     Args:
         input_bytes (bytes): Bytes to predict on.
-        input_type (str): The type of input. Can be 'image', 'text', 'video' or 'audio.
+        input_type (str, optional): The type of input. Can be 'image', 'text', 'video' or 'audio.
     """
+    if not input_type:
+      self.load_info()
+      if len(self.input_types) > 1:
+        raise UserError("Workflow has multiple input types. Please use workflow.predict().")
+      input_type = self.input_types[0]
+
     if input_type not in {'image', 'text', 'video', 'audio'}:
       raise UserError('Invalid input type it should be image, text, video or audio.')
     if not isinstance(input_bytes, bytes):
@@ -156,20 +171,26 @@ class Workflow(Lister, BaseClient):
 
     return self.predict(inputs=[input_proto])
 
-  def predict_by_url(self, url: str, input_type: str):
+  def predict_by_url(self, url: str, input_type: str = None):
     """Predicts the workflow based on the given URL.
 
     Args:
         url (str): The URL to predict.
-        input_type (str): The type of input. Can be 'image', 'text', 'video' or 'audio.
+        input_type (str, optional): The type of input. Can be 'image', 'text', 'video' or 'audio.
 
     Example:
         >>> from clarifai.client.workflow import Workflow
         >>> workflow = Workflow("url") # Example: https://clarifai.com/clarifai/main/workflows/Face-Sentiment
                       or
         >>> workflow = Workflow(user_id='user_id', app_id='app_id', workflow_id='workflow_id')
-        >>> workflow_prediction = workflow.predict_by_url('url', 'image')
+        >>> workflow_prediction = workflow.predict_by_url('url')
     """
+    if not input_type:
+      self.load_info()
+      if len(self.input_types) > 1:
+        raise UserError("Workflow has multiple input types. Please use workflow.predict().")
+      input_type = self.input_types[0]
+
     if input_type not in {'image', 'text', 'video', 'audio'}:
       raise UserError('Invalid input type it should be image, text, video or audio.')
 
@@ -245,6 +266,25 @@ class Workflow(Lister, BaseClient):
 
     self.logger.info(f"Exported workflow to {out_path}")
 
+  def load_info(self) -> None:
+    """Loads the workflow info."""
+    if not self.input_types:
+      request = service_pb2.GetWorkflowRequest(user_app_id=self.user_app_id, workflow_id=self.id)
+      response = self._grpc_request(self.STUB.GetWorkflow, request)
+      if response.status.code != status_code_pb2.SUCCESS:
+        raise Exception(f"Workflow Get failed with response {response.status!r}")
+
+      dict_response = MessageToDict(response, preserving_proto_field_name=True)
+      self.kwargs = self.process_response_keys(dict_response['workflow'])
+      self.workflow_info = resources_pb2.Workflow(**self.kwargs)
+
+      model = Model(
+          model_id=self.kwargs['nodes'][0]['model']['id'],
+          **self.kwargs['nodes'][0]['model'],
+          pat=self.pat)
+      model.load_input_types()
+      self.input_types = model.input_types
+
   def __getattr__(self, name):
     return getattr(self.workflow_info, name)
 

clarifai/datasets/export/inputs_annotations.py

@@ -192,9 +192,11 @@ class InputAnnotationDownloader:
         self._save_video_to_archive(new_archive, hosted_url, file_name)
       self.num_inputs += 1
 
-    if data_dict.get("concepts") or data_dict.get("regions"):
+    if data_dict.get("metadata") or data_dict.get("concepts") or data_dict.get("regions"):
       file_name = os.path.join(split, "annotations", input_.id + ".json")
-      annot_data = data_dict.get("regions", []) + data_dict.get("concepts", [])
+      annot_data = [{
+          "metadata": data_dict.get("metadata", {})
+      }] + data_dict.get("regions", []) + data_dict.get("concepts", [])
 
       self._save_annotation_to_archive(new_archive, annot_data, file_name)
       self.num_annotations += 1

clarifai/datasets/upload/utils.py

@@ -143,7 +143,7 @@ class DisplayUploadStatus:
 
     for data in dict_response["datasetVersionMetricsGroups"]:
       if isinstance(data["value"], str):
-        if data["value"].startswith("id-"):
+        if ("type" in data) and (data["type"] == "CONCEPT_ID"):
           data["metrics"].update({"Concept": data["value"]})
           data["metrics"].pop("regionLocationMatrix", None)
           dataset_statistics.append(data["metrics"])

clarifai/runners/models/model_run_locally.py

@@ -1,4 +1,3 @@
-import argparse
 import importlib.util
 import inspect
 import os
@@ -126,6 +125,7 @@ class ModelRunLocally:
         runner_id="n/a",
         nodepool_id="n/a",
         compute_cluster_id="n/a",
+        user_id="n/a",
     )
 
     # send an inference.
@@ -175,22 +175,16 @@ class ModelRunLocally:
       shutil.rmtree(self.temp_dir)
 
 
-def main():
-  parser = argparse.ArgumentParser()
-  parser.add_argument(
-      '--model_path', type=str, required=True, help='Path of the model folder to upload')
-  args = parser.parse_args()
+def main(model_path, run_model_server=False):
 
-  model_path = args.model_path
   manager = ModelRunLocally(model_path)
   manager.create_temp_venv()
 
   try:
     manager.install_requirements()
-    manager.test_model()
+    if run_model_server:
+      manager.run_model_server()
+    else:
+      manager.test_model()
   finally:
     manager.clean_up()
-
-
-if __name__ == "__main__":
-  main()

clarifai/runners/models/model_servicer.py

@@ -28,7 +28,7 @@ class ModelServicer(service_pb2_grpc.V2Servicer):
     """
 
     # Download any urls that are not already bytes.
-    ensure_urls_downloaded(self.url_fetcher, request)
+    ensure_urls_downloaded(request)
 
     try:
       return self.model_class.predict(request)
@@ -47,7 +47,7 @@ class ModelServicer(service_pb2_grpc.V2Servicer):
     returns an output.
     """
     # Download any urls that are not already bytes.
-    ensure_urls_downloaded(self.url_fetcher, request)
+    ensure_urls_downloaded(request)
 
     try:
       return self.model_class.generate(request)
@@ -71,7 +71,7 @@ class ModelServicer(service_pb2_grpc.V2Servicer):
 
     # Download any urls that are not already bytes.
     for req in request:
-      ensure_urls_downloaded(self.url_fetcher, req)
+      ensure_urls_downloaded(req)
 
     try:
       return self.model_class.stream(request_copy)

clarifai/runners/models/model_upload.py

@@ -1,4 +1,3 @@
-import argparse
 import os
 import time
 from string import Template
@@ -409,11 +408,11 @@ class ModelUploader:
         return False
 
 
-def main(folder, download_checkpoints):
+def main(folder, download_checkpoints, skip_dockerfile):
   uploader = ModelUploader(folder)
   if download_checkpoints:
     uploader.download_checkpoints()
-  if not args.skip_dockerfile:
+  if not skip_dockerfile:
     uploader.create_dockerfile()
   exists = uploader.check_model_exists()
   if exists:
@@ -425,25 +424,3 @@ def main(folder, download_checkpoints):
 
   input("Press Enter to continue...")
   uploader.upload_model_version(download_checkpoints)
-
-
-if __name__ == "__main__":
-  parser = argparse.ArgumentParser()
-  parser.add_argument(
-      '--model_path', type=str, help='Path of the model folder to upload', required=True)
-  # flag to default to not download checkpoints
-  parser.add_argument(
-      '--download_checkpoints',
-      action='store_true',
-      help=
-      'Flag to download checkpoints before uploading and including them in the tar file that is uploaded. Defaults to False, which will attempt to download them at docker build time.',
-  )
-  parser.add_argument(
-      '--skip_dockerfile',
-      action='store_true',
-      help=
-      'Flag to skip generating a dockerfile so that you can manually edit an already created dockerfile.',
-  )
-  args = parser.parse_args()
-
-  main(args.model_path, args.download_checkpoints)

clarifai/utils/cli.py

@@ -0,0 +1,45 @@
+import importlib
+import os
+import pkgutil
+
+import click
+import yaml
+
+
+def from_yaml(filename: str):
+  try:
+    with open(filename, 'r') as f:
+      return yaml.safe_load(f)
+  except yaml.YAMLError as e:
+    click.echo(f"Error reading YAML file: {e}", err=True)
+    return {}
+
+
+def dump_yaml(data, filename: str):
+  try:
+    with open(filename, 'w') as f:
+      yaml.dump(data, f)
+  except Exception as e:
+    click.echo(f"Error writing YAML file: {e}", err=True)
+
+
+def set_base_url(env):
+  environments = {
+      'prod': 'https://api.clarifai.com',
+      'staging': 'https://api-staging.clarifai.com',
+      'dev': 'https://api-dev.clarifai.com'
+  }
+
+  if env in environments:
+    return environments[env]
+  else:
+    raise ValueError("Invalid environment. Please choose from 'prod', 'staging', 'dev'.")
+
+
+# Dynamically find and import all command modules from the cli directory
+def load_command_modules():
+  package_dir = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'cli')
+
+  for _, module_name, _ in pkgutil.iter_modules([package_dir]):
+    if module_name != 'base':  # Skip the base.py file itself
+      importlib.import_module(f'clarifai.cli.{module_name}')

{clarifai-10.9.2.dist-info → clarifai-10.9.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: clarifai
-Version: 10.9.2
+Version: 10.9.4
 Summary: Clarifai Python SDK
 Home-page: https://github.com/Clarifai/clarifai-python
 Author: Clarifai
@@ -20,7 +20,7 @@ Classifier: Operating System :: OS Independent
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: clarifai-grpc >=10.9.4
+Requires-Dist: clarifai-grpc >=10.9.11
 Requires-Dist: clarifai-protocol >=0.0.6
 Requires-Dist: numpy >=1.22.0
 Requires-Dist: tqdm >=4.65.0
@@ -33,15 +33,17 @@ Requires-Dist: inquirerpy ==0.3.4
 Requires-Dist: tabulate >=0.9.0
 Requires-Dist: protobuf ==5.27.3
 Requires-Dist: fsspec ==2024.6.1
+Requires-Dist: click ==8.1.7
 Provides-Extra: all
 Requires-Dist: pycocotools ==2.0.6 ; extra == 'all'
 
 <h1 align="center">
-  <a href="https://www.clarifai.com/"><img alt="Clarifai" title="Clarifai" src="https://upload.wikimedia.org/wikipedia/commons/b/bc/Clarifai_Logo_FC_Web.png"></a>
+  <a href="https://www.clarifai.com/"><img alt="Clarifai" title="Clarifai" src="https://github.com/user-attachments/assets/623b883b-7fe5-4b95-bbfa-8691f5779af4"></a>
 </h1>
 
+
 <h2 align="center">
-Clarifai Python SDK</a>
+Clarifai Python SDK
 </h2>
 
 
@@ -51,6 +53,8 @@ Clarifai Python SDK</a>
   </a>
   <a href="https://pypi.org/project/clarifai" target="_blank"> <img src="https://img.shields.io/pypi/dm/clarifai" alt="PyPI - Downloads">
   </a>
+  <a href="https://img.shields.io/pypi/pyversions/clarifai" target="_blank"> <img src="https://img.shields.io/pypi/pyversions/clarifai" alt="PyPI - Versions">
+  </a>
 </p>
 
 
@@ -261,19 +265,19 @@ print(gpt_4_model)
 
 
 # Model Predict
-model_prediction = Model("https://clarifai.com/anthropic/completion/models/claude-v2").predict_by_bytes(b"Write a tweet on future of AI", input_type="text")
+model_prediction = Model("https://clarifai.com/anthropic/completion/models/claude-v2").predict_by_bytes(b"Write a tweet on future of AI")
 
 # Customizing Model Inference Output
-model_prediction = gpt_4_model.predict_by_bytes(b"Write a tweet on future of AI", "text", inference_params=dict(temperature=str(0.7), max_tokens=30))
+model_prediction = gpt_4_model.predict_by_bytes(b"Write a tweet on future of AI", inference_params=dict(temperature=str(0.7), max_tokens=30))
 # Return predictions having prediction confidence > 0.98
-model_prediction = model.predict_by_filepath(filepath="local_filepath", input_type, output_config={"min_value": 0.98}) # Supports image, text, audio, video
+model_prediction = model.predict_by_filepath(filepath="local_filepath", output_config={"min_value": 0.98}) # Supports image, text, audio, video
 
 # Supports prediction by url
-model_prediction = model.predict_by_url(url="url", input_type) # Supports image, text, audio, video
+model_prediction = model.predict_by_url(url="url") # Supports image, text, audio, video
 
 # Return predictions for specified interval of video
 video_input_proto = [input_obj.get_input_from_url("Input_id", video_url=BEER_VIDEO_URL)]
-model_prediction = model.predict(video_input_proto, input_type="video", output_config={"sample_ms": 2000})
+model_prediction = model.predict(video_input_proto, output_config={"sample_ms": 2000})
 ```
 #### Model Training
 ```python
@@ -383,12 +387,12 @@ from clarifai.client.workflow import Workflow
 
 # Workflow Predict
 workflow = Workflow("workflow_url") # Example: https://clarifai.com/clarifai/main/workflows/Face-Sentiment
-workflow_prediction = workflow.predict_by_url(url="url", input_type="image") # Supports image, text, audio, video
+workflow_prediction = workflow.predict_by_url(url="url") # Supports image, text, audio, video
 
 # Customizing Workflow Inference Output
 workflow = Workflow(user_id="user_id", app_id="app_id", workflow_id="workflow_id",
                   output_config={"min_value": 0.98}) # Return predictions having prediction confidence > 0.98
-workflow_prediction = workflow.predict_by_filepath(filepath="local_filepath", input_type="text") # Supports image, text, audio, video
+workflow_prediction = workflow.predict_by_filepath(filepath="local_filepath") # Supports image, text, audio, video
 ```
 
 #### Workflows Listing

{clarifai-10.9.2.dist-info → clarifai-10.9.4.dist-info}/RECORD

@@ -1,23 +1,26 @@
-clarifai/__init__.py,sha256=pIn_ah9SsqsljuRvOukQQ4KCQQ4aaDew3A8OtaeuEqU,23
+clarifai/__init__.py,sha256=60DFxKRLc905eL5FiLBS5Dd9SsZV4LlHGOJDJmZ3CTM,23
 clarifai/cli.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clarifai/errors.py,sha256=RwzTajwds51wLD0MVlMC5kcpBnzRpreDLlazPSBZxrg,2605
 clarifai/versions.py,sha256=jctnczzfGk_S3EnVqb2FjRKfSREkNmvNEwAAa_VoKiQ,222
+clarifai/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+clarifai/cli/base.py,sha256=N3dBmpH6XDSekMvwD_lrMpRe9CuwfR_RkJ612iDir6U,2228
+clarifai/cli/model.py,sha256=O0GZWbTF8ZMNpSn7KTYpMiMsAULAWkTFWBucaLqVcZg,1925
 clarifai/client/__init__.py,sha256=xI1U0l5AZdRThvQAXCLsd9axxyFzXXJ22m8LHqVjQRU,662
 clarifai/client/app.py,sha256=6pckYme1urV2YJjLIYfeZ-vH0Z5YSQa51jzIMcEfwug,38342
 clarifai/client/base.py,sha256=hSHOqkXbSKyaRDeylMMnkhUHCAHhEqno4KI0CXGziBA,7536
 clarifai/client/compute_cluster.py,sha256=lntZDLVDhS71Yj7mZrgq5uhnAuNPUnj48i3zMSuoUpk,8693
 clarifai/client/dataset.py,sha256=Xt8M9sP7RLeAqqTu17-8cpupThhkU9n_bl6jXtG_25A,31148
 clarifai/client/deployment.py,sha256=4gfvUvQY9adFS98B0vP9C5fR9OnDRV2JbUIdAkMymT8,2551
-clarifai/client/input.py,sha256=cEVRytrMF1gCgwHLbXlSbPSEQN8uHpUAoKcCdyHO1pc,44406
+clarifai/client/input.py,sha256=smAZqvBEVJlM59PQ3KhqbA6HaKHRNlSyQCcw917SbJ0,44322
 clarifai/client/lister.py,sha256=03KGMvs5RVyYqxLsSrWhNc34I8kiF1Ph0NeyEwu7nMU,2082
-clarifai/client/model.py,sha256=jNTyCxrME4vbU3Qw9nMSFwg4Ud8AuJrsmVYOMLBtbLI,83846
+clarifai/client/model.py,sha256=WmLBPm_rDzbPR_Cxo8gnntBnPiWFt3gYKiiKuJ9lH04,84652
 clarifai/client/module.py,sha256=FTkm8s9m-EaTKN7g9MnLhGJ9eETUfKG7aWZ3o1RshYs,4204
 clarifai/client/nodepool.py,sha256=DK8oqswjjrP6TqCqbw7Ge51Z7PxK3XmWZGLeUM3fd_A,10142
 clarifai/client/search.py,sha256=GaPWN6JmTQGZaCHr6U1yv0zqR6wKFl7i9IVLg2ul1CI,14254
 clarifai/client/user.py,sha256=0tcOk8_Yd1_ANj9E6sy9mz6s01V3qkmJS7pZVn_zUYo,17637
-clarifai/client/workflow.py,sha256=Wm4Fry6lGx8T43sBUqRI7v7sAmuvq_4Jft3vSW8UUJU,10516
+clarifai/client/workflow.py,sha256=BOmA1ilHxsw-yl_ZE1NOEshzV7owavnXTIG2UOD6PuA,12163
 clarifai/client/auth/__init__.py,sha256=7EwR0NrozkAUwpUnCsqXvE_p0wqx_SelXlSpKShKJK0,136
-clarifai/client/auth/helper.py,sha256=BuyI_smxuRq__8RF0Nv3lr_Rmb-CJVEyK4hv60VEV5A,14259
+clarifai/client/auth/helper.py,sha256=1QmQatzVdWDm4J20kgiHioM1lfHlgkGxWboO3z02QAI,14593
 clarifai/client/auth/register.py,sha256=2CMdBsoVLoTfjyksE6j7BM2tiEc73WKYvxnwDDgNn1k,536
 clarifai/client/auth/stub.py,sha256=xy4-fV0W8keCgXld4eOVzFQEIKxOktNwtL5bLztReug,4940
 clarifai/constants/base.py,sha256=ogmFSZYoF0YhGjHg5aiOc3MLqPr_poKAls6xaD0_C3U,89
@@ -29,14 +32,14 @@ clarifai/constants/search.py,sha256=yYEqTaFg-KdnpJE_Ytp-EPVHIIC395iNtZrpVlLIf4o,
 clarifai/constants/workflow.py,sha256=cECq1xdvf44MCdtK2AbkiuuwhyL-6OWZdQfYbsLKy_o,33
 clarifai/datasets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clarifai/datasets/export/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-clarifai/datasets/export/inputs_annotations.py,sha256=Mwo-wFRIYXsoe13bR2bfsXViA6DlJbWmNuPpD3yHSrE,9701
+clarifai/datasets/export/inputs_annotations.py,sha256=vqe3CBlWTgeBnIoRnjwIRnOj5QmH1480efSk8tv2mCg,9796
 clarifai/datasets/upload/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clarifai/datasets/upload/base.py,sha256=UIc0ufyIBCrb83_sFpv21L8FshsX4nwsLYQkdlJfzD4,2357
 clarifai/datasets/upload/features.py,sha256=jv2x7jGZKS-LMt87sEZNBwwOskHbP26XTMjoiaSA5pg,2024
 clarifai/datasets/upload/image.py,sha256=HlCsfEMu_C4GVecGSv52RUJ6laLW8H64Pfj_FQyX6qg,8580
 clarifai/datasets/upload/multimodal.py,sha256=2_s4SCotzacVMdnqkM7upSR3Ovxh3b9oW1hSWhKyxO4,2373
 clarifai/datasets/upload/text.py,sha256=boVJenfQZKf79aXu8CEP4g_ANzX5ROdd06g07O7RnXU,2198
-clarifai/datasets/upload/utils.py,sha256=h7mtN9FZXhQQbf47EXczgb-NTY2uOE9AJlE9u4-hDwI,9627
+clarifai/datasets/upload/utils.py,sha256=BerWhq40ZUN30z6VImlc93eZtT-1vI18AMgSOuNzJEM,9647
 clarifai/datasets/upload/loaders/README.md,sha256=aNRutSCTzLp2ruIZx74ZkN5AxpzwKOxMa7OzabnKpwg,2980
 clarifai/datasets/upload/loaders/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clarifai/datasets/upload/loaders/coco_captions.py,sha256=YfuNXplbdoH8N9ph7RyN9MfJTtOcJBG4ie1ow6-mELA,1516
@@ -60,10 +63,10 @@ clarifai/runners/dockerfile_template/Dockerfile.cuda.template,sha256=8uQp2sX_bIz
 clarifai/runners/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clarifai/runners/models/base_typed_model.py,sha256=OnAk08Lo2Y1fGiBc6JJ6UvJ8P435cTsikTNYDkStDpI,7790
 clarifai/runners/models/model_class.py,sha256=9JSPAr4U4K7xI0kSl-q0mHB06zknm2OR-8XIgBCto94,1611
-clarifai/runners/models/model_run_locally.py,sha256=eP_QFx3fuoKbf5VyRm-rfD9k6fLrXG7u1EaWz-Qjr68,6720
+clarifai/runners/models/model_run_locally.py,sha256=xbNcD0TMRlk52cUjJH-qenlkeiwS4YcOeb8eYy7KAEI,6583
 clarifai/runners/models/model_runner.py,sha256=3vzoastQxkGRDK8T9aojDsLNBb9A3IiKm6YmbFrE9S0,6241
-clarifai/runners/models/model_servicer.py,sha256=L5AuqKDZrsKOnv_Fz1Ld4-nzqehltLTsYAS7NIclm1g,2880
-clarifai/runners/models/model_upload.py,sha256=7Oi09Y93WUH7EKTzrKf6JQawZzLj3EnmWUDUgHOrFq8,17654
+clarifai/runners/models/model_servicer.py,sha256=X4715PVA5PBurRTYcwSEudg8fShGV6InAF4mmRlRcHg,2826
+clarifai/runners/models/model_upload.py,sha256=cfxHfNbnVLHjF50dDDVdAE5VDDTf8yF1o0HL5Rgvklc,16859
 clarifai/runners/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clarifai/runners/utils/data_handler.py,sha256=sxy9zlAgI6ETuxCQhUgEXAn2GCsaW1GxpK6GTaMne0g,6966
 clarifai/runners/utils/data_utils.py,sha256=R1iQ82TuQ9JwxCJk8yEB1Lyb0BYVhVbWJI9YDi1zGOs,318
@@ -72,6 +75,7 @@ clarifai/runners/utils/url_fetcher.py,sha256=-Hwjb1SURszn7zUVwi4Of0-nrksfZy-uqT4
 clarifai/schema/search.py,sha256=JjTi8ammJgZZ2OGl4K6tIA4zEJ1Fr2ASZARXavI1j5c,2448
 clarifai/urls/helper.py,sha256=tjoMGGHuWX68DUB0pk4MEjrmFsClUAQj2jmVEM_Sy78,4751
 clarifai/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+clarifai/utils/cli.py,sha256=avHgpKV3XP5V2L05Gpi458oiKJPV6qD5YSMdam52SlE,1175
 clarifai/utils/constants.py,sha256=MG_iHnSwNEyUZOpvsrTicNwaT4CIjmlK_Ixk_qqEX8g,142
 clarifai/utils/logging.py,sha256=rhutBRQJLtkNRz8IErNCgbIpvtl2fQ3D2otYcGqd3-Q,11565
 clarifai/utils/misc.py,sha256=ptjt1NtteDT0EhrPoyQ7mgWtvoAQ-XNncQaZvNHb0KI,2253
@@ -84,9 +88,9 @@ clarifai/workflows/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuF
 clarifai/workflows/export.py,sha256=vICRhIreqDSShxLKjHNM2JwzKsf1B4fdXB0ciMcA70k,1945
 clarifai/workflows/utils.py,sha256=nGeB_yjVgUO9kOeKTg4OBBaBz-AwXI3m-huSVj-9W18,1924
 clarifai/workflows/validate.py,sha256=yJq03MaJqi5AK3alKGJJBR89xmmjAQ31sVufJUiOqY8,2556
-clarifai-10.9.2.dist-info/LICENSE,sha256=mUqF_d12-qE2n41g7C5_sq-BMLOcj6CNN-jevr15YHU,555
-clarifai-10.9.2.dist-info/METADATA,sha256=BcdNjVzD08g4xW4RgxN-zkLGLsG2P590d6tAzY4f5ME,19479
-clarifai-10.9.2.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-clarifai-10.9.2.dist-info/entry_points.txt,sha256=qZOr_MIPG0dBBE1zringDJS_wXNGTAA_SQ-zcbmDHOw,82
-clarifai-10.9.2.dist-info/top_level.txt,sha256=wUMdCQGjkxaynZ6nZ9FAnvBUCgp5RJUVFSy2j-KYo0s,9
-clarifai-10.9.2.dist-info/RECORD,,
+clarifai-10.9.4.dist-info/LICENSE,sha256=mUqF_d12-qE2n41g7C5_sq-BMLOcj6CNN-jevr15YHU,555
+clarifai-10.9.4.dist-info/METADATA,sha256=eurSyw4GH8EzpxT4K--6kUzbC4Eo9BuBtw_4VabmMKI,19565
+clarifai-10.9.4.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+clarifai-10.9.4.dist-info/entry_points.txt,sha256=X9FZ4Z-i_r2Ud1RpZ9sNIFYuu_-9fogzCMCRUD9hyX0,51
+clarifai-10.9.4.dist-info/top_level.txt,sha256=wUMdCQGjkxaynZ6nZ9FAnvBUCgp5RJUVFSy2j-KYo0s,9
+clarifai-10.9.4.dist-info/RECORD,,

{clarifai-10.9.2.dist-info → clarifai-10.9.4.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.1.0)
+Generator: setuptools (75.2.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

clarifai-10.9.4.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+clarifai = clarifai.cli.base:cli

clarifai-10.9.2.dist-info/entry_points.txt

@@ -1,2 +0,0 @@
-[console_scripts]
-clarifai = clarifai.models.model_serving.cli.clarifai_clis:main