带知识库的智能体

git clone https://github.com/astreus-ai/examples
cd examples/agent-with-knowledge
npm install

npm install @astreus-ai/astreus

# .env
# LLM API key
OPENAI_API_KEY=sk-your-openai-api-key-here

# Knowledge database (required for RAG)
KNOWLEDGE_DB_URL=postgresql://username:password@localhost:5432/knowledge_db

# Main database for agent persistence
DB_URL=sqlite://./astreus.db

import { Agent } from '@astreus-ai/astreus';

// Create agent with knowledge enabled
const agent = await Agent.create({
  name: 'CosmosBot',
  model: 'gpt-4o',
  embeddingModel: 'text-embedding-3-small', // Specify embedding model directly
  knowledge: true,
  systemPrompt: 'You can search and retrieve information from scientific knowledge bases about the cosmos and universe.'
});

// Add knowledge from scientific book about the sun and cosmos
await agent.addKnowledgeFromFile(
  './data/The Sun\'s Light and Heat.pdf',
  { category: 'solar-physics', version: '1.0' }
);

// Agent automatically uses knowledge in conversations
const response = await agent.ask("What is Correction for Atmospheric Absorption? Explain.");
console.log(response); // Uses knowledge base automatically

npm run dev