Fix tool call type in agent template

jahooma · jahooma · commit e5c5d2bb1645 · 2025-08-01T11:40:00.000-07:00
diff --git a/common/src/types/agent-template.ts b/common/src/types/agent-template.ts
@@ -2,6 +2,7 @@ import type { Model } from '../constants'
 import type { AgentState, AgentTemplateType, ToolResult } from './session-state'
 import type { ToolName } from '../tools/constants'
 import type { z } from 'zod/v4'
+import { ToolCall } from '../util/types/agent-config'
 
 export type AgentTemplate<
   P = string | undefined,
@@ -33,7 +34,7 @@ export type AgentTemplate<
 }
 
 export type StepGenerator = Generator<
-  Omit<any, 'toolCallId'> | 'STEP' | 'STEP_ALL', // Generic tool call type
+  Omit<ToolCall, 'toolCallId'> | 'STEP' | 'STEP_ALL', // Generic tool call type
   void,
   { agentState: AgentState; toolResult: ToolResult | undefined }
 >

Original file line number	Diff line number	Diff line change
`@@ -2,6 +2,7 @@ import type { Model } from '../constants'`
`2`	`2`	`import type { AgentState, AgentTemplateType, ToolResult } from './session-state'`
`3`	`3`	`import type { ToolName } from '../tools/constants'`
`4`	`4`	`import type { z } from 'zod/v4'`
	`5`	`+import { ToolCall } from '../util/types/agent-config'`
`5`	`6`
`6`	`7`	`export type AgentTemplate<`
`7`	`8`	`P = string \| undefined,`
`@@ -33,7 +34,7 @@ export type AgentTemplate<`
`33`	`34`	`}`
`34`	`35`
`35`	`36`	`export type StepGenerator = Generator<`
`36`		`- Omit<any, 'toolCallId'> \| 'STEP' \| 'STEP_ALL', // Generic tool call type`
	`37`	`+ Omit<ToolCall, 'toolCallId'> \| 'STEP' \| 'STEP_ALL', // Generic tool call type`
`37`	`38`	`void,`
`38`	`39`	`{ agentState: AgentState; toolResult: ToolResult \| undefined }`
`39`	`40`	`>`