aipmodel 0.2.40__py3-none-any.whl

aipmodel/template.py

@@ -0,0 +1,67 @@
+from dotenv import load_dotenv
+from aipmodel.model_registry import MLOpsManager
+
+load_dotenv()
+
+manager = MLOpsManager(
+    # The following values will be picked up from the environment variables (.env) by default.
+    # You can uncomment and provide your own values below to override them.
+    
+    # CLEARML_API_SERVER_URL="your-endpoint_url",  
+    # CLEARML_USERNAME="your-clearml-username",    
+    # CLEARML_ACCESS_KEY="your-clearml-access-key",  
+    # CLEARML_SECRET_KEY="your-clearml-secret-key",
+    verbose=False  # Set to False to reduce logging output
+)
+
+# STEP 2: Upload local model
+print("\n--- STEP 2: Upload local model ---")
+local_model_id = manager.add_model(
+    source_type="local",
+    model_name="your_local_model",
+    source_path="path/to/your/local/model/folder",
+    code_path="path/to/your/local/model/model.py",  # Optional
+)
+print(f"Local Model ID: {local_model_id}\n")
+
+# STEP 3: Upload HuggingFace model
+print("\n--- STEP 3: uUpload HuggingFace model ---")
+hf_model_id = manager.add_model(
+    source_type="hf",
+    model_name="your_hf_model",
+    source_path="facebook/wav2vec2-base-960h"
+)
+print(f"HuggingFace Model ID: {hf_model_id}\n")
+
+# STEP 4: Upload model from your own S3 (e.g., AWS S3, MinIO, or Ceph)
+print("\n--- STEP 4: Upload model from your own S3 ---")
+s3_model_id = manager.add_model(
+    source_type="s3",
+    model_name="your_s3_model",
+    source_path="path/in/your/bucket/",  # Path to model in S3 bucket
+    code_path="path/to/your/local/model/model.py",  # Optional, remove if not using a script
+    external_ceph_endpoint_url="http://your-s3-endpoint.com",  # Example: "http://s3.example.com"
+    external_ceph_bucket_name="your-s3-bucket",
+    external_ceph_access_key="your-s3-access-key",
+    external_ceph_secret_key="your-s3-secret-key"
+)
+print(f"S3 Model ID: {s3_model_id}\n")
+
+# STEP 5: Download a model locally
+print("\n--- STEP 5: Download a model locally ---")
+manager.get_model(
+    model_name="your_hf_model",  # or any valid model name
+    local_dest="./downloaded_model/",
+)
+
+# STEP 6: List all models in your AIP project
+print("\n--- STEP 6: List all models ---")
+manager.list_models()
+
+# STEP 7: Get model information
+print("\n--- STEP 7: Get model information ---")
+manager.get_model_info("your_hf_model")
+
+# STEP 8: Delete a model
+print("\n--- STEP 8: Delete a model ---")
+manager.delete_model(model_id=local_model_id)

aipmodel/update_checker.py

@@ -0,0 +1,17 @@
+import importlib.metadata
+
+import requests
+
+
+def check_latest_version(package_name="aipmodel", auto_warn=True):
+    try:
+        current_version = importlib.metadata.version(package_name)
+        response = requests.get(f"https://pypi.org/pypi/{package_name}/json", timeout=5)
+        latest_version = response.json()["info"]["version"]
+        if current_version != latest_version:
+            if auto_warn:
+                print(f"[Update] A newer version ({latest_version}) is available. You are using {current_version}.")
+            return False, current_version, latest_version
+        return True, current_version, latest_version
+    except Exception:
+        return None, None, None

aipmodel-0.2.40.dist-info/METADATA

@@ -0,0 +1,26 @@
+Metadata-Version: 2.4
+Name: aipmodel
+Version: 0.2.40
+Summary: SDK for model registration, versioning, and storage
+Home-page: https://github.com/AIP-MLOPS/model-registry
+Author: AIP MLOPS Team
+Author-email: mohmmadweb@gmail.com
+Requires-Python: >=3.6
+Requires-Dist: boto3>=1.40.11
+Requires-Dist: clearml>=2.0.2
+Requires-Dist: fastapi>=0.115.0
+Requires-Dist: hf_xet
+Requires-Dist: huggingface-hub>=0.34.4
+Requires-Dist: pydantic>=2.9.2
+Requires-Dist: python-dotenv>=1.1.1
+Requires-Dist: requests>=2.32.4
+Requires-Dist: tabulate>=0.9.0
+Requires-Dist: tqdm>=4.67.1
+Requires-Dist: ruff
+Requires-Dist: uvicorn[standard]>=0.30.6
+Dynamic: author
+Dynamic: author-email
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

aipmodel-0.2.40.dist-info/RECORD

@@ -0,0 +1,9 @@
+aipmodel/CephS3Manager.py,sha256=wXd2UZTCWQDiT5TlZ5ftiXepHNBbzmluCyDcfcJdnd8,62124
+aipmodel/__init__.py,sha256=RjQjSCto30jJ_XZrDFTCzAcAAYRyQhISnzjXKbIMEmk,424
+aipmodel/model_registry.py,sha256=jK-q5KLlG0KmenRYGRXoV5ypnhsxO4--PRX6FRfJEhY,42411
+aipmodel/template.py,sha256=wwCxtawHZUmKnH3SGczs_wWHIBRmFlPZel2gSmSEZWg,2392
+aipmodel/update_checker.py,sha256=Jca6wpbRKKPuMC7D-OW7a-EmwRpDwGL49CKloKbKaM0,691
+aipmodel-0.2.40.dist-info/METADATA,sha256=tq9VJIkbCcqgGG6CZ3KtMfoJPiCb11ZdNmVyZCs8fDY,744
+aipmodel-0.2.40.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aipmodel-0.2.40.dist-info/top_level.txt,sha256=tCr0ZQgD1flFh3E9rxe4oN0IxpEse4RGcqB0l8hOlF4,9
+aipmodel-0.2.40.dist-info/RECORD,,

aipmodel-0.2.40.dist-info/WHEEL

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

aipmodel-0.2.40.dist-info/top_level.txt

@@ -0,0 +1 @@
+aipmodel