import { Mutagent } from '@mutagent/sdk';

const client = new Mutagent({ apiKey: process.env.MUTAGENT_API_KEY });

const job = await client.optimization.optimizePrompt({
  id: 42,  // Prompt ID
  body: {
    datasetId: 7,
    config: {
      maxIterations: 10,
      targetScore: 0.9,
      patience: 3,
      model: 'claude-sonnet-4-6',
      tuningParams: {
        evaluationModel: 'claude-sonnet-4-6',
      },
    },
    executionMode: 'worker_loop',
  },
});

console.log('Job started:', job);

const status = await client.optimization.getOptimization({
  id: 'job-uuid-here',
});

console.log('Status:', status);

const page = await client.optimization.listOptimizations({
  status: 'running',
  limit: 20,
  offset: 0,
});

for await (const job of page) {
  console.log(job.id, '-', job.status, `(${job.progress}%)`);
}

const progress = await client.optimization.getOptimizationProgress({
  id: 'job-uuid-here',
});

console.log('Progression:', progress);

await client.optimization.pauseOptimization({
  id: 'job-uuid-here',
});

console.log('Job paused');

await client.optimization.resumeOptimization({
  id: 'job-uuid-here',
});

console.log('Job resumed');

await client.optimization.cancelOptimization({
  id: 'job-uuid-here',
});

console.log('Job cancelled');

async function waitForOptimization(jobId: string) {
  while (true) {
    const status = await client.optimization.getOptimization({
      id: jobId,
    });

    // Access status fields from the response
    console.log(`Job status: ${JSON.stringify(status)}`);

    // Check for terminal states
    const statusStr = JSON.stringify(status);
    if (statusStr.includes('"completed"')) {
      console.log('Optimization complete');
      return status;
    }
    if (statusStr.includes('"failed"') || statusStr.includes('"cancelled"')) {
      throw new Error('Job ended without completing');
    }

    await new Promise(r => setTimeout(r, 5000));
  }
}

const result = await waitForOptimization('job-uuid-here');

interface OptimizeConfig {
  maxIterations?: number;      // Max optimization cycles (default: 10)
  targetScore?: number;        // Stop when score reached (0-1)
  patience?: number;           // Stop after N iterations with no improvement
  dryRun?: boolean;            // Validate config without running
  model?: string;              // Target LLM model
  tuningParams?: {             // Advanced tuning parameters
    evaluationModel?: string;  // Model used for evaluation scoring
    [key: string]: unknown;
  };
}

enum ExecutionMode {
  WorkerLoop = 'worker_loop',
  BunWorker = 'bun_worker',
}

interface OptimizeJobResponse {
  id: string;
  promptId: number;
  promptGroupId: string;
  datasetId: number;
  status: 'queued' | 'running' | 'paused' | 'completed' | 'failed' | 'cancelled';
  executionMode: 'worker_loop' | 'bun_worker';
  config: Record<string, unknown>;
  progress: number;
  currentIteration: number;
  maxIterations: number;
  currentStage: string | null;
  currentScore: number | null;
  bestScore: number | null;
  bestIteration: number | null;
  resultPromptId: number | null;
  error: string | null;
  createdAt: string;
  startedAt: string | null;
  completedAt: string | null;
}

interface ScoreProgression {
  iteration: number;
  score: number | null;
}