nolo-cli 0.1.45 → 0.1.46

package/ai/agent/_executeModel.ts

@@ -1,146 +0,0 @@
-import { RootState } from "../../app/store";
-import { Agent, Message } from "../../app/types";
-import { selectCurrentDialogConfig } from "../../chat/dialog/dialogSlice";
-import { read } from "../../database/dbSlice";
-import { fetchAgentContexts } from "../agent/fetchAgentContexts";
-import { filterAndCleanMessages } from "../../integrations/openai/filterAndCleanMessages";
-import { selectAllMsgs } from "../../chat/messages/messageSlice";
-import { generateRequestBody } from "../llm/generateRequestBody";
-import { getApiEndpoint } from "../llm/providers";
-import { selectCurrentServer } from "../../app/settings/settingSlice";
-import { selectCurrentToken } from "../../auth/authSlice";
-import { applyChatCompletionsStreamMode } from "../../integrations/openai/chatCompletionStreamMode";
-
-import { sendOpenAICompletionsRequest } from "../chat/sendOpenAICompletionsRequest";
-import { performFetchRequest } from "../chat/fetchUtils";
-import { updateTokensAction } from "../../chat/dialog/actions/updateTokensAction";
-import { extractCustomId } from "../../core/prefix";
-
-export const _executeModel = async (
-  options: {
-    isStreaming: boolean;
-    withAgentContext: boolean;
-    withChatHistory: boolean;
-    agentConfigOverrides?: Record<string, any>;
-  },
-  args: {
-    llmConfig?: Partial<Agent> & Pick<Agent, "provider" | "model">;
-    agentKey?: string;
-    agentConfig?: Partial<Agent> & Pick<Agent, "provider" | "model">;
-    content: any;
-    parentMessageId?: string;
-    billingDialogKey?: string;
-  },
-  thunkApi: any
-) => {
-  const { isStreaming, withAgentContext, withChatHistory, agentConfigOverrides } = options;
-  const { getState, dispatch, rejectWithValue } = thunkApi;
-  const { content } = args;
-  const state = getState() as RootState;
-
-  let agentConfig: Partial<Agent> & Pick<Agent, "provider" | "model">;
-  if (args.llmConfig) {
-    agentConfig = args.llmConfig;
-  } else if (args.agentConfig) {
-    agentConfig = args.agentConfig;
-  } else {
-    const agentKey = args.agentKey || selectCurrentDialogConfig(state)?.cybots?.[0];
-    if (!agentKey) {
-      const msg = "Model execution failed: No llmConfig, agentConfig, or agentKey provided.";
-      console.error(msg);
-      return rejectWithValue(msg);
-    }
-    try {
-      agentConfig = await dispatch(read({ dbKey: agentKey })).unwrap();
-    } catch (error: any) {
-      console.error(`_executeModel failed to load agent [${agentKey}]`, error);
-      return rejectWithValue(error.message);
-    }
-  }
-
-  try {
-    const resolvedConfig = agentConfigOverrides
-      ? { ...agentConfig, ...agentConfigOverrides }
-      : agentConfig;
-    const resolvedAgentConfig = resolvedConfig as Agent;
-    const agentContexts = withAgentContext
-      ? await fetchAgentContexts(resolvedAgentConfig.references, dispatch)
-      : {};
-
-    let messages: Message[];
-    if (withChatHistory) {
-      messages = filterAndCleanMessages(selectAllMsgs(state)) as unknown as Message[];
-      messages.push({ role: "user", content: args.content });
-    } else {
-      messages = [{ role: "user", content: args.content }];
-    }
-
-    const requestBody = generateRequestBody({
-      agentConfig: resolvedAgentConfig,
-      messages,
-      userInput: content,
-      contexts: agentContexts,
-    });
-    const bodyData = applyChatCompletionsStreamMode(requestBody, isStreaming) as any;
-    const currentDialogKey = selectCurrentDialogConfig(state)?.dbKey ?? "";
-
-    if (isStreaming) {
-      await sendOpenAICompletionsRequest({
-        bodyData,
-        agentConfig: resolvedAgentConfig,
-        thunkApi,
-        dialogKey: currentDialogKey,
-        parentMessageId: args.parentMessageId,
-      });
-    } else {
-      const response = await performFetchRequest({
-        agentConfig: resolvedAgentConfig,
-        api: getApiEndpoint(resolvedAgentConfig),
-        bodyData,
-        currentServer: selectCurrentServer(state),
-        token: selectCurrentToken(state),
-      });
-      if (!response.ok) {
-        let message = `Model request failed with HTTP ${response.status}`;
-        try {
-          const errorBody = await response.json();
-          const upstreamMessage =
-            typeof errorBody?.error?.message === "string"
-              ? errorBody.error.message
-              : typeof errorBody?.message === "string"
-                ? errorBody.message
-                : "";
-          if (upstreamMessage.trim()) message = upstreamMessage.trim();
-        } catch {
-          // Keep the status-based message when the response body is not JSON.
-        }
-        throw new Error(message);
-      }
-      const result = await response.json();
-      const content = result?.choices?.[0]?.message?.content;
-      if (typeof content !== "string") {
-        const upstreamMessage =
-          typeof result?.error?.message === "string" ? result.error.message : "";
-        throw new Error(
-          upstreamMessage.trim() ||
-          "Model response missing choices[0].message.content"
-        );
-      }
-      if (args.billingDialogKey && result?.usage) {
-        await updateTokensAction(
-          {
-            dialogId: extractCustomId(args.billingDialogKey),
-            dialogKey: args.billingDialogKey,
-            usage: result.usage,
-            agentConfig: resolvedAgentConfig,
-          },
-          thunkApi
-        );
-      }
-      return content;
-    }
-  } catch (error: any) {
-    console.error(`_executeModel failed`, error);
-    return rejectWithValue(error.message);
-  }
-};

package/ai/agent/agentSlice.ts

@@ -1,557 +0,0 @@
-// 路径: 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";
-import type { FormData as AgentFormData } from "../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;
-  }
-  if ("sharingLevel" in data) {
-    const sharingLevel = (data as any).sharingLevel;
-    if (
-      sharingLevel === "default" ||
-      sharingLevel === "split" ||
-      sharingLevel === "full"
-    ) {
-      changes.sharingLevel = sharingLevel;
-    }
-  }
-
-  // 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("../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("../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("../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;
-
-const agentReducer: (state: AgentState | undefined, action: any) => AgentState = slice.reducer;
-
-export default agentReducer;
-
-/** 读取 SSR 预载的公开 Agent 列表（首页 AI 广场） */
-export const selectSSRPublicAgents = (state: any): Agent[] =>
-  state.cybot?.pubCybots?.data ?? [];