@hoadev/trace-ai-sdk

SDK client cho hệ thống AI Trace API, dùng để track trace/span/event/LLM usage và query cost nhanh mà không cần viết request thủ công.

Cài đặt

bun add @hoadev/trace-ai-sdk

Khởi tạo client

import { createAiTraceClient } from '@hoadev/trace-ai-sdk'

const client = createAiTraceClient({
  baseUrl: 'http://localhost:3000',
  tenantId: 'tenant-001',
  timeoutMs: 10000
})

Dùng endpoint-level API

const trace = await client.traces.create({
  tenantId: 'tenant-001',
  serviceName: 'assistant-service',
  status: 'running'
})

const spans = await client.spans.list({ traceId: trace.traceId, limit: 20 })
console.log(spans.items.length, spans.pagination?.total)

Dùng helper lifecycle API

const traceSession = await client.helper.startTrace({
  tenantId: 'tenant-001',
  serviceName: 'assistant-service',
  chatId: 'chat-123'
})

const spanSession = await traceSession.startSpan({
  spanType: 'model_generation',
  name: 'Generate answer'
})

await spanSession.recordLlmUsage({
  model: 'gpt-4',
  provider: 'openai',
  promptTokens: 120,
  completionTokens: 80
})

await spanSession.endSuccess({ output: { text: 'done' } })
await traceSession.endSuccess()

Xử lý lỗi

import { AiTraceSdkError } from '@hoadev/trace-ai-sdk'

try {
  await client.pricing.getActive('openai', 'gpt-4')
} catch (error) {
  if (error instanceof AiTraceSdkError) {
    console.error(error.code, error.message, error.httpStatus)
  }
}