import { ZeroLeaks, createRuntimeTarget } from "@zeroleaks/sdk";
import { productionAgent } from "./agent";

const zeroleaks = new ZeroLeaks();

const target = createRuntimeTarget({
  definition: {
    name: "Support agent",
    provider: "custom",
    model: "production-model",
    instructions: SUPPORT_INSTRUCTIONS,
    tools: productionTools.map((tool) => ({
      name: tool.name,
      description: tool.description,
      inputSchema: tool.inputSchema,
      outputSchema: tool.outputSchema,
    })),
  },
  invoke: async ({ messages, sessionId, signal }) => {
    const result = await productionAgent({ messages, sessionId, signal });
    return {
      text: result.text,
      messages: result.messages,
      toolCalls: result.toolCalls,
      usage: result.usage,
    };
  },
  reset: async (sessionId) => {
    await productionAgent.reset(sessionId);
  },
});

const { scan, report } = await zeroleaks.runtimeScans.run(target, {
  eventConcurrency: 8,
  workerStallTimeoutMs: 6 * 60_000,
  scan: {
    scanMode: "full",
    knowledgeProfile: "production",
    attackSurfaces: [
      "direct_chat",
      "indirect_content",
      "tool_calling",
      "multi_turn",
    ],
  },
  onPoll: (current) => console.log(current.currentPhase),
});

console.log(scan.status, report.overallScore);