@downcity/agent 1.1.217 → 1.1.224

package/src/session/runtime/SessionPromptRuntime.ts

@@ -13,7 +13,7 @@ import type {
   SessionRecordV1,
   SessionMessageRecordV1,
 } from "@/executor/types/SessionRecords.js";
-import type { AgentSessionEvent } from "@/types/sdk/AgentSessionEvent.js";
+import type { SessionMutation } from "@/types/session/SessionMutation.js";
 import type { AgentSessionPromptInput } from "@/types/sdk/AgentSessionPrompt.js";
 import type { AgentSessionStopResult } from "@/types/sdk/AgentSessionStop.js";
 import type {
@@ -89,28 +89,23 @@ export interface SessionPromptRuntimeOptions {
   /**
    * 广播 session 事件。
    */
-  publish: (event: AgentSessionEvent) => void;
+  publish: (mutation: SessionMutation) => void;
 
   /**
    * 构造并持久化一条 user 消息。
    */
   createAndPersistUserMessage: (
     input: AgentSessionPromptInput,
+    turn_id: string,
+    input_type: "prompt" | "steer",
   ) => Promise<SessionUserMessageV1>;
 
-  /**
-   * 在 steer message 成功持久化后发送对应 action。
-   */
-  emit_steer_action: (input: {
-    /**
-     * steer message 所属的当前 turn 标识。
-     */
+  /** 持久化一条用户可见的 turn 错误 Message。 */
+  appendErrorMessage: (input: {
+    /** 错误所属 turn。 */
     turn_id: string;
-
-    /**
-     * 已成功持久化的 steer message。
-     */
-    message: SessionUserMessageV1;
+    /** 用户可见错误文本。 */
+    message: string;
   }) => Promise<void>;
 
   /**
@@ -141,7 +136,7 @@ export class SessionPromptRuntime {
   private readonly sessionId: string;
   private readonly publish: SessionPromptRuntimeOptions["publish"];
   private readonly createAndPersistUserMessage: SessionPromptRuntimeOptions["createAndPersistUserMessage"];
-  private readonly emit_steer_action: SessionPromptRuntimeOptions["emit_steer_action"];
+  private readonly appendErrorMessage: SessionPromptRuntimeOptions["appendErrorMessage"];
   private readonly executeTurn: SessionPromptRuntimeOptions["executeTurn"];
   private readonly stopTurn: SessionPromptRuntimeOptions["stopTurn"];
   private readonly queue: QueuedPrompt[] = [];
@@ -152,7 +147,7 @@ export class SessionPromptRuntime {
     this.sessionId = String(options.sessionId || "").trim();
     this.publish = options.publish;
     this.createAndPersistUserMessage = options.createAndPersistUserMessage;
-    this.emit_steer_action = options.emit_steer_action;
+    this.appendErrorMessage = options.appendErrorMessage;
     this.executeTurn = options.executeTurn;
     this.stopTurn = options.stopTurn;
     if (!this.sessionId) {
@@ -231,13 +226,18 @@ export class SessionPromptRuntime {
       const activeTurn = createActiveTurnState(turnId);
       this.activeTurn = activeTurn;
       this.publish({
-        type: "turn-start",
-        turnId,
+        mutation_id: nanoid(),
+        variant: "turn",
+        type: "start",
+        session_id: this.sessionId,
+        turn_id: turnId,
+        status: "running",
+        created_at: Date.now(),
       });
       current.deferredHandle.resolve(createTurnHandle(activeTurn));
 
       try {
-        await this.createAndPersistUserMessage(current.input);
+        await this.createAndPersistUserMessage(current.input, turnId, "prompt");
         const result = await this.executeTurn({
           turnId,
           promptInput: current.input,
@@ -260,10 +260,14 @@ export class SessionPromptRuntime {
         };
         activeTurn.result = finalResult;
         this.publish({
-          type: "turn-finish",
-          turnId,
+          mutation_id: nanoid(),
+          variant: "turn",
+          type: "finish",
+          session_id: this.sessionId,
+          turn_id: turnId,
+          status: stopped ? "stopped" : finalResult.success ? "completed" : "failed",
+          created_at: Date.now(),
           text: finalResult.text,
-          success: finalResult.success,
           ...(finalResult.error ? { error: finalResult.error } : {}),
         });
         activeTurn.deferredFinished.resolve(finalResult);
@@ -283,16 +287,24 @@ export class SessionPromptRuntime {
         };
         activeTurn.result = finalResult;
         if (message !== TURN_STOPPED_MESSAGE) {
-          this.publish({
-            type: "error",
-            message,
-          });
+          try {
+            await this.appendErrorMessage({
+              turn_id: turnId,
+              message,
+            });
+          } catch {
+            // Error Message 写入失败不能阻止 turn handle 收口。
+          }
         }
         this.publish({
-          type: "turn-finish",
-          turnId,
+          mutation_id: nanoid(),
+          variant: "turn",
+          type: "finish",
+          session_id: this.sessionId,
+          turn_id: turnId,
+          status: activeTurn.abortController.signal.aborted ? "stopped" : "failed",
+          created_at: Date.now(),
           text: "",
-          success: false,
           error: message,
         });
         activeTurn.deferredFinished.resolve(finalResult);
@@ -323,14 +335,23 @@ export class SessionPromptRuntime {
       cancelledTurn.result = finalResult;
       cancelledTurn.deferredFinished.resolve(finalResult);
       this.publish({
-        type: "turn-start",
-        turnId,
+        mutation_id: nanoid(),
+        variant: "turn",
+        type: "start",
+        session_id: this.sessionId,
+        turn_id: turnId,
+        status: "running",
+        created_at: Date.now(),
       });
       this.publish({
-        type: "turn-finish",
-        turnId,
+        mutation_id: nanoid(),
+        variant: "turn",
+        type: "finish",
+        session_id: this.sessionId,
+        turn_id: turnId,
+        status: "failed",
+        created_at: Date.now(),
         text: "",
-        success: false,
         error: QUEUED_PROMPT_CANCELLED_MESSAGE,
       });
       item.deferredHandle.resolve(createTurnHandle(cancelledTurn));
@@ -348,17 +369,13 @@ export class SessionPromptRuntime {
     for (let index = 0; index < drained.length; index += 1) {
       const item = drained[index];
       try {
-        const message = await this.createAndPersistUserMessage(item.input);
+        const message = await this.createAndPersistUserMessage(
+          item.input,
+          activeTurn.turnId,
+          "steer",
+        );
         item.deferredHandle.resolve(createTurnHandle(activeTurn));
         merged.push(message);
-        try {
-          await this.emit_steer_action({
-            turn_id: activeTurn.turnId,
-            message,
-          });
-        } catch {
-          // action 发送失败不应阻断已经持久化的 steer message。
-        }
       } catch {
         // 关键点（中文）：若某条消息持久化失败，把未处理部分重新放回队列头部，避免静默丢失。
         const remaining = drained.slice(index);

package/src/session/services/SessionStateService.ts

@@ -19,17 +19,14 @@ import {
 } from "@/session/storage/Metadata.js";
 import { touchSessionMetadata } from "@/session/storage/Persistence.js";
 import { ensureSessionTitle } from "@/session/SessionTitle.js";
-import { persistSdkAssistantResult } from "@/session/storage/Persistence.js";
 import { hydrateUserPromptFileParts } from "@executor/messages/SessionAttachmentMapper.js";
 import type { Executor } from "@executor/Executor.js";
-import type { JsonlSessionHistoryStore } from "@/executor/store/history/jsonl/JsonlSessionHistoryStore.js";
+import type { SessionHistoryStore } from "@/executor/store/history/SessionHistoryStore.js";
 import type {
   AgentSessionConfigSnapshot,
   AgentSessionSetInput,
 } from "@/types/agent/SessionTypes.js";
-import type {
-  AgentSessionEvent,
-} from "@/types/sdk/AgentSessionEvent.js";
+import type { SessionMutation } from "@/types/session/SessionMutation.js";
 import type { AgentSessionPromptInput } from "@/types/sdk/AgentSessionPrompt.js";
 import type {
   SessionActionRecordInputV1,
@@ -38,11 +35,20 @@ import type {
   SessionMessageRecordV1,
   SessionUserMessageV1,
 } from "@/executor/types/SessionRecords.js";
-import { to_session_action_record } from "@/executor/types/SessionRecords.js";
 import type { SessionLocalState } from "@/types/session/SessionLocalState.js";
 import { generateId } from "@/utils/Id.js";
 import type { Logger } from "@/utils/logger/Logger.js";
 import type { AgentModel } from "@/model/CityModelAdapter.js";
+import { SessionRecorder } from "@/session/recorder/SessionRecorder.js";
+import { normalize_session_user_parts } from "@/session/recorder/SessionRecorder.js";
+import {
+  from_ui_assistant_parts,
+  from_ui_user_parts,
+  to_executor_ui_message,
+} from "@/session/recorder/SessionMessageCodec.js";
+import fs from "fs-extra";
+import { getSdkAgentSessionMessagesPath } from "@/session/storage/Paths.js";
+import type { SessionMessage } from "@/types/session/SessionMessage.js";
 
 type SessionStateServiceOptions = {
   /**
@@ -63,7 +69,10 @@ type SessionStateServiceOptions = {
   /**
    * 当前 session 历史事实源。
    */
-  history_store: JsonlSessionHistoryStore;
+  history_store: SessionHistoryStore;
+
+  /** 当前 Session Message Recorder。 */
+  recorder: SessionRecorder;
 
   /**
    * 当前 session 执行器。
@@ -91,7 +100,7 @@ type SessionStateServiceOptions = {
   /**
    * 发布 session 事件。
    */
-  publish_event: (event: AgentSessionEvent) => void;
+  publish_event: (mutation: SessionMutation) => void;
 };
 
 type SessionSetOptions = {
@@ -114,7 +123,8 @@ export class SessionStateService {
   private readonly agent_id: string;
   private readonly project_root: string;
   private readonly session_id: string;
-  private readonly history_store: JsonlSessionHistoryStore;
+  private readonly history_store: SessionHistoryStore;
+  private readonly recorder: SessionRecorder;
   private readonly executor: Executor;
   private readonly state: SessionLocalState;
   private readonly logger: Logger;
@@ -127,6 +137,7 @@ export class SessionStateService {
     this.project_root = options.project_root;
     this.session_id = options.session_id;
     this.history_store = options.history_store;
+    this.recorder = options.recorder;
     this.executor = options.executor;
     this.state = options.state;
     this.logger = options.logger;
@@ -197,6 +208,7 @@ export class SessionStateService {
           : {}),
         ...(metadata.modelId ? { modelId: metadata.modelId } : {}),
       };
+      await this.restore_persisted_model();
     })();
     await this.state.initializePromise;
   }
@@ -206,7 +218,6 @@ export class SessionStateService {
    */
   async ensure_ready_for_execution(): Promise<void> {
     await this.initialize();
-    await this.restore_persisted_model();
     if (this.state.ensureConfiguredPromise) {
       await this.state.ensureConfiguredPromise;
       return;
@@ -239,6 +250,9 @@ export class SessionStateService {
   async set(input: AgentSessionSetInput, options?: SessionSetOptions): Promise<void> {
     const should_emit_action = options?.emit_action !== false;
     const previous_model_label = this.state.sessionConfig.modelLabel;
+    const previous_model_id = String(
+      this.state.sessionConfig.modelId || previous_model_label || "",
+    ).trim();
     const requested_model_id = String(input.modelId || "").trim();
     const next_model = input.model || (requested_model_id
       ? await this.resolve_model_by_id(requested_model_id)
@@ -251,19 +265,18 @@ export class SessionStateService {
       next_model &&
         should_emit_action &&
         this.state.sessionConfig.model &&
-        previous_model_label &&
-        next_model_label &&
-        previous_model_label !== next_model_label,
+        previous_model_id &&
+        next_model_id &&
+        previous_model_id !== next_model_id,
     );
+    const previous_model_name = previous_model_label || previous_model_id;
+    const next_model_name = next_model_label || next_model_id;
     const action_id = `model-switching:${this.session_id}:${Date.now()}:${generateId()}`;
 
     if (should_emit_model_switch_action) {
       await this.emit_action_event({
         id: action_id,
-        title: "Switching session model",
-        description: next_model_label
-          ? `Switching to ${next_model_label}.`
-          : undefined,
+        title: `Switching session model from ${previous_model_name} to ${next_model_name}`,
         state: "running",
       });
     }
@@ -287,7 +300,7 @@ export class SessionStateService {
       if (should_emit_model_switch_action) {
         await this.emit_action_event({
           id: action_id,
-          title: "Session model switch failed",
+          title: `Session model switch from ${previous_model_name} to ${next_model_name} failed`,
           description: error instanceof Error ? error.message : String(error),
           state: "failed",
         });
@@ -298,10 +311,7 @@ export class SessionStateService {
     if (should_emit_model_switch_action) {
       await this.emit_action_event({
         id: action_id,
-        title: "Session model switched",
-        description: this.state.sessionConfig.modelLabel
-          ? `Using ${this.state.sessionConfig.modelLabel}.`
-          : undefined,
+        title: `Session model switched from ${previous_model_name} to ${next_model_name}`,
         state: "completed",
       });
     }
@@ -331,7 +341,20 @@ export class SessionStateService {
     message?: SessionRecordV1 | null;
     text?: string;
   }): Promise<void> {
-    await this.executor.append_user_message(params);
+    const parts = params.message && "role" in params.message
+      ? from_ui_user_parts(params.message.parts)
+      : [{
+          part_id: `external-user-text:${Date.now()}`,
+          type: "text" as const,
+          text: String(params.text || "").trim(),
+          state: "done" as const,
+        }];
+    if (parts.length === 0) return;
+    await this.recorder.append_user_message({
+      turn_id: `external:${this.session_id}:${Date.now()}`,
+      input_type: "prompt",
+      parts,
+    });
     await this.ensure_title_from_history({ generate: true });
     await this.touch_metadata();
   }
@@ -343,7 +366,17 @@ export class SessionStateService {
     message?: SessionRecordV1 | null;
     fallbackText?: string;
   }): Promise<void> {
-    await this.executor.append_assistant_message(params);
+    const parts = params.message && "role" in params.message
+      ? from_ui_assistant_parts(params.message.parts)
+      : [{
+          part_id: `external-assistant-text:${Date.now()}`,
+          sequence: 1,
+          type: "text" as const,
+          text: String(params.fallbackText || "").trim(),
+          state: "done" as const,
+        }];
+    if (parts.length === 0) return;
+    await this.recorder.append_completed_assistant_message({ parts });
     await this.touch_metadata();
   }
 
@@ -351,11 +384,29 @@ export class SessionStateService {
    * 仅刷新当前 session metadata。
    */
   async touch_metadata(): Promise<void> {
+    const message_page = await this.recorder.list_messages({
+      limit: 500,
+      include_internal: true,
+    });
+    const messages_path = getSdkAgentSessionMessagesPath(
+      this.project_root,
+      this.agent_id,
+      this.session_id,
+    );
+    const history_bytes = await fs.stat(messages_path)
+      .then((file_stat) => file_stat.size)
+      .catch(() => 0);
+    const preview_text = resolve_message_preview(
+      message_page.items[message_page.items.length - 1],
+    ).slice(0, 180);
     await touchSessionMetadata({
       projectRoot: this.project_root,
       agentId: this.agent_id,
       sessionId: this.session_id,
       sessionConfig: this.state.sessionConfig,
+      message_count: message_page.total,
+      history_bytes,
+      ...(preview_text ? { preview_text } : {}),
     });
   }
 
@@ -390,8 +441,11 @@ export class SessionStateService {
     const next_title = String(next_metadata.title || "").trim();
     if (!next_title || next_title === before_title) return;
     this.publish_event({
-      type: "session-title",
-      sessionId: this.session_id,
+      mutation_id: generateId(),
+      variant: "session",
+      type: "title",
+      session_id: this.session_id,
+      created_at: Date.now(),
       title: next_title,
     });
   }
@@ -402,14 +456,8 @@ export class SessionStateService {
   async persist_assistant_result(
     assistant_message?: SessionMessageRecordV1 | null,
   ): Promise<void> {
-    await persistSdkAssistantResult({
-      projectRoot: this.project_root,
-      agentId: this.agent_id,
-      sessionId: this.session_id,
-      sessionConfig: this.state.sessionConfig,
-      executor: this.executor,
-      assistantMessage: assistant_message,
-    });
+    void assistant_message;
+    await this.touch_metadata();
   }
 
   /**
@@ -418,8 +466,23 @@ export class SessionStateService {
   async persist_action_event(
     event: SessionActionRecordV1,
   ): Promise<void> {
-    const message = to_session_action_record(event, this.session_id);
-    await this.history_store.write_record(message);
+    const existing = this.recorder.get_message(event.id);
+    if (!existing) {
+      const writer = await this.recorder.open_action_message({
+        message_id: event.id,
+        turn_id: event.metadata.turnId,
+        action_type: infer_action_type(event.id),
+        title: event.title,
+        description: event.description,
+      });
+      if (event.state === "completed") await writer.complete();
+      if (event.state === "failed") await writer.fail(event.description || event.title);
+    } else if (existing.type === "action" && event.state !== "running") {
+      await this.recorder.update_action_message(event.id, event.state, {
+        title: event.title,
+        description: event.description,
+      });
+    }
     await this.touch_metadata();
   }
 
@@ -427,21 +490,26 @@ export class SessionStateService {
    * 写入并发布一条 action。
    */
   async emit_action_event(input: EmitActionInput): Promise<void> {
-    const event = to_session_action_record(
-      {
-        ...input,
-        id:
-          String(input.id || "").trim() ||
-          `action:${this.session_id}:${Date.now()}:${generateId()}`,
+    const action_id = String(input.id || "").trim() ||
+      `action:${this.session_id}:${Date.now()}`;
+    const turn_id = "turnId" in input
+      ? input.turnId
+      : input.metadata?.turnId;
+    await this.persist_action_event({
+      type: "action",
+      id: action_id,
+      title: input.title,
+      ...(input.description ? { description: input.description } : {}),
+      state: input.state,
+      metadata: {
+        v: 1,
+        ts: input.metadata?.ts || Date.now(),
+        sessionId: this.session_id,
+        ...(turn_id
+          ? { turnId: turn_id }
+          : {}),
       },
-      this.session_id,
-    );
-    try {
-      await this.persist_action_event(event);
-    } catch {
-      // action 持久化失败不应阻断宿主操作。
-    }
-    this.publish_event(event);
+    });
   }
 
   /**
@@ -449,43 +517,22 @@ export class SessionStateService {
    */
   async create_and_persist_user_prompt_message(
     input: AgentSessionPromptInput,
+    turn_id: string,
+    input_type: "prompt" | "steer" = "prompt",
   ): Promise<SessionUserMessageV1> {
     const query = input.query;
-    if (typeof query === "string") {
-      const message = this.history_store.userText({
-        text: query.trim(),
-        metadata: {
-          sessionId: this.session_id,
-        },
-      }) as SessionUserMessageV1;
-      await this.executor.append_user_message({
-        message,
-      });
-      await this.ensure_title_from_history({ generate: true });
-      await this.touch_metadata();
-      return message;
-    }
-
-    // query 是 parts 数组，先把本地图片附件转换为模型可消费的 data URL。
-    const parts = await hydrateUserPromptFileParts(
-      Array.isArray(query) ? query : [],
-      this.project_root,
-    );
-    const message: SessionUserMessageV1 = {
-      id: `u:${this.session_id}:${generateId()}`,
-      role: "user",
-      parts,
-      metadata: {
-        v: 1,
-        ts: Date.now(),
-        sessionId: this.session_id,
-        source: "ingress",
-        kind: "normal",
-      },
-    };
-    await this.executor.append_user_message({
-      message,
+    const ui_parts = typeof query === "string"
+      ? [{ type: "text" as const, text: query.trim() }]
+      : await hydrateUserPromptFileParts(
+          Array.isArray(query) ? query : [],
+          this.project_root,
+        );
+    const canonical = await this.recorder.append_user_message({
+      turn_id,
+      input_type,
+      parts: normalize_session_user_parts(ui_parts),
     });
+    const message = to_executor_ui_message(canonical) as SessionUserMessageV1;
     await this.ensure_title_from_history({ generate: true });
     await this.touch_metadata();
     return message;
@@ -502,10 +549,36 @@ export class SessionStateService {
       : [];
     if (normalized_messages.length <= 0) return;
     for (const message of normalized_messages) {
-      await this.executor.append_user_message({
-        message,
+      await this.recorder.append_user_message({
+        turn_id: String(message.metadata?.turnId || `deferred:${this.session_id}:${Date.now()}`),
+        input_type: "steer",
+        parts: from_ui_user_parts(message.parts),
       });
     }
     await this.touch_metadata();
   }
 }
+
+function resolve_message_preview(message: SessionMessage | undefined): string {
+  if (!message) return "";
+  if (message.type === "user") {
+    return message.parts
+      .flatMap((part) => part.type === "text" ? [part.text] : [])
+      .join("")
+      .trim();
+  }
+  if (message.type === "assistant") {
+    return message.parts
+      .flatMap((part) => part.type === "text" ? [part.text] : [])
+      .join("")
+      .trim();
+  }
+  if (message.type === "action") {
+    return [message.title, message.description].filter(Boolean).join("\n");
+  }
+  return message.message.trim();
+}
+
+function infer_action_type(message_id: string): string {
+  return String(message_id || "").split(":")[0] || "action";
+}