change z.toJSONSchema to use input schema

Author: charleslien

backend/src/templates/strings.ts

@@ -188,7 +188,9 @@ export async function getAgentPrompt<T extends StringField>(
       addendum += '```json\n'
       try {
         // Convert Zod schema to JSON schema for display
-        const jsonSchema = z.toJSONSchema(agentTemplate.outputSchema)
+        const jsonSchema = z.toJSONSchema(agentTemplate.outputSchema, {
+          io: 'input',
+        })
         delete jsonSchema['$schema'] // Remove the $schema field for cleaner display
         addendum += JSON.stringify(jsonSchema, null, 2)
       } catch {

backend/src/tools/prompts.ts

@@ -15,7 +15,9 @@ function paramsSection(schema: z.ZodObject, endsAgentStep: boolean) {
           .describe('Easp flag must be set to true'),
       })
     : schema
-  const jsonSchema = z.toJSONSchema(schemaWithEndsAgentStepParam)
+  const jsonSchema = z.toJSONSchema(schemaWithEndsAgentStepParam, {
+    io: 'input',
+  })
   delete jsonSchema.description
   delete jsonSchema['$schema']
   const paramsDescription = Object.keys(jsonSchema.properties ?? {}).length

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

@@ -1,7 +1,7 @@
 import z from 'zod/v4'
 
-import { llmToolCallSchema } from './list'
 import { publishedTools } from './constants'
+import { llmToolCallSchema } from './list'
 
 /**
  * Compiles all tool definitions into a single TypeScript definition file content.
@@ -19,7 +19,7 @@ export function compileToolDefinitions(): string {
       // Convert Zod schema to TypeScript interface using JSON schema
       let typeDefinition: string
       try {
-        const jsonSchema = z.toJSONSchema(parameterSchema)
+        const jsonSchema = z.toJSONSchema(parameterSchema, { io: 'input' })
         typeDefinition = jsonSchemaToTypeScript(jsonSchema)
       } catch (error) {
         console.warn(`Failed to convert schema for ${toolName}:`, error)

common/src/util/zod-schema.ts

@@ -5,13 +5,14 @@ import z from 'zod/v4'
  */
 export function schemaToJsonStr(
   schema: z.ZodTypeAny | undefined | Record<string, any>,
+  options?: Parameters<typeof z.toJSONSchema>[1],
 ): string {
   if (!schema) return 'None'
 
   try {
     // Handle Zod schemas
     if (schema instanceof z.ZodType) {
-      const jsonSchema = z.toJSONSchema(schema)
+      const jsonSchema = z.toJSONSchema(schema, options)
       delete jsonSchema['$schema']
       return JSON.stringify(jsonSchema, null, 2)
     }

web/src/components/docs/mdx/schema-display.tsx

@@ -7,11 +7,11 @@ import { DynamicAgentTemplateSchema } from '@codebuff/common/types/dynamic-agent
 import { CodeDemo } from './code-demo'
 
 export function SchemaDisplay() {
-  const schemaString = schemaToJsonStr(CodebuffConfigSchema)
+  const schemaString = schemaToJsonStr(CodebuffConfigSchema, {io: 'input'})
   return <CodeDemo language="json">{schemaString}</CodeDemo>
 }
 
 export function AgentTemplateSchemaDisplay() {
-  const schemaString = schemaToJsonStr(DynamicAgentTemplateSchema)
+  const schemaString = schemaToJsonStr(DynamicAgentTemplateSchema, {io: 'input'})
   return <CodeDemo language="json">{schemaString}</CodeDemo>
 }