import { track } from '@burn0/burn0'

// Wrap any async function that makes API calls
await track('chat', { userId: 'user123' }, async () => {
  const response = await openai.chat.completions.create({
    model: 'gpt-4o-mini',
    messages: [{ role: 'user', content: 'Hello' }],
  })
  return response
})

track(feature: string, metadata: Record<string, string | number | boolean>, fn: () => Promise<void>): Promise<void>

import { startSpan } from '@burn0/burn0'

const span = startSpan('chat', { userId: 'user123' })

// ... make API calls ...

span.end()

app.post('/api/chat', async (req, res) => {
  const { userId, message } = req.body

  await track('chat', { userId }, async () => {
    const response = await openai.chat.completions.create({
      model: 'gpt-4o-mini',
      messages: [{ role: 'user', content: message }],
    })
    res.json({ response: response.choices[0].message.content })
  })
})

// Summarize feature
await track('summarize', { documentId: doc.id }, async () => {
  await anthropic.messages.create({ ... })
})

// Search feature
await track('search', { query: q }, async () => {
  await openai.embeddings.create({ ... })
  await pinecone.query({ ... })
})