Skip to content

Commit 6afefd8

Browse files
committed
chores: format fix
1 parent a5274f0 commit 6afefd8

File tree

3 files changed

+28
-48
lines changed
  • 01-tutorials/workshop/session4

3 files changed

+28
-48
lines changed
Lines changed: 10 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -1,38 +1,32 @@
1-
import os
2-
31
from google.adk.planners import BuiltInPlanner
4-
from google.adk.planners import PlanReActPlanner
5-
from google.adk.tools.agent_tool import AgentTool
6-
from google.adk.tools.mcp_tool.mcp_toolset import MCPToolset, StreamableHTTPConnectionParams
2+
from google.adk.tools.mcp_tool.mcp_toolset import (
3+
MCPToolset,
4+
StreamableHTTPConnectionParams,
5+
)
76
from google.genai import types
8-
from veadk import Agent, Runner
9-
from veadk.a2a.remote_ve_agent import RemoteVeAgent
7+
from veadk import Agent
108
from veadk.config import getenv
11-
from veadk.knowledgebase.knowledgebase import KnowledgeBase
12-
from veadk.memory.long_term_memory import LongTermMemory
139
from veadk.memory.short_term_memory import ShortTermMemory
14-
from veadk.tools.demo_tools import get_city_weather
15-
from veadk.utils.mcp_utils import get_mcp_params
1610

1711
short_term_memory = ShortTermMemory(backend="local")
1812

19-
###Shopping cart MCP Server
13+
# Shopping cart MCP Server
2014
shopping_mcp_tools_url = getenv("SHOPPING_CART_MCP_TOOLS_URL")
2115
shopping_api_key = getenv("SHOPPING_CART_MCP_API_KEY")
2216
mcp_shopping_cart = MCPToolset(
23-
connection_params=StreamableHTTPConnectionParams(
17+
connection_params=StreamableHTTPConnectionParams(
2418
url=shopping_mcp_tools_url,
2519
headers={"Authorization": f"Bearer {shopping_api_key}"},
2620
),
2721
)
2822

29-
agent_model_name=getenv("MODEL_AGENT_NAME")
23+
agent_model_name = getenv("MODEL_AGENT_NAME")
3024

3125
agent: Agent = Agent(
3226
name="shopping_cart_advanced",
3327
model_name=agent_model_name,
3428
instruction="You are an shopping cart agent.",
35-
#planner=PlanReActPlanner(),
29+
# planner=PlanReActPlanner(),
3630
planner=BuiltInPlanner(
3731
thinking_config=types.ThinkingConfig(
3832
include_thoughts=True,
@@ -42,4 +36,4 @@
4236
short_term_memory=short_term_memory,
4337
)
4438

45-
root_agent = agent
39+
root_agent = agent

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

Lines changed: 9 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -1,25 +1,19 @@
1-
import os
2-
31
from google.adk.planners import BuiltInPlanner
4-
from google.adk.planners import PlanReActPlanner
5-
from google.adk.tools.agent_tool import AgentTool
6-
from google.adk.tools.mcp_tool.mcp_toolset import MCPToolset, StreamableHTTPConnectionParams
2+
from google.adk.tools.mcp_tool.mcp_toolset import (
3+
MCPToolset,
4+
StreamableHTTPConnectionParams,
5+
)
76
from google.genai import types
8-
from veadk import Agent, Runner
9-
from veadk.a2a.remote_ve_agent import RemoteVeAgent
7+
from veadk import Agent
108
from veadk.config import getenv
11-
from veadk.knowledgebase.knowledgebase import KnowledgeBase
12-
from veadk.memory.long_term_memory import LongTermMemory
139
from veadk.memory.short_term_memory import ShortTermMemory
14-
from veadk.tools.demo_tools import get_city_weather
15-
from veadk.utils.mcp_utils import get_mcp_params
1610

1711
short_term_memory = ShortTermMemory(backend="local")
1812

1913
amap_mcp_tool_api_key = getenv("AMAP_MCP_TOOL_API_KEY")
2014
amap_mcp_tool_url = getenv("AMAP_MCP_TOOL_URL")
2115
amap_mcp_tool = MCPToolset(
22-
connection_params=StreamableHTTPConnectionParams(
16+
connection_params=StreamableHTTPConnectionParams(
2317
url=amap_mcp_tool_url,
2418
headers={"Authorization": f"Bearer {amap_mcp_tool_api_key}"},
2519
),
@@ -28,19 +22,18 @@
2822
github_tool_url = getenv("GITHUB_TOOL_URL")
2923
github_tool_api_key = getenv("GITHUB_TOOL_API_KEY")
3024
github_mcp_tool = MCPToolset(
31-
connection_params=StreamableHTTPConnectionParams(
25+
connection_params=StreamableHTTPConnectionParams(
3226
url=github_tool_url,
3327
headers={"Authorization": f"Bearer {github_tool_api_key}"},
3428
),
3529
)
3630

37-
agent_model_name=getenv("MODEL_AGENT_NAME")
38-
31+
agent_model_name = getenv("MODEL_AGENT_NAME")
3932
agent: Agent = Agent(
4033
name="amap_tool_agent",
4134
model_name=agent_model_name,
4235
instruction="You are an map agent.",
43-
#planner=PlanReActPlanner(),
36+
# planner=PlanReActPlanner(),
4437
planner=BuiltInPlanner(
4538
thinking_config=types.ThinkingConfig(
4639
include_thoughts=True,

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

Lines changed: 9 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -1,38 +1,31 @@
1-
import os
2-
31
from google.adk.planners import BuiltInPlanner
4-
from google.adk.planners import PlanReActPlanner
5-
from google.adk.tools.agent_tool import AgentTool
6-
from google.adk.tools.mcp_tool.mcp_toolset import MCPToolset, StreamableHTTPConnectionParams
2+
from google.adk.tools.mcp_tool.mcp_toolset import (
3+
MCPToolset,
4+
StreamableHTTPConnectionParams,
5+
)
76
from google.genai import types
8-
from veadk import Agent, Runner
9-
from veadk.a2a.remote_ve_agent import RemoteVeAgent
7+
from veadk import Agent
108
from veadk.config import getenv
11-
from veadk.knowledgebase.knowledgebase import KnowledgeBase
12-
from veadk.memory.long_term_memory import LongTermMemory
139
from veadk.memory.short_term_memory import ShortTermMemory
14-
from veadk.tools.demo_tools import get_city_weather
15-
from veadk.utils.mcp_utils import get_mcp_params
1610

1711
short_term_memory = ShortTermMemory(backend="local")
1812

19-
#amap_mcp_api_key
13+
# amap_mcp_api_key
2014
amap_mcp_tool_set_api_key = getenv("AMAP_MCP_TOOL_SET_API_KEY")
2115
amap_mcp_tool_set_url = getenv("AMAP_MCP_TOOL_SET_URL")
2216
amap_mcp_tool_set = MCPToolset(
23-
connection_params=StreamableHTTPConnectionParams(
17+
connection_params=StreamableHTTPConnectionParams(
2418
url=amap_mcp_tool_set_url,
2519
headers={"Authorization": f"Bearer {amap_mcp_tool_set_api_key}"},
2620
),
2721
)
2822

29-
agent_model_name=getenv("MODEL_AGENT_NAME")
30-
23+
agent_model_name = getenv("MODEL_AGENT_NAME")
3124
agent: Agent = Agent(
3225
name="amap_tool_set_agent",
3326
model_name=agent_model_name,
3427
instruction="You are an map agent.",
35-
#planner=PlanReActPlanner(),
28+
# planner=PlanReActPlanner(),
3629
planner=BuiltInPlanner(
3730
thinking_config=types.ThinkingConfig(
3831
include_thoughts=True,

0 commit comments

Comments
 (0)