deepcode-ai 1.1.30 → 1.1.32

package/dist/index.js

@@ -2256,6 +2256,9 @@ import { Effect as Effect9 } from "effect";
 import { z as z10 } from "zod";
 import { Effect as Effect10 } from "effect";
 import { z as z11 } from "zod";
+import { Effect as Effect11 } from "effect";
+import { z as z12 } from "zod";
+import { zodToJsonSchema as zodToJsonSchema2 } from "zod-to-json-schema";
 import path14 from "path";
 import { readdir as readdir5, lstat as lstat3 } from "fs/promises";
 var DeepCodeError = class extends Error {
@@ -3682,24 +3685,25 @@ Execute this task using the available tools. Return a summary of what was done.`
    */
   async executeTaskWithLLM(prompt, session, mode, options, taskType) {
     const taskPrompt = this.createInternalPromptMessage(prompt);
-    const allowedToolNames = this.allowedToolNamesForTaskType(mode, taskType);
+    const allowedToolNames = this.allowedToolNamesForTaskType(mode, taskType, this.getRevealedTools(session));
     const resolvedModel = session.model ?? resolveConfiguredModelForProvider(this.config, session.provider);
     const toolProfile = resolveModelExecutionProfile(session.provider, resolvedModel);
-    const toolDefinitions = this.toolDefinitionsForNames(allowedToolNames, toolProfile.toolSchemaMode);
-    const textToolFallbackEnabled = toolDefinitions.length > 0 && toolProfile.toolCallStrategy !== "native";
     const maxTaskIterations = 10;
     let taskIterations = 0;
     let finalAssistantText = "";
     while (taskIterations < maxTaskIterations) {
       taskIterations++;
       this.enforceBudget(session.id);
+      const toolDefinitions = this.toolDefinitionsForNames(allowedToolNames, toolProfile.toolSchemaMode);
+      const textToolFallbackEnabled = toolDefinitions.length > 0 && toolProfile.toolCallStrategy !== "native";
       const chunks = this.providerManager.chat(
         this.messagesForSystemPrompt(
           session,
           this.systemPromptForMode(mode),
           true,
           [taskPrompt],
-          textToolFallbackEnabled ? buildFallbackToolCallPrompt(allowedToolNames) : void 0
+          textToolFallbackEnabled ? buildFallbackToolCallPrompt(allowedToolNames) : void 0,
+          mode === "build" ? this.buildDeferredToolsHint(session) : void 0
         ),
         {
           preferredProvider: options.provider ?? session.provider,
@@ -3760,7 +3764,19 @@ ${assistantText}` : assistantText;
         break;
       }
       for (const call of toolCalls) {
-        const result = await this.executeTool(call, session, mode, options.signal, allowedToolNames, options.onToolActivity);
+        const result = await this.executeTool(
+          call,
+          session,
+          mode,
+          options.signal,
+          allowedToolNames,
+          options.onToolActivity,
+          (names) => {
+            for (const name of names) {
+              if (this.tools.get(name)) allowedToolNames.add(name);
+            }
+          }
+        );
         this.sessions.addMessage(session.id, {
           role: "tool",
           source: "tool",
@@ -3781,12 +3797,15 @@ ${assistantText}` : assistantText;
     const toolProfile = resolveModelExecutionProfile(session.provider, resolvedModel);
     const baseAllowedToolNames = turnStrategy.allowTools ? this.allowedToolNamesForMode(mode) : /* @__PURE__ */ new Set();
     const allowedToolNames = this.applyToolOverrides(baseAllowedToolNames, options);
-    const toolDefinitions = turnStrategy.allowTools ? this.toolDefinitionsForNames(allowedToolNames, toolProfile.toolSchemaMode) : [];
-    const textToolFallbackEnabled = toolDefinitions.length > 0 && toolProfile.toolCallStrategy !== "native";
+    for (const name of this.getRevealedTools(session)) {
+      if (this.tools.get(name)) allowedToolNames.add(name);
+    }
     while (iterations < maxIterations) {
       iterations += 1;
       options.onIteration?.(iterations, maxIterations);
       this.enforceBudget(session.id);
+      const toolDefinitions = turnStrategy.allowTools ? this.toolDefinitionsForNames(allowedToolNames, toolProfile.toolSchemaMode) : [];
+      const textToolFallbackEnabled = toolDefinitions.length > 0 && toolProfile.toolCallStrategy !== "native";
       await this.compressContextIfNeeded(session, turnStrategy.systemPrompt, options);
       const chunks = this.providerManager.chat(
         this.messagesForSystemPrompt(
@@ -3794,7 +3813,8 @@ ${assistantText}` : assistantText;
           turnStrategy.systemPrompt,
           turnStrategy.allowTools,
           [],
-          textToolFallbackEnabled ? buildFallbackToolCallPrompt(allowedToolNames) : void 0
+          textToolFallbackEnabled ? buildFallbackToolCallPrompt(allowedToolNames) : void 0,
+          mode === "build" ? this.buildDeferredToolsHint(session) : void 0
         ),
         {
           preferredProvider: options.provider ?? session.provider,
@@ -3860,7 +3880,19 @@ ${assistantText}` : assistantText;
       }
       if (toolCalls.length === 0) break;
       for (const call of toolCalls) {
-        const result = await this.executeTool(call, session, mode, options.signal, allowedToolNames, options.onToolActivity);
+        const result = await this.executeTool(
+          call,
+          session,
+          mode,
+          options.signal,
+          allowedToolNames,
+          options.onToolActivity,
+          (names) => {
+            for (const name of names) {
+              if (this.tools.get(name)) allowedToolNames.add(name);
+            }
+          }
+        );
         this.sessions.addMessage(session.id, {
           role: "tool",
           source: "tool",
@@ -3890,7 +3922,7 @@ ${assistantText}` : assistantText;
     }
     return { path: entry.path, restored: true };
   }
-  async executeTool(call, session, mode, signal, allowedToolNames = this.allowedToolNamesForMode(mode), onToolActivity) {
+  async executeTool(call, session, mode, signal, allowedToolNames = this.allowedToolNamesForMode(mode), onToolActivity, onRevealTools) {
     if (!this.isToolAllowed(call.name, mode)) {
       const modeHint = mode === "plan" ? "Switch to BUILD mode (press Tab in the TUI) to enable this tool." : "";
       return {
@@ -3948,6 +3980,12 @@ ${assistantText}` : assistantText;
         const stack = this.undoStacks.get(session.id) ?? [];
         stack.push({ path: filePath, previousContent });
         this.undoStacks.set(session.id, stack);
+      },
+      revealTools: (names) => {
+        const current = this.getRevealedTools(session);
+        session.metadata.revealedTools = [.../* @__PURE__ */ new Set([...current, ...names])];
+        this.sessions.save(session);
+        onRevealTools?.(names);
       }
     };
     try {
@@ -4032,9 +4070,23 @@ ${assistantText}` : assistantText;
   }
   allowedToolNamesForMode(mode) {
     return new Set(
-      this.tools.list().filter((tool) => this.isToolAllowed(tool.name, mode)).map((tool) => tool.name)
+      this.tools.list().filter((tool) => this.isToolAllowed(tool.name, mode) && !tool.deferred).map((tool) => tool.name)
     );
   }
+  getRevealedTools(session) {
+    return Array.isArray(session.metadata.revealedTools) ? session.metadata.revealedTools : [];
+  }
+  buildDeferredToolsHint(session) {
+    const deferred = this.tools.listDeferred();
+    if (deferred.length === 0) return void 0;
+    const revealed = new Set(this.getRevealedTools(session));
+    const unrevealed = deferred.filter((t2) => !revealed.has(t2.name));
+    if (unrevealed.length === 0) return void 0;
+    return [
+      "Deferred tools (not in schema \u2014 call tool_search with a keyword to activate):",
+      ...unrevealed.map((t2) => `- ${t2.name}: ${t2.description.slice(0, 100)}`)
+    ].join("\n");
+  }
   applyToolOverrides(base, options) {
     if (!options.allowedTools && !options.disallowedTools) return base;
     let names = options.allowedTools ? new Set(options.allowedTools) : new Set(base);
@@ -4043,10 +4095,14 @@ ${assistantText}` : assistantText;
     }
     return new Set([...names].filter((n) => base.has(n)));
   }
-  allowedToolNamesForTaskType(mode, taskType) {
+  allowedToolNamesForTaskType(mode, taskType, revealedTools) {
     if (taskType === "research") return /* @__PURE__ */ new Set([...PLAN_ALLOWED_TOOLS]);
     if (taskType === "verify") return /* @__PURE__ */ new Set(["read_file", "list_dir", "analyze_code", "search_text", "bash"]);
-    return this.allowedToolNamesForMode(mode);
+    const base = this.allowedToolNamesForMode(mode);
+    for (const name of revealedTools ?? []) {
+      if (this.tools.get(name)) base.add(name);
+    }
+    return base;
   }
   toolDefinitionsForNames(names, schemaMode = "full") {
     return this.tools.list().filter((tool) => names.has(tool.name)).map((tool) => ({
@@ -4071,7 +4127,7 @@ ${assistantText}` : assistantText;
   systemPromptForMode(mode) {
     return mode === "plan" ? PLAN_SYSTEM_PROMPT : BUILD_SYSTEM_PROMPT;
   }
-  messagesForSystemPrompt(session, systemPrompt, toolsEnabled, extraMessages = [], fallbackToolPrompt) {
+  messagesForSystemPrompt(session, systemPrompt, toolsEnabled, extraMessages = [], fallbackToolPrompt, deferredToolsHint) {
     return [
       {
         id: "mode_system",
@@ -4085,6 +4141,12 @@ ${assistantText}` : assistantText;
         content: this.runtimeContextPrompt(session, toolsEnabled),
         createdAt: session.createdAt
       },
+      ...deferredToolsHint ? [{
+        id: "deferred_tools_system",
+        role: "system",
+        content: deferredToolsHint,
+        createdAt: session.createdAt
+      }] : [],
       ...fallbackToolPrompt ? [{
         id: "tool_fallback_system",
         role: "system",
@@ -4492,6 +4554,9 @@ var ToolRegistry = class {
   list() {
     return [...this.tools.values()];
   }
+  listDeferred() {
+    return [...this.tools.values()].filter((t2) => t2.deferred);
+  }
   descriptions() {
     return this.list().map((tool) => `- ${tool.name}: ${tool.description}`).join("\n");
   }
@@ -4502,6 +4567,7 @@ function adaptMcpTool(client, tool, serverName) {
     name: qualifiedName,
     description: tool.description ?? tool.name,
     parameters: z22.record(z22.unknown()).default({}),
+    deferred: true,
     execute: (args) => Effect3.tryPromise({
       try: () => client.callTool(tool.name, args),
       catch: (e) => e instanceof Error ? e : new Error(String(e))
@@ -6350,6 +6416,7 @@ var ProviderManager = class {
       const isPreferred = providerId === options.preferredProvider;
       const providerModel = isPreferred ? options.model : resolveConfiguredModelForProvider(this.config, providerId);
       if (!isPreferred && !providerModel) continue;
+      validateModelForProvider(providerId, providerModel);
       for (let attempt = 0; attempt <= this.retries; attempt += 1) {
         let emitted = false;
         try {
@@ -6387,6 +6454,7 @@ var ProviderManager = class {
       );
     }
     const model = normalizeProviderModelId(providerId, configuredModel);
+    validateModelForProvider(providerId, model);
     const started = Date.now();
     const controller = new AbortController();
     const timeout = setTimeout(() => controller.abort(), options.timeoutMs ?? 15e3);
@@ -6417,6 +6485,26 @@ var ProviderManager = class {
     }
   }
 };
+var OPENROUTER_SUFFIX_RE = /:(?:free|nitro|extended|beta|floor|online)$/i;
+function validateModelForProvider(providerId, model) {
+  if (!model || providerId === "openrouter") return;
+  if (OPENROUTER_SUFFIX_RE.test(model)) {
+    throw new ProviderError(
+      `Model "${model}" uses an OpenRouter-specific variant suffix (e.g. :free, :nitro). Set provider to "openrouter" instead of "${providerId}".`,
+      providerId
+    );
+  }
+  const slash = model.indexOf("/");
+  if (slash > 0) {
+    const prefix = model.slice(0, slash);
+    if (PROVIDER_IDS.includes(prefix) && prefix !== providerId) {
+      throw new ProviderError(
+        `Model "${model}" belongs to provider "${prefix}" but is being used with "${providerId}". Switch provider to "${prefix}" or use a model native to "${providerId}".`,
+        providerId
+      );
+    }
+  }
+}
 function backoffMs(attempt) {
   return Math.min(250 * 2 ** attempt, 2e3);
 }
@@ -7750,6 +7838,45 @@ function createTaskTool(subagents, worktree) {
     })
   });
 }
+function createToolSearchTool(registry) {
+  return defineTool({
+    name: "tool_search",
+    description: "Search and activate deferred tools (MCP integrations) by name or description keyword. Call this before using a tool that is not in the current schema. Matched tools are revealed and available in subsequent calls this session.",
+    parameters: z12.object({
+      query: z12.string().min(1).describe("Keyword to search in tool names and descriptions")
+    }),
+    execute: (args, context) => Effect11.tryPromise({
+      try: async () => {
+        const query = args.query.toLowerCase();
+        const deferred = registry.listDeferred();
+        if (deferred.length === 0) {
+          return "No deferred tools are configured. Add MCP servers in .deepcode/config.json to enable integrations.";
+        }
+        const matches = deferred.filter(
+          (t2) => t2.name.toLowerCase().includes(query) || t2.description.toLowerCase().includes(query)
+        );
+        if (matches.length === 0) {
+          const available = deferred.map((t2) => `- ${t2.name}: ${t2.description.slice(0, 100)}`).join("\n");
+          return `No deferred tools match "${args.query}".
+
+All available deferred tools:
+${available}`;
+        }
+        context.revealTools?.(matches.map((t2) => t2.name));
+        const schemas = matches.map((t2) => ({
+          name: t2.name,
+          description: t2.description,
+          parameters: zodToJsonSchema2(t2.parameters, { target: "jsonSchema7" })
+        }));
+        return [
+          `Revealed ${matches.length} tool(s) \u2014 available for calls in this session:`,
+          JSON.stringify(schemas, null, 2)
+        ].join("\n\n");
+      },
+      catch: (e) => e instanceof Error ? e : new Error(String(e))
+    })
+  });
+}
 var PROJECT_MARKER2 = ".git";
 var SKIP_DIRS = /* @__PURE__ */ new Set([
   ".git",
@@ -9912,6 +10039,7 @@ async function createRuntime(options) {
       tools.register(tool);
     }
   }
+  tools.register(createToolSearchTool(tools));
   const agent = new Agent(
     providers,
     tools,
@@ -28163,7 +28291,7 @@ function parseVersion2(version) {
   if (!match) return null;
   return [Number(match[1]), Number(match[2]), Number(match[3])];
 }
-var VERSION = "1.1.30".length > 0 ? "1.1.30" : "0.0.0-dev";
+var VERSION = "1.1.32".length > 0 ? "1.1.32" : "0.0.0-dev";
 var updateCommand = {
   name: "update",
   description: "Check published DeepCode versions",
@@ -29800,6 +29928,8 @@ var AppContainer = ({ cwd, config, provider, model, resumeSessionId }) => {
   const lastSubmittedPromptRef = useRef17(null);
   const runStartedAtRef = useRef17(null);
   const streamingResponseLengthRef = useRef17(0);
+  const pendingTextBufferRef = useRef17("");
+  const taskStreamsBufferRef = useRef17({});
   const drainingQueueRef = useRef17(false);
   const messageQueueRef = useRef17([]);
   const sessionShellAllowlistRef = useRef17(/* @__PURE__ */ new Set());
@@ -30076,6 +30206,28 @@ var AppContainer = ({ cwd, config, provider, model, resumeSessionId }) => {
     }, 250);
     return () => clearInterval(interval);
   }, [isRunning]);
+  useEffect27(() => {
+    const id = setInterval(() => {
+      const text = pendingTextBufferRef.current;
+      if (text) {
+        pendingTextBufferRef.current = "";
+        setPendingAssistantText((prev) => prev + text);
+      }
+      const taskBuf = taskStreamsBufferRef.current;
+      const taskKeys = Object.keys(taskBuf);
+      if (taskKeys.length > 0) {
+        taskStreamsBufferRef.current = {};
+        setTaskStreams((prev) => {
+          const next = { ...prev };
+          for (const taskId of taskKeys) {
+            next[taskId] = (next[taskId] ?? "") + taskBuf[taskId];
+          }
+          return next;
+        });
+      }
+    }, 50);
+    return () => clearInterval(id);
+  }, []);
   useEffect27(() => {
     let mounted = true;
     const initialize = async () => {
@@ -30325,6 +30477,8 @@ var AppContainer = ({ cwd, config, provider, model, resumeSessionId }) => {
       historyManager.addItem({ type: "user", text: prompt }, Date.now());
       lastSubmittedPromptRef.current = prompt;
       setPromptSuggestion(null);
+      pendingTextBufferRef.current = "";
+      taskStreamsBufferRef.current = {};
       setPendingAssistantText("");
       setIsRunning(true);
       setIsReceivingContent(false);
@@ -30345,15 +30499,12 @@ var AppContainer = ({ cwd, config, provider, model, resumeSessionId }) => {
           signal: controller.signal,
           onChunk: (text) => {
             streamingResponseLengthRef.current += text.length;
-            setPendingAssistantText((prev) => prev + text);
+            pendingTextBufferRef.current += text;
             setIsReceivingContent(true);
           },
           onChunkForTask: (taskId, text) => {
             streamingResponseLengthRef.current += text.length;
-            setTaskStreams((prev) => ({
-              ...prev,
-              [taskId]: (prev[taskId] ?? "") + text
-            }));
+            taskStreamsBufferRef.current[taskId] = (taskStreamsBufferRef.current[taskId] ?? "") + text;
             setIsReceivingContent(true);
           },
           onUsage: (inputTokens, outputTokens) => {
@@ -30409,6 +30560,8 @@ var AppContainer = ({ cwd, config, provider, model, resumeSessionId }) => {
         );
       } finally {
         abortRef.current = null;
+        pendingTextBufferRef.current = "";
+        taskStreamsBufferRef.current = {};
         setPendingAssistantText("");
         setIsRunning(false);
         setLiveToolCalls([]);