@downcity/agent 1.1.161 → 1.1.163

package/src/session/runtime/SessionPromptRuntime.ts

@@ -146,9 +146,7 @@ export class SessionPromptRuntime {
   prompt(input: AgentSessionPromptInput): Promise<AgentSessionTurnHandle> {
     const deferredHandle = createDeferred<AgentSessionTurnHandle>();
     this.queue.push({
-      input: {
-        query: String(input.query || "").trim(),
-      },
+      input,
       deferredHandle,
     });
     this.ensureProcessing();

package/src/session/services/SessionStateService.ts

@@ -35,6 +35,7 @@ import type {
   SessionUserMessageV1,
 } from "@/executor/types/SessionMessages.js";
 import type { SessionLocalState } from "@/types/session/SessionLocalState.js";
+import { generateId } from "@/utils/Id.js";
 
 type SessionStateServiceOptions = {
   /**
@@ -307,12 +308,36 @@ export class SessionStateService {
   async create_and_persist_user_prompt_message(
     input: AgentSessionPromptInput,
   ): Promise<SessionUserMessageV1> {
-    const message = this.history_store.userText({
-      text: String(input.query || "").trim(),
+    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.appendUserMessage({
+        message,
+      });
+      await this.ensure_title_from_history({ generate: true });
+      await this.touch_metadata();
+      return message;
+    }
+
+    // query 是 parts 数组，直接用其构造 user 消息
+    const parts = Array.isArray(query) ? query : [];
+    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",
       },
-    }) as SessionUserMessageV1;
+    };
     await this.executor.appendUserMessage({
       message,
     });

package/src/session/services/SessionTurnService.ts

@@ -7,13 +7,14 @@
  * - metadata、title、assistant 结果持久化等状态副作用交由 SessionStateService 处理。
  */
 
-import { extractTextFromUiMessage } from "@/executor/messages/UIMessageTransformer.js";
+import { extractTextFromParts, extractTextFromUiMessage } from "@/executor/messages/UIMessageTransformer.js";
 import type { Executor } from "@executor/Executor.js";
 import type { AgentSessionEvent } from "@/types/sdk/AgentSessionEvent.js";
 import type {
   AgentSessionSubscriber,
   AgentSessionUnsubscribe,
 } from "@/types/sdk/AgentSessionEvent.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";
@@ -118,14 +119,11 @@ export class SessionTurnService {
    * 追加一条新的 Session prompt。
    */
   async prompt(input: AgentSessionPromptInput): Promise<AgentSessionTurnHandle> {
-    const query = String(input.query || "").trim();
-    if (!query) {
+    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({
-      query,
-    });
+    return await this.prompt_runtime.prompt(input);
   }
 
   /**
@@ -205,8 +203,10 @@ export class SessionTurnService {
       pendingAssistantFileParts: [],
       ...(input.abortSignal ? { abortSignal: input.abortSignal } : {}),
     };
+    const query = input.promptInput.query;
+    const executor_query = typeof query === "string" ? query : extractTextFromParts(query);
     const result = await this.executor.run({
-      query: input.promptInput.query,
+      query: executor_query,
       runContext: run_context,
     });
     await this.state_service.persist_assistant_result(result.assistantMessage);

package/src/types/sdk/AgentSessionPrompt.ts

@@ -6,16 +6,57 @@
  * - 首条输入、运行中补充输入、排队到下一轮的输入，调用侧都使用同一结构。
  */
 
+import type { UIDataTypes, UITools, UIMessagePart } from "ai";
+
+/**
+ * Session user message part 类型。
+ *
+ * 说明（中文）
+ * - 与 AI SDK 标准 `UIMessagePart` 等价，但已固定泛型参数，避免调用侧引入 `ai` 包。
+ * - 可直接用于 `session.prompt({ query: [...parts] })` 传入 text parts、file parts 等。
+ */
+export type SessionUserMessagePart = UIMessagePart<UIDataTypes, UITools>;
+
 /**
  * Session prompt 输入。
  */
 export interface AgentSessionPromptInput {
   /**
-   * 当前这次要追加到 Session 的用户文本。
+   * 当前这次要追加到 Session 的用户文本或 parts 数组。
    *
    * 说明（中文）
-   * - 调用侧永远只传“新的用户输入”。
+   * - 支持两种格式：
+   *   1. `string`：纯文本用户输入，Session 会将其包装为 `role="user"` 的 UIMessage。
+   *   2. `SessionUserMessagePart[]`：AI SDK 标准 user message parts 数组，可直接携带 text parts、file parts 等。
+   * - 调用侧永远只传"新的用户输入"。
    * - 它是否并入当前 turn，还是排到下一 turn，由 Session 内部决定。
    */
-  query: string;
+  query: string | SessionUserMessagePart[];
+}
+
+/**
+ * 判断 prompt 输入是否为空。
+ *
+ * 说明（中文）
+ * - `string`：trim 后为空即视为空。
+ * - `SessionUserMessagePart[]`：数组为空或仅包含空文本时视为空。
+ */
+export function isAgentSessionPromptInputEmpty(input: AgentSessionPromptInput): boolean {
+  const query = input.query;
+  if (typeof query === "string") {
+    return query.trim() === "";
+  }
+  if (Array.isArray(query)) {
+    if (query.length === 0) {
+      return true;
+    }
+    // 如果所有 parts 都是空文本，也视为空
+    return query.every((part) => {
+      if (part && typeof part === "object" && "type" in part && part.type === "text") {
+        return String(part.text ?? "").trim() === "";
+      }
+      return false; // 非文本 part（如 file）不算空
+    });
+  }
+  return true;
 }