quivr/backend/packages/files/parsers/notebook.py

from langchain_community.document_loaders import NotebookLoader
from models import File

from .common import process_file


def process_ipnyb(
    file: File, brain_id, original_file_name, integration=None, integration_link=None
):
    return process_file(
        file=file,
        loader_class=NotebookLoader,
        brain_id=brain_id,
        original_file_name=original_file_name,
        integration=integration,
        integration_link=integration_link,
    )