quivr/backend/parsers/csv.py

from langchain.document_loaders import CSVLoader
from models.files import File
from models.settings import CommonsDep

from .common import process_file


def process_csv(
    commons: CommonsDep,
    file: File,
    enable_summarization,
    brain_id,
    user_openai_api_key,
):
    return process_file(
        commons,
        file,
        CSVLoader,
        enable_summarization,
        brain_id,
        user_openai_api_key,
    )