vibesurf 0.1.10__tar.gz → 0.1.11__tar.gz

{vibesurf-0.1.10 → vibesurf-0.1.11}/.env.example

@@ -46,6 +46,7 @@ VIBESURF_BACKEND_PORT=
 VIBESURF_EXTENSION=
 VIBESURF_WORKSPACE=
 VIBESURF_DATABASE_URL=
+VIBESURF_DEBUG=
 
 BROWSER_EXECUTION_PATH=
 BROWSER_USER_DATA=

{vibesurf-0.1.10 → vibesurf-0.1.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vibesurf
-Version: 0.1.10
+Version: 0.1.11
 Summary: VibeSurf: A powerful browser assistant for vibe surfing
 Author: Shao Warm
 License: Apache-2.0
@@ -37,7 +37,11 @@ Requires-Dist: aiosqlite>=0.21.0
 Requires-Dist: rich>=13.0.0
 Requires-Dist: greenlet>=3.2.4
 Requires-Dist: getmac>=0.9.5
-Requires-Dist: browser-use==0.7.1
+Requires-Dist: browser-use==0.7.7
+Requires-Dist: markdown-pdf>=1.9
+Requires-Dist: nanoid>=2.0.0
+Requires-Dist: markdownify>=1.2.0
+Requires-Dist: pathvalidate>=3.3.1
 Dynamic: license-file
 
 # VibeSurf: A powerful browser assistant for vibe surfing

{vibesurf-0.1.10 → vibesurf-0.1.11}/pyproject.toml

@@ -38,7 +38,11 @@ dependencies = [
     "rich>=13.0.0",
     "greenlet>=3.2.4",
     "getmac>=0.9.5",
-    "browser-use==0.7.1",
+    "browser-use==0.7.7",
+    "markdown-pdf>=1.9",
+    "nanoid>=2.0.0",
+    "markdownify>=1.2.0",
+    "pathvalidate>=3.3.1",
 ]
 
 [project.urls]
@@ -58,7 +62,7 @@ packages = [
     "vibe_surf.agents.prompts",
     "vibe_surf.browser",
     "vibe_surf.browser.watchdogs",
-    "vibe_surf.controller",
+    "vibe_surf.tools",
     "vibe_surf.llm",
     "vibe_surf.backend"
 ]

{vibesurf-0.1.10 → vibesurf-0.1.11}/tests/test_agents.py

@@ -14,7 +14,9 @@ from vibe_surf.browser.browser_manager import BrowserManager
 from vibe_surf.browser.agent_browser_session import AgentBrowserSession
 
 from vibe_surf.browser.agent_browser_session import AgentBrowserSession
-from vibe_surf.controller.vibesurf_tools import VibeSurfController
+from vibe_surf.browser.agen_browser_profile import AgentBrowserProfile
+from vibe_surf.tools.browser_use_tools import BrowserUseTools
+from vibe_surf.tools.vibesurf_tools import VibeSurfTools
 from vibe_surf.llm.openai_compatible import ChatOpenAICompatible
 from vibe_surf.agents.browser_use_agent import BrowserUseAgent
 from vibe_surf.agents.vibe_surf_agent import VibeSurfAgent
@@ -25,7 +27,7 @@ async def run_single_bu_agent():
         browser_exec_path = "C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe"
     else:
         browser_exec_path = "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome"
-    browser_profile = BrowserProfile(
+    browser_profile = AgentBrowserProfile(
         executable_path=browser_exec_path,
         user_data_dir=os.path.abspath('./tmp/chrome/profiles/default'),
         headless=False,
@@ -34,24 +36,24 @@ async def run_single_bu_agent():
     # Use SwarmBrowserSession instead of BrowserSession to disable DVD animation
     main_browser_session = AgentBrowserSession(browser_profile=browser_profile)
     await main_browser_session.start()
-    controller = VibeSurfController()
+    bu_tools = BrowserUseTools()
 
-    llm = ChatOpenAICompatible(model='gemini-2.5-pro',
+    llm = ChatOpenAICompatible(model='gemini-2.5-flash',
                                base_url=os.getenv("OPENAI_ENDPOINT"),
                                api_key=os.getenv("OPENAI_API_KEY"))
-    # task = "Search Google for 'what is browser automation' and tell me the top 3 results"
-    task = r"""
-    1. 在新的tab 导航到 https://github.com/
-    2. 在新的tab 导航到 https://vibemotion.co/
-    3. 在新的tab 导航到 https://browser-use.com/
-    4. 分别总结所有tab的内容(在一步中使用parallel的extract操作, 不要分开三步)，然后保存到 tabs_summary.txt
-    """
+    task = "Search Google for 'Elon Mask' and tell me the top 3 results"
+    # task = r"""
+    # 1. 在新的tab 导航到 https://github.com/
+    # 2. 在新的tab 导航到 https://vibemotion.co/
+    # 3. 在新的tab 导航到 https://browser-use.com/
+    # 4. 分别总结所有tab的内容(在一步中使用parallel的extract操作, 不要分开三步)，然后保存到 tabs_summary.txt
+    # """
     agent = BrowserUseAgent(task=task,
                             llm=llm,
                             browser_session=main_browser_session,
-                            controller=controller,
+                            tools=bu_tools,
                             task_id=main_browser_session.id,
-                            file_system_path="./tmp/agent_workspace")
+                            file_system_path="./tmp/single_bu_tests")
     history = await agent.run()
     print(history.final_result())
     await main_browser_session.kill()
@@ -63,7 +65,7 @@ async def run_multi_bu_agents():
         browser_exec_path = "C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe"
     else:
         browser_exec_path = "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome"
-    browser_profile = BrowserProfile(
+    browser_profile = AgentBrowserProfile(
         executable_path=browser_exec_path,
         user_data_dir=os.path.abspath('./tmp/chrome/profiles/default'),
         headless=False,
@@ -82,12 +84,15 @@ async def run_multi_bu_agents():
         }
     }
     # Use SwarmBrowserSession instead of BrowserSession to disable DVD animation
+    # Use SwarmBrowserSession instead of BrowserSession to disable DVD animation
     main_browser_session = AgentBrowserSession(browser_profile=browser_profile)
     await main_browser_session.start()
+    bu_tools = BrowserUseTools()
     browser_manager = BrowserManager(main_browser_session=main_browser_session)
-    controller = VibeSurfController()
-    # await controller.register_mcp_clients(mcp_server_config)
-    llm = ChatOpenAICompatible(model='gemini-2.5-flash', base_url=os.getenv("OPENAI_ENDPOINT"),
+    # await tools.register_mcp_clients(mcp_server_config)
+
+    llm = ChatOpenAICompatible(model='gemini-2.5-flash',
+                               base_url=os.getenv("OPENAI_ENDPOINT"),
                                api_key=os.getenv("OPENAI_API_KEY"))
     agent_browser_sessions = await asyncio.gather(
         browser_manager.register_agent("agent-1"),
@@ -98,8 +103,8 @@ async def run_multi_bu_agents():
         BrowserUseAgent(task=task,
                         llm=llm,
                         browser_session=agent_browser_sessions[i],
-                        controller=controller,
-                        file_system_path="./tmp/agent_workspace")
+                        tools=bu_tools,
+                        file_system_path="./tmp/multi_bu_tests")
         for i, task in enumerate([
             # 'Search Google for weather in Tokyo',
             # 'Check Reddit front page title',
@@ -122,8 +127,8 @@ async def run_multi_bu_agents():
         print(await agent_browser_sessions[i].get_tabs())
         print(ret.final_result())
     await browser_manager.close()
-    await controller.unregister_mcp_clients()
     await main_browser_session.kill()
+    await bu_tools.unregister_mcp_clients()
 
 
 async def test_vibe_surf_agent():
@@ -133,37 +138,29 @@ async def test_vibe_surf_agent():
         browser_exec_path = "C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe"
     else:
         browser_exec_path = "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome"
-    from screeninfo import get_monitors
-    primary_monitor = get_monitors()[0]
-    browser_profile = BrowserProfile(
+    browser_profile = AgentBrowserProfile(
         executable_path=browser_exec_path,
         user_data_dir=os.path.abspath('./tmp/chrome/profiles/default'),
         headless=False,
-        keep_alive=True,
-        highlight_elements=True,
-        window_size={"width": 1100, "height": 1280}
-        # window_size={"width": primary_monitor.width, "height": primary_monitor.height}
+        keep_alive=True
     )
-    
-    # Initialize components
+    # Use SwarmBrowserSession instead of BrowserSession to disable DVD animation
     main_browser_session = AgentBrowserSession(browser_profile=browser_profile)
     await main_browser_session.start()
-    
+    vs_tools = VibeSurfTools()
     browser_manager = BrowserManager(main_browser_session=main_browser_session)
-    controller = VibeSurfController()
-    
-    llm = ChatOpenAICompatible(
-        model='gemini-2.5-flash',
-        base_url=os.getenv("OPENAI_ENDPOINT"),
-        api_key=os.getenv("OPENAI_API_KEY")
-    )
+    llm = ChatOpenAICompatible(model='gemini-2.5-flash',
+                               base_url=os.getenv("OPENAI_ENDPOINT"),
+                               api_key=os.getenv("OPENAI_API_KEY"))
+
     
     # Create VibeSurfAgent
     agent = VibeSurfAgent(
         llm=llm,
         browser_manager=browser_manager,
-        controller=controller,
-        workspace_dir=os.path.abspath("./tmp/swarm_surf_workspace_test")
+        tools=vs_tools,
+        workspace_dir=os.path.abspath("./tmp/vibesurf_tests"),
+        calculate_token_cost=True
     )
     
     try:
@@ -194,11 +191,11 @@ async def test_vibe_surf_agent():
 
         # Test 4: Browser parallel task
         print("🧪 Testing browser parallel tasks...")
-        browser_task = "Search for Dify, n8n, browser-use and get latest news"
+        browser_task = "Search for Dify, n8n, browser-use and click into their own homepage, take screenshot and save"
         result4 = await agent.run(browser_task)
         print(f"✅ Browser task result:")
         pprint.pprint(result4)
-        with open("./tmp/swarm_surf_workspace_test/parallel_test.md", "w", encoding='utf-8') as fw:
+        with open("./tmp/vibesurf_tests/parallel_test.md", "w", encoding='utf-8') as fw:
             fw.write(result4)
         assert result4 is not None and len(result4) > 0
         print("🎉 All VibeSurfAgent tests passed!")
@@ -219,35 +216,31 @@ async def test_vibe_surf_agent_control():
         browser_exec_path = "C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe"
     else:
         browser_exec_path = "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome"
-    
-    browser_profile = BrowserProfile(
+    browser_profile = AgentBrowserProfile(
         executable_path=browser_exec_path,
-        user_data_dir=os.path.abspath('./tmp/chrome/profiles/control_test'),
-        headless=False,  # Keep visible for control testing
+        user_data_dir=os.path.abspath('./tmp/chrome/profiles/default'),
+        headless=False,
         keep_alive=True
     )
-    
-    # Initialize components
+    # Use SwarmBrowserSession instead of BrowserSession to disable DVD animation
     main_browser_session = AgentBrowserSession(browser_profile=browser_profile)
     await main_browser_session.start()
-    
+    vs_tools = VibeSurfTools()
     browser_manager = BrowserManager(main_browser_session=main_browser_session)
-    controller = VibeSurfController()
-    
-    llm = ChatOpenAICompatible(
-        model='gemini-2.5-flash',
-        base_url=os.getenv("OPENAI_ENDPOINT"),
-        api_key=os.getenv("OPENAI_API_KEY")
-    )
+    llm = ChatOpenAICompatible(model='gemini-2.5-pro',
+                               base_url=os.getenv("OPENAI_ENDPOINT"),
+                               api_key=os.getenv("OPENAI_API_KEY"))
     
+
     # Create VibeSurfAgent
     agent = VibeSurfAgent(
         llm=llm,
         browser_manager=browser_manager,
-        controller=controller,
-        workspace_dir="./tmp/swarm_surf_control_test"
+        tools=vs_tools,
+        workspace_dir=os.path.abspath("./tmp/vibesurf_tests"),
+        calculate_token_cost=True
     )
-    
+
     try:
         print("🧪 Testing VibeSurfAgent control functionality...")
         
@@ -375,6 +368,6 @@ async def test_vibe_surf_agent_control():
 
 if __name__ == "__main__":
     # asyncio.run(run_single_bu_agent())
-    asyncio.run(run_multi_bu_agents())
-    # asyncio.run(test_vibe_surf_agent())
+    # asyncio.run(run_multi_bu_agents())
+    asyncio.run(test_vibe_surf_agent())
     # asyncio.run(test_vibe_surf_agent_control())

{vibesurf-0.1.10 → vibesurf-0.1.11}/tests/test_backend_api.py

@@ -206,33 +206,33 @@ async def test_llm_profile_management():
 
 
 async def test_controller_configuration():
-    """Test controller/MCP server configuration"""
+    """Test tools/MCP server configuration"""
     print("\n🧪 Testing Controller Configuration...")
     async with BackendAPITester() as tester:
         await tester.wait_for_backend()
 
-        # 1. Get current controller config
-        print("\n📋 Testing get controller config...")
-        config_resp = await tester.get("/config/controller")
+        # 1. Get current tools config
+        print("\n📋 Testing get tools config...")
+        config_resp = await tester.get("/config/tools")
         print(f"Controller config response: {config_resp}")
 
-        # 2. Update controller config (using new Pydantic model)
-        print("\n✏️ Testing update controller config...")
+        # 2. Update tools config (using new Pydantic model)
+        print("\n✏️ Testing update tools config...")
         controller_config = {
             "exclude_actions": ["scroll_up", "scroll_down"],
             "max_actions_per_task": 150,
             "display_files_in_done_text": False
         }
 
-        update_resp = await tester.post("/config/controller", controller_config)
+        update_resp = await tester.post("/config/tools", controller_config)
         if update_resp["status"] == 200:
             print("✅ Controller configuration updated successfully")
         else:
             print(f"❌ Controller config update failed: {update_resp}")
 
         # 3. Verify the update
-        print("\n🔍 Testing verify controller config update...")
-        verify_resp = await tester.get("/config/controller")
+        print("\n🔍 Testing verify tools config update...")
+        verify_resp = await tester.get("/config/tools")
         if verify_resp["status"] == 200:
             print("✅ Controller configuration verified")
 
@@ -440,7 +440,7 @@ async def test_configuration_status():
             status_data = status_resp["data"]
             print(f"Overall status: {status_data.get('overall_status')}")
             print(f"LLM profiles: {status_data.get('llm_profiles')}")
-            print(f"Controller initialized: {status_data.get('controller', {}).get('initialized')}")
+            print(f"Controller initialized: {status_data.get('tools', {}).get('initialized')}")
             print(f"Browser manager initialized: {status_data.get('browser_manager', {}).get('initialized')}")
             print(f"Swarm agent initialized: {status_data.get('swarm_agent', {}).get('initialized')}")
 

vibesurf-0.1.10/tests/test_controller.py → vibesurf-0.1.11/tests/test_tools.py

@@ -10,9 +10,9 @@ from dotenv import load_dotenv
 load_dotenv()
 
 
-async def test_controller_with_mcp():
+async def test_tools_with_mcp():
     import os
-    from vibe_surf.controller.vibesurf_tools import VibeSurfController
+    from vibe_surf.tools.vibesurf_tools import VibeSurfTools
 
     mcp_server_config = {
         "mcpServers": {
@@ -43,7 +43,7 @@ async def test_controller_with_mcp():
         }
     }
 
-    controller = VibeSurfController(mcp_server_config=mcp_server_config)
+    controller = VibeSurfTools(mcp_server_config=mcp_server_config)
     await controller.register_mcp_clients()
     pdb.set_trace()
     # action_name = "mcp.desktop-commander.start_process"
@@ -71,7 +71,7 @@ async def test_controller_with_mcp():
     #     pid = int(result.split("\n")[0].split("PID")[-1].strip())
     #     pdb.set_trace()
     #     action_name = "mcp.desktop-commander.read_process_output"
-    #     action_info = controller.registry.registry.actions[action_name]
+    #     action_info = tools.registry.registry.actions[action_name]
     #     param_model = action_info.param_model
     #     print(param_model.model_json_schema())
     #     params = {"pid": pid}
@@ -80,7 +80,7 @@ async def test_controller_with_mcp():
     #     output_result = ""
     #     while True:
     #         time.sleep(1)
-    #         result = await controller.act(action_model)
+    #         result = await tools.act(action_model)
     #         result = result.extracted_content
     #         if result:
     #             output_result = result
@@ -90,15 +90,34 @@ async def test_controller_with_mcp():
 
 
 async def test_filesystem():
-    from browser_use.filesystem.file_system import FileSystem
+    from vibe_surf.tools.file_system import CustomFileSystem
 
     file_system_path = r"E:\AIBrowser\VibeSurf\tmp\vibesurf_workspace"
-    filesystem = FileSystem(file_system_path)
-    result = await filesystem.read_file(
-        "E:/AIBrowser/VibeSurf/tmp/vibesurf_workspace/068b3165-ddb7-7a49-8000-d4df8a16ae2d/upload_files/Ayers Application Form.md")
+    filesystem = CustomFileSystem(file_system_path)
+    result = await filesystem.create_file("reports/final_report.html")
     print(result)
+    result = filesystem.get_absolute_path("reports/final_report.html")
+    print(result)
+
+
+async def test_bu_tools():
+    import os
+    from vibe_surf.tools.browser_use_tools import BrowserUseTools
+
+    tools = BrowserUseTools()
+    print(tools.registry.registry.actions.keys())
+
+
+async def test_vibesurf_tools():
+    import os
+    from vibe_surf.tools.vibesurf_tools import VibeSurfTools
+
+    tools = VibeSurfTools()
+    print(tools.registry.registry.actions.keys())
 
 
 if __name__ == '__main__':
-    # asyncio.run(test_controller_with_mcp())
-    asyncio.run(test_filesystem())
+    # asyncio.run(test_tools_with_mcp())
+    # asyncio.run(test_filesystem())
+    # asyncio.run(test_bu_tools())
+    asyncio.run(test_vibesurf_tools())

{vibesurf-0.1.10 → vibesurf-0.1.11}/vibe_surf/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.1.10'
-__version_tuple__ = version_tuple = (0, 1, 10)
+__version__ = version = '0.1.11'
+__version_tuple__ = version_tuple = (0, 1, 11)
 
-__commit_id__ = commit_id = 'g3e1f64f34'
+__commit_id__ = commit_id = 'g542374226'