cosmos-agent-sdk 0.0.1a0__tar.gz

cosmos_agent_sdk-0.0.1a0/PKG-INFO

@@ -0,0 +1,101 @@
+Metadata-Version: 2.4
+Name: cosmos-agent-sdk
+Version: 0.0.1a0
+Summary: Python client for the Auggie v2 SDK / RPC protocol
+Author-email: Augment Code <support@augmentcode.com>
+License-Expression: MIT
+Project-URL: Homepage, https://augmentcode.com
+Project-URL: Repository, https://github.com/augmentcode/cosmos-agent-sdk
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Provides-Extra: ws
+Requires-Dist: websocket-client>=1.7.0; extra == "ws"
+Provides-Extra: mongo
+Requires-Dist: pymongo>=4.6.0; extra == "mongo"
+Provides-Extra: cloud
+Requires-Dist: httpx>=0.27.0; extra == "cloud"
+Provides-Extra: dev
+Requires-Dist: ruff>=0.6.0; extra == "dev"
+Requires-Dist: mypy>=1.0.0; extra == "dev"
+Requires-Dist: pytest>=8.0.0; extra == "dev"
+Requires-Dist: websocket-client>=1.7.0; extra == "dev"
+Requires-Dist: pymongo>=4.6.0; extra == "dev"
+Requires-Dist: mongomock>=4.1.0; extra == "dev"
+Requires-Dist: httpx>=0.27.0; extra == "dev"
+
+# cosmos-agent-sdk — Python client
+
+RPC-first Python client for the Auggie v2 SDK / RPC protocol. It implements the
+shared spec under [`../../spec`](../../spec) and passes the shared
+[`../../conformance`](../../conformance) fixtures, in lockstep with the
+TypeScript client.
+
+## Layers
+
+- **Transport** (`cosmos_agent_sdk.transport`) — JSONL framing, request/response
+  correlation by `id`, protocol-version handshake, and message routing. Spawns
+  `auggie-v2 --mode rpc` (`SubprocessTransport`); a `FakeTransport` replays
+  canned wire messages through the same code paths for tests/conformance. Single
+  session per process (multi-session = a client-side process pool).
+- **Protocol** (`cosmos_agent_sdk.protocol`) — typed command / response / event /
+  extension-UI models mirroring the four JSON Schemas. Deep external structures
+  are forwarded verbatim as permissive objects (see the spec's "Permissiveness
+  policy").
+- **Facade** (`cosmos_agent_sdk.client.AgentClient`) — the ergonomic API:
+  `prompt`, `wait_for_idle` (idle = the `agent_end` event), the full typed
+  command surface, and bidirectional extension-UI round-trips (reply-expecting
+  methods are answered; fire-and-forget methods are not).
+
+```python
+from cosmos_agent_sdk import AgentClient, SubprocessTransport
+
+transport = SubprocessTransport("auggie-v2")
+transport.start()
+with AgentClient(transport) as client:
+    client.prompt("List the files in this repo")
+    client.wait_for_idle()
+    print(client.get_last_assistant_text()["data"]["text"])
+```
+
+## Toolchain
+
+- [uv](https://docs.astral.sh/uv/) for environment management and building.
+- [ruff](https://docs.astral.sh/ruff/) for lint / format.
+- [mypy](https://mypy-lang.org/) for type-checking.
+- [pytest](https://docs.pytest.org/) for the unit suite.
+
+## Commands
+
+```bash
+uv sync --extra dev          # create a venv and install dev dependencies
+uv run ruff check .          # lint
+uv run ruff format .         # format
+uv run mypy cosmos_agent_sdk # type-check
+uv run pytest                # unit tests
+uv build                     # build the sdist + wheel
+```
+
+## Conformance
+
+The cross-client conformance runner is JavaScript. The Python client binds to it
+through a thin JS shim, [`conformance-adapter.mjs`](./conformance-adapter.mjs),
+that shells out to the real Python client
+(`python -m cosmos_agent_sdk.conformance`) driven over an in-memory fake
+transport replaying the fixture wire messages through the real
+parsing / correlation / extension-UI code paths (v0 does not spawn a real
+`auggie-v2` binary).
+
+```bash
+# from clients/python/ — requires the conformance deps installed once:
+( cd ../../conformance && npm install )
+uv sync --extra dev
+make conformance             # or: node conformance-adapter.mjs
+```
+
+Exits non-zero if any of the 40 shared vectors fails.

cosmos_agent_sdk-0.0.1a0/README.md

@@ -0,0 +1,70 @@
+# cosmos-agent-sdk — Python client
+
+RPC-first Python client for the Auggie v2 SDK / RPC protocol. It implements the
+shared spec under [`../../spec`](../../spec) and passes the shared
+[`../../conformance`](../../conformance) fixtures, in lockstep with the
+TypeScript client.
+
+## Layers
+
+- **Transport** (`cosmos_agent_sdk.transport`) — JSONL framing, request/response
+  correlation by `id`, protocol-version handshake, and message routing. Spawns
+  `auggie-v2 --mode rpc` (`SubprocessTransport`); a `FakeTransport` replays
+  canned wire messages through the same code paths for tests/conformance. Single
+  session per process (multi-session = a client-side process pool).
+- **Protocol** (`cosmos_agent_sdk.protocol`) — typed command / response / event /
+  extension-UI models mirroring the four JSON Schemas. Deep external structures
+  are forwarded verbatim as permissive objects (see the spec's "Permissiveness
+  policy").
+- **Facade** (`cosmos_agent_sdk.client.AgentClient`) — the ergonomic API:
+  `prompt`, `wait_for_idle` (idle = the `agent_end` event), the full typed
+  command surface, and bidirectional extension-UI round-trips (reply-expecting
+  methods are answered; fire-and-forget methods are not).
+
+```python
+from cosmos_agent_sdk import AgentClient, SubprocessTransport
+
+transport = SubprocessTransport("auggie-v2")
+transport.start()
+with AgentClient(transport) as client:
+    client.prompt("List the files in this repo")
+    client.wait_for_idle()
+    print(client.get_last_assistant_text()["data"]["text"])
+```
+
+## Toolchain
+
+- [uv](https://docs.astral.sh/uv/) for environment management and building.
+- [ruff](https://docs.astral.sh/ruff/) for lint / format.
+- [mypy](https://mypy-lang.org/) for type-checking.
+- [pytest](https://docs.pytest.org/) for the unit suite.
+
+## Commands
+
+```bash
+uv sync --extra dev          # create a venv and install dev dependencies
+uv run ruff check .          # lint
+uv run ruff format .         # format
+uv run mypy cosmos_agent_sdk # type-check
+uv run pytest                # unit tests
+uv build                     # build the sdist + wheel
+```
+
+## Conformance
+
+The cross-client conformance runner is JavaScript. The Python client binds to it
+through a thin JS shim, [`conformance-adapter.mjs`](./conformance-adapter.mjs),
+that shells out to the real Python client
+(`python -m cosmos_agent_sdk.conformance`) driven over an in-memory fake
+transport replaying the fixture wire messages through the real
+parsing / correlation / extension-UI code paths (v0 does not spawn a real
+`auggie-v2` binary).
+
+```bash
+# from clients/python/ — requires the conformance deps installed once:
+( cd ../../conformance && npm install )
+uv sync --extra dev
+make conformance             # or: node conformance-adapter.mjs
+```
+
+Exits non-zero if any of the 40 shared vectors fails.

cosmos_agent_sdk-0.0.1a0/cosmos_agent_sdk/__init__.py

@@ -0,0 +1,197 @@
+"""Python client for the Auggie v2 SDK / RPC protocol.
+
+RPC-first client mirroring the shared spec under ``spec/``: a JSONL transport
+that spawns ``auggie-v2 --mode rpc`` (single session per process), a typed
+command / response / event / extension-UI layer, and an ergonomic
+:class:`AgentClient` facade with ``prompt`` / ``wait_for_idle`` and bidirectional
+extension-UI round-trips. Conformance parity with the TypeScript client is
+proved by the shared fixtures (see ``clients/python/conformance-adapter.mjs``).
+"""
+
+from __future__ import annotations
+
+from .acp import (
+    AcpClient,
+    AcpCost,
+    AcpEventListener,
+    AcpNotInitializedError,
+    AcpPermissionHandler,
+    AcpRpcError,
+    AcpWarning,
+    AcpWarningListener,
+    MappedSessionUpdate,
+    UsageUpdateEvent,
+    map_session_update,
+)
+from .client import (
+    AgentClient,
+    CommandError,
+    ExtensionUIResponder,
+    RpcError,
+    make_confirm_all_responder,
+)
+from .cloud import (
+    AugmentCredentials,
+    CloudAgentCapability,
+    CloudAgentClient,
+    CloudHttpError,
+    CosmosTransport,
+    CredentialsNotFoundError,
+    HttpClient,
+    McpServerConfig,
+    UnselectableCapabilityError,
+    UrllibHttpClient,
+    capability_from,
+    default_http_client,
+    resolve_credentials,
+    resolve_mcp_servers,
+)
+from .redaction import (
+    DEFAULT_REDACTION_POLICY,
+    RedactionMode,
+    RedactionPolicy,
+    redact,
+)
+from .session_reader import (
+    CURRENT_SESSION_VERSION,
+    ParsedSession,
+    PerModelUsage,
+    SessionStats,
+    SessionTreeNode,
+    UsageAggregation,
+    aggregate_stats,
+    aggregate_usage,
+    build_tree,
+    deterministic_id_generator,
+    header,
+    latest_queue_state,
+    migrate_session_entries,
+    parse_session,
+    read_session,
+    session_entries,
+    watch_session,
+)
+from .session_sink import (
+    EntryUsage,
+    RollupModelSpend,
+    SessionEntryDocument,
+    SessionRollupDocument,
+    SessionSink,
+    session_id_of,
+    split_header_and_entries,
+    to_session_documents,
+    to_session_rollup,
+)
+from .protocol import (
+    ASYNC_COMMANDS,
+    COMMAND_TYPES,
+    EVENT_TYPES,
+    FIRE_AND_FORGET_METHODS,
+    IDLE_EVENT,
+    NO_DATA_COMMANDS,
+    PROTOCOL_VERSION,
+    REPLY_EXPECTING_METHODS,
+    classify_stdout,
+    is_idle_event,
+    is_reply_expecting,
+    is_response_success,
+    make_extension_ui_response,
+)
+from .transport import (
+    FakeTransport,
+    ProtocolVersionMismatch,
+    SubprocessTransport,
+    Transport,
+    WebSocketConnection,
+    WebSocketConnector,
+    WebSocketTransport,
+    assert_protocol_version,
+    decode_line,
+    encode_line,
+)
+
+__all__ = [
+    "ASYNC_COMMANDS",
+    "CURRENT_SESSION_VERSION",
+    "DEFAULT_REDACTION_POLICY",
+    "RedactionMode",
+    "RedactionPolicy",
+    "redact",
+    "ParsedSession",
+    "PerModelUsage",
+    "SessionStats",
+    "SessionTreeNode",
+    "UsageAggregation",
+    "aggregate_stats",
+    "aggregate_usage",
+    "build_tree",
+    "deterministic_id_generator",
+    "header",
+    "latest_queue_state",
+    "migrate_session_entries",
+    "parse_session",
+    "read_session",
+    "session_entries",
+    "watch_session",
+    "EntryUsage",
+    "RollupModelSpend",
+    "SessionEntryDocument",
+    "SessionRollupDocument",
+    "SessionSink",
+    "session_id_of",
+    "split_header_and_entries",
+    "to_session_documents",
+    "to_session_rollup",
+    "COMMAND_TYPES",
+    "EVENT_TYPES",
+    "FIRE_AND_FORGET_METHODS",
+    "IDLE_EVENT",
+    "NO_DATA_COMMANDS",
+    "PROTOCOL_VERSION",
+    "REPLY_EXPECTING_METHODS",
+    "AcpClient",
+    "AcpCost",
+    "AcpEventListener",
+    "AcpNotInitializedError",
+    "AcpPermissionHandler",
+    "AcpRpcError",
+    "AcpWarning",
+    "AcpWarningListener",
+    "AgentClient",
+    "AugmentCredentials",
+    "CloudAgentCapability",
+    "CloudAgentClient",
+    "CloudHttpError",
+    "CommandError",
+    "CosmosTransport",
+    "CredentialsNotFoundError",
+    "HttpClient",
+    "McpServerConfig",
+    "UnselectableCapabilityError",
+    "UrllibHttpClient",
+    "capability_from",
+    "default_http_client",
+    "resolve_credentials",
+    "resolve_mcp_servers",
+    "MappedSessionUpdate",
+    "UsageUpdateEvent",
+    "ExtensionUIResponder",
+    "FakeTransport",
+    "ProtocolVersionMismatch",
+    "RpcError",
+    "SubprocessTransport",
+    "Transport",
+    "WebSocketConnection",
+    "WebSocketConnector",
+    "WebSocketTransport",
+    "assert_protocol_version",
+    "classify_stdout",
+    "decode_line",
+    "encode_line",
+    "is_idle_event",
+    "is_reply_expecting",
+    "is_response_success",
+    "make_confirm_all_responder",
+    "map_session_update",
+    "make_extension_ui_response",
+]

cosmos_agent_sdk-0.0.1a0/cosmos_agent_sdk/acp/__init__.py

@@ -0,0 +1,43 @@
+"""ACP-transport adapter for the Auggie v2 SDK (CSS-1414).
+
+An additive sibling of the RPC client: :class:`AcpClient` drives
+``auggie-v2 --acp`` as an Agent Client Protocol *client* over the child's stdio
+and re-exposes the SAME ``on_event`` / ``on_warning`` / ``prompt`` /
+``prompt_and_wait`` / ``wait_for_idle`` / ``close`` ergonomics as
+:class:`~cosmos_agent_sdk.AgentClient`, so consumer code looks the same on either
+transport. ``on_event`` also delivers the SDK-owned ``UsageUpdateEvent`` for the
+ACP ``usage_update`` notification (CSS-2258); see ``events.py``.
+
+Zero base dependencies: the minimal ACP client (newline-delimited JSON-RPC 2.0
+over stdio) is hand-rolled here and is fully synchronous, matching the RPC
+client's threads + :class:`threading.Condition` model.
+"""
+
+from __future__ import annotations
+
+from .client import (
+    PROTOCOL_VERSION,
+    AcpClient,
+    AcpEventListener,
+    AcpNotInitializedError,
+    AcpPermissionHandler,
+    AcpRpcError,
+    AcpWarning,
+    AcpWarningListener,
+)
+from .events import AcpCost, MappedSessionUpdate, UsageUpdateEvent, map_session_update
+
+__all__ = [
+    "PROTOCOL_VERSION",
+    "AcpClient",
+    "AcpCost",
+    "AcpEventListener",
+    "AcpNotInitializedError",
+    "AcpPermissionHandler",
+    "AcpRpcError",
+    "AcpWarning",
+    "AcpWarningListener",
+    "MappedSessionUpdate",
+    "UsageUpdateEvent",
+    "map_session_update",
+]