run command pre-commit to format the files

wangyue.demon · wangyue.demon · commit 9e92ad0967d8 · 2025-12-23T09:58:20.000+08:00
diff --git a/01-tutorials/workshop/session4/http_to_mcp/client.py b/01-tutorials/workshop/session4/http_to_mcp/client.py
@@ -1,7 +1,6 @@
 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
@@ -12,26 +11,25 @@
     # Step 0: setup running configs
     app_name = "travel_planner_advanced"
     user_id = "agentkit_user"
-    session_id = f"agentkit_sample_session"
+    session_id = "agentkit_sample_session"
     base_url = "http://127.0.0.1:8000"
     api_key = "test key"
 
-    task_num = 1   
-    
+    task_num = 1
 
     # Step 1: create a session
     def create_session():
         create_session_request = CreateSessionRequest(
-            session_id = session_id + f"_{random.randint(1, 9999)}",
+            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
@@ -48,20 +46,25 @@ def create_session():
     # 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,
+            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:
+        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())
-            
 
+    asyncio.run(send_request_parallel())
diff --git a/01-tutorials/workshop/session4/http_to_mcp/shopping_cart_agent/agent.py b/01-tutorials/workshop/session4/http_to_mcp/shopping_cart_agent/agent.py
@@ -1,38 +1,32 @@
-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.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 import Agent
 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(
+    connection_params=StreamableHTTPConnectionParams(
         url=shopping_mcp_tools_url,
         headers={"Authorization": f"Bearer {shopping_api_key}"},
     ),
 )
 
-agent_model_name=getenv("MODEL_AGENT_NAME")
+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=PlanReActPlanner(),
     planner=BuiltInPlanner(
         thinking_config=types.ThinkingConfig(
             include_thoughts=True,
@@ -42,4 +36,4 @@
     short_term_memory=short_term_memory,
 )
 
-root_agent = agent
+root_agent = agent
diff --git a/01-tutorials/workshop/session4/tool_search/amap_all_tool_agent/agent.py b/01-tutorials/workshop/session4/tool_search/amap_all_tool_agent/agent.py
@@ -1,25 +1,19 @@
-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.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 import Agent
 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(
+    connection_params=StreamableHTTPConnectionParams(
         url=amap_mcp_tool_url,
         headers={"Authorization": f"Bearer {amap_mcp_tool_api_key}"},
     ),
@@ -28,19 +22,19 @@
 github_tool_url = getenv("GITHUB_TOOL_URL")
 github_tool_api_key = getenv("GITHUB_TOOL_API_KEY")
 github_mcp_tool = MCPToolset(
-        connection_params=StreamableHTTPConnectionParams(
+    connection_params=StreamableHTTPConnectionParams(
         url=github_tool_url,
         headers={"Authorization": f"Bearer {github_tool_api_key}"},
     ),
 )
 
-agent_model_name=getenv("MODEL_AGENT_NAME")
+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=PlanReActPlanner(),
     planner=BuiltInPlanner(
         thinking_config=types.ThinkingConfig(
             include_thoughts=True,
diff --git a/01-tutorials/workshop/session4/tool_search/amap_tool_search_agent/agent.py b/01-tutorials/workshop/session4/tool_search/amap_tool_search_agent/agent.py
@@ -1,38 +1,32 @@
-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.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 import Agent
 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_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(
+    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_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=PlanReActPlanner(),
     planner=BuiltInPlanner(
         thinking_config=types.ThinkingConfig(
             include_thoughts=True,
diff --git a/02-use-cases/beginner/realtime_voice/agent.py b/02-use-cases/beginner/realtime_voice/agent.py
diff --git a/02-use-cases/beginner/realtime_voice/core_utils.py b/02-use-cases/beginner/realtime_voice/core_utils.py
