Refactor _set_browser_use_version_and_source to accept an optional source parameter

Author: MagMueller

browser_use/agent/service.py

@@ -198,7 +198,7 @@ def __init__(
 
 		# Action setup
 		self._setup_action_models()
-		self._set_browser_use_version_and_source()
+		self._set_browser_use_version_and_source(source)
 		self.initial_actions = self._convert_initial_actions(initial_actions) if initial_actions else None
 
 		# Model setup
@@ -312,7 +312,7 @@ def _set_message_context(self) -> str | None:
 				self.settings.message_context = f'Available actions: {self.unfiltered_actions}'
 		return self.settings.message_context
 
-	def _set_browser_use_version_and_source(self) -> None:
+	def _set_browser_use_version_and_source(self, source_override: Optional[str] = None) -> None:
 		"""Get the version and source of the browser-use package (git or pip in a nutshell)"""
 		try:
 			# First check for repository-specific files
@@ -337,8 +337,8 @@ def _set_browser_use_version_and_source(self) -> None:
 		except Exception:
 			version = 'unknown'
 			source = 'unknown'
-		if self.source is not None:
-			source = self.source
+		if source_override is not None:
+			source = source_override
 		logger.debug(f'Version: {version}, Source: {source}')
 		self.version = version
 		self.source = source