clarifai 9.8.1__tar.gz → 9.9.0__tar.gz

{clarifai-9.8.1/clarifai.egg-info → clarifai-9.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: clarifai
-Version: 9.8.1
+Version: 9.9.0
 Summary: Clarifai Python SDK
 Home-page: https://github.com/Clarifai/clarifai-python
 Author: Clarifai
@@ -46,7 +46,7 @@ This is the official Python client for interacting with our powerful [API](https
 
 **Clarifai Community**: [https://clarifai.com/explore](https://clarifai.com/explore)
 
-**Python SDK Docs**: [https://clarifai-python.readthedocs.io/en/latest/index.html](https://clarifai-python.readthedocs.io/en/latest/index.html)
+**Python SDK Docs**: [https://docs.clarifai.com/python-sdk/api-reference](https://docs.clarifai.com/python-sdk/api-reference)
 
 
 ---
@@ -211,7 +211,23 @@ all_workflow = app.list_workflow()
 # List all workflow in community filtered by description
 all_face_community_workflows = App().list_workflows(filter_by={"query": "face"}, only_in_app=False) # Get all face related workflows
 ```
+#### Workflow Create
+Create a new workflow specified by a yaml config file.
+```python
+# Note: CLARIFAI_PAT must be set as env variable.
+from clarifai.client.app import App
+app = App(app_id="app_id", user_id="user_id")
+workflow = app.create_workflow(config_filepath="config.yml")
+```
 
+#### Workflow Export
+Export an existing workflow from Clarifai as a local yaml file.
+```python
+# Note: CLARIFAI_PAT must be set as env variable.
+from clarifai.client.workflow import Workflow
+workflow = Workflow("https://clarifai.com/clarifai/main/workflows/Demographics")
+workflow.export('demographics_workflow.yml')
+```
 
 ## More Examples
 See many more code examples in this [repo](https://github.com/Clarifai/examples).

{clarifai-9.8.1 → clarifai-9.9.0}/README.md

@@ -30,7 +30,7 @@ This is the official Python client for interacting with our powerful [API](https
 
 **Clarifai Community**: [https://clarifai.com/explore](https://clarifai.com/explore)
 
-**Python SDK Docs**: [https://clarifai-python.readthedocs.io/en/latest/index.html](https://clarifai-python.readthedocs.io/en/latest/index.html)
+**Python SDK Docs**: [https://docs.clarifai.com/python-sdk/api-reference](https://docs.clarifai.com/python-sdk/api-reference)
 
 
 ---
@@ -195,7 +195,23 @@ all_workflow = app.list_workflow()
 # List all workflow in community filtered by description
 all_face_community_workflows = App().list_workflows(filter_by={"query": "face"}, only_in_app=False) # Get all face related workflows
 ```
+#### Workflow Create
+Create a new workflow specified by a yaml config file.
+```python
+# Note: CLARIFAI_PAT must be set as env variable.
+from clarifai.client.app import App
+app = App(app_id="app_id", user_id="user_id")
+workflow = app.create_workflow(config_filepath="config.yml")
+```
 
+#### Workflow Export
+Export an existing workflow from Clarifai as a local yaml file.
+```python
+# Note: CLARIFAI_PAT must be set as env variable.
+from clarifai.client.workflow import Workflow
+workflow = Workflow("https://clarifai.com/clarifai/main/workflows/Demographics")
+workflow.export('demographics_workflow.yml')
+```
 
 ## More Examples
 See many more code examples in this [repo](https://github.com/Clarifai/examples).

{clarifai-9.8.1 → clarifai-9.9.0}/clarifai/client/app.py

@@ -1,5 +1,8 @@
+import os
+import uuid
 from typing import Any, Dict, List
 
+import yaml
 from clarifai_grpc.grpc.api import resources_pb2, service_pb2
 from clarifai_grpc.grpc.api.status import status_code_pb2
 from google.protobuf.json_format import MessageToDict
@@ -10,22 +13,30 @@ from clarifai.client.input import Inputs
 from clarifai.client.lister import Lister
 from clarifai.client.model import Model
 from clarifai.client.module import Module
+from clarifai.client.search import Search
 from clarifai.client.workflow import Workflow
 from clarifai.errors import UserError
 from clarifai.urls.helper import ClarifaiUrlHelper
-from clarifai.utils.logging import get_logger
+from clarifai.utils.logging import display_workflow_tree, get_logger
+from clarifai.workflows.utils import get_yaml_output_info_proto, is_same_yaml_model
+from clarifai.workflows.validate import validate
 
 
 class App(Lister, BaseClient):
   """App is a class that provides access to Clarifai API endpoints related to App information."""
 
-  def __init__(self, url_init: str = "", app_id: str = "", **kwargs):
+  def __init__(self,
+               url_init: str = "",
+               app_id: str = "",
+               base_url: str = "https://api.clarifai.com",
+               **kwargs):
     """Initializes an App object.
 
     Args:
         url_init (str): The URL to initialize the app object.
         app_id (str): The App ID for the App to interact with.
-        **kwargs: Additional keyword arguments to be passed to the ClarifaiAuthHelper.
+        base_url (str): Base API url. Default "https://api.clarifai.com"
+        **kwargs: Additional keyword arguments to be passed to the App.
             - name (str): The name of the app.
             - description (str): The description of the app.
     """
@@ -37,7 +48,7 @@ class App(Lister, BaseClient):
     self.kwargs = {**kwargs, 'id': app_id}
     self.app_info = resources_pb2.App(**self.kwargs)
     self.logger = get_logger(logger_level="INFO", name=__name__)
-    BaseClient.__init__(self, user_id=self.user_id, app_id=self.id)
+    BaseClient.__init__(self, user_id=self.user_id, app_id=self.id, base=base_url)
     Lister.__init__(self)
 
   def list_datasets(self) -> List[Dataset]:
@@ -236,30 +247,96 @@ class App(Lister, BaseClient):
 
     return Model(model_id=model_id, **kwargs)
 
-  def create_workflow(self, workflow_id: str, **kwargs) -> Workflow:
+  def create_workflow(self,
+                      config_filepath: str,
+                      generate_new_id: bool = False,
+                      display: bool = True) -> Workflow:
     """Creates a workflow for the app.
 
     Args:
-        workflow_id (str): The workflow ID for the workflow to create.
-        **kwargs: Additional keyword arguments to be passed to the workflow.
+        config_filepath (str): The path to the yaml workflow config file.
+        generate_new_id (bool): If True, generate a new workflow ID.
+        display (bool): If True, display the workflow nodes tree.
 
     Returns:
-        Workflow: A Workflow object for the specified workflow ID.
+        Workflow: A Workflow object for the specified workflow config.
 
     Example:
         >>> from clarifai.client.app import App
         >>> app = App(app_id="app_id", user_id="user_id")
-        >>> workflow = app.create_workflow(workflow_id="workflow_id")
+        >>> workflow = app.create_workflow(config_filepath="config.yml")
     """
+    if not os.path.exists(config_filepath):
+      raise UserError(f"Workflow config file not found at {config_filepath}")
+
+    with open(config_filepath, 'r') as file:
+      data = yaml.safe_load(file)
+
+    data = validate(data)
+    workflow = data['workflow']
+
+    # Get all model objects from the workflow nodes.
+    all_models = []
+    for node in workflow['nodes']:
+      output_info = get_yaml_output_info_proto(node['model'].get('output_info', None))
+      try:
+        model = self.model(
+            node['model']['model_id'],
+            node['model'].get('model_version_id', ""),
+            user_id=node['model'].get('user_id', ""),
+            app_id=node['model'].get('app_id', ""))
+      except Exception as e:
+        if "Model does not exist" in str(e):
+          model = self.create_model(
+              **{k: v
+                 for k, v in node['model'].items() if k != 'output_info'})
+          model_version = model.create_model_version(output_info=output_info)
+          all_models.append(model_version.model_info)
+          continue
+
+      # If the model version ID is specified, or if the yaml model is the same as the one in the api
+      if node["model"].get("model_version_id", "") or is_same_yaml_model(
+          model.model_info, node["model"]):
+        all_models.append(model.model_info)
+      else:  # Create a new model version
+        model = model.create_model_version(output_info=output_info)
+        all_models.append(model.model_info)
+
+    # Convert nodes to resources_pb2.WorkflowNodes.
+    nodes = []
+    for i, yml_node in enumerate(workflow['nodes']):
+      node = resources_pb2.WorkflowNode(
+          id=yml_node['id'],
+          model=all_models[i],
+      )
+      # Add node inputs if they exist, i.e. if these nodes do not connect directly to the input.
+      if yml_node.get("node_inputs"):
+        for ni in yml_node.get("node_inputs"):
+          node.node_inputs.append(resources_pb2.NodeInput(node_id=ni['node_id']))
+      nodes.append(node)
+
+    workflow_id = workflow['id']
+    if generate_new_id:
+      workflow_id = str(uuid.uuid4())
+
+    # Create the workflow.
     request = service_pb2.PostWorkflowsRequest(
-        user_app_id=self.user_app_id, workflows=[resources_pb2.Workflow(id=workflow_id, **kwargs)])
+        user_app_id=self.user_app_id,
+        workflows=[resources_pb2.Workflow(id=workflow_id, nodes=nodes)])
+
     response = self._grpc_request(self.STUB.PostWorkflows, request)
     if response.status.code != status_code_pb2.SUCCESS:
       raise Exception(response.status)
     self.logger.info("\nWorkflow created\n%s", response.status)
-    kwargs.update({'app_id': self.id, 'user_id': self.user_id})
 
-    return Workflow(workflow_id=workflow_id, **kwargs)
+    dict_response = MessageToDict(response, preserving_proto_field_name=True)
+    # Display the workflow nodes tree.
+    if display:
+      display_workflow_tree(dict_response["workflows"][0]["nodes"])
+    kwargs = self.process_response_keys(dict_response[list(dict_response.keys())[1]][0],
+                                        "workflow")
+
+    return Workflow(**kwargs)
 
   def create_module(self, module_id: str, description: str, **kwargs) -> Module:
     """Creates a module for the app.
@@ -329,8 +406,16 @@ class App(Lister, BaseClient):
         >>> app = App(app_id="app_id", user_id="user_id")
         >>> model_v1 = app.model(model_id="model_id", model_version_id="model_version_id")
     """
-    request = service_pb2.GetModelRequest(
-        user_app_id=self.user_app_id, model_id=model_id, version_id=model_version_id)
+    # Change user_app_id based on whether user_id or app_id is specified.
+    if kwargs.get("user_id") or kwargs.get("app_id"):
+      request = service_pb2.GetModelRequest(
+          user_app_id=self.auth_helper.get_user_app_id_proto(
+              kwargs.get("user_id"), kwargs.get("app_id")),
+          model_id=model_id,
+          version_id=model_version_id)
+    else:
+      request = service_pb2.GetModelRequest(
+          user_app_id=self.user_app_id, model_id=model_id, version_id=model_version_id)
     response = self._grpc_request(self.STUB.GetModel, request)
 
     if response.status.code != status_code_pb2.SUCCESS:
@@ -471,6 +556,22 @@ class App(Lister, BaseClient):
       raise Exception(response.status)
     self.logger.info("\nModule Deleted\n%s", response.status)
 
+  def search(self, **kwargs) -> Model:
+    """Returns a Search object for the user and app ID.
+
+    Args:
+        see the Search class in clarifai.client.search for kwargs.
+
+    Returns:
+        Search: A Search object for the user and app ID.
+
+    Example:
+        >>> from clarifai.client.app import App
+        >>> app = App(app_id="app_id", user_id="user_id")
+        >>> search_client = app.search(top_k=12, metric="euclidean")
+    """
+    return Search(**kwargs)
+
   def __getattr__(self, name):
     return getattr(self.app_info, name)
 

{clarifai-9.8.1/clarifai_utils → clarifai-9.9.0/clarifai}/client/base.py

@@ -2,7 +2,7 @@ import os
 from datetime import datetime
 from typing import Any, Callable
 
-from google.protobuf.json_format import MessageToDict  # noqa
+from google.protobuf import struct_pb2
 from google.protobuf.timestamp_pb2 import Timestamp
 from google.protobuf.wrappers_pb2 import BoolValue
 
@@ -71,7 +71,10 @@ class BaseClient:
     try:
       datetime_obj = datetime.strptime(date_str, '%Y-%m-%dT%H:%M:%S.%fZ')
     except ValueError:
-      datetime_obj = datetime.strptime(date_str, '%Y-%m-%dT%H:%M:%SZ')
+      try:
+        datetime_obj = datetime.strptime(date_str, '%Y-%m-%dT%H:%M:%SZ')
+      except ValueError:
+        return Timestamp()
 
     # Convert the datetime object to a Timestamp object
     timestamp_obj = Timestamp()
@@ -99,8 +102,12 @@ class BaseClient:
             value = self.convert_string_to_timestamp(value)
           elif key in ['workflow_recommended']:
             value = BoolValue(value=True)
-          elif key in ['metadata', 'fields_map', 'params']:
-            continue  # TODO Fix "app_duplication",proto struct
+          elif key in ['fields_map', 'params']:
+            value_s = struct_pb2.Struct()
+            value_s.update(value)
+            value = value_s
+          elif key in ['metadata']:
+            continue  # TODO Fix "app_duplication"
           new_item[key] = convert_recursive(value)
         return new_item
       elif isinstance(item, list):

{clarifai-9.8.1/clarifai_utils → clarifai-9.9.0/clarifai}/client/dataset.py

@@ -29,13 +29,18 @@ ClarifaiDatasetType = TypeVar('ClarifaiDatasetType', VisualClassificationDataset
 class Dataset(Lister, BaseClient):
   """Dataset is a class that provides access to Clarifai API endpoints related to Dataset information."""
 
-  def __init__(self, url_init: str = "", dataset_id: str = "", **kwargs):
+  def __init__(self,
+               url_init: str = "",
+               dataset_id: str = "",
+               base_url: str = "https://api.clarifai.com",
+               **kwargs):
     """Initializes a Dataset object.
 
     Args:
         url_init (str): The URL to initialize the dataset object.
         dataset_id (str): The Dataset ID within the App to interact with.
-        **kwargs: Additional keyword arguments to be passed to the ClarifaiAuthHelper.
+        base_url (str): Base API url. Default "https://api.clarifai.com"
+        **kwargs: Additional keyword arguments to be passed to the Dataset.
     """
     if url_init != "" and dataset_id != "":
       raise UserError("You can only specify one of url_init or dataset_id.")
@@ -51,7 +56,7 @@ class Dataset(Lister, BaseClient):
     self.chunk_size = 128  # limit max protos in a req
     self.task = None  # Upload dataset type
     self.input_object = Inputs(user_id=self.user_id, app_id=self.app_id)
-    BaseClient.__init__(self, user_id=self.user_id, app_id=self.app_id)
+    BaseClient.__init__(self, user_id=self.user_id, app_id=self.app_id, base=base_url)
     Lister.__init__(self)
 
   def _concurrent_annot_upload(self, annots: List[List[resources_pb2.Annotation]]

{clarifai-9.8.1 → clarifai-9.9.0}/clarifai/client/input.py

@@ -23,12 +23,18 @@ from clarifai.utils.misc import BackoffIterator, Chunker
 class Inputs(Lister, BaseClient):
   """Inputs is a class that provides access to Clarifai API endpoints related to Input information."""
 
-  def __init__(self, user_id: str = "", app_id: str = "", logger_level: str = "INFO", **kwargs):
+  def __init__(self,
+               user_id: str = "",
+               app_id: str = "",
+               logger_level: str = "INFO",
+               base_url: str = "https://api.clarifai.com",
+               **kwargs):
     """Initializes an Input object.
 
     Args:
         user_id (str): A user ID for authentication.
         app_id (str): An app ID for the application to interact with.
+        base_url (str): Base API url. Default "https://api.clarifai.com"
         **kwargs: Additional keyword arguments to be passed to the Input
     """
     self.user_id = user_id
@@ -36,7 +42,7 @@ class Inputs(Lister, BaseClient):
     self.kwargs = {**kwargs}
     self.input_info = resources_pb2.Input(**self.kwargs)
     self.logger = get_logger(logger_level=logger_level, name=__name__)
-    BaseClient.__init__(self, user_id=self.user_id, app_id=self.app_id)
+    BaseClient.__init__(self, user_id=self.user_id, app_id=self.app_id, base=base_url)
     Lister.__init__(self)
 
   def _get_proto(self,
@@ -121,8 +127,8 @@ class Inputs(Lister, BaseClient):
         Input: An Input object for the specified input ID.
 
     Example:
-        >>> from clarifai.client.input import Input
-        >>> input_obj = Input()
+        >>> from clarifai.client.input import Inputs
+        >>> input_obj = Inputs()
         >>> input_proto = input_obj.get_input_from_url(input_id = 'demo', image_url='https://samples.clarifai.com/metro-north.jpg')
     """
     if not any((image_url, video_url, audio_url, text_url)):
@@ -163,8 +169,8 @@ class Inputs(Lister, BaseClient):
         Input: An Input object for the specified input ID.
 
     Example:
-        >>> from clarifai.client.input import Input
-        >>> input_obj = Input()
+        >>> from clarifai.client.input import Inputs
+        >>> input_obj = Inputs()
         >>> input_proto = input_obj.get_input_from_file(input_id = 'demo', video_file='file_path')
     """
     if not any((image_file, video_file, audio_file, text_file)):
@@ -205,8 +211,8 @@ class Inputs(Lister, BaseClient):
         Input: An Input object for the specified input ID.
 
     Example:
-        >>> from clarifai.client.input import Input
-        >>> input_obj = Input()
+        >>> from clarifai.client.input import Inputs
+        >>> input_obj = Inputs()
         >>> image = open('demo.jpg', 'rb').read()
         >>> video = open('demo.mp4', 'rb').read()
         >>> input_proto = input_obj.get_input_from_bytes(input_id = 'demo',image_bytes =image, video_bytes=video)
@@ -240,8 +246,8 @@ class Inputs(Lister, BaseClient):
         list of Input: A list of Input objects for the specified folder.
 
     Example:
-        >>> from clarifai.client.input import Input
-        >>> input_obj = Input()
+        >>> from clarifai.client.input import Inputs
+        >>> input_obj = Inputs()
         >>> input_protos = input_obj.get_image_inputs_from_folder(folder_path='demo_folder')
     """
     input_protos = []
@@ -270,8 +276,8 @@ class Inputs(Lister, BaseClient):
         Text: An Input object for the specified input ID.
 
     Example:
-        >>> from clarifai.client.input import Input
-        >>> input_obj = Input()
+        >>> from clarifai.client.input import Inputs
+        >>> input_obj = Inputs()
         >>> input_protos = input_obj.get_text_input(input_id = 'demo', raw_text = 'This is a test')
     """
     text_pb = resources_pb2.Text(raw=raw_text)
@@ -296,8 +302,8 @@ class Inputs(Lister, BaseClient):
         inputs: List of inputs
 
     Example:
-        >>> from clarifai.client.input import Input
-        >>> input_obj = Input()
+        >>> from clarifai.client.input import Inputs
+        >>> input_obj = Inputs()
         >>> input_protos = input_obj.get_inputs_from_csv(csv_path='filepath', input_type='text', csv_type='raw')
     """
     input_protos = []
@@ -357,8 +363,8 @@ class Inputs(Lister, BaseClient):
         list of Input: A list of Input objects for the specified folder.
 
     Example:
-        >>> from clarifai.client.input import Input
-        >>> input_obj = Input()
+        >>> from clarifai.client.input import Inputs
+        >>> input_obj = Inputs()
         >>> input_protos = input_obj.get_text_inputs_from_folder(folder_path='demo_folder')
     """
     input_protos = []
@@ -385,8 +391,8 @@ class Inputs(Lister, BaseClient):
         An annotation object for the specified input ID.
 
     Example:
-        >>> from clarifai.client.input import Input
-        >>> input_obj = Input()
+        >>> from clarifai.client.input import Inputs
+        >>> input_obj = Inputs()
         >>> input_obj.get_annotation_proto(input_id='demo', label='demo', annotations=[x_min, y_min, x_max, y_max])
     """
     if not isinstance(annotations, list):
@@ -424,8 +430,8 @@ class Inputs(Lister, BaseClient):
         An annotation object for the specified input ID.
 
     Example:
-        >>> from clarifai.client.input import Input
-        >>> input_obj = Input()
+        >>> from clarifai.client.input import Inputs
+        >>> input_obj = Inputs()
         >>> input_obj.get_mask_proto(input_id='demo', label='demo', polygons=[[[x,y],...,[x,y]],...])
     """
     if not isinstance(polygons, list):
@@ -471,8 +477,8 @@ class Inputs(Lister, BaseClient):
         input_job_id: job id for the upload request.
 
     Example:
-        >>> from clarifai.client.input import Input
-        >>> input_obj = Input(user_id = 'user_id', app_id = 'demo_app')
+        >>> from clarifai.client.input import Inputs
+        >>> input_obj = Inputs(user_id = 'user_id', app_id = 'demo_app')
         >>> input_obj.upload_from_url(input_id='demo', image_url='https://samples.clarifai.com/metro-north.jpg')
     """
     input_pb = self.get_input_from_url(input_id, image_url, video_url, audio_url, text_url,
@@ -501,8 +507,8 @@ class Inputs(Lister, BaseClient):
         input_job_id: job id for the upload request.
 
     Example:
-        >>> from clarifai.client.input import Input
-        >>> input_obj = Input(user_id = 'user_id', app_id = 'demo_app')
+        >>> from clarifai.client.input import Inputs
+        >>> input_obj = Inputs(user_id = 'user_id', app_id = 'demo_app')
         >>> input_obj.upload_from_file(input_id='demo', audio_file='demo.mp3')
     """
     input_pb = self.get_input_from_file(input_id, image_file, video_file, audio_file, text_file,
@@ -531,8 +537,8 @@ class Inputs(Lister, BaseClient):
         input_job_id: job id for the upload request.
 
     Example:
-        >>> from clarifai.client.input import Input
-        >>> input_obj = Input(user_id = 'user_id', app_id = 'demo_app')
+        >>> from clarifai.client.input import Inputs
+        >>> input_obj = Inputs(user_id = 'user_id', app_id = 'demo_app')
         >>> image = open('demo.jpg', 'rb').read()
         >>> input_obj.upload_from_bytes(input_id='demo', image_bytes=image)
     """
@@ -553,8 +559,8 @@ class Inputs(Lister, BaseClient):
         input_job_id (str): job id for the upload request.
 
     Example:
-        >>> from clarifai.client.input import Input
-        >>> input_obj = Input(user_id = 'user_id', app_id = 'demo_app')
+        >>> from clarifai.client.input import Inputs
+        >>> input_obj = Inputs(user_id = 'user_id', app_id = 'demo_app')
         >>> input_obj.upload_text(input_id = 'demo', raw_text = 'This is a test')
     """
     input_pb = self._get_proto(

{clarifai-9.8.1/clarifai_utils → clarifai-9.9.0/clarifai}/client/model.py

@@ -5,6 +5,7 @@ from typing import Dict, 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.lister import Lister
@@ -22,6 +23,7 @@ class Model(Lister, BaseClient):
                model_id: str = "",
                model_version: Dict = {'id': ""},
                output_config: Dict = {'min_value': 0},
+               base_url: str = "https://api.clarifai.com",
                **kwargs):
     """Initializes a Model object.
 
@@ -34,7 +36,8 @@ class Model(Lister, BaseClient):
           max_concepts (int): The maximum number of concepts to return.
           select_concepts (list[Concept]): The concepts to select.
           sample_ms (int): The number of milliseconds to sample.
-        **kwargs: Additional keyword arguments to be passed to the ClarifaiAuthHelper.
+        base_url (str): Base API url. Default "https://api.clarifai.com"
+        **kwargs: Additional keyword arguments to be passed to the Model.
     """
     if url_init != "" and model_id != "":
       raise UserError("You can only specify one of url_init or model_id.")
@@ -49,9 +52,75 @@ class Model(Lister, BaseClient):
                    'output_info': {'output_config': output_config}}
     self.model_info = resources_pb2.Model(**self.kwargs)
     self.logger = get_logger(logger_level="INFO")
-    BaseClient.__init__(self, user_id=self.user_id, app_id=self.app_id)
+    BaseClient.__init__(self, user_id=self.user_id, app_id=self.app_id, base=base_url)
     Lister.__init__(self)
 
+  def create_model_version(self, **kwargs) -> 'Model':
+    """Creates a model version for the Model.
+
+    Args:
+        **kwargs: Additional keyword arguments to be passed to Model Version.
+          - description (str): The description of the model version.
+          - concepts (list[Concept]): The concepts to associate with the model version.
+          - output_info (resources_pb2.OutputInfo(): The output info to associate with the model version.
+
+    Returns:
+        Model: A Model object for the specified model ID.
+
+    Example:
+        >>> from clarifai.client.model import Model
+        >>> model = Model("model_url")
+                    or
+        >>> model = Model(model_id='model_id', user_id='user_id', app_id='app_id')
+        >>> model_version = model.create_model_version(description='model_version_description')
+    """
+    request = service_pb2.PostModelVersionsRequest(
+        user_app_id=self.user_app_id,
+        model_id=self.id,
+        model_versions=[resources_pb2.ModelVersion(**kwargs)])
+
+    response = self._grpc_request(self.STUB.PostModelVersions, request)
+    if response.status.code != status_code_pb2.SUCCESS:
+      raise Exception(response.status)
+    self.logger.info("\nModel Version created\n%s", response.status)
+
+    kwargs.update({'app_id': self.app_id, 'user_id': self.user_id})
+    dict_response = MessageToDict(response, preserving_proto_field_name=True)
+    kwargs = self.process_response_keys(dict_response['model'], 'model')
+
+    return Model(**kwargs)
+
+  def list_versions(self) -> List['Model']:
+    """Lists all the versions for the model.
+
+    Returns:
+        List[Model]: A list of Model objects for the versions of the model.
+
+    Example:
+        >>> from clarifai.client.model import Model
+        >>> 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')
+        >>> all_model_versions = model.list_versions()
+    """
+    request_data = dict(
+        user_app_id=self.user_app_id,
+        model_id=self.id,
+        per_page=self.default_page_size,
+    )
+    all_model_versions_info = list(
+        self.list_all_pages_generator(self.STUB.ListModelVersions,
+                                      service_pb2.ListModelVersionsRequest, request_data))
+
+    for model_version_info in all_model_versions_info:
+      model_version_info['id'] = model_version_info['model_version_id']
+      del model_version_info['model_version_id']
+
+    return [
+        Model(model_id=self.id, **dict(self.kwargs, model_version=model_version_info))
+        for model_version_info in all_model_versions_info
+    ]
+
   def predict(self, inputs: List[Input]):
     """Predicts the model based on the given inputs.
 

{clarifai-9.8.1/clarifai_utils → clarifai-9.9.0/clarifai}/client/module.py

@@ -16,6 +16,7 @@ class Module(Lister, BaseClient):
                url_init: str = "",
                module_id: str = "",
                module_version: Dict = {'id': ""},
+               base_url: str = "https://api.clarifai.com",
                **kwargs):
     """Initializes a Module object.
 
@@ -23,7 +24,8 @@ class Module(Lister, BaseClient):
             url_init (str): The URL to initialize the module object.
             module_id (str): The Module ID to interact with.
             module_version (dict): The Module Version to interact with.
-            **kwargs: Additional keyword arguments to be passed to the ClarifaiAuthHelper.
+            base_url (str): Base API url. Default "https://api.clarifai.com"
+            **kwargs: Additional keyword arguments to be passed to the Module.
         """
     if url_init != "" and module_id != "":
       raise UserError("You can only specify one of url_init or module_id.")
@@ -38,7 +40,7 @@ class Module(Lister, BaseClient):
     self.kwargs = {**kwargs, 'id': module_id, 'module_version': module_version}
     self.module_info = resources_pb2.Module(**self.kwargs)
     self.logger = get_logger(logger_level="INFO")
-    BaseClient.__init__(self, user_id=self.user_id, app_id=self.app_id)
+    BaseClient.__init__(self, user_id=self.user_id, app_id=self.app_id, base=base_url)
     Lister.__init__(self)
 
   def list_versions(self) -> List['Module']: