from agno.knowledge.pdf_url import PDFUrlKnowledgeBase
from agno.vectordb.pgvector import PgVector, SearchType
# 数据库 URL
db_url = "postgresql+psycopg://ai:ai@localhost:5532/ai"
# 初始化混合向量数据库
hybrid_db = PgVector(
table_name="recipes",
db_url=db_url,
search_type=SearchType.hybrid # 混合搜索
)
# 使用混合搜索加载 PDF 知识库
knowledge_base = PDFUrlKnowledgeBase(
urls=["https://agno-public.s3.amazonaws.com/recipes/ThaiRecipes.pdf"],
vector_db=hybrid_db,
)
# 将数据加载到数据库中(首次设置)
knowledge_base.load(recreate=True, upsert=True)
# 运行混合搜索查询
results = hybrid_db.search("chicken coconut soup", limit=5)
print("Hybrid Search Results:", results)