chenmingxuan
e1ea099592
Install and update artemis-use with task diagnostics and plugin guidance. Preserve cross-device file receipts, review canonical reports, guard stage transitions and reduce small history summaries. Improve exploration lifecycle, request recovery and browser live connections.
65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
"""Minimal task-owned attachment server. Run through ARTEMIS, not as a shell tool."""
|
|
|
|
import hashlib
|
|
import os
|
|
from pathlib import Path
|
|
import time
|
|
import uuid
|
|
|
|
from mcp.server.fastmcp import FastMCP
|
|
from mcp.types import CallToolResult, TextContent, ToolAnnotations
|
|
|
|
mcp = FastMCP("evidence-demo")
|
|
|
|
|
|
@mcp.tool(
|
|
annotations=ToolAnnotations(readOnlyHint=False, destructiveHint=False, idempotentHint=False)
|
|
)
|
|
def write_note(text: str) -> CallToolResult:
|
|
"""Save at most 16 KiB of user-requested text as a new task attachment."""
|
|
encoded = text.encode("utf-8")
|
|
root = os.environ.get("ARTEMIS_ARTIFACTS_DIR")
|
|
if not root or len(encoded) > 16384:
|
|
return CallToolResult(
|
|
isError=True,
|
|
content=[TextContent(type="text", text="任务目录缺失或内容超过 16 KiB；未写文件。")],
|
|
structuredContent={
|
|
"artemis": {
|
|
"version": 1,
|
|
"execution": "not_started",
|
|
"side_effect": "none",
|
|
"error_code": "invalid_input",
|
|
"retryable": False,
|
|
}
|
|
},
|
|
)
|
|
directory = Path(root)
|
|
directory.mkdir(parents=True, exist_ok=True)
|
|
name = f"note-{uuid.uuid4().hex}.txt"
|
|
path = directory / name
|
|
with path.open("xb") as handle:
|
|
handle.write(encoded)
|
|
digest = hashlib.sha256(path.read_bytes()).hexdigest()
|
|
return CallToolResult(
|
|
content=[TextContent(type="text", text=f"已保存并读回校验：{name}")],
|
|
structuredContent={
|
|
"artemis": {
|
|
"version": 1,
|
|
"execution": "completed",
|
|
"side_effect": "confirmed",
|
|
"retryable": False,
|
|
"source": {
|
|
"kind": "file",
|
|
"target_id": "task-artifacts",
|
|
"label": "任务文件",
|
|
"captured_at": time.time(),
|
|
},
|
|
"attachments": [{"name": name, "sha256": digest, "media_type": "text/plain"}],
|
|
}
|
|
},
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
mcp.run(transport="stdio")
|