Python SDK

pip install retrace

import retrace

retrace.configure(
    api_key="rt_live_...",
    base_url="https://api.retrace.yashbogam.me",
    project_id="my-project",
)

@retrace.record(name="research-agent")
def research(topic: str) -> str:
    response = client.models.generate_content(
        model="gemini-3.1-pro-preview",
        messages=[{"role": "user", "content": f"Research {topic}"}]
    )
    return response.choices[0].message.content

with retrace.record(name="agent", input={"prompt": "hello"}) as t:
    result = agent.run("hello")
    t.output = result

with retrace.record(name="pipeline") as t:
    t.metadata = {"environment": "staging", "version": "1.2.0"}
    result = run_pipeline()
    t.output = result

with retrace.record(name="agent") as t:
    span = t._recorder.start_span(
        name="web_search",
        span_type=retrace.SpanType.TOOL_CALL,
        input={"query": "quantum computing"}
    )
    results = search("quantum computing")
    t._recorder.end_span(span.id, output=results)

from retrace.interceptors.gemini import install_gemini_interceptor
install_gemini_interceptor()

retrace.configure(api_key="rt_live_...", transport="http")

import os
os.environ["RETRACE_ENABLED"] = "false"