podstack 1.3.5__tar.gz → 1.3.7__tar.gz

{podstack-1.3.5 → podstack-1.3.7}/PKG-INFO

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

{podstack-1.3.5 → podstack-1.3.7}/podstack/gpu_runner.py

@@ -205,10 +205,17 @@ class GPUExecutionResult:
 
     @classmethod
     def from_dict(cls, data: Dict[str, Any]) -> "GPUExecutionResult":
+        # The result API returns output as []ExecutionOutputResponse objects.
+        # Join their content fields into a single string.
+        raw_output = data.get("output", "")
+        if isinstance(raw_output, list):
+            output = "".join(item.get("content", "") for item in raw_output if item.get("content"))
+        else:
+            output = raw_output or ""
         return cls(
             execution_id=data.get("execution_id", ""),
             status=data.get("status", "unknown"),
-            output=data.get("output", ""),
+            output=output,
             error=data.get("error"),
             gpu_seconds=data.get("gpu_seconds", 0.0),
             cost_paise=data.get("cost_cents", data.get("cost_paise", data.get("actual_cost_paise", 0))),
@@ -437,7 +444,8 @@ def _stream_install(cmd, description):
 # Install pip packages using uv (faster)
 _pip_packages = {repr(pip_list)}
 _stream_install(
-    ["uv", "pip", "install", "--system", "--progress"] + _pip_packages,
+    ["uv", "pip", "install", "--system", "--progress",
+     "--extra-index-url", "https://pypi.org/simple/"] + _pip_packages,
     f"Installing pip packages with uv: {{' '.join(_pip_packages)}}"
 )
 ''')
@@ -446,7 +454,8 @@ _stream_install(
 # Install pip packages
 _pip_packages = {repr(pip_list)}
 _stream_install(
-    [sys.executable, "-m", "pip", "install", "--progress-bar", "on"] + _pip_packages,
+    [sys.executable, "-m", "pip", "install", "--progress-bar", "on",
+     "--extra-index-url", "https://pypi.org/simple/"] + _pip_packages,
     f"Installing pip packages: {{' '.join(_pip_packages)}}"
 )
 ''')
@@ -812,10 +821,12 @@ _stream_install(
                     raise ConnectionError(f"Failed to get result after {max_retries} attempts: {e}")
                 time.sleep(2 * (attempt + 1))
 
-        # Prefer the streamed output over the API result's output field,
-        # since __PODSTACK_RESULT__ is printed to stdout during execution.
-        if output_buffer:
-            result.output = "".join(output_buffer)
+        # If the API result's output doesn't have the result marker but the
+        # streaming buffer does (e.g. race between completed status and last
+        # stdout event), use the streaming buffer instead.
+        streamed = "".join(output_buffer)
+        if "__PODSTACK_RESULT__" not in result.output and "__PODSTACK_RESULT__" in streamed:
+            result.output = streamed
 
         if result.success:
             print(f"\n[Podstack] Completed in {result.gpu_seconds:.1f}s (cost: ₹{result.cost_paise/100:.2f})")

{podstack-1.3.5 → podstack-1.3.7}/podstack.egg-info/PKG-INFO

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

{podstack-1.3.5 → podstack-1.3.7}/pyproject.toml

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