molecule-sdk-python/README.md
Molecule AI SDK-Dev 74f6813b52
All checks were successful
Test / test (3.12) (pull_request) Successful in 1m59s
Test / test (3.13) (pull_request) Successful in 2m12s
Test / test (3.11) (pull_request) Successful in 2m16s
docs(sdk): add molecule_agent coverage to README — closes documentation gap
README title was misleading (# molecule_plugin) and the file only documented
molecule_plugin while completely omitting molecule_agent. Users landing on the
PyPI page had no indication that a remote-agent client was included.

New README structure:
- Covers both packages with a quick-reference table at the top
- Full molecule_agent quick-start (RemoteAgentClient, A2A delegation,
  two delivery paths, plugin install)
- molecule_plugin section kept with existing content
- Both-packages section for shared conventions

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-05-13 07:50:56 +00:00

224 lines
7.1 KiB
Markdown

# molecule-ai-sdk — Python SDK for Molecule AI
Two packages, one install:
| Package | Purpose |
|---------|---------|
| `molecule_agent` | **Remote-agent client.** Run an agent outside the platform's Docker network — it registers, pulls secrets, heartbeats, and participates in A2A delegation. |
| `molecule_plugin` | **Plugin-authoring SDK.** Bundle rules, skills, and per-runtime adaptors for any Molecule AI workspace. |
Published on PyPI as [`molecule-ai-sdk`](https://pypi.org/project/molecule-ai-sdk/).
```bash
pip install molecule-ai-sdk
```
---
## molecule_agent — Remote-Agent Client
Write an agent that runs on a laptop, VM, or any machine outside the platform's Docker network. The SDK handles registration, authentication, heartbeat, and A2A communication via the Phase 30 HTTP contract.
### Quick Start
```python
from molecule_agent import RemoteAgentClient
client = RemoteAgentClient(
workspace_id="550e8400-e29b-41d4-a716-446655440000",
platform_url="https://your-platform.example.com",
agent_card={"name": "my-remote-agent", "skills": []},
)
client.register() # mints and persists the auth token
secrets = client.pull_secrets() # returns {"OPENAI_API_KEY": "sk-..."}
client.run_heartbeat_loop() # background heartbeat + state-poll; detects pause/delete
```
### One-liner bootstrap (poll mode)
```bash
pip install molecule-ai-sdk
WORKSPACE_ID=... PLATFORM_URL=... AGENT_TOKEN=... \
python -m molecule_agent connect --handler my_module:handle_message
```
Picks `PollDelivery` automatically when no public URL is available — works behind NAT with no inbound firewall holes. `SIGTERM`/`SIGINT` shut the loop down cleanly.
### A2A Delegation
```python
from molecule_agent import RemoteAgentClient
client = RemoteAgentClient(workspace_id="...", platform_url="...")
# Sync: wait for the peer's response
response = client.delegate(peer_workspace_id, "Summarise the Q1 report")
# Async: get a task_id, poll for result
task_id = client.delegate(peer_workspace_id, "Audit the homepage", async_delegate=True)
result = client.check_delegation_status(task_id)
```
### Inbound Messages (Two delivery paths)
**Poll mode** (default, no public URL needed):
```python
from molecule_agent import RemoteAgentClient, PollDelivery
client = RemoteAgentClient(workspace_id="...", platform_url="...")
client.register()
def handle(msg):
reply = f"Got: {msg.text}"
client.reply(msg, reply) # routes to /notify (canvas user) or /a2a (peer)
client.run_agent_loop(handler=handle) # uses PollDelivery internally
```
**Push mode** (requires a public URL, lower latency):
```python
from molecule_agent import RemoteAgentClient, PushDelivery, A2AServer
server = A2AServer(agent_id="...", inbound_url="https://your-agent.example.com/a2a/inbound")
server.start_in_background()
client = RemoteAgentClient(workspace_id="...", platform_url="...")
client.reported_url = server.inbound_url # register with public URL
client.register()
client.run_agent_loop(handler=handle, delivery=PushDelivery(client, server))
```
### Plugin Install
Agents can install plugins from the platform registry at runtime:
```python
client.install_plugin(source="local://my-plugin")
# or from a tarball
client.install_plugin_from_tarball("/path/to/plugin.tar.gz", expected_sha256="...")
```
### All public exports
```python
from molecule_agent import (
RemoteAgentClient, # Main entry point
A2AServer, # Push-mode inbound HTTP server
PollDelivery, # Default poll-mode delivery
PushDelivery, # Push-mode delivery (needs public URL)
InboundMessage, # Inbound message object
MessageHandler, # Handler callable signature
WorkspaceState, # Pause / running / deleted
PeerInfo, # Peer workspace metadata
compute_plugin_sha256,
verify_plugin_sha256,
)
```
See `examples/remote-agent/run.py` for a full runnable demo.
---
## molecule_plugin — Plugin Authoring SDK
A Molecule AI plugin is a directory that bundles rules, skills, and per-runtime install adaptors. Any plugin that conforms to this contract is installable on any Molecule AI workspace whose runtime supports it.
### Quick Start
```bash
# Clone the template
cp -r template/ my-plugin/
# Edit my-plugin/plugin.yaml, rules/, skills/, adapters/
```
Validate:
```python
from molecule_plugin import validate_manifest
errors = validate_manifest("my-plugin/plugin.yaml")
assert not errors, errors
```
### CLI
```bash
python -m molecule_plugin validate plugin my-plugin/
python -m molecule_plugin validate workspace workspace-configs-templates/claude-code-default/
python -m molecule_plugin validate org org-templates/molecule-dev/
python -m molecule_plugin validate channel channels.yaml
```
Exit code 0 when valid, 1 when errors found — suitable for CI. Add `-q` / `--quiet` to suppress success lines.
### Writing a Custom Adaptor
The default `AgentskillsAdaptor` handles rules + skills. Write a custom adaptor when you need to:
- Register runtime tools dynamically — `ctx.register_tool(name, fn)`
- Register DeepAgents sub-agents — `ctx.register_subagent(name, spec)`
- Write to a non-standard memory file — `ctx.append_to_memory(filename, content)`
```python
from molecule_plugin import InstallContext, InstallResult
class Adaptor:
def __init__(self, plugin_name: str, runtime: str):
self.plugin_name, self.runtime = plugin_name, runtime
async def install(self, ctx: InstallContext) -> InstallResult:
ctx.register_subagent("my-agent", {"prompt": "...", "tools": [...]})
return InstallResult(plugin_name=self.plugin_name, runtime=self.runtime, source="plugin")
async def uninstall(self, ctx: InstallContext) -> None:
pass
```
### Resolution order
For `(plugin_name, runtime)`:
1. **Platform registry** — curated, set by the Molecule AI team
2. **Plugin-shipped**`<plugin_root>/adapters/<runtime>.py` (what this SDK helps you build)
3. **Raw-drop fallback** — copies files, no tools wired
### Testing locally
```python
import asyncio
from pathlib import Path
from molecule_plugin import AgentskillsAdaptor, InstallContext
ctx = InstallContext(
configs_dir=Path("/tmp/configs"),
workspace_id="local",
runtime="claude_code",
plugin_root=Path("./my-plugin"),
)
asyncio.run(AgentskillsAdaptor("my-plugin", "claude_code").install(ctx))
# check /tmp/configs/CLAUDE.md, /tmp/configs/skills/
```
### Supported runtimes
`claude_code`, `deepagents`, `langgraph`, `crewai`, `autogen`, `openclaw`. See the live list:
```bash
curl "$PLATFORM_URL/plugins"
```
---
## Both packages
- **Python:** `>=3.11`, no external async dependencies in `molecule_agent`
(uses blocking `requests` so it embeds in any event loop).
- **Error handling:** Network errors in loops are logged and swallowed so a
transient platform hiccup does not take a remote agent offline. API-level
errors (4xx) propagate via `raise_for_status()`.
- **Token security:** Auth token cached at `~/.molecule/<workspace_id>/.auth_token`
with `0600` permissions.
- **Full documentation:** See `CLAUDE.md` for architecture, platform API
endpoints, SDK conventions, and known issues.