Reinstate write_todos as a tool forces a next step, but with exception

Author: jahooma

common/src/tools/constants.ts

@@ -14,7 +14,6 @@ export const TOOLS_WHICH_WONT_FORCE_NEXT_STEP = [
   'add_message',
   'update_subgoal',
   'create_plan',
-  'write_todos',
 ]
 
 // List of all available tools

packages/agent-runtime/src/run-agent-step.ts

@@ -368,8 +368,25 @@ export const runAgentStep = async (
     toolResults.filter(
       (result) => !TOOLS_WHICH_WONT_FORCE_NEXT_STEP.includes(result.toolName),
     ).length === 0
+
+  // Exception: if the only tool call is write_todos and all todos are completed, then end turn.
+  let hasOnlyFinishedTodos = false
+  if (toolCalls.length === 1 && toolCalls[0].toolName === 'write_todos') {
+    const todos = toolCalls[0].input.todos as
+      | {
+          task: string
+          completed: boolean
+        }[]
+      | undefined
+    if (todos && todos.every((todo) => todo.completed)) {
+      hasOnlyFinishedTodos = true
+    }
+  }
+
   let shouldEndTurn =
-    toolCalls.some((call) => call.toolName === 'end_turn') || hasNoToolResults
+    toolCalls.some((call) => call.toolName === 'end_turn') ||
+    hasNoToolResults ||
+    hasOnlyFinishedTodos
 
   agentState = {
     ...agentState,