Function createDiscoveryTool

packages/ai-code-mode/src/create-discovery-tool.ts:55–108

(
  lazyTools: Array<CodeModeTool>,
  lazyToolsConfig?: LazyToolsConfig,
)

 * description appears in this tool's own catalog (mirroring the system prompt).
 */
export function createDiscoveryTool(
  lazyTools: Array<CodeModeTool>,
  lazyToolsConfig?: LazyToolsConfig,
): ServerTool<
  typeof discoverInputSchema,
  typeof discoverOutputSchema,
  'discover_tools'
> {
  const lazyMap = new Map(lazyTools.map((t) => [t.name, t]))
  const include = lazyToolsConfig?.includeDescription ?? 'none'
  const catalog = lazyTools
    .map((t) =>
      renderLazyCatalogEntry(
        `${EXTERNAL_PREFIX}${t.name}`,
        t.description,
        include,
      ),
    )
    .join(', ')

  return toolDefinition({
    name: 'discover_tools' as const,
    description:
      `Discover full TypeScript signatures for additional sandbox APIs before ` +
      `using them inside execute_typescript. Discoverable tools: [${catalog}]. ` +
      `Pass the names exactly as shown (the external_ prefix is optional).`,
    inputSchema: discoverInputSchema,
    outputSchema: discoverOutputSchema,
  }).server(async ({ toolNames }) => {
    const tools: Array<{
      name: string
      description: string
      typeStub: string
    }> = []
    const errors: Array<string> = []

    for (const name of toolNames) {
      const tool = lazyMap.get(stripExternalPrefix(name))
      if (!tool) {
        errors.push(`Unknown tool: '${name}'. Discoverable tools: [${catalog}]`)
        continue
      }
      const binding = toolToBinding(tool, EXTERNAL_PREFIX)
      const typeStub = generateTypeStubs({ [binding.name]: binding })
      tools.push({
        name: binding.name,
        description: tool.description,
        typeStub,
      })
    }

    return errors.length > 0 ? { tools, errors } : { tools }
  })
}