Daniel Marques commited on
Commit
6318d9a
1 Parent(s): d5863e7

feat: add message tamplate

Browse files
Files changed (1) hide show
  1. main.py +1 -2
main.py CHANGED
@@ -23,7 +23,6 @@ from prompt_template_utils import get_prompt_template
23
 
24
  # from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
25
  from langchain.vectorstores import Chroma
26
- from werkzeug.utils import secure_filename
27
 
28
  from constants import CHROMA_SETTINGS, EMBEDDING_MODEL_NAME, PERSIST_DIRECTORY, MODEL_ID, MODEL_BASENAME
29
 
@@ -164,7 +163,7 @@ async def create_upload_file(file: Union[UploadFile, None] = None):
164
  if file.filename == "":
165
  raise HTTPException(status_code=400, detail="No selected file")
166
  if file:
167
- filename = secure_filename(file.filename)
168
  folder_path = "SOURCE_DOCUMENTS"
169
 
170
  if not os.path.exists(folder_path):
 
23
 
24
  # from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
25
  from langchain.vectorstores import Chroma
 
26
 
27
  from constants import CHROMA_SETTINGS, EMBEDDING_MODEL_NAME, PERSIST_DIRECTORY, MODEL_ID, MODEL_BASENAME
28
 
 
163
  if file.filename == "":
164
  raise HTTPException(status_code=400, detail="No selected file")
165
  if file:
166
+ filename = file.filename
167
  folder_path = "SOURCE_DOCUMENTS"
168
 
169
  if not os.path.exists(folder_path):