podstack 1.3.1__tar.gz → 1.3.3__tar.gz

{podstack-1.3.1 → podstack-1.3.3}/PKG-INFO

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

{podstack-1.3.1 → podstack-1.3.3}/podstack/__init__.py

@@ -54,7 +54,7 @@ Decorators:
         ...
 """
 
-__version__ = "1.3.1"
+__version__ = "1.3.2"
 
 from .client import Client
 from .notebook import Notebook, NotebookStatus
@@ -83,6 +83,7 @@ from . import gpu_runner
 from .gpu_runner import (
     GPURunner,
     GPUExecutionResult,
+    RunnerList,
     run as run_on_gpu,
     list_runners,
 )
@@ -209,6 +210,7 @@ __all__ = [
     "gpu_runner",
     "GPURunner",
     "GPUExecutionResult",
+    "RunnerList",
     "run_on_gpu",
     "list_runners",
     # Annotations

{podstack-1.3.1 → podstack-1.3.3}/podstack/annotations.py

@@ -142,6 +142,12 @@ class GPUConfig:
 
     def __call__(self, func: Callable) -> Callable:
         """Decorator usage - executes function on remote GPU."""
+        # Capture runner config at decoration time so it survives across
+        # multiple calls (the global is cleared after the first call).
+        global _current_runner_config
+        _captured_runner_config = _current_runner_config
+        _current_runner_config = None  # clear here, not inside wrapper
+
         @functools.wraps(func)
         def wrapper(*args, **kwargs):
             if not self.remote:
@@ -213,22 +219,20 @@ if __podstack_result__ is not None:
     print(repr(__podstack_result__))
 """
 
-            # Check for active runner config from @podstack.runner() decorator
-            global _current_runner_config
+            # Use the runner config captured at decoration time
             runner_name = None
             effective_pip = self.pip
             effective_uv = self.uv
             effective_conda = self.conda
-            if _current_runner_config is not None:
-                runner_name = _current_runner_config.get("image")
+            if _captured_runner_config is not None:
+                runner_name = _captured_runner_config.get("image")
                 # Merge package lists from runner config
-                if _current_runner_config.get("pip") and not effective_pip:
-                    effective_pip = _current_runner_config["pip"]
-                if _current_runner_config.get("uv") and not effective_uv:
-                    effective_uv = _current_runner_config["uv"]
-                if _current_runner_config.get("conda") and not effective_conda:
-                    effective_conda = _current_runner_config["conda"]
-                _current_runner_config = None  # Clear after use
+                if _captured_runner_config.get("pip") and not effective_pip:
+                    effective_pip = _captured_runner_config["pip"]
+                if _captured_runner_config.get("uv") and not effective_uv:
+                    effective_uv = _captured_runner_config["uv"]
+                if _captured_runner_config.get("conda") and not effective_conda:
+                    effective_conda = _captured_runner_config["conda"]
 
             # Execute on remote GPU with improved error handling
             # Auto-enable streaming in Jupyter notebooks for real-time output

{podstack-1.3.1 → podstack-1.3.3}/podstack/gpu_runner.py

@@ -35,6 +35,33 @@ def is_jupyter() -> bool:
     return False
 
 
+class RunnerList(list):
+    """A list of runners with a .show() method for tabular display."""
+
+    def show(self):
+        """Print runners in a formatted table with S.NO."""
+        if not self:
+            print("No runners available.")
+            return
+
+        # Column widths
+        no_w = 5
+        name_w = 45
+        type_w = 6
+        desc_w = 50
+
+        header = f"{'S.NO':<{no_w}} {'Name':<{name_w}} {'Type':<{type_w}} {'Description'}"
+        print(header)
+        print("-" * (no_w + name_w + type_w + desc_w + 3))
+        for i, r in enumerate(self, 1):
+            name = r.get("name", "")
+            rtype = r.get("type", "")
+            desc = r.get("description", "")
+            if len(desc) > desc_w:
+                desc = desc[:desc_w - 3] + "..."
+            print(f"{i:<{no_w}} {name:<{name_w}} {rtype:<{type_w}} {desc}")
+
+
 class OutputStreamer:
     """Handles real-time output streaming from GPU executions."""
 
@@ -184,7 +211,7 @@ class GPUExecutionResult:
             output=data.get("output", ""),
             error=data.get("error"),
             gpu_seconds=data.get("gpu_seconds", 0.0),
-            cost_paise=data.get("actual_cost_paise", 0),
+            cost_paise=data.get("cost_cents", data.get("cost_paise", data.get("actual_cost_paise", 0))),
             gpu_type=data.get("gpu_type", ""),
             gpu_count=data.get("gpu_count", 1),
             success=data.get("status") == "completed"
@@ -978,12 +1005,16 @@ _stream_install(
         # Fallback: use the same base URL
         return self.api_url
 
-    def list_runners(self) -> list:
+    def list_runners(self) -> RunnerList:
         """
         List available GPU/CPU runners.
 
         Returns:
-            List of runner dictionaries with name, type, description, image, libraries, scenarios.
+            RunnerList of runner dictionaries. Call .show() for tabular display.
+
+        Example:
+            podstack.list_runners().show()  # Print table with S.NO
+            runners = podstack.list_runners()  # Just fetch list
         """
         platform_url = self._get_platform_url()
         url = f"{platform_url}/api/v1/runners"
@@ -1004,7 +1035,7 @@ _stream_install(
                 raise RuntimeError(f"Failed to list runners: {error_msg}")
 
             try:
-                return response.json()
+                return RunnerList(response.json())
             except Exception:
                 raise RuntimeError(f"Invalid JSON response: {response.text[:200]}")
 
@@ -1181,17 +1212,16 @@ def run(
     )
 
 
-def list_runners() -> list:
+def list_runners() -> RunnerList:
     """
     List available GPU/CPU runners.
 
     Returns:
-        List of runner dictionaries with name, type, description, image, libraries, scenarios.
+        RunnerList of runner dicts. Call .show() for tabular display.
 
     Example:
-        runners = podstack.gpu_runner.list_runners()
-        for r in runners:
-            print(f"{r['name']} ({r['type']}): {r['description']}")
+        podstack.list_runners().show()   # Print table with S.NO
+        runners = podstack.list_runners()  # Just fetch list
     """
     return get_runner().list_runners()
 

{podstack-1.3.1 → podstack-1.3.3}/podstack.egg-info/PKG-INFO

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

{podstack-1.3.1 → podstack-1.3.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "podstack"
-version = "1.3.1"
+version = "1.3.3"
 description = "Official Python SDK for Podstack GPU Notebook Platform"
 readme = "README.md"
 license = "MIT"