import { trace, models } from "@fallom/trace";
import OpenAI from "openai";
// Initialize both trace and models
await trace.init({ apiKey: "your-fallom-api-key" });
models.init({ apiKey: "your-fallom-api-key" });
const openrouter = trace.wrapOpenAI(
new OpenAI({
baseURL: "https://openrouter.ai/api/v1",
apiKey: process.env.OPENROUTER_API_KEY,
})
);
// Get assigned model - could be GPT-4o, Claude, Gemini, etc.
const modelId = await models.get("my-experiment", "session-123", {
fallback: "openai/gpt-4o-mini",
});
trace.setSession("my-experiment", "session-123");
const response = await openrouter.chat.completions.create({
model: modelId,
messages: [{ role: "user", content: "Hello!" }],
});