From bbac41cc76aaba8b63d00ceb40b435e2b59a8af9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Magnus=20M=C3=BCller?=
 <67061560+MagMueller@users.noreply.github.com>
Date: Tue, 18 Feb 2025 12:54:31 -0800
Subject: [PATCH] Clean up add_tool_message

---
 browser_use/agent/message_manager/service.py | 15 ++-------------
 1 file changed, 2 insertions(+), 13 deletions(-)

diff --git a/browser_use/agent/message_manager/service.py b/browser_use/agent/message_manager/service.py
index cb10bdf2a..054615a97 100644
--- a/browser_use/agent/message_manager/service.py
+++ b/browser_use/agent/message_manager/service.py
@@ -85,13 +85,7 @@ class MessageManager:
 			tool_calls=tool_calls,
 		)
 		self._add_message_with_tokens(example_tool_call)
-		tool_message = ToolMessage(
-			content='Browser started',
-			tool_call_id=str(self.state.tool_id),
-		)
-		self._add_message_with_tokens(tool_message)
-
-		self.state.tool_id += 1
+		self.add_tool_message(content='Browser started')
 
 		placeholder_message = HumanMessage(content='[Your task history memory starts here]')
 		self._add_message_with_tokens(placeholder_message)
@@ -155,12 +149,7 @@ class MessageManager:
 
 		self._add_message_with_tokens(msg)
 		# empty tool response
-		tool_message = ToolMessage(
-			content='',
-			tool_call_id=str(self.state.tool_id),
-		)
-		self._add_message_with_tokens(tool_message)
-		self.state.tool_id += 1
+		self.add_tool_message(content='')
 
 	def add_plan(self, plan: Optional[str], position: int = -1) -> None:
 		if plan: