Skip to main content
Pre-release

This release is a pre-release and may not be stable for production use.

cosmos-agent-sdk — Python client

RPC-first Python client for the Auggie v2 SDK / RPC protocol. It implements the shared spec under ../../spec and passes the shared ../../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).
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 for environment management and building.
  • ruff for lint / format.
  • mypy for type-checking.
  • pytest for the unit suite.

Commands

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, 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).

# 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.

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

cosmos_agent_sdk-0.0.1a0.tar.gz (111.8 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

cosmos_agent_sdk-0.0.1a0-py3-none-any.whl (86.7 kB view details)

Uploaded Python 3

File details

Details for the file cosmos_agent_sdk-0.0.1a0.tar.gz.

File metadata

  • Download URL: cosmos_agent_sdk-0.0.1a0.tar.gz
  • Upload date:
  • Size: 111.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for cosmos_agent_sdk-0.0.1a0.tar.gz
Algorithm Hash digest
SHA256 7a7955a68021ec016b51b5b406eed77aed1c8f6d36d0b70575dc7b08d3c70ce7
MD5 faecd26b23121e412921f72dc244ed29
BLAKE2b-256 484623ebe87ef8f3c9974ebcf94d41fbacabeb50bdc8324791f7a36cd2ca0586

See more details on using hashes here.

Provenance

The following attestation bundles were made for cosmos_agent_sdk-0.0.1a0.tar.gz:

Publisher: release-pypi.yml on augmentcode/cosmos-agent-sdk

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file cosmos_agent_sdk-0.0.1a0-py3-none-any.whl.

File metadata

File hashes

Hashes for cosmos_agent_sdk-0.0.1a0-py3-none-any.whl
Algorithm Hash digest
SHA256 9219bbfc330ac76ade4c3a70073da512affc46644d84bbeef65d6b327f034817
MD5 8d019cf9301e9122c4059fb5552bab19
BLAKE2b-256 ff0ea418109340ed50e05d52a1cf6a400c62c1f496e7f36c2d79f597e64ae53d

See more details on using hashes here.

Provenance

The following attestation bundles were made for cosmos_agent_sdk-0.0.1a0-py3-none-any.whl:

Publisher: release-pypi.yml on augmentcode/cosmos-agent-sdk

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Release history Release notifications | RSS feed

This release

0.0.1a0 This release

2 files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page