nolo-cli 0.1.39 → 0.1.41

package/ai/agent/toolFailureGuard.ts

@@ -0,0 +1,99 @@
+import { canonicalizeToolName } from "../tools/toolNameAliases";
+
+export interface ToolFailureGuardState {
+  signature: string | null;
+  count: number;
+}
+
+export interface ToolFailureGuardToolCall {
+  function: {
+    name: string;
+    arguments?: unknown;
+  };
+}
+
+function parseToolResult(content: string | null | undefined): any {
+  if (typeof content !== "string" || !content.trim()) return null;
+  try {
+    return JSON.parse(content);
+  } catch {
+    return null;
+  }
+}
+
+function pickString(value: unknown): string | null {
+  return typeof value === "string" && value.trim() ? value.trim() : null;
+}
+
+function getToolError(parsed: any): string | null {
+  if (!parsed || typeof parsed !== "object") return null;
+  const error = pickString(parsed.error);
+  if (error) return error;
+  const message = pickString(parsed.message);
+  if (message && (parsed.ok === false || parsed.success === false || parsed.status === "error")) {
+    return message;
+  }
+  if (parsed.ok === false) return "tool returned ok=false";
+  if (parsed.success === false) return "tool returned success=false";
+  return null;
+}
+
+function stableStringify(value: unknown): string {
+  if (Array.isArray(value)) {
+    return `[${value.map((item) => stableStringify(item)).join(",")}]`;
+  }
+  if (value && typeof value === "object") {
+    return `{${Object.entries(value as Record<string, unknown>)
+      .sort(([left], [right]) => left.localeCompare(right))
+      .map(([key, item]) => `${JSON.stringify(key)}:${stableStringify(item)}`)
+      .join(",")}}`;
+  }
+  return JSON.stringify(value);
+}
+
+function normalizeToolArgumentsForFailureSignature(rawArguments: unknown): string {
+  if (typeof rawArguments !== "string") return stableStringify(rawArguments ?? null);
+  const trimmed = rawArguments.trim();
+  if (!trimmed) return "";
+  try {
+    return stableStringify(JSON.parse(trimmed));
+  } catch {
+    return trimmed;
+  }
+}
+
+export function recordConsecutiveToolFailure(
+  guard: ToolFailureGuardState,
+  toolCall: ToolFailureGuardToolCall,
+  toolResult: string,
+  maxConsecutiveFailures = 3,
+): string | null {
+  const parsed = parseToolResult(toolResult);
+  const error = getToolError(parsed);
+  if (!error) {
+    guard.signature = null;
+    guard.count = 0;
+    return null;
+  }
+
+  const toolName = canonicalizeToolName(toolCall.function.name);
+  const signature = stableStringify({
+    toolName,
+    arguments: normalizeToolArgumentsForFailureSignature(toolCall.function.arguments),
+    error,
+  });
+  if (signature === guard.signature) {
+    guard.count += 1;
+  } else {
+    guard.signature = signature;
+    guard.count = 1;
+  }
+
+  if (guard.count < maxConsecutiveFailures) return null;
+
+  return [
+    `已停止工具循环：${toolName} 使用同一组参数连续 ${guard.count} 次返回同一个错误。`,
+    `错误：${error}`,
+    "请先调整参数、换一种操作路径，或向用户确认后再继续；不要继续重复同一个失败调用。",
+  ].join("\n");
+}

package/ai/agent/types.ts

@@ -0,0 +1,73 @@
+// 文件路径: ai/agent/types.ts
+
+export interface AgentRuntimeOptions {
+    parallelSessionId?: string;
+    parallelBranchId?: string;
+    parallelLabel?: string;
+    parallelIndex?: number;
+
+    /**
+     * 在本次调用中额外开放给 LLM 的工具名称。
+     * 会与 Agent 本身的 tools 数组合并后再传给 LLM。
+     * 名字要和 toolRegistry 中的 schema.name 对齐，例如 "addTableRow"。
+     */
+    extraTools?: string[];
+
+    /**
+     * 当前调用关注的“编辑目标”，用于后续构造 editingContext。
+     * 这里只定义接口，不涉及任何持久化或具体实现。
+     */
+    editingTarget?: {
+        /**
+         * 语义标签，比如: "table" | "page" | "article" | "code" 等。
+         * 未来可以扩展，但不需要改 Prompt 结构。
+         */
+        kind: string;
+        /**
+         * 可选：具体对象的 key，例如表的 dbKey、页面的 pageKey。
+         * 如果不传，后面可以从全局 state 中推断。
+         */
+        key?: string;
+        /**
+         * 可选：用于补充当前编辑对象的人类可读标题。
+         * 适用于对象不在本地 Redux 中、但调用方已知元数据的场景。
+         */
+        title?: string;
+        /**
+         * 可选：调用方预构造的一段简短说明，供 prompt 构造时使用。
+         */
+        summary?: string;
+        /**
+         * 可选：与当前编辑目标相关的轻量元数据。
+         * 例如应用的 framework、url、fileNames 等。
+         */
+        metadata?: Record<string, unknown>;
+    };
+
+    /**
+     * 本轮调用的图片生成配置，仅作用于这一轮请求。
+     * 会在调用前与 Agent / Dialog 的默认配置合并后传给 LLM。
+     */
+    imageConfigOverride?: {
+        /** 是否在本轮启用图片输出（不写或 true 表示按默认；false 可强制关闭） */
+        enabled?: boolean;
+        /** 本轮强制切换到另一个图片模型变体，仅作用于本次调用 */
+        imageModelOverride?: string;
+
+        /** 本轮期望的宽高比，不指定则使用上层（Dialog / Agent / 模型）的默认值 */
+        aspectRatio?:
+        | "1:1"
+        | "2:3"
+        | "3:2"
+        | "3:4"
+        | "4:3"
+        | "4:5"
+        | "5:4"
+        | "9:16"
+        | "16:9"
+        | "21:9";
+
+        /** 本轮期望的分辨率大小，不指定则使用上层默认值 */
+        imageSize?: "1K" | "2K" | "4K";
+    };
+}

package/ai/agent/utils/imageOutput.ts

@@ -0,0 +1,37 @@
+import type { Agent } from "../../../app/types";
+import { getModelInfo } from "../../llm/getModelContextWindow";
+import { getPublicImageAgentMode } from "./publicImageAgentMode";
+
+const LEGACY_IMAGE_MODELS = new Set([
+  "google/gemini-3-pro-image-preview",
+  "google/gemini-3.1-flash-image-preview",
+]);
+
+export function supportsImageGeneration(agent: Agent): boolean {
+  if ((agent as any).hasImageOutput === true) return true;
+
+  // Move continuous-agent check before model registry early return
+  if (getPublicImageAgentMode(agent as any) === "continuous") return true;
+
+  const modelName = String(agent.model || "").trim();
+  if (!modelName) return false;
+
+  const modelInfo = getModelInfo(modelName);
+  if (modelInfo) {
+    return !!(modelInfo.hasImageOutput ?? (modelInfo as any).supportsImageOutput);
+  }
+
+  const normalized = modelName.toLowerCase();
+  if (LEGACY_IMAGE_MODELS.has(normalized)) return true;
+
+  if ((agent as any).imageConfig?.enabled && agent.hasVision) return true;
+
+  if (getPublicImageAgentMode(agent as any) === "continuous") return true;
+
+  return (
+    normalized.includes("image-preview") ||
+    normalized.includes("flash-image") ||
+    normalized.includes("seedream") ||
+    normalized.includes("flux.")
+  );
+}

package/ai/agent/utils/publicImageAgentMode.ts

@@ -0,0 +1,26 @@
+import type { PublicImageAgentMode } from "../../../app/types";
+
+export type { PublicImageAgentMode };
+
+export const getPublicImageAgentMode = (
+  agent?: Partial<{
+    imageWorkflow: PublicImageAgentMode;
+    provider: string;
+    model: string;
+    imageConfig: { enabled?: boolean } | null;
+  }>
+): PublicImageAgentMode | null => {
+  if (!agent) return null;
+  if (agent.imageWorkflow) return agent.imageWorkflow;
+  return null;
+};
+
+export const getPublicImageAgentDefaultProfile = (mode: PublicImageAgentMode) => {
+  if (mode === "generate") {
+    return { quality: "medium", size: "1024x1024", outputFormat: "png" } as const;
+  }
+  if (mode === "edit") {
+    return { quality: "medium", size: "auto", outputFormat: "png" } as const;
+  }
+  return { quality: "low", size: "auto", outputFormat: "png" } as const;
+};