import { Execlave } from '@execlave/sdk';

// Initialize the SDK
const ag = new Execlave({
  apiKey: 'exe_prod_xxx', // or set EXECLAVE_API_KEY env var
  environment: 'production',
});

// Register your agent (idempotent — call on startup)
const agent = await ag.registerAgent({
  agentId: 'my-chatbot',
  name: 'Customer Support Bot',
  type: 'chatbot',
  platform: 'custom',
});

// Trace an LLM call
const trace = ag.startTrace({ agentId: 'my-chatbot', sessionId: 'sess_123' });
trace.setInput(userQuestion);

const answer = await llm.call(userQuestion);

trace.setOutput(answer).setModel('gpt-4-turbo').setTokens(150, 300).setCost(0.012);
trace.finish(); // submits to buffer; flushed in background

// Or use the wrap() helper for automatic tracing
const tracedCall = ag.wrap(
  async (question: string) => {
    return await llm.call(question);
  },
  { agentId: 'my-chatbot' },
);

const result = await tracedCall('Hello!');

// Prompt management
const version = await agent.deployPrompt({
  promptTemplate: 'You are a helpful assistant. Answer: {question}',
  systemMessage: 'Be concise.',
  modelName: 'gpt-4-turbo',
  changeSummary: 'Improved conciseness',
});

// Kill-switch aware
try {
  const trace = ag.startTrace();
  // ...
} catch (err) {
  if (err instanceof AgentPausedError) {
    return 'Service temporarily unavailable.';
  }
  throw err;
}

// Graceful shutdown (flushes remaining traces)
await ag.shutdown();