feat(sdk): enable code_search in sdk (#254)

Co-authored-by: James Grugett <jahooma@gmail.com>

Author: brandonkachen

bun.lock

@@ -63,6 +63,7 @@ export const publishedTools = [
 ] as const
 
 export type ToolName = (typeof toolNames)[number]
+export type PublishedToolName = (typeof publishedTools)[number]
 
 export type $ToolParams<T extends ToolName = ToolName> = {
   toolName: T

common/src/tools/constants.ts

@@ -23,7 +23,7 @@ import { updateSubgoalParams } from './params/tool/update-subgoal'
 import { webSearchParams } from './params/tool/web-search'
 import { writeFileParams } from './params/tool/write-file'
 
-import type { ToolName, $ToolParams } from './constants'
+import type { ToolName, $ToolParams, PublishedToolName } from './constants'
 import type { ToolMessage } from '../types/messages/codebuff-message'
 import type {
   ToolCallPart,
@@ -125,3 +125,5 @@ export const clientToolNames = clientToolCallSchema.def.options.map(
 export type ClientToolCall<T extends ClientToolName = ClientToolName> = z.infer<
   typeof clientToolCallSchema
 > & { toolName: T } & Omit<ToolCallPart, 'type'>
+
+export type PublishedClientToolName = ClientToolName & PublishedToolName

common/src/tools/list.ts

@@ -58,7 +58,8 @@
     "diff": "8.0.2",
     "zod": "^4.0.0",
     "@vscode/tree-sitter-wasm": "0.1.4",
-    "web-tree-sitter": "0.25.6"
+    "web-tree-sitter": "0.25.6",
+    "@vscode/ripgrep": "1.15.14"
   },
   "devDependencies": {
     "@types/diff": "8.0.0",

sdk/package.json

@@ -4,6 +4,7 @@ import {
   type RunState,
 } from './run-state'
 import { changeFile } from './tools/change-file'
+import { codeSearch } from './tools/code-search'
 import { getFiles } from './tools/read-files'
 import { runTerminalCommand } from './tools/run-terminal-command'
 import { WebSocketHandler } from './websocket-client'
@@ -16,14 +17,18 @@ import { MAX_AGENT_STEPS_DEFAULT } from '../../common/src/constants/agents'
 import { toolNames } from '../../common/src/tools/constants'
 import {
   clientToolCallSchema,
+  PublishedClientToolName,
   type ClientToolCall,
   type ClientToolName,
   type CodebuffToolOutput,
 } from '../../common/src/tools/list'
 
 import type { CustomToolDefinition } from './custom-tool'
 import type { AgentDefinition } from '../../common/src/templates/initial-agents-dir/types/agent-definition'
-import type { ToolName } from '../../common/src/tools/constants'
+import type {
+  PublishedToolName,
+  ToolName,
+} from '../../common/src/tools/constants'
 import type {
   ToolResultOutput,
   ToolResultPart,
@@ -38,7 +43,7 @@ export type CodebuffClientOptions = {
   onError: (error: { message: string }) => void
   overrideTools?: Partial<
     {
-      [K in ClientToolName]: (
+      [K in ClientToolName & PublishedToolName]: (
         input: ClientToolCall<K>['input'],
       ) => Promise<CodebuffToolOutput<K>>
     } & {
@@ -321,7 +326,7 @@ export class CodebuffClient {
     }
 
     try {
-      let override = this.overrideTools[toolName as ClientToolName]
+      let override = this.overrideTools[toolName as PublishedClientToolName]
       if (!override && toolName === 'str_replace') {
         // Note: write_file and str_replace have the same implementation, so reuse their write_file override.
         override = this.overrideTools['write_file']
@@ -337,6 +342,21 @@ export class CodebuffClient {
           ...input,
           cwd: input.cwd ?? this.cwd,
         } as Parameters<typeof runTerminalCommand>[0])
+      } else if (toolName === 'code_search') {
+        result = await codeSearch({
+          projectPath: this.cwd,
+          ...input,
+        } as Parameters<typeof codeSearch>[0])
+      } else if (toolName === 'run_file_change_hooks') {
+        // No-op: SDK doesn't run file change hooks
+        result = [
+          {
+            type: 'json',
+            value: {
+              message: 'File change hooks are not supported in SDK mode',
+            },
+          },
+        ]
       } else {
         throw new Error(
           `Tool not implemented in SDK. Please provide an override or modify your agent to not use this tool: ${toolName}`,

sdk/src/client.ts

@@ -0,0 +1,97 @@
+import { spawn } from 'child_process'
+import * as path from 'path'
+
+import { rgPath } from '@vscode/ripgrep'
+
+import type { CodebuffToolOutput } from '../../../common/src/tools/list'
+
+export function codeSearch({
+  projectPath,
+  pattern,
+  flags,
+  cwd,
+}: {
+  projectPath: string
+  pattern: string
+  flags?: string
+  cwd?: string
+}): Promise<CodebuffToolOutput<'code_search'>> {
+  return new Promise((resolve) => {
+    let stdout = ''
+    let stderr = ''
+
+    const flagsArray = (flags || '').split(' ').filter(Boolean)
+    let searchCwd = projectPath
+    if (cwd) {
+      const requestedPath = path.resolve(projectPath, cwd)
+      // Ensure the search path is within the project directory
+      if (!requestedPath.startsWith(projectPath)) {
+        resolve([
+          {
+            type: 'json',
+            value: {
+              errorMessage: `Invalid cwd: Path '${cwd}' is outside the project directory.`,
+            },
+          },
+        ])
+        return
+      }
+      searchCwd = requestedPath
+    }
+
+    const args = [...flagsArray, pattern, '.']
+
+    const childProcess = spawn(rgPath, args, {
+      cwd: searchCwd,
+      stdio: ['ignore', 'pipe', 'pipe'],
+    })
+
+    childProcess.stdout.on('data', (data) => {
+      stdout += data.toString()
+    })
+
+    childProcess.stderr.on('data', (data) => {
+      stderr += data.toString()
+    })
+
+    childProcess.on('close', (code) => {
+      // Truncate output to prevent memory issues
+      const maxLength = 10000
+      const truncatedStdout =
+        stdout.length > maxLength
+          ? stdout.substring(0, maxLength) + '\n\n[Output truncated]'
+          : stdout
+
+      const maxErrorLength = 1000
+      const truncatedStderr =
+        stderr.length > maxErrorLength
+          ? stderr.substring(0, maxErrorLength) + '\n\n[Error output truncated]'
+          : stderr
+
+      const result = {
+        stdout: truncatedStdout,
+        ...(truncatedStderr && { stderr: truncatedStderr }),
+        ...(code !== null && { exitCode: code }),
+        message: 'Code search completed',
+      }
+
+      resolve([
+        {
+          type: 'json',
+          value: result,
+        },
+      ])
+    })
+
+    childProcess.on('error', (error) => {
+      resolve([
+        {
+          type: 'json',
+          value: {
+            errorMessage: `Failed to execute ripgrep: ${error.message}. Make sure ripgrep is installed and available in PATH.`,
+          },
+        },
+      ])
+    })
+  })
+}

sdk/src/tools/code-search.ts

@@ -0,0 +1,21 @@
+import type { CodebuffToolOutput } from '../../../common/src/tools/list'
+
+export function runFileChangeHooks({
+  files,
+}: {
+  files: string[]
+}): Promise<CodebuffToolOutput<'run_file_change_hooks'>> {
+  // In the SDK, we don't have access to codebuff.json configuration
+  // or the hook running infrastructure, so this is a no-op
+  
+  return Promise.resolve([
+    {
+      type: 'json',
+      value: [
+        {
+          errorMessage: 'No file change hooks were triggered for the specified files. File change hooks are not supported in the SDK environment.',
+        },
+      ],
+    },
+  ])
+}