Remove TaskHandlerResult and use TaskResult directly. (#93)

* remove TaskHandlerResult

* changelog

* update notebooks

Author: nerdai

CHANGELOG.md

@@ -14,6 +14,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/).
 
 ### Changed
 
+- Remove TaskHandlerResult and use TaskResult directly (#93)
 - Improve `NextStepDecision` to allow for only one next_step or task_result (#88)
 
 ## [0.0.6] - 2025-07-27

docs/notebooks/first_llm_agent.ipynb

@@ -219,7 +219,7 @@
     {
      "data": {
       "text/plain": [
-       "Task(id_='25aa12c3-ae1a-415c-a186-102b4412c268', instruction='Add one to fifty-five point three. Use an appropriate tool!')"
+       "Task(id_='38bb636c-3dd2-4916-a45f-f3fb2ada8bec', instruction='Add one to fifty-five point three. Use an appropriate tool!')"
       ]
      },
      "execution_count": 13,
@@ -249,7 +249,7 @@
       "DEBUG (llm_agents_fs.TaskHandler) :      💬 ASSISTANT: \n",
       "INFO (llm_agents_fs.TaskHandler) :      🛠️ Executing Tool Call: add_one\n",
       "INFO (llm_agents_fs.TaskHandler) :      ✅ Successful Tool Call: 56.3\n",
-      "INFO (llm_agents_fs.TaskHandler) :      ✅ Step Result: The result of adding one to 55.3 is 56.3.\n",
+      "INFO (llm_agents_fs.TaskHandler) :      ✅ Step Result: The result of adding one to fifty-five point three is fifty-six point three.\n",
       "DEBUG (llm_agents_fs.TaskHandler) :      🧵 Rollout: assistant: The current instruction is 'Add one to fifty-five point three. Use an appropriate tool!'\n",
       "assistant: I need to make a tool call(s) to `add_one`.\n",
       "tool: {\n",
@@ -262,7 +262,7 @@
       "    \"content\": \"56.3\",\n",
       "    \"error\": false\n",
       "}\n",
-      "assistant: The result of adding one to 55.3 is 56.3.\n",
+      "assistant: The result of adding one to fifty-five point three is fifty-six point three.\n",
       "DEBUG (llm_agents_fs.TaskHandler) :      ---NEXT STEP PROMPT: You are overseeing an assistant's\n",
       "progress in accomplishing a user instruction. Provided below is the assistant's\n",
       "current response to the original task instruction. Also provided, is an\n",
@@ -277,7 +277,7 @@
       "</user-instruction>\n",
       "\n",
       "<current-response>\n",
-      "The result of adding one to 55.3 is 56.3.\n",
+      "The result of adding one to fifty-five point three is fifty-six point three.\n",
       "</current-response>\n",
       "\n",
       "<thinking-process>\n",
@@ -293,17 +293,140 @@
       "    \"content\": \"56.3\",\n",
       "    \"error\": false\n",
       "}\n",
-      "assistant: The result of adding one to 55.3 is 56.3.\n",
+      "assistant: The result of adding one to fifty-five point three is fifty-six point three.\n",
       "</thinking-process>\n",
       "\n",
-      "DEBUG (llm_agents_fs.TaskHandler) :      ---NEXT STEP: {\"task_step\":{\"id_\":\"915cca85-0b6b-477d-921e-0dce7b8b3533\",\"task_id\":\"add-one-to-55-point-three\",\"instruction\":\"Add one to fifty-five point three using the tool add_one.\"},\"task_result\":{\"task_id\":\"add-one-to-55-point-three\",\"content\":\"The result of adding one to 55.3 is 56.3.\"}}\n",
+      "DEBUG (llm_agents_fs.TaskHandler) :      ---NEXT STEP: {\"kind\":\"next_step\",\"content\":\"I need to make a tool call to `add_one` with the number '55.3' as an argument.\"}\n",
+      "INFO (llm_agents_fs.TaskHandler) :      🧠 New Step: I need to make a tool call to `add_one` with the number '55.3' as an argument.\n",
+      "INFO (llm_agents_fs.TaskHandler) :      ⚙️ Processing Step: I need to make a tool call to `add_one` with the number '55.3' as an argument.\n",
+      "DEBUG (llm_agents_fs.TaskHandler) :      🧵 Rollout: assistant: The current instruction is 'Add one to fifty-five point three. Use an appropriate tool!'\n",
+      "assistant: I need to make a tool call(s) to `add_one`.\n",
+      "tool: {\n",
+      "    \"tool_call\": {\n",
+      "        \"tool_name\": \"add_one\",\n",
+      "        \"arguments\": {\n",
+      "            \"x\": \"55.3\"\n",
+      "        }\n",
+      "    },\n",
+      "    \"content\": \"56.3\",\n",
+      "    \"error\": false\n",
+      "}\n",
+      "assistant: The result of adding one to fifty-five point three is fifty-six point three.\n",
+      "DEBUG (llm_agents_fs.TaskHandler) :      💬 SYSTEM: You are a helpful assistant.\n",
+      "\n",
+      "Here is some past dialogue and context, where another assistant was working\n",
+      "towards completing the task.\n",
+      "\n",
+      "<history>\n",
+      "assistant: The current instruction is 'Add one to fifty-five point three. Use an appropriate tool!'\n",
+      "assistant: I need to make a tool call(s) to `add_one`.\n",
+      "tool: {\n",
+      "    \"tool_call\": {\n",
+      "        \"tool_name\": \"add_one\",\n",
+      "        \"arguments\": {\n",
+      "            \"x\": \"55.3\"\n",
+      "        }\n",
+      "    },\n",
+      "    \"content\": \"56.3\",\n",
+      "    \"error\": false\n",
+      "}\n",
+      "assistant: The result of adding one to fifty-five point three is fifty-six point three.\n",
+      "</history>\n",
+      "\n",
+      "DEBUG (llm_agents_fs.TaskHandler) :      💬 USER INPUT: I need to make a tool call to `add_one` with the number '55.3' as an argument.\n",
+      "DEBUG (llm_agents_fs.TaskHandler) :      💬 ASSISTANT: \n",
+      "INFO (llm_agents_fs.TaskHandler) :      🛠️ Executing Tool Call: add_one\n",
+      "INFO (llm_agents_fs.TaskHandler) :      ✅ Successful Tool Call: 56.3\n",
+      "INFO (llm_agents_fs.TaskHandler) :      ✅ Step Result: The result of adding one to fifty-five point three is fifty-six point three.\n",
+      "DEBUG (llm_agents_fs.TaskHandler) :      🧵 Rollout: assistant: The current instruction is 'Add one to fifty-five point three. Use an appropriate tool!'\n",
+      "assistant: I need to make a tool call(s) to `add_one`.\n",
+      "tool: {\n",
+      "    \"tool_call\": {\n",
+      "        \"tool_name\": \"add_one\",\n",
+      "        \"arguments\": {\n",
+      "            \"x\": \"55.3\"\n",
+      "        }\n",
+      "    },\n",
+      "    \"content\": \"56.3\",\n",
+      "    \"error\": false\n",
+      "}\n",
+      "assistant: The result of adding one to fifty-five point three is fifty-six point three.assistant: The current instruction is 'I need to make a tool call to `add_one` with the number '55.3' as an argument.'\n",
+      "assistant: I need to make a tool call(s) to `add_one`.\n",
+      "tool: {\n",
+      "    \"tool_call\": {\n",
+      "        \"tool_name\": \"add_one\",\n",
+      "        \"arguments\": {\n",
+      "            \"x\": \"55.3\"\n",
+      "        }\n",
+      "    },\n",
+      "    \"content\": \"56.3\",\n",
+      "    \"error\": false\n",
+      "}\n",
+      "assistant: The result of adding one to fifty-five point three is fifty-six point three.\n",
+      "DEBUG (llm_agents_fs.TaskHandler) :      ---NEXT STEP PROMPT: You are overseeing an assistant's\n",
+      "progress in accomplishing a user instruction. Provided below is the assistant's\n",
+      "current response to the original task instruction. Also provided, is an\n",
+      "internal 'thinking' process of the assistant that the user has not seen.\n",
+      "\n",
+      "Determine if the current the response is sufficient to answer the original task\n",
+      "instruction. In the case that it is not, provide a new instruction to the\n",
+      "assistant in order to help them improve upon their current response.\n",
+      "\n",
+      "<user-instruction>\n",
+      "Add one to fifty-five point three. Use an appropriate tool!\n",
+      "</user-instruction>\n",
+      "\n",
+      "<current-response>\n",
+      "The result of adding one to fifty-five point three is fifty-six point three.\n",
+      "</current-response>\n",
+      "\n",
+      "<thinking-process>\n",
+      "assistant: The current instruction is 'Add one to fifty-five point three. Use an appropriate tool!'\n",
+      "assistant: I need to make a tool call(s) to `add_one`.\n",
+      "tool: {\n",
+      "    \"tool_call\": {\n",
+      "        \"tool_name\": \"add_one\",\n",
+      "        \"arguments\": {\n",
+      "            \"x\": \"55.3\"\n",
+      "        }\n",
+      "    },\n",
+      "    \"content\": \"56.3\",\n",
+      "    \"error\": false\n",
+      "}\n",
+      "assistant: The result of adding one to fifty-five point three is fifty-six point three.assistant: The current instruction is 'I need to make a tool call to `add_one` with the number '55.3' as an argument.'\n",
+      "assistant: I need to make a tool call(s) to `add_one`.\n",
+      "tool: {\n",
+      "    \"tool_call\": {\n",
+      "        \"tool_name\": \"add_one\",\n",
+      "        \"arguments\": {\n",
+      "            \"x\": \"55.3\"\n",
+      "        }\n",
+      "    },\n",
+      "    \"content\": \"56.3\",\n",
+      "    \"error\": false\n",
+      "}\n",
+      "assistant: The result of adding one to fifty-five point three is fifty-six point three.\n",
+      "</thinking-process>\n",
+      "\n",
+      "DEBUG (llm_agents_fs.TaskHandler) :      ---NEXT STEP: {\"kind\":\"final_result\",\"content\":\"The result of adding one to fifty-five point three is fifty-six point three.\"}\n",
       "INFO (llm_agents_fs.TaskHandler) :      No new step required.\n",
-      "INFO (llm_agents_fs.LLMAgent) :      🏁 Task completed: The result of adding one to 55.3 is 56.3.\n"
+      "INFO (llm_agents_fs.LLMAgent) :      🏁 Task completed: The result of adding one to fifty-five point three is fifty-six point three.\n"
      ]
+    },
+    {
+     "data": {
+      "text/plain": [
+       "TaskResult(task_id='38bb636c-3dd2-4916-a45f-f3fb2ada8bec', content='The result of adding one to fifty-five point three is fifty-six point three.')"
+      ]
+     },
+     "execution_count": 14,
+     "metadata": {},
+     "output_type": "execute_result"
     }
    ],
    "source": [
-    "result = await agent.run(task)"
+    "handler = agent.run(task)\n",
+    "await handler"
    ]
   },
   {
@@ -315,7 +438,7 @@
     {
      "data": {
       "text/plain": [
-       "TaskResult(task_id='25aa12c3-ae1a-415c-a186-102b4412c268', content='The result of adding one to 55.3 is 56.3.')"
+       "TaskResult(task_id='38bb636c-3dd2-4916-a45f-f3fb2ada8bec', content='The result of adding one to fifty-five point three is fifty-six point three.')"
       ]
      },
      "execution_count": 15,
@@ -324,7 +447,8 @@
     }
    ],
    "source": [
-    "result.task_result"
+    "result = handler.result()\n",
+    "result"
    ]
   },
   {
@@ -337,12 +461,12 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "The result of adding one to 55.3 is 56.3.\n"
+      "The result of adding one to fifty-five point three is fifty-six point three.\n"
      ]
     }
    ],
    "source": [
-    "print(result.task_result)"
+    "print(result)"
    ]
   },
   {
@@ -367,12 +491,24 @@
       "    \"content\": \"56.3\",\n",
       "    \"error\": false\n",
       "}\n",
-      "assistant: The result of adding one to 55.3 is 56.3.\n"
+      "assistant: The result of adding one to fifty-five point three is fifty-six point three.assistant: The current instruction is 'I need to make a tool call to `add_one` with the number '55.3' as an argument.'\n",
+      "assistant: I need to make a tool call(s) to `add_one`.\n",
+      "tool: {\n",
+      "    \"tool_call\": {\n",
+      "        \"tool_name\": \"add_one\",\n",
+      "        \"arguments\": {\n",
+      "            \"x\": \"55.3\"\n",
+      "        }\n",
+      "    },\n",
+      "    \"content\": \"56.3\",\n",
+      "    \"error\": false\n",
+      "}\n",
+      "assistant: The result of adding one to fifty-five point three is fifty-six point three.\n"
      ]
     }
    ],
    "source": [
-    "print(result.rollout)"
+    "print(handler.rollout)"
    ]
   }
  ],