From c011455dc4fc7cc171bdcacbbd7e9b1ce20d7a2b Mon Sep 17 00:00:00 2001 From: Huu Le <39040748+leehuwuj@users.noreply.github.com> Date: Thu, 12 Sep 2024 13:51:48 +0700 Subject: [PATCH] fix cannot upload file (#286) --- templates/components/services/python/file.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/templates/components/services/python/file.py b/templates/components/services/python/file.py index 9441db6e..d5267181 100644 --- a/templates/components/services/python/file.py +++ b/templates/components/services/python/file.py @@ -3,7 +3,7 @@ import os from io import BytesIO from pathlib import Path -from typing import Any, List, Tuple +from typing import List, Optional, Tuple from app.engine.index import IndexConfig, get_index from llama_index.core import VectorStoreIndex @@ -72,7 +72,12 @@ def store_and_parse_file(file_name, file_data, extension) -> List[Document]: return documents @staticmethod - def process_file(file_name: str, base64_content: str, params: Any) -> List[str]: + def process_file( + file_name: str, base64_content: str, params: Optional[dict] = None + ) -> List[str]: + if params is None: + params = {} + file_data, extension = PrivateFileService.preprocess_base64_file(base64_content) # Add the nodes to the index and persist it