import Anthropic from "@anthropic-ai/sdk";

const anthropic = new Anthropic({
  apiKey: process.env.SHINZO_API_KEY,
  baseURL: "https://api.app.shinzo.ai/spotlight/anthropic"
});

// Define tools for the agent
const tools: Anthropic.Messages.Tool[] = [
  {
    name: "search_knowledge_base",
    description: "Search the knowledge base for answers to customer questions",
    input_schema: {
      type: "object" as const,
      properties: {
        query: {
          type: "string",
          description: "The search query"
        }
      },
      required: ["query"]
    }
  }
];

// Agent conversation loop
async function runAgent(userMessage: string) {
  const messages: Anthropic.Messages.MessageParam[] = [
    { role: "user", content: userMessage }
  ];

  let response = await anthropic.messages.create({
    model: "claude-sonnet-4-5",
    max_tokens: 2000,
    tools,
    messages
  });

  // Handle tool calls
  while (response.stop_reason === "tool_use") {
    const toolUse = response.content.find(
      (block): block is Anthropic.Messages.ToolUseBlock =>
        block.type === "tool_use"
    );

    if (!toolUse) break;

    // Execute tool (searchKnowledgeBase is your implementation)
    const toolResult = await searchKnowledgeBase(
      (toolUse.input as { query: string }).query
    );

    // Add assistant response and tool result to messages
    messages.push({ role: "assistant", content: response.content });
    messages.push({
      role: "user",
      content: [
        {
          type: "tool_result",
          tool_use_id: toolUse.id,
          content: JSON.stringify(toolResult)
        }
      ]
    });

    // Continue conversation
    response = await anthropic.messages.create({
      model: "claude-sonnet-4-5",
      max_tokens: 2000,
      tools,
      messages
    });
  }

  return response;
}

const response = await runAgent("How do I reset my password?");
console.log(response.content[0]);