blade-agent-kit 0.4.1__py3-none-any.whl

blade_agent_kit/__init__.py

@@ -0,0 +1,3 @@
+from blade_agent_kit.client import BladeAgentClient
+
+__all__ = ["BladeAgentClient"]

blade_agent_kit/client.py

@@ -0,0 +1,138 @@
+"""Minimal REST client for Blade Agent (v1: no Socket.IO)."""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+
+class BladeAgentClient:
+    def __init__(
+        self,
+        base_url: str,
+        *,
+        token: str | None = None,
+        timeout: float = 60.0,
+    ) -> None:
+        self._base = base_url.rstrip("/")
+        self._token = token
+        self._timeout = timeout
+
+    def _headers(self) -> dict[str, str]:
+        h: dict[str, str] = {"Accept": "application/json"}
+        if self._token:
+            h["Authorization"] = f"Bearer {self._token}"
+        return h
+
+    def _client(self) -> httpx.Client:
+        return httpx.Client(
+            base_url=self._base,
+            timeout=self._timeout,
+            headers=self._headers(),
+        )
+
+    def health(self) -> dict[str, Any]:
+        with self._client() as c:
+            r = c.get("/api/health")
+            r.raise_for_status()
+            return r.json()
+
+    def list_sessions(
+        self,
+        *,
+        limit: int = 100,
+        offset: int = 0,
+        template_id_prefix: str | None = None,
+        q: str | None = None,
+    ) -> list[dict[str, Any]]:
+        """Return session items from paginated GET /api/sessions."""
+        params: dict[str, Any] = {"limit": limit, "offset": offset}
+        if template_id_prefix:
+            params["template_id_prefix"] = template_id_prefix
+        if q:
+            params["q"] = q.strip()
+        with self._client() as c:
+            r = c.get("/api/sessions", params=params)
+            r.raise_for_status()
+            payload = r.json()
+        if isinstance(payload, list):
+            return payload
+        if isinstance(payload, dict):
+            items = payload.get("items")
+            if isinstance(items, list):
+                return items
+        raise TypeError(f"unexpected /api/sessions response: {type(payload).__name__}")
+
+    def list_sessions_page(
+        self,
+        *,
+        limit: int = 20,
+        offset: int = 0,
+        template_id_prefix: str | None = None,
+        q: str | None = None,
+    ) -> dict[str, Any]:
+        """Return full paginated payload: items, total, limit, offset."""
+        params: dict[str, Any] = {"limit": limit, "offset": offset}
+        if template_id_prefix:
+            params["template_id_prefix"] = template_id_prefix
+        if q:
+            params["q"] = q.strip()
+        with self._client() as c:
+            r = c.get("/api/sessions", params=params)
+            r.raise_for_status()
+            payload = r.json()
+        if not isinstance(payload, dict):
+            raise TypeError(f"unexpected /api/sessions response: {type(payload).__name__}")
+        return payload
+
+    def get_session(self, session_id: str) -> dict[str, Any]:
+        with self._client() as c:
+            r = c.get(f"/api/sessions/{session_id}")
+            r.raise_for_status()
+            return r.json()
+
+    def get_history(self, session_id: str) -> dict[str, Any]:
+        with self._client() as c:
+            r = c.get(f"/api/sessions/{session_id}/history")
+            r.raise_for_status()
+            return r.json()
+
+    def create_session(self, intent: str = "") -> dict[str, Any]:
+        with self._client() as c:
+            r = c.post("/api/sessions", json={"intent": intent})
+            r.raise_for_status()
+            return r.json()
+
+    def delete_session(self, session_id: str) -> dict[str, Any]:
+        with self._client() as c:
+            r = c.delete(f"/api/sessions/{session_id}")
+            r.raise_for_status()
+            return r.json()
+
+    def list_dir(self, session_id: str, dir_path: str = ".") -> list[dict[str, Any]]:
+        with self._client() as c:
+            r = c.get(f"/api/sessions/{session_id}/ls/{_encode_ls_path(dir_path)}")
+            r.raise_for_status()
+            return r.json()
+
+    def get_checkpoints(self, session_id: str) -> dict[str, Any]:
+        with self._client() as c:
+            r = c.get(f"/api/sessions/{session_id}/checkpoints")
+            r.raise_for_status()
+            return r.json()
+
+
+def _encode_ls_path(dir_path: str) -> str:
+    """Encode path for GET /api/sessions/{id}/ls/{dir_path}.
+
+    Root ``.`` must be sent as ``%2E``: ``/ls/.`` is normalized to ``/ls`` by HTTP
+    clients and does not match the server route (404). Aligns with browser
+    ``encodeURIComponent`` for nested paths (``/`` → ``%2F``).
+    """
+    from urllib.parse import quote
+
+    normalized = (dir_path or ".").strip()
+    if normalized == ".":
+        return "%2E"
+    return quote(normalized, safe="")

blade_agent_kit-0.4.1.dist-info/METADATA

@@ -0,0 +1,6 @@
+Metadata-Version: 2.4
+Name: blade-agent-kit
+Version: 0.4.1
+Summary: REST client for Blade Agent (AgentKit Python, v1 without Socket.IO)
+Requires-Python: <3.13,>=3.12
+Requires-Dist: httpx>=0.28.1

blade_agent_kit-0.4.1.dist-info/RECORD

@@ -0,0 +1,5 @@
+blade_agent_kit/__init__.py,sha256=IdRFOMy_fUi_ZMRe3vZo5YVUgJVTG0bUG3_Ev5HnrkY,84
+blade_agent_kit/client.py,sha256=NeY6l8-qIWTYRjNZH8Q8l42T-DFCAC5-diPIhXCOnnM,4577
+blade_agent_kit-0.4.1.dist-info/METADATA,sha256=PtFlmaFud_iFcZxTOBVOx4vVlCC418diPqQozI6MqnI,195
+blade_agent_kit-0.4.1.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+blade_agent_kit-0.4.1.dist-info/RECORD,,

blade_agent_kit-0.4.1.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.29.0
+Root-Is-Purelib: true
+Tag: py3-none-any