agent_observability.py

"""A runnable day-one observability path for an LLM agent."""

import infrai_observability as infrai


def run_tool(agent, tool_name, tool):
    """Count a tool run and capture its exception with agent context."""
    infrai.metrics.report(
        type="counter", name="agent.tool.started", value=1,
        tags={"agent": agent, "tool": tool_name},
    )
    try:
        result = tool()
    except Exception as exc:
        infrai.errors.capture(
            message=f"{agent}/{tool_name}: {exc}",
            level="error",
            exception=__import__("traceback").format_exc(),
            context={"agent": agent, "step": tool_name},
        )
        raise
    infrai.metrics.report(
        type="counter", name="agent.tool.succeeded", value=1,
        tags={"agent": agent, "tool": tool_name},
    )
    return result


def main():
    answer = run_tool("research-agent", "local-answer", lambda: "observability chosen")
    print(answer)


if __name__ == "__main__":
    main()