Tracing

@decimalai.trace(agent_name="my-agent")
def run_agent(query: str) -> str:
    response = openai.chat.completions.create(
        model="gpt-4o",
        messages=[{"role": "user", "content": query}],
    )
    return response.choices[0].message.content

# Every call to run_agent() creates a trace
result = run_agent("What is the capital of France?")

with decimalai.start_trace(agent_name="my-agent") as ctx:
    # Everything inside this block is part of one trace
    decimalai.log_llm_call(
        model="gpt-4o",
        input=[{"role": "user", "content": "Hello"}],
        output={"content": "Hi there!"},
    )

decimalai.log_llm_call(
    model="gpt-4o",
    input=[{"role": "user", "content": "Hello"}],
    output={"content": "Hi there!"},
    input_tokens=5,
    output_tokens=3,
    latency_ms=320,
)

decimalai.log_tool_call(
    name="search_flights",
    input={"from": "SFO", "to": "NRT"},
    output={"results": [...]},
    latency_ms=150,
)

@decimalai.tool
def search_flights(origin: str, destination: str) -> list:
    """Search for available flights."""
    return flight_api.search(origin, destination)

# Tool calls are auto-logged when used inside a traced function