62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
测试在 D:\\aaa\\aiagent\\user_data 下创建 aaa.md（与线上一致的 file_write_tool）。
|
||
|
||
用法（在 backend 目录）:
|
||
.\\venv\\Scripts\\python.exe scripts\\test_write_user_data_aaa_md.py
|
||
|
||
可选环境变量:
|
||
TEST_MD_CONTENT 写入内容，默认一行时间戳
|
||
TEST_USE_ABSPATH 设为 1 时使用绝对路径 D:\\aaa\\aiagent\\user_data\\aaa.md，否则用相对路径 user_data/aaa.md
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import asyncio
|
||
import os
|
||
import sys
|
||
from datetime import datetime, timezone
|
||
from pathlib import Path
|
||
|
||
BACKEND_DIR = Path(__file__).resolve().parents[1]
|
||
|
||
|
||
def main() -> int:
|
||
os.chdir(BACKEND_DIR)
|
||
sys.path.insert(0, str(BACKEND_DIR))
|
||
|
||
from app.services.builtin_tools import _local_file_workspace_root, file_write_tool
|
||
|
||
root = _local_file_workspace_root()
|
||
content = os.environ.get(
|
||
"TEST_MD_CONTENT",
|
||
f"# aaa\\n\\nwritten by test_write_user_data_aaa_md.py at {datetime.now(timezone.utc).isoformat()}\\n",
|
||
).replace("\\n", "\n")
|
||
|
||
if os.environ.get("TEST_USE_ABSPATH", "").strip() in ("1", "true", "yes"):
|
||
target = root / "user_data" / "aaa.md"
|
||
file_path_arg = str(target)
|
||
else:
|
||
file_path_arg = "user_data/aaa.md"
|
||
|
||
print("workspace root:", root)
|
||
print("file_path:", file_path_arg)
|
||
print("content bytes (utf-8):", len(content.encode("utf-8")))
|
||
|
||
raw = asyncio.run(file_write_tool(file_path_arg, content, "w"))
|
||
print("tool return:", raw)
|
||
|
||
resolved = (root / "user_data" / "aaa.md").resolve()
|
||
if not resolved.is_file():
|
||
print("FAIL: file missing:", resolved, file=sys.stderr)
|
||
return 2
|
||
on_disk = resolved.read_text(encoding="utf-8", errors="replace")
|
||
print("OK: on disk", resolved)
|
||
print("--- file head ---")
|
||
print(on_disk[:500])
|
||
print("---")
|
||
return 0
|
||
|
||
|
||
if __name__ == "__main__":
|
||
raise SystemExit(main())
|