podstack 1.3.20__tar.gz → 1.4.0__tar.gz

{podstack-1.3.20 → podstack-1.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: podstack
-Version: 1.3.20
+Version: 1.4.0
 Summary: Official Python SDK for Podstack GPU Notebook Platform
 Author-email: Podstack <support@podstack.ai>
 License-Expression: MIT

{podstack-1.3.20 → podstack-1.4.0}/podstack/__init__.py

@@ -54,7 +54,7 @@ Decorators:
         ...
 """
 
-__version__ = "1.3.2"
+__version__ = "1.4.0"
 
 from .client import Client
 from .notebook import Notebook, NotebookStatus
@@ -86,6 +86,31 @@ from .gpu_runner import (
     RunnerList,
     run as run_on_gpu,
     list_runners,
+    get_runner_info,
+    pip_install,
+)
+
+# Pipeline module import
+from . import pipelines
+from .pipelines import (
+    step,
+    pipeline,
+    StepConfig,
+    PipelineDef,
+    list_pipelines,
+    get_pipeline,
+    get_pipeline_run,
+    trigger_pipeline,
+    cancel_pipeline_run,
+    archive_pipeline,
+)
+
+# Storage module import
+from . import storage
+from .storage import (
+    StorageClient, BucketList, VolumeList,
+    create_bucket, get_bucket, list_buckets, delete_bucket,
+    create_volume, get_volume, list_volumes, delete_volume,
 )
 
 # Annotations module import
@@ -103,6 +128,13 @@ from .annotations import (
     get_environment,
     get_auto_shutdown_minutes,
     enable_remote_execution,
+    deploy,
+    DeployConfig,
+)
+from .gpu_runner import (
+    get_deployment,
+    stop_deployment,
+    delete_deployment,
 )
 
 
@@ -206,6 +238,31 @@ __all__ = [
     "WalletBalance",
     # Registry
     "registry",
+    # Pipelines
+    "pipelines",
+    "step",
+    "pipeline",
+    "StepConfig",
+    "PipelineDef",
+    "list_pipelines",
+    "get_pipeline",
+    "get_pipeline_run",
+    "trigger_pipeline",
+    "cancel_pipeline_run",
+    "archive_pipeline",
+    # Storage
+    "storage",
+    "StorageClient",
+    "BucketList",
+    "VolumeList",
+    "create_bucket",
+    "get_bucket",
+    "list_buckets",
+    "delete_bucket",
+    "create_volume",
+    "get_volume",
+    "list_volumes",
+    "delete_volume",
     # GPU Runner
     "gpu_runner",
     "GPURunner",
@@ -213,6 +270,8 @@ __all__ = [
     "RunnerList",
     "run_on_gpu",
     "list_runners",
+    "get_runner_info",
+    "pip_install",
     # Annotations
     "annotations",
     "gpu",
@@ -227,4 +286,10 @@ __all__ = [
     "get_environment",
     "get_auto_shutdown_minutes",
     "enable_remote_execution",
+    # Deploy
+    "deploy",
+    "DeployConfig",
+    "get_deployment",
+    "stop_deployment",
+    "delete_deployment",
 ]

{podstack-1.3.20 → podstack-1.4.0}/podstack/annotations.py

@@ -37,6 +37,7 @@ from typing import Optional, Dict, Any, Callable, Union
 from contextlib import contextmanager
 
 from . import registry
+from .pipelines import step, pipeline, StepConfig, PipelineDef
 from .gpu_runner import (
     GPURunner,
     GPUExecutionResult,
@@ -802,6 +803,154 @@ class ModelRegistry:
 model = ModelRegistry()
 
 
+class DeployConfig:
+    """
+    Deploy configuration decorator.
+
+    Deploys the decorated function as a persistent REST API endpoint.
+
+    Usage:
+        @podstack.deploy(name="sentiment-api", gpu="A10", fraction=50)
+        def predict(data):
+            return {"sentiment": "positive", "text": data["text"]}
+
+        result = predict()  # Deploys and returns endpoint info
+    """
+
+    def __init__(
+        self,
+        name: str,
+        gpu: str = "A10",
+        count: int = 1,
+        fraction: int = 100,
+        pip: Union[str, list] = None,
+        uv: Union[str, list] = None,
+        function: str = None,
+        image: str = None,
+    ):
+        self.name = name
+        self.gpu = gpu
+        self.count = count
+        self.fraction = fraction
+        self.pip = pip
+        self.uv = uv
+        self.function = function
+        self.image = image
+
+    def __call__(self, func: Callable) -> Callable:
+        @functools.wraps(func)
+        def wrapper(*args, **kwargs):
+            runner = get_runner()
+
+            # Get function source
+            source = inspect.getsource(func)
+            source = textwrap.dedent(source)
+
+            # Remove decorator lines
+            lines = source.split('\n')
+            clean_lines = []
+            skip_decorator = False
+            paren_depth = 0
+            for line in lines:
+                stripped = line.strip()
+                if stripped.startswith('@podstack.deploy') or stripped.startswith('@deploy'):
+                    skip_decorator = True
+                    paren_depth += line.count('(') - line.count(')')
+                    if paren_depth <= 0:
+                        skip_decorator = False
+                    continue
+                if skip_decorator:
+                    paren_depth += line.count('(') - line.count(')')
+                    if paren_depth <= 0:
+                        skip_decorator = False
+                    continue
+                if stripped.startswith('@'):
+                    continue
+                clean_lines.append(line)
+            source = '\n'.join(clean_lines)
+
+            function_name = self.function or func.__name__
+            pip_packages = self.pip
+            if isinstance(pip_packages, list):
+                pip_packages = " ".join(pip_packages)
+
+            # POST to deployment API
+            import httpx
+            headers = runner._get_headers()
+            url = f"{runner.api_url}/api/v1/deployments/from-source"
+            with httpx.Client(timeout=30.0) as client:
+                response = client.post(url, headers=headers, json={
+                    "name": self.name,
+                    "source": source,
+                    "function_name": function_name,
+                    "gpu_type": self.gpu,
+                    "gpu_fraction": self.fraction,
+                    "gpu_count": self.count,
+                    "pip_packages": pip_packages or "",
+                    "image": self.image or "",
+                })
+                if response.status_code >= 400:
+                    raise PodstackError(f"Deployment failed: {response.text}")
+                data = response.json()
+
+            deployment_id = data.get("deployment_id")
+            print(f"[Podstack] Deploying '{self.name}' as REST API...")
+            print(f"[Podstack] Deployment ID: {deployment_id}")
+            print(f"[Podstack] Status: {data.get('status')}")
+
+            # Poll for active status (up to 120s)
+            status_url = f"{runner.api_url}/api/v1/deployments/{deployment_id}"
+            for _ in range(24):
+                time.sleep(5)
+                with httpx.Client(timeout=30.0) as client:
+                    status_resp = client.get(status_url, headers=headers)
+                    if status_resp.status_code >= 400:
+                        continue
+                    status_data = status_resp.json()
+                status = status_data.get("status")
+                if status == "active":
+                    endpoint = status_data.get("service_url", "")
+                    print(f"[Podstack] Endpoint live: POST {endpoint}/predict")
+                    return {"deployment_id": deployment_id, "status": "active", "endpoint": f"{endpoint}/predict"}
+                elif status == "failed":
+                    error = status_data.get("error_message", "Unknown error")
+                    print(f"[Podstack] Deployment failed: {error}")
+                    raise PodstackError(f"Deployment failed: {error}")
+                print(f"[Podstack] Status: {status}...")
+
+            print(f"[Podstack] Deployment still starting. Check status with:")
+            print(f"  podstack.get_deployment('{deployment_id}')")
+            return {"deployment_id": deployment_id, "status": "starting"}
+
+        return wrapper
+
+
+def deploy(
+    name: str,
+    gpu: str = "A10",
+    count: int = 1,
+    fraction: int = 100,
+    pip: Union[str, list] = None,
+    uv: Union[str, list] = None,
+    function: str = None,
+    image: str = None,
+) -> DeployConfig:
+    """Deploy a function as a persistent REST API endpoint.
+
+    Examples:
+        @podstack.deploy(name="sentiment-api", gpu="A10", fraction=25)
+        def predict(data):
+            return {"sentiment": "positive"}
+
+        result = predict()  # Deploys and returns endpoint info
+        print(result["endpoint"])
+    """
+    return DeployConfig(
+        name=name, gpu=gpu, count=count, fraction=fraction,
+        pip=pip, uv=uv, function=function, image=image,
+    )
+
+
 def get_gpu_config() -> Dict[str, Any]:
     """Get current GPU configuration."""
     return _current_gpu_config.copy()
@@ -832,6 +981,13 @@ __all__ = [
     "enable_remote_execution",
     "is_remote_execution_enabled",
     "GPUConfig",
+    "deploy",
+    "DeployConfig",
+    # Pipeline
+    "step",
+    "pipeline",
+    "StepConfig",
+    "PipelineDef",
     # Exceptions
     "PodstackError",
     "PodstackTimeoutError",