umai-agent-sdk 0.1.0

Python SDK for UMAI Agent Mesh Governance and guardrails.

UMAI Agent SDK for Python

Production Python SDK for UMAI Agent Mesh Governance, guardrails, signed agent identity, and agent work-tree observability.

pip install umai-agent-sdk
pip install umai-agent-sdk[openai]

from umai import UmaiClient
from umai.stores import FileIdentityStore

umai = UmaiClient(endpoint="https://api.umai.ai", api_key="...")
agent = umai.agent(
    "support-agent",
    identity_store=FileIdentityStore(allow_plaintext_private_key=True),
)

await agent.register(bootstrap_token="...")
run = await agent.start_run(guardrail_id="gr-prod")
decision = await agent.guard_tool_input(
    guardrail_id="gr-prod",
    run_id=run.run_id,
    tool_name="crm.lookup",
    payload_summary="Lookup customer profile",
    messages=[{"role": "assistant", "content": "Call crm.lookup"}],
)
await agent.complete_run(run.run_id)

The SDK is a convenience layer over UMAI HTTP APIs. Every SDK feature maps to public UMAI endpoints and can be implemented without the SDK.