waken-sdk 0.1.0__tar.gz

waken_sdk-0.1.0/PKG-INFO

@@ -0,0 +1,46 @@
+Metadata-Version: 2.4
+Name: waken-sdk
+Version: 0.1.0
+Summary: Waken SDK: persistent AI agents hosted on the cloud, from Python
+License: MIT
+Project-URL: Homepage, https://waken.sh
+Project-URL: Documentation, https://waken.sh/docs/api
+Project-URL: Source, https://github.com/medhychabour/waken
+Keywords: ai,agents,waken,claude-code,codex,sdk
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+
+# waken-sdk
+
+Persistent AI agents hosted on the cloud, from Python. Installed as `waken-sdk`, imported as `waken`. A thin client over the [Waken HTTP API](https://waken.sh/docs/api), no dependencies, Python 3.9+.
+
+```bash
+pip install waken-sdk
+```
+
+```python
+import os
+from waken import Waken
+
+waken = Waken(api_key=os.environ["WAKEN_KEY"])
+
+agent = waken.agents.create(
+    name="bookkeeper",
+    model={"provider": "anthropic", "api_key": os.environ["ANTHROPIC_API_KEY"]},
+    spend_cap_usd=10,
+)
+
+reply = waken.agents.message(agent["id"], "Sort my March receipts by vendor.")
+print(reply["text"])
+
+cp = waken.checkpoints.take(agent["id"])          # a point you can rewind to
+waken.checkpoints.restore(agent["id"], cp["id"])  # rewind: files, memory, conversation
+```
+
+Everything the API does is here: `agents` (create, list, get, delete, message, sleep, wake, restart, usage, resize, set_model, set_harness, logs, exec), `checkpoints` (take, list, restore), `files` (list, read, read_text, write, mkdir, move, delete, upload_tar), `env` (list, set), `triggers` (schedules and the agent's incoming webhook), `webhooks` (events on all your agents), `harnesses()`. Any other route: `waken.request(method, path, body)`.
+
+Errors raise `WakenError` with `.status` and the API message.
+
+Keys are created in the dashboard, Developers. An agent runs on your own model key, passed at creation and never stored by Waken.

waken_sdk-0.1.0/README.md

@@ -0,0 +1,32 @@
+# waken-sdk
+
+Persistent AI agents hosted on the cloud, from Python. Installed as `waken-sdk`, imported as `waken`. A thin client over the [Waken HTTP API](https://waken.sh/docs/api), no dependencies, Python 3.9+.
+
+```bash
+pip install waken-sdk
+```
+
+```python
+import os
+from waken import Waken
+
+waken = Waken(api_key=os.environ["WAKEN_KEY"])
+
+agent = waken.agents.create(
+    name="bookkeeper",
+    model={"provider": "anthropic", "api_key": os.environ["ANTHROPIC_API_KEY"]},
+    spend_cap_usd=10,
+)
+
+reply = waken.agents.message(agent["id"], "Sort my March receipts by vendor.")
+print(reply["text"])
+
+cp = waken.checkpoints.take(agent["id"])          # a point you can rewind to
+waken.checkpoints.restore(agent["id"], cp["id"])  # rewind: files, memory, conversation
+```
+
+Everything the API does is here: `agents` (create, list, get, delete, message, sleep, wake, restart, usage, resize, set_model, set_harness, logs, exec), `checkpoints` (take, list, restore), `files` (list, read, read_text, write, mkdir, move, delete, upload_tar), `env` (list, set), `triggers` (schedules and the agent's incoming webhook), `webhooks` (events on all your agents), `harnesses()`. Any other route: `waken.request(method, path, body)`.
+
+Errors raise `WakenError` with `.status` and the API message.
+
+Keys are created in the dashboard, Developers. An agent runs on your own model key, passed at creation and never stored by Waken.

waken_sdk-0.1.0/pyproject.toml

@@ -0,0 +1,21 @@
+[build-system]
+requires = ["setuptools>=68"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "waken-sdk"
+version = "0.1.0"
+description = "Waken SDK: persistent AI agents hosted on the cloud, from Python"
+readme = "README.md"
+license = { text = "MIT" }
+requires-python = ">=3.9"
+keywords = ["ai", "agents", "waken", "claude-code", "codex", "sdk"]
+classifiers = ["Programming Language :: Python :: 3", "License :: OSI Approved :: MIT License"]
+
+[project.urls]
+Homepage = "https://waken.sh"
+Documentation = "https://waken.sh/docs/api"
+Source = "https://github.com/medhychabour/waken"
+
+[tool.setuptools]
+packages = ["waken"]

waken_sdk-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

waken_sdk-0.1.0/waken/__init__.py

@@ -0,0 +1,227 @@
+"""Waken SDK for Python. A thin, dependency-free client over the HTTP API (https://waken.sh/docs/api).
+
+    from waken import Waken
+    waken = Waken(api_key=os.environ["WAKEN_KEY"])
+    agent = waken.agents.create(name="bookkeeper", model={"provider": "anthropic", "api_key": "sk-ant-..."})
+    print(waken.agents.message(agent["id"], "Sort my March receipts by vendor.")["text"])
+"""
+from __future__ import annotations
+
+import json
+import urllib.error
+import urllib.parse
+import urllib.request
+from typing import Any, Optional
+
+__all__ = ["Waken", "WakenError"]
+__version__ = "0.1.0"
+
+
+class WakenError(Exception):
+    def __init__(self, status: int, message: str, body: Any = None):
+        super().__init__(message)
+        self.status, self.body = status, body
+
+
+class Waken:
+    def __init__(self, api_key: str, base_url: str = "https://api.waken.sh", timeout: float = 60.0):
+        if not api_key:
+            raise ValueError("api_key is required: create one in the dashboard, Developers")
+        self._key, self._base, self._timeout = api_key, base_url.rstrip("/"), timeout
+        self.agents = _Agents(self)
+        self.checkpoints = _Checkpoints(self)
+        self.files = _Files(self)
+        self.env = _Env(self)
+        self.triggers = _Triggers(self)
+        self.webhooks = _Webhooks(self)
+
+    # Low level: any route. body may be a dict (sent as JSON), str or bytes (sent as is), or None.
+    def request(self, method: str, path: str, body: Any = None, timeout: Optional[float] = None, raw: bool = False, content_type: Optional[str] = None) -> Any:
+        headers = {"Authorization": f"Bearer {self._key}"}
+        data = None
+        if body is not None:
+            if isinstance(body, (str, bytes)):
+                data = body.encode() if isinstance(body, str) else body
+                headers["Content-Type"] = content_type or "application/octet-stream"
+            else:
+                data = json.dumps(body).encode()
+                headers["Content-Type"] = "application/json"
+        req = urllib.request.Request(self._base + path, data=data, method=method, headers=headers)
+        try:
+            with urllib.request.urlopen(req, timeout=timeout or self._timeout) as r:
+                payload = r.read()
+        except urllib.error.HTTPError as e:
+            text = e.read().decode(errors="replace")
+            try:
+                j = json.loads(text)
+                msg = j.get("error") if isinstance(j, dict) else None
+            except ValueError:
+                j, msg = text, None
+            raise WakenError(e.code, msg or f"{method} {path}: HTTP {e.code}", j) from None
+        if raw:
+            return payload
+        if not payload:
+            return None
+        try:
+            return json.loads(payload)
+        except ValueError:
+            return {"raw": payload.decode(errors="replace")}
+
+    def harnesses(self) -> list:
+        return self.request("GET", "/harnesses")
+
+
+class _Agents:
+    def __init__(self, c: Waken):
+        self._c = c
+
+    def create(self, **agent: Any) -> dict:
+        """name, harness, model={provider, name, api_key}, env, spend_cap_usd, idle_s, external_id, mem_mib, disk_gib,
+        desktop, image, command, port, registry_auth, repo, repo_ref, repo_token."""
+        return self._c.request("POST", "/agents", agent, timeout=180)
+
+    def list(self, external_id: Optional[str] = None) -> list:
+        q = f"?external_id={urllib.parse.quote(external_id)}" if external_id else ""
+        return self._c.request("GET", "/agents" + q)
+
+    def get(self, id: str) -> dict:
+        return self._c.request("GET", f"/agents/{id}")
+
+    def delete(self, id: str) -> dict:
+        return self._c.request("DELETE", f"/agents/{id}")
+
+    def message(self, id: str, text: str, timeout: float = 600) -> dict:
+        """Talk to the agent. Wakes it if needed; the call stays open until it answers."""
+        return self._c.request("POST", f"/agents/{id}/message", {"text": text}, timeout=timeout)
+
+    def sleep(self, id: str) -> dict:
+        return self._c.request("POST", f"/agents/{id}/sleep", timeout=120)
+
+    def wake(self, id: str) -> dict:
+        return self._c.request("POST", f"/agents/{id}/wake", timeout=120)
+
+    def restart(self, id: str) -> dict:
+        return self._c.request("POST", f"/agents/{id}/restart")
+
+    def usage(self, id: str) -> dict:
+        return self._c.request("GET", f"/agents/{id}/usage")
+
+    def resize(self, id: str, mem_mib: Optional[int] = None, disk_gib: Optional[int] = None) -> dict:
+        body = {k: v for k, v in (("mem_mib", mem_mib), ("disk_gib", disk_gib)) if v}
+        return self._c.request("PUT", f"/agents/{id}/size", body, timeout=180)
+
+    def set_model(self, id: str, model: dict) -> dict:
+        return self._c.request("PUT", f"/agents/{id}/model", {"model": model})
+
+    def set_harness(self, id: str, harness: str, model: dict) -> dict:
+        return self._c.request("PUT", f"/agents/{id}/harness", {"harness": harness, "model": model}, timeout=180)
+
+    def logs(self, id: str, lines: int = 100) -> dict:
+        return self._c.request("GET", f"/agents/{id}/logs?lines={lines}")
+
+    def exec(self, id: str, command: str, timeout_s: int = 60) -> dict:
+        return self._c.request("POST", f"/agents/{id}/exec", {"command": command, "timeout_s": timeout_s}, timeout=timeout_s + 30)
+
+
+class _Checkpoints:
+    def __init__(self, c: Waken):
+        self._c = c
+
+    def take(self, id: str) -> dict:
+        return self._c.request("POST", f"/agents/{id}/snapshot", timeout=180)
+
+    def list(self, id: str) -> list:
+        return self._c.request("GET", f"/agents/{id}/snapshots")
+
+    def restore(self, id: str, snapshot_id: Optional[str] = None) -> dict:
+        """Rewind to a checkpoint, or to the latest one when snapshot_id is omitted."""
+        return self._c.request("POST", f"/agents/{id}/restore", {"snapshot_id": snapshot_id} if snapshot_id else {}, timeout=180)
+
+
+class _Files:
+    """Paths are relative to the agent's working folder."""
+
+    def __init__(self, c: Waken):
+        self._c = c
+
+    def list(self, id: str, path: str = "") -> Any:
+        return self._c.request("GET", f"/agents/{id}/files/list?path={urllib.parse.quote(path)}")
+
+    def read(self, id: str, path: str) -> bytes:
+        return self._c.request("GET", f"/agents/{id}/files/download?path={urllib.parse.quote(path)}", raw=True)
+
+    def read_text(self, id: str, path: str) -> str:
+        return self.read(id, path).decode()
+
+    def write(self, id: str, path: str, content: Any) -> Any:
+        return self._c.request("PUT", f"/agents/{id}/files/write?path={urllib.parse.quote(path)}", content)
+
+    def mkdir(self, id: str, path: str) -> Any:
+        return self._c.request("POST", f"/agents/{id}/files/mkdir", {"path": path})
+
+    def move(self, id: str, src: str, dst: str) -> Any:
+        return self._c.request("POST", f"/agents/{id}/files/move", {"from": src, "to": dst})
+
+    def delete(self, id: str, path: str) -> Any:
+        return self._c.request("DELETE", f"/agents/{id}/files/delete?path={urllib.parse.quote(path)}")
+
+    def upload_tar(self, id: str, tar_gz: bytes) -> dict:
+        """Upload a whole folder as a gzipped tarball."""
+        return self._c.request("POST", f"/agents/{id}/files", tar_gz, timeout=300, content_type="application/gzip")
+
+
+class _Env:
+    """Environment variables of an agent. Values are never returned."""
+
+    def __init__(self, c: Waken):
+        self._c = c
+
+    def list(self, id: str) -> dict:
+        return self._c.request("GET", f"/agents/{id}/env")
+
+    def set(self, id: str, env: dict, unset: Optional[list] = None, reload: bool = False) -> Any:
+        return self._c.request("PUT", f"/agents/{id}/env", {"env": env, "unset": unset or [], "reload": reload})
+
+
+class _Triggers:
+    """Schedules, and the incoming webhook of an agent."""
+
+    def __init__(self, c: Waken):
+        self._c = c
+
+    def list(self, id: str) -> list:
+        return self._c.request("GET", f"/agents/{id}/triggers")
+
+    def create(self, id: str, cron: str, message: str) -> dict:
+        return self._c.request("POST", f"/agents/{id}/triggers", {"cron": cron, "message": message})
+
+    def delete(self, id: str, tid: str) -> Any:
+        return self._c.request("DELETE", f"/agents/{id}/triggers/{tid}")
+
+    def run(self, id: str, tid: str) -> Any:
+        return self._c.request("POST", f"/agents/{id}/triggers/{tid}/run", timeout=600)
+
+    def webhook(self, id: str) -> dict:
+        return self._c.request("GET", f"/agents/{id}/webhook")
+
+    def rotate_webhook(self, id: str) -> dict:
+        return self._c.request("POST", f"/agents/{id}/webhook/rotate")
+
+
+class _Webhooks:
+    """Outgoing webhooks: events on all your agents."""
+
+    def __init__(self, c: Waken):
+        self._c = c
+
+    def list(self) -> list:
+        return self._c.request("GET", "/webhooks")
+
+    def create(self, url: str, events: Optional[list] = None) -> dict:
+        body: dict = {"url": url}
+        if events:
+            body["events"] = events
+        return self._c.request("POST", "/webhooks", body)
+
+    def delete(self, wid: str) -> Any:
+        return self._c.request("DELETE", f"/webhooks/{wid}")

waken_sdk-0.1.0/waken_sdk.egg-info/PKG-INFO

@@ -0,0 +1,46 @@
+Metadata-Version: 2.4
+Name: waken-sdk
+Version: 0.1.0
+Summary: Waken SDK: persistent AI agents hosted on the cloud, from Python
+License: MIT
+Project-URL: Homepage, https://waken.sh
+Project-URL: Documentation, https://waken.sh/docs/api
+Project-URL: Source, https://github.com/medhychabour/waken
+Keywords: ai,agents,waken,claude-code,codex,sdk
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+
+# waken-sdk
+
+Persistent AI agents hosted on the cloud, from Python. Installed as `waken-sdk`, imported as `waken`. A thin client over the [Waken HTTP API](https://waken.sh/docs/api), no dependencies, Python 3.9+.
+
+```bash
+pip install waken-sdk
+```
+
+```python
+import os
+from waken import Waken
+
+waken = Waken(api_key=os.environ["WAKEN_KEY"])
+
+agent = waken.agents.create(
+    name="bookkeeper",
+    model={"provider": "anthropic", "api_key": os.environ["ANTHROPIC_API_KEY"]},
+    spend_cap_usd=10,
+)
+
+reply = waken.agents.message(agent["id"], "Sort my March receipts by vendor.")
+print(reply["text"])
+
+cp = waken.checkpoints.take(agent["id"])          # a point you can rewind to
+waken.checkpoints.restore(agent["id"], cp["id"])  # rewind: files, memory, conversation
+```
+
+Everything the API does is here: `agents` (create, list, get, delete, message, sleep, wake, restart, usage, resize, set_model, set_harness, logs, exec), `checkpoints` (take, list, restore), `files` (list, read, read_text, write, mkdir, move, delete, upload_tar), `env` (list, set), `triggers` (schedules and the agent's incoming webhook), `webhooks` (events on all your agents), `harnesses()`. Any other route: `waken.request(method, path, body)`.
+
+Errors raise `WakenError` with `.status` and the API message.
+
+Keys are created in the dashboard, Developers. An agent runs on your own model key, passed at creation and never stored by Waken.

waken_sdk-0.1.0/waken_sdk.egg-info/SOURCES.txt

@@ -0,0 +1,7 @@
+README.md
+pyproject.toml
+waken/__init__.py
+waken_sdk.egg-info/PKG-INFO
+waken_sdk.egg-info/SOURCES.txt
+waken_sdk.egg-info/dependency_links.txt
+waken_sdk.egg-info/top_level.txt

waken_sdk-0.1.0/waken_sdk.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

waken_sdk-0.1.0/waken_sdk.egg-info/top_level.txt

@@ -0,0 +1 @@
+waken