vibesurf 0.1.22__tar.gz → 0.1.24__tar.gz

{vibesurf-0.1.22 → vibesurf-0.1.24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vibesurf
-Version: 0.1.22
+Version: 0.1.24
 Summary: VibeSurf: A powerful browser assistant for vibe surfing
 Author: Shao Warm
 License: Apache-2.0
@@ -43,6 +43,7 @@ Requires-Dist: nanoid>=2.0.0
 Requires-Dist: markdownify>=1.2.0
 Requires-Dist: pathvalidate>=3.3.1
 Requires-Dist: dashscope>=1.24.5
+Requires-Dist: yfinance>=0.2.66
 Dynamic: license-file
 
 # VibeSurf: A powerful browser assistant for vibe surfing

{vibesurf-0.1.22 → vibesurf-0.1.24}/pyproject.toml

@@ -44,6 +44,7 @@ dependencies = [
     "markdownify>=1.2.0",
     "pathvalidate>=3.3.1",
     "dashscope>=1.24.5",
+    "yfinance>=0.2.66",
 ]
 
 [project.urls]

{vibesurf-0.1.22 → vibesurf-0.1.24}/tests/test_agents.py

@@ -18,9 +18,11 @@ 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 browser_use.llm.deepseek.chat import ChatDeepSeek
 from vibe_surf.agents.browser_use_agent import BrowserUseAgent
 from vibe_surf.agents.vibe_surf_agent import VibeSurfAgent
 
+
 async def run_single_bu_agent():
     import platform
     if platform.system() != "Darwin":
@@ -46,9 +48,13 @@ async def run_single_bu_agent():
     #                            base_url=os.getenv("ALIBABA_ENDPOINT"),
     #                            api_key=os.getenv("ALIBABA_API_KEY"))
 
-    llm = ChatOpenAICompatible(model='kimi-k2-turbo-preview',
-                               base_url=os.getenv("MOONSHOT_ENDPOINT"),
-                               api_key=os.getenv("MOONSHOT_API_KEY"))
+    # llm = ChatOpenAICompatible(model='kimi-k2-turbo-preview',
+    #                            base_url=os.getenv("MOONSHOT_ENDPOINT"),
+    #                            api_key=os.getenv("MOONSHOT_API_KEY"))
+
+    llm = ChatOpenAICompatible(model='deepseek-reasoner',
+                               base_url=os.getenv("DEEPSEEK_ENDPOINT"),
+                               api_key=os.getenv("DEEPSEEK_API_KEY"))
 
     task = "Search Google for 'Elon Mask' and tell me the top 3 results"
 
@@ -159,20 +165,22 @@ async def test_vibe_surf_agent():
     await main_browser_session.start()
     vs_tools = VibeSurfTools()
     browser_manager = BrowserManager(main_browser_session=main_browser_session)
-    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"))
+
+    llm = ChatOpenAICompatible(model='deepseek-chat',
+                               base_url=os.getenv("DEEPSEEK_ENDPOINT"),
+                               api_key=os.getenv("DEEPSEEK_API_KEY"))
 
-    
     # Create VibeSurfAgent
     agent = VibeSurfAgent(
         llm=llm,
         browser_manager=browser_manager,
         tools=vs_tools,
-        workspace_dir=os.path.abspath("./tmp/vibesurf_tests"),
-        calculate_token_cost=True
+        workspace_dir=os.path.abspath("./tmp/vibesurf_tests")
     )
-    
+
     try:
         # Test 1: Simple task (should not require browser)
         print("🧪 Testing simple task...")
@@ -200,16 +208,16 @@ async def test_vibe_surf_agent():
         # print("🎉 All VibeSurfAgent tests passed!")
 
         # Test 4: Browser parallel task
-        print("🧪 Testing browser parallel tasks...")
-        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/vibesurf_tests/parallel_test.md", "w", encoding='utf-8') as fw:
-            fw.write(result4)
-        assert result4 is not None and len(result4) > 0
+        # print("🧪 Testing browser parallel tasks...")
+        # 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/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!")
-        
+
     except Exception as e:
         print(f"❌ VibeSurfAgent test failed: {e}")
         raise e
@@ -240,7 +248,6 @@ async def test_vibe_surf_agent_control():
     llm = ChatOpenAICompatible(model='gemini-2.5-pro',
                                base_url=os.getenv("OPENAI_ENDPOINT"),
                                api_key=os.getenv("OPENAI_API_KEY"))
-    
 
     # Create VibeSurfAgent
     agent = VibeSurfAgent(
@@ -253,69 +260,69 @@ async def test_vibe_surf_agent_control():
 
     try:
         print("🧪 Testing VibeSurfAgent control functionality...")
-        
+
         # Test 1: Status check when idle
         print("📊 Testing initial status...")
         status = agent.get_status()
         print(f"Initial status: {status.overall_status}")
         assert status.overall_status == "idle"
-        
+
         # Test 2: Start a long-running browser task
         print("🚀 Starting long-running browser task...")
         browser_task = "Search for Dify, n8n, langflow and gather relative information, and generate a detailed report for comparison"
-        
+
         # Start task in background
         async def run_task():
             return await agent.run(browser_task)
-        
+
         task_coroutine = asyncio.create_task(run_task())
-        
+
         # Wait a bit for task to start
         await asyncio.sleep(10)
-        
+
         # Test 3: Check status during execution
         print("📊 Checking status during execution...")
         status = agent.get_status()
         print(f"Running status: {status.overall_status}")
         print(f"Progress: {status.progress}")
         print(f"Active agents: {len(status.agent_statuses)}")
-        
+
         # Test 4: Pause execution
         print("⏸️ Testing pause functionality...")
         pause_result = await agent.pause("Testing pause functionality")
         print(f"Pause result: {pause_result.success} - {pause_result.message}")
         assert pause_result.success
-        
+
         # Check status after pause
         await asyncio.sleep(1)
         status = agent.get_status()
         print(f"Paused status: {status.overall_status}")
         assert status.overall_status == "paused"
-        
+
         # Test 5: Resume execution
         print("▶️ Testing resume functionality...")
         resume_result = await agent.resume("Testing resume functionality")
         print(f"Resume result: {resume_result.success} - {resume_result.message}")
         assert resume_result.success
-        
+
         # Check status after resume
         await asyncio.sleep(1)
         status = agent.get_status()
         print(f"Resumed status: {status.overall_status}")
-        
+
         # Let it run a bit more
         await asyncio.sleep(50)
-        
+
         # Test 6: Stop execution
         print("🛑 Testing stop functionality...")
         stop_result = await agent.stop("Testing stop functionality")
         print(f"Stop result: {stop_result.success} - {stop_result.message}")
-        
+
         # Check status after stop (should be stopped even if stop had issues)
         await asyncio.sleep(1)
         status = agent.get_status()
         print(f"Stopped status: {status.overall_status}")
-        
+
         # Wait for task to complete (it should be cancelled)
         try:
             result = await asyncio.wait_for(task_coroutine, timeout=3)
@@ -329,38 +336,38 @@ async def test_vibe_surf_agent_control():
                 pass
         except asyncio.CancelledError:
             print("✅ Task was cancelled as expected")
-        
+
         # Verify stop worked (may have timed out but should still be effective)
         if stop_result.success:
             assert status.overall_status == "idle"
         else:
             print(f"⚠️ Stop operation had issues but continuing: {stop_result.message}")
-        
+
         # Test 7: Test simple task control (should work quickly)
         print("🔄 Testing control on simple task...")
         simple_task = "Find out who is the founder of Browser-Use."
-        
+
         async def run_simple_task():
             return await agent.run(simple_task)
-        
+
         simple_task_coroutine = asyncio.create_task(run_simple_task())
-        
+
         # Pause quickly
         await asyncio.sleep(0.5)
         pause_result = await agent.pause("Testing simple task pause")
         print(f"Simple task pause: {pause_result.success}")
-        
+
         # Resume
         await asyncio.sleep(0.5)
         resume_result = await agent.resume("Testing simple task resume")
         print(f"Simple task resume: {resume_result.success}")
-        
+
         # Let it complete
         simple_result = await simple_task_coroutine
         print(f"Simple task completed: {len(simple_result) > 0}")
         print(simple_result)
         print("🎉 All VibeSurfAgent control tests passed!")
-        
+
     except Exception as e:
         print(f"❌ VibeSurfAgent control test failed: {e}")
         import traceback
@@ -377,7 +384,7 @@ async def test_vibe_surf_agent_control():
 
 
 if __name__ == "__main__":
-    asyncio.run(run_single_bu_agent())
+    # asyncio.run(run_single_bu_agent())
     # asyncio.run(run_multi_bu_agents())
-    # asyncio.run(test_vibe_surf_agent())
+    asyncio.run(test_vibe_surf_agent())
     # asyncio.run(test_vibe_surf_agent_control())

{vibesurf-0.1.22 → vibesurf-0.1.24}/tests/test_browser.py

@@ -117,7 +117,6 @@ async def test_browser_state_capture(manager: BrowserManager):
     #         f"🔍 DIAGNOSIS: {agent_name} navigation completed at {task_end - start_time:.3f}s (duration: {task_end - task_start:.3f}s)")
     #     return None
 
-
     # Navigate to different pages
     await asyncio.gather(
         navigate_with_timing(agent1, "https://www.python.org", "agent1"),
@@ -240,6 +239,29 @@ async def test_browser_state_capture(manager: BrowserManager):
     print(main_tabs)
 
 
+async def get_all_css_selector(browser_session: AgentBrowserSession):
+    target_id = await browser_session.navigate_to_url("https://github.com/", new_tab=True)
+    result = await browser_session.cdp_client.send.Target.attachToTarget({'targetId': target_id, 'flatten': True})
+    session_id = result['sessionId']
+    doc_result = await browser_session.cdp_client.send.DOM.getDocument(session_id=session_id)
+    document_node_id = doc_result['root']['nodeId']
+    from browser_use.dom.service import DomService
+    dom_service = DomService(browser_session)
+    pdb.set_trace()
+    # Query selector all
+    query_params = {'nodeId': document_node_id}
+    result = await browser_session.cdp_client.send.DOM.querySelectorAll(query_params, session_id=session_id)
+
+    elements = []
+
+    # Convert node IDs to backend node IDs
+    for node_id in result['nodeIds']:
+        # Get backend node ID
+        describe_params = {'nodeId': node_id}
+        node_result = await browser_session.cdp_client.send.DOM.describeNode(describe_params, session_id=session_id)
+        pdb.set_trace()
+
+
 async def main():
     """
     Main function to run all browser session tests.
@@ -273,7 +295,8 @@ async def main():
             # await test_manual_page_assignment(manager)
             # await test_agent_cleanup(manager)
             # await test_agent_tab_isolation(manager)
-            await test_browser_state_capture(manager)
+            # await test_browser_state_capture(manager)
+            await get_all_css_selector(main_browser_session)
 
     except Exception as e:
         logging.error(f"An error occurred during tests: {e}", exc_info=True)

{vibesurf-0.1.22 → vibesurf-0.1.24}/tests/test_tools.py

@@ -116,8 +116,19 @@ async def test_vibesurf_tools():
     print(tools.registry.registry.actions.keys())
 
 
+async def test_finance_tools():
+    from vibe_surf.tools.finance_tools import FinanceDataRetriever
+
+    retriever = FinanceDataRetriever('TSLA')
+
+    result = retriever.get_finance_data(["get_news"])
+
+    pdb.set_trace()
+
+
 if __name__ == '__main__':
     # asyncio.run(test_tools_with_mcp())
     # asyncio.run(test_filesystem())
     # asyncio.run(test_bu_tools())
-    asyncio.run(test_vibesurf_tools())
+    # asyncio.run(test_vibesurf_tools())
+    asyncio.run(test_finance_tools())

{vibesurf-0.1.22 → vibesurf-0.1.24}/vibe_surf/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.1.22'
-__version_tuple__ = version_tuple = (0, 1, 22)
+__version__ = version = '0.1.24'
+__version_tuple__ = version_tuple = (0, 1, 24)
 
-__commit_id__ = commit_id = 'g95ec3c7e7'
+__commit_id__ = commit_id = 'gebf62a182'

{vibesurf-0.1.22 → vibesurf-0.1.24}/vibe_surf/agents/prompts/vibe_surf_prompt.py

@@ -65,6 +65,15 @@ You will receive contextual information including:
 - Include file references in task descriptions when relevant
 - All file operations automatically resolve relative to the workspace directory
 
+## Skills Command Processing
+- When users input commands in `/skill_name` format, please use the corresponding skill action:
+- **Tab Targeting[Optional]**: Such as `/crawl @1234` → Execute `skill_crawl` with tab_id "1234"
+- **Parameter Processing**: Sometimes user provide uncompleted or simple prompt, please convert it to correct and optimized params. Such as convert natural language to valid JavaScript for code skill
+- **Special Cases**: `skill_deep_research` only returns guidelines only, then follow guidelines to conduct actual research
+- **Execution Policy**: Skill actions execute only once (no need to retry if errors occur), and all results - whether successful or failed - should be presented to users in structured markdown format.
+- **Follow-up Operations**: When users input skill operations without specifying additional tasks, do not automatically perform subsequent operations. Only perform additional tool operations when users specifically request actions like saving results to files or writing reports.
+- After `/search` completion, NEVER use browser agent to deeply investigate search result (unless explicitly emphasized by the user). The user usually only need the search results. Just return the search results.
+
 ## Language Adaptability
 
 **Critical**: Your output language must match the user's request language. If the user communicates in Chinese, respond in Chinese. If in English, respond in English. Maintain consistency throughout the interaction.
@@ -77,6 +86,7 @@ Before executing any action:
 3. **Plan Resource Usage**: Consider tab management and session optimization
 4. **Validate Completeness**: Ensure all user requirements are addressed
 
+
 Execute with precision, leverage concurrent capabilities for efficiency, and deliver professional results that exceed expectations.
 """
 

{vibesurf-0.1.22 → vibesurf-0.1.24}/vibe_surf/agents/vibe_surf_agent.py

@@ -479,6 +479,16 @@ async def _vibesurf_agent_node_impl(state: VibeSurfState) -> VibeSurfState:
                     llm=vibesurf_agent.llm,
                     file_system=vibesurf_agent.file_system,
                 )
+                if action_name in ["skill_search", "skill_crawl", "skill_summary", "skill_finance"]:
+                    state.current_step = "END"
+                    # Format final response
+                    final_response = f"{result.extracted_content}"
+                    await log_agent_activity(state, agent_name, "result", final_response)
+                    state.final_response = final_response
+                    logger.debug(final_response)
+                    state.is_complete = True
+                    return state
+
                 state.current_step = "vibesurf_agent"
                 if result.extracted_content:
                     vibesurf_agent.message_history.append(
@@ -1479,8 +1489,9 @@ Please continue with your assigned work, incorporating this guidance only if it'
         """
         activity_entry = {
             "agent_name": 'user',
-            "agent_status": 'request',  # working, result, error
-            "agent_msg": f"{new_task}"
+            "agent_status": 'additional_request',  # working, result, error
+            "agent_msg": f"{new_task}",
+            "timestamp": datetime.now().isoformat()
         }
         self.activity_logs.append(activity_entry)
 

vibesurf-0.1.24/vibe_surf/backend/api/agent.py

@@ -0,0 +1,38 @@
+"""
+Agent API endpoints
+"""
+from typing import List
+from fastapi import APIRouter, HTTPException
+
+from vibe_surf.logger import get_logger
+
+logger = get_logger(__name__)
+
+router = APIRouter(prefix="/agent", tags=["agent"])
+
+
+@router.get("/get_all_skills", response_model=List[str])
+async def get_all_skills():
+    """
+    Get all available skill names from the VibeSurf tools registry.
+    Returns skill names with the 'skill_' prefix removed.
+    """
+    try:
+        from ..shared_state import vibesurf_tools
+        if not vibesurf_tools:
+            raise HTTPException(status_code=500, detail="VibeSurf tools not initialized")
+        
+        # Get all action names from the registry
+        all_actions = vibesurf_tools.registry.registry.actions.keys()
+        
+        # Filter for actions that start with 'skill_' and remove the prefix
+        skill_names = [
+            action_name.replace('skill_', '') 
+            for action_name in all_actions 
+            if action_name.startswith('skill_')
+        ]
+        logger.info(skill_names)
+        return skill_names
+        
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Failed to get skills: {str(e)}")

{vibesurf-0.1.22 → vibesurf-0.1.24}/vibe_surf/backend/api/config.py

@@ -234,7 +234,9 @@ async def update_llm_profile(
         
         # Return updated profile
         updated_profile = await LLMProfileQueries.get_profile(db, profile_name)
-        
+        from ..shared_state import current_llm_profile_name
+        if current_llm_profile_name != profile_name:
+            current_llm_profile_name = None
         # Use safe extraction to avoid greenlet issues
         return LLMProfileResponse(**_profile_to_response_dict(updated_profile))
         

{vibesurf-0.1.22 → vibesurf-0.1.24}/vibe_surf/backend/api/task.py

@@ -287,7 +287,7 @@ async def stop_task(control_request: TaskControlRequest):
                 active_task["end_time"] = datetime.now()
 
             # Clear active task
-            # clear_active_task()
+            clear_active_task()
 
             return {
                 "success": True,

{vibesurf-0.1.22 → vibesurf-0.1.24}/vibe_surf/backend/main.py

@@ -21,6 +21,7 @@ from .api.activity import router as activity_router
 from .api.config import router as config_router
 from .api.browser import router as browser_router
 from .api.voices import router as voices_router
+from .api.agent import router as agent_router
 
 # Import shared state
 from . import shared_state
@@ -53,6 +54,7 @@ app.include_router(activity_router, prefix="/api", tags=["activity"])
 app.include_router(config_router, prefix="/api", tags=["config"])
 app.include_router(browser_router, prefix="/api", tags=["browser"])
 app.include_router(voices_router, prefix="/api", tags=["voices"])
+app.include_router(agent_router, prefix="/api", tags=["agent"])
 
 # Global variable to control browser monitoring task
 browser_monitor_task = None

{vibesurf-0.1.22 → vibesurf-0.1.24}/vibe_surf/backend/utils/llm_factory.py

@@ -142,8 +142,9 @@ def create_llm_from_profile(llm_profile) -> BaseChatModel:
             )
 
         elif provider == "deepseek":
-            return ChatDeepSeek(
+            return ChatOpenAICompatible(
                 model=model,
+                base_url="https://api.deepseek.com",
                 api_key=api_key,
                 **common_params
             )

{vibesurf-0.1.22 → vibesurf-0.1.24}/vibe_surf/browser/agent_browser_session.py

@@ -407,11 +407,11 @@ class AgentBrowserSession(BrowserSession):
         self.logger.info('🚫 VibeSurfBrowserSession: AboutBlankWatchdog disabled - no DVD animation will be shown')
 
         # Initialize DownloadsWatchdog
-        DownloadsWatchdog.model_rebuild()
-        self._downloads_watchdog = DownloadsWatchdog(event_bus=self.event_bus, browser_session=self)
-        self._downloads_watchdog.attach_to_session()
-        if self.browser_profile.auto_download_pdfs:
-            self.logger.info('📄 PDF auto-download enabled for this session')
+        # DownloadsWatchdog.model_rebuild()
+        # self._downloads_watchdog = DownloadsWatchdog(event_bus=self.event_bus, browser_session=self)
+        # self._downloads_watchdog.attach_to_session()
+        # if self.browser_profile.auto_download_pdfs:
+        #     self.logger.info('📄 PDF auto-download enabled for this session')
 
         # Initialize LocalBrowserWatchdog
         LocalBrowserWatchdog.model_rebuild()

{vibesurf-0.1.22 → vibesurf-0.1.24}/vibe_surf/chrome_extension/scripts/api-client.js

@@ -498,6 +498,11 @@ class VibeSurfAPIClient {
     return this.get('/browser/all-tabs');
   }
 
+  // Agent APIs - Get available skills
+  async getAllSkills() {
+    return this.get('/agent/get_all_skills');
+  }
+
   // Utility methods
   delay(ms) {
     return new Promise(resolve => setTimeout(resolve, ms));

{vibesurf-0.1.22 → vibesurf-0.1.24}/vibe_surf/chrome_extension/scripts/main.js

@@ -340,7 +340,7 @@ class VibeSurfApp {
             <div class="input-container">
               <div class="input-main">
                 <div class="textarea-container">
-                  <textarea id="task-input" class="task-input" placeholder="Describe your browsing task..." rows="3"></textarea>
+                  <textarea id="task-input" class="task-input" placeholder="Ask anything (/ for skills, @ to specify tab)" rows="3"></textarea>
                   <div class="input-actions">
                     <button id="attach-file-btn" class="action-btn attach-btn" title="Attach Files">
                       <svg width="16" height="16" viewBox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg">