@downcity/agent 1.1.238 → 1.1.259

package/src/session/services/SessionTurnService.ts

@@ -1,368 +0,0 @@
-/**
- * SessionTurnService：本地 Session turn 编排服务。
- *
- * 关键点（中文）
- * - 统一管理 prompt actor 调度、turn 执行与 Session 事件流派发。
- * - 该服务负责把 SessionPromptRuntime 与 Executor 连接起来。
- * - metadata、title、assistant 结果持久化等状态副作用交由 SessionStateService 处理。
- */
-
-import { extractTextFromParts, extractTextFromUiMessage } from "@/executor/messages/UIMessageTransformer.js";
-import type { Executor } from "@executor/Executor.js";
-import type { SessionMutation } from "@/types/session/SessionMutation.js";
-import { isAgentSessionPromptInputEmpty } from "@/types/sdk/AgentSessionPrompt.js";
-import type { AgentSessionPromptInput } from "@/types/sdk/AgentSessionPrompt.js";
-import type { AgentSessionStopResult } from "@/types/sdk/AgentSessionStop.js";
-import type { AgentSessionTurnHandle } from "@/types/sdk/AgentSessionTurn.js";
-import type { SessionRunResult } from "@/executor/types/SessionRun.js";
-import type {
-  SessionRecordV1,
-  SessionMessageRecordV1,
-  SessionUserMessageV1,
-} from "@/executor/types/SessionRecords.js";
-import { SessionEventHub } from "@/session/runtime/SessionEventHub.js";
-import { SessionPromptRuntime } from "@/session/runtime/SessionPromptRuntime.js";
-import type { SessionRunContext } from "@/types/executor/SessionRunContext.js";
-import type { SessionQueueCommand } from "@/types/session/SessionQueueCommand.js";
-import { SessionStateService } from "@/session/services/SessionStateService.js";
-import {
-  SessionAssistantMessageWriter,
-  SessionRecorder,
-} from "@/session/recorder/SessionRecorder.js";
-import { from_ui_assistant_parts } from "@/session/recorder/SessionMessageCodec.js";
-import { generateId } from "@/utils/Id.js";
-
-type SessionTurnServiceOptions = {
-  /**
-   * 当前 session 标识。
-   */
-  session_id: string;
-
-  /**
-   * 当前项目根目录。
-   */
-  project_root: string;
-
-  /**
-   * 当前 session 执行器。
-   */
-  executor: Executor;
-
-  /**
-   * 当前 session 状态服务。
-   */
-  state_service: SessionStateService;
-
-  /**
-   * 当前 session 共享事件总线。
-   */
-  event_hub: SessionEventHub;
-
-  /** 当前 Session Message Recorder。 */
-  recorder: SessionRecorder;
-};
-
-/**
- * 本地 Session turn 编排服务。
- */
-export class SessionTurnService {
-  private readonly session_id: string;
-  private readonly project_root: string;
-  private readonly executor: Executor;
-  private readonly state_service: SessionStateService;
-  private readonly event_hub: SessionEventHub;
-  private readonly recorder: SessionRecorder;
-  private readonly prompt_runtime: SessionPromptRuntime;
-  private active_run_context: SessionRunContext | null = null;
-  private request_active_history_reload: (() => void) | null = null;
-
-  constructor(options: SessionTurnServiceOptions) {
-    this.session_id = options.session_id;
-    this.project_root = options.project_root;
-    this.executor = options.executor;
-    this.state_service = options.state_service;
-    this.event_hub = options.event_hub;
-    this.recorder = options.recorder;
-    this.prompt_runtime = new SessionPromptRuntime({
-      sessionId: this.session_id,
-      publish: (event) => {
-        this.publish_event(event);
-      },
-      createAndPersistUserMessage: async (input, turn_id, input_type) => {
-        return await this.state_service.create_and_persist_user_prompt_message(
-          input,
-          turn_id,
-          input_type,
-        );
-      },
-      appendErrorMessage: async ({ turn_id, message }) => {
-        await this.recorder.append_error_message({
-          scope: "turn",
-          turn_id,
-          code: "turn_execution_failed",
-          message,
-          recoverable: true,
-        });
-      },
-      executeTurn: async ({ turnId, promptInput, onStepMerge, abortSignal }) => {
-        return await this.execute_prompt_turn({
-          turnId,
-          promptInput,
-          onStepMerge,
-          abortSignal,
-        });
-      },
-      stopTurn: () => this.executor.stop(),
-    });
-  }
-
-  /**
-   * 发布一条 session 事件。
-   */
-  publish_event(mutation: SessionMutation): void {
-    this.event_hub.publish(mutation);
-  }
-
-  /**
-   * 判断当前 turn 调度器是否活跃。
-   */
-  is_prompt_runtime_active(): boolean {
-    return this.prompt_runtime.isActive();
-  }
-
-  /**
-   * 追加一条新的 Session prompt。
-   */
-  async prompt(input: AgentSessionPromptInput): Promise<AgentSessionTurnHandle> {
-    if (isAgentSessionPromptInputEmpty(input)) {
-      throw new Error("session.prompt requires a non-empty query");
-    }
-    await this.state_service.ensure_runnable();
-    return await this.prompt_runtime.prompt(input);
-  }
-
-  /**
-   * 把 configured state command 加入当前 Session 的统一输入队列。
-   */
-  enqueue_command(command: SessionQueueCommand): void {
-    this.prompt_runtime.enqueue_command(command);
-  }
-
-  /**
-   * 把一次显式历史压缩加入当前 Session 的统一输入队列。
-   */
-  async compact(): Promise<void> {
-    await this.state_service.ensure_runnable();
-    const command_id = generateId();
-    this.enqueue_command({
-      type: "command",
-      command_id,
-      scope: "session",
-      execute: async ({ turn_id }) => {
-        const active_run_context = this.active_run_context;
-        const run_context = active_run_context ||
-          this.create_compaction_run_context(turn_id);
-        const result = await this.executor.compact_history(run_context);
-        if (
-          result.compacted &&
-          active_run_context !== null &&
-          active_run_context === this.active_run_context
-        ) {
-          this.request_active_history_reload?.();
-        }
-        await this.state_service.touch_metadata();
-        if (result.compacted) {
-          return;
-        }
-        if (!result.compacted && result.reason === "nothing_to_compact") {
-          await this.state_service.emit_action_event({
-            id: `compacting:${this.session_id}:${command_id}`,
-            title: "Session records already compact",
-            description: "The Session has no active records to compact.",
-            state: "completed",
-            turnId: turn_id,
-          });
-        }
-      },
-    });
-  }
-
-  /**
-   * 停止当前 turn，并取消尚未被吸收的排队 prompt。
-   */
-  async stop(): Promise<AgentSessionStopResult> {
-    return this.prompt_runtime.stop();
-  }
-
-  /**
-   * 执行单轮 prompt turn。
-   */
-  private async execute_prompt_turn(input: {
-    turnId: string;
-    promptInput: AgentSessionPromptInput;
-    onStepMerge: () => Promise<SessionUserMessageV1[]>;
-    abortSignal?: AbortSignal;
-  }): Promise<{
-    text: string;
-    success: boolean;
-    assistantMessage?: SessionMessageRecordV1 | null;
-    error?: string;
-  }> {
-    const assistant_writer_ref: {
-      current: SessionAssistantMessageWriter | null;
-    } = { current: null };
-    let assistant_segment_index = 0;
-    let history_reload_requested = false;
-    const ensure_assistant_writer = async (): Promise<SessionAssistantMessageWriter> => {
-      if (assistant_writer_ref.current) return assistant_writer_ref.current;
-      assistant_segment_index += 1;
-      assistant_writer_ref.current = await this.recorder.open_assistant_message({
-        turn_id: input.turnId,
-        segment_index: assistant_segment_index,
-      });
-      return assistant_writer_ref.current;
-    };
-    const run_context: SessionRunContext = {
-      turnId: input.turnId,
-      sessionId: this.session_id,
-      projectRoot: this.project_root,
-      onStepCallback: async () => {
-        if (
-          this.prompt_runtime.has_pending_command() &&
-          assistant_writer_ref.current
-        ) {
-          await assistant_writer_ref.current.complete();
-          assistant_writer_ref.current = null;
-        }
-        const merged = await input.onStepMerge();
-        if (merged.length > 0 && assistant_writer_ref.current) {
-          await assistant_writer_ref.current.complete();
-          assistant_writer_ref.current = null;
-        }
-        return merged;
-      },
-      hasPendingStepInput: () => this.prompt_runtime.has_pending_prompt(),
-      consume_history_reload: () => {
-        const requested = history_reload_requested;
-        history_reload_requested = false;
-        return requested;
-      },
-      onUiMessageChunkCallback: async (chunk) => {
-        if (!is_assistant_content_chunk(chunk.type)) return;
-        const writer = await ensure_assistant_writer();
-        await writer.apply_chunk(chunk);
-      },
-      onActionCallback: async (event) => {
-        await this.state_service.persist_action_event(event);
-      },
-      injectedUserMessages: [],
-      deferredPersistedUserMessages: [],
-      pendingAssistantFileParts: [],
-      ...(input.abortSignal ? { abortSignal: input.abortSignal } : {}),
-    };
-    const query = input.promptInput.query;
-    const executor_query = typeof query === "string" ? query : extractTextFromParts(query);
-    this.active_run_context = run_context;
-    this.request_active_history_reload = () => {
-      history_reload_requested = true;
-    };
-    let result: SessionRunResult;
-    try {
-      result = await this.executor.run({
-        query: executor_query,
-        runContext: run_context,
-      });
-    } finally {
-      if (this.active_run_context === run_context) {
-        this.active_run_context = null;
-        this.request_active_history_reload = null;
-      }
-    }
-    const final_assistant_parts = result.assistantMessage
-      ? from_ui_assistant_parts(result.assistantMessage.parts)
-      : [];
-    const final_assistant_writer = assistant_writer_ref.current;
-    if (final_assistant_writer) {
-      for (const part of final_assistant_parts) {
-        if (part.type === "file") {
-          await final_assistant_writer.append_file_part(part);
-        }
-      }
-      if (input.abortSignal?.aborted) {
-        await final_assistant_writer.stop();
-      } else if (result.success) {
-        await final_assistant_writer.complete();
-      } else {
-        await final_assistant_writer.fail(result.error);
-      }
-    } else if (result.assistantMessage) {
-      if (final_assistant_parts.length > 0) {
-        assistant_segment_index += 1;
-        const fallback_writer = await this.recorder.open_assistant_message({
-          turn_id: input.turnId,
-          segment_index: assistant_segment_index,
-        });
-        for (const part of final_assistant_parts) await fallback_writer.upsert_part(part);
-        if (result.success) await fallback_writer.complete();
-        else await fallback_writer.fail(result.error);
-      }
-    }
-    if (!result.success && !input.abortSignal?.aborted && result.error) {
-      await this.recorder.append_error_message({
-        scope: "turn",
-        turn_id: input.turnId,
-        code: "turn_execution_failed",
-        message: result.error,
-        recoverable: true,
-      });
-    }
-    await this.state_service.persist_assistant_result(result.assistantMessage);
-    await this.state_service.persist_deferred_user_messages(
-      result.deferredPersistedUserMessages,
-    );
-    if (result.compact_required) {
-      await this.executor.compact_history(run_context);
-      await this.state_service.touch_metadata();
-    }
-    return {
-      text: result.assistantMessage
-        ? extractTextFromUiMessage(result.assistantMessage)
-        : "",
-      success: result.success,
-      ...(result.assistantMessage
-        ? { assistantMessage: result.assistantMessage }
-        : {}),
-      ...(result.error ? { error: result.error } : {}),
-    };
-  }
-
-  /**
-   * 为 turn 开始前执行的 compact command 构造最小运行上下文。
-   */
-  private create_compaction_run_context(turn_id: string): SessionRunContext {
-    return {
-      turnId: turn_id,
-      sessionId: this.session_id,
-      projectRoot: this.project_root,
-      onActionCallback: async (event) => {
-        await this.state_service.persist_action_event(event);
-      },
-      injectedUserMessages: [],
-      deferredPersistedUserMessages: [],
-      pendingAssistantFileParts: [],
-    };
-  }
-}
-
-function is_assistant_content_chunk(type: string): boolean {
-  return (
-    type === "text-start" ||
-    type === "text-delta" ||
-    type === "text-end" ||
-    type === "reasoning-start" ||
-    type === "reasoning-delta" ||
-    type === "reasoning-end" ||
-    type.startsWith("tool-") ||
-    type === "file"
-  );
-}

package/src/session/services/SessionViewService.ts

@@ -1,335 +0,0 @@
-/**
- * SessionViewService：本地 Session 查询与派生视图服务。
- *
- * 关键点（中文）
- * - 统一管理 info/records/system/fork 这类读取型能力。
- * - 该服务只负责拼装查询结果，不负责编排 prompt turn。
- * - 需要写入持久化状态的地方仍委托给 SessionStateService。
- */
-
-import { nanoid } from "nanoid";
-import { buildSessionInfo } from "@/session/browse/Browse.js";
-import { ensureSessionTitle } from "@/session/SessionTitle.js";
-import { readSessionMetadata } from "@/session/storage/Metadata.js";
-import { buildSessionSystemBlocks } from "@/session/SessionSystemBuilder.js";
-import type { SessionHistoryStore } from "@/executor/store/history/SessionHistoryStore.js";
-import type { SessionSystemComposer } from "@/executor/composer/system/SessionSystemComposer.js";
-import type {
-  AgentSessionForkInput,
-  AgentSessionInfo,
-  AgentSessionSystemBlock,
-  AgentSessionSystemSnapshot,
-} from "@/types/agent/SessionTypes.js";
-import type { AgentSession } from "@/types/agent/SessionActor.js";
-import type { SessionRecordV1 } from "@/executor/types/SessionRecords.js";
-import { SessionStateService } from "@/session/services/SessionStateService.js";
-import type { SessionRunContext } from "@/types/executor/SessionRunContext.js";
-import type { Logger } from "@/utils/logger/Logger.js";
-import { SessionRecorder } from "@/session/recorder/SessionRecorder.js";
-import type { SessionMessage } from "@/types/session/SessionMessage.js";
-
-type SessionViewServiceOptions<TSession extends Pick<AgentSession, "set">> = {
-  /**
-   * 当前 agent 稳定标识。
-   */
-  agent_id: string;
-
-  /**
-   * 当前项目根目录。
-   */
-  project_root: string;
-
-  /**
-   * 当前 session 标识。
-   */
-  session_id: string;
-
-  /**
-   * 当前 session 历史事实源。
-   */
-  history_store: SessionHistoryStore;
-
-  /** 当前 Session Message Recorder。 */
-  recorder: SessionRecorder;
-
-  /**
-   * 当前 session 状态服务。
-   */
-  state_service: SessionStateService;
-
-  /**
-   * 当前 session 运行日志器。
-   */
-  logger: Logger;
-
-  /**
-   * 判断当前 session 是否正在执行。
-   */
-  is_executing: () => boolean;
-
-  /**
-   * 读取 instruction system blocks。
-   */
-  get_instruction_system_blocks: () => AgentSessionSystemBlock[];
-
-  /**
-   * 读取受托管 plugin system blocks。
-   */
-  get_managed_plugin_system_blocks: () => Promise<AgentSessionSystemBlock[]>;
-
-  /**
-   * 读取显式注册 plugin system blocks。
-   */
-  get_plugin_system_blocks: () => Promise<AgentSessionSystemBlock[]>;
-
-  /**
-   * 可选自定义 system composer。
-   *
-   * 关键点（中文）
-   * - 仅当调用方覆盖了 system composer 时传入。
-   * - 默认 SDK system snapshot 仍保留 block 级来源信息。
-   */
-  custom_system_composer?: SessionSystemComposer;
-
-  /**
-   * 创建一个新的本地 Session 实例。
-   */
-  create_fork_session: (session_id: string) => Promise<{
-    session: TSession;
-    history_store: SessionHistoryStore;
-    recorder: SessionRecorder;
-    state_service: SessionStateService;
-  }>;
-};
-
-/**
- * 本地 Session 查询与派生视图服务。
- */
-export class SessionViewService<TSession extends Pick<AgentSession, "set">> {
-  private readonly agent_id: string;
-  private readonly project_root: string;
-  private readonly session_id: string;
-  private readonly history_store: SessionHistoryStore;
-  private readonly recorder: SessionRecorder;
-  private readonly state_service: SessionStateService;
-  private readonly logger: Logger;
-  private readonly is_executing: SessionViewServiceOptions<TSession>["is_executing"];
-  private readonly get_instruction_system_blocks: SessionViewServiceOptions<TSession>["get_instruction_system_blocks"];
-  private readonly get_managed_plugin_system_blocks: SessionViewServiceOptions<TSession>["get_managed_plugin_system_blocks"];
-  private readonly get_plugin_system_blocks: SessionViewServiceOptions<TSession>["get_plugin_system_blocks"];
-  private readonly custom_system_composer?: SessionSystemComposer;
-  private readonly create_fork_session: SessionViewServiceOptions<TSession>["create_fork_session"];
-
-  constructor(options: SessionViewServiceOptions<TSession>) {
-    this.agent_id = options.agent_id;
-    this.project_root = options.project_root;
-    this.session_id = options.session_id;
-    this.history_store = options.history_store;
-    this.recorder = options.recorder;
-    this.state_service = options.state_service;
-    this.logger = options.logger;
-    this.is_executing = options.is_executing;
-    this.get_instruction_system_blocks = options.get_instruction_system_blocks;
-    this.get_managed_plugin_system_blocks =
-      options.get_managed_plugin_system_blocks;
-    this.get_plugin_system_blocks = options.get_plugin_system_blocks;
-    this.custom_system_composer = options.custom_system_composer;
-    this.create_fork_session = options.create_fork_session;
-  }
-
-  /**
-   * 读取当前 session 详情。
-   */
-  async get_info(): Promise<AgentSessionInfo> {
-    const [metadata, messages] = await Promise.all([
-      readSessionMetadata({
-        projectRoot: this.project_root,
-        agentId: this.agent_id,
-        sessionId: this.session_id,
-      }),
-      this.history_store.list_records(),
-    ]);
-    return await this.build_info({
-      metadata,
-      messages,
-    });
-  }
-
-  private async build_info(input: {
-    /**
-     * 当前 session metadata。
-     */
-    metadata: Awaited<ReturnType<typeof readSessionMetadata>>;
-
-    /**
-     * 当前 session 消息列表。
-     */
-    messages: SessionRecordV1[];
-  }): Promise<AgentSessionInfo> {
-    const metadata_with_title = input.metadata.title
-      ? input.metadata
-      : await ensureSessionTitle({
-          projectRoot: this.project_root,
-          agentId: this.agent_id,
-          sessionId: this.session_id,
-          messages: input.messages,
-          logger: this.logger,
-        });
-    return buildSessionInfo({
-      projectRoot: this.project_root,
-      agentId: this.agent_id,
-      sessionId: this.session_id,
-      metadata: metadata_with_title,
-      messages: input.messages,
-      executing: this.is_executing(),
-    });
-  }
-
-  /**
-   * 读取当前 session 生效的 system 快照。
-   */
-  async system(): Promise<AgentSessionSystemSnapshot> {
-    const blocks = this.custom_system_composer
-      ? await this.resolve_custom_system_blocks(this.custom_system_composer)
-      : await buildSessionSystemBlocks({
-          agentId: this.agent_id,
-          projectRoot: this.project_root,
-          sessionId: this.session_id,
-          createdAt: this.state_service.get_created_at(),
-          timezone: this.state_service.get_timezone(),
-          getInstructionSystemBlocks: this.get_instruction_system_blocks,
-          getManagedPluginSystemBlocks: this.get_managed_plugin_system_blocks,
-          getPluginSystemBlocks: this.get_plugin_system_blocks,
-        });
-    return {
-      sessionId: this.session_id,
-      session: {
-        agentId: this.agent_id,
-        sessionId: this.session_id,
-        projectRoot: this.project_root,
-        createdAt: new Date(this.state_service.get_created_at()).toISOString(),
-        timezone: this.state_service.get_timezone(),
-      },
-      blocks,
-    };
-  }
-
-  private async resolve_custom_system_blocks(
-    composer: SessionSystemComposer,
-  ): Promise<AgentSessionSystemBlock[]> {
-    const run_context: SessionRunContext = {
-      sessionId: this.session_id,
-      injectedUserMessages: [],
-      deferredPersistedUserMessages: [],
-      pendingAssistantFileParts: [],
-    };
-    const messages = await composer.resolve(run_context);
-    const blocks: AgentSessionSystemBlock[] = [];
-    messages.forEach((message, index) => {
-      const content = this.stringify_system_content(
-        (message as { content?: unknown }).content,
-      );
-      if (!content) return;
-      blocks.push({
-        source: "session",
-        name: `${composer.name || "custom_system"}:${index + 1}`,
-        content,
-      });
-    });
-    return blocks;
-  }
-
-  /**
-   * 从当前 session 创建一个分叉会话。
-   */
-  async fork(input?: AgentSessionForkInput | string): Promise<TSession> {
-    const message_id =
-      typeof input === "string"
-        ? String(input || "").trim() || undefined
-        : String(input?.messageId || "").trim() || undefined;
-    const messages = await this.recorder.list_history_messages();
-    const fork_messages =
-      !message_id
-        ? messages
-        : this.resolve_fork_messages(messages, message_id);
-    const action_id = `history-forking:${this.session_id}:${Date.now()}:${nanoid(8)}`;
-
-    await this.state_service.emit_action_event({
-      id: action_id,
-      title: "Forking session records",
-      description: `Preparing ${String(fork_messages.length)} messages for the new session.`,
-      state: "running",
-    });
-
-    try {
-      const forked_bundle = await this.create_fork_session(
-        `fork-${Date.now()}-${nanoid(8)}`,
-      );
-      const forked = forked_bundle.session;
-      const session_config = this.state_service.get_config();
-      if (session_config.model) {
-        await forked_bundle.state_service.set(
-          {
-            model: session_config.model,
-            ...(session_config.modelId
-              ? { modelId: session_config.modelId }
-              : {}),
-          },
-          { emit_action: false },
-        );
-      }
-      await this.append_fork_messages(forked_bundle, fork_messages);
-      await this.state_service.emit_action_event({
-        id: action_id,
-        title: "Session records forked",
-        description: `Created ${String((forked as { id?: unknown }).id || "")} with ${String(fork_messages.length)} messages.`,
-        state: "completed",
-      });
-      return forked;
-    } catch (error) {
-      await this.state_service.emit_action_event({
-        id: action_id,
-        title: "Session records fork failed",
-        description: error instanceof Error ? error.message : String(error),
-        state: "failed",
-      });
-      throw error;
-    }
-  }
-
-  private resolve_fork_messages(
-    messages: SessionMessage[],
-    message_id: string,
-  ): SessionMessage[] {
-    const target_index = messages.findIndex(
-      (message) => message.message_id === message_id,
-    );
-    if (target_index < 0) {
-      throw new Error(
-        `Cannot fork session "${this.session_id}": messageId "${message_id}" not found.`,
-      );
-    }
-    return messages.slice(0, target_index + 1);
-  }
-
-  private async append_fork_messages(
-    forked_bundle: {
-      history_store: SessionHistoryStore;
-      recorder: SessionRecorder;
-      state_service: SessionStateService;
-    },
-    messages: SessionMessage[],
-  ): Promise<void> {
-    await forked_bundle.recorder.import_messages(messages);
-  }
-
-  private stringify_system_content(content: unknown): string {
-    if (typeof content === "string") return content.trim();
-    if (content === null || content === undefined) return "";
-    try {
-      return JSON.stringify(content);
-    } catch {
-      return String(content || "").trim();
-    }
-  }
-}