Only publish a subset of tool type definitions

Author: jahooma

common/src/tools/compile-tool-definitions.ts

@@ -1,13 +1,15 @@
 import z from 'zod/v4'
 
-import { llmToolCallSchema } from './list'
+import { llmToolCallSchema, publishedTools } from './list'
 
 /**
  * Compiles all tool definitions into a single TypeScript definition file content.
  * This generates type definitions for all available tools and their parameters.
  */
 export function compileToolDefinitions(): string {
-  const toolEntries = Object.entries(llmToolCallSchema)
+  const toolEntries = publishedTools.map(
+    (toolName) => [toolName, llmToolCallSchema[toolName]] as const,
+  )
 
   const toolInterfaces = toolEntries
     .map(([toolName, toolDef]) => {

common/src/tools/list.ts

@@ -48,6 +48,26 @@ export const llmToolCallSchema = {
   [K in ToolName]: ToolParams<K>
 }
 
+export const publishedTools = [
+  'add_message',
+  'code_search',
+  'end_turn',
+  'find_files',
+  'read_docs',
+  'read_files',
+  'run_file_change_hooks',
+  'run_terminal_command',
+  'set_messages',
+  'set_output',
+  'spawn_agents',
+  'str_replace',
+  'think_deeply',
+  'web_search',
+  'write_file',
+  // 'spawn_agents_async',
+  // 'spawn_agent_inline',
+] as const
+
 export const clientToolCallSchema = {
   // Tools that require an id and objective
   add_subgoal: ['id', 'objective', 'status', 'plan', 'log'],

common/src/util/types/agent-definition.d.ts

@@ -231,15 +231,13 @@ export type TerminalTools = 'run_terminal_command' | 'run_file_change_hooks'
 /**
  * Web and browser tools
  */
-export type WebTools = 'browser_logs' | 'web_search' | 'read_docs'
+export type WebTools = 'web_search' | 'read_docs'
 
 /**
  * Agent management tools
  */
 export type AgentTools =
   | 'spawn_agents'
-  | 'spawn_agents_async'
-  | 'send_agent_message'
   | 'set_messages'
   | 'add_message'
 
@@ -248,9 +246,6 @@ export type AgentTools =
  */
 export type PlanningTools =
   | 'think_deeply'
-  | 'create_plan'
-  | 'add_subgoal'
-  | 'update_subgoal'
 
 /**
  * Output and control tools

common/src/util/types/tools.d.ts

@@ -3,10 +3,7 @@
  */
 export type ToolName =
   | 'add_message'
-  | 'add_subgoal'
-  | 'browser_logs'
   | 'code_search'
-  | 'create_plan'
   | 'end_turn'
   | 'find_files'
   | 'read_docs'
@@ -16,11 +13,8 @@ export type ToolName =
   | 'set_messages'
   | 'set_output'
   | 'spawn_agents'
-  | 'spawn_agents_async'
-  | 'spawn_agent_inline'
   | 'str_replace'
   | 'think_deeply'
-  | 'update_subgoal'
   | 'web_search'
   | 'write_file'
 
@@ -29,10 +23,7 @@ export type ToolName =
  */
 export interface ToolParamsMap {
   add_message: AddMessageParams
-  add_subgoal: AddSubgoalParams
-  browser_logs: BrowserLogsParams
   code_search: CodeSearchParams
-  create_plan: CreatePlanParams
   end_turn: EndTurnParams
   find_files: FindFilesParams
   read_docs: ReadDocsParams
@@ -42,11 +33,8 @@ export interface ToolParamsMap {
   set_messages: SetMessagesParams
   set_output: SetOutputParams
   spawn_agents: SpawnAgentsParams
-  spawn_agents_async: SpawnAgentsAsyncParams
-  spawn_agent_inline: SpawnAgentInlineParams
   str_replace: StrReplaceParams
   think_deeply: ThinkDeeplyParams
-  update_subgoal: UpdateSubgoalParams
   web_search: WebSearchParams
   write_file: WriteFileParams
 }
@@ -59,34 +47,6 @@ export interface AddMessageParams {
   content: string
 }
 
-/**
- * Add a new subgoal for tracking progress. To be used for complex requests that can't be solved in a single step, as you may forget what happened!
- */
-export interface AddSubgoalParams {
-  /** A unique identifier for the subgoal. Try to choose the next sequential integer that is not already in use. */
-  id: string
-  /** The objective of the subgoal, concisely and clearly stated. */
-  objective: string
-  /** The status of the subgoal. */
-  status: 'NOT_STARTED' | 'IN_PROGRESS' | 'COMPLETE' | 'ABORTED'
-  /** A plan for the subgoal. */
-  plan?: string
-  /** A log message for the subgoal progress. */
-  log?: string
-}
-
-/**
- * Parameters for browser_logs tool
- */
-export interface BrowserLogsParams {
-  /** The type of browser action to perform (e.g., "navigate"). */
-  type: string
-  /** The URL to navigate to. */
-  url: string
-  /** When to consider navigation successful. Defaults to 'load'. */
-  waitUntil?: 'load' | 'domcontentloaded' | 'networkidle0'
-}
-
 /**
  * Search for string patterns in the project's files. This tool uses ripgrep (rg), a fast line-oriented search tool. Use this tool only when read_files is not sufficient to find the files you need.
  */
@@ -99,16 +59,6 @@ export interface CodeSearchParams {
   cwd?: string
 }
 
-/**
- * Generate a detailed markdown plan for complex tasks.
- */
-export interface CreatePlanParams {
-  /** The path including the filename of a markdown file that will be overwritten with the plan. */
-  path: string
-  /** A detailed plan to solve the user's request. */
-  plan: string
-}
-
 /**
  * End your turn, regardless of any new tool results that might be coming. This will allow the user to type another prompt.
  */
@@ -193,32 +143,6 @@ export interface SpawnAgentsParams {
   }[]
 }
 
-/**
- * Parameters for spawn_agents_async tool
- */
-export interface SpawnAgentsAsyncParams {
-  agents: {
-    /** Agent to spawn */
-    agent_type: string
-    /** Prompt to send to the agent */
-    prompt?: string
-    /** Parameters object for the agent (if any) */
-    params?: Record<string, any>
-  }[]
-}
-
-/**
- * Spawn a single agent that runs within the current message history.
- */
-export interface SpawnAgentInlineParams {
-  /** Agent to spawn */
-  agent_type: string
-  /** Prompt to send to the agent */
-  prompt?: string
-  /** Parameters object for the agent (if any) */
-  params?: Record<string, any>
-}
-
 /**
  * Replace strings in a file with new strings.
  */
@@ -242,20 +166,6 @@ export interface ThinkDeeplyParams {
   thought: string
 }
 
-/**
- * Update a subgoal in the context given the id, and optionally the status or plan, or a new log to append. Feel free to update any combination of the status, plan, or log in one invocation.
- */
-export interface UpdateSubgoalParams {
-  /** The id of the subgoal to update. */
-  id: string
-  /** Change the status of the subgoal. */
-  status?: 'NOT_STARTED' | 'IN_PROGRESS' | 'COMPLETE' | 'ABORTED'
-  /** Change the plan for the subgoal. */
-  plan?: string
-  /** Add a log message to the subgoal. This will create a new log entry and append it to the existing logs. Use this to record your progress and any new information you learned as you go. */
-  log?: string
-}
-
 /**
  * Search the web for current information using Linkup API.
  */

sdk/src/types/agent-definition.ts

@@ -231,15 +231,13 @@ export type TerminalTools = 'run_terminal_command' | 'run_file_change_hooks'
 /**
  * Web and browser tools
  */
-export type WebTools = 'browser_logs' | 'web_search' | 'read_docs'
+export type WebTools = 'web_search' | 'read_docs'
 
 /**
  * Agent management tools
  */
 export type AgentTools =
   | 'spawn_agents'
-  | 'spawn_agents_async'
-  | 'send_agent_message'
   | 'set_messages'
   | 'add_message'
 
@@ -248,9 +246,6 @@ export type AgentTools =
  */
 export type PlanningTools =
   | 'think_deeply'
-  | 'create_plan'
-  | 'add_subgoal'
-  | 'update_subgoal'
 
 /**
  * Output and control tools