@downcity/agent 1.1.338 → 1.1.400

package/src/executor/core-engine/CoreEngineMessageState.ts

@@ -2,36 +2,22 @@
  * CoreEngine 模型消息运行态。
  *
  * 关键点（中文）
- * - CoreEngine 同时维护 session 语义消息与模型消息。
- * - 新增 user 消息可优先做增量转换，失败时再回退为全量重算。
- * - assistant UI 消息只需要进入 session 语义基线；模型侧使用 SDK 返回的 response messages。
+ * - CoreEngine 只维护本 Turn 后续模型 Step 需要的标准 ModelMessage。
+ * - canonical SessionMessage 的持久化由 SessionMessages 独立拥有。
  */
 
-import type { ModelMessage, Tool } from "ai";
-import type { SessionRecordV1 } from "@/executor/types/SessionRecords.js";
-import {
-  pick_merged_user_messages,
-  to_model_messages,
-} from "@executor/messages/SessionMessageCodec.js";
+import type { ModelMessage } from "@downcity/type";
+import { session_messages_to_model_messages } from "@executor/messages/SessionModelMessages.js";
+import type { SessionMessage, SessionUserMessage } from "@/types/session/SessionMessage.js";
 
 /**
  * CoreEngine 单轮执行期间的消息基线。
  */
 export class CoreEngineMessageState {
-  /**
-   * 当前运行时 session 语义消息。
-   */
-  private sessionMessages: SessionRecordV1[];
-
   /**
    * 当前模型侧消息基线。
    */
-  private currentModelMessages: ModelMessage[];
-
-  /**
-   * 当前轮可用工具集合。
-   */
-  private readonly tools: Record<string, Tool>;
+  private current_model_messages: ModelMessage[];
 
   /**
    * 当前项目根目录，用于解析历史中的相对路径 file part。
@@ -39,57 +25,34 @@ export class CoreEngineMessageState {
   private readonly project_root?: string;
 
   private constructor(params: {
-    /**
-     * 当前运行时 session 语义消息。
-     */
-    sessionMessages: SessionRecordV1[];
     /**
      * 当前模型侧消息基线。
      */
-    modelMessages: ModelMessage[];
-    /**
-     * 当前轮可用工具集合。
-     */
-    tools: Record<string, Tool>;
+    model_messages: ModelMessage[];
     /**
      * 当前项目根目录。
      */
     project_root?: string;
   }) {
-    this.sessionMessages = params.sessionMessages;
-    this.currentModelMessages = params.modelMessages;
-    this.tools = params.tools;
+    this.current_model_messages = params.model_messages;
     this.project_root = params.project_root;
   }
 
   /**
-   * 基于初始 session 消息创建运行态。
+   * 基于初始模型消息创建运行态。
    */
   static async create(params: {
     /**
-     * 初始 session 语义消息。
-     */
-    messages: SessionRecordV1[];
-    /**
-     * 当前轮可用工具集合。
+     * 初始标准模型消息。
      */
-    tools: Record<string, Tool>;
+    messages: ModelMessage[];
     /**
      * 当前项目根目录。
      */
     project_root?: string;
   }): Promise<CoreEngineMessageState> {
-    const sessionMessages = Array.isArray(params.messages)
-      ? [...params.messages]
-      : [];
     return new CoreEngineMessageState({
-      sessionMessages,
-      modelMessages: await to_model_messages(
-        sessionMessages,
-        params.tools,
-        params.project_root,
-      ),
-      tools: params.tools,
+      model_messages: Array.isArray(params.messages) ? [...params.messages] : [],
       project_root: params.project_root,
     });
   }
@@ -97,41 +60,33 @@ export class CoreEngineMessageState {
   /**
    * 读取当前模型消息。
    */
-  get modelMessages(): ModelMessage[] {
-    return this.currentModelMessages;
+  get model_messages(): ModelMessage[] {
+    return this.current_model_messages;
   }
 
   /**
-   * 把 step 间新增的 user 消息并入两份基线。
+   * 把 Step 间新增的 canonical User Message 并入模型基线。
    */
-  async appendMergedUserMessages(
-    messages: SessionRecordV1[],
+  async append_merged_user_messages(
+    messages: SessionUserMessage[],
   ): Promise<ModelMessage[]> {
-    const mergedMessages = pick_merged_user_messages(messages);
-    if (mergedMessages.length === 0) return [];
-    return await this.appendSessionMessagesAsModelMessages(mergedMessages);
+    if (messages.length === 0) return [];
+    return await this.append_session_messages(messages);
   }
 
   /**
    * 追加内部生成的 user nudge 消息。
    */
-  async appendUserTextMessage(message: SessionRecordV1): Promise<void> {
-    await this.appendSessionMessagesAsModelMessages([message]);
-  }
-
-  /**
-   * 追加 assistant UI 消息到 session 语义基线。
-   */
-  appendRuntimeSessionMessage(message: SessionRecordV1): void {
-    this.sessionMessages = [...this.sessionMessages, message];
+  async append_user_message(message: SessionUserMessage): Promise<void> {
+    await this.append_session_messages([message]);
   }
 
   /**
    * 追加 SDK 返回的模型 response messages。
    */
-  appendModelMessages(messages: ModelMessage[]): void {
+  append_model_messages(messages: ModelMessage[]): void {
     if (!Array.isArray(messages) || messages.length === 0) return;
-    this.currentModelMessages = [...this.currentModelMessages, ...messages];
+    this.current_model_messages = [...this.current_model_messages, ...messages];
   }
 
   /**
@@ -141,46 +96,22 @@ export class CoreEngineMessageState {
    * Session 语义消息与持久化历史仍保持完整，等待 turn 收口后再单独归档。
    */
   replace_model_messages(messages: ModelMessage[]): void {
-    this.currentModelMessages = Array.isArray(messages) ? [...messages] : [];
+    this.current_model_messages = Array.isArray(messages) ? [...messages] : [];
   }
 
-  /**
-   * 使用 canonical records 原子替换两份消息基线。
-   *
-   * 关键点（中文）
-   * - 显式 compact 会重写持久化历史，旧的 SessionRecord 与 ModelMessage 必须同时失效。
-   * - tools 使用当前 step 的执行视图，避免配置 command 生效后仍按旧工具解码历史。
-   */
-  async replace_session_messages(
-    messages: SessionRecordV1[],
-    tools: Record<string, Tool>,
-  ): Promise<void> {
-    this.sessionMessages = Array.isArray(messages) ? [...messages] : [];
-    this.currentModelMessages = await to_model_messages(
-      this.sessionMessages,
-      tools,
-      this.project_root,
-    );
+  /** 使用 Composer 重新生成的模型历史原子替换当前基线。 */
+  replace_model_history(messages: ModelMessage[]): void {
+    this.current_model_messages = Array.isArray(messages) ? [...messages] : [];
   }
 
-  private async appendSessionMessagesAsModelMessages(
-    messages: SessionRecordV1[],
+  private async append_session_messages(
+    messages: SessionMessage[],
   ): Promise<ModelMessage[]> {
-    this.sessionMessages = [...this.sessionMessages, ...messages];
-    const modelMessages = await to_model_messages(
+    const model_messages = await session_messages_to_model_messages(
       messages,
-      this.tools,
-      this.project_root,
-    );
-    if (modelMessages.length > 0) {
-      this.currentModelMessages = [...this.currentModelMessages, ...modelMessages];
-      return modelMessages;
-    }
-    this.currentModelMessages = await to_model_messages(
-      this.sessionMessages,
-      this.tools,
       this.project_root,
     );
-    return [];
+    this.current_model_messages = [...this.current_model_messages, ...model_messages];
+    return model_messages;
   }
 }