qzl/social-app
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: 移除 crewai agent 架构相关代码并更新 LLM 配置

Browse Source
This commit is contained in:
qzl
2026-03-04 11:37:09 +08:00
parent 87399f74c8
commit b02a322bf3
71 changed files with 1045 additions and 7499 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/tests/integration/v1/agent/__init__.py
View File
backend/tests/integration/v1/agent/test_chat_routes.py
-82
View File
@@ -1,82 +0,0 @@
from __future__ import annotations
from uuid import UUID
import pytest
from fastapi.testclient import TestClient
from app import app
from core.auth.models import CurrentUser
from v1.agent.dependencies import get_agent_service
from v1.agent.schemas import RunAgentInput
from v1.users.dependencies import get_current_user
class FakeAgentService:
async def prepare_resume(self, run_id: str, input_data: RunAgentInput):
return None
async def stream_run(self, input_data: RunAgentInput):
yield 'data: {"type": "RUN_STARTED", "runId": "r1"}\n\n'
yield 'data: {"type": "TEXT_MESSAGE_START", "messageId": "m1"}\n\n'
yield 'data: {"type": "TEXT_MESSAGE_CONTENT", "delta": "Hello"}\n\n'
yield 'data: {"type": "TEXT_MESSAGE_END", "messageId": "m1"}\n\n'
yield 'data: {"type": "RUN_FINISHED", "runId": "r1"}\n\n'
async def stream_resume(self, run_id: str, input_data: RunAgentInput):
yield 'data: {"type": "RUN_STARTED", "runId": "r1"}\n\n'
yield 'data: {"type": "TEXT_MESSAGE_START", "messageId": "m2"}\n\n'
yield 'data: {"type": "TEXT_MESSAGE_CONTENT", "delta": "Resumed"}\n\n'
yield 'data: {"type": "TEXT_MESSAGE_END", "messageId": "m2"}\n\n'
yield 'data: {"type": "RUN_FINISHED", "runId": "r1"}\n\n'
def _get_test_user() -> CurrentUser:
return CurrentUser(id=UUID("00000000-0000-0000-0000-000000000001"))
@pytest.fixture
def client() -> TestClient:
app.dependency_overrides[get_current_user] = _get_test_user
app.dependency_overrides[get_agent_service] = lambda: FakeAgentService()
yield TestClient(app)
app.dependency_overrides.clear()
class TestChatRoutes:
def test_run_route_streams_sse_events(self, client: TestClient):
payload = {
"threadId": "t1",
"runId": "r1",
"state": {},
"messages": [],
"tools": [],
"context": [],
"forwardedProps": {},
}
response = client.post("/api/v1/agent/runs", json=payload)
assert response.status_code == 200
assert response.headers["content-type"] == "text/event-stream; charset=utf-8"
events = response.text.split("\n\n")
assert 'data: {"type": "RUN_STARTED"' in events[0]
assert 'data: {"type": "TEXT_MESSAGE_START"' in events[1]
def test_resume_route_streams_sse_events(self, client: TestClient):
payload = {
"threadId": "t1",
"runId": "r1",
"state": {},
"messages": [],
"tools": [],
"context": [],
"forwardedProps": {},
"resume": {"interruptId": "int-1", "payload": {"decision": "approved"}},
}
response = client.post("/api/v1/agent/runs/r1/resume", json=payload)
assert response.status_code == 200
assert response.headers["content-type"] == "text/event-stream; charset=utf-8"
events = response.text.split("\n\n")
assert 'data: {"type": "RUN_STARTED"' in events[0]
assert 'data: {"type": "TEXT_MESSAGE_CONTENT", "delta": "Resumed"' in events[2]
backend/tests/integration/v1/agent/test_interrupt_resume_flow.py
-144
View File
@@ -1,144 +0,0 @@
from __future__ import annotations
import json
from uuid import UUID
import pytest
from fastapi.testclient import TestClient
from app import app
from core.auth.models import CurrentUser
from v1.agent.dependencies import get_agent_service
from v1.agent.schemas import RunAgentInput
from v1.users.dependencies import get_current_user
class FakeAgentServiceWithInterrupt:
async def prepare_resume(self, run_id: str, input_data: RunAgentInput):
return None
async def stream_run(self, input_data: RunAgentInput):
yield 'data: {"type": "RUN_STARTED", "runId": "' + input_data.runId + '"}\n\n'
yield 'data: {"type": "TEXT_MESSAGE_START", "messageId": "m1"}\n\n'
yield 'data: {"type": "TEXT_MESSAGE_CONTENT", "delta": "Let me navigate"}\n\n'
yield 'data: {"type": "TOOL_CALL", "toolName": "ui.navigate_to", "args": {"path": "/home"}}\n\n'
yield (
'data: {"type": "RUN_FINISHED", "runId": "'
+ input_data.runId
+ '", "outcome": "interrupt", "interrupt": {"id": "int-1", "reason": "frontend_tool", "payload": {"toolName": "ui.navigate_to", "args": {"path": "/home"}}}}\n\n'
)
async def stream_resume(self, run_id: str, input_data: RunAgentInput):
if input_data.resume and input_data.resume.get("interruptId") == "int-1":
payload = input_data.resume.get("payload", {})
yield 'data: {"type": "RUN_STARTED", "runId": "' + run_id + '"}\n\n'
yield (
'data: {"type": "TOOL_RESULT", "toolName": "ui.navigate_to", "result": '
+ json.dumps(payload.get("result", {}))
+ "}\n\n"
)
yield 'data: {"type": "TEXT_MESSAGE_START", "messageId": "m2"}\n\n'
yield 'data: {"type": "TEXT_MESSAGE_CONTENT", "delta": "Navigation completed"}\n\n'
yield 'data: {"type": "RUN_FINISHED", "runId": "' + run_id + '"}\n\n'
else:
yield (
'data: {"type": "RUN_FINISHED", "runId": "'
+ run_id
+ '", "outcome": "error"}\n\n'
)
def _get_test_user() -> CurrentUser:
return CurrentUser(id=UUID("00000000-0000-0000-0000-000000000001"))
@pytest.fixture
def client() -> TestClient:
app.dependency_overrides[get_current_user] = _get_test_user
app.dependency_overrides[get_agent_service] = (
lambda: FakeAgentServiceWithInterrupt()
)
yield TestClient(app)
app.dependency_overrides.clear()
class TestInterruptResumeFlow:
def test_frontend_tool_interrupt_then_resume_with_result(self, client: TestClient):
payload = {
"threadId": "t1",
"runId": "r1",
"state": {},
"messages": [{"role": "user", "content": "Navigate to home"}],
"tools": [{"name": "ui.navigate_to", "execution_target": "frontend"}],
"context": [],
"forwardedProps": {},
}
response = client.post("/api/v1/agent/runs", json=payload)
assert response.status_code == 200
events = response.text.split("\n\n")
interrupt_event = [e for e in events if '"outcome": "interrupt"' in e][0]
assert '"id": "int-1"' in interrupt_event
assert '"reason": "frontend_tool"' in interrupt_event
resume_payload = {
"threadId": "t1",
"runId": "r1",
"state": {},
"messages": [],
"tools": [],
"context": [],
"forwardedProps": {},
"resume": {
"interruptId": "int-1",
"payload": {"result": {"success": True}},
},
}
resume_response = client.post(
"/api/v1/agent/runs/r1/resume", json=resume_payload
)
assert resume_response.status_code == 200
resume_events = resume_response.text.split("\n\n")
tool_result_event = [e for e in resume_events if '"type": "TOOL_RESULT"' in e][
0
]
assert '"toolName": "ui.navigate_to"' in tool_result_event
assert '"success": true' in tool_result_event.lower()
def test_backend_tool_approval_rejected(self, client: TestClient):
payload = {
"threadId": "t2",
"runId": "r2",
"state": {},
"messages": [{"role": "user", "content": "Transfer funds"}],
"tools": [
{
"name": "srv.transfer_funds",
"execution_target": "backend",
"requires_approval": True,
}
],
"context": [],
"forwardedProps": {},
}
response = client.post("/api/v1/agent/runs", json=payload)
assert response.status_code == 200
resume_payload = {
"threadId": "t2",
"runId": "r2",
"state": {},
"messages": [],
"tools": [],
"context": [],
"forwardedProps": {},
"resume": {
"interruptId": "int-1",
"payload": {"decision": "rejected", "reason": "User denied"},
},
}
resume_response = client.post(
"/api/v1/agent/runs/r2/resume", json=resume_payload
)
assert resume_response.status_code == 200