refactor: 简化 AgentScope 运行时模块与 prompt 系统

2026-03-15 17:14:15 +08:00
parent 61997f3613
commit 072c09d99d
32 changed files with 750 additions and 1863 deletions
@@ -1,123 +0,0 @@
-from __future__ import annotations
-
-from types import SimpleNamespace
-from typing import Any, cast
-from uuid import uuid4
-
-from ag_ui.core import RunAgentInput
-from fastapi import HTTPException
-import pytest
-
-from core.auth.models import CurrentUser
-from v1.agent import router as agent_router
-
-
-def _resume_input_with_tool_message() -> RunAgentInput:
-    return RunAgentInput.model_validate(
-        {
-            "threadId": "00000000-0000-0000-0000-000000000001",
-            "runId": "run-resume-1",
-            "state": {},
-            "messages": [
-                {
-                    "id": "tool-1",
-                    "role": "tool",
-                    "toolCallId": "call-1",
-                    "content": '{"toolName":"navigate_to_route","result":{"ok":true}}',
-                }
-            ],
-            "tools": [],
-            "context": [],
-            "forwardedProps": {},
-        }
-    )
-
-
-@pytest.mark.asyncio
-async def test_enqueue_resume_rejects_without_tool_contract() -> None:
-    request = RunAgentInput.model_validate(
-        {
-            "threadId": "00000000-0000-0000-0000-000000000001",
-            "runId": "run-resume-invalid",
-            "state": {},
-            "messages": [{"id": "u1", "role": "user", "content": "continue"}],
-            "tools": [],
-            "context": [],
-            "forwardedProps": {},
-        }
-    )
-
-    class _Service:
-        async def enqueue_resume(self, **kwargs):  # noqa: ANN003
-            del kwargs
-            raise AssertionError("enqueue_resume should not be called")
-
-    with pytest.raises(HTTPException) as exc_info:
-        await agent_router.enqueue_resume(
-            thread_id="00000000-0000-0000-0000-000000000001",
-            request=request,
-            service=cast(Any, _Service()),
-            current_user=CurrentUser(id=uuid4(), email="user@example.com"),
-        )
-
-    assert exc_info.value.status_code == 422
-
-
-@pytest.mark.asyncio
-async def test_enqueue_resume_rejects_when_rate_limited(
-    monkeypatch: pytest.MonkeyPatch,
-) -> None:
-    request = _resume_input_with_tool_message()
-
-    async def _deny_run(*, user_id: str) -> bool:
-        del user_id
-        return False
-
-    monkeypatch.setattr(agent_router, "_allow_run_request", _deny_run)
-
-    class _Service:
-        async def enqueue_resume(self, **kwargs):  # noqa: ANN003
-            del kwargs
-            raise AssertionError("enqueue_resume should not be called")
-
-    with pytest.raises(HTTPException) as exc_info:
-        await agent_router.enqueue_resume(
-            thread_id="00000000-0000-0000-0000-000000000001",
-            request=request,
-            service=cast(Any, _Service()),
-            current_user=CurrentUser(id=uuid4(), email="user@example.com"),
-        )
-
-    assert exc_info.value.status_code == 429
-
-
-@pytest.mark.asyncio
-async def test_enqueue_resume_accepts_valid_tool_contract(
-    monkeypatch: pytest.MonkeyPatch,
-) -> None:
-    request = _resume_input_with_tool_message()
-
-    async def _allow_run(*, user_id: str) -> bool:
-        del user_id
-        return True
-
-    monkeypatch.setattr(agent_router, "_allow_run_request", _allow_run)
-
-    class _Service:
-        async def enqueue_resume(self, **kwargs):  # noqa: ANN003
-            return SimpleNamespace(
-                task_id="task-resume-1",
-                thread_id=kwargs["thread_id"],
-                run_id=kwargs["run_input"].run_id,
-                created=False,
-            )
-
-    result = await agent_router.enqueue_resume(
-        thread_id="00000000-0000-0000-0000-000000000001",
-        request=request,
-        service=cast(Any, _Service()),
-        current_user=CurrentUser(id=uuid4(), email="user@example.com"),
-    )
-
-    assert result.task_id == "task-resume-1"
-    assert result.run_id == "run-resume-1"