chore: no changes needed for calendar message card

2026-03-11 21:06:02 +08:00
parent 98f22a2127
commit a8dacbe81f
8 changed files with 590 additions and 0 deletions
@@ -100,3 +100,105 @@ async def test_agent_sse_closed_loop_live() -> None:
            )
        )
        assert len(list(rows.scalars().all())) >= 1
+
+
+@pytest.mark.asyncio
+@pytest.mark.live
+async def test_agent_runs_events_history_live_with_image_input() -> None:
+    if os.getenv("AGENT_LIVE_INTEGRATION") != "1":
+        pytest.skip("set AGENT_LIVE_INTEGRATION=1 to run live integration test")
+
+    async with httpx.AsyncClient(timeout=30.0) as client:
+        token = await _live_access_token(client)
+        headers = {"Authorization": f"Bearer {token}"}
+        thread_id = str(uuid4())
+
+        run_resp = await client.post(
+            f"{BASE_URL}/api/v1/agent/runs",
+            headers=headers,
+            json={
+                "threadId": thread_id,
+                "runId": "run-live-image-1",
+                "state": {},
+                "messages": [
+                    {
+                        "id": "u1",
+                        "role": "user",
+                        "content": [
+                            {"type": "text", "text": "请描述图片里的内容"},
+                            {
+                                "type": "binary",
+                                "data": "aGVsbG8=",
+                                "mimeType": "image/png",
+                            },
+                        ],
+                    }
+                ],
+                "tools": [],
+                "context": [],
+                "forwardedProps": {},
+            },
+        )
+        assert run_resp.status_code == 202
+
+        events_url = f"{BASE_URL}/api/v1/agent/runs/{thread_id}/events"
+        event_names: list[str] = []
+        async with client.stream(
+            "GET", events_url, headers=headers, timeout=20.0
+        ) as sse_resp:
+            assert sse_resp.status_code == 200
+            assert sse_resp.headers.get("content-type", "").startswith(
+                "text/event-stream"
+            )
+            async for line in sse_resp.aiter_lines():
+                if line.startswith("event:"):
+                    event_name = line.split(":", 1)[1].strip()
+                    event_names.append(event_name)
+                    if event_name in {"RUN_FINISHED", "RUN_ERROR"}:
+                        break
+
+        assert "RUN_STARTED" in event_names
+        assert "RUN_FINISHED" in event_names or "RUN_ERROR" in event_names
+
+        history_resp = await client.get(
+            f"{BASE_URL}/api/v1/agent/runs/{thread_id}/history",
+            headers=headers,
+        )
+        assert history_resp.status_code == 200
+        history = history_resp.json()
+        assert history.get("type") == "STATE_SNAPSHOT"
+        snapshot = history.get("snapshot", {})
+        assert snapshot.get("scope") == "history_day"
+        messages = snapshot.get("messages", [])
+        user_messages = [
+            item
+            for item in messages
+            if isinstance(item, dict) and item.get("role") == "user"
+        ]
+        assert user_messages
+        attachments = user_messages[0].get("attachments")
+        assert isinstance(attachments, list)
+        assert attachments and isinstance(attachments[0], dict)
+        assert isinstance(attachments[0].get("path"), str)
+
+    async with AsyncSessionLocal() as session:
+        session_row = await session.get(AgentChatSession, UUID(thread_id))
+        assert session_row is not None
+        assert session_row.message_count >= 1
+        assert session_row.total_tokens >= 0
+        assert session_row.total_cost >= 0
+
+        rows = await session.execute(
+            select(AgentChatMessage).where(
+                AgentChatMessage.session_id == UUID(thread_id)
+            )
+        )
+        all_messages = list(rows.scalars().all())
+        assert all_messages
+        user_rows = [row for row in all_messages if str(row.role) == "user"]
+        assert user_rows
+        metadata = user_rows[0].metadata_json or {}
+        attachments = metadata.get("attachments")
+        assert isinstance(attachments, list)
+        assert attachments and isinstance(attachments[0], dict)
+        assert isinstance(attachments[0].get("path"), str)