ab268d8eba
- 更新 .gitignore 以排除测试相关目录。 - 增强 ChatbotAgent 和 DeepAgent 以利用 BaseState 进行状态管理。 - 引入 PresentArtifacts 工具，允许代理向用户展示输出文件。 - 在 state.py 中添加新的状态管理功能，包括 merge_artifacts 和 BaseState。 - 创建 AgentArtifactsCard 组件以在前端显示工件。 - 在 AgentChatComponent 和 AgentInputArea 中集成工件处理。 - 更新 MessageInputComponent 和 ToolCallRenderer 以更好地处理工件的 UI 展示。 - 添加工件状态管理和标准化函数的测试。 - 更新 docker-compose 以使用最新的沙箱配置器镜像。 - 在路线图中记录新功能。
78 lines
2.7 KiB
Python
78 lines
2.7 KiB
Python
from yuxi.agents.backends.sandbox import (
|
|
VIRTUAL_PATH_PREFIX,
|
|
ensure_thread_dirs,
|
|
sandbox_outputs_dir,
|
|
sandbox_uploads_dir,
|
|
)
|
|
from yuxi.agents.state import merge_artifacts
|
|
from yuxi.agents.toolkits.buildin.tools import _normalize_presented_artifact_path
|
|
from yuxi.services.chat_service import extract_agent_state
|
|
|
|
|
|
def _runtime_with_thread(thread_id: str):
|
|
context = type("RuntimeContext", (), {"thread_id": thread_id})()
|
|
return type("RuntimeStub", (), {"context": context})()
|
|
|
|
|
|
def test_merge_artifacts_deduplicates_and_preserves_order():
|
|
assert merge_artifacts(
|
|
["/home/gem/user-data/outputs/a.md"],
|
|
["/home/gem/user-data/outputs/a.md", "/home/gem/user-data/outputs/b.md"],
|
|
) == [
|
|
"/home/gem/user-data/outputs/a.md",
|
|
"/home/gem/user-data/outputs/b.md",
|
|
]
|
|
|
|
|
|
def test_normalize_presented_artifact_path_accepts_host_path():
|
|
thread_id = "artifacts-host-path"
|
|
ensure_thread_dirs(thread_id)
|
|
output_file = sandbox_outputs_dir(thread_id) / "report.md"
|
|
output_file.write_text("# demo", encoding="utf-8")
|
|
|
|
normalized = _normalize_presented_artifact_path(str(output_file), _runtime_with_thread(thread_id))
|
|
|
|
assert normalized == f"{VIRTUAL_PATH_PREFIX}/outputs/report.md"
|
|
|
|
|
|
def test_normalize_presented_artifact_path_accepts_virtual_path():
|
|
thread_id = "artifacts-virtual-path"
|
|
ensure_thread_dirs(thread_id)
|
|
output_file = sandbox_outputs_dir(thread_id) / "summary.txt"
|
|
output_file.write_text("demo", encoding="utf-8")
|
|
|
|
normalized = _normalize_presented_artifact_path(
|
|
f"{VIRTUAL_PATH_PREFIX}/outputs/summary.txt",
|
|
_runtime_with_thread(thread_id),
|
|
)
|
|
|
|
assert normalized == f"{VIRTUAL_PATH_PREFIX}/outputs/summary.txt"
|
|
|
|
|
|
def test_normalize_presented_artifact_path_rejects_non_outputs_path():
|
|
thread_id = "artifacts-reject-path"
|
|
ensure_thread_dirs(thread_id)
|
|
upload_file = sandbox_uploads_dir(thread_id) / "note.txt"
|
|
upload_file.write_text("demo", encoding="utf-8")
|
|
|
|
try:
|
|
_normalize_presented_artifact_path(str(upload_file), _runtime_with_thread(thread_id))
|
|
except ValueError as exc:
|
|
assert f"{VIRTUAL_PATH_PREFIX}/outputs/" in str(exc)
|
|
else:
|
|
raise AssertionError("expected ValueError for non-outputs file")
|
|
|
|
|
|
def test_extract_agent_state_includes_artifacts():
|
|
state = extract_agent_state(
|
|
{
|
|
"todos": [{"content": "done", "status": "completed"}],
|
|
"files": {"/tmp/demo.txt": {"content": ["x"]}},
|
|
"artifacts": ["/home/gem/user-data/outputs/demo.txt"],
|
|
}
|
|
)
|
|
|
|
assert state["todos"] == [{"content": "done", "status": "completed"}]
|
|
assert state["files"] == {"/tmp/demo.txt": {"content": ["x"]}}
|
|
assert state["artifacts"] == ["/home/gem/user-data/outputs/demo.txt"]
|