import { KnowledgeBase } from "@agentium/core";

const kb = new KnowledgeBase({
  name: "Product Docs",
  vectorStore: myVectorStore,
  collection: "product_docs", // optional, defaults to sanitized name
});

await kb.initialize();
await kb.add({ id: "doc-1", content: "Our API supports webhooks..." });
const results = await kb.search("How do I set up webhooks?");

const kb = new KnowledgeBase({
  name: "Support KB",
  vectorStore: vectorStore,
});

await kb.initialize();
await kb.addDocuments(docs);

const agent = new Agent({
  name: "SupportBot",
  model: openai("gpt-4o"),
  tools: [kb.asTool({ toolName: "search_kb", topK: 5 })],
  instructions: "Answer using the knowledge base when relevant.",
});

const result = await agent.run("How do I reset my password?");
// Agent calls search_kb internally and uses retrieved docs in its answer