[Render Preview] Zod 4 & AI SDK 5 (#239)

Author: charleslien

.agents/changes-reviewer.ts

@@ -47,7 +47,7 @@ Use the following guidelines to review the changes and suggest improvements:
     // Step 1: Get list of changed files from git diff
     const { toolResult: gitDiffResult } = yield {
       toolName: 'run_terminal_command',
-      args: {
+      input: {
         command: 'git diff HEAD --name-only',
         process_type: 'SYNC',
         timeout_seconds: 30,
@@ -57,7 +57,7 @@ Use the following guidelines to review the changes and suggest improvements:
     // Step 2: Get untracked files from git status
     const { toolResult: gitStatusResult } = yield {
       toolName: 'run_terminal_command',
-      args: {
+      input: {
         command: 'git status --porcelain',
         process_type: 'SYNC',
         timeout_seconds: 30,
@@ -67,7 +67,7 @@ Use the following guidelines to review the changes and suggest improvements:
     // Step 3: Run full git diff to see the actual changes
     yield {
       toolName: 'run_terminal_command',
-      args: {
+      input: {
         command: 'git diff HEAD',
         process_type: 'SYNC',
         timeout_seconds: 30,
@@ -97,7 +97,7 @@ Use the following guidelines to review the changes and suggest improvements:
     if (allFilesToRead.length > 0) {
       yield {
         toolName: 'read_files',
-        args: {
+        input: {
           paths: allFilesToRead,
         },
       }
@@ -106,7 +106,7 @@ Use the following guidelines to review the changes and suggest improvements:
     // Step 5: Put words in the AI's mouth to get it to spawn the file explorer.
     yield {
       toolName: 'add_message',
-      args: {
+      input: {
         role: 'assistant',
         content:
           'Now I will spawn a file explorer to find any missing codebase context, and then review the changes.',

.agents/claude4-gemini-thinking.ts

@@ -324,7 +324,7 @@ User cwd: {CODEBUFF_USER_CWD}
     while (true) {
       yield {
         toolName: 'spawn_agents',
-        args: {
+        input: {
           agents: [
             {
               agent_type: 'thinker',

.agents/file-explorer.ts

@@ -58,7 +58,7 @@ const definition: AgentDefinition = {
       ),
       { toolResult: spawnResult } = yield {
         toolName: 'spawn_agents',
-        args: {
+        input: {
           agents: filePickerPrompts.map((promptText) => ({
             agent_type: 'file-picker',
             prompt: promptText,
@@ -67,7 +67,7 @@ const definition: AgentDefinition = {
       }
     yield {
       toolName: 'set_output',
-      args: {
+      input: {
         results: spawnResult,
       },
     }

.agents/file-picker.ts

@@ -1,6 +1,6 @@
 import { publisher, version } from './constants'
 
-import type { AgentDefinition } from './types/agent-definition'
+import type { AgentDefinition, ToolCall } from './types/agent-definition'
 
 const definition: AgentDefinition = {
   id: 'file-picker',
@@ -37,8 +37,8 @@ In your report, please give an analysis that includes the full paths of files th
   handleSteps: function* ({ agentState, prompt, params }) {
     yield {
       toolName: 'find_files',
-      args: { prompt: prompt ?? '' },
-    }
+      input: { prompt: prompt ?? '' },
+    } satisfies ToolCall
     yield 'STEP_ALL'
   },
 }

.agents/opensource/file-picker.ts

@@ -1,4 +1,4 @@
-import type { AgentDefinition } from '../types/agent-definition'
+import type { AgentDefinition, ToolCall } from '../types/agent-definition'
 
 const definition: AgentDefinition = {
   id: 'oss-model-file-picker',
@@ -36,8 +36,8 @@ In your report, please give an analysis that includes the full paths of files th
   handleSteps: function* ({ agentState, prompt, params }) {
     yield {
       toolName: 'find_files',
-      args: { prompt: prompt ?? "Find files related to the user's request" },
-    }
+      input: { prompt: prompt ?? "Find files related to the user's request" },
+    } satisfies ToolCall
     yield 'STEP_ALL'
   },
 }

.agents/researcher.ts

@@ -1,6 +1,6 @@
 import { publisher, version } from './constants'
 
-import type { AgentDefinition } from './types/agent-definition'
+import type { AgentDefinition, ToolCall } from './types/agent-definition'
 
 const definition: AgentDefinition = {
   id: 'researcher',
@@ -37,8 +37,8 @@ Always end your response with the end_turn tool.`,
   handleSteps: function* ({ agentState, prompt, params }) {
     yield {
       toolName: 'web_search',
-      args: { query: prompt ?? '', depth: 'standard' },
-    }
+      input: { query: prompt ?? '', depth: 'standard' },
+    } satisfies ToolCall
     yield 'STEP_ALL'
   },
 }

.agents/types/agent-definition.ts

@@ -116,7 +116,7 @@ export interface AgentDefinition {
   /** Programmatically step the agent forward and run tools.
    *
    * You can either yield:
-   * - A tool call object with toolName and args properties.
+   * - A tool call object with toolName and input properties.
    * - 'STEP' to run agent's model and generate one assistant message.
    * - 'STEP_ALL' to run the agent's model until it uses the end_turn tool or stops includes no tool calls in a message.
    *
@@ -126,7 +126,7 @@ export interface AgentDefinition {
    * function* handleSteps({ agentStep, prompt, params}) {
    *   const { toolResult } = yield {
    *     toolName: 'read_files',
-   *     args: { paths: ['file1.txt', 'file2.txt'] }
+   *     input: { paths: ['file1.txt', 'file2.txt'] }
    *   }
    *   yield 'STEP_ALL'
    * }
@@ -136,7 +136,7 @@ export interface AgentDefinition {
    *   while (true) {
    *     yield {
    *       toolName: 'spawn_agents',
-   *       args: {
+   *       input: {
    *         agents: [
    *         {
    *           agent_type: 'thinker',
@@ -191,7 +191,7 @@ export interface AgentStepContext {
 export type ToolCall<T extends ToolName = ToolName> = {
   [K in T]: {
     toolName: K
-    args?: Tools.GetToolParams<K>
+    input?: Tools.GetToolParams<K>
   }
 }[T]
 

backend/package.json

@@ -24,27 +24,28 @@
     "bun": ">=1.2.11"
   },
   "dependencies": {
-    "@ai-sdk/google-vertex": "2.2.24",
-    "@ai-sdk/openai": "1.3.22",
+    "@ai-sdk/google-vertex": "3.0.6",
+    "@ai-sdk/openai": "2.0.11",
     "@codebuff/billing": "workspace:*",
     "@codebuff/common": "workspace:*",
     "@codebuff/internal": "workspace:*",
     "@google-cloud/vertexai": "1.10.0",
     "@google/generative-ai": "0.24.1",
-    "ai": "4.3.16",
+    "@jitl/quickjs-wasmfile-release-sync": "0.31.0",
+    "@openrouter/ai-sdk-provider": "1.1.2",
+    "ai": "5.0.0",
     "cors": "^2.8.5",
     "diff": "5.2.0",
     "dotenv": "16.4.5",
     "express": "4.19.2",
     "gpt-tokenizer": "2.8.1",
     "ignore": "5.3.2",
-    "@jitl/quickjs-wasmfile-release-sync": "0.31.0",
-    "quickjs-emscripten-core": "0.31.0",
     "lodash": "*",
     "openai": "^4.78.1",
     "pino": "9.4.0",
     "postgres": "3.4.4",
     "posthog-node": "^4.14.0",
+    "quickjs-emscripten-core": "0.31.0",
     "ts-pattern": "5.3.1",
     "ws": "8.18.0",
     "zod": "3.25.67",

backend/src/__tests__/agent-registry.test.ts

@@ -60,6 +60,57 @@ const mockStaticTemplates: Record<string, AgentTemplate> = {
 // We'll spy on the validation functions instead of mocking the entire module
 
 describe('Agent Registry', () => {
+  beforeAll(() => {
+    // Mock the database module
+    mockModule('@codebuff/common/db', () => ({
+      default: {
+        select: () => ({
+          from: () => ({
+            where: () => ({
+              orderBy: () => ({
+                limit: () => Promise.resolve([]),
+              }),
+              then: (fn: (rows: any[]) => any) => fn([]),
+            }),
+          }),
+        }),
+      },
+    }))
+
+    // Mock the schema module
+    mockModule('@codebuff/common/db/schema', () => ({
+      agentConfig: {
+        id: 'id',
+        publisher_id: 'publisher_id',
+        version: 'version',
+        major: 'major',
+        minor: 'minor',
+        patch: 'patch',
+        data: 'data',
+      },
+    }))
+
+    // Mock drizzle-orm
+    mockModule('drizzle-orm', () => ({
+      and: (...args: any[]) => ({ type: 'and', args }),
+      desc: (field: any) => ({ type: 'desc', field }),
+      eq: (field: any, value: any) => ({ type: 'eq', field, value }),
+    }))
+
+    // Mock logger
+    mockModule('../util/logger', () => ({
+      logger: {
+        debug: () => {},
+        error: () => {},
+        warn: () => {},
+      },
+    }))
+
+    // Mock static agent templates
+    mockModule('@codebuff/backend/templates/agent-list', () => ({
+      agentTemplates: mockStaticTemplates,
+    }))
+  })
   let mockFileContext: ProjectFileContext
 
   beforeAll(() => {

backend/src/__tests__/loop-agent-steps.test.ts

@@ -193,15 +193,15 @@ describe('loopAgentSteps STEP behavior', () => {
 
       if (stepCount === 1) {
         // First call: Execute a tool, then STEP
-        yield { toolName: 'read_files', args: { paths: ['file1.txt'] } }
+        yield { toolName: 'read_files', input: { paths: ['file1.txt'] } }
         yield 'STEP' // Should pause here
       } else if (stepCount === 2) {
         // Second call: Should continue from here, not call LLM
         yield {
           toolName: 'write_file',
-          args: { path: 'output.txt', content: 'test' },
+          input: { path: 'output.txt', content: 'test' },
         }
-        yield { toolName: 'end_turn', args: {} }
+        yield { toolName: 'end_turn', input: {} }
       }
     })() as StepGenerator
 
@@ -258,12 +258,12 @@ describe('loopAgentSteps STEP behavior', () => {
     // it should complete without calling the LLM at all (since it ends with end_turn)
 
     const mockGenerator = (function* () {
-      yield { toolName: 'read_files', args: { paths: ['file1.txt'] } }
+      yield { toolName: 'read_files', input: { paths: ['file1.txt'] } }
       yield {
         toolName: 'write_file',
-        args: { path: 'output.txt', content: 'test' },
+        input: { path: 'output.txt', content: 'test' },
       }
-      yield { toolName: 'end_turn', args: {} }
+      yield { toolName: 'end_turn', input: {} }
     })() as StepGenerator
 
     mockTemplate.handleSteps = () => mockGenerator