fix: stabilize MindSpace local workflows

Signed-off-by: john <cynell@139.com>

scripts/mindspace-agent-jobs-e2e.mjs

@@ -333,6 +333,7 @@ try {
 
   const finalJob = await pollJob(owner.cookie, jobId);
   assert.equal(finalJob.status, 'completed', JSON.stringify(finalJob));
+  assert.ok(finalJob.sessionId, '任务完成后应关联 Agent 会话');
   assert.ok(finalJob.resultPageId, '任务完成后应生成页面草稿');
 
   const pageDetail = await request(`/api/mindspace/v1/pages/${finalJob.resultPageId}`, {
@@ -348,11 +349,14 @@ try {
   assert.equal(draftPages.response.status, 200, JSON.stringify(draftPages.body));
   assert.equal(draftPages.body.data.length, 1);
 
-  const [usageRows] = await pool.query(
-    `SELECT COUNT(*) AS count FROM h5_usage_records WHERE user_id = ?`,
-    [owner.id],
+  const [billingRows] = await pool.query(
+    `SELECT last_input_tokens, last_output_tokens
+     FROM h5_session_billing_state
+     WHERE agent_session_id = ? AND user_id = ?`,
+    [finalJob.sessionId, owner.id],
   );
-  assert.equal(Number(usageRows[0]?.count ?? 0), 1, 'Agent 运行后应记录一次扣费');
+  assert.equal(Number(billingRows[0]?.last_input_tokens ?? 0), 12, 'Agent 运行后应记录输入 Token');
+  assert.equal(Number(billingRows[0]?.last_output_tokens ?? 0), 20, 'Agent 运行后应记录输出 Token');
 
   console.log('MindSpace agent jobs E2E passed');
 } finally {