sxandie commited on
Commit
fa3cfe7
·
verified ·
1 Parent(s): 679d697

Update src/upload_file.py

Browse files
Files changed (1) hide show
  1. src/upload_file.py +3 -3
src/upload_file.py CHANGED
@@ -14,7 +14,7 @@ class UploadFile:
14
  """
15
 
16
  @staticmethod
17
- def process_uploaded_files(files_dir: List, chatbot: List, rag_with_dropdown: str) -> Tuple:
18
  """
19
  Prepares and saves a VectorDB from uploaded files.
20
 
@@ -25,7 +25,7 @@ class UploadFile:
25
  Returns:
26
  Tuple: A tuple containing an empty string and the updated chatbot instance.
27
  """
28
- if rag_with_dropdown == "Upload docs to chat with:":
29
  prepare_vectordb_instance = PrepareVectorDB(data_directory=files_dir,
30
  persist_directory=APP_CONFIG.custom_persist_directory,
31
  embedding_model_engine=APP_CONFIG.embedding_model_engine,
@@ -36,5 +36,5 @@ class UploadFile:
36
  (" ", "Uploaded files are ready for querying."))
37
  else:
38
  chatbot.append(
39
- (" ", "If you want to upload your own PDF, please select 'rag_with' from the dropdown."))
40
  return "", chatbot
 
14
  """
15
 
16
  @staticmethod
17
+ def process_uploaded_files(files_dir: List, chatbot: List, llm = rag_with_dropdown: str) -> Tuple:
18
  """
19
  Prepares and saves a VectorDB from uploaded files.
20
 
 
25
  Returns:
26
  Tuple: A tuple containing an empty string and the updated chatbot instance.
27
  """
28
+ if rag_with_dropdown == "Upload new data":
29
  prepare_vectordb_instance = PrepareVectorDB(data_directory=files_dir,
30
  persist_directory=APP_CONFIG.custom_persist_directory,
31
  embedding_model_engine=APP_CONFIG.embedding_model_engine,
 
36
  (" ", "Uploaded files are ready for querying."))
37
  else:
38
  chatbot.append(
39
+ (" ", "If you want to upload your own PDF, please select 'Upload new data' from the dropdown."))
40
  return "", chatbot