@downcity/agent 1.1.287 → 1.1.310

package/src/session/SessionState.ts

@@ -8,41 +8,31 @@
  */
 
 import {
-  inferAgentModelLabel,
+  infer_agent_model_label,
   read_agent_model_context_window,
 } from "@/agent/AgentModel.js";
-import {
-  patchSessionModelLabel,
-  readSessionMetadata,
-  resolveSystemTimezone,
-  writeSessionMetadata,
-} from "@/session/storage/Metadata.js";
-import { touchSessionMetadata } from "@/session/storage/Metadata.js";
-import { ensureSessionTitle } from "@/session/SessionTitle.js";
+import { resolve_system_timezone } from "@/session/storage/Metadata.js";
+import { ensure_session_title } from "@/session/SessionTitle.js";
 import type {
   AgentSessionConfigSnapshot,
-  AgentSessionSetInput,
 } from "@/types/agent/SessionTypes.js";
+import type { AgentModel } from "@/agent/AgentModel.js";
 import type { SessionLocalState } from "@/types/session/SessionLocalState.js";
-import type { SessionModelQueueCommand } from "@/types/session/SessionQueue.js";
-import { generateId } from "@/utils/Id.js";
+import { generate_id } from "@/utils/Id.js";
 import type { Logger } from "@/utils/logger/Logger.js";
 import { SessionMessages } from "@/session/SessionMessages.js";
 import { to_executor_history } from "@/session/messages/SessionMessageCodec.js";
 import type { SessionMessage } from "@/types/session/SessionMessage.js";
-import type {
-  SessionConfiguredCommandResult,
-  SessionSetOptions,
-  SessionStateOptions,
-} from "@/types/session/SessionState.js";
+import type { SessionStateOptions } from "@/types/session/SessionState.js";
+import type { SessionStore } from "@/types/store/SessionStore.js";
 
 /**
  * 本地 Session 配置与 Metadata 状态管理器。
  */
 export class SessionState {
   private readonly agent_id: string;
-  private readonly project_root: string;
   private readonly session_id: string;
+  private readonly store: SessionStore;
   private readonly messages: SessionMessages;
   private readonly state: SessionLocalState;
   private readonly logger: Logger;
@@ -52,8 +42,8 @@ export class SessionState {
 
   constructor(options: SessionStateOptions) {
     this.agent_id = options.agent_id;
-    this.project_root = options.project_root;
     this.session_id = options.session_id;
+    this.store = options.store;
     this.messages = options.messages;
     this.state = options.state;
     this.logger = options.logger;
@@ -94,27 +84,18 @@ export class SessionState {
       return;
     }
     this.state.initialize_promise = (async () => {
-      const metadata = await readSessionMetadata({
-        projectRoot: this.project_root,
-        agentId: this.agent_id,
-        sessionId: this.session_id,
-      });
+      const metadata = await this.store.read_metadata();
       const created_at =
-        typeof metadata.createdAt === "number" ? metadata.createdAt : Date.now();
+        typeof metadata.created_at === "number" ? metadata.created_at : Date.now();
       const timezone =
         typeof metadata.timezone === "string" && metadata.timezone.trim()
           ? metadata.timezone.trim()
-          : resolveSystemTimezone();
-      await writeSessionMetadata({
-        projectRoot: this.project_root,
-        agentId: this.agent_id,
-        sessionId: this.session_id,
-        meta: {
-          ...metadata,
-          agentId: this.agent_id,
-          createdAt: created_at,
-          timezone,
-        },
+          : resolve_system_timezone();
+      await this.store.write_metadata({
+        ...metadata,
+        agent_id: this.agent_id,
+        created_at: created_at,
+        timezone,
       });
       this.state.created_at = created_at;
       this.state.timezone = timezone;
@@ -160,76 +141,29 @@ export class SessionState {
   /**
    * 写入当前 session 配置。
    */
-  async set(
-    input: AgentSessionSetInput,
-    options?: SessionSetOptions,
-  ): Promise<SessionConfiguredCommandResult> {
-    const should_emit_action = options?.emit_action !== false;
-    const previous_model_label = this.state.session_config.modelLabel;
-    const next_model = input.model;
-    const next_model_label = next_model
-      ? inferAgentModelLabel(next_model)
-      : undefined;
-    const previous_model_name = String(previous_model_label || "").trim();
-    const next_model_name = String(next_model_label || "").trim();
-    const should_emit_model_switch_action = Boolean(
-      next_model &&
-        should_emit_action &&
-        this.state.session_config.model &&
-        previous_model_name &&
-        next_model_name &&
-        previous_model_name !== next_model_name,
-    );
-    const action_id = `model-switching:${this.session_id}:${Date.now()}:${generateId()}`;
-
+  async set_model(model: AgentModel): Promise<AgentSessionConfigSnapshot> {
+    const next_model_label = infer_agent_model_label(model);
     const next_config: AgentSessionConfigSnapshot = {
       ...this.state.session_config,
+      model,
+      model_label: next_model_label,
+      model_context_window: read_agent_model_context_window(model),
     };
-    if (next_model) {
-      next_config.model = next_model;
-      next_config.modelLabel = next_model_label;
-      next_config.model_context_window =
-        read_agent_model_context_window(next_model);
-    }
-    await patchSessionModelLabel({
-      projectRoot: this.project_root,
-      agentId: this.agent_id,
-      sessionId: this.session_id,
-      model: next_config.model,
+    const metadata = await this.store.read_metadata();
+    await this.store.write_metadata({
+      ...metadata,
+      agent_id: this.agent_id,
+      updated_at: Date.now(),
+      ...(next_model_label ? { model_label: next_model_label } : {}),
     });
     this.state.session_config = next_config;
-
-    if (options?.emit_action === false) {
-      await this.apply_model_command({
-        type: "session_model",
-        command_id: generateId(),
-        config: next_config,
-      });
-      return {};
-    }
-
-    return {
-      command: {
-        type: "session_model",
-        command_id: generateId(),
-        config: next_config,
-        ...(should_emit_model_switch_action
-          ? {
-              action_id,
-              action_title:
-                `Session model switched from ${previous_model_name} to ${next_model_name}`,
-            }
-          : {}),
-      },
-    };
+    return next_config;
   }
 
   /** 在 Session Step 检查点提交模型配置。 */
-  async apply_model_command(
-    command: SessionModelQueueCommand,
-  ): Promise<void> {
+  apply_model_config(config: AgentSessionConfigSnapshot): void {
     this.state.effective_session_config = {
-      ...command.config,
+      ...config,
     };
   }
 
@@ -241,14 +175,17 @@ export class SessionState {
     const preview_text = resolve_message_preview(
       stats.latest_message || undefined,
     ).slice(0, 180);
-    await touchSessionMetadata({
-      projectRoot: this.project_root,
-      agentId: this.agent_id,
-      sessionId: this.session_id,
-      sessionConfig: this.state.session_config,
+    const metadata = await this.store.read_metadata();
+    await this.store.write_metadata({
+      ...metadata,
+      agent_id: this.agent_id,
+      updated_at: Date.now(),
+      ...(this.state.session_config.model_label
+        ? { model_label: this.state.session_config.model_label }
+        : {}),
       message_count: stats.message_count,
-      history_bytes: stats.history_bytes,
-      ...(preview_text ? { preview_text } : {}),
+      historyBytes: stats.history_bytes,
+      ...(preview_text ? { preview_text: preview_text } : {}),
     });
   }
 
@@ -265,24 +202,19 @@ export class SessionState {
       this.session_id,
       await this.messages.context_snapshot(),
     );
-    const before_metadata = await readSessionMetadata({
-      projectRoot: this.project_root,
-      agentId: this.agent_id,
-      sessionId: this.session_id,
-    });
+    const before_metadata = await this.store.read_metadata();
     const before_title = String(before_metadata.title || "").trim();
-    const next_metadata = await ensureSessionTitle({
-      projectRoot: this.project_root,
-      agentId: this.agent_id,
-      sessionId: this.session_id,
+    const next_metadata = await ensure_session_title({
+      session_id: this.session_id,
+      store: this.store,
       messages,
       ...(input?.generate
         ? {
             model: this.get_model(),
           }
         : {}),
-      ...(this.state.session_config.modelLabel
-        ? { modelLabel: this.state.session_config.modelLabel }
+      ...(this.state.session_config.model_label
+        ? { model_label: this.state.session_config.model_label }
         : {}),
       logger: this.logger,
       generate: input?.generate === true,
@@ -290,7 +222,7 @@ export class SessionState {
     const next_title = String(next_metadata.title || "").trim();
     if (!next_title || next_title === before_title) return;
     this.publish_event({
-      mutation_id: generateId(),
+      mutation_id: generate_id(),
       variant: "session",
       type: "title",
       session_id: this.session_id,

package/src/session/SessionSystem.ts

@@ -47,10 +47,10 @@ function create_session_info(
 ): AgentSessionSystemSessionInfo {
   const created_at = Number.isFinite(input.created_at) ? input.created_at : 0;
   return {
-    agentId: String(input.agent_id || "").trim(),
-    sessionId: String(input.session_id || "").trim(),
-    projectRoot: String(input.project_root || "").trim(),
-    createdAt: new Date(created_at).toISOString(),
+    agent_id: String(input.agent_id || "").trim(),
+    session_id: String(input.session_id || "").trim(),
+    project_root: String(input.project_root || "").trim(),
+    created_at: new Date(created_at).toISOString(),
     timezone: String(input.timezone || "").trim() || "UTC",
   };
 }
@@ -60,9 +60,9 @@ function create_session_system_block(
 ): AgentSessionSystemBlock {
   const content = [
     "Current session context:",
-    `You are serving agent "${session.agentId}" in session "${session.sessionId}".`,
-    `The current project root is "${session.projectRoot}".`,
-    `This session was created at ${session.createdAt}, with ${session.timezone} as its reference timezone.`,
+    `You are serving agent "${session.agent_id}" in session "${session.session_id}".`,
+    `The current project root is "${session.project_root}".`,
+    `This session was created at ${session.created_at}, with ${session.timezone} as its reference timezone.`,
     "This creation time is a stable reference for the session and does not represent the current time for every run.",
     "If the user message provides a newer current time, a relative time, or other dynamic context, prioritize the dynamic information from the user message.",
   ].join("\n");

package/src/session/SessionTitle.ts

@@ -12,11 +12,8 @@ import type { SessionHistoryMetaV1 } from "@/executor/types/SessionHistoryMeta.j
 import type { SessionRecordV1 } from "@/executor/types/SessionRecords.js";
 import { is_session_message_record } from "@/executor/types/SessionRecords.js";
 import type { Logger } from "@/utils/logger/Logger.js";
-import {
-  normalizeSessionTitle,
-  readSessionMetadata,
-  writeSessionMetadata,
-} from "@/session/storage/Metadata.js";
+import { normalize_session_title } from "@/session/storage/Metadata.js";
+import type { SessionStore } from "@/types/store/SessionStore.js";
 
 const GENERATED_SESSION_TITLE_MAX_CHARS = 24;
 
@@ -24,20 +21,13 @@ const GENERATED_SESSION_TITLE_MAX_CHARS = 24;
  * 标题持久化参数。
  */
 export interface EnsureSessionTitleParams {
-  /**
-   * 当前项目根目录。
-   */
-  projectRoot: string;
-
-  /**
-   * 当前 agent 稳定标识。
-   */
-  agentId: string;
+  /** 当前 Session 的领域持久化入口。 */
+  store: SessionStore;
 
   /**
-   * 当前 sessionId。
+   * 当前 session_id。
    */
-  sessionId: string;
+  session_id: string;
 
   /**
    * 当前 session 已落盘消息。
@@ -52,7 +42,7 @@ export interface EnsureSessionTitleParams {
   /**
    * 当前模型展示标签；仅用于排障日志，不参与生成逻辑。
    */
-  modelLabel?: string;
+  model_label?: string;
 
   /**
    * 当前 session 运行日志器；标题生成失败时仅记录摘要，不影响主流程。
@@ -102,7 +92,7 @@ function normalizeGeneratedTitle(input: string): string | undefined {
     .split("\n")
     .map((line) => line.trim())
     .find(Boolean);
-  const title = normalizeSessionTitle(
+  const title = normalize_session_title(
     String(firstLine || "")
       .replace(/^["'`“”‘’]+|["'`“”‘’]+$/g, "")
       .replace(/^标题[:：]\s*/i, "")
@@ -144,7 +134,7 @@ async function logSessionTitleDiagnostic(input: {
   /**
    * 当前 session 标识。
    */
-  sessionId: string;
+  session_id: string;
 
   /**
    * 日志级别。
@@ -169,7 +159,7 @@ async function logSessionTitleDiagnostic(input: {
   if (!input.logger) return;
   try {
     await input.logger.log(input.level, input.message, {
-      sessionId: input.sessionId,
+      session_id: input.session_id,
       ...input.details,
     });
   } catch {
@@ -186,12 +176,12 @@ async function generateSessionTitle(input: {
   /**
    * 当前 session 标识。
    */
-  sessionId: string;
+  session_id: string;
 
   /**
    * 当前模型展示标签；仅用于排障日志。
    */
-  modelLabel?: string;
+  model_label?: string;
 
   /**
    * 首条用户消息文本。
@@ -224,11 +214,11 @@ async function generateSessionTitle(input: {
     if (!generatedTitle) {
       await logSessionTitleDiagnostic({
         logger: input.logger,
-        sessionId: input.sessionId,
+        session_id: input.session_id,
         level: "warn",
         message: "[agent] session_title.empty",
         details: {
-          modelLabel: input.modelLabel || null,
+          model_label: input.model_label || null,
           firstUserTextLength: input.firstUserText.length,
           rawTitleLength: String(text || "").length,
         },
@@ -239,11 +229,11 @@ async function generateSessionTitle(input: {
     const effectiveError = observedStreamError || error;
     await logSessionTitleDiagnostic({
       logger: input.logger,
-      sessionId: input.sessionId,
+      session_id: input.session_id,
       level: "warn",
       message: "[agent] session_title.generate_failed",
       details: {
-        modelLabel: input.modelLabel || null,
+        model_label: input.model_label || null,
         firstUserTextLength: input.firstUserText.length,
         ...summarizeTitleError(effectiveError),
       },
@@ -256,10 +246,10 @@ async function generateSessionTitle(input: {
 /**
  * 确保当前 session meta 中持久化 title。
  */
-export async function ensureSessionTitle(
+export async function ensure_session_title(
   input: EnsureSessionTitleParams,
 ): Promise<SessionHistoryMetaV1> {
-  const current = await readSessionMetadata(input);
+  const current = await input.store.read_metadata();
   if (current.title) return current;
 
   const firstUserText = resolveFirstUserText(input.messages);
@@ -269,13 +259,13 @@ export async function ensureSessionTitle(
   if (!input.model || !firstUserText) {
     await logSessionTitleDiagnostic({
       logger: input.logger,
-      sessionId: input.sessionId,
+      session_id: input.session_id,
       level: "debug",
       message: "[agent] session_title.skipped",
       details: {
         reason: !input.model ? "missing_model" : "missing_first_user_text",
-        modelLabel: input.modelLabel || null,
-        messageCount: input.messages.length,
+        model_label: input.model_label || null,
+        message_count: input.messages.length,
       },
     });
     return current;
@@ -283,8 +273,8 @@ export async function ensureSessionTitle(
 
   const generatedTitle = await generateSessionTitle({
     model: input.model,
-    sessionId: input.sessionId,
-    modelLabel: input.modelLabel,
+    session_id: input.session_id,
+    model_label: input.model_label,
     firstUserText,
     logger: input.logger,
   });
@@ -294,11 +284,6 @@ export async function ensureSessionTitle(
     ...current,
     title: generatedTitle,
   };
-  await writeSessionMetadata({
-    projectRoot: input.projectRoot,
-    agentId: input.agentId,
-    sessionId: input.sessionId,
-    meta: generatedMeta,
-  });
+  await input.store.write_metadata(generatedMeta);
   return generatedMeta;
 }