fix ToolCall type for agent-config

Author: charleslien

.agents/file-picker.ts

@@ -28,9 +28,9 @@ const config: AgentConfig = {
   stepPrompt:
     'Do not use the find_files tool or any tools again. Just give your response.',
   handleSteps: function* ({ agentState, prompt, params }) {
-    yield {
+    const toolResult = yield {
       toolName: 'find_files',
-      args: { prompt },
+      args: { prompt: prompt ?? '' },
     }
     yield 'STEP_ALL'
   },

.agents/types/agent-config.d.ts

@@ -186,10 +186,12 @@ export interface AgentStepContext {
 /**
  * Tool call object for handleSteps generator
  */
-export interface ToolCall<T extends ToolName = ToolName> {
-  toolName: T
-  args?: Tools.GetToolParams<T>
-}
+export type ToolCall<T extends ToolName = ToolName> = {
+  [K in T]: {
+    toolName: K
+    args?: Tools.GetToolParams<K>
+  }
+}[T]
 
 /**
  * Result from executing a tool

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

@@ -186,10 +186,12 @@ export interface AgentStepContext {
 /**
  * Tool call object for handleSteps generator
  */
-export interface ToolCall<T extends ToolName = ToolName> {
-  toolName: T
-  args?: Tools.GetToolParams<T>
-}
+export type ToolCall<T extends ToolName = ToolName> = {
+  [K in T]: {
+    toolName: K
+    args?: Tools.GetToolParams<K>
+  }
+}[T]
 
 /**
  * Result from executing a tool