From ef36632ccf22df2f6e691bd46f49790cb59fcfbb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Magnus=20M=C3=BCller?=
 <67061560+MagMueller@users.noreply.github.com>
Date: Sat, 5 Jul 2025 23:12:05 +0200
Subject: [PATCH] Refactor observability imports to maintain consistency across
 modules

---
 browser_use/agent/message_manager/service.py | 3 +--
 browser_use/browser/session.py               | 3 +--
 browser_use/controller/registry/service.py   | 2 +-
 browser_use/controller/service.py            | 3 +--
 4 files changed, 4 insertions(+), 7 deletions(-)

diff --git a/browser_use/agent/message_manager/service.py b/browser_use/agent/message_manager/service.py
index 1b56bac09..36e5dcd75 100644
--- a/browser_use/agent/message_manager/service.py
+++ b/browser_use/agent/message_manager/service.py
@@ -3,8 +3,6 @@ from __future__ import annotations
 import json
 import logging
 
-from observability import observe_debug
-
 from browser_use.agent.message_manager.views import (
 	HistoryItem,
 )
@@ -24,6 +22,7 @@ from browser_use.llm.messages import (
 	SystemMessage,
 	UserMessage,
 )
+from browser_use.observability import observe_debug
 from browser_use.utils import match_url_with_domain_pattern, time_execution_sync
 
 logger = logging.getLogger(__name__)
diff --git a/browser_use/browser/session.py b/browser_use/browser/session.py
index 5074a5c72..0259b18b5 100644
--- a/browser_use/browser/session.py
+++ b/browser_use/browser/session.py
@@ -16,9 +16,8 @@ from pathlib import Path
 from typing import Any, Self
 from urllib.parse import urlparse
 
-from observability import observe_debug
-
 from browser_use.config import CONFIG
+from browser_use.observability import observe_debug
 from browser_use.utils import _log_pretty_path, _log_pretty_url
 
 from .utils import normalize_url
diff --git a/browser_use/controller/registry/service.py b/browser_use/controller/registry/service.py
index 43a4207e5..f372ad65d 100644
--- a/browser_use/controller/registry/service.py
+++ b/browser_use/controller/registry/service.py
@@ -8,7 +8,6 @@ from inspect import Parameter, iscoroutinefunction, signature
 from types import UnionType
 from typing import Any, Generic, Optional, TypeVar, Union, get_args, get_origin
 
-from observability import observe_debug
 from pydantic import BaseModel, Field, RootModel, create_model
 
 from browser_use.browser import BrowserSession
@@ -21,6 +20,7 @@ from browser_use.controller.registry.views import (
 )
 from browser_use.filesystem.file_system import FileSystem
 from browser_use.llm.base import BaseChatModel
+from browser_use.observability import observe_debug
 from browser_use.telemetry.service import ProductTelemetry
 from browser_use.utils import match_url_with_domain_pattern, time_execution_async
 
diff --git a/browser_use/controller/service.py b/browser_use/controller/service.py
index 216987938..acdddfce1 100644
--- a/browser_use/controller/service.py
+++ b/browser_use/controller/service.py
@@ -7,8 +7,6 @@ import re
 from collections.abc import Awaitable, Callable
 from typing import Any, Generic, TypeVar, cast
 
-from observability import observe_debug
-
 try:
 	from lmnr import Laminar  # type: ignore
 except ImportError:
@@ -37,6 +35,7 @@ from browser_use.controller.views import (
 from browser_use.filesystem.file_system import FileSystem
 from browser_use.llm.base import BaseChatModel
 from browser_use.llm.messages import UserMessage
+from browser_use.observability import observe_debug
 from browser_use.utils import time_execution_sync
 
 logger = logging.getLogger(__name__)