RAG Example

User Question → Agent → KnowledgeBase Tool → Vector Search → Context → LLM → Answer

import {
  Agent,
  openai,
  KnowledgeBase,
  InMemoryVectorStore,
  OpenAIEmbedding,
} from "@agentium/core";

const embedder = new OpenAIEmbedding();
const vectorStore = new InMemoryVectorStore(1536);

const kb = new KnowledgeBase({
  name: "company-docs",
  vectorStore,
});

await kb.initialize();

// Add documents
await kb.addDocuments([
  {
    id: "pricing",
    content: "Our Pro plan costs $49/month and includes unlimited agents, 100k API calls, and priority support.",
    metadata: { category: "pricing" },
  },
  {
    id: "features",
    content: "Agentium supports multi-agent teams, workflows, RAG, and multi-modal inputs including images and audio.",
    metadata: { category: "features" },
  },
  {
    id: "setup",
    content: "Install with npm install @agentium/core. Set your OPENAI_API_KEY environment variable to get started.",
    metadata: { category: "getting-started" },
  },
]);

// Create agent with KB as a tool
const agent = new Agent({
  name: "support-bot",
  model: openai("gpt-4o"),
  instructions: "You are a support agent. Use the search tool to find answers from the knowledge base. Always cite sources.",
  tools: [kb.asTool()],
  logLevel: "info",
});

const result = await agent.run("How much does the Pro plan cost?");
console.log(result.text);
// "The Pro plan costs $49/month and includes unlimited agents, 100k API calls, and priority support."

import {
  Agent,
  openai,
  KnowledgeBase,
  QdrantVectorStore,
  OpenAIEmbedding,
} from "@agentium/core";

const embedder = new OpenAIEmbedding();

const vectorStore = new QdrantVectorStore({
  url: "http://localhost:6334",
  collectionName: "docs",
  dimensions: 1536,
});

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

await kb.initialize();

// Add documents (only needed once)
await kb.addDocuments([
  { id: "doc-1", content: "Your document content here..." },
  { id: "doc-2", content: "Another document..." },
]);

const agent = new Agent({
  name: "rag-agent",
  model: openai("gpt-4o"),
  instructions: "Answer questions using the knowledge base.",
  tools: [kb.asTool({ topK: 3, description: "Search the documentation" })],
});

const result = await agent.run("What does the documentation say about X?");
console.log(result.text);

import {
  Agent,
  openai,
  KnowledgeBase,
  MongoDBVectorStore,
  OpenAIEmbedding,
} from "@agentium/core";

const embedder = new OpenAIEmbedding();

const vectorStore = new MongoDBVectorStore({
  uri: "mongodb+srv://user:pass@cluster.mongodb.net",
  dbName: "myapp",
  collectionName: "documents",
  indexName: "vector_index",
  dimensions: 1536,
});

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

await kb.initialize();

// Add and query as above

import {
  Agent, openai, KnowledgeBase,
  InMemoryVectorStore, OpenAIEmbedding,
} from "@agentium/core";

const kb = new KnowledgeBase({
  name: "Product Docs",
  vectorStore: new InMemoryVectorStore(new OpenAIEmbedding()),
  searchMode: "hybrid",
});

await kb.initialize();
await kb.addDocuments([
  { id: "pricing", content: "Our Pro plan costs $49/month with unlimited API calls." },
  { id: "sla", content: "Enterprise SLA guarantees 99.9% uptime with 4-hour response time." },
]);

const agent = new Agent({
  name: "support-bot",
  model: openai("gpt-4o"),
  tools: [kb.asTool({ topK: 3, searchMode: "hybrid" })],
  instructions: "Answer using the knowledge base. Cite sources.",
});

const result = await agent.run("What's the SLA uptime guarantee?");

kb.asTool({
  toolName: "search_docs",                   // Custom tool name
  description: "Search company documentation", // Custom description
  topK: 5,                                    // Number of results
  minScore: 0.7,                              // Minimum relevance score
  formatResults: (results) => {               // Custom result formatting
    return results
      .map((r) => `[${r.document.id}] ${r.document.content}`)
      .join("\n\n");
  },
});