| from phi.knowledge.combined import CombinedKnowledgeBase | |
| from phi.vectordb.pgvector import PgVector2 | |
| from src.knowledge_bases.json import json_knowledge_base | |
| from src.databases.postgres import sqlalchemy_engine | |
| knowledge_base = CombinedKnowledgeBase( | |
| sources=[ | |
| json_knowledge_base, | |
| # website_knowledge_base, | |
| # local_pdf_knowledge_base, | |
| ], | |
| vector_db=PgVector2( | |
| # Table name: ai.combined_documents | |
| collection="combined_documents", | |
| db_engine=sqlalchemy_engine | |
| ), | |
| ) | |