agentbase-autoheal

agentbase-sdk

Deploy. Observe. Heal. The Python SDK for AgentBase - the operating system for AI agents.

Install

pip install agentbase-sdk

Quick Start (2 lines to instrument)

import agentbase_sdk as ab

ab.init(api_key="ab_live_...", agent_name="my-agent")
ab.instrument_openai()  # auto-trace every LLM call

# That's it. Every OpenAI call is now traced with:
# - Token counts & costs
# - Latency
# - Error detection
# - Auto-heal on failures

Manual Tracing

import agentbase_sdk as ab

ab.init(api_key="ab_live_...", agent_name="support-bot")

with ab.trace("handle-ticket") as t:
    t.set_input(user_question)

    with ab.span("classify", span_type="llm") as s:
        s.set_model("gpt-4o")
        intent = classify(user_question)
        s.set_output(intent)
        s.set_tokens(prompt_tokens=200, completion_tokens=20)

    with ab.span("search-kb", span_type="tool") as s:
        s.set_tool("knowledge_base")
        results = search(intent)
        s.set_output(str(results))

    with ab.span("respond", span_type="llm") as s:
        s.set_model("gpt-4o")
        answer = generate(intent, results)
        s.set_output(answer)
        s.set_tokens(prompt_tokens=800, completion_tokens=200)

    t.set_output(answer)

Decorators

from agentbase_sdk import observe, heal

@observe(as_trace=True)
def handle_support_ticket(question: str) -> str:
    return my_agent.run(question)

@heal(retry=3, fallback=lambda: "Sorry, please try again.")
def risky_api_call(query: str) -> str:
    return external_api.search(query)

Auto-Instrumentation

ab.instrument_openai()      # OpenAI SDK
ab.instrument_anthropic()   # Anthropic SDK
ab.instrument_langchain()   # LangChain
ab.instrument_crewai()      # CrewAI

Auto-Heal

When your agent encounters errors, AgentBase automatically:

Dashboard

View all traces, costs, and heal events at your AgentBase dashboard.

Links

• API Documentation
• Website