pi-sidecar-client 0.1.0.dev2__tar.gz → 0.1.0.dev4__tar.gz

{pi_sidecar_client-0.1.0.dev2 → pi_sidecar_client-0.1.0.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pi-sidecar-client
-Version: 0.1.0.dev2
+Version: 0.1.0.dev4
 Summary: Python client for the Pi SDK sidecar service
 License-Expression: Apache-2.0
 Project-URL: Repository, https://github.com/myk-org/pi-sidecar

{pi_sidecar_client-0.1.0.dev2 → pi_sidecar_client-0.1.0.dev4}/pi_sidecar_client/__init__.py

@@ -1,6 +1,7 @@
 import asyncio
 import inspect
 import os
+import tempfile
 from collections.abc import Callable
 from dataclasses import dataclass
 from typing import Any
@@ -11,6 +12,7 @@ from simple_logger.logger import get_logger
 logger = get_logger(name=__name__, level=os.environ.get("LOG_LEVEL", "INFO"))
 
 SIDECAR_URL = os.environ.get("SIDECAR_URL", "http://127.0.0.1:9100")
+DEFAULT_CWD = tempfile.gettempdir()
 
 __all__ = [
     "AIResult",
@@ -48,7 +50,8 @@ _usage_recorder: Callable | None = None
 def set_usage_recorder(callback: Callable) -> None:
     """Register a callback for recording AI token usage.
 
-    The callback signature:
+    The callback may be sync or async:
+        def recorder(*, request_id, result, call_type, prompt_chars, ai_provider, ai_model)
         async def recorder(*, request_id, result, call_type, prompt_chars, ai_provider, ai_model)
     """
     global _usage_recorder
@@ -115,6 +118,7 @@ class SidecarClient:
     def __init__(self, base_url: str = SIDECAR_URL):
         self._base_url = base_url.rstrip("/")
         self._client = httpx.AsyncClient(base_url=self._base_url, timeout=600.0)
+        self._closed = False
 
     async def health(self) -> dict:
         """Check sidecar health."""
@@ -140,7 +144,7 @@ class SidecarClient:
         provider: str,
         model: str,
         system_prompt: str,
-        cwd: str = "/tmp",
+        cwd: str = DEFAULT_CWD,
         custom_tools: list | None = None,
     ) -> str:
         """Create a new AI session. Returns session_id."""
@@ -202,6 +206,7 @@ class SidecarClient:
     async def close(self) -> None:
         """Close the HTTP client."""
         await self._client.aclose()
+        self._closed = True
 
 
 # Singleton client
@@ -211,7 +216,7 @@ _client: SidecarClient | None = None
 def get_sidecar_client() -> SidecarClient:
     """Get the singleton sidecar client."""
     global _client
-    if _client is None:
+    if _client is None or _client._closed:
         _client = SidecarClient()
     return _client
 
@@ -250,7 +255,7 @@ async def call_ai(
                 provider=ai_provider,
                 model=ai_model,
                 system_prompt=system_prompt or "You are a helpful assistant.",
-                cwd=cwd or "/tmp",
+                cwd=cwd or DEFAULT_CWD,
                 custom_tools=custom_tools,
             )
             created_session = True
@@ -289,9 +294,10 @@ async def call_ai_once(
 ) -> AIResult:
     """Single-shot AI call with automatic session cleanup.
 
-    Creates a session, sends the prompt, and always deletes the session.
-    Use this for one-off calls (analysis, bug creation, etc.).
-    Use ``call_ai`` directly for multi-turn conversations (peer debate).
+    Creates a session, sends the prompt, and deletes the session.
+    Cleanup is best-effort — if deletion fails, result.session_id is
+    preserved so the caller can retry cleanup.
+    Use ``call_ai`` directly for multi-turn conversations.
     """
     result = await call_ai(
         prompt,
@@ -327,13 +333,20 @@ async def check_sidecar_available() -> tuple[bool, str]:
     """Check if the sidecar service is available and ready."""
     try:
         client = get_sidecar_client()
-        resp = await client._client.get("/health")
-        data = resp.json()
-        if resp.status_code == 200 and data.get("status") == "ok":
+        data = await client.health()
+        if data.get("status") == "ok":
             return True, "Sidecar is ready"
-        if resp.status_code == 503 and data.get("status") == "starting":
+        if data.get("status") == "starting":
             return False, f"Sidecar starting: {data.get('message', 'model discovery in progress')}"
-        return False, f"Sidecar unhealthy (HTTP {resp.status_code}): {data}"
+        return False, f"Sidecar unhealthy: {data}"
+    except httpx.HTTPStatusError as e:
+        if e.response.status_code == 503:
+            try:
+                data = e.response.json()
+                return False, f"Sidecar starting: {data.get('message', 'model discovery in progress')}"
+            except ValueError:
+                pass
+        return False, f"Sidecar unhealthy (HTTP {e.response.status_code})"
     except Exception as e:
         return False, f"Sidecar unavailable: {e}"
 
@@ -343,6 +356,9 @@ async def run_parallel_with_limit(
     max_concurrency: int = 5,
 ) -> list:
     """Run async tasks in parallel with concurrency limit."""
+    if max_concurrency < 1:
+        raise ValueError("max_concurrency must be >= 1")
+
     semaphore = asyncio.Semaphore(max_concurrency)
 
     async def limited(coro):

{pi_sidecar_client-0.1.0.dev2 → pi_sidecar_client-0.1.0.dev4}/pi_sidecar_client.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pi-sidecar-client
-Version: 0.1.0.dev2
+Version: 0.1.0.dev4
 Summary: Python client for the Pi SDK sidecar service
 License-Expression: Apache-2.0
 Project-URL: Repository, https://github.com/myk-org/pi-sidecar

{pi_sidecar_client-0.1.0.dev2 → pi_sidecar_client-0.1.0.dev4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pi-sidecar-client"
-version = "0.1.0.dev2"
+version = "0.1.0.dev4"
 description = "Python client for the Pi SDK sidecar service"
 license = "Apache-2.0"
 requires-python = ">=3.10"