OpenAI Integration

npm install costlens openai

import { CostLens } from 'costlens';
import OpenAI from 'openai';

const openai = new OpenAI({ apiKey: process.env.OPENAI_API_KEY });
const costlens = new CostLens({ 
  apiKey: process.env.COSTLENS_API_KEY 
});

// Wrap your client
const tracked = costlens.wrapOpenAI(openai);

// Use it exactly like normal OpenAI
const result = await tracked.chat.completions.create({
  model: 'gpt-4',
  messages: [{ role: 'user', content: 'Hello!' }]
});
// ✅ Automatically tracked!

import { CostLens } from 'costlens';
import OpenAI from 'openai';

const openai = new OpenAI({ apiKey: process.env.OPENAI_API_KEY });
const costlens = new CostLens({ 
  apiKey: process.env.COSTLENS_API_KEY 
});

const params = {
  model: 'gpt-4',
  messages: [{ role: 'user', content: 'Hello!' }]
};

const start = Date.now();
const result = await openai.chat.completions.create(params);
await costlens.trackOpenAI(params, result, Date.now() - start);

const costlens = new CostLens({ 
  apiKey: process.env.COSTLENS_API_KEY,
  enableCache: true  // Enable caching
});

const tracked = costlens.wrapOpenAI(openai);

// Cache this response for 1 hour
const result = await tracked.chat.completions.create(
  { model: 'gpt-4', messages: [...] },
  { cacheTTL: 3600000 }
);

const stream = await tracked.chat.completions.stream({
  model: 'gpt-4',
  messages: [{ role: 'user', content: 'Tell me a story' }]
});

for await (const chunk of stream) {
  process.stdout.write(chunk.choices[0]?.delta?.content || '');
}
// ✅ Automatically tracked after stream completes

import { CostLens } from 'costlens';
import OpenAI from 'openai';

// Initialize
const costlens = new CostLens({ 
  apiKey: process.env.COSTLENS_API_KEY 
});
const openai = new OpenAI({
  apiKey: process.env.OPENAI_API_KEY
});

// Make your API call
const params = {
  model: 'gpt-4',
  messages: [
    { role: 'system', content: 'You are a helpful assistant.' },
    { role: 'user', content: 'Hello!' }
  ]
};

const start = Date.now();
const result = await openai.chat.completions.create(params);

// Track the call
await costlens.trackOpenAI(params, result, Date.now() - start);

console.log(result.choices[0].message.content);

const params = {
  model: 'gpt-4',
  messages: [...]
};

const start = Date.now();
const result = await openai.chat.completions.create(params);

// Pass promptId as 4th parameter
await costlens.trackOpenAI(
  params, 
  result, 
  Date.now() - start,
  'customer-support-v2'
);

const start = Date.now();
try {
  const result = await openai.chat.completions.create(params);
  await costlens.trackOpenAI(params, result, Date.now() - start);
  return result;
} catch (error) {
  await costlens.trackError(
    'openai',
    params.model,
    JSON.stringify(params.messages),
    error,
    Date.now() - start
  );
  throw error;
}