nolo-cli 0.1.13 → 0.1.14

package/ai/agent/agentSlice.ts

@@ -1,2 +1,545 @@
+// 路径: ai/agent/agentSlice.ts
+
+import { asyncThunkCreator, buildCreateSlice, type PayloadAction } from "@reduxjs/toolkit";
+import type { Agent, AgentRuntimeBinding, ReferenceItem } from "app/types";
+
+import { write, patch, remove } from "database/dbSlice";
+import { createCybotKey, createAgentKey } from "database/keys";
+import { DataType } from "create/types";
 import { ulid } from "ulid";
-export const createAgent = () => ({ id: ulid() });
+import type { FormData as AgentFormData } from "ai/agent/createAgentSchema";
+
+const createSliceWithThunks = buildCreateSlice({
+  creators: { asyncThunk: asyncThunkCreator },
+});
+
+/** Slice State 定义 */
+interface AgentState {
+  pubCybots: {
+    loading: boolean;
+    error: string | null;
+    data: Agent[];
+  };
+}
+
+/** runLlm 参数（通用 LLM 调用） */
+interface RunLlmArgs {
+  /** 直接传入内置 llmConfig，完全不依赖 Agent 数据 */
+  llmConfig?: Partial<Agent> & Pick<Agent, "provider" | "model">;
+  agentKey?: string;
+  /** 兼容旧调用：直接传入 agentConfig，跳过 DB 读取 */
+  agentConfig?: Partial<Agent> & Pick<Agent, "provider" | "model">;
+  content: unknown;
+  isStreaming?: boolean;
+  parentMessageId?: string;
+  /** 覆盖 Agent 配置中的 system prompt */
+  systemPromptOverride?: string;
+  /** 覆盖 Agent 配置中的工具列表（tool id 数组） */
+  toolsOverride?: string[];
+  billingDialogKey?: string;
+}
+
+/** runAgent 参数（通用 Agent 调用，多轮工具循环） */
+interface RunAgentArgs {
+  agentKey: string;
+  content: unknown;
+  parentMessageId?: string;
+  billingDialogKey?: string;
+}
+
+/** createAgent 参数（新建 Agent） */
+interface CreateAgentArgs {
+  userId: string;
+  formData: AgentFormData; // 已通过表单或 Tool 构造的完整数据
+  spaceId?: string; // 可选的空间 ID，如果提供，则将 Agent 添加到该空间
+}
+
+/** updateAgent 参数（更新 Agent，支持部分字段 patch） */
+interface UpdateAgentArgs {
+  userId: string;
+  agentId: string; // 纯 id（不带 db path 前缀）
+  formData: Partial<AgentFormData>; // 允许只传需要修改的字段
+  previousAgent?: Partial<Agent>; // UI 编辑时可以传，用来保持公共副本同步
+}
+
+const initialState: AgentState = {
+  pubCybots: {
+    loading: false,
+    error: null,
+    data: [],
+  },
+};
+
+const normalizeAgentReferences = (references: any[]): ReferenceItem[] => {
+  if (!Array.isArray(references)) return [];
+  return references.map((ref) => ({
+    dbKey: ref.dbKey || "",
+    title: ref.title || "",
+    type: ref.type === "page" ? "knowledge" : ref.type || "knowledge",
+  }));
+};
+
+/**
+ * 创建场景：表单数据 -> 持久化数据（全量）
+ */
+const processAgentCreateForm = (formData: AgentFormData, userId: string) => {
+  const isPublic = !!formData.isPublic;
+  const machineId = typeof (formData as any).machineId === "string"
+    ? (formData as any).machineId.trim()
+    : "";
+
+  const result: any = {
+    ...formData,
+    // tags: "a, b" -> ["a", "b"]
+    tags: formData.tags
+      ? formData.tags
+        .split(",")
+        .map((s) => s.trim())
+        .filter(Boolean)
+      : [],
+    // 归一化 references
+    references: normalizeAgentReferences(formData.references || []),
+    // 非公开时，强制清空白名单，避免脏数据
+    whitelist: isPublic ? formData.whitelist || [] : [],
+  };
+
+  delete result.machineId;
+  if (formData.apiSource === "cli" && machineId) {
+    const binding: AgentRuntimeBinding = {
+      ...(result.runtimeBinding && typeof result.runtimeBinding === "object"
+        ? result.runtimeBinding
+        : {}),
+      machineId,
+      ownerUserId: userId,
+    };
+    result.runtimeBinding = binding;
+  }
+
+  return result;
+};
+
+/**
+ * 更新场景：部分表单字段 -> patch changes
+ * - 只对传进来的字段做转换 / 归一化
+ * - 未出现在 formData 里的字段一律不修改
+ */
+const processAgentUpdateChanges = (
+  data: Partial<AgentFormData>,
+  userId: string,
+  previousAgent?: Partial<Agent>
+) => {
+  const changes: any = {};
+
+  // 基本字符串字段
+  if ("name" in data) {
+    changes.name = String(data.name ?? "").trim();
+  }
+  if ("model" in data) {
+    changes.model = (data.model ?? "").trim();
+  }
+  if ("provider" in data) {
+    changes.provider = (data.provider ?? "").trim();
+  }
+  if ("prompt" in data) {
+    changes.prompt = (data.prompt ?? "").trim();
+  }
+  if ("introduction" in data) {
+    changes.introduction = (data.introduction ?? "").trim();
+  }
+  if ("customProviderUrl" in data) {
+    changes.customProviderUrl = (data.customProviderUrl ?? "").trim();
+  }
+  if ("apiKey" in data) {
+    changes.apiKey = (data.apiKey ?? "").trim();
+  }
+
+  // 简单标志位
+  if ("hasVision" in data && data.hasVision !== undefined) {
+    changes.hasVision = !!data.hasVision;
+  }
+  if ("apiSource" in data && data.apiSource) {
+    changes.apiSource = data.apiSource;
+  }
+  if ("cliProvider" in data) {
+    changes.cliProvider = (data as any).cliProvider || "";
+  }
+  if ("machineId" in data) {
+    const machineId = String((data as any).machineId ?? "").trim();
+    
+    // Determine effective apiSource: prefer data.apiSource, fall back to previousAgent.apiSource
+    const effectiveApiSource = data.apiSource ?? previousAgent?.apiSource;
+    
+    if (machineId) {
+      // Persist runtimeBinding when effective apiSource is 'cli' or undefined
+      // undefined = tool/legacy partial update without apiSource context → preserve backward compatibility
+      if (effectiveApiSource === "cli" || effectiveApiSource === undefined) {
+        const binding: AgentRuntimeBinding = {
+          ...((changes.runtimeBinding && typeof changes.runtimeBinding === "object")
+            ? changes.runtimeBinding
+            : {}),
+          machineId,
+          ownerUserId: userId,
+        };
+        changes.runtimeBinding = binding;
+      }
+      // If apiSource is explicitly non-cli and machineId is supplied, do not create binding
+    } else {
+      // CLEAR branch: clear runtimeBinding when effectiveApiSource is 'cli' or undefined (backward-compatible)
+      if (effectiveApiSource === "cli" || effectiveApiSource === undefined) {
+        changes.runtimeBinding = null;
+      }
+    }
+  }
+  if ("useServerProxy" in data && data.useServerProxy !== undefined) {
+    changes.useServerProxy = !!data.useServerProxy;
+  }
+
+  // greeting / tools 直接透传
+  if ("greeting" in data) {
+    changes.greeting = data.greeting as any;
+  }
+  if ("tools" in data) {
+    changes.tools = Array.isArray(data.tools) ? data.tools.slice() : [];
+  }
+
+  // 数值字段
+  const numericKeys: (keyof AgentFormData)[] = [
+    "inputPrice",
+    "outputPrice",
+    "temperature",
+    "top_p",
+    "frequency_penalty",
+    "presence_penalty",
+    "max_tokens",
+  ];
+  numericKeys.forEach((key) => {
+    if (key in data) {
+      const raw = data[key];
+      if (raw === undefined || raw === null) {
+        changes[key] = raw as any;
+      } else {
+        const num = Number(raw as any);
+        changes[key] = Number.isNaN(num) ? raw : num;
+      }
+    }
+  });
+
+  // ⚠ 唯一实质改动：reasoning_effort 允许传 null，用于“清空该字段”
+  // 原来是：if ("reasoning_effort" in data && data.reasoning_effort) { ... }
+  // 这样会把 null 吞掉，无法让后端 patch + deepMerge 删除字段。
+  if ("reasoning_effort" in data) {
+    changes.reasoning_effort = (data as any).reasoning_effort;
+  }
+
+  // tags: string 或 string[]
+  if ("tags" in data) {
+    const raw = (data as any).tags;
+    let arr: string[] = [];
+    if (Array.isArray(raw)) {
+      arr = raw.map((s) => String(s || "").trim()).filter(Boolean);
+    } else if (typeof raw === "string") {
+      arr = raw
+        .split(",")
+        .map((s) => s.trim())
+        .filter(Boolean);
+    }
+    changes.tags = arr;
+  }
+
+  // references
+  if ("references" in data) {
+    changes.references = normalizeAgentReferences(
+      ((data.references as any) || []) as any[]
+    );
+  }
+
+  // whitelist + isPublic
+  if ("whitelist" in data) {
+    changes.whitelist = (data.whitelist as string[]) || [];
+  }
+
+  if ("isPublic" in data) {
+    changes.isPublic = !!data.isPublic;
+    if (!changes.isPublic) {
+      // 一旦改为私有，强制清空白名单
+      changes.whitelist = [];
+    }
+  }
+
+  return changes;
+};
+
+export const slice = createSliceWithThunks({
+  // 注意：保持 name = "cybot"，以兼容原有 Redux state 结构
+  name: "cybot",
+  initialState,
+  reducers: (create) => ({
+    /**
+     * SSR 首屏：服务端预取公开 Agent 列表后注入，走 __PRELOADED_STATE__ 链路
+     */
+    setSSRPublicAgents: create.reducer(
+      (state, action: PayloadAction<Agent[]>) => {
+        state.pubCybots.data = Array.isArray(action.payload) ? action.payload : [];
+        state.pubCybots.loading = false;
+        state.pubCybots.error = null;
+      }
+    ),
+
+    /**
+     * 通用 LLM 调用（不带 Agent 上下文 / 历史）
+     */
+    runLlm: create.asyncThunk(async (args: RunLlmArgs, thunkApi) => {
+      const overrides: Record<string, any> = {};
+      if (args.systemPromptOverride !== undefined) overrides.prompt = args.systemPromptOverride;
+      if (args.toolsOverride !== undefined) overrides.tools = args.toolsOverride;
+      const { _executeModel } = await import("ai/agent/_executeModel");
+      return _executeModel(
+        {
+          isStreaming: args.isStreaming ?? false,
+          withAgentContext: false,
+          withChatHistory: false,
+          agentConfigOverrides: Object.keys(overrides).length ? overrides : undefined,
+        },
+        args,
+        thunkApi
+      );
+    }),
+
+    /**
+     * 通用 Agent 调用（带 Agent 上下文，多轮工具循环）
+     *
+     * 使用客户端 runAgentClientLoop：
+     * - 每轮调用 LLM（非流式）
+     * - 遇到 tool_calls 时通过 findToolExecutor 本地执行工具
+     * - 循环直到无工具调用或触发其他运行时停止条件
+     */
+    runAgent: create.asyncThunk(async (args: RunAgentArgs, thunkApi) => {
+      const { runAgentClientLoop } = await import("ai/agent/runAgentClientLoop");
+      const { content: loopContent, toolCallCount } = await runAgentClientLoop(
+        {
+          agentKey: args.agentKey,
+          content: args.content,
+          parentMessageId: args.parentMessageId,
+          billingDialogKey: args.billingDialogKey,
+        },
+        thunkApi
+      );
+      return loopContent;
+    }),
+
+    /**
+     * 聊天轮次流式 Agent 调用
+     */
+    streamAgentChatTurn: create.asyncThunk(async (args: any, thunkApi) => {
+      const { streamAgentChatTurnHandler } = await import("ai/agent/streamAgentChatTurn");
+      return streamAgentChatTurnHandler(args, thunkApi);
+    }),
+
+    /**
+     * 创建 Agent：
+     * - 写入用户私有路径
+     * - 如 isPublic=true，则同时写入公共路径
+     * - 返回完整 Agent 对象（包含 id / meta 字段）
+     */
+    createAgent: create.asyncThunk(
+      async ({ userId, formData, spaceId }: CreateAgentArgs, thunkApi) => {
+        const processed = processAgentCreateForm(formData, userId);
+
+        const now = Date.now();
+        const id = ulid();
+
+        const privateKey = createAgentKey.private(userId, id);
+        const publicKey = createAgentKey.public(id);
+
+        const agent: Agent = {
+          ...(processed as any),
+          id,
+          type: DataType.AGENT,
+          userId,
+          createdAt: now,
+          updatedAt: now,
+          dialogCount: 0,
+          messageCount: 0,
+          tokenCount: 0,
+          spaceId: spaceId, // 记录 spaceId
+        };
+
+        // 写入私有副本
+        await thunkApi
+          .dispatch(
+            write({
+              data: agent,
+              customKey: privateKey,
+            })
+          )
+          .unwrap();
+
+        // 如需公开，再写入公共副本
+        if (agent.isPublic) {
+          await thunkApi
+            .dispatch(
+              write({
+                data: agent,
+                customKey: publicKey,
+              })
+            )
+            .unwrap();
+        }
+
+        return agent;
+      }
+    ),
+
+    /**
+     * 更新 Agent（支持局部字段 patch）：
+     * - patch 私有副本
+     * - 如提供 previousAgent，则同步更新 / 删除公共副本
+     *
+     * 注意：
+     * - Tool 场景下一般不提供 previousAgent，此时只保证私有副本被更新；
+     *   公共副本（应用市场）不做强一致保证。
+     */
+    updateAgent: create.asyncThunk(
+      async (
+        { userId, agentId, formData, previousAgent }: UpdateAgentArgs,
+        thunkApi
+      ) => {
+        const normalizedAgentId = (() => {
+          const raw = agentId.trim();
+          if (raw.startsWith("agent-") || raw.startsWith("cybot-")) {
+            const parts = raw.split("-");
+            if (parts.length >= 3) return parts[parts.length - 1];
+          }
+          return raw;
+        })();
+
+        // 【兼容层】双前缀处理：
+        // 1. 如果 agentId 是以 cybot- 开头，或者 previousAgent.type 是 cybot，说明这是存量旧数据。
+        // 2. 存量数据必须使用 createCybotKey 才能正确定位到数据库中的位置。
+        // 3. 新数据则默认使用 createAgentKey (agent- 前缀)。
+        let privateKey = createAgentKey.private(userId, normalizedAgentId);
+        let publicKey = createAgentKey.public(normalizedAgentId);
+
+        if (agentId.startsWith("cybot-") || previousAgent?.type === "cybot") {
+          privateKey = createCybotKey.private(userId, normalizedAgentId);
+          publicKey = createCybotKey.public(normalizedAgentId);
+        }
+
+        const changes = processAgentUpdateChanges(formData || {}, userId, previousAgent);
+
+        // 1) 检查本地是否存在
+        let localExists = false;
+        try {
+          const { db } = thunkApi.extra as any;
+          const localData = await db.get(privateKey);
+          localExists = !!localData;
+        } catch (e) {
+          // ignore
+        }
+
+        if (localExists) {
+          // 有本地数据，直接 patch
+          await thunkApi
+            .dispatch(
+              patch({
+                dbKey: privateKey,
+                changes,
+              })
+            )
+            .unwrap();
+        } else if (previousAgent) {
+          // 无本地数据，但有 UI 传来的 previousAgent，用 write 回填
+          const merged = {
+            ...previousAgent,
+            ...changes,
+            id: normalizedAgentId,
+            type: previousAgent.type || DataType.AGENT,
+            userId,
+          };
+          await thunkApi
+            .dispatch(
+              write({
+                data: merged,
+                customKey: privateKey,
+              })
+            )
+            .unwrap();
+        } else {
+          // 既无本地也无 previousAgent，尝试标准 path (可能会失败 if remote also fails or path throws)
+          // 但既然到了 update，大概率之前 read 过。
+          // 兜底调用 patch，让 patch 内部去报错
+          await thunkApi
+            .dispatch(
+              patch({
+                dbKey: privateKey,
+                changes,
+              })
+            )
+            .unwrap();
+        }
+
+        // 2) 如提供 previousAgent，则尝试保持公共副本同步
+        if (previousAgent) {
+          const wasPublic = !!previousAgent.isPublic;
+          const hasIsPublicChange = Object.prototype.hasOwnProperty.call(
+            changes,
+            "isPublic"
+          );
+          const nowPublic = hasIsPublicChange
+            ? !!(changes as any).isPublic
+            : wasPublic;
+
+          if (nowPublic) {
+            const mergedPublic: Agent = {
+              ...(previousAgent as any),
+              ...(changes as any),
+              id: normalizedAgentId,
+              type: previousAgent.type || DataType.AGENT,
+              userId,
+            };
+
+            await thunkApi
+              .dispatch(
+                write({
+                  data: mergedPublic,
+                  customKey: publicKey,
+                })
+              )
+              .unwrap();
+          } else if (wasPublic && !nowPublic) {
+            await thunkApi
+              .dispatch(remove(publicKey))
+              .unwrap();
+          }
+        }
+
+        // 3) 返回“私有视角”的最新 Agent（主要给前端本地状态使用）
+        const base = previousAgent ?? ({} as Partial<Agent>);
+        const mergedPrivate: Agent = {
+          ...(base as any),
+          ...(changes as any),
+          id: normalizedAgentId,
+          type: base.type || DataType.AGENT,
+          userId,
+        };
+
+        return mergedPrivate;
+      }
+    ),
+  }),
+});
+
+export const {
+  runLlm,
+  runAgent,
+  streamAgentChatTurn,
+  createAgent,
+  updateAgent,
+  setSSRPublicAgents,
+} = slice.actions;
+
+export default slice.reducer;
+
+/** 读取 SSR 预载的公开 Agent 列表（首页 AI 广场） */
+export const selectSSRPublicAgents = (state: any): Agent[] =>
+  state.cybot?.pubCybots?.data ?? [];

package/ai/agent/appWorkingMemory.ts

@@ -0,0 +1,126 @@
+import type { Message } from "chat/messages/types";
+import type { ToolRun } from "ai/tools/toolRunSlice";
+
+const APP_TOOL_NAMES = new Set([
+  "appList",
+  "appRead",
+  "appDeploy",
+  "appPreflight",
+  "appDelete",
+]);
+
+const contentToText = (content: Message["content"] | undefined): string => {
+  if (!content) return "";
+  if (typeof content === "string") return content;
+  if (Array.isArray(content)) {
+    return content
+      .map((part) => ("text" in part ? part.text ?? "" : ""))
+      .join("\n");
+  }
+  return "";
+};
+
+const parseAppListEntries = (text: string): string[] =>
+  text
+    .split("\n")
+    .map((line) => line.trim())
+    .filter((line) => line.startsWith("- **") && line.includes("(appId:"));
+
+export const buildRecentAppToolMemory = (
+  messages: Message[],
+  toolRuns: ToolRun[],
+): string | null => {
+  const messageById = new Map(messages.map((msg) => [msg.id, msg]));
+  const recentRuns = [...toolRuns]
+    .filter((run) => APP_TOOL_NAMES.has(run.toolName))
+    .sort((a, b) => (b.startedAt ?? 0) - (a.startedAt ?? 0))
+    .slice(0, 8);
+
+  if (recentRuns.length === 0) return null;
+
+  const lines: string[] = [
+    "这些信息来自当前对话里最近的 app 工具调用，不依赖右侧编辑态。",
+  ];
+
+  const latestDeployLike = recentRuns.find((run) =>
+    ["appDeploy", "appRead", "appPreflight"].includes(run.toolName),
+  );
+  if (latestDeployLike) {
+    const input = latestDeployLike.input ?? {};
+    const parts = [
+      `- 最近一次关键 app 操作: ${latestDeployLike.toolName}`,
+      typeof input.appId === "string" ? `appId=${input.appId}` : null,
+      typeof input.name === "string" ? `name=${input.name}` : null,
+      typeof input.framework === "string" ? `framework=${input.framework}` : null,
+    ].filter(Boolean);
+    lines.push(parts.join("，"));
+  }
+
+  const relatedMessages = recentRuns
+    .map((run) => messageById.get(run.messageId))
+    .filter((msg): msg is Message => !!msg);
+
+  const appListMessage = relatedMessages.find((msg) => msg.toolName === "appList");
+  if (appListMessage) {
+    const entries = parseAppListEntries(contentToText(appListMessage.content)).slice(0, 5);
+    if (entries.length > 0) {
+      lines.push("- 最近一次 appList 结果：");
+      lines.push(...entries.map((entry) => `  ${entry}`));
+    }
+  }
+
+  const latestReadMessage = relatedMessages.find((msg) => msg.toolName === "appRead");
+  if (latestReadMessage) {
+    const text = contentToText(latestReadMessage.content);
+    const appId = text.match(/- appId:\s*(.+)/)?.[1]?.trim();
+    const url = text.match(/- 访问地址:\s*(.+)/)?.[1]?.trim();
+    if (appId || url) {
+      lines.push(
+        [
+          "- 最近一次 appRead 真值:",
+          appId ? `appId=${appId}` : null,
+          url ? `url=${url}` : null,
+        ]
+          .filter(Boolean)
+          .join("，"),
+      );
+    }
+  }
+
+  const latestPreflightRun = recentRuns.find((run) => run.toolName === "appPreflight");
+  if (latestPreflightRun) {
+    lines.push(
+      [
+        "- 最近一次 appPreflight:",
+        latestPreflightRun.status === "failed" ? "失败" : "已执行",
+        latestPreflightRun.outputSummary ? `摘要=${latestPreflightRun.outputSummary}` : null,
+      ]
+        .filter(Boolean)
+        .join("，"),
+    );
+  }
+
+  const latestDeployMessage = relatedMessages.find((msg) => msg.toolName === "appDeploy");
+  if (latestDeployMessage) {
+    const text = contentToText(latestDeployMessage.content);
+    const appId = text.match(/- appId:\s*(.+)/)?.[1]?.trim();
+    const url = text.match(/- 访问地址:\s*(.+)/)?.[1]?.trim();
+    if (appId || url) {
+      lines.push(
+        [
+          "- 最近一次 appDeploy 结果:",
+          appId ? `appId=${appId}` : null,
+          url ? `url=${url}` : null,
+        ]
+          .filter(Boolean)
+          .join("，"),
+      );
+    }
+  }
+
+  lines.push(
+    "- 如果用户说“刚才那个 app / 那个网站”，优先把它理解为上面最近一次被读取、预检或部署的应用；若仍有歧义，再用 appList 或 appRead 确认。",
+  );
+
+  return lines.join("\n");
+};

package/ai/agent/avatarUtils.ts

@@ -0,0 +1,24 @@
+// packages/ai/agent/avatarUtils.ts
+// Resolves an agent's avatarFileId → a displayable URL
+
+import { buildDatabaseFileContentUrl } from "database/fileUrl";
+
+/**
+ * Returns a URL for displaying the agent avatar, or null.
+ * - If avatarFileId is an http(s) URL → use directly
+ * - Otherwise build server file-content URL
+ */
+export function resolveAvatarUrl(
+  avatarFileId: string | undefined | null,
+  server: string | undefined | null
+): string | null {
+  if (!avatarFileId) return null;
+  if (avatarFileId.startsWith("http") || avatarFileId.startsWith("blob:")) {
+    return avatarFileId;
+  }
+  if (avatarFileId.startsWith("/")) {
+    if (!server) return avatarFileId;
+    return `${server}${avatarFileId}`;
+  }
+  return buildDatabaseFileContentUrl(server, avatarFileId);
+}