PyMuPDF Loader

pip install "upsonic[loaders]"

pip install rapidocr-onnxruntime

from upsonic import Agent, Task, KnowledgeBase
from upsonic.loaders import PyMuPDFLoader, PyMuPDFLoaderConfig
from upsonic.embeddings import OpenAIEmbedding, OpenAIEmbeddingConfig
from upsonic.text_splitter import RecursiveChunker, RecursiveChunkingConfig
from upsonic.vectordb import ChromaProvider, ChromaConfig, ConnectionConfig, Mode

# Configure loader
loader_config = PyMuPDFLoaderConfig(
    extraction_mode="hybrid",
    text_extraction_method="dict",
    preserve_layout=True
)
loader = PyMuPDFLoader(loader_config)

# Setup KnowledgeBase
embedding = OpenAIEmbedding(OpenAIEmbeddingConfig())
chunker = RecursiveChunker(RecursiveChunkingConfig())
vectordb = ChromaProvider(ChromaConfig(
    collection_name="pymupdf_docs",
    vector_size=1536,
    connection=ConnectionConfig(mode=Mode.IN_MEMORY)
))

kb = KnowledgeBase(
    sources=["document.pdf"],
    embedding_provider=embedding,
    vectordb=vectordb,
    loaders=[loader],
    splitters=[chunker]
)

# Query with Agent
agent = Agent("openai/gpt-4o")
task = Task("Summarize the document", context=[kb])
result = agent.do(task)
print(result)