import { PromptManager } from '@prompt-manager/core';
import { MemoryStorage } from '@prompt-manager/storage-memory';
import { SimpleTemplate } from '@prompt-manager/template-simple';

const manager = new PromptManager({
  storage: new MemoryStorage(),
  templates: new SimpleTemplate(),
});

// Register a prompt
await manager.register({
  definition: {
    name: 'greeting',
    version: '1.0.0',
    template: {
      user: 'Hello, {{name}}!',
    },
  },
});

// Render the prompt
const result = await manager.render({
  name: 'greeting',
  options: {
    variables: { name: 'World' },
  },
});

console.log(result.messages); // [{ role: 'user', content: 'Hello, World!' }]

const execution: ExecutionData = {
  promptName: 'greeting',
  version: '1.0.0',
  variables: { name: 'World' },
  response: 'Hi there!',
  latencyMs: 150,
  cost: 0.0001,
};

await manager.logExecution({ data: execution });

const metrics = await manager.getMetrics({
  filters: {
    promptName: 'greeting',
    version: '1.0.0',
  },
});

console.log(metrics.executionCount);
console.log(metrics.avgLatencyMs);
console.log(metrics.totalCost);