deepcode-hku 1.0.2__tar.gz → 1.0.3__tar.gz

{deepcode_hku-1.0.2/deepcode_hku.egg-info → deepcode_hku-1.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: deepcode-hku
-Version: 1.0.2
+Version: 1.0.3
 Summary: AI Research Engine - Transform research papers into working code automatically
 Home-page: https://github.com/HKUDS/DeepCode
 Author: DeepCodeTeam

{deepcode_hku-1.0.2 → deepcode_hku-1.0.3}/__init__.py

@@ -5,7 +5,7 @@ DeepCode - AI Research Engine
 ⚡ Transform research papers into working code automatically
 """
 
-__version__ = "1.0.2"
+__version__ = "1.0.3"
 __author__ = "DeepCode Team"
 __url__ = "https://github.com/HKUDS/DeepCode"
 

{deepcode_hku-1.0.2 → deepcode_hku-1.0.3/deepcode_hku.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: deepcode-hku
-Version: 1.0.2
+Version: 1.0.3
 Summary: AI Research Engine - Transform research papers into working code automatically
 Home-page: https://github.com/HKUDS/DeepCode
 Author: DeepCodeTeam

{deepcode_hku-1.0.2 → deepcode_hku-1.0.3}/deepcode_hku.egg-info/SOURCES.txt

@@ -24,6 +24,7 @@ deepcode_hku.egg-info/top_level.txt
 prompts/code_prompts.py
 schema/mcp-agent.config.schema.json
 tools/__init__.py
+tools/bocha_search_server.py
 tools/code_implementation_server.py
 tools/code_indexer.py
 tools/code_reference_indexer.py

{deepcode_hku-1.0.2 → deepcode_hku-1.0.3}/mcp_agent.config.yaml

@@ -1,5 +1,8 @@
 $schema: ./schema/mcp-agent.config.schema.json
 
+# Default search server configuration
+# Options: "brave" or "bocha-mcp"
+default_search_server: "brave"
 
 execution_engine: asyncio
 logger:
@@ -64,6 +67,12 @@ mcp:
       env:
         PYTHONPATH: "."
       description: "Code reference indexer server - Provides intelligent code reference search from indexed repositories"
+    bocha-mcp:
+      command: "python3"
+      args: ["tools/bocha_search_server.py"]
+      env:
+        PYTHONPATH: "."
+        BOCHA_API_KEY: ""
 
 openai:
   # Secrets (API keys, etc.) are stored in an mcp_agent.secrets.yaml file which can be gitignored

deepcode_hku-1.0.3/tools/bocha_search_server.py

@@ -0,0 +1,219 @@
+import os
+import sys
+import json
+
+import httpx
+from dotenv import load_dotenv
+from mcp.server.fastmcp import FastMCP
+
+load_dotenv()
+
+
+# Initialize FastMCP server
+server = FastMCP(
+    "bocha-search-mcp",
+    prompt="""
+# Bocha Search MCP Server
+
+Bocha is a Chinese search engine for AI, This server provides tools for searching the web using Bocha Search API.
+It allows you to get enhanced search details from billions of web documents, including weather, news, wikis, healthcare, train tickets, images, and more.
+
+## Available Tools
+
+### 1. bocha_web_search
+Search with Bocha Web Search and get enhanced search details from billions of web documents, including page titles, urls, summaries, site names, site icons, publication dates, image links, and more.
+
+### 2. bocha_ai_search
+Search with Bocha AI Search, recognizes the semantics of search terms and additionally returns structured modal cards with content from vertical domains.
+
+## Output Format
+
+All search results will be formatted as text with clear sections for each
+result item, including:
+
+- Bocha Web search: Title, URL, Description, Published date and Site name
+- Bocha AI search: Title, URL, Description, Published date, Site name, and structured data card
+
+If the API key is missing or invalid, appropriate error messages will be returned.
+""",
+)
+
+
+@server.tool()
+async def bocha_web_search(
+    query: str, freshness: str = "noLimit", count: int = 10
+) -> str:
+    """Search with Bocha Web Search and get enhanced search details from billions of web documents,
+    including page titles, urls, summaries, site names, site icons, publication dates, image links, and more.
+
+    Args:
+        query: Search query (required)
+        freshness: The time range for the search results. (Available options YYYY-MM-DD, YYYY-MM-DD..YYYY-MM-DD, noLimit, oneYear, oneMonth, oneWeek, oneDay. Default is noLimit)
+        count: Number of results (1-50, default 10)
+    """
+    # Get API key from environment
+    boch_api_key = os.environ.get("BOCHA_API_KEY", "")
+
+    if not boch_api_key:
+        return (
+            "Error: Bocha API key is not configured. Please set the "
+            "BOCHA_API_KEY environment variable."
+        )
+
+    # Endpoint
+    endpoint = "https://api.bochaai.com/v1/web-search?utm_source=bocha-mcp-local"
+
+    try:
+        payload = {
+            "query": query,
+            "summary": True,
+            "freshness": freshness,
+            "count": count,
+        }
+
+        headers = {
+            "Authorization": f"Bearer {boch_api_key}",
+            "Content-Type": "application/json",
+        }
+
+        async with httpx.AsyncClient() as client:
+            response = await client.post(
+                endpoint, headers=headers, json=payload, timeout=10.0
+            )
+
+            response.raise_for_status()
+            resp = response.json()
+            if "data" not in resp:
+                return "Search error."
+
+            data = resp["data"]
+
+            if "webPages" not in data:
+                return "No results found."
+
+            results = []
+            for result in data["webPages"]["value"]:
+                results.append(
+                    f"Title: {result['name']}\n"
+                    f"URL: {result['url']}\n"
+                    f"Description: {result['summary']}\n"
+                    f"Published date: {result['datePublished']}\n"
+                    f"Site name: {result['siteName']}"
+                )
+
+            return "\n\n".join(results)
+
+    except httpx.HTTPStatusError as e:
+        return f"Bocha Web Search API HTTP error occurred: {e.response.status_code} - {e.response.text}"
+    except httpx.RequestError as e:
+        return f"Error communicating with Bocha Web Search API: {str(e)}"
+    except Exception as e:
+        return f"Unexpected error: {str(e)}"
+
+
+@server.tool()
+async def bocha_ai_search(
+    query: str, freshness: str = "noLimit", count: int = 10
+) -> str:
+    """Search with Bocha AI Search, recognizes the semantics of search terms
+    and additionally returns structured modal cards with content from vertical domains.
+
+    Args:
+        query: Search query (required)
+        freshness: The time range for the search results. (Available options noLimit, oneYear, oneMonth, oneWeek, oneDay. Default is noLimit)
+        count: Number of results (1-50, default 10)
+    """
+    # Get API key from environment
+    boch_api_key = os.environ.get("BOCHA_API_KEY", "")
+
+    if not boch_api_key:
+        return (
+            "Error: Bocha API key is not configured. Please set the "
+            "BOCHA_API_KEY environment variable."
+        )
+
+    # Endpoint
+    endpoint = "https://api.bochaai.com/v1/ai-search?utm_source=bocha-mcp-local"
+
+    try:
+        payload = {
+            "query": query,
+            "freshness": freshness,
+            "count": count,
+            "answer": False,
+            "stream": False,
+        }
+
+        headers = {
+            "Authorization": f"Bearer {boch_api_key}",
+            "Content-Type": "application/json",
+        }
+
+        async with httpx.AsyncClient() as client:
+            response = await client.post(
+                endpoint, headers=headers, json=payload, timeout=10.0
+            )
+
+            response.raise_for_status()
+            response = response.json()
+            results = []
+            if "messages" in response:
+                for message in response["messages"]:
+                    content = {}
+                    try:
+                        content = json.loads(message["content"])
+                    except (json.JSONDecodeError, TypeError):
+                        content = {}
+
+                    # 网页
+                    if message["content_type"] == "webpage":
+                        if "value" in content:
+                            for item in content["value"]:
+                                results.append(
+                                    f"Title: {item['name']}\n"
+                                    f"URL: {item['url']}\n"
+                                    f"Description: {item['summary']}\n"
+                                    f"Published date: {item['datePublished']}\n"
+                                    f"Site name: {item['siteName']}"
+                                )
+                    elif (
+                        message["content_type"] != "image"
+                        and message["content"] != "{}"
+                    ):
+                        results.append(message["content"])
+
+            if not results:
+                return "No results found."
+
+            return "\n\n".join(results)
+
+    except httpx.HTTPStatusError as e:
+        return f"Bocha AI Search API HTTP error occurred: {e.response.status_code} - {e.response.text}"
+    except httpx.RequestError as e:
+        return f"Error communicating with Bocha AI Search API: {str(e)}"
+    except Exception as e:
+        return f"Unexpected error: {str(e)}"
+
+
+def main():
+    """Initialize and run the MCP server."""
+
+    # Check for required environment variables
+    if "BOCHA_API_KEY" not in os.environ:
+        print(
+            "Error: BOCHA_API_KEY environment variable is required",
+            file=sys.stderr,
+        )
+        print(
+            "Get a Bocha API key from: " "https://open.bochaai.com",
+            file=sys.stderr,
+        )
+        sys.exit(1)
+
+    print("Starting Bocha Search MCP server...", file=sys.stderr)
+
+    server.run(transport="stdio")
+
+
+if __name__ == "__main__":
+    main()

{deepcode_hku-1.0.2 → deepcode_hku-1.0.3}/workflows/agent_orchestration_engine.py

@@ -31,7 +31,7 @@ import json
 import os
 import re
 import yaml
-from typing import Callable, Dict, Optional, Tuple
+from typing import Callable, Dict, List, Optional, Tuple
 
 # MCP Agent imports
 from mcp_agent.agents.agent import Agent
@@ -99,6 +99,57 @@ def get_preferred_llm_class(config_path: str = "mcp_agent.secrets.yaml"):
         return OpenAIAugmentedLLM
 
 
+def get_default_search_server(config_path: str = "mcp_agent.config.yaml"):
+    """
+    Get the default search server from configuration.
+
+    Args:
+        config_path: Path to the main configuration file
+
+    Returns:
+        str: The default search server name ("brave" or "bocha-mcp")
+    """
+    try:
+        if os.path.exists(config_path):
+            with open(config_path, "r", encoding="utf-8") as f:
+                config = yaml.safe_load(f)
+
+            default_server = config.get("default_search_server", "brave")
+            print(f"🔍 Using search server: {default_server}")
+            return default_server
+        else:
+            print(f"⚠️ Config file {config_path} not found, using default: brave")
+            return "brave"
+    except Exception as e:
+        print(f"⚠️ Error reading config file {config_path}: {e}")
+        print("🔍 Falling back to default search server: brave")
+        return "brave"
+
+
+def get_search_server_names(
+    additional_servers: Optional[List[str]] = None,
+) -> List[str]:
+    """
+    Get server names list with the configured default search server.
+
+    Args:
+        additional_servers: Optional list of additional servers to include
+
+    Returns:
+        List[str]: List of server names including the default search server
+    """
+    default_search = get_default_search_server()
+    server_names = [default_search]
+
+    if additional_servers:
+        # Add additional servers, avoiding duplicates
+        for server in additional_servers:
+            if server not in server_names:
+                server_names.append(server)
+
+    return server_names
+
+
 def extract_clean_json(llm_output: str) -> str:
     """
     Extract clean JSON from LLM output, removing all extra text and formatting.
@@ -193,7 +244,7 @@ async def run_research_analyzer(prompt_text: str, logger) -> str:
         analyzer_agent = Agent(
             name="ResearchAnalyzerAgent",
             instruction=PAPER_INPUT_ANALYZER_PROMPT,
-            server_names=["brave"],
+            server_names=get_search_server_names(),
         )
 
         async with analyzer_agent:
@@ -341,12 +392,12 @@ async def run_code_analyzer(paper_dir: str, logger) -> str:
     algorithm_analysis_agent = Agent(
         name="AlgorithmAnalysisAgent",
         instruction=PAPER_ALGORITHM_ANALYSIS_PROMPT,
-        server_names=["filesystem", "brave"],
+        server_names=get_search_server_names(additional_servers=["filesystem"]),
     )
     code_planner_agent = Agent(
         name="CodePlannerAgent",
         instruction=CODE_PLANNING_PROMPT,
-        server_names=["brave"],
+        server_names=get_search_server_names(),
     )
 
     code_aggregator_agent = ParallelLLM(
@@ -931,9 +982,7 @@ async def run_chat_planning_agent(user_input: str, logger) -> str:
         chat_planning_agent = Agent(
             name="ChatPlanningAgent",
             instruction=CHAT_AGENT_PLANNING_PROMPT,
-            server_names=[
-                "brave"
-            ],  # Add tools if needed for web search or other capabilities
+            server_names=get_search_server_names(),  # Dynamic search server configuration
         )
 
         async with chat_planning_agent: