refactor: 重构 Agent 模块为 AgentScope,删除旧版 CrewAI/LiteLLM 实现

This commit is contained in:
qzl
2026-03-11 20:51:56 +08:00
parent 177ed616bf
commit 145e3dc615
149 changed files with 5120 additions and 11356 deletions
@@ -97,6 +97,35 @@ class FakeRepo:
del data
return MagicMock()
async def list_subscribed_items_by_date_range(
self,
subscriber_id: UUID,
start_at: datetime,
end_at: datetime,
):
del subscriber_id, start_at, end_at
return []
async def get_user_subscriptions(self, subscriber_id: UUID):
del subscriber_id
return []
async def get_subscriptions_by_item_id(self, item_id: UUID):
del item_id
return []
async def get_subscription(self, item_id: UUID, subscriber_id: UUID):
del item_id, subscriber_id
return None
async def update_subscription_status(
self, item_id: UUID, subscriber_id: UUID, status
):
del item_id, subscriber_id, status
async def delete_subscriptions_by_item_id(self, item_id: UUID):
del item_id
@pytest.fixture
def mock_session() -> AsyncMock:
@@ -106,8 +135,15 @@ def mock_session() -> AsyncMock:
return session
@pytest.fixture
def mock_inbox_repository() -> MagicMock:
return MagicMock()
@pytest.mark.asyncio
async def test_create_success(mock_session: AsyncMock) -> None:
async def test_create_success(
mock_session: AsyncMock, mock_inbox_repository: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
request = ScheduleItemCreateRequest(
title="Test Event",
@@ -117,6 +153,7 @@ async def test_create_success(mock_session: AsyncMock) -> None:
repository=FakeRepo(None),
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=mock_inbox_repository,
)
result = await service.create(request)
@@ -126,7 +163,9 @@ async def test_create_success(mock_session: AsyncMock) -> None:
@pytest.mark.asyncio
async def test_create_invalid_end_at(mock_session: AsyncMock) -> None:
async def test_create_invalid_end_at(
mock_session: AsyncMock, mock_inbox_repository: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
request = ScheduleItemCreateRequest(
title="Test Event",
@@ -137,6 +176,7 @@ async def test_create_invalid_end_at(mock_session: AsyncMock) -> None:
repository=FakeRepo(None),
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=mock_inbox_repository,
)
with pytest.raises(HTTPException) as exc_info:
@@ -146,13 +186,16 @@ async def test_create_invalid_end_at(mock_session: AsyncMock) -> None:
@pytest.mark.asyncio
async def test_get_by_id_success(mock_session: AsyncMock) -> None:
async def test_get_by_id_success(
mock_session: AsyncMock, mock_inbox_repository: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
item = _create_mock_schedule_item()
service = ScheduleItemService(
repository=FakeRepo(item),
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=mock_inbox_repository,
)
result = await service.get_by_id(item.id)
@@ -161,12 +204,15 @@ async def test_get_by_id_success(mock_session: AsyncMock) -> None:
@pytest.mark.asyncio
async def test_get_by_id_not_found(mock_session: AsyncMock) -> None:
async def test_get_by_id_not_found(
mock_session: AsyncMock, mock_inbox_repository: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
service = ScheduleItemService(
repository=FakeRepo(None),
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=mock_inbox_repository,
)
with pytest.raises(HTTPException) as exc_info:
@@ -176,13 +222,16 @@ async def test_get_by_id_not_found(mock_session: AsyncMock) -> None:
@pytest.mark.asyncio
async def test_update_success(mock_session: AsyncMock) -> None:
async def test_update_success(
mock_session: AsyncMock, mock_inbox_repository: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
item = _create_mock_schedule_item()
service = ScheduleItemService(
repository=FakeRepo(item),
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=mock_inbox_repository,
)
result = await service.update(item.id, ScheduleItemUpdateRequest(title="Updated"))
@@ -191,13 +240,16 @@ async def test_update_success(mock_session: AsyncMock) -> None:
@pytest.mark.asyncio
async def test_delete_success(mock_session: AsyncMock) -> None:
async def test_delete_success(
mock_session: AsyncMock, mock_inbox_repository: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
item = _create_mock_schedule_item()
service = ScheduleItemService(
repository=FakeRepo(item),
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=mock_inbox_repository,
)
await service.delete(item.id)
@@ -206,7 +258,9 @@ async def test_delete_success(mock_session: AsyncMock) -> None:
@pytest.mark.asyncio
async def test_create_maps_metadata_to_extra_metadata(mock_session: AsyncMock) -> None:
async def test_create_maps_metadata_to_extra_metadata(
mock_session: AsyncMock, mock_inbox_repository: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
captured: dict | None = None
@@ -232,6 +286,7 @@ async def test_create_maps_metadata_to_extra_metadata(mock_session: AsyncMock) -
repository=CaptureRepo(None),
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=mock_inbox_repository,
)
await service.create(request)
@@ -244,7 +299,9 @@ async def test_create_maps_metadata_to_extra_metadata(mock_session: AsyncMock) -
@pytest.mark.asyncio
async def test_update_maps_metadata_to_extra_metadata(mock_session: AsyncMock) -> None:
async def test_update_maps_metadata_to_extra_metadata(
mock_session: AsyncMock, mock_inbox_repository: MagicMock
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
item = _create_mock_schedule_item()
captured: dict | None = None
@@ -261,6 +318,7 @@ async def test_update_maps_metadata_to_extra_metadata(mock_session: AsyncMock) -
repository=CaptureRepo(item),
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=mock_inbox_repository,
)
await service.update(
@@ -285,6 +343,7 @@ async def test_update_maps_metadata_to_extra_metadata(mock_session: AsyncMock) -
@pytest.mark.asyncio
async def test_update_maps_null_metadata_to_extra_metadata_null(
mock_session: AsyncMock,
mock_inbox_repository: MagicMock,
) -> None:
user_id = UUID("00000000-0000-0000-0000-000000000001")
item = _create_mock_schedule_item()
@@ -302,6 +361,7 @@ async def test_update_maps_null_metadata_to_extra_metadata_null(
repository=CaptureRepo(item),
session=mock_session,
current_user=CurrentUser(id=user_id),
inbox_repository=mock_inbox_repository,
)
await service.update(