From 421e962258ea7ac117aca7fc749666bd4d9da700 Mon Sep 17 00:00:00 2001
From: liangxinbing <1580466765@qq.com>
Date: Tue, 18 Mar 2025 22:57:47 +0800
Subject: [PATCH] add workspace_root for Config and update Manus

---
 app/agent/manus.py | 14 +++++---------
 app/config.py      |  5 +++++
 2 files changed, 10 insertions(+), 9 deletions(-)

diff --git a/app/agent/manus.py b/app/agent/manus.py
index 8cba71a..d7ec2f9 100644
--- a/app/agent/manus.py
+++ b/app/agent/manus.py
@@ -1,9 +1,8 @@
-import os
-from pathlib import Path
-
 from pydantic import Field
 
 from app.agent.browser import BrowserAgent
+from app.config import config
+from app.prompt.browser import NEXT_STEP_PROMPT as BROWSER_NEXT_STEP_PROMPT
 from app.prompt.manus import NEXT_STEP_PROMPT, SYSTEM_PROMPT
 from app.tool import Terminate, ToolCollection
 from app.tool.browser_use_tool import BrowserUseTool
@@ -11,9 +10,6 @@ from app.tool.python_execute import PythonExecute
 from app.tool.str_replace_editor import StrReplaceEditor
 
 
-initial_working_directory = Path(os.getcwd()) / "workspace"
-
-
 class Manus(BrowserAgent):
     """
     A versatile general-purpose agent that uses planning to solve various tasks.
@@ -28,7 +24,7 @@ class Manus(BrowserAgent):
         "A versatile agent that can solve various tasks using multiple tools"
     )
 
-    system_prompt: str = SYSTEM_PROMPT.format(directory=initial_working_directory)
+    system_prompt: str = SYSTEM_PROMPT.format(directory=config.workspace_root)
     next_step_prompt: str = NEXT_STEP_PROMPT
 
     max_observe: int = 10000
@@ -55,8 +51,8 @@ class Manus(BrowserAgent):
         )
 
         if browser_in_use:
-            # Override with parent class's prompt temporarily to get browser context
-            self.next_step_prompt = BrowserAgent.next_step_prompt
+            # Override with browser-specific prompt temporarily to get browser context
+            self.next_step_prompt = BROWSER_NEXT_STEP_PROMPT
 
         # Call parent's think method
         result = await super().think()
diff --git a/app/config.py b/app/config.py
index 6d724ec..0be771b 100644
--- a/app/config.py
+++ b/app/config.py
@@ -222,5 +222,10 @@ class Config:
     def search_config(self) -> Optional[SearchSettings]:
         return self._config.search_config
 
+    @property
+    def workspace_root(self) -> Path:
+        """Get the workspace root directory"""
+        return WORKSPACE_ROOT
+
 
 config = Config()