Update handleSteps to be called again after llm generation finished

Author: jahooma

.agents/factory/base.ts

@@ -70,7 +70,8 @@ export const base = (model: ModelName): Omit<SecretAgentDefinition, 'id'> => ({
         },
       } as any
 
-      yield 'STEP'
+      const { stepsComplete } = yield 'STEP'
+      if (stepsComplete) break
     }
   },
 })

.agents/factory/thinking-base.ts

@@ -70,7 +70,9 @@ export const thinkingBase = (
           ],
         },
       } satisfies ToolCall
-      yield 'STEP'
+
+      const { stepsComplete } = yield 'STEP'
+      if (stepsComplete) break
     }
   },
 })

.agents/types/agent-definition.ts

@@ -166,12 +166,14 @@ export interface AgentDefinition {
    * }
    * }
    */
-  handleSteps?: (
-    context: AgentStepContext,
-  ) => Generator<
+  handleSteps?: (context: AgentStepContext) => Generator<
     ToolCall | 'STEP' | 'STEP_ALL',
     void,
-    { agentState: AgentState; toolResult: string | undefined }
+    {
+      agentState: AgentState
+      toolResult: string | undefined
+      stepsComplete: boolean
+    }
   >
 }