feat(agent-chat): complete core workflow and strengthen auth rate limiting

2026-02-25 16:51:12 +08:00
parent 53c72e48e6
commit cd40b2b4f4
62 changed files with 3441 additions and 3 deletions
@@ -0,0 +1,38 @@
+from __future__ import annotations
+
+from datetime import datetime, timezone
+from decimal import Decimal
+from uuid import UUID
+
+from models.agent_chat_session import AgentChatSession, AgentChatSessionStatus
+from v1.agent_chat.service import select_recent_session
+
+
+def test_recent_session_home_default_selection() -> None:
+    sessions = [
+        AgentChatSession(
+            id=UUID("00000000-0000-0000-0000-0000000000a1"),
+            user_id=UUID("00000000-0000-0000-0000-0000000000c1"),
+            title="older",
+            status=AgentChatSessionStatus.COMPLETED,
+            last_activity_at=datetime(2026, 2, 25, 8, 0, tzinfo=timezone.utc),
+            message_count=2,
+            total_tokens=100,
+            total_cost=Decimal("0.010000"),
+        ),
+        AgentChatSession(
+            id=UUID("00000000-0000-0000-0000-0000000000a2"),
+            user_id=UUID("00000000-0000-0000-0000-0000000000c1"),
+            title="newer",
+            status=AgentChatSessionStatus.RUNNING,
+            last_activity_at=datetime(2026, 2, 25, 9, 0, tzinfo=timezone.utc),
+            message_count=3,
+            total_tokens=120,
+            total_cost=Decimal("0.020000"),
+        ),
+    ]
+
+    selected = select_recent_session(sessions)
+
+    assert selected is not None
+    assert selected.id == UUID("00000000-0000-0000-0000-0000000000a2")