Axiom AI SDK provides an API to wrap your AI calls with observability instrumentation.
github.com/axiomhq/ai
axiomhq/ai