quivr/backend/parsers/code_python.py

15 lines
421 B
Python
Raw Normal View History

from langchain.document_loaders import PythonLoader
from models import File
from .common import process_file
async def process_python(file: File, enable_summarization, brain_id, user_openai_api_key):
return await process_file(
file=file,
loader_class=PythonLoader,
enable_summarization=enable_summarization,
brain_id=brain_id,
user_openai_api_key=user_openai_api_key,
)