vrr/agent-zero
2
0
Fork 0
mirror of https://github.com/agent0ai/agent-zero.git synced 2026-04-29 20:20:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

better filename check

Browse source
This commit is contained in:
linkliti 2026-01-24 21:54:21 +03:00
parent c4846e7e9d
commit e5669daf67
9 changed files with 75 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

6
python/api/api_message.py
View file

@ -6,7 +6,7 @@ from python.helpers.api import ApiHandler, Request, Response
from python.helpers import files, projects
from python.helpers.print_style import PrintStyle
from python.helpers.projects import activate_project
from werkzeug.utils import secure_filename
from python.helpers.security import safe_filename
from initialize import initialize_agent
import threading
@ -57,9 +57,9 @@ class ApiMessage(ApiHandler):
continue
try:
filename = secure_filename(attachment["filename"])
filename = safe_filename(attachment["filename"])
if not filename:
continue
raise ValueError("Invalid filename")
# Decode base64 content
file_content = base64.b64decode(attachment["base64"])