clarifai 10.9.5__tar.gz → 10.10.1__tar.gz

{clarifai-10.9.5/clarifai.egg-info → clarifai-10.10.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: clarifai
-Version: 10.9.5
+Version: 10.10.1
 Summary: Clarifai Python SDK
 Home-page: https://github.com/Clarifai/clarifai-python
 Author: Clarifai

clarifai-10.10.1/clarifai/__init__.py

@@ -0,0 +1 @@
+__version__ = "10.10.1"

clarifai-10.10.1/clarifai/cli/README.md

@@ -0,0 +1,87 @@
+# Clarifai CLI
+
+## Overview
+
+Clarifai offers a user-friendly interface for deploying your local model into production with Clarifai, featuring:
+
+* A convenient command-line interface (CLI)
+* Easy implementation and testing in Python
+* No need for MLops expertise.
+
+## Compute Orchestration
+
+Quick example for deploying a `visual-classifier` model
+
+### Login
+
+First, login to cli using clarifai account details in a config file as shown below:
+
+```bash
+$ clarifai login --config <config-filepath>
+```
+
+### Setup
+
+To prepare for deployment step, we have to setup a Compute Cluster with Nodepool of required server config to deploy the model.
+
+So, First, create a new Compute Cluster
+```bash
+$ clarifai computecluster create --config <compute-cluster-config-filepath>
+```
+
+Then, create a new Nodepool in the created Compute Cluster
+```bash
+$ clarifai nodepool create --config <nodepool-config-filepath>
+```
+
+### Deployment
+
+After setup, we can deploy the `visual-classifier` model using a deployment config file as shown below:
+
+```bash
+$ clarifai deployment create --config <deployment-config-filepath>
+```
+
+### List Resources
+
+List out existing Compute Clusters:
+
+```bash
+$ clarifai computecluster list
+```
+
+List out existing Nodepools:
+
+```bash
+$ clarifai nodepool list --compute_cluster_id <compute-cluster-id>
+```
+
+List out existing Deployments:
+
+```bash
+$ clarifai deployment list --nodepool_id <nodepool-id>
+```
+
+### Delete Resources
+
+Delete existing Deployment:
+
+```bash
+$ clarifai deployment delete --nodepool_id <nodepool-id> --deployment_id <deployment-id>
+```
+
+Delete existing Nodepool:
+
+```bash
+$ clarifai nodepool delete --compute_cluster_id <compute-cluster-id> --nodepool_id <nodepool-id>
+```
+
+Delete existing Compute Clusters:
+
+```bash
+$ clarifai computecluster delete --compute_cluster_id <compute-cluster-id>
+```
+
+## Learn More
+
+* [Example Configs](https://github.com/Clarifai/examples/tree/main/ComputeOrchestration/configs)

{clarifai-10.9.5 → clarifai-10.10.1}/clarifai/cli/base.py

@@ -5,7 +5,38 @@ import click
 from ..utils.cli import dump_yaml, from_yaml, load_command_modules, set_base_url
 
 
-@click.group()
+class CustomMultiGroup(click.Group):
+
+  def group(self, *args, **kwargs):
+    """Behaves the same as `click.Group.group()` except if passed
+        a list of names, all after the first will be aliases for the first.
+        """
+
+    def decorator(f):
+      aliased_group = []
+      if isinstance(args[0], list):
+        # we have a list so create group aliases
+        _args = [args[0][0]] + list(args[1:])
+        for alias in args[0][1:]:
+          grp = super(CustomMultiGroup, self).group(alias, *args[1:], **kwargs)(f)
+          grp.short_help = "Alias for '{}'".format(_args[0])
+          aliased_group.append(grp)
+      else:
+        _args = args
+
+      # create the main group
+      grp = super(CustomMultiGroup, self).group(*_args, **kwargs)(f)
+
+      # for all of the aliased groups, share the main group commands
+      for aliased in aliased_group:
+        aliased.commands = grp.commands
+
+      return grp
+
+    return decorator
+
+
+@click.group(cls=CustomMultiGroup)
 @click.pass_context
 def cli(ctx):
   """Clarifai CLI"""
@@ -49,9 +80,16 @@ def login(ctx, config, env, user_id):
   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"])
+  elif 'user_id' in ctx.obj:
+    ctx.obj['user_id'] = ctx.obj.get('user_id', "")
     os.environ["CLARIFAI_USER_ID"] = ctx.obj['user_id']
+  elif 'CLARIFAI_USER_ID' in os.environ:
+    ctx.obj['user_id'] = os.environ["CLARIFAI_USER_ID"]
+  else:
+    user_id = click.prompt("Pass the User ID here", type=str)
+    os.environ["CLARIFAI_USER_ID"] = user_id
+    ctx.obj['user_id'] = user_id
+    click.echo("User ID saved successfully.")
 
   if env:
     ctx.obj['env'] = env

clarifai-10.10.1/clarifai/cli/compute_cluster.py

@@ -0,0 +1,51 @@
+import click
+from clarifai.cli.base import cli
+from clarifai.client.user import User
+from clarifai.utils.cli import display_co_resources
+
+
+@cli.group(['computecluster', 'cc'])
+def computecluster():
+  """Manage Compute Clusters: create, delete, list"""
+  pass
+
+
+@computecluster.command()
+@click.option(
+    '--config',
+    type=click.Path(exists=True),
+    required=True,
+    help='Path to the compute cluster config file.')
+@click.option(
+    '-cc_id',
+    '--compute_cluster_id',
+    required=False,
+    help='New Compute Cluster ID for the compute cluster to create.')
+@click.pass_context
+def create(ctx, config, compute_cluster_id):
+  """Create a new Compute Cluster with the given config file."""
+  user = User(user_id=ctx.obj['user_id'], pat=ctx.obj['pat'], base_url=ctx.obj['base_url'])
+  if compute_cluster_id:
+    user.create_compute_cluster(config, compute_cluster_id=compute_cluster_id)
+  else:
+    user.create_compute_cluster(config)
+
+
+@computecluster.command()
+@click.option('--page_no', required=False, help='Page number to list.', default=1)
+@click.option('--per_page', required=False, help='Number of items per page.', default=16)
+@click.pass_context
+def list(ctx, page_no, per_page):
+  """List all compute clusters for the user."""
+  user = User(user_id=ctx.obj['user_id'], pat=ctx.obj['pat'], base_url=ctx.obj['base_url'])
+  response = user.list_compute_clusters(page_no, per_page)
+  display_co_resources(response, "Compute Cluster")
+
+
+@computecluster.command()
+@click.option('-cc_id', '--compute_cluster_id', help='Compute Cluster ID of the user to delete.')
+@click.pass_context
+def delete(ctx, compute_cluster_id):
+  """Deletes a compute cluster for the user."""
+  user = User(user_id=ctx.obj['user_id'], pat=ctx.obj['pat'], base_url=ctx.obj['base_url'])
+  user.delete_compute_clusters([compute_cluster_id])

clarifai-10.10.1/clarifai/cli/deployment.py

@@ -0,0 +1,84 @@
+import click
+from clarifai.cli.base import cli
+from clarifai.client.nodepool import Nodepool
+from clarifai.utils.cli import display_co_resources, from_yaml
+
+
+@cli.group(['deployment', 'dpl'])
+def deployment():
+  """Manage Deployments: create, delete, list"""
+  pass
+
+
+@deployment.command()
+@click.option(
+    '-np_id',
+    '--nodepool_id',
+    required=False,
+    help='Nodepool ID for the Nodepool to interact with.')
+@click.option(
+    '--config',
+    type=click.Path(exists=True),
+    required=True,
+    help='Path to the deployment config file.')
+@click.option(
+    '-dpl_id',
+    '--deployment_id',
+    required=False,
+    help='New deployment ID for the deployment to create.')
+@click.pass_context
+def create(ctx, nodepool_id, config, deployment_id):
+  """Create a new Deployment with the given config file."""
+  if not nodepool_id:
+    deployment_config = from_yaml(config)
+    nodepool_id = deployment_config['deployment']['nodepools'][0]['id']
+
+  nodepool = Nodepool(
+      nodepool_id=nodepool_id,
+      user_id=ctx.obj['user_id'],
+      pat=ctx.obj['pat'],
+      base_url=ctx.obj['base_url'])
+  if deployment_id:
+    nodepool.create_deployment(config, deployment_id=deployment_id)
+  else:
+    nodepool.create_deployment(config)
+
+
+@deployment.command()
+@click.option(
+    '-np_id',
+    '--nodepool_id',
+    required=True,
+    help='Nodepool ID for the Nodepool to interact with.')
+@click.option('--page_no', required=False, help='Page number to list.', default=1)
+@click.option('--per_page', required=False, help='Number of items per page.', default=16)
+@click.pass_context
+def list(ctx, nodepool_id, page_no, per_page):
+  """List all deployments for the nodepool."""
+
+  nodepool = Nodepool(
+      nodepool_id=nodepool_id,
+      user_id=ctx.obj['user_id'],
+      pat=ctx.obj['pat'],
+      base_url=ctx.obj['base_url'])
+  response = nodepool.list_deployments(page_no=page_no, per_page=per_page)
+  display_co_resources(response, "Deployment")
+
+
+@deployment.command()
+@click.option(
+    '-np_id',
+    '--nodepool_id',
+    required=True,
+    help='Nodepool ID for the Nodepool to interact with.')
+@click.option('-dpl_id', '--deployment_id', help='Deployment ID of the nodepool to delete.')
+@click.pass_context
+def delete(ctx, nodepool_id, deployment_id):
+  """Deletes a deployment for the nodepool."""
+
+  nodepool = Nodepool(
+      nodepool_id=nodepool_id,
+      user_id=ctx.obj['user_id'],
+      pat=ctx.obj['pat'],
+      base_url=ctx.obj['base_url'])
+  nodepool.delete_deployments([deployment_id])

{clarifai-10.9.5 → clarifai-10.10.1}/clarifai/cli/model.py

@@ -2,7 +2,7 @@ import click
 from clarifai.cli.base import cli
 
 
-@cli.group()
+@cli.group(['model'])
 def model():
   """Manage models: upload, test locally"""
   pass

clarifai-10.10.1/clarifai/cli/nodepool.py

@@ -0,0 +1,91 @@
+import click
+from clarifai.cli.base import cli
+from clarifai.client.compute_cluster import ComputeCluster
+from clarifai.utils.cli import display_co_resources, dump_yaml, from_yaml
+
+
+@cli.group(['nodepool', 'np'])
+def nodepool():
+  """Manage Nodepools: create, delete, list"""
+  pass
+
+
+@nodepool.command()
+@click.option(
+    '-cc_id',
+    '--compute_cluster_id',
+    required=False,
+    help='Compute Cluster ID for the compute cluster to interact with.')
+@click.option(
+    '--config',
+    type=click.Path(exists=True),
+    required=True,
+    help='Path to the nodepool config file.')
+@click.option(
+    '-np_id', '--nodepool_id', required=False, help='New Nodepool ID for the nodepool to create.')
+@click.pass_context
+def create(ctx, compute_cluster_id, config, nodepool_id):
+  """Create a new Nodepool with the given config file."""
+
+  nodepool_config = from_yaml(config)
+  if not compute_cluster_id:
+    if 'compute_cluster' not in nodepool_config['nodepool']:
+      click.echo(
+          "Please provide a compute cluster ID either in the config file or using --compute_cluster_id flag",
+          err=True)
+      return
+    compute_cluster_id = nodepool_config['nodepool']['compute_cluster']['id']
+  else:
+    if 'compute_cluster' not in nodepool_config['nodepool']:
+      nodepool_config['nodepool']['compute_cluster']['id'] = compute_cluster_id
+      dump_yaml(config, nodepool_config)
+
+  compute_cluster = ComputeCluster(
+      compute_cluster_id=compute_cluster_id,
+      user_id=ctx.obj['user_id'],
+      pat=ctx.obj['pat'],
+      base_url=ctx.obj['base_url'])
+  if nodepool_id:
+    compute_cluster.create_nodepool(config, nodepool_id=nodepool_id)
+  else:
+    compute_cluster.create_nodepool(config)
+
+
+@nodepool.command()
+@click.option(
+    '-cc_id',
+    '--compute_cluster_id',
+    required=True,
+    help='Compute Cluster ID for the compute cluster to interact with.')
+@click.option('--page_no', required=False, help='Page number to list.', default=1)
+@click.option('--per_page', required=False, help='Number of items per page.', default=16)
+@click.pass_context
+def list(ctx, compute_cluster_id, page_no, per_page):
+  """List all nodepools for the user."""
+
+  compute_cluster = ComputeCluster(
+      compute_cluster_id=compute_cluster_id,
+      user_id=ctx.obj['user_id'],
+      pat=ctx.obj['pat'],
+      base_url=ctx.obj['base_url'])
+  response = compute_cluster.list_nodepools(page_no, per_page)
+  display_co_resources(response, "Nodepool")
+
+
+@nodepool.command()
+@click.option(
+    '-cc_id',
+    '--compute_cluster_id',
+    required=True,
+    help='Compute Cluster ID for the compute cluster to interact with.')
+@click.option('-np_id', '--nodepool_id', help='Nodepool ID of the user to delete.')
+@click.pass_context
+def delete(ctx, compute_cluster_id, nodepool_id):
+  """Deletes a nodepool for the user."""
+
+  compute_cluster = ComputeCluster(
+      compute_cluster_id=compute_cluster_id,
+      user_id=ctx.obj['user_id'],
+      pat=ctx.obj['pat'],
+      base_url=ctx.obj['base_url'])
+  compute_cluster.delete_nodepools([nodepool_id])

{clarifai-10.9.5 → clarifai-10.10.1}/clarifai/client/auth/helper.py

@@ -321,6 +321,7 @@ Additionally, these optional params are supported:
 
   def set_base(self, base: str):
     """ Set the base domain for the API. """
+    base = DEFAULT_BASE if base is None else base
     self._base = https_cache(base_https_cache, base)
 
   def set_ui(self, ui: str):

{clarifai-10.9.5 → clarifai-10.10.1}/clarifai/client/compute_cluster.py

@@ -102,12 +102,12 @@ class ComputeCluster(Lister, BaseClient):
       nodepool["visibility"] = resources_pb2.Visibility(**nodepool["visibility"])
     return nodepool
 
-  def create_nodepool(self, nodepool_id: str, config_filepath: str) -> Nodepool:
+  def create_nodepool(self, config_filepath: str, nodepool_id: str = None) -> Nodepool:
     """Creates a nodepool for the compute cluster.
 
     Args:
-        nodepool_id (str): The nodepool ID for the nodepool to create.
         config_filepath (str): The path to the nodepool config file.
+        nodepool_id (str): New nodepool ID for the nodepool to create.
 
     Returns:
         Nodepool: A Nodepool object for the specified nodepool ID.
@@ -115,7 +115,7 @@ class ComputeCluster(Lister, BaseClient):
     Example:
         >>> from clarifai.client.compute_cluster import ComputeCluster
         >>> compute_cluster = ComputeCluster(compute_cluster_id="compute_cluster_id", user_id="user_id")
-        >>> nodepool = compute_cluster.create_nodepool(nodepool_id="nodepool_id", config_filepath="config.yml")
+        >>> nodepool = compute_cluster.create_nodepool(config_filepath="config.yml")
     """
     if not os.path.exists(config_filepath):
       raise UserError(f"Nodepool config file not found at {config_filepath}")
@@ -123,7 +123,8 @@ class ComputeCluster(Lister, BaseClient):
     nodepool_config = self._process_nodepool_config(config_filepath)
 
     if 'id' in nodepool_config:
-      nodepool_id = nodepool_config['id']
+      if nodepool_id is None:
+        nodepool_id = nodepool_config['id']
       nodepool_config.pop('id')
 
     request = service_pb2.PostNodepoolsRequest(

{clarifai-10.9.5 → clarifai-10.10.1}/clarifai/client/dataset.py

@@ -9,6 +9,7 @@ from typing import Dict, Generator, List, Optional, Tuple, Type, TypeVar, Union
 
 import requests
 from clarifai_grpc.grpc.api import resources_pb2, service_pb2
+from clarifai_grpc.grpc.api.resources_pb2 import Input
 from clarifai_grpc.grpc.api.service_pb2 import MultiInputResponse
 from clarifai_grpc.grpc.api.status import status_code_pb2, status_pb2
 from google.protobuf.json_format import MessageToDict
@@ -190,6 +191,26 @@ class Dataset(Lister, BaseClient):
       }
       yield Dataset.from_auth_helper(self.auth_helper, **kwargs)
 
+  def list_inputs(self, page_no: int = None, per_page: int = None,
+                  input_type: str = None) -> Generator[Input, None, None]:
+    """Lists all the inputs for the dataset.
+
+    Args:
+        page_no (int): The page number to list.
+        per_page (int): The number of items per page.
+        input_type (str): The type of input to list. Options: 'image', 'video', 'audio', 'text'.
+
+    Yields:
+        Input: Input objects in the dataset.
+
+    Example:
+        >>> from clarifai.client.dataset import Dataset
+        >>> dataset = Dataset(dataset_id='dataset_id', user_id='user_id', app_id='app_id')
+        >>> all_dataset_inputs = list(dataset.list_inputs())
+    """
+    return self.input_object.list_inputs(
+        dataset_id=self.id, page_no=page_no, per_page=per_page, input_type=input_type)
+
   def __iter__(self):
     return iter(DatasetExportReader(archive_url=self.archive_zip()))
 

{clarifai-10.9.5 → clarifai-10.10.1}/clarifai/client/deployment.py

@@ -49,7 +49,8 @@ class Deployment(Lister, BaseClient):
     Returns:
         resources_pb2.RunnerSelector: A RunnerSelector object for the given deployment_id.
     """
-    return resources_pb2.RunnerSelector(deployment_id=deployment_id, user_id=user_id)
+    return resources_pb2.RunnerSelector(
+        deployment=resources_pb2.Deployment(id=deployment_id, user_id=user_id))
 
   def __getattr__(self, name):
     return getattr(self.deployment_info, name)

{clarifai-10.9.5 → clarifai-10.10.1}/clarifai/client/input.py

@@ -867,6 +867,33 @@ class Inputs(Lister, BaseClient):
       raise Exception(response.status)
     self.logger.info("\nInputs Deleted\n%s", response.status)
 
+  def delete_annotations(self, input_ids: List[str], annotation_ids: List[str] = []) -> None:
+    """Delete list of annotations of input objects from the app.
+
+    Args:
+        input_ids (Input): List of input objects for which annotations to delete.
+        annotation_ids (List[str]): List of annotation ids to delete.
+
+    Example:
+        >>> from clarifai.client.user import User
+        >>> input_obj = User(user_id="user_id").app(app_id="app_id").inputs()
+        >>> input_obj.delete_annotations(input_ids=['input_id_1', 'input_id_2'])
+
+    Note:
+        'annotation_ids' are optional but if the are provided, the number and order in
+        'annotation_ids' and 'input_ids' should match
+    """
+    if not isinstance(input_ids, list):
+      raise UserError("input_ids must be a list of input ids")
+    if annotation_ids and len(input_ids) != len(annotation_ids):
+      raise UserError("Number of provided annotation_ids and input_ids should match.")
+    request = service_pb2.DeleteAnnotationsRequest(
+        user_app_id=self.user_app_id, ids=annotation_ids, input_ids=input_ids)
+    response = self._grpc_request(self.STUB.DeleteAnnotations, request)
+    if response.status.code != status_code_pb2.SUCCESS:
+      raise Exception(response.status)
+    self.logger.info("\nAnnotations Deleted\n%s", response.status)
+
   def download_inputs(self, inputs: List[Input]) -> List[bytes]:
     """Download list of input objects from the app.
 

{clarifai-10.9.5 → clarifai-10.10.1}/clarifai/client/nodepool.py

@@ -136,12 +136,12 @@ class Nodepool(Lister, BaseClient):
     nodepool = resources_pb2.Nodepool(id=nodepool_id, compute_cluster=compute_cluster)
     return resources_pb2.RunnerSelector(nodepool=nodepool)
 
-  def create_deployment(self, deployment_id: str, config_filepath: str) -> Deployment:
+  def create_deployment(self, config_filepath: str, deployment_id: str = None) -> Deployment:
     """Creates a deployment for the nodepool.
 
     Args:
-        deployment_id (str): The deployment ID for the deployment to create.
         config_filepath (str): The path to the deployment config file.
+        deployment_id (str): New deployment ID for the deployment to create.
 
     Returns:
         Deployment: A Deployment object for the specified deployment ID.
@@ -149,7 +149,7 @@ class Nodepool(Lister, BaseClient):
     Example:
         >>> from clarifai.client.nodepool import Nodepool
         >>> nodepool = Nodepool(nodepool_id="nodepool_id", user_id="user_id")
-        >>> deployment = nodepool.create_deployment(deployment_id="deployment_id", config_filepath="config.yml")
+        >>> deployment = nodepool.create_deployment(config_filepath="config.yml")
     """
     if not os.path.exists(config_filepath):
       raise UserError(f"Deployment config file not found at {config_filepath}")
@@ -157,7 +157,8 @@ class Nodepool(Lister, BaseClient):
     deployment_config = self._process_deployment_config(config_filepath)
 
     if 'id' in deployment_config:
-      deployment_id = deployment_config['id']
+      if deployment_id is None:
+        deployment_id = deployment_config['id']
       deployment_config.pop('id')
 
     request = service_pb2.PostDeploymentsRequest(

{clarifai-10.9.5 → clarifai-10.10.1}/clarifai/client/user.py

@@ -222,13 +222,13 @@ class User(Lister, BaseClient):
       compute_cluster["visibility"] = resources_pb2.Visibility(**compute_cluster["visibility"])
     return compute_cluster
 
-  def create_compute_cluster(self, compute_cluster_id: str,
-                             config_filepath: str) -> ComputeCluster:
+  def create_compute_cluster(self, config_filepath: str,
+                             compute_cluster_id: str = None) -> ComputeCluster:
     """Creates a compute cluster for the user.
 
     Args:
-        compute_cluster_id (str): The compute cluster ID for the compute cluster to create.
         config_filepath (str): The path to the compute cluster config file.
+        compute_cluster_id (str): New compute cluster ID for the compute cluster to create.
 
     Returns:
         ComputeCluster: A Compute Cluster object for the specified compute cluster ID.
@@ -236,7 +236,7 @@ class User(Lister, BaseClient):
     Example:
         >>> from clarifai.client.user import User
         >>> client = User(user_id="user_id")
-        >>> compute_cluster = client.create_compute_cluster(compute_cluster_id="compute_cluster_id", config_filepath="config.yml")
+        >>> compute_cluster = client.create_compute_cluster(config_filepath="config.yml")
     """
     if not os.path.exists(config_filepath):
       raise UserError(f"Compute Cluster config file not found at {config_filepath}")
@@ -244,7 +244,8 @@ class User(Lister, BaseClient):
     compute_cluster_config = self._process_compute_cluster_config(config_filepath)
 
     if 'id' in compute_cluster_config:
-      compute_cluster_id = compute_cluster_config['id']
+      if compute_cluster_id is None:
+        compute_cluster_id = compute_cluster_config['id']
       compute_cluster_config.pop('id')
 
     request = service_pb2.PostComputeClustersRequest(

{clarifai-10.9.5 → clarifai-10.10.1}/clarifai/datasets/upload/multimodal.py

@@ -6,7 +6,6 @@ from google.protobuf.struct_pb2 import Struct
 
 from clarifai.client.input import Inputs
 from clarifai.datasets.upload.base import ClarifaiDataLoader, ClarifaiDataset
-from clarifai.utils.misc import get_uuid
 
 
 class MultiModalDataset(ClarifaiDataset):
@@ -36,7 +35,6 @@ class MultiModalDataset(ClarifaiDataset):
       image_bytes = data_item.image_bytes
       text = data_item.text
       labels = data_item.labels if isinstance(data_item.labels, list) else [data_item.labels]
-      id = get_uuid(8)
       input_id = f"{self.dataset_id}-{id}" if data_item.id is None else f"{self.dataset_id}-{str(data_item.id)}"
       if data_item.metadata is not None:
         metadata.update(data_item.metadata)

clarifai-10.9.5/clarifai/runners/dockerfile_template/Dockerfile.cpu.template → clarifai-10.10.1/clarifai/runners/dockerfile_template/Dockerfile.template

@@ -1,16 +1,16 @@
-ARG PYTHON_VERSION=${PYTHON_VERSION}
-FROM public.ecr.aws/docker/library/python:${PYTHON_VERSION}-slim-bookworm as build
+FROM --platform=$TARGETPLATFORM ${BASE_IMAGE} as build
 
-# Set the working directory to /app
-WORKDIR /app
+ENV DEBIAN_FRONTEND=noninteractive
 
+#############################
+# User specific requirements
+#############################
 COPY requirements.txt .
-# Install requirements and cleanup before leaving this line.
-# Note(zeiler): this could be in a future template as {{model_python_deps}}
-RUN python -m pip install -r requirements.txt && rm -rf /root/.cache
 
-# Install Clarifai SDK
-RUN python -m pip install clarifai
+# Install requirements and clarifai package and cleanup before leaving this line.
+# Note(zeiler): this could be in a future template as {{model_python_deps}}
+RUN pip install --no-cache-dir -r requirements.txt && \
+    pip install --no-cache-dir clarifai
 
 # These will be set by the templaing system.
 ENV CLARIFAI_PAT=${CLARIFAI_PAT}
@@ -20,12 +20,22 @@ ENV CLARIFAI_NODEPOOL_ID=${CLARIFAI_NODEPOOL_ID}
 ENV CLARIFAI_COMPUTE_CLUSTER_ID=${CLARIFAI_COMPUTE_CLUSTER_ID}
 ENV CLARIFAI_API_BASE=${CLARIFAI_API_BASE}
 
+# Set the NUMBA cache dir to /tmp
+ENV NUMBA_CACHE_DIR=/tmp/numba_cache
+ENV HOME=/tmp
+
+# Set the working directory to /app
+WORKDIR /app
+
 # Copy the current folder into /app/model_dir that the SDK will expect.
+# Note(zeiler): would be nice to exclude checkpoints in case they were pre-downloaded.
 COPY . /app/model_dir/${name}
 
 # Add the model directory to the python path.
-ENV PYTHONPATH "${PYTHONPATH}:/app/model_dir/${name}"
+ENV PYTHONPATH=${PYTHONPATH}:/app/model_dir/${name}
+
+ENTRYPOINT ["python", "-m", "clarifai.runners.server"]
 
 # Finally run the clarifai entrypoint to start the runner loop and local dev server.
 # Note(zeiler): we may want to make this a clarifai CLI call.
-CMD ["python", "-m", "clarifai.runners.server", "--model_path", "/app/model_dir/${name}"]
+CMD ["--model_path", "/app/model_dir/main"]