quivr/backend/core/parsers/csv.py

23 lines
438 B
Python
Raw Normal View History

from langchain.document_loaders import CSVLoader
2023-06-28 20:39:27 +03:00
from models.files import File
2023-06-19 23:46:25 +03:00
from models.settings import CommonsDep
from .common import process_file
2023-05-13 00:05:31 +03:00
2023-05-22 09:39:55 +03:00
def process_csv(
commons: CommonsDep,
2023-06-28 20:39:27 +03:00
file: File,
enable_summarization,
2023-06-28 20:39:27 +03:00
brain_id,
user_openai_api_key,
):
return process_file(
commons,
file,
CSVLoader,
enable_summarization,
2023-06-28 20:39:27 +03:00
brain_id,
user_openai_api_key,
)