podstack 1.2.1__tar.gz → 1.2.2__tar.gz

{podstack-1.2.1 → podstack-1.2.2}/PKG-INFO

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

{podstack-1.2.1 → podstack-1.2.2}/podstack/__init__.py

@@ -54,7 +54,7 @@ Decorators:
         ...
 """
 
-__version__ = "1.2.1"
+__version__ = "1.2.2"
 
 from .client import Client
 from .notebook import Notebook, NotebookStatus

{podstack-1.2.1 → podstack-1.2.2}/podstack/gpu_runner.py

@@ -38,10 +38,11 @@ def is_jupyter() -> bool:
 class OutputStreamer:
     """Handles real-time output streaming from GPU executions."""
 
-    def __init__(self, execution_id: str, api_url: str, headers: Dict[str, str], timeout: float = 30.0):
+    def __init__(self, execution_id: str, api_url: str, headers: Dict[str, str], timeout: float = 30.0, api_key: str = None):
         self.execution_id = execution_id
         self.api_url = api_url
         self.headers = headers
+        self.api_key = api_key
         self.timeout = timeout
         self._output_buffer: List[str] = []
         self._status = "running"
@@ -56,10 +57,14 @@ class OutputStreamer:
             Dict with event data (type, content, timestamp, etc.)
         """
         url = f"{self.api_url}/api/v1/executions/{self.execution_id}/stream"
+        # Pass token as query param for SSE — proxies often strip Authorization header on streaming connections
+        params = {}
+        if self.api_key:
+            params["token"] = self.api_key
 
         try:
             with httpx.Client(timeout=httpx.Timeout(self.timeout, read=None)) as client:
-                with client.stream("GET", url, headers=self.headers) as response:
+                with client.stream("GET", url, headers=self.headers, params=params) as response:
                     if response.status_code != 200:
                         raise RuntimeError(f"Failed to connect to stream: HTTP {response.status_code}")
 
@@ -249,7 +254,7 @@ class GPURunner:
         return {
             "Authorization": f"Bearer {self.api_key}",
             "Content-Type": "application/json",
-            "User-Agent": "podstack-python-sdk/1.0.3"
+            "User-Agent": "podstack-python-sdk/1.2.2"
         }
 
     def _build_annotation(
@@ -607,7 +612,8 @@ _stream_install(
             execution_id=execution_id,
             api_url=self.api_url,
             headers=self._get_headers(),
-            timeout=self.timeout
+            timeout=self.timeout,
+            api_key=self.api_key
         )
         yield from streamer.stream(show_output=show_output)
 
@@ -750,6 +756,12 @@ _stream_install(
                         final_event = event
                         break
 
+        except RuntimeError as e:
+            if "HTTP 401" in str(e):
+                # Auth failed on stream — fall back to polling
+                print(f"\n[Podstack] Streaming auth failed, falling back to polling...")
+                return self._run_with_polling(execution_id, gpu, count, timeout, 2.0, max_retries, 300, cancel_on_timeout)
+            raise
         except (ConnectionError, httpx.ConnectError) as e:
             # Try to recover and get the result
             logger.warning(f"Stream connection lost: {e}")

{podstack-1.2.1 → podstack-1.2.2}/podstack.egg-info/PKG-INFO

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

{podstack-1.2.1 → podstack-1.2.2}/pyproject.toml

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