41 lines
1.6 KiB
Python
41 lines
1.6 KiB
Python
"""子智能体特有工具实现（read_mediafile）。
|
||
|
||
read_mediafile 直接读取项目内媒体文件并返回 base64。
|
||
（原 search_workspace 工具已移除：子智能体需要搜索时改用 run_command
|
||
执行 rg / grep / find，走主进程沙箱链路，避免进程内遍历大目录导致内存失控。）
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import base64
|
||
import mimetypes
|
||
from pathlib import Path
|
||
from typing import Any, Dict
|
||
|
||
|
||
async def handle_read_mediafile(project_path: Path, arguments: Dict[str, Any]) -> Dict[str, Any]:
|
||
"""read_mediafile 实现：直接读取项目内媒体文件并返回 base64。"""
|
||
path = arguments.get("path")
|
||
if not path:
|
||
return {"success": False, "error": "path 不能为空"}
|
||
try:
|
||
abs_path = (project_path / path).resolve()
|
||
abs_path.relative_to(project_path)
|
||
except Exception:
|
||
return {"success": False, "error": "非法路径，超出项目根目录"}
|
||
|
||
if not abs_path.exists() or not abs_path.is_file():
|
||
return {"success": False, "error": f"文件不存在: {path}"}
|
||
|
||
mime, _ = mimetypes.guess_type(str(abs_path))
|
||
if not mime or (not mime.startswith("image/") and not mime.startswith("video/")):
|
||
return {"success": False, "error": "禁止的文件类型"}
|
||
|
||
try:
|
||
data = abs_path.read_bytes()
|
||
b64 = base64.b64encode(data).decode("utf-8")
|
||
file_type = "image" if mime.startswith("image/") else "video"
|
||
return {"success": True, "path": path, "mime": mime, "type": file_type, "b64": b64}
|
||
except Exception as exc:
|
||
return {"success": False, "error": str(exc)}
|