Vercel AI SDK Integration

Trace your Vercel AI SDK calls with one function wrapper.

Install

npm install nyraxis-sdk

Quick start

import { generateText } from "ai"
import { openai } from "@ai-sdk/openai"
import { traceVercelAI } from "nyraxis-sdk"

const prompt = "Explain AI governance in 3 sentences"
const result = await generateText({ model: openai("gpt-4o"), prompt })

await traceVercelAI(result, prompt, "gpt-4o", {
  apiKey: "nyx_your_api_key",
  baseUrl: "http://localhost:8000",
  agentName: "my-next-app",
})

With LangChain.js

import { LangChainCallbackHandler } from "nyraxis-sdk"
import { ChatOpenAI } from "@langchain/openai"

const handler = new LangChainCallbackHandler({
  apiKey: "nyx_your_api_key",
  agentName: "my-langchain-js-agent",
})

const llm = new ChatOpenAI({ model: "gpt-4o", callbacks: [handler] })
const response = await llm.invoke("Hello")
await handler.flush()

What gets captured

• Model name, prompt, completion
• Token counts (prompt + completion)
• Cost auto-calculated
• Governance policies evaluated server-side

Other integrations

• LangChain
• CrewAI
• LlamaIndex