clarifai 11.5.4__py3-none-any.whl → 11.5.6__py3-none-any.whl

clarifai/__init__.py

@@ -1 +1 @@
-__version__ = "11.5.4"
+__version__ = "11.5.6"

clarifai/client/model.py

@@ -528,7 +528,10 @@ class Model(Lister, BaseClient):
             inference_params = kwargs.get('inference_params', {})
             output_config = kwargs.get('output_config', {})
             return self.client._predict_by_proto(
-                inputs=inputs, inference_params=inference_params, output_config=output_config
+                inputs=inputs,
+                # method_name="PostModelOutputs",
+                inference_params=inference_params,
+                output_config=output_config,
             )
 
         return self.client.predict(*args, **kwargs)

clarifai/runners/models/model_builder.py

@@ -457,8 +457,8 @@ class ModelBuilder:
         Returns the method signatures for the model class in YAML format.
         """
         model_class = self.load_model_class(mocking=True)
-        method_info = model_class._get_method_info()
-        signatures = {method.name: method.signature for method in method_info.values()}
+        method_infos = model_class._get_method_infos()
+        signatures = {method.name: method.signature for method in method_infos.values()}
         return signatures_to_yaml(signatures)
 
     def get_method_signatures(self, mocking=True):
@@ -472,8 +472,8 @@ class ModelBuilder:
           list: A list of method signatures for the model class.
         """
         model_class = self.load_model_class(mocking=mocking)
-        method_info = model_class._get_method_info()
-        signatures = [method.signature for method in method_info.values()]
+        method_infos = model_class._get_method_infos()
+        signatures = [method.signature for method in method_infos.values()]
         return signatures
 
     @property
@@ -550,6 +550,11 @@ class ModelBuilder:
             "inference_compute_info not found in the config file"
         )
         inference_compute_info = self.config.get('inference_compute_info')
+        # Ensure cpu_limit is a string if it exists and is an int
+        if 'cpu_limit' in inference_compute_info and isinstance(
+            inference_compute_info['cpu_limit'], int
+        ):
+            inference_compute_info['cpu_limit'] = str(inference_compute_info['cpu_limit'])
         return json_format.ParseDict(inference_compute_info, resources_pb2.ComputeInfo())
 
     def check_model_exists(self):

clarifai/runners/models/model_class.py

@@ -1,7 +1,7 @@
 import inspect
 import itertools
-import logging
 import os
+import threading
 import traceback
 from abc import ABC
 from collections import abc
@@ -20,11 +20,16 @@ from clarifai.runners.utils.method_signatures import (
     serialize,
     signatures_to_json,
 )
+from clarifai.runners.utils.model_utils import is_proto_style_method
+from clarifai.utils.logging import logger
 
 _METHOD_INFO_ATTR = '_cf_method_info'
 
 _RAISE_EXCEPTIONS = os.getenv("RAISE_EXCEPTIONS", "false").lower() in ("true", "1")
 
+FALLBACK_METHOD_PROTO = 'PostModelOutputs'
+FALLBACK_METHOD_PYTHON = 'predict'
+
 
 class ModelClass(ABC):
     '''
@@ -55,22 +60,33 @@ class ModelClass(ABC):
             yield item.x + ' ' + str(item.y)
     '''
 
+    def __init__(self):
+        super().__init__()
+        self._thread_local = threading.local()
+
     @staticmethod
     def method(func):
         setattr(func, _METHOD_INFO_ATTR, _MethodInfo(func))
         return func
 
     def set_output_context(self, prompt_tokens=None, completion_tokens=None):
-        """This is used to set the prompt and completion tokens in the Output proto"""
-        self._prompt_tokens = prompt_tokens
-        self._completion_tokens = completion_tokens
+        """Set the prompt and completion tokens for the Output proto.
+        In batch mode, call this once per output, in order, before returning each output.
+        """
+        if not hasattr(self._thread_local, 'token_contexts'):
+            self._thread_local.token_contexts = []
+        self._thread_local.token_contexts.append((prompt_tokens, completion_tokens))
 
     def load_model(self):
         """Load the model."""
 
     def _handle_get_signatures_request(self) -> service_pb2.MultiOutputResponse:
-        methods = self._get_method_info()
-        signatures = {method.name: method.signature for method in methods.values()}
+        methods = self._get_method_infos()
+        signatures = {
+            method.name: method.signature
+            for method in methods.values()
+            if method.signature is not None
+        }
         resp = service_pb2.MultiOutputResponse(
             status=status_pb2.Status(code=status_code_pb2.SUCCESS)
         )
@@ -99,18 +115,61 @@ class ModelClass(ABC):
         outputs = []
         try:
             # TODO add method name field to proto
-            method_name = 'predict'
+            # to support old callers who might not pass in the method name we have a few defaults.
+            # first we look for a PostModelOutputs method that is implemented as protos and use that
+            # if it exists.
+            # if not we default to 'predict'.
+            method_name = None
             if len(request.inputs) > 0 and '_method_name' in request.inputs[0].data.metadata:
                 method_name = request.inputs[0].data.metadata['_method_name']
+            if method_name is None and FALLBACK_METHOD_PROTO in self._get_method_infos():
+                _info = self._get_method_infos(FALLBACK_METHOD_PROTO)
+                if _info.proto_method:
+                    method_name = FALLBACK_METHOD_PROTO
+            if method_name is None:
+                method_name = FALLBACK_METHOD_PYTHON
             if (
                 method_name == '_GET_SIGNATURES'
             ):  # special case to fetch signatures, TODO add endpoint for this
                 return self._handle_get_signatures_request()
-            if method_name not in self._get_method_info():
+            if method_name not in self._get_method_infos():
                 raise ValueError(f"Method {method_name} not found in model class")
             method = getattr(self, method_name)
-            method_info = method._cf_method_info
+            method_info = self._get_method_infos(method_name)
             signature = method_info.signature
+            proto_method = method_info.proto_method
+
+            # If this is an old predict(proto) -> proto method, just call it and return
+            # the response.
+            if proto_method:
+                out_proto = method(request)
+                # if we already have out_proto.status.code set then return
+                if out_proto.status.code != status_code_pb2.ZERO:
+                    return out_proto
+
+                successes = [
+                    out.status.code == status_code_pb2.SUCCESS for out in out_proto.outputs
+                ]
+                if all(successes):
+                    # If all outputs are successful, we can return the response.
+                    out_proto.status.CopyFrom(
+                        status_pb2.Status(code=status_code_pb2.SUCCESS, description='Success')
+                    )
+                    return out_proto
+                if any(successes):
+                    # If some outputs are successful and some are not, we return a mixed status.
+                    out_proto.status.CopyFrom(
+                        status_pb2.Status(
+                            code=status_code_pb2.MIXED_STATUS, description='Mixed Status'
+                        )
+                    )
+                    return out_proto
+                # If all outputs are failures, we return a failure status.
+                out_proto.status.CopyFrom(
+                    status_pb2.Status(code=status_code_pb2.FAILURE, description='Failed')
+                )
+                return out_proto
+
             python_param_types = method_info.python_param_types
             for input in request.inputs:
                 # check if input is in old format
@@ -121,6 +180,7 @@ class ModelClass(ABC):
                         input.data, signature.input_fields
                     )
                     input.data.CopyFrom(new_data)
+
             # convert inputs to python types
             inputs = self._convert_input_protos_to_python(
                 request.inputs, signature.input_fields, python_param_types
@@ -148,7 +208,7 @@ class ModelClass(ABC):
         except Exception as e:
             if _RAISE_EXCEPTIONS:
                 raise
-            logging.exception("Error in predict")
+            logger.exception("Error in predict")
             return service_pb2.MultiOutputResponse(
                 status=status_pb2.Status(
                     code=status_code_pb2.FAILURE,
@@ -161,11 +221,12 @@ class ModelClass(ABC):
         self, request: service_pb2.PostModelOutputsRequest
     ) -> Iterator[service_pb2.MultiOutputResponse]:
         try:
+            assert len(request.inputs) == 1, "Generate requires exactly one input"
             method_name = 'generate'
             if len(request.inputs) > 0 and '_method_name' in request.inputs[0].data.metadata:
                 method_name = request.inputs[0].data.metadata['_method_name']
             method = getattr(self, method_name)
-            method_info = method._cf_method_info
+            method_info = self._get_method_infos(method_name)
             signature = method_info.signature
             python_param_types = method_info.python_param_types
             for input in request.inputs:
@@ -207,7 +268,7 @@ class ModelClass(ABC):
         except Exception as e:
             if _RAISE_EXCEPTIONS:
                 raise
-            logging.exception("Error in generate")
+            logger.exception("Error in generate")
             yield service_pb2.MultiOutputResponse(
                 status=status_pb2.Status(
                     code=status_code_pb2.FAILURE,
@@ -227,7 +288,7 @@ class ModelClass(ABC):
             if len(request.inputs) > 0 and '_method_name' in request.inputs[0].data.metadata:
                 method_name = request.inputs[0].data.metadata['_method_name']
             method = getattr(self, method_name)
-            method_info = method._cf_method_info
+            method_info = self._get_method_infos(method_name)
             signature = method_info.signature
             python_param_types = method_info.python_param_types
 
@@ -282,7 +343,7 @@ class ModelClass(ABC):
         except Exception as e:
             if _RAISE_EXCEPTIONS:
                 raise
-            logging.exception("Error in stream")
+            logger.exception("Error in stream")
             yield service_pb2.MultiOutputResponse(
                 status=status_pb2.Status(
                     code=status_code_pb2.FAILURE,
@@ -333,12 +394,18 @@ class ModelClass(ABC):
             data = DataConverter.convert_output_data_to_old_format(proto.data)
             proto.data.CopyFrom(data)
         proto.status.code = status_code_pb2.SUCCESS
-        if hasattr(self, "_prompt_tokens") and self._prompt_tokens is not None:
-            proto.prompt_tokens = self._prompt_tokens
-        if hasattr(self, "_completion_tokens") and self._completion_tokens is not None:
-            proto.completion_tokens = self._completion_tokens
-        self._prompt_tokens = None
-        self._completion_tokens = None
+        # Per-output token context support
+        token_contexts = getattr(self._thread_local, 'token_contexts', None)
+        prompt_tokens = completion_tokens = None
+        if token_contexts and len(token_contexts) > 0:
+            prompt_tokens, completion_tokens = token_contexts.pop(0)
+            # If this was the last, clean up
+            if len(token_contexts) == 0:
+                del self._thread_local.token_contexts
+        if prompt_tokens is not None:
+            proto.prompt_tokens = prompt_tokens
+        if completion_tokens is not None:
+            proto.completion_tokens = completion_tokens
         return proto
 
     @classmethod
@@ -359,28 +426,44 @@ class ModelClass(ABC):
                     continue
                 methods[name] = method_info
         # check for generic predict(request) -> response, etc. methods
-        # for name in ('predict', 'generate', 'stream'):
-        #  if hasattr(cls, name):
-        #    method = getattr(cls, name)
-        #    if not hasattr(method, _METHOD_INFO_ATTR):  # not already put in registry
-        #      methods[name] = _MethodInfo(method)
+        # older models never had generate or stream so don't bother with them.
+        for name in [FALLBACK_METHOD_PROTO]:  # , 'GenerateModelOutputs', 'StreamModelOutputs'):
+            if hasattr(cls, name) and name not in methods:
+                method = getattr(cls, name)
+                if not callable(method):
+                    continue
+                if is_proto_style_method(method):
+                    # If this is a proto-style method, we can add it to the registry as a special case.
+                    methods[name] = _MethodInfo(method, proto_method=True)
         # set method table for this class in the registry
         return methods
 
     @classmethod
-    def _get_method_info(cls, func_name=None):
+    def _get_method_infos(cls, func_name=None):
+        # FIXME: this is a re-use of the _METHOD_INFO_ATTR attribute to store the method info
+        # for all methods on the class. Should use a different attribute name to avoid confusion.
         if not hasattr(cls, _METHOD_INFO_ATTR):
             setattr(cls, _METHOD_INFO_ATTR, cls._register_model_methods())
-        method_info = getattr(cls, _METHOD_INFO_ATTR)
+        method_infos = getattr(cls, _METHOD_INFO_ATTR)
         if func_name:
-            return method_info[func_name]
-        return method_info
+            return method_infos[func_name]
+        return method_infos
 
 
 class _MethodInfo:
-    def __init__(self, method):
+    def __init__(self, method, proto_method=False):
+        """Initialize a MethodInfo instance.
+
+        Args:
+            method: The method to wrap.
+            old_method: If True, this is an old proto-style method that returns a proto directly.
+        """
         self.name = method.__name__
-        self.signature = build_function_signature(method)
+        self.proto_method = proto_method
+        if not proto_method:
+            self.signature = build_function_signature(method)
+        else:
+            self.signature = None
         self.python_param_types = {
             p.name: p.annotation
             for p in inspect.signature(method).parameters.values()

clarifai/runners/models/model_runner.py

@@ -108,9 +108,19 @@ class ModelRunner(BaseRunner, HealthProbeRequestHandler):
         ensure_urls_downloaded(request, auth_helper=self._auth_helper)
 
         resp = self.model.predict_wrapper(request)
-        if resp.status.code != status_code_pb2.SUCCESS:
+        # if we have any non-successful code already it's an error we can return.
+        if (
+            resp.status.code != status_code_pb2.SUCCESS
+            and resp.status.code != status_code_pb2.ZERO
+        ):
             return service_pb2.RunnerItemOutput(multi_output_response=resp)
-        successes = [o.status.code == status_code_pb2.SUCCESS for o in resp.outputs]
+        successes = []
+        for output in resp.outputs:
+            if not output.HasField('status') or not output.status.code:
+                raise Exception(
+                    "Output must have a status code, please check the model implementation."
+                )
+            successes.append(output.status.code == status_code_pb2.SUCCESS)
         if all(successes):
             status = status_pb2.Status(
                 code=status_code_pb2.SUCCESS,
@@ -141,7 +151,11 @@ class ModelRunner(BaseRunner, HealthProbeRequestHandler):
         ensure_urls_downloaded(request, auth_helper=self._auth_helper)
 
         for resp in self.model.generate_wrapper(request):
-            if resp.status.code != status_code_pb2.SUCCESS:
+            # if we have any non-successful code already it's an error we can return.
+            if (
+                resp.status.code != status_code_pb2.SUCCESS
+                and resp.status.code != status_code_pb2.ZERO
+            ):
                 yield service_pb2.RunnerItemOutput(multi_output_response=resp)
                 continue
             successes = []
@@ -174,10 +188,12 @@ class ModelRunner(BaseRunner, HealthProbeRequestHandler):
         self, runner_item_iterator: Iterator[service_pb2.RunnerItem]
     ) -> Iterator[service_pb2.RunnerItemOutput]:
         # Call the generate() method the underlying model implements.
-        for resp in self.model.stream_wrapper(
-            pmo_iterator(runner_item_iterator, self._auth_helper)
-        ):
-            if resp.status.code != status_code_pb2.SUCCESS:
+        for resp in self.model.stream_wrapper(pmo_iterator(runner_item_iterator)):
+            # if we have any non-successful code already it's an error we can return.
+            if (
+                resp.status.code != status_code_pb2.SUCCESS
+                and resp.status.code != status_code_pb2.ZERO
+            ):
                 yield service_pb2.RunnerItemOutput(multi_output_response=resp)
                 continue
             successes = []

clarifai/runners/models/openai_class.py

@@ -34,6 +34,7 @@ class OpenAIModelClass(ModelClass):
     model = None
 
     def __init__(self) -> None:
+        super().__init__()
         if self.client is None:
             raise NotImplementedError("Subclasses must set the 'client' class attribute")
         if self.model is None:

clarifai/runners/utils/code_script.py

@@ -18,7 +18,9 @@ def has_signature_method(
     :param method_signatures: List of MethodSignature objects to search in.
     :return: True if a method with the given name exists, False otherwise.
     """
-    return any(method_signature.name == name for method_signature in method_signatures)
+    return any(
+        method_signature.name == name for method_signature in method_signatures if method_signature
+    )
 
 
 def generate_client_script(
@@ -153,6 +155,8 @@ model = Model("{model_ui_url}",
     # Generate method signatures
     method_signatures_str = []
     for method_signature in method_signatures:
+        if method_signature is None:
+            continue
         method_name = method_signature.name
         client_script_str = f'response = model.{method_name}('
         annotations = _get_annotations_source(method_signature)

clarifai/runners/utils/model_utils.py

@@ -1,3 +1,4 @@
+import inspect
 import os
 import shlex
 import signal
@@ -8,6 +9,7 @@ import time
 
 import psutil
 import requests
+from clarifai_grpc.grpc.api import service_pb2
 
 from clarifai.utils.logging import logger
 
@@ -133,3 +135,50 @@ def wait_for_server(base_url: str, timeout: int = None) -> None:
                 raise TimeoutError("Server did not become ready within timeout period")
         except requests.exceptions.RequestException:
             time.sleep(1)
+
+
+def is_proto_style_method(method):
+    """
+    Determines if the given method is likely an old-style proto method:
+    - Has a 'request' parameter after 'self'
+    - Optionally, returns a known proto response type
+    """
+    try:
+        sig = inspect.signature(method)
+        params = list(sig.parameters.values())
+
+        # Must have at least 'self' and one argument
+        if len(params) < 2:
+            return False
+
+        # First parameter should be 'self'
+        if params[0].name != 'self':
+            return False
+        # Second param typically should be named 'request'
+        request_param = params[1]
+        if request_param.name != 'request':
+            return False
+        # Optionally: check annotation is a proto type
+        # (If signature is incomplete, this part will gracefully fall through)
+        return_annotation = sig.return_annotation
+        # If type annotation is available, check it's PostModelOutputsRequest
+        if (
+            request_param.annotation != inspect.Parameter.empty
+            and request_param.annotation != service_pb2.PostModelOutputsRequest
+        ):
+            return False
+        # If return annotation is available, check it's MultiOutputResponse
+        if (
+            return_annotation != inspect.Signature.empty
+            and return_annotation != service_pb2.MultiOutputResponse
+        ):
+            return False
+        if (
+            request_param.annotation is inspect.Parameter.empty
+            and return_annotation is inspect.Signature.empty
+        ):
+            return True  # signature OK, even if signature is empty
+        return True
+
+    except (ValueError, TypeError):
+        return False

{clarifai-11.5.4.dist-info → clarifai-11.5.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: clarifai
-Version: 11.5.4
+Version: 11.5.6
 Home-page: https://github.com/Clarifai/clarifai-python
 Author: Clarifai
 Author-email: support@clarifai.com
@@ -32,7 +32,7 @@ Requires-Dist: click>=8.1.7
 Requires-Dist: requests>=2.32.3
 Requires-Dist: aiohttp>=3.10.0
 Requires-Dist: uv==0.7.12
-Requires-Dist: ruff==0.11.4
+Requires-Dist: psutil==7.0.0
 Provides-Extra: all
 Requires-Dist: pycocotools>=2.0.7; extra == "all"
 Dynamic: author

{clarifai-11.5.4.dist-info → clarifai-11.5.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-clarifai/__init__.py,sha256=dX3Oi94KTZEq35Tm3ca-zFtWYOWbwopKBoxPCzh9NFo,23
+clarifai/__init__.py,sha256=BxuXNC41hShXWv3SgslBYu8ufM3yw21Lbp4jemirGjY,23
 clarifai/cli.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clarifai/errors.py,sha256=GXa6D4v_L404J83jnRNFPH7s-1V9lk7w6Ws99f1g-AY,2772
 clarifai/versions.py,sha256=ecSuEB_nOL2XSoYHDw2n23XUbm_KPOGjudMXmQrGdS8,224
@@ -24,7 +24,7 @@ clarifai/client/dataset.py,sha256=OgdpZkQ_vYmRxL8-qphcNozpvPV1bWTlte9Jv6UkKb8,35
 clarifai/client/deployment.py,sha256=QBf0tzkKBEpzNgmOEmWUJMOlVWdFEFc70Y44o8y75Gs,2875
 clarifai/client/input.py,sha256=jpX47qwn7aUBBIEuSSLHF5jk70XaWEh0prD065c9b-E,51205
 clarifai/client/lister.py,sha256=1YEm2suNxPaJO4x9V5szgD_YX6N_00vgSO-7m0HagY8,2208
-clarifai/client/model.py,sha256=Oh2zCa7XA_pBAEx9D8ME81H_BBqD7wJYoeB5CEjlI5E,89989
+clarifai/client/model.py,sha256=8D_L6nuL4_hBAKgwseYhoAeKS9u3ky0zczkcJghxFe8,90072
 clarifai/client/model_client.py,sha256=4gIS0mKBdiNMA1x_6Wo6H7WbfLsmQix64EpONcQjQV4,37129
 clarifai/client/module.py,sha256=jLViQYvVV3FmRN_ivvbk83uwsx7CgYGeEx4dYAr6yD4,4537
 clarifai/client/nodepool.py,sha256=Y5zQ0JLdTjAp2TmVnx7AAOwaB2YUslk3nl7s6BQ90FQ,16415
@@ -75,12 +75,12 @@ clarifai/runners/dockerfile_template/Dockerfile.template,sha256=DUH7F0-uLOV0LTjn
 clarifai/runners/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clarifai/runners/models/dummy_openai_model.py,sha256=pcmAVbqTTGG4J3BLVjKfvM_SQ-GET_XexIUdLcr9Zvo,8373
 clarifai/runners/models/mcp_class.py,sha256=RdKn7rW4vYol0VRDZiLTSMfkqjLhO1ijXAQ0Rq0Jfnw,6647
-clarifai/runners/models/model_builder.py,sha256=D47H1zm3_Xr-nlPYb_1dY8dxlMiM93kU4QYGEtyxK88,63403
-clarifai/runners/models/model_class.py,sha256=-euUF-eHUi4KXR_e1pIwvToDZ13CM6TSz2FolzildjM,16069
+clarifai/runners/models/model_builder.py,sha256=lzXf1H4oQ4l672xHoXSaT4bi0z0vTj3YnziT7RRRLIg,63695
+clarifai/runners/models/model_class.py,sha256=Ndh437BNMkpFBo6B108GuKL8sGYaGnSplZ6FxOgd_v8,20010
 clarifai/runners/models/model_run_locally.py,sha256=6-6WjEKc0ba3gAv4wOLdMs2XOzS3b-2bZHJS0wdVqJY,20088
-clarifai/runners/models/model_runner.py,sha256=CE5UkxW8wCm-g8dn_i1TNkZDORpOGRoVwZK-Uk75Xws,8407
+clarifai/runners/models/model_runner.py,sha256=tZTX1XKMlniJEmd1WMjcwGfej5NCWqv23HZ4xrG8YV8,9153
 clarifai/runners/models/model_servicer.py,sha256=415RqiXCPH-1WgFHrtZMDH1nb8gaQDpVlrQ_tvtbjOg,4523
-clarifai/runners/models/openai_class.py,sha256=aXlk5W6LWkh-A4eZYi74DeLW0i_86_9DYYGxpJHXI0w,6688
+clarifai/runners/models/openai_class.py,sha256=OVYe4dWJPhskyZn53X9yJorzKZA7mjRCqYbu1rzRTFU,6715
 clarifai/runners/models/visual_classifier_class.py,sha256=1ZoLfCT2crrgRbejjTMAIwpTRgQMiH9N9yflOVpFxSg,2721
 clarifai/runners/models/visual_detector_class.py,sha256=ky4oFAkGCKPpGPdgaOso-n6D3HcmnbKee_8hBsNiV8U,2883
 clarifai/runners/pipeline_steps/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -88,12 +88,12 @@ clarifai/runners/pipeline_steps/pipeline_step_builder.py,sha256=E6Ce3b0RolYLMJHa
 clarifai/runners/pipelines/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clarifai/runners/pipelines/pipeline_builder.py,sha256=z_bCwjwQPFa_1AYkorhh5r6t6r5hC5K2D8Z1LTEzIpg,12801
 clarifai/runners/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-clarifai/runners/utils/code_script.py,sha256=4wkInN_7MgCvMt9KLJA5_13iF5UFdeQLEhs0T5GvsHQ,13830
+clarifai/runners/utils/code_script.py,sha256=RxN4UOBIa61WW1U5y-gC-dwr_tY8ZAKr1nR8sySNWt4,13922
 clarifai/runners/utils/const.py,sha256=MK7lTzzJKbOiyiUtG_jlJXfz_xNKMn5LjkQ9vjbttXE,1538
 clarifai/runners/utils/data_utils.py,sha256=HRpMYR2O0OiDpXXhOManLHTeomC4bFnXMHVAiT_12yE,20856
 clarifai/runners/utils/loader.py,sha256=K5Y8MPbIe5STw2gDnrL8KqFgKNxEo7bz-RV0ip1T4PM,10900
 clarifai/runners/utils/method_signatures.py,sha256=qdHaO8ZIgP6BBXXMhMPhcQ46dse-XMP2t4VJCNG7O3Q,18335
-clarifai/runners/utils/model_utils.py,sha256=Cn3_89ZMhIR6f61H-btcK_Luwq0CMAx5fMxPxwngo68,4530
+clarifai/runners/utils/model_utils.py,sha256=eTIGOZpt0icOGfVnmYM9yO9g7CTsb_plLkzvJIH1QRo,6288
 clarifai/runners/utils/openai_convertor.py,sha256=ZlIrvvfHttD_DavLvmKZdL8gNq_TQvQtZVnYamwdWz4,8248
 clarifai/runners/utils/pipeline_validation.py,sha256=RWWWUA3mNCXHCSNpuofMGfXfWEYe7LrYKwIc-1VFvQs,6444
 clarifai/runners/utils/serializers.py,sha256=pI7GqMTC0T3Lu_X8v8TO4RiplO-gC_49Ns37jYwsPtg,7908
@@ -118,9 +118,9 @@ clarifai/workflows/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuF
 clarifai/workflows/export.py,sha256=HvUYG9N_-UZoRR0-_tdGbZ950_AeBqawSppgUxQebR0,1913
 clarifai/workflows/utils.py,sha256=ESL3INcouNcLKCh-nMpfXX-YbtCzX7tz7hT57_RGQ3M,2079
 clarifai/workflows/validate.py,sha256=UhmukyHkfxiMFrPPeBdUTiCOHQT5-shqivlBYEyKTlU,2931
-clarifai-11.5.4.dist-info/licenses/LICENSE,sha256=mUqF_d12-qE2n41g7C5_sq-BMLOcj6CNN-jevr15YHU,555
-clarifai-11.5.4.dist-info/METADATA,sha256=9PlKYam9cZ_1StX7ECiTXQMpS1ci4LjgvKhNi59W-8c,22736
-clarifai-11.5.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-clarifai-11.5.4.dist-info/entry_points.txt,sha256=X9FZ4Z-i_r2Ud1RpZ9sNIFYuu_-9fogzCMCRUD9hyX0,51
-clarifai-11.5.4.dist-info/top_level.txt,sha256=wUMdCQGjkxaynZ6nZ9FAnvBUCgp5RJUVFSy2j-KYo0s,9
-clarifai-11.5.4.dist-info/RECORD,,
+clarifai-11.5.6.dist-info/licenses/LICENSE,sha256=mUqF_d12-qE2n41g7C5_sq-BMLOcj6CNN-jevr15YHU,555
+clarifai-11.5.6.dist-info/METADATA,sha256=dE9Pb314d6-aAeFuNetMEyt6GsbveEuJdqnjTbeln7E,22737
+clarifai-11.5.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+clarifai-11.5.6.dist-info/entry_points.txt,sha256=X9FZ4Z-i_r2Ud1RpZ9sNIFYuu_-9fogzCMCRUD9hyX0,51
+clarifai-11.5.6.dist-info/top_level.txt,sha256=wUMdCQGjkxaynZ6nZ9FAnvBUCgp5RJUVFSy2j-KYo0s,9
+clarifai-11.5.6.dist-info/RECORD,,