RAG 管道

入库文档、嵌入，并在聊天中检索相关上下文。将 @agentskit/rag 与任意嵌入器及向量存储配合使用。

设置

import { createRAG } from '@agentskit/rag'
import { openaiEmbedder } from '@agentskit/adapters'

const rag = createRAG({
  embed: openaiEmbedder({ apiKey: process.env.OPENAI_API_KEY! }),
  store: yourVectorStore,  // SQLite, Redis, or in-memory
  chunkSize: 512,
  chunkOverlap: 50,
})

文档入库

await rag.ingest([
  { id: 'readme', content: readFileSync('README.md', 'utf-8'), source: 'README.md' },
  { id: 'guide', content: readFileSync('docs/guide.md', 'utf-8'), source: 'guide.md' },
])

直接搜索

const results = await rag.search('how to configure tools', { topK: 3 })
results.forEach(doc => {
  console.log(`[${doc.source}] ${doc.content.slice(0, 100)}...`)
})

与聊天配合

createRAG 返回 Retriever——可直接传给 useChat 或 runtime：

import { useChat } from '@agentskit/react'

function RAGChat() {
  const chat = useChat({
    adapter: yourAdapter,
    retriever: rag,  // retrieved context auto-injected into system prompt
  })
  // ... render chat UI
}

自定义分块

const rag = createRAG({
  embed: yourEmbedder,
  store: yourStore,
  split: (text) => text.split('\n\n'),  // paragraph-based chunking
})