feat: add http to mcp and tool search tutorial

Author: tian-yuan

01-tutorials/workshop/session4/http_to_mcp/client.py

@@ -0,0 +1,67 @@
+import requests
+import httpx
+import random
+import json
+
+from google.adk.cli.adk_web_server import CreateSessionRequest, RunAgentRequest
+from google.genai.types import Content, Part
+import asyncio
+
+
+if __name__ == "__main__":
+    # Step 0: setup running configs
+    app_name = "travel_planner_advanced"
+    user_id = "agentkit_user"
+    session_id = f"agentkit_sample_session"
+    base_url = "http://127.0.0.1:8000"
+    api_key = "test key"
+
+    task_num = 1   
+    
+
+    # Step 1: create a session
+    def create_session():
+        create_session_request = CreateSessionRequest(
+            session_id = session_id + f"_{random.randint(1, 9999)}",
+        )
+
+        response = requests.post(
+            url=f"{base_url}/apps/{app_name}/users/{user_id}/sessions/{create_session_request.session_id}",
+            headers={"Authorization": f"Bearer {api_key}"},
+        )
+  
+        print(f"[create session] Response from server: {response.json()}")
+    
+        return create_session_request.session_id
+
+    # Step 2: run agent with SSE
+    run_agent_request = RunAgentRequest(
+        app_name=app_name,
+        user_id=user_id,
+        session_id=create_session(),
+        new_message=Content(parts=[Part(text="商品购买")], role="user"),
+        stream=True,
+    )
+
+    print("[run agent] Event from server:")
+
+    # 3. Handle streaming events
+    async def send_request(message: str):
+        run_agent_request = RunAgentRequest(
+        app_name=app_name,
+        user_id=user_id,
+        session_id=create_session(),
+        new_message=Content(parts=[Part(text=message)], role="user"),
+        stream=True,
+        )
+
+        with httpx.stream("POST", f"{base_url}/run_sse", json=run_agent_request.model_dump(exclude_none=True), timeout=120, headers={"Authorization": f"Bearer {api_key}"}) as r:
+            for line in r.iter_lines():
+                print(line)
+                
+    async def send_request_parallel():
+        tasks = [send_request("找到 vivobook 商品并下单购买") for _ in range(task_num)]
+        await asyncio.gather(*tasks)
+    asyncio.run(send_request_parallel())
+            
+

01-tutorials/workshop/session4/http_to_mcp/readme.md

@@ -0,0 +1,20 @@
+# shopping cart agent 使用说明
+
+## 配置文件
+```markdown
+# 在当前目录下面配置 config.yaml 文件如下：
+# api_key 为模型使用
+model:
+  agent:
+      api_key: 
+      name: doubao-seed-1-6-250615
+
+# for deployment & web search
+volcengine:
+  access_key: 
+  secret_key: 
+
+# for shopping cart mcp server，本 demo 中，如下配置必须填写
+shopping_cart_mcp_api_key: 
+shopping_cart_mcp_tools_url: 
+```

01-tutorials/workshop/session4/http_to_mcp/shopping_cart_agent/__init__.py

@@ -0,0 +1 @@
+from . import agent  # noqa

01-tutorials/workshop/session4/http_to_mcp/shopping_cart_agent/agent.py

@@ -0,0 +1,45 @@
+import os
+
+from google.adk.planners import BuiltInPlanner
+from google.adk.planners import PlanReActPlanner
+from google.adk.tools.agent_tool import AgentTool
+from google.adk.tools.mcp_tool.mcp_toolset import MCPToolset, StreamableHTTPConnectionParams
+from google.genai import types
+from veadk import Agent, Runner
+from veadk.a2a.remote_ve_agent import RemoteVeAgent
+from veadk.config import getenv
+from veadk.knowledgebase.knowledgebase import KnowledgeBase
+from veadk.memory.long_term_memory import LongTermMemory
+from veadk.memory.short_term_memory import ShortTermMemory
+from veadk.tools.demo_tools import get_city_weather
+from veadk.utils.mcp_utils import get_mcp_params
+
+short_term_memory = ShortTermMemory(backend="local")
+
+###Shopping cart MCP Server
+shopping_mcp_tools_url = getenv("SHOPPING_CART_MCP_TOOLS_URL")
+shopping_api_key = getenv("SHOPPING_CART_MCP_API_KEY")
+mcp_shopping_cart = MCPToolset(
+        connection_params=StreamableHTTPConnectionParams(
+        url=shopping_mcp_tools_url,
+        headers={"Authorization": f"Bearer {shopping_api_key}"},
+    ),
+)
+
+agent_model_name=getenv("MODEL_AGENT_NAME")
+
+agent: Agent = Agent(
+    name="shopping_cart_advanced",
+    model_name=agent_model_name,
+    instruction="You are an shopping cart agent.",
+    #planner=PlanReActPlanner(),
+    planner=BuiltInPlanner(
+        thinking_config=types.ThinkingConfig(
+            include_thoughts=True,
+        ),
+    ),
+    tools=[mcp_shopping_cart],
+    short_term_memory=short_term_memory,
+)
+
+root_agent = agent

01-tutorials/workshop/session4/tool_search/amap_all_tool_agent/__init__.py

@@ -0,0 +1 @@
+from . import agent  # noqa

01-tutorials/workshop/session4/tool_search/amap_all_tool_agent/agent.py

@@ -0,0 +1,53 @@
+import os
+
+from google.adk.planners import BuiltInPlanner
+from google.adk.planners import PlanReActPlanner
+from google.adk.tools.agent_tool import AgentTool
+from google.adk.tools.mcp_tool.mcp_toolset import MCPToolset, StreamableHTTPConnectionParams
+from google.genai import types
+from veadk import Agent, Runner
+from veadk.a2a.remote_ve_agent import RemoteVeAgent
+from veadk.config import getenv
+from veadk.knowledgebase.knowledgebase import KnowledgeBase
+from veadk.memory.long_term_memory import LongTermMemory
+from veadk.memory.short_term_memory import ShortTermMemory
+from veadk.tools.demo_tools import get_city_weather
+from veadk.utils.mcp_utils import get_mcp_params
+
+short_term_memory = ShortTermMemory(backend="local")
+
+amap_mcp_tool_api_key = getenv("AMAP_MCP_TOOL_API_KEY")
+amap_mcp_tool_url = getenv("AMAP_MCP_TOOL_URL")
+amap_mcp_tool = MCPToolset(
+        connection_params=StreamableHTTPConnectionParams(
+        url=amap_mcp_tool_url,
+        headers={"Authorization": f"Bearer {amap_mcp_tool_api_key}"},
+    ),
+)
+
+github_tool_url = getenv("GITHUB_TOOL_URL")
+github_tool_api_key = getenv("GITHUB_TOOL_API_KEY")
+github_mcp_tool = MCPToolset(
+        connection_params=StreamableHTTPConnectionParams(
+        url=github_tool_url,
+        headers={"Authorization": f"Bearer {github_tool_api_key}"},
+    ),
+)
+
+agent_model_name=getenv("MODEL_AGENT_NAME")
+
+agent: Agent = Agent(
+    name="amap_tool_agent",
+    model_name=agent_model_name,
+    instruction="You are an map agent.",
+    #planner=PlanReActPlanner(),
+    planner=BuiltInPlanner(
+        thinking_config=types.ThinkingConfig(
+            include_thoughts=True,
+        ),
+    ),
+    tools=[amap_mcp_tool, github_mcp_tool],
+    short_term_memory=short_term_memory,
+)
+
+root_agent = agent

01-tutorials/workshop/session4/tool_search/amap_tool_search_agent/__init__.py

@@ -0,0 +1 @@
+from . import agent  # noqa

01-tutorials/workshop/session4/tool_search/amap_tool_search_agent/agent.py

@@ -0,0 +1,45 @@
+import os
+
+from google.adk.planners import BuiltInPlanner
+from google.adk.planners import PlanReActPlanner
+from google.adk.tools.agent_tool import AgentTool
+from google.adk.tools.mcp_tool.mcp_toolset import MCPToolset, StreamableHTTPConnectionParams
+from google.genai import types
+from veadk import Agent, Runner
+from veadk.a2a.remote_ve_agent import RemoteVeAgent
+from veadk.config import getenv
+from veadk.knowledgebase.knowledgebase import KnowledgeBase
+from veadk.memory.long_term_memory import LongTermMemory
+from veadk.memory.short_term_memory import ShortTermMemory
+from veadk.tools.demo_tools import get_city_weather
+from veadk.utils.mcp_utils import get_mcp_params
+
+short_term_memory = ShortTermMemory(backend="local")
+
+#amap_mcp_api_key
+amap_mcp_tool_set_api_key = getenv("AMAP_MCP_TOOL_SET_API_KEY")
+amap_mcp_tool_set_url = getenv("AMAP_MCP_TOOL_SET_URL")
+amap_mcp_tool_set = MCPToolset(
+        connection_params=StreamableHTTPConnectionParams(
+        url=amap_mcp_tool_set_url,
+        headers={"Authorization": f"Bearer {amap_mcp_tool_set_api_key}"},
+    ),
+)
+
+agent_model_name=getenv("MODEL_AGENT_NAME")
+
+agent: Agent = Agent(
+    name="amap_tool_set_agent",
+    model_name=agent_model_name,
+    instruction="You are an map agent.",
+    #planner=PlanReActPlanner(),
+    planner=BuiltInPlanner(
+        thinking_config=types.ThinkingConfig(
+            include_thoughts=True,
+        ),
+    ),
+    tools=[amap_mcp_tool_set],
+    short_term_memory=short_term_memory,
+)
+
+root_agent = agent

01-tutorials/workshop/session4/tool_search/readme.md

@@ -0,0 +1,27 @@
+# tool search agent 使用说明
+
+## 配置文件
+```markdown
+# 在当前目录下面配置 config.yaml 文件如下：
+# api_key 为模型使用
+model:
+  agent:
+      api_key: 
+      name: doubao-seed-1-6-250615
+
+# for deployment & web search
+volcengine:
+  access_key: 
+  secret_key: 
+
+# for amap mcp tool server
+amap_mcp_tool_api_key: 
+amap_mcp_tool_url: 
+
+# for github mcp tool
+github_tool_url: 
+
+# for amap mcp tool set server
+amap_mcp_tool_set_api_key: 
+amap_mcp_tool_set_url: 
+```