nolo-cli 0.1.40 → 0.1.42

package/ai/tools/agent/agentUpdateShared.ts

@@ -0,0 +1,311 @@
+import type { Agent } from "../../../app/types";
+import type { FormData as AgentFormData } from "../../agent/createAgentSchema";
+import {
+  type AgentUpdateField,
+  AGENT_UPDATE_FIELD_NAMES,
+} from "../../policy/selfUpdateFields";
+import { ToolResultError } from "../../tools/toolResultError";
+
+type ReasoningEffort = "low" | "medium" | "high";
+
+export type ReferenceArg = {
+  dbKey: string;
+  title?: string;
+  type?: "knowledge" | "instruction" | "page";
+};
+
+export type GreetingMenuItemArg = {
+  id: string;
+  label: string;
+  userMessage?: string;
+};
+
+export type GreetingConfigArg = {
+  text?: string;
+  menu?: GreetingMenuItemArg[];
+};
+
+export type AgentUpdateArgsShape = {
+  __confirmedSelfEvolution?: boolean;
+  name?: string;
+  model?: string;
+  provider?: string;
+  prompt?: string;
+  introduction?: string;
+  greeting?: string | GreetingConfigArg;
+  isPublic?: boolean;
+  tags?: string[] | string;
+  tools?: string[];
+  references?: ReferenceArg[];
+  temperature?: number;
+  top_p?: number;
+  frequency_penalty?: number;
+  presence_penalty?: number;
+  max_tokens?: number;
+  reasoning_effort?: ReasoningEffort;
+};
+
+export type UpdateAgentToolArgs = AgentUpdateArgsShape & {
+  agentId: string;
+};
+
+export type UpdateSelfToolArgs = AgentUpdateArgsShape;
+
+export const agentUpdateFieldSchemaProperties = {
+  name: { type: "string", description: "Agent 的新名称。" },
+  model: { type: "string", description: "模型名称，例如 'gpt-4o-mini'。" },
+  provider: { type: "string", description: "模型提供方，例如 'openai'。若不确定可留空。" },
+  prompt: { type: "string", description: "系统提示词，描述 Agent 的角色和行为。" },
+  introduction: { type: "string", description: "面向终端用户的简介文案。" },
+  greeting: {
+    description: "欢迎配置：纯文本字符串，或包含欢迎语 + 菜单的对象。",
+    anyOf: [
+      { type: "string", description: "纯文本欢迎语。" },
+      {
+        type: "object",
+        description: "结构化欢迎配置。",
+        properties: {
+          text: { type: "string", description: "欢迎语文本。" },
+          menu: {
+            type: "array",
+            description: "快捷菜单项。",
+            items: {
+              type: "object",
+              properties: {
+                id: { type: "string", description: "菜单项唯一标识。" },
+                label: { type: "string", description: "按钮显示文案。" },
+                userMessage: { type: "string", description: "点击后发送给 Agent 的消息。" },
+              },
+              required: ["id", "label"],
+            },
+          },
+        },
+      },
+    ],
+  },
+  isPublic: { type: "boolean", description: "是否公开到应用市场。" },
+  tags: {
+    type: "array",
+    items: { type: "string" },
+    description: "标签列表；传空数组 [] 表示清空。",
+  },
+  tools: {
+    type: "array",
+    items: { type: "string" },
+    description: "允许调用的工具名称数组。",
+  },
+  references: {
+    type: "array",
+    description: "引用集合；传空数组 [] 表示清空。",
+    items: {
+      type: "object",
+      properties: {
+        dbKey: { type: "string", description: "引用条目的数据库键。" },
+        title: { type: "string", description: "引用在 UI 中展示的标题。" },
+        type: {
+          type: "string",
+          enum: ["knowledge", "instruction", "page"],
+          description: "引用类型。",
+        },
+      },
+      required: ["dbKey"],
+    },
+  },
+  temperature: { type: "number", description: "采样温度，0~2。" },
+  top_p: { type: "number", description: "nucleus sampling，0~1。" },
+  frequency_penalty: { type: "number", description: "重复惩罚，-2~2。" },
+  presence_penalty: { type: "number", description: "新话题激励，-2~2。" },
+  max_tokens: { type: "number", description: "单次回答最大 token 数。" },
+  reasoning_effort: {
+    type: "string",
+    enum: ["low", "medium", "high"],
+    description: "推理强度。",
+  },
+} as const;
+
+type AgentPatch = Partial<AgentFormData> & {
+  greeting?: string | GreetingConfigArg;
+  tags?: string[] | string;
+  references?: ReferenceArg[];
+};
+
+export const validateUpdateArgs = (
+  userId: string | undefined,
+  options?: { requireAgentId?: boolean; agentId?: string },
+): void => {
+  if (!userId) throw new Error("更新 Agent 失败：当前未登录或缺少 userId。");
+  if (options?.requireAgentId && !options?.agentId?.trim()) {
+    throw new Error("更新 Agent 失败：必须提供非空的 agentId。");
+  }
+};
+
+export const fetchAgentByDbKey = async (
+  dbKey: string,
+  db: any,
+): Promise<Agent | undefined> => {
+  if (!db) return undefined;
+  return db.get(dbKey).catch(() => undefined);
+};
+
+export const extractAgentId = (dbKey: string): string => {
+  const parts = dbKey.trim().split("-");
+  return parts.length >= 3 ? parts[parts.length - 1] : dbKey.trim();
+};
+
+export const buildPatch = (args: AgentUpdateArgsShape): AgentPatch => {
+  const patch: AgentPatch = {};
+  const {
+    name,
+    model,
+    provider,
+    prompt,
+    introduction,
+    greeting,
+    isPublic,
+    tags,
+    tools,
+    references,
+    temperature,
+    top_p,
+    frequency_penalty,
+    presence_penalty,
+    max_tokens,
+    reasoning_effort,
+  } = args;
+
+  if (name !== undefined) patch.name = String(name).trim();
+  if (model !== undefined) patch.model = String(model).trim();
+  if (provider !== undefined) patch.provider = String(provider).trim();
+  if (prompt !== undefined) patch.prompt = prompt;
+  if (introduction !== undefined) patch.introduction = introduction;
+  if (greeting !== undefined) patch.greeting = greeting;
+  if (isPublic !== undefined) patch.isPublic = isPublic;
+  if (tags !== undefined) patch.tags = tags as any;
+  if (tools !== undefined) patch.tools = tools ?? [];
+  if (references !== undefined) patch.references = references as any;
+  if (temperature !== undefined) patch.temperature = temperature;
+  if (top_p !== undefined) patch.top_p = top_p;
+  if (frequency_penalty !== undefined) patch.frequency_penalty = frequency_penalty;
+  if (presence_penalty !== undefined) patch.presence_penalty = presence_penalty;
+  if (max_tokens !== undefined) patch.max_tokens = max_tokens;
+  if (reasoning_effort !== undefined) patch.reasoning_effort = reasoning_effort;
+
+  return patch;
+};
+
+const AGENT_UPDATE_FIELD_NAME_SET = new Set<string>(AGENT_UPDATE_FIELD_NAMES);
+
+export const listRequestedFields = (
+  args: Record<string, unknown>,
+): AgentUpdateField[] =>
+  Object.keys(args).filter(
+    (key): key is AgentUpdateField =>
+      key !== "agentId" &&
+      key !== "__confirmedSelfEvolution" &&
+      AGENT_UPDATE_FIELD_NAME_SET.has(key) &&
+      args[key] !== undefined,
+  );
+
+export const assertAgentUpdateConfirmation = ({
+  scope,
+  requestedFields,
+  confirmed,
+  autoApprovedFields = [],
+}: {
+  scope: "self" | "generic";
+  requestedFields: AgentUpdateField[];
+  confirmed?: boolean;
+  autoApprovedFields?: AgentUpdateField[];
+}): void => {
+  if (requestedFields.length === 0 || confirmed) return;
+
+  const allowedFields =
+    scope === "self"
+      ? requestedFields.filter((field) => autoApprovedFields.includes(field))
+      : [];
+  const blockedFields =
+    scope === "self"
+      ? requestedFields.filter((field) => !autoApprovedFields.includes(field))
+      : requestedFields;
+
+  if (blockedFields.length === 0) return;
+
+  const displayPrefix = scope === "self" ? "updateSelf" : "updateAgent";
+  throw new ToolResultError(
+    scope === "self"
+      ? `当前自我更新字段需要先确认：${blockedFields.join(", ")}`
+      : `当前通用 Agent 更新需要先确认：${blockedFields.join(", ")}`,
+    {
+      code: "agent_update_requires_confirmation",
+      retryable: false,
+      displayData: `${displayPrefix} 需要先确认：${blockedFields.join(", ")}`,
+      rawData: {
+        error: "agent_update_requires_confirmation",
+        message:
+          scope === "self"
+            ? "当前自我更新包含需要用户确认的字段。"
+            : "当前通用 Agent 更新默认要求用户确认。",
+        policy: {
+          capability: scope === "self" ? "self_update" : "agent_update",
+          scope,
+          decision: "ask",
+          requestedFields,
+          allowedFields,
+          blockedFields,
+          autoApprovedFields,
+        },
+      },
+    },
+  );
+};
+
+export const buildRawDataWithUpdateInfo = (
+  agent: Agent,
+  previousAgent: Agent,
+  requestedFields: AgentUpdateField[],
+) => {
+  const changes: Record<string, { o: any; n: any }> = {};
+  for (const key of requestedFields) {
+    const newVal = (agent as any)[key];
+    const oldVal = (previousAgent as any)[key];
+    if (JSON.stringify(newVal) !== JSON.stringify(oldVal)) {
+      changes[key] = { o: oldVal, n: newVal };
+    }
+  }
+
+  return {
+    ...agent,
+    _isUpdate: true,
+    _changes: Object.keys(changes).length > 0 ? changes : undefined,
+  };
+};
+
+export const formatUpdatedAgentOutput = (agent: Agent): string =>
+  [
+    `✅ 已更新 Agent：${agent.name ?? "(名称未变)"}`,
+    `- ID: ${agent.id}`,
+    `- 是否公开: ${agent.isPublic ? "是" : "否"}`,
+    agent.model && `- 模型: ${agent.model}`,
+    (agent as any).provider && `- Provider: ${(agent as any).provider}`,
+    agent.tags?.length && `- 标签: ${agent.tags.join(", ")}`,
+    typeof agent.temperature === "number" && `- temperature: ${agent.temperature}`,
+  ]
+    .filter(Boolean)
+    .join("\n");
+
+export const buildUpdateThunkPreviousAgent = (
+  previousAgent: Agent,
+  userId: string,
+): Agent => {
+  const ownerUserId = String(previousAgent.userId ?? "").trim();
+  if (previousAgent.isPublic && ownerUserId !== userId) {
+    return {
+      ...previousAgent,
+      isPublic: false,
+    };
+  }
+  return previousAgent;
+};
+
+// TODO: Persist lightweight audit events for updateSelf/updateAgent so version history
+// and future rollback can inspect who changed what without rebuilding it from dialogs.

package/ai/tools/agent/callAgentTool.ts

@@ -0,0 +1,139 @@
+// 文件路径: ai/tools/agent/callAgentTool.ts
+
+import type { RootState } from "../../../app/store";
+import { runAgent } from "../../agent/agentSlice";
+import { runAgentBackground } from "../../agent/runAgentBackground";
+
+
+/**
+ * callAgent 工具 Schema
+ *
+ * 用于开展一个子对话（sub-dialog），只关心执行结果。支持两种执行模式：
+ * - "server"（默认）：通过服务端 agent run 接口后台异步执行，结果写回 DB 并通过 SSE 回传，
+ *   适合大多数子任务，也能复用服务端 runtime auto-routing。
+ * - "client"：客户端同步执行，在当前 Redux 上下文中直接调用 runAgent，
+ *   适合确认只需本地轻量执行的子任务。
+ *
+ * 若需要用户立刻看到流式回复，请改用 runStreamingAgent。
+ */
+export const callAgentFunctionSchema = {
+    name: "callAgent",
+    description:
+        "调用一个指定的 Agent 执行一次子任务并返回结果。" +
+        "通常用于：将复杂子问题委托给其他 Agent，例如自动评测、多 Agent 对比、抓取结果的结构化处理等。",
+    parameters: {
+        type: "object",
+        properties: {
+            agentKey: {
+                type: "string",
+                description:
+                    "要调用的 Agent 的唯一 ID。",
+            },
+            task: {
+                type: "string",
+                description:
+                    "委托给该 Agent 的子任务描述（自然语言）。建议在此包含必要的上下文说明（例如当前表结构、题库说明等）。",
+            },
+            input: {
+                description:
+                    "可选。JSON 或字符串，将作为本次子任务的附加输入。通常用于传递抓取到的原始数据、题库、上下文片段等。",
+            },
+            mode: {
+                type: "string",
+                enum: ["client", "server"],
+                description:
+                    "执行模式：'server'（默认）为服务端后台异步执行，能复用 runtime auto-routing；" +
+                    "'client' 为客户端同步执行，适合明确只需本地轻量完成的任务。",
+                default: "server",
+            },
+            serverBase: {
+                type: "string",
+                description:
+                    "可选。目标 Agent 所在的 nolo server origin，例如 Windows 机器通过 Cloudflare 暴露的 https://win.example.com。" +
+                    "仅 server 模式使用；跨域目标必须由服务端 AGENT_TOOL_ALLOWED_SERVER_BASES 明确放行。" +
+                    "如果目标 Agent 记录声明了 delegation.serverBase / runtimeServerBase，服务端会在未显式传入时自动路由。",
+            },
+        },
+        required: ["agentKey", "task"],
+    },
+};
+
+interface CallAgentArgs {
+    agentKey: string;
+    task: string;
+    input?: any;
+    mode?: "client" | "server";
+    serverBase?: string;
+}
+
+/**
+ * callAgent 工具执行函数
+ * @param args - { agentKey, task, input, mode }
+ * @param thunkApi - Redux thunkApi
+ * @param context - { parentMessageId }，用于关联父消息
+ */
+export async function callAgentFunc(
+    args: CallAgentArgs,
+    thunkApi: any,
+    context?: { parentMessageId: string }
+): Promise<{ rawData: any; displayData?: string }> {
+    const { agentKey, task, input, mode = "server", serverBase } = args;
+    const { dispatch, getState } = thunkApi;
+    const state = getState() as RootState;
+
+    if (!agentKey) {
+        throw new Error("callAgent: 缺少 agentKey 参数。");
+    }
+    if (!task || typeof task !== "string") {
+        throw new Error("callAgent: 缺少有效的 task 文本描述。");
+    }
+
+    // 组装 content：统一用「指令 + 输入」的简单文本协议
+    let content: string;
+    if (input === undefined || input === null) {
+        content = task;
+    } else if (typeof input === "string") {
+        content = `${task}\n\n--- INPUT (text) ---\n${input}`;
+    } else {
+        const jsonStr = JSON.stringify(input, null, 2);
+        content = `${task}\n\n--- INPUT (json) ---\n${jsonStr}`;
+    }
+
+    const parentMessageId = context?.parentMessageId;
+
+    try {
+        if (mode === "server") {
+            // 服务端后台执行：适合耗时/计算密集型任务
+            // 通过 /api/agent/run?background=true 派发到服务端，SSE 监听结果
+            const result = await dispatch(
+                runAgentBackground({
+                    agentKey,
+                    userInput: content,
+                    ...(serverBase ? { serverBase } : {}),
+                })
+            ).unwrap();
+
+            return {
+                rawData: result.content ?? result,
+                displayData: `✅ callAgent(服务端) 执行完成，dialogId: ${result.dialogId}`,
+            };
+        }
+
+        // client 模式（默认）：客户端同步执行，结果直接返回给调用方
+        const result = await dispatch(
+            runAgent({
+                agentKey,
+                content,
+                parentMessageId,
+            })
+        ).unwrap();
+
+        return {
+            rawData: result,
+            displayData: "✅ callAgent(客户端) 子任务执行完成。",
+        };
+    } catch (e: any) {
+        const msg = e?.message || String(e);
+        throw new Error(`callAgent 调用 Agent [${agentKey}] 时出错: ${msg}`);
+    }
+}