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
    ),
)