filename fix
This commit is contained in:
parent
cf27dfddeb
commit
97a17c77d7
3 changed files with 10 additions and 5 deletions
|
|
@ -194,9 +194,9 @@ async def upload_and_ingest_user_file(
|
||||||
content = await upload_file.read()
|
content = await upload_file.read()
|
||||||
|
|
||||||
# Create temporary file
|
# Create temporary file
|
||||||
|
safe_filename = upload_file.filename.replace(" ", "_").replace("/", "_")
|
||||||
temp_fd, temp_path = tempfile.mkstemp(
|
temp_fd, temp_path = tempfile.mkstemp(
|
||||||
suffix=f"_{upload_file.filename}",
|
suffix=f"_{safe_filename}"
|
||||||
prefix="langflow_upload_"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
|
|
||||||
|
|
@ -119,9 +119,9 @@ async def langflow_upload_ingest_task(
|
||||||
content = await upload_file.read()
|
content = await upload_file.read()
|
||||||
|
|
||||||
# Create temporary file
|
# Create temporary file
|
||||||
|
safe_filename = upload_file.filename.replace(" ", "_").replace("/", "_")
|
||||||
temp_fd, temp_path = tempfile.mkstemp(
|
temp_fd, temp_path = tempfile.mkstemp(
|
||||||
suffix=f"_{upload_file.filename}",
|
suffix=f"_{safe_filename}"
|
||||||
prefix="langflow_upload_"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
# Write content to temp file
|
# Write content to temp file
|
||||||
|
|
|
||||||
|
|
@ -371,7 +371,12 @@ class LangflowFileProcessor(TaskProcessor):
|
||||||
content = f.read()
|
content = f.read()
|
||||||
|
|
||||||
# Create file tuple for upload
|
# Create file tuple for upload
|
||||||
filename = os.path.basename(item)
|
temp_filename = os.path.basename(item)
|
||||||
|
# Extract original filename from temp file suffix (remove tmp prefix)
|
||||||
|
if "_" in temp_filename:
|
||||||
|
filename = temp_filename.split("_", 1)[1] # Get everything after first _
|
||||||
|
else:
|
||||||
|
filename = temp_filename
|
||||||
content_type, _ = mimetypes.guess_type(filename)
|
content_type, _ = mimetypes.guess_type(filename)
|
||||||
if not content_type:
|
if not content_type:
|
||||||
content_type = 'application/octet-stream'
|
content_type = 'application/octet-stream'
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue