import { Agent, openai, PageIndexToolkit } from "@agentium/core";

const pageindex = new PageIndexToolkit({
  apiKey: process.env.PAGEINDEX_API_KEY,
});

const agent = new Agent({
  name: "document-analyst",
  model: openai("gpt-4o"),
  instructions: "Analyze uploaded documents. Answer questions accurately with citations.",
  tools: [...pageindex.getTools()],
});

const result = await agent.run(
  "Submit https://example.com/annual-report.pdf and then summarize the revenue breakdown by segment."
);

// After submitting a document
await agent.run("What were the total operating expenses in Q3 2024?");

await agent.run(
  "Compare the risk factors section between the 2023 and 2024 annual reports."
);

const agent = new Agent({
  name: "extractor",
  model: openai("gpt-4o"),
  instructions: "Extract structured data from documents with page citations.",
  tools: [...pageindex.getTools()],
  outputType: z.object({
    items: z.array(z.object({
      field: z.string(),
      value: z.string(),
      page: z.number(),
    })),
  }),
});

import { Agent, openai, PageIndexToolkit, InMemoryKnowledge } from "@agentium/core";

const agent = new Agent({
  name: "hybrid-knowledge",
  model: openai("gpt-4o"),
  tools: [...new PageIndexToolkit({ apiKey: "..." }).getTools()],
  knowledge: new InMemoryKnowledge({ /* local vector search for quick lookups */ }),
  instructions: "Use PageIndex for complex document analysis. Use knowledge search for quick factual lookups.",
});