import { Adaline } from '@adaline/client';
import OpenAI from 'openai';

// Initialize the Adaline client, must setup ADALINE_API_KEY environment variable prior to using the SDK
const adaline = new Adaline();

// Get your deployed prompt configuration
const deployment = await adaline.getLatestDeployment({
  promptId: 'your-prompt-id',
  deploymentEnvironmentId: 'your-deployment-environment-id'
});

// Initialize monitoring for your project
const monitor = adaline.initMonitor({
  projectId: 'your-project-id',
  flushInterval: 5, // flush every 5 seconds
  maxBufferSize: 10  // or when buffer reaches 10 items
});

// Create a trace for the entire user interaction
const trace = monitor.logTrace({
  name: 'Chat Completion',
  status: 'unknown',
  sessionId: 'user-session-123'
});

// Log the LLM call as a span
const llmSpan = trace.logSpan({
  name: 'OpenAI GPT-4 Call',
  status: 'unknown',
  promptId: deployment.id,
  deploymentId: deployment.id
});

try {
  // Make your LLM call with the deployed configuration
  const openai = new OpenAI();
  const response = await openai.chat.completions.create({
    model: deployment.prompt.config.model,
    messages: deployment.prompt.messages,
    ...deployment.prompt.config.settings
  });

  // Update span with successful result
  llmSpan.update({
    status: 'success',
    content: {
      type: 'Model',
      provider: deployment.prompt.config.providerName,
      model: deployment.prompt.config.model,
      input: JSON.stringify(deployment.prompt.messages),
      output: JSON.stringify(response.choices[0].message)
    }
  });
  
  trace.update({ status: 'success' });
} catch (error) {
  // Log errors
  llmSpan.update({ status: 'failure' });
  trace.update({ status: 'failure' });
}

// End tracking, this will mark the trace (and all its spans) as complete and ready to be flushed
trace.end();

// Logs are automatically flushed every 5 seconds, or manually flush:
await monitor.flush();

const controller = await adaline.initLatestDeployment({
  promptId: 'your-prompt-id',
  deploymentEnvironmentId: 'your-deployment-environment-id',
  refreshInterval: 60 // refresh every 60 seconds
});

// Always get the latest cached prompt deployment
const deployment = await controller.get();

// Or force a fresh fetch of the prompt deployment
const refreshedDeployment = await controller.get(true);

// Stop when done
controller.stop();