Track the source with an parameter

Author: MagMueller

browser_use/agent/service.py

@@ -153,6 +153,7 @@ def __init__(
 		save_playwright_script_path: Optional[str] = None,
 		enable_memory: bool = True,
 		memory_config: Optional[MemoryConfig] = None,
+		source: Optional[str] = None,
 	):
 		if page_extraction_llm is None:
 			page_extraction_llm = llm
@@ -336,7 +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
 		logger.debug(f'Version: {version}, Source: {source}')
 		self.version = version
 		self.source = source

eval/service.py

@@ -544,7 +544,13 @@ async def run_agent_with_tracing(
 
 		browser = browser or Browser()
 
-		agent = Agent(task=task.confirmed_task, llm=llm, browser=browser, use_vision=use_vision)
+		agent = Agent(
+			task=task.confirmed_task,
+			llm=llm,
+			browser=browser,
+			use_vision=use_vision,
+			source='eval_platform',  # Override source detection
+		)
 
 		# Pass our hook functions
 		result = await agent.run(max_steps=max_steps, on_step_start=tracker.on_step_start, on_step_end=tracker.on_step_end)