feature: add search tool research agent

python/valuecell/agents/research_agent/core.py

@@ -3,8 +3,6 @@
 
 from agno.agent import Agent
 from agno.db.in_memory import InMemoryDb
-from agno.models.google import Gemini
-from agno.models.openrouter import OpenRouter
 from edgar import set_identity
 from loguru import logger
 
@@ -16,28 +14,28 @@
 from valuecell.agents.research_agent.sources import (
     fetch_event_sec_filings,
     fetch_periodic_sec_filings,
+    web_search,
 )
 from valuecell.agents.utils.context import build_ctx_from_dep
 from valuecell.core.agent.responses import streaming
 from valuecell.core.types import BaseAgent, StreamResponse
 from valuecell.utils.env import agent_debug_mode_enabled
-
-
-def _get_model_based_on_env():
-    model_id = os.getenv("RESEARCH_AGENT_MODEL_ID")
-    if os.getenv("GOOGLE_API_KEY"):
-        return Gemini(id=model_id or "gemini-2.5-flash")
-    return OpenRouter(id=model_id or "google/gemini-2.5-flash", max_tokens=None)
+from valuecell.utils.model import get_model
 
 
 class ResearchAgent(BaseAgent):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
+        tools = [
+            fetch_periodic_sec_filings,
+            fetch_event_sec_filings,
+            web_search,
+        ]
         self.knowledge_research_agent = Agent(
-            model=_get_model_based_on_env(),
+            model=get_model("RESEARCH_AGENT_MODEL_ID"),
             instructions=[KNOWLEDGE_AGENT_INSTRUCTION],
             expected_output=KNOWLEDGE_AGENT_EXPECTED_OUTPUT,
-            tools=[fetch_periodic_sec_filings, fetch_event_sec_filings],
+            tools=tools,
             knowledge=knowledge,
             db=InMemoryDb(),
             # context
@@ -81,15 +79,3 @@ async def stream(
         logger.info("Financial data analysis completed")
 
         yield streaming.done()
-
-
-if __name__ == "__main__":
-    import asyncio
-
-    async def main():
-        agent = ResearchAgent()
-        query = "Provide a summary of Apple's 2024 all quarterly and annual reports."
-        async for response in agent.stream(query, "test_session", "test_task"):
-            print(response)
-
-    asyncio.run(main())

python/valuecell/agents/research_agent/sources.py

@@ -1,8 +1,12 @@
+import os
 from datetime import date, datetime
 from pathlib import Path
 from typing import Iterable, List, Optional, Sequence
 
 import aiofiles
+from agno.agent import Agent
+from agno.models.google import Gemini
+from agno.models.openrouter import OpenRouter
 from edgar import Company
 from edgar.entity.filings import EntityFilings
 
@@ -187,3 +191,37 @@ async def fetch_event_sec_filings(
         filtered = filtered[:limit]
 
     return await _write_and_ingest(filtered, Path(get_knowledge_path()))
+
+
+async def web_search(query: str) -> str:
+    """Search web for the given query and return a summary of the top results.
+
+    Args:
+        query: The search query string.
+
+    Returns:
+        A summary of the top search results.
+    """
+
+    if os.getenv("WEB_SEARCH_PROVIDER", "google").lower() == "google" and os.getenv(
+        "GOOGLE_API_KEY"
+    ):
+        return await _web_search_google(query)
+
+    model = OpenRouter(id="perplexity/sonar", max_tokens=None)
+    response = await Agent(model=model).arun(query)
+    return response.content
+
+
+async def _web_search_google(query: str) -> str:
+    """Search Google for the given query and return a summary of the top results.
+
+    Args:
+        query: The search query string.
+
+    Returns:
+        A summary of the top search results.
+    """
+    model = Gemini(id="gemini-2.5-flash", search=True)
+    response = await Agent(model=model).arun(query)
+    return response.content

python/valuecell/core/coordinate/planner.py

@@ -12,14 +12,12 @@
 
 import asyncio
 import logging
-import os
 from datetime import datetime
 from typing import Callable, List, Optional
 
 from a2a.types import AgentCard
 from agno.agent import Agent
 from agno.db.in_memory import InMemoryDb
-from agno.models.openrouter import OpenRouter
 
 from valuecell.core.agent.connect import RemoteConnections
 from valuecell.core.coordinate.planner_prompts import (
@@ -30,6 +28,7 @@
 from valuecell.core.types import UserInput
 from valuecell.utils import generate_uuid
 from valuecell.utils.env import agent_debug_mode_enabled
+from valuecell.utils.model import get_model
 
 from .models import ExecutionPlan, PlannerInput, PlannerResponse
 
@@ -152,10 +151,7 @@ async def _analyze_input_and_create_tasks(
         """
         # Create planning agent with appropriate tools and instructions
         agent = Agent(
-            model=OpenRouter(
-                id=os.getenv("PLANNER_MODEL_ID", "google/gemini-2.5-flash"),
-                max_tokens=None,
-            ),
+            model=get_model("PLANNER_MODEL_ID"),
             tools=[
                 # TODO: enable UserControlFlowTools when stable
                 # UserControlFlowTools(),