tetra-rp 0.12.0__py3-none-any.whl → 0.13.0__py3-none-any.whl

tetra_rp/cli/utils/skeleton.py

@@ -0,0 +1,101 @@
+"""Project skeleton creation utilities."""
+
+from pathlib import Path
+from typing import Dict, List, Any
+
+from tetra_rp.config import get_paths
+
+
+def get_template_directory() -> Path:
+    """Get the path to the templates directory."""
+    return Path(__file__).parent.parent / "templates"
+
+
+def load_template_files(template_name: str) -> Dict[str, Any]:
+    """Load template files from filesystem."""
+    template_dir = get_template_directory() / template_name
+
+    if not template_dir.exists():
+        raise ValueError(f"Template '{template_name}' not found in {template_dir}")
+
+    files = {}
+
+    # Load all files from the template directory
+    for file_path in template_dir.iterdir():
+        if file_path.is_file():
+            relative_path = file_path.name
+
+            # Special handling for config.json - return as callable that generates tetra config
+            if file_path.name == "config.json":
+                config_content = file_path.read_text()
+                files[".tetra/config.json"] = lambda content=config_content: content
+            else:
+                files[relative_path] = file_path.read_text()
+
+    return files
+
+
+def get_available_templates() -> Dict[str, Dict[str, Any]]:
+    """Get available project templates from filesystem."""
+    template_dir = get_template_directory()
+    templates = {}
+
+    # Template descriptions
+    descriptions = {
+        "basic": "Simple remote function example",
+        "advanced": "Multi-function project with dependencies",
+        "gpu-compute": "GPU-optimized compute workload",
+        "web-api": "FastAPI web service deployment",
+    }
+
+    # Discover templates from filesystem
+    for template_path in template_dir.iterdir():
+        if template_path.is_dir():
+            template_name = template_path.name
+            try:
+                templates[template_name] = {
+                    "description": descriptions.get(
+                        template_name, f"{template_name} template"
+                    ),
+                    "files": load_template_files(template_name),
+                }
+            except Exception as e:
+                print(f"Warning: Failed to load template '{template_name}': {e}")
+
+    return templates
+
+
+def create_project_skeleton(
+    template_name: str, template_info: Dict[str, Any], force: bool = False
+) -> List[str]:
+    """Create project skeleton from template."""
+    created_files = []
+
+    # Create .tetra directory using centralized config
+    paths = get_paths()
+    paths.ensure_tetra_dir()
+
+    # Create files from template
+    for file_path, content in template_info["files"].items():
+        path = Path(file_path)
+
+        # Create parent directories if needed
+        path.parent.mkdir(parents=True, exist_ok=True)
+
+        # Skip existing files unless force is True
+        if path.exists() and not force:
+            continue
+
+        # Get content (could be string or callable)
+        if callable(content):
+            file_content = content()
+        else:
+            file_content = content
+
+        # Write file
+        with open(path, "w") as f:
+            f.write(file_content)
+
+        created_files.append(str(path))
+
+    return created_files

tetra_rp/client.py

@@ -15,7 +15,6 @@ def remote(
     dependencies: Optional[List[str]] = None,
     system_dependencies: Optional[List[str]] = None,
     accelerate_downloads: bool = True,
-    hf_models_to_cache: Optional[List[str]] = None,
     **extra,
 ):
     """
@@ -33,8 +32,6 @@ def remote(
             environment before executing the function. Defaults to None.
         accelerate_downloads (bool, optional): Enable download acceleration for dependencies and models.
             Defaults to True.
-        hf_models_to_cache (List[str], optional): List of HuggingFace model IDs to pre-cache using
-            download acceleration. Defaults to None.
         extra (dict, optional): Additional parameters for the execution of the resource. Defaults to an empty dict.
 
     Returns:
@@ -47,7 +44,6 @@ def remote(
             resource_config=my_resource_config,
             dependencies=["numpy", "pandas"],
             accelerate_downloads=True,
-            hf_models_to_cache=["gpt2", "bert-base-uncased"]
         )
         async def my_function(data):
             # Function logic here
@@ -64,7 +60,6 @@ def remote(
                 dependencies,
                 system_dependencies,
                 accelerate_downloads,
-                hf_models_to_cache,
                 extra,
             )
         else:
@@ -82,7 +77,6 @@ def remote(
                     dependencies,
                     system_dependencies,
                     accelerate_downloads,
-                    hf_models_to_cache,
                     *args,
                     **kwargs,
                 )

tetra_rp/config.py

@@ -0,0 +1,29 @@
+"""Configuration management for tetra-rp CLI."""
+
+from pathlib import Path
+from typing import NamedTuple
+
+
+class TetraPaths(NamedTuple):
+    """Paths for tetra-rp configuration and data."""
+
+    tetra_dir: Path
+    config_file: Path
+    deployments_file: Path
+
+    def ensure_tetra_dir(self) -> None:
+        """Ensure the .tetra directory exists."""
+        self.tetra_dir.mkdir(exist_ok=True)
+
+
+def get_paths() -> TetraPaths:
+    """Get standardized paths for tetra-rp configuration."""
+    tetra_dir = Path.cwd() / ".tetra"
+    config_file = tetra_dir / "config.json"
+    deployments_file = tetra_dir / "deployments.json"
+
+    return TetraPaths(
+        tetra_dir=tetra_dir,
+        config_file=config_file,
+        deployments_file=deployments_file,
+    )

tetra_rp/execute_class.py

@@ -203,7 +203,6 @@ def create_remote_class(
     dependencies: Optional[List[str]],
     system_dependencies: Optional[List[str]],
     accelerate_downloads: bool,
-    hf_models_to_cache: Optional[List[str]],
     extra: dict,
 ):
     """
@@ -222,7 +221,6 @@ def create_remote_class(
             self._dependencies = dependencies or []
             self._system_dependencies = system_dependencies or []
             self._accelerate_downloads = accelerate_downloads
-            self._hf_models_to_cache = hf_models_to_cache
             self._extra = extra
             self._constructor_args = args
             self._constructor_kwargs = kwargs
@@ -307,7 +305,6 @@ def create_remote_class(
                     dependencies=self._dependencies,
                     system_dependencies=self._system_dependencies,
                     accelerate_downloads=self._accelerate_downloads,
-                    hf_models_to_cache=self._hf_models_to_cache,
                     instance_id=self._instance_id,
                     create_new_instance=not hasattr(
                         self, "_stub"

tetra_rp/protos/remote_execution.py

@@ -81,10 +81,6 @@ class FunctionRequest(BaseModel):
         default=True,
         description="Enable download acceleration for dependencies and models",
     )
-    hf_models_to_cache: Optional[List[str]] = Field(
-        default=None,
-        description="List of HuggingFace model IDs to pre-cache using acceleration",
-    )
 
     @model_validator(mode="after")
     def validate_execution_requirements(self) -> "FunctionRequest":

tetra_rp/stubs/live_serverless.py

@@ -68,7 +68,6 @@ class LiveServerlessStub(RemoteExecutorStub):
         dependencies,
         system_dependencies,
         accelerate_downloads,
-        hf_models_to_cache,
         *args,
         **kwargs,
     ):
@@ -79,7 +78,6 @@ class LiveServerlessStub(RemoteExecutorStub):
             "dependencies": dependencies,
             "system_dependencies": system_dependencies,
             "accelerate_downloads": accelerate_downloads,
-            "hf_models_to_cache": hf_models_to_cache,
         }
 
         # Thread-safe cache access
@@ -110,7 +108,7 @@ class LiveServerlessStub(RemoteExecutorStub):
 
         if response.stdout:
             for line in response.stdout.splitlines():
-                log.info(f"Remote | {line}")
+                print(line)
 
         if response.success:
             if response.result is None:

tetra_rp/stubs/registry.py

@@ -31,7 +31,6 @@ def _create_live_serverless_stub(resource, **extra):
         dependencies,
         system_dependencies,
         accelerate_downloads,
-        hf_models_to_cache,
         *args,
         **kwargs,
     ) -> dict:
@@ -43,7 +42,6 @@ def _create_live_serverless_stub(resource, **extra):
             dependencies,
             system_dependencies,
             accelerate_downloads,
-            hf_models_to_cache,
             *args,
             **kwargs,
         )
@@ -78,7 +76,6 @@ def _(resource, **extra):
         dependencies,
         system_dependencies,
         accelerate_downloads,
-        hf_models_to_cache,
         *args,
         **kwargs,
     ) -> dict:
@@ -103,7 +100,6 @@ def _(resource, **extra):
         dependencies,
         system_dependencies,
         accelerate_downloads,
-        hf_models_to_cache,
         *args,
         **kwargs,
     ) -> dict:

{tetra_rp-0.12.0.dist-info → tetra_rp-0.13.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tetra_rp
-Version: 0.12.0
+Version: 0.13.0
 Summary: A Python library for distributed inference and serving of machine learning models
 Author-email: Marut Pandya <pandyamarut@gmail.com>, Patrick Rachford <prachford@icloud.com>, Dean Quinanola <dean.quinanola@runpod.io>
 License: MIT
@@ -14,6 +14,9 @@ Requires-Dist: cloudpickle>=3.1.1
 Requires-Dist: runpod
 Requires-Dist: python-dotenv>=1.0.0
 Requires-Dist: pydantic>=2.0.0
+Requires-Dist: rich>=14.0.0
+Requires-Dist: typer>=0.12.0
+Requires-Dist: questionary>=2.0.0
 
 # Tetra: Serverless computing for AI workloads
 
@@ -359,6 +362,7 @@ if __name__ == "__main__":
 ```python
 import asyncio
 from tetra_rp import remote, LiveServerless, GpuGroup, PodTemplate
+import base64
 
 # Advanced GPU configuration with consolidated template overrides
 sd_config = LiveServerless(

{tetra_rp-0.12.0.dist-info → tetra_rp-0.13.0.dist-info}/RECORD

@@ -1,7 +1,24 @@
 tetra_rp/__init__.py,sha256=_D3Wbtv9tBh_WGS4Si5uQbsefL63GqqjoGTN3R8P6fA,769
-tetra_rp/client.py,sha256=urSVh0j9didd9U8lboPv3TtFYURp2XO6ReOICr9Xrls,3414
-tetra_rp/execute_class.py,sha256=jYNFalqqjKvvCz1zzodRvOkrLQd2FYnLYa4EElEYp8w,12243
+tetra_rp/client.py,sha256=cJRDCzMcU7OMwvlFSV5thjOBMysRWnzHFGko-ebYHBI,3073
+tetra_rp/config.py,sha256=9FYzouOam0PVoBIITpwHu1N5dGd1ueLHjWMaZYmdquI,760
+tetra_rp/execute_class.py,sha256=pm-wupP_rGwlmS_3B5O2g8a5qINlVSWuGUc1E7pveUw,12075
 tetra_rp/logger.py,sha256=gk5-PWp3k_GQ5DxndsRkBCX0jarp_3lgZ1oiTFuThQg,1125
+tetra_rp/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tetra_rp/cli/main.py,sha256=89rSSHWG9fxXFmCR0QzRIQUd9N_nvlEy3ImzQdGs8jw,2090
+tetra_rp/cli/commands/__init__.py,sha256=gQ6tnU0Rvm0-ESWFUBU-KDl5dpNOpUTG509hXOQQjwY,27
+tetra_rp/cli/commands/deploy.py,sha256=dznG7NI04ihfccn5YIjBbIxMRM19h-B60jTTWazGppc,10609
+tetra_rp/cli/commands/init.py,sha256=smRqSrxSzW_Rh-HBOLDdxOw-SDVC6uNOWTJAjF078c4,3020
+tetra_rp/cli/commands/resource.py,sha256=XMsSAYaCtw42tUz7vJsK53lZRrKX51fMMfQv8sdExlc,5679
+tetra_rp/cli/commands/run.py,sha256=XHbsTqzgz48qA_I1rc9MiZAh2H2OKRMNr6JPOGVaoXI,3781
+tetra_rp/cli/templates/advanced/main.py,sha256=gfPETln1Re0R7Ysd8WlMon1myFmCtpJk6k6a1AGnK0Q,1221
+tetra_rp/cli/templates/advanced/utils.py,sha256=8BfZW2_qXS7bYyiO62KhEib_TqUfIxer2jqXoIctK1c,647
+tetra_rp/cli/templates/basic/main.py,sha256=fItaje-ahvQKsvVcA_HNVCy0grw9fLb1kd07eU5jvAQ,677
+tetra_rp/cli/templates/gpu-compute/main.py,sha256=x-8W2ciT_bnLACcdKrMPwX4HqkLCeX-RKC1E6IzXnAI,1602
+tetra_rp/cli/templates/web-api/api.py,sha256=7ucd6Mg9Y3aVNuG21Jh_ToiBWe0R3NaYIN-dExaRN_U,2009
+tetra_rp/cli/templates/web-api/main.py,sha256=lRu0CVbPuCfvVQrp5IQf4KlR1dpC70fd7J008GXmtvs,846
+tetra_rp/cli/utils/__init__.py,sha256=Ytcdqe_Kzy37LSnsBj1Drkrx0Uo7TWExjQZ2mwW_nMg,27
+tetra_rp/cli/utils/deployment.py,sha256=2hskoGEnZYTJMR2on37X4kaQEtBDCKoWK_7RQr1VEY0,5123
+tetra_rp/cli/utils/skeleton.py,sha256=zOo1sudIPA8sJa3tToV-XpkJMdsMtYzYAqSv-QH9nq4,3184
 tetra_rp/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tetra_rp/core/api/__init__.py,sha256=oldrEKMwxYoBPLvPfVlaFS3wfUtTTxCN6-HzlpTh6vE,124
 tetra_rp/core/api/runpod.py,sha256=3TTx1fkXMLZ2R5JCrQYPEn8dhdUsBt8i5OEwAfaKQ_k,10451
@@ -27,12 +44,13 @@ tetra_rp/core/utils/json.py,sha256=q0r7aEdfh8kKVeHGeh9fBDfuhHYNopSreislAMB6HhM,1
 tetra_rp/core/utils/lru_cache.py,sha256=drwKg-DfLbeBRGTzuxKqNKMQq0EuZV15LMTZIOyZuVk,2618
 tetra_rp/core/utils/singleton.py,sha256=lSXgEQGX9nzhrc05GMpThn9SHKG45iajBbSEtwCcNyI,632
 tetra_rp/protos/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tetra_rp/protos/remote_execution.py,sha256=flKJG0U4ked84cXyF4Gfs_7fBgLsEVOzBv8ZWB9UlP0,5648
+tetra_rp/protos/remote_execution.py,sha256=InVGKgkShi-muoNjGz-l7-cBwOZzaTnBdFr6LhVE9JE,5482
 tetra_rp/stubs/__init__.py,sha256=ozKsHs8q0T7o2qhQEquub9hqomh1Htys53mMraaRu2E,72
-tetra_rp/stubs/live_serverless.py,sha256=2bQ47Hq4D8o3YydkxR6QVuLjYrzYS427r3067beYK2A,4550
-tetra_rp/stubs/registry.py,sha256=akJREObyvTAVp6Rvxw8O49W52eJAjByleLU_phMyaDI,3351
+tetra_rp/stubs/live_serverless.py,sha256=gIjhizgX5MDifxXo5JBONS3fmGipwRAgv9_sydc_cxU,4451
+tetra_rp/stubs/registry.py,sha256=OdqAHFJIhPBAaR-S_spJNgRwpO96xEqhfAENhomqmI4,3235
 tetra_rp/stubs/serverless.py,sha256=BM_a5Ml5VADBYu2WRNmo9qnicP8NnXDGl5ywifulbD0,947
-tetra_rp-0.12.0.dist-info/METADATA,sha256=u2OHbJV-8KIiOca3SenK2J6I8ms13P8Jc64KGNhFAyE,28077
-tetra_rp-0.12.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-tetra_rp-0.12.0.dist-info/top_level.txt,sha256=bBay7JTDwJXsTYvVjrwno9hnF-j0q272lk65f2AcPjU,9
-tetra_rp-0.12.0.dist-info/RECORD,,
+tetra_rp-0.13.0.dist-info/METADATA,sha256=9IiJ4LYhMcTUkf2Qgp2U_rh8y8XdAK0WDGe8tWSFXpE,28182
+tetra_rp-0.13.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+tetra_rp-0.13.0.dist-info/entry_points.txt,sha256=FJi3ytBLKCcyz4tXJhOM3XNKzTxpnl3AauiPH2lHvWY,48
+tetra_rp-0.13.0.dist-info/top_level.txt,sha256=bBay7JTDwJXsTYvVjrwno9hnF-j0q272lk65f2AcPjU,9
+tetra_rp-0.13.0.dist-info/RECORD,,

tetra_rp-0.13.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+flash = tetra_rp.cli.main:app