clarifai 11.2.4rc2__py3-none-any.whl → 11.3.0__py3-none-any.whl

clarifai/utils/protobuf.py

@@ -0,0 +1,143 @@
+from google.protobuf import struct_pb2, symbol_database, wrappers_pb2
+from google.protobuf.descriptor import FieldDescriptor
+from google.protobuf.message import Message
+from google.protobuf.timestamp_pb2 import Timestamp
+
+from clarifai.utils.logging import logger
+
+# Initialize the Protobuf symbol database for dynamic message type resolution
+db = symbol_database.Default()
+
+# Define standard wrapper types for Protobuf well-known types
+WRAPPER_TYPES = {
+    wrappers_pb2.Int32Value,
+    wrappers_pb2.Int64Value,
+    wrappers_pb2.StringValue,
+    wrappers_pb2.UInt32Value,
+    wrappers_pb2.UInt64Value,
+}
+
+
+def dict_to_protobuf(pb_obj: Message, data: dict) -> None:
+  """Recursively convert a nested dictionary to a Protobuf message object.
+
+    Args:
+        pb_obj: The target Protobuf message instance to populate.
+        data: Source dictionary containing the data to convert.
+    """
+  for field, value in data.items():
+    if field not in pb_obj.DESCRIPTOR.fields_by_name:
+      logger.warning(f"Ignoring unknown field '{field}' in message '{pb_obj.DESCRIPTOR.name}'")
+      continue
+
+    field_descriptor = pb_obj.DESCRIPTOR.fields_by_name[field]
+
+    try:
+      # Handle repeated fields (lists)
+      if field_descriptor.label == FieldDescriptor.LABEL_REPEATED:
+        _handle_repeated_field(pb_obj, field_descriptor, field, value)
+
+      # Handle message fields (nested messages)
+      elif field_descriptor.type == FieldDescriptor.TYPE_MESSAGE:
+        _handle_message_field(pb_obj, field_descriptor, field, value)
+
+      # Handle scalar fields
+      else:
+        if value:
+          setattr(pb_obj, field, value)
+
+    except Exception as e:
+      logger.error(f"Error processing field '{field}': {str(e)}")
+      raise
+
+
+def _handle_repeated_field(pb_obj: Message, field_descriptor: FieldDescriptor, field: str,
+                           value: list) -> None:
+  """Process repeated fields (both scalar and message types)."""
+  if not isinstance(value, list):
+    logger.warning(f"Expected list for repeated field '{field}', got {type(value).__name__}")
+    return
+
+  repeated_field = getattr(pb_obj, field)
+
+  # Handle repeated message fields
+  if field_descriptor.type == FieldDescriptor.TYPE_MESSAGE:
+    for item in value:
+      if isinstance(item, dict):
+        item_msg = repeated_field.add()
+        dict_to_protobuf(item_msg, item)
+      elif isinstance(item, Message):
+        repeated_field.add().CopyFrom(item)
+      else:
+        logger.warning(f"Invalid type {type(item).__name__} in repeated message field '{field}'")
+  else:
+    # Handle repeated scalar fields
+    try:
+      repeated_field.extend(value)
+    except TypeError as e:
+      logger.error(f"Type mismatch in repeated scalar field '{field}': {str(e)}")
+      raise
+
+
+def _handle_message_field(pb_obj: Message, field_descriptor: FieldDescriptor, field: str,
+                          value: object) -> None:
+  """Process message-type fields including special types."""
+  msg_class = db.GetSymbol(field_descriptor.message_type.full_name)
+  target_field = getattr(pb_obj, field)
+
+  # Handle special message types
+  if msg_class is Timestamp:
+    _set_timestamp_value(target_field, value)
+  elif msg_class is struct_pb2.Struct:
+    _set_struct_value(target_field, value)
+  elif msg_class in WRAPPER_TYPES:
+    _set_wrapper_value(target_field, msg_class, value)
+  else:
+    # Handle nested messages
+    if isinstance(value, dict):
+      nested_pb = msg_class()
+      dict_to_protobuf(nested_pb, value)
+      target_field.CopyFrom(nested_pb)
+    elif isinstance(value, Message):
+      target_field.CopyFrom(value)
+    else:
+      logger.warning(f"Invalid type {type(value).__name__} for message field '{field}'")
+
+
+def _set_timestamp_value(target_field: Message, value: object) -> None:
+  """Set timestamp value from dict or numeric timestamp."""
+  ts = Timestamp()
+  if isinstance(value, dict):
+    ts.seconds = value.get('seconds', 0)
+    ts.nanos = value.get('nanos', 0)
+  elif isinstance(value, (int, float)):
+    ts.seconds = int(value)
+    ts.nanos = int((value - ts.seconds) * 1e9)
+  else:
+    logger.warning(f"Unsupported timestamp format: {type(value).__name__}")
+  target_field.CopyFrom(ts)
+
+
+def _set_struct_value(target_field: Message, value: object) -> None:
+  """Convert dictionary to Protobuf Struct."""
+  if not isinstance(value, dict):
+    logger.warning(f"Expected dict for Struct field, got {type(value).__name__}")
+    return
+
+  struct = struct_pb2.Struct()
+  try:
+    struct.update(value)
+  except ValueError as e:
+    logger.error(f"Invalid value in Struct: {str(e)}")
+    raise
+  target_field.CopyFrom(struct)
+
+
+def _set_wrapper_value(target_field: Message, wrapper_type: type, value: object) -> None:
+  """Set value for wrapper types (e.g., Int32Value)."""
+  try:
+    wrapper = wrapper_type(value=value)
+  except TypeError as e:
+    logger.error(f"Invalid value for {wrapper_type.__name__}: {str(e)}")
+    raise
+  target_field.CopyFrom(wrapper)

{clarifai-11.2.4rc2.dist-info → clarifai-11.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: clarifai
-Version: 11.2.4rc2
+Version: 11.3.0
 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>=11.2.6
+Requires-Dist: clarifai-grpc>=11.3.2
 Requires-Dist: clarifai-protocol>=0.0.22
 Requires-Dist: numpy>=1.22.0
 Requires-Dist: tqdm>=4.65.0
@@ -42,6 +42,7 @@ Dynamic: description
 Dynamic: description-content-type
 Dynamic: home-page
 Dynamic: license
+Dynamic: license-file
 Dynamic: provides-extra
 Dynamic: requires-dist
 Dynamic: requires-python

clarifai-11.3.0.dist-info/RECORD

@@ -0,0 +1,107 @@
+clarifai/__init__.py,sha256=AKQVpPKUjXA31bxC6zOaWuLMOUf9UCfopNi6s5UU5_g,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/README.md,sha256=YGApHfeUyu5P0Pdth-mqQCQftWHDxz6bugDlvDXDhOE,1942
+clarifai/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+clarifai/cli/__main__.py,sha256=utJ2F40cl0jPHcYdTlGZRqpPfZ0CtVYB-8Ft0b2fWD4,72
+clarifai/cli/base.py,sha256=ZG1Ial89wGVqt4bSFYmIoRH_OZtw2yxlq3Ihp9e025U,7592
+clarifai/cli/compute_cluster.py,sha256=dma_dgzSg5pYQcxlWhoKYdMFPO2KS7HzsN81ploNB-E,2083
+clarifai/cli/deployment.py,sha256=wBWkYfLr6W73LScF6NlQM-reXWUP38VfAudltjpnFpk,3822
+clarifai/cli/model.py,sha256=M1dEbR7t0Ay9cgUe73knIvZ5y1UJedM8FoNrmuYxWOA,12789
+clarifai/cli/nodepool.py,sha256=7nTNd9T-UEPFE36vbxwdUHu2B3NECbN-lSTb_DQYw18,3874
+clarifai/client/__init__.py,sha256=xI1U0l5AZdRThvQAXCLsd9axxyFzXXJ22m8LHqVjQRU,662
+clarifai/client/app.py,sha256=FnKvKksYZwdry0e4Obh-trdSO1mv6QcVAa3kzKiQMpU,38340
+clarifai/client/base.py,sha256=hSHOqkXbSKyaRDeylMMnkhUHCAHhEqno4KI0CXGziBA,7536
+clarifai/client/compute_cluster.py,sha256=EvW9TJjPvInUlggfg1A98sxoWH8_PY5rCVXZhsj6ac0,8705
+clarifai/client/dataset.py,sha256=6DqfXQtiHoOJq0vPw2fz5ATo5zxQMmlhKG9IBwhnXeU,32078
+clarifai/client/deployment.py,sha256=I3UJLu2NB1GlKZaIFTI_q-3P5ZTtIoNEuKxB-oHMvD4,2682
+clarifai/client/input.py,sha256=obMAHMDU1OwfXZ8KraOnGFlWzlW-3F7Ob_2lcOQMlhY,46339
+clarifai/client/lister.py,sha256=03KGMvs5RVyYqxLsSrWhNc34I8kiF1Ph0NeyEwu7nMU,2082
+clarifai/client/model.py,sha256=Zm_GoTNy-HL6zmld_PeKGqnQgaLhLPqlNb4BcFBX8bY,77125
+clarifai/client/model_client.py,sha256=X0RndVfTdfa4PZaOtSick0VKC3AkQakT-qrkPqaFKuI,19971
+clarifai/client/module.py,sha256=FTkm8s9m-EaTKN7g9MnLhGJ9eETUfKG7aWZ3o1RshYs,4204
+clarifai/client/nodepool.py,sha256=la3vTFrO4LX8zm2eQ5jqf2L0-kQ63Dano8FibadoZbk,10152
+clarifai/client/search.py,sha256=GaPWN6JmTQGZaCHr6U1yv0zqR6wKFl7i9IVLg2ul1CI,14254
+clarifai/client/user.py,sha256=WOv66ww2wx9isIge3V-xTHCkqC6akl2jEGAxzT_Ugfs,17642
+clarifai/client/workflow.py,sha256=5VjZ2D8cudLznR8yhrwNawOmjxUhkJllZMKM6pn-4gs,12170
+clarifai/client/auth/__init__.py,sha256=7EwR0NrozkAUwpUnCsqXvE_p0wqx_SelXlSpKShKJK0,136
+clarifai/client/auth/helper.py,sha256=Ngw5IDkOWvnOz5YwViVk55z3mC52MyezLc0G3WxLqok,14643
+clarifai/client/auth/register.py,sha256=2CMdBsoVLoTfjyksE6j7BM2tiEc73WKYvxnwDDgNn1k,536
+clarifai/client/auth/stub.py,sha256=FrA5QpEFY4sK6fvPVhTg19ROLHkYSRo343nNqiVE7xw,4963
+clarifai/constants/base.py,sha256=ogmFSZYoF0YhGjHg5aiOc3MLqPr_poKAls6xaD0_C3U,89
+clarifai/constants/dataset.py,sha256=vjK3IlgXu31HycuvjRSzEQSqhU6xfj5TIgo6IpyUWoc,609
+clarifai/constants/input.py,sha256=WcHwToUVIK9ItAhDefaSohQHCLNeR55PSjZ0BFnoZ3U,28
+clarifai/constants/model.py,sha256=Um1hLfMFlh5R_vtP3Z6P-o6zon-tdbLcKVIl4PucrV4,438
+clarifai/constants/rag.py,sha256=WcHwToUVIK9ItAhDefaSohQHCLNeR55PSjZ0BFnoZ3U,28
+clarifai/constants/search.py,sha256=yYEqTaFg-KdnpJE_Ytp-EPVHIIC395iNtZrpVlLIf4o,101
+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=3AtUBrMIjw8H3ehDsJFYcBFoAZ1QKQo1hXTMsHh8f20,10159
+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=GK69WvUYnks5G26Z5L5XSisBIZILLv7lYhS2y8BJCt0,2031
+clarifai/datasets/upload/image.py,sha256=HlCsfEMu_C4GVecGSv52RUJ6laLW8H64Pfj_FQyX6qg,8580
+clarifai/datasets/upload/multimodal.py,sha256=_NpNQak9KMn0NOiOr48MYnXL0GQZ1LXKhwdYF1HhrHs,2377
+clarifai/datasets/upload/text.py,sha256=dpRMNz49EyKau0kwksEaNV6TLBUf5lSr7t5g3pG2byM,2298
+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
+clarifai/datasets/upload/loaders/coco_detection.py,sha256=_I_yThw435KS9SH7zheBbJDK3zFgjTImBsES__ijjMk,2831
+clarifai/datasets/upload/loaders/imagenet_classification.py,sha256=i7W5F6FTB3LwLmhPgjZHmbCbS3l4LmjsuBFKtjxl1pU,1962
+clarifai/datasets/upload/loaders/xview_detection.py,sha256=hk8cZdYZimm4KOaZvBjYcC6ikURZMn51xmn7pXZT3HE,6052
+clarifai/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+clarifai/models/api.py,sha256=d3FQQlG0mNDLrfEvchqaVcq4Tgb_TqryNnJtwp3c7sE,10961
+clarifai/modules/README.md,sha256=mx8pVx6cPp-pP4LcFPT_nX3ngGmhygVK0WiXeD3cbIo,367
+clarifai/modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+clarifai/modules/css.py,sha256=kadCEunmyh5h2yf0-4aysE3ZcZ6qaQcxuAgDXS96yF8,2020
+clarifai/modules/pages.py,sha256=iOoM3RNRMgXlV0qBqcdQofxoXo2RuRQh0h9c9BIS0-I,1383
+clarifai/modules/style.css,sha256=j7FNPZVhLPj35vvBksAJ90RuX5sLuqzDR5iM2WIEhiA,6073
+clarifai/rag/__init__.py,sha256=wu3PzAzo7uqgrEzuaC9lY_3gj1HFiR3GU3elZIKTT5g,40
+clarifai/rag/rag.py,sha256=bqUWnfdf91OYMucEK0_rJXDwg0oKjz5c7eda-9CPXu8,12680
+clarifai/rag/utils.py,sha256=yr1jAcbpws4vFGBqlAwPPE7v1DRba48g8gixLFw8OhQ,4070
+clarifai/runners/__init__.py,sha256=cDJ31l41dDsqW4Xn6sFMkKxxdIMTnGH9IW6sVkq0TNw,207
+clarifai/runners/server.py,sha256=xHDLdhQApCgYG19QOKXqJNCGNyw1Vsvobq3UdryDrVc,4132
+clarifai/runners/dockerfile_template/Dockerfile.template,sha256=5cjv7U8PmWa3DB_5B1CqSYh_6GE0E0np52TIAa7EIDE,2312
+clarifai/runners/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+clarifai/runners/models/model_builder.py,sha256=vIUZII9n5wL0SGV4xR6wfsdRvrMv4pSHr6P-KdYpikI,36343
+clarifai/runners/models/model_class.py,sha256=J33aq5OqKu_1iVnaSAbEQCaE_ina8X46PlQ5AiK80_o,14806
+clarifai/runners/models/model_run_locally.py,sha256=H7FKUBzZ_EPPj1b6P59qbOYr3mUJjJTlD5gavH5e80o,17746
+clarifai/runners/models/model_runner.py,sha256=T4Qn_x0vky7XdeS54bvipzEmKZMra1tQdAu_u01yyjc,6503
+clarifai/runners/models/model_servicer.py,sha256=A--b1P71PBCAMJCpy_-fpNDkfCVdvdMh1LleW15dSas,3037
+clarifai/runners/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+clarifai/runners/utils/code_script.py,sha256=NutTcbJWChjSb_WLNwkesavzcpP_-wh_R4ySWt0VPcI,8971
+clarifai/runners/utils/const.py,sha256=9qnOC1Bt6SGLQ9XCQEQ6519XhW4gzcztsV1Rgej67Vw,1055
+clarifai/runners/utils/data_types.py,sha256=l_yd09GIlAZ_0bJdY558BuNSKTM48rx7fYA7-HNxETY,12415
+clarifai/runners/utils/data_utils.py,sha256=dVGl7ZW6t0BztzfIOKV-WMeAEyEjwxzU3ud5fgL5Jmo,12843
+clarifai/runners/utils/loader.py,sha256=Sl0m29RDtMPx2cIiSbbDFtKHQj2ktXQ5CnkvaHi-zDc,8804
+clarifai/runners/utils/method_signatures.py,sha256=2SSStcBuBUbIgvFgAJFt4By0ZHos2c7bDWJOiUxQxZQ,18190
+clarifai/runners/utils/openai_convertor.py,sha256=qRhq_nGacaLFD_-8oAbPaDhGZJVMtKbgb2A4X7FlhD0,4564
+clarifai/runners/utils/serializers.py,sha256=S4sRsOVvH191vAGTRTAAdwLlQwlK4T5QVRDGPptg9nQ,7191
+clarifai/runners/utils/url_fetcher.py,sha256=v_8JOWmkyFAzsBulsieKX7Nfjy1Yg7wGSZeqfEvw2cg,1640
+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=FUFJJb8Jzohpt132JE43xHNzWtX0N8fbWROxJISfCM0,5072
+clarifai/utils/config.py,sha256=MxuQ1U4EOT4yVOEcRyv3OUPnoY_A_J5cZKHJGiuuo30,2883
+clarifai/utils/constants.py,sha256=fdCmw8fjgo8eoWu85t95ZoRTrgXC5_XVXFyB8_VQTtw,219
+clarifai/utils/logging.py,sha256=I2Sj1X-phYcINXe4-pW22Tq6Sz5PlwcxYQh7zORBbeM,13634
+clarifai/utils/misc.py,sha256=gensDjU5pV44iFzUzmM_Lf6PZX63GFxRXJX91esEjAg,2999
+clarifai/utils/model_train.py,sha256=Mndqy5GNu7kjQHjDyNVyamL0hQFLGSHcWhOuPyOvr1w,8005
+clarifai/utils/protobuf.py,sha256=P2deeGfs3VajkfQgPn7ROkqZ-4kXnWRxHCsy_XS7yKI,4954
+clarifai/utils/evaluation/__init__.py,sha256=PYkurUrXrGevByj7RFb6CoU1iC7fllyQSfnnlo9WnY8,69
+clarifai/utils/evaluation/helpers.py,sha256=aZeHLI7oSmU5YDWQp5GdkYW5qbHx37nV9xwunKTAwWM,18549
+clarifai/utils/evaluation/main.py,sha256=sQAuMk0lPclXCYvy_rS7rYteo2xh9Ju13VNvbyGt_VM,15779
+clarifai/utils/evaluation/testset_annotation_parser.py,sha256=iZfLw6oR1qgJ3MHMbOZXcGBLu7btSDn0VqdiAzpIm4g,5002
+clarifai/workflows/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+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-11.3.0.dist-info/licenses/LICENSE,sha256=mUqF_d12-qE2n41g7C5_sq-BMLOcj6CNN-jevr15YHU,555
+clarifai-11.3.0.dist-info/METADATA,sha256=_Y8jWVUURPYvFkK2Ky0MSr0dZSjY9S_LkLMBtBZh4Bo,22472
+clarifai-11.3.0.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+clarifai-11.3.0.dist-info/entry_points.txt,sha256=X9FZ4Z-i_r2Ud1RpZ9sNIFYuu_-9fogzCMCRUD9hyX0,51
+clarifai-11.3.0.dist-info/top_level.txt,sha256=wUMdCQGjkxaynZ6nZ9FAnvBUCgp5RJUVFSy2j-KYo0s,9
+clarifai-11.3.0.dist-info/RECORD,,

{clarifai-11.2.4rc2.dist-info → clarifai-11.3.0.dist-info}/WHEEL

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

clarifai/client/cli/base_cli.py

@@ -1,88 +0,0 @@
-import click
-import os
-import yaml
-
-@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)
-        print("Loaded config from file.")
-        print(f"Config: {ctx.obj}")
-    else:
-        ctx.obj = {}
-
-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'}
-    return environments.get(env, 'https://api.clarifai.com')
-
-
-@cli.command()
-@click.option('--config', type=click.Path(), required=False, help='Path to the config file')
-@click.option('-e', '--env', required=False, help='Environment', 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')
-    
-    click.echo("Login successful.")
-
-# Import the model CLI commands to register them
-from clarifai.client.cli.model_cli import model  # Ensure this is the correct import path
-
-
-if __name__ == '__main__':
-    cli()

clarifai/client/cli/model_cli.py

@@ -1,29 +0,0 @@
-import click
-from clarifai.client.cli.base_cli import cli
-
-@cli.group()
-def model():
-    """Manage models: upload, test locally"""
-    pass
-
-@model.command()
-@click.argument('model_path', type=click.Path(exists=True))
-@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.argument('model_path', type=click.Path(exists=True))
-def test_locally(model_path):
-    """Test model locally."""
-    try:
-        from clarifai.runners.models import run_test_locally
-        run_test_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)
-

clarifai/runners/dockerfile_template/Dockerfile.cpu.template

@@ -1,31 +0,0 @@
-ARG BASE_IMAGE=${BASE_IMAGE}
-FROM ${BASE_IMAGE} as build
-
-# Set the working directory to /app
-WORKDIR /app
-
-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 pip install --no-cache-dir -r requirements.txt
-
-# Install Clarifai SDK
-RUN pip install --no-cache-dir clarifai
-
-# These will be set by the templaing system.
-ENV CLARIFAI_PAT=${CLARIFAI_PAT}
-ENV CLARIFAI_USER_ID=${CLARIFAI_USER_ID}
-ENV CLARIFAI_RUNNER_ID=${CLARIFAI_RUNNER_ID}
-ENV CLARIFAI_NODEPOOL_ID=${CLARIFAI_NODEPOOL_ID}
-ENV CLARIFAI_COMPUTE_CLUSTER_ID=${CLARIFAI_COMPUTE_CLUSTER_ID}
-ENV CLARIFAI_API_BASE=${CLARIFAI_API_BASE}
-
-# Copy the current folder into /app/model_dir that the SDK will expect.
-COPY . /app/model_dir/${name}
-
-# Add the model directory to the python path.
-ENV PYTHONPATH "${PYTHONPATH}:/app/model_dir/${name}"
-
-# 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 ["-m", "clarifai.runners.server", "--model_path", "/app/model_dir/${name}"]