NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-12-18 19:32:21 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

update bigquery agent and oauth calendar agent according to new toolset interface

Browse Source 
PiperOrigin-RevId: 757949732
This commit is contained in:
Xiang (Sean) Zhou
2025-05-12 16:05:45 -07:00
committed by Copybara-Service
parent 6a04ff84ba
commit d19927b1bc
2 changed files with 18 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
contributing/samples/oauth_calendar_agent/agent.py
View File

@@ -45,12 +45,12 @@ SCOPES = ["https://www.googleapis.com/auth/calendar"]
calendar_tool_set.configure_auth(
client_id=oauth_client_id, client_secret=oauth_client_secret
)
get_calendar_events = calendar_tool_set.get_tool("calendar_events_get")
# list_calendar_events = calendar_tool_set.get_tool("calendar_events_list")
# you can replace below customized list_calendar_events tool with above ADK
# build-in google calendar tool which is commented for now to acheive same
# effect.
calendar_tool_set.set_tool_filter(
# you can also replace below customized `list_calendar_events` with build-in
# google calendar tool by adding `calendar_events_list` in the filter list
lambda tool, ctx=None: tool.name
in ["calendar_events_get"]
)
def list_calendar_events(
@@ -210,6 +210,6 @@ root_agent = Agent(
Currnet time: {_time}
""",
tools=[list_calendar_events, get_calendar_events],
tools=[list_calendar_events, calendar_tool_set],
before_agent_callback=update_time,
)