Skip to main content

​
Example

Install arxiv using this guide
knowledge_base.py
from phi.knowledge.arxiv import ArxivKnowledgeBase
from phi.vectordb.pgvector import PgVector

from resources import vector_db

knowledge_base = ArxivKnowledgeBase(
    queries=["Generative AI", "Machine Learning"],
    # Table name: llm.arxiv_documents
    vector_db=PgVector(
        table_name="arxiv_documents",
        db_url=vector_db.get_db_connection_local(),
    ),
)

​
ArxivKnowledgeBase Params

ParameterTypeDefaultDescription
queriesList[str][]Queries to search
readerArxivReaderArxivReader()A ArxivReader that reads the articles and converts them into Documents for the vector database

​
AgentKnowledge Params

ArxivKnowledgeBase is a subclass of the AgentKnowledge class and has access to the same params
ParameterTypeDefaultDescription
readerOptional[Reader]NoneReader to read the documents
vector_dbOptional[VectorDb]NoneVector db to store the knowledge base
num_documentsint2Number of relevant documents to return on search
optimize_onOptional[int]1000Number of documents to optimize the vector db on
driverstr"knowledge"Driver for the Assistant knowledge