deepcode-ai 1.1.29 → 1.1.31

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 {
@@ -3129,21 +3132,21 @@ function resolveTurnStrategy(input, mode, policy) {
       intent
     };
   }
-  if (mode === "plan") {
+  if (intent.kind === "direct_utility") {
     return {
       allowTools: true,
       shouldPlan: false,
-      systemPrompt: PLAN_SYSTEM_PROMPT,
-      kind: "task",
+      systemPrompt: UTILITY_SYSTEM_PROMPT,
+      kind: "utility",
       intent
     };
   }
-  if (intent.kind === "direct_utility") {
+  if (mode === "plan") {
     return {
       allowTools: true,
       shouldPlan: false,
-      systemPrompt: UTILITY_SYSTEM_PROMPT,
-      kind: "utility",
+      systemPrompt: PLAN_SYSTEM_PROMPT,
+      kind: "task",
       intent
     };
   }
@@ -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))
@@ -7750,6 +7816,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 +10017,7 @@ async function createRuntime(options) {
       tools.register(tool);
     }
   }
+  tools.register(createToolSearchTool(tools));
   const agent = new Agent(
     providers,
     tools,
@@ -28163,7 +28269,7 @@ function parseVersion2(version) {
   if (!match) return null;
   return [Number(match[1]), Number(match[2]), Number(match[3])];
 }
-var VERSION = "1.1.29".length > 0 ? "1.1.29" : "0.0.0-dev";
+var VERSION = "1.1.31".length > 0 ? "1.1.31" : "0.0.0-dev";
 var updateCommand = {
   name: "update",
   description: "Check published DeepCode versions",