Add Human-in-the-Loop Controls

type CheckpointState = {
  conversationId: string;
  messages: { role: string; content: string }[];
  pendingAction: string;
  requestedAt: string; // ISO timestamp
};

import OpenAI from "openai";

const client = new OpenAI({
  baseURL: "https://opendunes.com/api/v1",
  apiKey: process.env.OPENDUNES_API_KEY,
});

const CHECKPOINT_SIGNAL = "CHECKPOINT_REQUIRED:";

async function runAgentLoop(initialMessages: { role: string; content: string }[]) {
  const messages = [...initialMessages];

  while (true) {
    const response = await client.chat.completions.create({
      model: "anthropic/claude-opus-4.8",
      messages,
      max_tokens: 1024,
    });

    const assistantMessage = response.choices[0].message;
    messages.push({ role: "assistant", content: assistantMessage.content ?? "" });

    const content = assistantMessage.content ?? "";

    if (content.startsWith(CHECKPOINT_SIGNAL)) {
      // Extract the action description after the signal prefix
      const pendingAction = content.slice(CHECKPOINT_SIGNAL.length).trim();
      const approval = await waitForApproval(messages, pendingAction);

      if (!approval) {
        messages.push({ role: "user", content: "Action rejected. Do not proceed. Summarize what you planned to do and stop." });
      } else {
        messages.push({ role: "user", content: "Approved. Continue." });
      }
      continue;
    }

    // No checkpoint signal — task is complete
    return content;
  }
}

const systemPrompt = `You are a deployment assistant. 
When you are about to take an irreversible action (delete, deploy, send, charge), 
stop and output exactly: CHECKPOINT_REQUIRED: <one-sentence description of the action>.
Do nothing else until you receive "Approved." or "Rejected."`;

async function waitForApproval(
  messages: { role: string; content: string }[],
  pendingAction: string,
  timeoutMs = 30 * 60 * 1000, // 30 minutes
): Promise<boolean> {
  const conversationId = crypto.randomUUID();
  const deadline = Date.now() + timeoutMs;

  await saveCheckpoint({ conversationId, messages, pendingAction, requestedAt: new Date().toISOString() });
  await notifyReviewer(conversationId, pendingAction);

  while (Date.now() < deadline) {
    const decision = await pollDecision(conversationId);
    if (decision === "approved") return true;
    if (decision === "rejected") return false;
    await new Promise((r) => setTimeout(r, 3000)); // poll every 3s
  }

  // Timed out — treat as rejected
  await resolveCheckpoint(conversationId, "timeout");
  return false;
}