File size: 527 Bytes
383415a
 
6a7abb7
383415a
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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
    ),
)