[Add] browser-use and main.py

browser-use/browser_use/logging_config.py

@@ -0,0 +1,132 @@
+import logging
+import os
+import sys
+
+from dotenv import load_dotenv
+
+load_dotenv()
+
+
+def addLoggingLevel(levelName, levelNum, methodName=None):
+	"""
+	Comprehensively adds a new logging level to the `logging` module and the
+	currently configured logging class.
+
+	`levelName` becomes an attribute of the `logging` module with the value
+	`levelNum`. `methodName` becomes a convenience method for both `logging`
+	itself and the class returned by `logging.getLoggerClass()` (usually just
+	`logging.Logger`). If `methodName` is not specified, `levelName.lower()` is
+	used.
+
+	To avoid accidental clobberings of existing attributes, this method will
+	raise an `AttributeError` if the level name is already an attribute of the
+	`logging` module or if the method name is already present
+
+	Example
+	-------
+	>>> addLoggingLevel('TRACE', logging.DEBUG - 5)
+	>>> logging.getLogger(__name__).setLevel('TRACE')
+	>>> logging.getLogger(__name__).trace('that worked')
+	>>> logging.trace('so did this')
+	>>> logging.TRACE
+	5
+
+	"""
+	if not methodName:
+		methodName = levelName.lower()
+
+	if hasattr(logging, levelName):
+		raise AttributeError(f'{levelName} already defined in logging module')
+	if hasattr(logging, methodName):
+		raise AttributeError(f'{methodName} already defined in logging module')
+	if hasattr(logging.getLoggerClass(), methodName):
+		raise AttributeError(f'{methodName} already defined in logger class')
+
+	# This method was inspired by the answers to Stack Overflow post
+	# http://stackoverflow.com/q/2183233/2988730, especially
+	# http://stackoverflow.com/a/13638084/2988730
+	def logForLevel(self, message, *args, **kwargs):
+		if self.isEnabledFor(levelNum):
+			self._log(levelNum, message, args, **kwargs)
+
+	def logToRoot(message, *args, **kwargs):
+		logging.log(levelNum, message, *args, **kwargs)
+
+	logging.addLevelName(levelNum, levelName)
+	setattr(logging, levelName, levelNum)
+	setattr(logging.getLoggerClass(), methodName, logForLevel)
+	setattr(logging, methodName, logToRoot)
+
+
+def setup_logging():
+	# Try to add RESULT level, but ignore if it already exists
+	try:
+		addLoggingLevel('RESULT', 35)  # This allows ERROR, FATAL and CRITICAL
+	except AttributeError:
+		pass  # Level already exists, which is fine
+
+	log_type = os.getenv('BROWSER_USE_LOGGING_LEVEL', 'info').lower()
+
+	# Check if handlers are already set up
+	if logging.getLogger().hasHandlers():
+		return
+
+	# Clear existing handlers
+	root = logging.getLogger()
+	root.handlers = []
+
+	class BrowserUseFormatter(logging.Formatter):
+		def format(self, record):
+			if isinstance(record.name, str) and record.name.startswith('browser_use.'):
+				record.name = record.name.split('.')[-2]
+			return super().format(record)
+
+	# Setup single handler for all loggers
+	console = logging.StreamHandler(sys.stdout)
+
+	# adittional setLevel here to filter logs
+	if log_type == 'result':
+		console.setLevel('RESULT')
+		console.setFormatter(BrowserUseFormatter('%(message)s'))
+	else:
+		console.setFormatter(BrowserUseFormatter('%(levelname)-8s [%(name)s] %(message)s'))
+
+	# Configure root logger only
+	root.addHandler(console)
+
+	# switch cases for log_type
+	if log_type == 'result':
+		root.setLevel('RESULT')  # string usage to avoid syntax error
+	elif log_type == 'debug':
+		root.setLevel(logging.DEBUG)
+	else:
+		root.setLevel(logging.INFO)
+
+	# Configure browser_use logger
+	browser_use_logger = logging.getLogger('browser_use')
+	browser_use_logger.propagate = False  # Don't propagate to root logger
+	browser_use_logger.addHandler(console)
+	browser_use_logger.setLevel(root.level)  # Set same level as root logger
+
+	logger = logging.getLogger('browser_use')
+	# logger.info('BrowserUse logging setup complete with level %s', log_type)
+	# Silence third-party loggers
+	for logger in [
+		'WDM',
+		'httpx',
+		'selenium',
+		'playwright',
+		'urllib3',
+		'asyncio',
+		'langchain',
+		'openai',
+		'httpcore',
+		'charset_normalizer',
+		'anthropic._base_client',
+		'PIL.PngImagePlugin',
+		'trafilatura.htmlprocessing',
+		'trafilatura',
+	]:
+		third_party = logging.getLogger(logger)
+		third_party.setLevel(logging.ERROR)
+		third_party.propagate = False