clarifai 11.6.4rc1__py3-none-any.whl → 11.6.5__py3-none-any.whl

clarifai/utils/cli.py

@@ -220,3 +220,130 @@ def validate_context_auth(pat: str, user_id: str, api_base: str = None):
             logger.error(f"❌ Validation failed: \n{error_msg}")
             logger.error("Please check your credentials and try again.")
         raise click.Abort()  # Exit without saving the configuration
+
+
+def customize_ollama_model(model_path, model_name, port, context_length):
+    """Customize the Ollama model name in the cloned template files.
+    Args:
+     model_path: Path to the cloned model directory
+     model_name: The model name to set (e.g., 'llama3.1', 'mistral')
+
+    """
+    model_py_path = os.path.join(model_path, "1", "model.py")
+
+    if not os.path.exists(model_py_path):
+        logger.warning(f"Model file {model_py_path} not found, skipping model name customization")
+        return
+
+    try:
+        # Read the model.py file
+        with open(model_py_path, 'r') as file:
+            content = file.read()
+        if model_name:
+            # Replace the default model name in the load_model method
+            content = content.replace(
+                'self.model = os.environ.get("OLLAMA_MODEL_NAME", \'llama3.2\')',
+                f'self.model = os.environ.get("OLLAMA_MODEL_NAME", \'{model_name}\')',
+            )
+
+        if port:
+            # Replace the default port variable in the model.py file
+            content = content.replace("PORT = '23333'", f"PORT = '{port}'")
+
+        if context_length:
+            # Replace the default context length variable in the model.py file
+            content = content.replace(
+                "context_length = '8192'", f"context_length = '{context_length}'"
+            )
+
+        # Write the modified content back to model.py
+        with open(model_py_path, 'w') as file:
+            file.write(content)
+
+    except Exception as e:
+        logger.error(f"Failed to customize Ollama model name in {model_py_path}: {e}")
+        raise
+
+
+def check_ollama_installed():
+    """Check if the Ollama CLI is installed."""
+    try:
+        import subprocess
+
+        result = subprocess.run(
+            ['ollama', '--version'], capture_output=True, text=True, check=False
+        )
+        if result.returncode == 0:
+            return True
+        else:
+            return False
+    except FileNotFoundError:
+        return False
+
+
+def _is_package_installed(package_name):
+    """Helper function to check if a single package in requirements.txt is installed."""
+    import importlib.metadata
+
+    try:
+        importlib.metadata.distribution(package_name)
+        logger.debug(f"✅ {package_name} - installed")
+        return True
+    except importlib.metadata.PackageNotFoundError:
+        logger.debug(f"❌ {package_name} - not installed")
+        return False
+    except Exception as e:
+        logger.warning(f"Error checking {package_name}: {e}")
+        return False
+
+
+def check_requirements_installed(model_path):
+    """Check if all dependencies in requirements.txt are installed."""
+    import re
+    from pathlib import Path
+
+    requirements_path = Path(model_path) / "requirements.txt"
+
+    if not requirements_path.exists():
+        logger.warning(f"requirements.txt not found at {requirements_path}")
+        return True
+
+    try:
+        package_pattern = re.compile(r'^([a-zA-Z0-9_-]+)')
+
+        # Getting package name and version (for logging)
+        requirements = [
+            (match.group(1), pack)
+            for line in requirements_path.read_text().splitlines()
+            if (pack := line.strip())
+            and not line.startswith('#')
+            and (match := package_pattern.match(line))
+        ]
+
+        if not requirements:
+            logger.info("No dependencies found in requirements.txt")
+            return True
+
+        logger.info(f"Checking {len(requirements)} dependencies...")
+
+        missing = [
+            full_req
+            for package_name, full_req in requirements
+            if not _is_package_installed(package_name)
+        ]
+
+        if not missing:
+            logger.info(f"✅ All {len(requirements)} dependencies are installed!")
+            return True
+
+        # Report missing packages
+        logger.error(
+            f"❌ {len(missing)} of {len(requirements)} required packages are missing in the current environment"
+        )
+        logger.error("\n".join(f"  - {pkg}" for pkg in missing))
+        logger.warning(f"To install: pip install -r {requirements_path}")
+        return False
+
+    except Exception as e:
+        logger.error(f"Failed to check requirements: {e}")
+        return False

clarifai/utils/constants.py

@@ -64,3 +64,8 @@ DEFAULT_LOCAL_RUNNER_NODEPOOL_CONFIG = {
 }
 DEFAULT_OLLAMA_MODEL_REPO = "https://github.com/Clarifai/runners-examples"
 DEFAULT_OLLAMA_MODEL_REPO_BRANCH = "ollama"
+
+STATUS_OK = "200 OK"
+STATUS_MIXED = "207 MIXED"
+STATUS_FAIL = "500 FAIL"
+STATUS_UNKNOWN = "UNKNOWN"

clarifai/utils/misc.py

@@ -78,6 +78,53 @@ def get_from_env(key: str, env_key: str) -> str:
         )
 
 
+def get_from_dict_env_or_config(key: str, env_key: str, **data) -> str:
+    """Get a value from a dictionary, environment variable, or CLI config context."""
+    # First try the provided data/kwargs
+    if key in data and data[key]:
+        return data[key]
+
+    # Then try environment variables
+    if env_key in os.environ and os.environ[env_key]:
+        return os.environ[env_key]
+
+    # Finally try CLI config context as fallback
+    try:
+        from clarifai.utils.config import Config
+        from clarifai.utils.constants import DEFAULT_CONFIG
+
+        config = Config.from_yaml(filename=DEFAULT_CONFIG)
+        current_context = config.current
+
+        # Convert env_key to the attribute name expected by Context
+        # e.g., CLARIFAI_PAT -> pat, CLARIFAI_USER_ID -> user_id, CLARIFAI_API_BASE -> api_base
+        if env_key == "CLARIFAI_PAT":
+            attr_name = "pat"
+        elif env_key == "CLARIFAI_USER_ID":
+            attr_name = "user_id"
+        elif env_key == "CLARIFAI_API_BASE":
+            attr_name = "api_base"
+        else:
+            # For other cases, convert CLARIFAI_SOMETHING to something
+            attr_name = env_key.replace("CLARIFAI_", "").lower()
+
+        if hasattr(current_context, attr_name):
+            value = getattr(current_context, attr_name)
+            if value:
+                return value
+    except Exception:
+        # If CLI config loading fails, fall through to raise error
+        pass
+
+    # If all methods fail, raise an error suggesting clarifai login
+    raise UserError(
+        f"Configuration Required. Could not find '{key}'. Please provide it in one of the following ways:\n\n"
+        f"- Pass '{key}' as a named parameter to your function.\n"
+        f"- Set the {env_key} environment variable in your environment.\n"
+        f"- Run `clarifai login` in your terminal to configure CLI authentication."
+    )
+
+
 def concept_relations_accumulation(
     relations_dict: Dict[str, Any], subject_concept: str, object_concept: str, predicate: str
 ) -> Dict[str, Any]:
@@ -242,7 +289,8 @@ class GitHubDownloader:
             return owner, repo, branch, folder_path
 
         except Exception as e:
-            raise ValueError(f"Failed to parse GitHub URL: {e}")
+            logger.error(f"Failed to parse GitHub URL: {e}")
+            sys.exit(1)
 
     def get_folder_contents(self, owner: str, repo: str, path: str, branch: str = 'main') -> list:
         api_url = f"https://api.github.com/repos/{owner}/{repo}/contents/{path}"

{clarifai-11.6.4rc1.dist-info → clarifai-11.6.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: clarifai
-Version: 11.6.4rc1
+Version: 11.6.5
 Home-page: https://github.com/Clarifai/clarifai-python
 Author: Clarifai
 Author-email: support@clarifai.com
@@ -19,7 +19,7 @@ Classifier: Operating System :: OS Independent
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: clarifai-grpc>=11.6.1
+Requires-Dist: clarifai-grpc>=11.6.4
 Requires-Dist: clarifai-protocol>=0.0.25
 Requires-Dist: numpy>=1.22.0
 Requires-Dist: tqdm>=4.65.0

clarifai-11.6.5.dist-info/RECORD

@@ -0,0 +1,127 @@
+clarifai/__init__.py,sha256=v1UMok1JAGjDjikTzsL5Gx86ndvD68_szNQj1A3xaVk,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
+clarifai/cli/README.md,sha256=u-7Mgg1ZrygEokXS3RJepddq3GjhkSVkdUzXvxnCoIE,2633
+clarifai/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+clarifai/cli/__main__.py,sha256=7nPbLW7Jr2shkgMPvnxpn4xYGMvIcnqluJ69t9w4H_k,74
+clarifai/cli/base.py,sha256=rMmbjo5VIiAq35cTFb9IrNPVFBfiXU366_rv7V4ULus,8889
+clarifai/cli/compute_cluster.py,sha256=8Xss0Obrp6l1XuxJe0luOqU_pf8vXGDRi6jyIe8qR6k,2282
+clarifai/cli/deployment.py,sha256=9C4I6_kyMxRkWl6h681wc79-3mAtDHtTUaxRv05OZMs,4262
+clarifai/cli/model.py,sha256=bL6jiM8UMBZUG7yDPVNvrulWgV2-PfGWj3R52kyG0PI,41276
+clarifai/cli/nodepool.py,sha256=H6OIdUW_EiyDUwZogzEDoYmVwEjLMsgoDlPyE7gjIuU,4245
+clarifai/cli/pipeline.py,sha256=smWPCK9kLCqnjTCb3w8BAeiAcowY20Bdxfk-OCzCi0I,10601
+clarifai/cli/pipeline_step.py,sha256=Unrq63w5rjwyhHHmRhV-enztO1HuVTnimAXltNCotQs,3814
+clarifai/cli/templates/__init__.py,sha256=HbMlZuYOMyVJde73ijNAevmSRUpIttGlHdwyO4W-JOs,44
+clarifai/cli/templates/model_templates.py,sha256=q1dKF5jHhX_1g4qtiYcEbJdWQaxCv2BSEAXI4AR2JOo,9709
+clarifai/cli/templates/pipeline_step_templates.py,sha256=HU1BoU7wG71MviQAvyecxT_qo70XhTtPGYtoIQ-U-l0,1663
+clarifai/cli/templates/pipeline_templates.py,sha256=mfHrEoRxICIv00zxfgIct2IpxcMmZ6zjHG8WLF1TPcI,4409
+clarifai/client/__init__.py,sha256=KXvZFE9TCJf1k_GNUHCZ4icsUlKr1lz0cnBR91LuY8M,765
+clarifai/client/app.py,sha256=eMXFYea-bopfcQ30JuAQqdhzhEsxAB3w4vDOsXOVO6c,41445
+clarifai/client/base.py,sha256=aduMVVSPUSQ-MsDpj4yuHmCTUNbCTPt76dhRZ14euVg,10284
+clarifai/client/compute_cluster.py,sha256=q-JuBai4pKpSeDAn-P61IKyGjizX-IV-O5huWAYx5DU,10279
+clarifai/client/dataset.py,sha256=OgdpZkQ_vYmRxL8-qphcNozpvPV1bWTlte9Jv6UkKb8,35299
+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=DeB-es6rp0FJQVY46oO2eR_-Ux5tTmkM_qYsN1X6WcE,90793
+clarifai/client/model_client.py,sha256=4gIS0mKBdiNMA1x_6Wo6H7WbfLsmQix64EpONcQjQV4,37129
+clarifai/client/module.py,sha256=jLViQYvVV3FmRN_ivvbk83uwsx7CgYGeEx4dYAr6yD4,4537
+clarifai/client/nodepool.py,sha256=Qjk_-GfjtdlQdR5hHh28kRENW7rLi6zNqxeFYf7dioc,16473
+clarifai/client/pipeline.py,sha256=Hy3qnSX1pcoi-OAtdzr-qxkRYi1CxsaUzsfS3GDtETM,14358
+clarifai/client/runner.py,sha256=5xCiqByGGscfNm0IjHelhDTx8-9l8G0C3HL-3YZogK8,2253
+clarifai/client/search.py,sha256=3LLfATrdU43a0mRNITmJV-E53bhfafZkYsbwkTtlnyU,15661
+clarifai/client/user.py,sha256=ssfqHAE_GOh1c4dX74HH1xC09Rv4pVWxD08PTIfSjS8,22071
+clarifai/client/workflow.py,sha256=Bqh8lAmJcSbviJebckchTxucYlU11UQEhFSov7elpsk,13417
+clarifai/client/auth/__init__.py,sha256=7EwR0NrozkAUwpUnCsqXvE_p0wqx_SelXlSpKShKJK0,136
+clarifai/client/auth/helper.py,sha256=eitq1uQBoJ1TPjsTtbJ9g7eqT3mFm9vgc4HQR1s-zNg,16283
+clarifai/client/auth/register.py,sha256=xIg1OoZ_GGnc0KQKII-R6q_cKN1JXW3BGcU2eyS-Kkc,871
+clarifai/client/auth/stub.py,sha256=nKod8eNWaBnmsaNZ59dEt42TTcwC9mMgCzEMYPZggxc,9537
+clarifai/constants/base.py,sha256=ogmFSZYoF0YhGjHg5aiOc3MLqPr_poKAls6xaD0_C3U,89
+clarifai/constants/dataset.py,sha256=9saFE_VxCTCMlPTQbOU4KM_SzN_TQjwsrsFnpsc_Epo,556
+clarifai/constants/input.py,sha256=WcHwToUVIK9ItAhDefaSohQHCLNeR55PSjZ0BFnoZ3U,28
+clarifai/constants/model.py,sha256=ghcdvKIbx-g22H9RwxmMfUt1CrKz5Oa-iTExt0VpUVU,463
+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=1PFjIw30dRO9NSUa_yvox_07W30tlAnrk4mEAVgfXBY,10980
+clarifai/datasets/upload/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+clarifai/datasets/upload/base.py,sha256=8b13tmAokgZVH0d-HRpU-O15Swa46eJy2Fp6zU7jQkU,2441
+clarifai/datasets/upload/features.py,sha256=Yy0MP1F_ev2IbWnEyXDLt5SaEb11bXSUu5ux8tD1oxY,2115
+clarifai/datasets/upload/image.py,sha256=0Fs02KK1WuDE24RAJ2ZOdktva1iOqzE25J9jJJYJUDk,10122
+clarifai/datasets/upload/multimodal.py,sha256=IZYjmyxVJpNjwyNyzptQENZcCKDbdTU_t70QSdK23Ec,2762
+clarifai/datasets/upload/text.py,sha256=L2IEbickNhZcmrhbBfQciCXQf4JPcqWulRWJ7HDpJLc,2482
+clarifai/datasets/upload/utils.py,sha256=VdTn7AqIWB8D60rX9YD4T-4fi-7xMEocHATJvvaPjxQ,10508
+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=v6e8lh6BCUFOlurnvDfBEY-drl6JhpIE9nt2Qnz7XQ8,1623
+clarifai/datasets/upload/loaders/coco_detection.py,sha256=0FsGV84MT0IkerXoe9hdxQSr9wd96j9cPOne-rhzBPw,3190
+clarifai/datasets/upload/loaders/imagenet_classification.py,sha256=w2Tiot_YK0QYiyAVbWSxoKthZgUsRo9LiWWY2lVzPrQ,2247
+clarifai/datasets/upload/loaders/xview_detection.py,sha256=dukodZRdyVH-s5MtmCDtKwseKGwSCWAK2Uwuwa6kJu4,8035
+clarifai/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+clarifai/models/api.py,sha256=YhbGqZMC5mH_ZqnbaEmgfJdrIhSkLevpWih8ZqC3BL4,12193
+clarifai/modules/README.md,sha256=mx8pVx6cPp-pP4LcFPT_nX3ngGmhygVK0WiXeD3cbIo,367
+clarifai/modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+clarifai/modules/css.py,sha256=Jg67jXo2mZoGQp7M-At74ywyk4cN90s-aJTLjfqsaB4,2176
+clarifai/modules/pages.py,sha256=DZVO7YjIjYQdUyHYF-zxLkc5qkYC5KNnUxGiKQGo3Uk,1509
+clarifai/modules/style.css,sha256=j7FNPZVhLPj35vvBksAJ90RuX5sLuqzDR5iM2WIEhiA,6073
+clarifai/rag/__init__.py,sha256=wu3PzAzo7uqgrEzuaC9lY_3gj1HFiR3GU3elZIKTT5g,40
+clarifai/rag/rag.py,sha256=EG3GoFrHFCmA70Tz49_0Jo1-3WIaHSgWGHecPeErcdc,14170
+clarifai/rag/utils.py,sha256=_gVZdABuMnraCKViLruV75x0F3IpgFXN6amYSGE5_xc,4462
+clarifai/runners/__init__.py,sha256=wXLaSljH7qLeJCrZdKEnlQh2tNqTQAIZKWOu2rZ6wGs,279
+clarifai/runners/server.py,sha256=E7VtQcGuFYS_CIfrBIvdRVgjLwzHO9LEUpD_gtU-ye4,4873
+clarifai/runners/dockerfile_template/Dockerfile.template,sha256=nEnIMqzhAXDbd0Ht7QQm0U7AVPIHWQD6kYnFZ0TKJUM,2428
+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=6DEwMo-Aq5CSWxTgMNgQISq2H37HXlkPauOgnmSWhZk,64704
+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=RxhYawEX6vJKTZZ_AmcHwo2ONwT7V7e3W-vXuunY7eI,11874
+clarifai/runners/models/model_servicer.py,sha256=415RqiXCPH-1WgFHrtZMDH1nb8gaQDpVlrQ_tvtbjOg,4523
+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
+clarifai/runners/pipeline_steps/pipeline_step_builder.py,sha256=X2xWg4QM4kX3cr8JkG1RmAiVE8-XH8XbomjmXa16zmY,21333
+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=dzvHKgD-3B25KtQVCnCLz9rGzvgrxi4w7DoVkd6AxaI,14312
+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=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
+clarifai/runners/utils/url_fetcher.py,sha256=pGkrpj692FEvMsV6pNbmidfYKzQV7APzjja_fwd1Aig,4792
+clarifai/runners/utils/data_types/__init__.py,sha256=iBtmPkIoLK9ew6ZiXElGt2YBBTDLEA0fmxE_eOYzvhk,478
+clarifai/runners/utils/data_types/data_types.py,sha256=UBHTNPshr94qUs2KqkYis0VlArm23rcSG_EyO_Uetgc,19823
+clarifai/schema/search.py,sha256=o9-ct8ulLZByB3RCVwZWPgaDwdcW7cM5s-g8oyAz89s,2421
+clarifai/urls/helper.py,sha256=z6LnLGgLHxD8scFtyRdxqYIRJNhxqPkfLe53UtTLUBY,11727
+clarifai/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+clarifai/utils/cli.py,sha256=-r4qtzBQHRo3XeO4GSRwGHv5-_ndTcJ_4IkFY3eXvSg,11574
+clarifai/utils/config.py,sha256=jMSWYxJfp_D8eoGqz-HTdsngn5bg_1ymjLidYz6rdLA,7073
+clarifai/utils/constants.py,sha256=klcdSiVccrdJVRXL9qq08p2lWL3KerXcQOTIG6y9298,2414
+clarifai/utils/logging.py,sha256=0we53uTqUvzrulC86whu-oeWNxn1JjJL0OQ98Bwf9vo,15198
+clarifai/utils/misc.py,sha256=ld7_lls4CrjHzKchsSgJs4T-rVpUVEzRy4MDWBEQlyA,22812
+clarifai/utils/model_train.py,sha256=0XSAoTkSsrwf4f-W9yw2mkXZtkal7LBLJSoi86CFCn4,9250
+clarifai/utils/protobuf.py,sha256=VMhnNsPuWQ16VarKm8BOr5zccXMe26UlrxdJxIzEZNM,6220
+clarifai/utils/evaluation/__init__.py,sha256=PYkurUrXrGevByj7RFb6CoU1iC7fllyQSfnnlo9WnY8,69
+clarifai/utils/evaluation/helpers.py,sha256=0t6eIDXeZEoiVvnmHTnsIF_-v4BzrQW1hFaqc_psifU,21079
+clarifai/utils/evaluation/main.py,sha256=N_sfRuMjHrUeuWN0Pzms65M1PbkQkgYg3WoQVaDR1Jw,17764
+clarifai/utils/evaluation/testset_annotation_parser.py,sha256=Nmodfi5BYFYEbybWcC8tmU5-wtwRBsWIbnpd3OvKSmA,5414
+clarifai/workflows/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+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.6.5.dist-info/licenses/LICENSE,sha256=mUqF_d12-qE2n41g7C5_sq-BMLOcj6CNN-jevr15YHU,555
+clarifai-11.6.5.dist-info/METADATA,sha256=2wCjPH9Fg3ojVf27yb5CBrnA2a2ZGuzEoikvnf5U4cY,22737
+clarifai-11.6.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+clarifai-11.6.5.dist-info/entry_points.txt,sha256=X9FZ4Z-i_r2Ud1RpZ9sNIFYuu_-9fogzCMCRUD9hyX0,51
+clarifai-11.6.5.dist-info/top_level.txt,sha256=wUMdCQGjkxaynZ6nZ9FAnvBUCgp5RJUVFSy2j-KYo0s,9
+clarifai-11.6.5.dist-info/RECORD,,

{clarifai-11.6.4rc1.dist-info → clarifai-11.6.5.dist-info}/WHEEL

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

clarifai/cli/model_templates.py

@@ -1,243 +0,0 @@
-"""Templates for model initialization."""
-
-from clarifai import __version__
-
-
-def get_model_class_template() -> str:
-    """Return the template for a basic ModelClass-based model."""
-    return '''from typing import Iterator, List
-from clarifai.runners.models.model_class import ModelClass
-from clarifai.runners.util.data_utils import Param
-
-class MyModel(ModelClass):
-    """A custom model implementation using ModelClass."""
-
-    def load_model(self):
-        """Load the model here.
-        # TODO: please fill in
-        # Add your model loading logic here
-        """
-        pass
-
-    @ModelClass.method
-    def predict(
-        self,
-        prompt: str = "",
-        chat_history: List[dict] = None,
-        max_tokens: int = Param(default=256, description="The maximum number of tokens to generate. Shorter token lengths will provide faster performance."),
-        temperature: float = Param(default=1.0, description="A decimal number that determines the degree of randomness in the response"),
-        top_p: float = Param(default=1.0, description="An alternative to sampling with temperature, where the model considers the results of the tokens with top_p probability mass."),
-    ) -> str:
-        """This is the method that will be called when the runner is run. It takes in an input and returns an output."""
-        # TODO: please fill in
-        # Implement your prediction logic here
-        pass  # Replace with your actual logic
-
-    @ModelClass.method
-    def generate(
-        self,
-        prompt: str = "",
-        chat_history: List[dict] = None,
-        max_tokens: int = Param(default=256, description="The maximum number of tokens to generate. Shorter token lengths will provide faster performance."),
-        temperature: float = Param(default=1.0, description="A decimal number that determines the degree of randomness in the response"),
-        top_p: float = Param(default=1.0, description="An alternative to sampling with temperature, where the model considers the results of the tokens with top_p probability mass."),
-    ) -> Iterator[str]:
-        """Example yielding a streamed response."""
-        # TODO: please fill in
-        # Implement your generation logic here
-        pass # Replace with your actual logic
-'''
-
-
-def get_mcp_model_class_template() -> str:
-    """Return the template for an MCPModelClass-based model."""
-    return '''from typing import Any
-
-from fastmcp import FastMCP  # use fastmcp v2 not the built in mcp
-from pydantic import Field
-
-from clarifai.runners.models.mcp_class import MCPModelClass
-
-# TODO: please fill in
-# Configure your FastMCP server
-server = FastMCP("my-mcp-server", instructions="", stateless_http=True)
-
-
-# TODO: please fill in
-# Add your tools, resources, and prompts here
-@server.tool("example_tool", description="An example tool")
-def example_tool(input_param: Any = Field(description="Example input parameter")):
-    """Example tool implementation."""
-    # TODO: please fill in
-    # Implement your tool logic here
-    return f"Processed: {input_param}"
-
-
-# Static resource example
-@server.resource("config://version")
-def get_version():
-    """Example static resource."""
-    # TODO: please fill in
-    # Return your resource data
-    return "1.0.0"
-
-
-@server.prompt()
-def example_prompt(text: str) -> str:
-    """Example prompt template."""
-    # TODO: please fill in
-    # Define your prompt template
-    return f"Process this text: {text}"
-
-
-class MyModel(MCPModelClass):
-    """A custom model implementation using MCPModelClass."""
-
-    def get_server(self) -> FastMCP:
-        """Return the FastMCP server instance."""
-        return server
-'''
-
-
-def get_openai_model_class_template() -> str:
-    """Return the template for an OpenAIModelClass-based model."""
-    return '''from typing import List
-from openai import OpenAI
-from clarifai.runners.models.openai_class import OpenAIModelClass
-from clarifai.runners.util.data_utils import Param
-from clarifai.runners.utils.openai_convertor import build_openai_messages
-
-class MyModel(OpenAIModelClass):
-    """A custom model implementation using OpenAIModelClass."""
-
-    # TODO: please fill in
-    # Configure your OpenAI-compatible client for local model
-    client = OpenAI(
-        api_key="local-key",  # TODO: please fill in - use your local API key
-        base_url="http://localhost:8000/v1",  # TODO: please fill in - your local model server endpoint
-    )
-
-    # TODO: please fill in
-    # Specify the model name to use
-    model = "my-local-model"  # TODO: please fill in - replace with your local model name
-
-    def load_model(self):
-        """Optional: Add any additional model loading logic here."""
-        # TODO: please fill in (optional)
-        # Add any initialization logic if needed
-        pass
-
-    @OpenAIModelClass.method
-    def predict(
-        self,
-        prompt: str = "",
-        chat_history: List[dict] = None,
-        max_tokens: int = Param(default=256, description="The maximum number of tokens to generate. Shorter token lengths will provide faster performance."),
-        temperature: float = Param(default=1.0, description="A decimal number that determines the degree of randomness in the response"),
-        top_p: float = Param(default=1.0, description="An alternative to sampling with temperature, where the model considers the results of the tokens with top_p probability mass."),
-    ) -> str:
-        """Run a single prompt completion using the OpenAI client."""
-        # TODO: please fill in
-        # Implement your prediction logic here
-        messages = build_openai_messages(prompt, chat_history)
-        response = self.client.chat.completions.create(
-            model=self.model,
-            messages=messages,
-            max_completion_tokens=max_tokens,
-            temperature=temperature,
-            top_p=top_p,
-        )
-        return response.choices[0].message.content
-
-    @OpenAIModelClass.method
-    def generate(
-        self,
-        prompt: str = "",
-        chat_history: List[dict] = None,
-        max_tokens: int = Param(default=256, description="The maximum number of tokens to generate. Shorter token lengths will provide faster performance."),
-        temperature: float = Param(default=1.0, description="A decimal number that determines the degree of randomness in the response"),
-        top_p: float = Param(default=1.0, description="An alternative to sampling with temperature, where the model considers the results of the tokens with top_p probability mass."),
-    ):
-        """Stream a completion response using the OpenAI client."""
-        # TODO: please fill in
-        # Implement your streaming logic here
-        messages = build_openai_messages(prompt, chat_history)
-        stream = self.client.chat.completions.create(
-            model=self.model,
-            messages=messages,
-            max_completion_tokens=max_tokens,
-            temperature=temperature,
-            top_p=top_p,
-            stream=True,
-        )
-        for chunk in stream:
-            if chunk.choices:
-                text = (chunk.choices[0].delta.content
-                        if (chunk and chunk.choices[0].delta.content) is not None else '')
-                yield text
-'''
-
-
-def get_config_template(model_type_id: str = "text-to-text") -> str:
-    """Return the template for config.yaml."""
-    return f'''# Configuration file for your Clarifai model
-
-model:
-  id: "my-model"  # TODO: please fill in - replace with your model ID
-  user_id: "user_id"  # TODO: please fill in - replace with your user ID
-  app_id: "app_id"  # TODO: please fill in - replace with your app ID
-  model_type_id: "{model_type_id}"  # TODO: please fill in - replace if different model type ID
-
-build_info:
-  python_version: "3.12"
-
-# TODO: please fill in - adjust compute requirements for your model
-inference_compute_info:
-  cpu_limit: "1"  # TODO: please fill in - Amount of CPUs to use as a limit
-  cpu_memory: "1Gi"  # TODO: please fill in - Amount of CPU memory to use as a limit
-  cpu_requests: "0.5"  # TODO: please fill in - Amount of CPUs to use as a minimum
-  cpu_memory_requests: "512Mi"  # TODO: please fill in - Amount of CPU memory to use as a minimum
-  num_accelerators: 1  # TODO: please fill in - Amount of GPU/TPUs to use
-  accelerator_type: ["NVIDIA-*"]  # TODO: please fill in - type of accelerators requested
-  accelerator_memory: "1Gi"  # TODO: please fill in - Amount of accelerator/GPU memory to use as a minimum
-
-# TODO: please fill in (optional) - add checkpoints section if needed
-# checkpoints:
-#   type: "huggingface"  # supported type
-#   repo_id: "your-model-repo"  # for huggingface
-#   when: "build"  # or "runtime", "upload"
-'''
-
-
-def get_requirements_template(model_type_id: str = None) -> str:
-    """Return the template for requirements.txt."""
-    requirements = f'''# Clarifai SDK - required
-clarifai>={__version__}
-'''
-    if model_type_id == "mcp":
-        requirements += "fastmcp\n"
-    elif model_type_id == "openai":
-        requirements += "openai\n"
-    requirements += '''
-# TODO: please fill in - add your model's dependencies here
-# Examples:
-# torch>=2.0.0
-# transformers>=4.30.0
-# numpy>=1.21.0
-# pillow>=9.0.0
-'''
-    return requirements
-
-
-# Mapping of model type IDs to their corresponding templates
-MODEL_TYPE_TEMPLATES = {
-    "mcp": get_mcp_model_class_template,
-    "openai": get_openai_model_class_template,
-}
-
-
-def get_model_template(model_type_id: str = None) -> str:
-    """Get the appropriate model template based on model_type_id."""
-    if model_type_id in MODEL_TYPE_TEMPLATES:
-        return MODEL_TYPE_TEMPLATES[model_type_id]()
-    return get_model_class_template()