DocumentKnowledgeBase 会读取本地文档文件,将它们转换为向量嵌入并加载到向量数据库中。

用法

我们在此示例中使用本地 PgVector 数据库。请确保它正在运行

pip install textract
from agno.knowledge.document import DocumentKnowledgeBase
from agno.vectordb.pgvector import PgVector

knowledge_base = DocumentKnowledgeBase(
    path="data/docs",
    # 表名:ai.documents
    vector_db=PgVector(
        table_name="documents",
        db_url="postgresql+psycopg://ai:ai@localhost:5532/ai",
    ),
)

然后将 knowledge_baseAgent 一起使用:

from agno.agent import Agent
from knowledge_base import knowledge_base

agent = Agent(
    knowledge=knowledge_base,
    search_knowledge=True,
)
agent.knowledge.load(recreate=False)

agent.print_response("就知识库中的某个问题问我")

参数

参数类型默认描述
documentsList[Document]-要用作知识库的 Document 对象列表

DocumentKnowledgeBaseAgentKnowledge 类的子类,可以访问相同的参数。

开发者资源