nolo-cli 0.1.23 → 0.1.29

package/ai/agent/guidedCreation/capabilities.ts

@@ -0,0 +1,84 @@
+import type { GuidedAgentCapabilityId } from "./types";
+
+type CapabilityDefinition = {
+  id: GuidedAgentCapabilityId;
+  label: {
+    zhCN: string;
+    en: string;
+  };
+  description: {
+    zhCN: string;
+    en: string;
+  };
+  toolIds: string[];
+};
+
+export const GUIDED_AGENT_CAPABILITIES: Record<
+  GuidedAgentCapabilityId,
+  CapabilityDefinition
+> = {
+  webSearch: {
+    id: "webSearch",
+    label: { zhCN: "联网搜索", en: "Search the web" },
+    description: {
+      zhCN: "需要获取外部网页、资料或最新信息时使用。",
+      en: "Use when the agent needs external webpages, research, or current information.",
+    },
+    toolIds: ["exa_search", "fetchWebpage"],
+  },
+  docs: {
+    id: "docs",
+    label: { zhCN: "读写文档", en: "Read and write docs" },
+    description: {
+      zhCN: "需要读取、创建或更新知识页和文档时使用。",
+      en: "Use when the agent should read, create, or update docs.",
+    },
+    toolIds: ["createDoc", "updateDoc", "read"],
+  },
+  tables: {
+    id: "tables",
+    label: { zhCN: "处理表格", en: "Work with tables" },
+    description: {
+      zhCN: "需要创建表、追加记录或整理结构化数据时使用。",
+      en: "Use when the agent should create tables or manage structured rows.",
+    },
+    toolIds: ["createTable", "addTableRow", "addTableRows", "read"],
+  },
+  agents: {
+    id: "agents",
+    label: { zhCN: "调用其他 AI", en: "Call other AI agents" },
+    description: {
+      zhCN: "需要多 Agent 协作、评审或分工时使用。",
+      en: "Use when the agent should collaborate with other agents.",
+    },
+    toolIds: ["callAgent", "startAgentDialog"],
+  },
+  apps: {
+    id: "apps",
+    label: { zhCN: "创建应用或代码", en: "Create apps or code" },
+    description: {
+      zhCN: "需要生成应用、代码或使用 App Builder 能力时使用。",
+      en: "Use when the agent should build apps, code, or App Builder outputs.",
+    },
+    toolIds: ["appDeploy", "appList", "appRead"],
+  },
+};
+
+export const mapCapabilityIdsToToolIds = (
+  capabilityIds: GuidedAgentCapabilityId[]
+): string[] => {
+  const seen = new Set<string>();
+  const result: string[] = [];
+
+  for (const id of capabilityIds) {
+    const definition = GUIDED_AGENT_CAPABILITIES[id];
+    if (!definition) continue;
+    for (const toolId of definition.toolIds) {
+      if (seen.has(toolId)) continue;
+      seen.add(toolId);
+      result.push(toolId);
+    }
+  }
+
+  return result;
+};

package/ai/agent/guidedCreation/draft.ts

@@ -0,0 +1,62 @@
+import type { FormData as AgentFormData } from "../../agent/createAgentSchema";
+import type { GuidedAgentDraft, GuidedAgentValidationResult } from "./types";
+
+const trim = (value: unknown) =>
+  typeof value === "string" ? value.trim() : "";
+
+export const validateGuidedAgentDraft = (
+  draft: GuidedAgentDraft
+): GuidedAgentValidationResult => {
+  const missing: string[] = [];
+
+  if (!trim(draft.name)) missing.push("name");
+  if (!trim(draft.model)) missing.push("model");
+  if (!trim(draft.provider)) missing.push("provider");
+  if (!trim(draft.prompt)) missing.push("prompt");
+  if (!trim(draft.introduction)) missing.push("introduction");
+
+  return missing.length === 0 ? { ok: true } : { ok: false, missing };
+};
+
+export const mergeGuidedAgentDraft = (
+  current: GuidedAgentDraft,
+  next: Partial<GuidedAgentDraft>
+): GuidedAgentDraft => ({
+  ...current,
+  ...next,
+  capabilityIds: next.capabilityIds ?? current.capabilityIds,
+  toolIds: next.toolIds ?? current.toolIds,
+  references: next.references ?? current.references,
+  tags: next.tags ?? current.tags,
+  unresolved: next.unresolved ?? current.unresolved,
+});
+
+export const buildAgentFormDataFromGuidedDraft = (
+  draft: GuidedAgentDraft
+): AgentFormData => ({
+  name: trim(draft.name),
+  model: trim(draft.model),
+  provider: trim(draft.provider),
+  apiSource: "platform",
+  useServerProxy: true,
+  hasVision: false,
+  prompt: trim(draft.prompt),
+  introduction: trim(draft.introduction),
+  greeting: `你好，我是${trim(draft.name) || "你的 AI 助手"}。你想先让我帮你做什么？`,
+  isPublic: draft.isPublic === true,
+  tags: draft.tags.map((tag) => trim(tag)).filter(Boolean).join(", "),
+  tools: Array.from(new Set(draft.toolIds.filter(Boolean))),
+  references: draft.references
+    .filter((reference) => reference.selected === true)
+    .map((reference) => ({
+      dbKey: reference.dbKey,
+      title: reference.title || "",
+      type: reference.type === "instruction" ? "instruction" : "knowledge",
+    })),
+  linkedSpaces: [],
+  customProviderUrl: "",
+  apiKey: "",
+  inputPrice: 0,
+  outputPrice: 0,
+  whitelist: [],
+});

package/ai/agent/guidedCreation/types.ts

@@ -0,0 +1,38 @@
+import type { ReferenceItem } from "../../../app/types";
+
+export type GuidedAgentCapabilityId =
+  | "webSearch"
+  | "docs"
+  | "tables"
+  | "agents"
+  | "apps";
+
+export type GuidedAgentReferenceChoice = ReferenceItem & {
+  selected: boolean;
+  reason?: string;
+};
+
+export type GuidedAgentDraft = {
+  name: string;
+  introduction: string;
+  prompt: string;
+  promptSummary: string;
+  provider: string;
+  model: string;
+  isPublic: boolean;
+  capabilityIds: GuidedAgentCapabilityId[];
+  toolIds: string[];
+  references: GuidedAgentReferenceChoice[];
+  tags: string[];
+  unresolved: string[];
+};
+
+export type GuidedAgentAssistantMessage = {
+  id: string;
+  role: "user" | "assistant";
+  content: string;
+};
+
+export type GuidedAgentValidationResult =
+  | { ok: true }
+  | { ok: false; missing: string[] };

package/ai/agent/hooks/fetchPublicAgents.ts

@@ -1,10 +1,10 @@
-import { getDb } from "database/client/db";
+import { getDb } from "../../../database/client/db";
 import { pino } from "pino";
-import { pubAgentKeys } from "database/keys";
-import { Agent } from "app/types";
-import { supportsImageGeneration } from "ai/agent/utils/imageOutput";
-import { getRecordTimestamp, isTombstoneRecord } from "database/tombstones";
-import { buildSortMeta, sortAgents } from "ai/agent/utils/sortUtils";
+import { pubAgentKeys } from "../../../database/keys";
+import { Agent } from "../../../app/types";
+import { supportsImageGeneration } from "../../agent/utils/imageOutput";
+import { getRecordTimestamp, isTombstoneRecord } from "../../../database/tombstones";
+import { buildSortMeta, sortAgents } from "../../agent/utils/sortUtils";
 
 const logger = pino({ name: "fetchPublicAgents" });
 

package/ai/agent/hooks/useAgentConfig.ts

@@ -1,60 +1,88 @@
 import { useEffect, useState } from "react";
 
-import type { Agent } from "app/types";
-import { useAppDispatch, useAppSelector } from "app/store";
-import { read } from "database/dbSlice";
+import type { Agent } from "../../../app/types";
+import { useAppDispatch, useAppSelector } from "../../../app/store";
+import { read, selectById } from "../../../database/dbSlice";
+import { selectRuntimeSnapshot } from "../../../app/stateViews/runtime";
 
-import { selectCurrentDialogConfig } from "chat/dialog/dialogSlice";
-import { getPrimaryDialogAgentId } from "chat/dialog/dialogAgents";
+import {
+  selectCurrentDialogConfig,
+  selectCurrentDialogKey,
+} from "../../../chat/dialog/dialogSlice";
+import { getPrimaryDialogAgentId } from "../../../chat/dialog/dialogAgents";
 
 interface UseAgentConfigResult {
   agentConfig: Agent | null;
   isLoading: boolean;
+  loadState: "idle" | "loading" | "ready" | "error";
 }
 
 const useAgentConfig = () => {
   const dispatch = useAppDispatch();
-  const [agentConfig, setAgentConfig] = useState<Agent | null>(null);
-  const [isLoading, setIsLoading] = useState(false);
+  const [loadState, setLoadState] =
+    useState<UseAgentConfigResult["loadState"]>("idle");
 
+  const currentDialogKey = useAppSelector(selectCurrentDialogKey);
   const currentDialogConfig = useAppSelector(selectCurrentDialogConfig);
   const agentKey = getPrimaryDialogAgentId(currentDialogConfig);
+  const { currentToken, currentServer } = useAppSelector(selectRuntimeSnapshot);
+  const agentConfig = useAppSelector((state) =>
+    agentKey ? ((selectById(state, agentKey) as Agent | null) ?? null) : null
+  );
 
   useEffect(() => {
+    if (currentDialogKey && !currentDialogConfig) {
+      setLoadState("loading");
+      return;
+    }
+
     if (!agentKey) {
-      setAgentConfig(null);
-      setIsLoading(false);
+      setLoadState("idle");
+      return;
+    }
+
+    if (agentConfig) {
+      setLoadState("ready");
+      return;
+    }
+
+    if (!currentToken) {
+      setLoadState("loading");
       return;
     }
 
     let cancelled = false;
-    setAgentConfig(null);
-    setIsLoading(true);
+    setLoadState("loading");
 
     void (async () => {
       try {
-        const result = (await dispatch(read({ dbKey: agentKey })).unwrap()) as Agent | null;
+        await dispatch(read({ dbKey: agentKey })).unwrap();
         if (cancelled) return;
-        setAgentConfig(result);
+        setLoadState("ready");
       } catch (error) {
         if (cancelled) return;
         console.warn("[useAgentConfig] Failed to load agent config:", error);
-        setAgentConfig(null);
-      } finally {
-        if (!cancelled) {
-          setIsLoading(false);
-        }
+        setLoadState("error");
       }
     })();
 
     return () => {
       cancelled = true;
     };
-  }, [agentKey, dispatch]);
+  }, [
+    agentConfig,
+    agentKey,
+    currentDialogConfig,
+    currentDialogKey,
+    currentServer,
+    currentToken,
+    dispatch,
+  ]);
 
   return {
-    agentConfig,
-    isLoading,
+    agentConfig: agentConfig ?? null,
+    isLoading: loadState === "loading",
+    loadState,
   } satisfies UseAgentConfigResult;
 };
 

package/ai/agent/hooks/useAgentDialog.ts

@@ -2,7 +2,7 @@
 import { useCallback } from "react";
 import { useTranslation } from "react-i18next";
 import toast from "react-hot-toast";
-import { useCreateDialog } from "chat/dialog/useCreateDialog";
+import { useCreateDialog } from "../../../chat/dialog/useCreateDialog";
 
 interface UseAgentDialogOptions {
   spaceId?: string | null;

package/ai/agent/hooks/useAgentFormValidation.ts

@@ -3,26 +3,26 @@
 import { useForm } from "react-hook-form";
 import { zodResolver } from "@hookform/resolvers/zod";
 import { useCallback } from "react";
-import { useAppDispatch } from "app/store";
-import { useAuth } from "auth/hooks/useAuth";
-import { useCreateDialog } from "chat/dialog/useCreateDialog";
-import { createAgentKey } from "database/keys";
+import { useAppDispatch } from "../../../app/store";
+import { useAuth } from "../../../auth/hooks/useAuth";
+import { useCreateDialog } from "../../../chat/dialog/useCreateDialog";
+import { createAgentKey } from "../../../database/keys";
 import { useTranslation } from "react-i18next";
-import { useAppSelector } from "app/store";
+import { useAppSelector } from "../../../app/store";
 import {
   addContentToSpace,
   selectCurrentSpace,
   selectCurrentSpaceId,
   updateContentTitle,
-} from "create/space/spaceSlice";
+} from "../../../create/space/spaceSlice";
 import {
   getCreateAgentSchema,
   FormData,
   normalizeReferences,
 } from "../createAgentSchema";
-import { createAgent, updateAgent } from "ai/agent/agentSlice";
-import { ContentType } from "app/types";
-import { DEFAULT_MODEL } from "ai/llm/providers";
+import { createAgent, updateAgent } from "../../agent/agentSlice";
+import { ContentType } from "../../../app/types";
+import { DEFAULT_MODEL } from "../../llm/providers";
 
 type AgentEditIdentityInput = Partial<ExtendedFormData> & {
   dbKey?: string;

package/ai/agent/hooks/usePublicAgents.ts

@@ -1,28 +1,28 @@
 // 文件路径: ai/agent/hooks/usePublicAgents.ts
 
 import { useEffect, useState, useCallback, useRef } from "react";
-import { useAppDispatch, useAppSelector } from "app/store";
+import { useAppDispatch, useAppSelector } from "../../../app/store";
 
-import type { Agent } from "app/types";
+import type { Agent } from "../../../app/types";
 import {
   selectCurrentServer,
   selectSyncServers,
-} from "app/settings/settingSlice";
-import { selectUserId } from "auth/authSlice";
+} from "../../../app/settings/settingSlice";
+import { selectUserId } from "../../../auth/authSlice";
 
-import { fetchPublicAgents as fetchLocal } from "ai/agent/hooks/fetchPublicAgents";
-import { remove } from "database/dbSlice";
-import { getAllServers } from "database/actions/common";
+import { fetchPublicAgents as fetchLocal } from "../../agent/hooks/fetchPublicAgents";
+import { remove } from "../../../database/dbSlice";
+import { getAllServers } from "../../../database/actions/common";
 import {
   getRecordTimestamp,
   isTombstoneRecord,
-} from "database/tombstones";
+} from "../../../database/tombstones";
 import {
   buildSortMeta,
   sortAgents,
   toTimeMs,
   MAgent
-} from "ai/agent/utils/sortUtils";
+} from "../../agent/utils/sortUtils";
 
 /* -------------------------------------------------------------------------- */
 /* 类型定义                                                                   */

package/ai/agent/machineRunPermissions.ts

@@ -65,13 +65,31 @@ export function resolveMachineRunPermissionPolicy(agentConfig: any): MachineRunP
 }
 
 const WRITE_INTENT_RE =
-  /\b(write|overwrite|edit|modify|delete|remove|rename|move|create|save|patch|apply patch|mkdir|touch|rm|rmdir|del|erase|copy-item|move-item|remove-item|set-content|add-content|new-item)\b|[\u5199\u5220\u6539\u4FEE\u4FDD\u5B58\u521B\u5EFA\u79FB\u52A8]/i;
+  /\b(write|overwrite|edit|modify|delete|remove|rename|move|create|save|patch|apply patch|mkdir|touch|rm|rmdir|del|erase|copy-item|move-item|remove-item|set-content|add-content|new-item)\b|(\u5199\u5165|\u5199\u6587\u4EF6|\u5220\u9664|\u79FB\u9664|\u91CD\u547D\u540D|\u79FB\u52A8|\u521B\u5EFA|\u65B0\u5EFA|\u4FDD\u5B58|\u4FEE\u6539|\u7F16\u8F91|\u6253\u8865\u4E01|\u5E94\u7528\u8865\u4E01)/i;
 
 const SHELL_INTENT_RE =
-  /\b(run|execute|exec|shell|terminal|powershell|pwsh|bash|zsh|cmd\.exe|npm|bun|node|python|git|curl|wget|ssh|scp|chmod|chown|sudo)\b|[\u8FD0\u884C\u6267\u884C\u547D\u4EE4\u7EC8\u7AEF]/i;
+  /\b(run|execute|exec)\s+(?:a\s+)?(?:shell|command|terminal|powershell|pwsh|bash|zsh|cmd\.exe|npm|bun|node|python|git|curl|wget|ssh|scp|chmod|chown|sudo|tests?|test suite)\b|\b(open|start)\s+(?:a\s+)?(?:shell|terminal)\b|\b(powershell|pwsh|bash|zsh|cmd\.exe|ssh|sudo)\b|(\u8FD0\u884C\u547D\u4EE4|\u6267\u884C\u547D\u4EE4|\u547D\u4EE4\u884C|\u7EC8\u7AEF|\u63A7\u5236\u53F0)/i;
+
+const NEGATION_RE =
+  /\b(do not|don't|dont|never|no need to|without|非目标|不要|不需要|禁止|无需)\b|[\u4E0D\u975E]/i;
+
+function stripNegatedPolicySentences(task: string): string {
+  return task
+    .split(/([\n.;!?。；！？])/)
+    .reduce<string[]>((segments, part, index, parts) => {
+      if (index % 2 === 1) return segments;
+      const separator = parts[index + 1] ?? "";
+      const segment = `${part}${separator}`;
+      const mentionsRestrictedAction = WRITE_INTENT_RE.test(segment) || SHELL_INTENT_RE.test(segment);
+      if (mentionsRestrictedAction && NEGATION_RE.test(segment)) return segments;
+      segments.push(segment);
+      return segments;
+    }, [])
+    .join("");
+}
 
 export function assertMachineRunAllowed(userInput: string, policy: MachineRunPermissionPolicy) {
-  const task = String(userInput || "");
+  const task = stripNegatedPolicySentences(String(userInput || ""));
   if (!policy.allowFilesystemWrite && WRITE_INTENT_RE.test(task)) {
     throw new Error(WRITE_DENIED);
   }

package/ai/agent/persistMessageWithFixedId.ts

@@ -1,7 +1,7 @@
-import { DataType } from "create/types";
-import { write } from "database/dbSlice";
-import { addUserMessage } from "chat/messages/messageSlice";
-import type { Message } from "chat/messages/types";
+import { DataType } from "../../create/types";
+import { write } from "../../database/dbSlice";
+import { addUserMessage } from "../../chat/messages/messageSlice";
+import type { Message } from "../../chat/messages/types";
 
 export async function persistMessageWithFixedId(
   dispatch: any,

package/ai/agent/planSlice.ts

@@ -3,7 +3,7 @@
 // 支持普通步骤和反思步骤（reflect step）
 
 import { type PayloadAction, createSlice } from "@reduxjs/toolkit";
-import type { RootState } from "app/store";
+import type { RootState } from "../../app/store";
 
 // --- Interfaces ---
 

package/ai/agent/referenceUtils.ts

@@ -1,11 +1,11 @@
-import type { AppDispatch } from "app/store";
-import type { ReferenceItem } from "app/types";
-import { read } from "database/dbSlice";
+import type { AppDispatch } from "../../app/store";
+import type { ReferenceItem } from "../../app/types";
+import { read } from "../../database/dbSlice";
 import {
   extractRuntimePageCapabilities,
   resolveSkillGraphFromRoots,
   type SkillRuntimePageLike,
-} from "ai/skills/referenceRuntime";
+} from "../skills/referenceRuntime";
 
 export type ResolvedReferenceAssets = {
   references: ReferenceItem[];

package/ai/agent/runAgentBackground.ts

@@ -12,9 +12,9 @@
 //   }));
 
 import { createAsyncThunk } from "@reduxjs/toolkit";
-import type { RootState } from "app/store";
-import { selectCurrentServer } from "app/settings/settingSlice";
-import { selectCurrentToken } from "auth/authSlice";
+import type { RootState } from "../../app/store";
+import { selectCurrentServer } from "../../app/settings/settingSlice";
+import { selectCurrentToken } from "../../auth/authSlice";
 import { createSSEParser } from "../chat/parseMultilineSSE";
 
 export type DialogStatus = "pending" | "running" | "done" | "failed" | "cancelled" | "reconnecting";

package/ai/agent/runAgentClientLoop.ts

@@ -5,18 +5,18 @@
 // - 工具执行通过 executeToolCall 在客户端完成
 // - 每轮：调 LLM → 有 tool_calls → 执行工具 → 追加 tool 消息 → 继续
 
-import { RootState } from "app/store";
-import { Message } from "app/types";
-import { read } from "database/dbSlice";
-import { fetchAgentContexts } from "ai/agent/fetchAgentContexts";
-import { generateRequestBody } from "ai/llm/generateRequestBody";
-import { getApiEndpoint } from "ai/llm/providers";
-import { selectCurrentServer } from "app/settings/settingSlice";
-import { selectCurrentToken } from "auth/authSlice";
-import { performFetchRequest } from "ai/chat/fetchUtils";
-import { executeToolCall } from "ai/agent/executeToolCall";
-import { extractCustomId } from "core/prefix";
-import { updateTokensAction } from "chat/dialog/actions/updateTokensAction";
+import { RootState } from "../../app/store";
+import { Message } from "../../app/types";
+import { read } from "../../database/dbSlice";
+import { fetchAgentContexts } from "../agent/fetchAgentContexts";
+import { generateRequestBody } from "../llm/generateRequestBody";
+import { getApiEndpoint } from "../llm/providers";
+import { selectCurrentServer } from "../../app/settings/settingSlice";
+import { selectCurrentToken } from "../../auth/authSlice";
+import { performFetchRequest } from "../chat/fetchUtils";
+import { executeToolCall } from "../agent/executeToolCall";
+import { extractCustomId } from "../../core/prefix";
+import { updateTokensAction } from "../../chat/dialog/actions/updateTokensAction";
 
 export interface RunAgentClientLoopArgs {
   agentKey: string;

package/ai/agent/runtimeGuidance.ts

@@ -1,4 +1,4 @@
-import { canonicalizeToolNames } from "ai/tools/toolNameAliases";
+import { canonicalizeToolNames } from "../tools/toolNameAliases";
 import { buildContextLayerContractBlock } from "./contextLayerContract";
 import { buildStartupProtocolBlock } from "./startupProtocol";
 
@@ -54,7 +54,7 @@ const buildWebResearchToolPolicyBlock = (
   return [
     "--- 生产环境网页研究工具策略 ---",
     "生产环境网页研究优先使用 fetchWebpage、站点 Markdown / llms.txt、或专用浏览/搜索工具。",
-    "不要用 execShell/execBash 调 curl、grep、sed 等命令抓网页或截取网页段落；生产环境通常会禁用 dev shell，反复尝试只会浪费回合。",
+    "不要用 execShell 调 curl、grep、sed 等命令抓网页或截取网页段落；生产环境通常会禁用 dev shell，反复尝试只会浪费回合。",
     "如果网页内容过长或锚点段落没有被单独提取，先寻找该文档站提供的 Markdown 版本、独立页面、llms.txt 索引或更具体 URL，再继续回答。",
   ].join("\n");
 };

package/ai/agent/server/fetchPublicAgents.ts

@@ -1,9 +1,9 @@
-import serverDb from "database/server/db";
-import { pubAgentKeys } from "database/keys";
-import { Agent } from "app/types";
-import { sortAgents } from "ai/agent/utils/sortUtils";
-import { supportsImageGeneration } from "ai/agent/utils/imageOutput";
-import { isTombstoneRecord } from "database/tombstones";
+import serverDb from "../../../database/server/db";
+import { pubAgentKeys } from "../../../database/keys";
+import { Agent } from "../../../app/types";
+import { sortAgents } from "../../agent/utils/sortUtils";
+import { supportsImageGeneration } from "../../agent/utils/imageOutput";
+import { isTombstoneRecord } from "../../../database/tombstones";
 
 export interface FetchPublicAgentsOptions {
   limit?: number;
@@ -150,7 +150,12 @@ export async function fetchPublicAgents(
 
   if (searchName) {
     const kw = searchName.toLowerCase();
-    list = list.filter((agent) => agent.name?.toLowerCase().includes(kw));
+    list = list.filter((agent) => {
+      const nameMatch = agent.name?.toLowerCase().includes(kw);
+      const introMatch = agent.introduction?.toLowerCase().includes(kw);
+      const tagMatch = agent.tags?.some((tag) => tag.toLowerCase().includes(kw));
+      return nameMatch || introMatch || tagMatch;
+    });
   }
 
   if (userId) {

package/ai/agent/startParallelAgentStreams.ts

@@ -1,20 +1,20 @@
-import type { RootState } from "app/store";
-import type { DialogConfig } from "app/types";
-import { patch, read, write } from "database/dbSlice";
-import { DataType } from "create/types";
-import { createDialogMessageKeyAndId } from "database/keys";
-import { streamAgentChatTurn } from "ai/agent/agentSlice";
-import { createDialog } from "chat/dialog/dialogSlice";
-import { serializeMessageContent } from "chat/messages/messageContent";
-import { addToolMessage, selectAllMsgs } from "chat/messages/messageSlice";
-import type { Message } from "chat/messages/types";
-import { extractCustomId } from "core/prefix";
+import type { RootState } from "../../app/store";
+import type { DialogConfig } from "../../app/types";
+import { patch, read, write } from "../../database/dbSlice";
+import { DataType } from "../../create/types";
+import { createDialogMessageKeyAndId } from "../../database/keys";
+import { streamAgentChatTurn } from "../agent/agentSlice";
+import { createDialog } from "../../chat/dialog/dialogSlice";
+import { serializeMessageContent } from "../../chat/messages/messageContent";
+import { addToolMessage, selectAllMsgs } from "../../chat/messages/messageSlice";
+import type { Message } from "../../chat/messages/types";
+import { extractCustomId } from "../../core/prefix";
 import {
   calculateParallelBranchPricing,
   summarizeParallelBudget,
   summarizeParallelCosts,
   type ParallelBudgetSummary,
-} from "ai/tools/agent/parallelBudget";
+} from "../tools/agent/parallelBudget";
 
 type ParallelDisplayMode = "inline" | "folded";