Commit Graph
74 Commits
Author SHA1 Message Date
Xiang (Sean) ZhouandCopybara-Service b92e068155 chore: fix function tool UT
PiperOrigin-RevId: 764567867
2025-05-28 22:30:42 -07:00
Xiang (Sean) ZhouandCopybara-Service a0714b8afc ADK changes
PiperOrigin-RevId: 764564638
2025-05-28 22:19:24 -07:00
Xiang (Sean) ZhouandCopybara-Service b70e74c450 chore: auto format mcp tools
PiperOrigin-RevId: 764547299
2025-05-28 21:14:42 -07:00
Xiang (Sean) ZhouandCopybara-Service 791ac9eb68 refactor: refactor cli to use agent loader
PiperOrigin-RevId: 764546019
2025-05-28 21:09:40 -07:00
Xiang (Sean) ZhouandCopybara-Service 623957c0a8 refactor: refine agent_loader to load the most common folder structure first and don't including unnecessary case
PiperOrigin-RevId: 764545194
2025-05-28 21:06:07 -07:00
Xiang (Sean) ZhouandCopybara-Service c7ce987676 chore: reformat load_memory_tool
PiperOrigin-RevId: 764402270
2025-05-28 13:52:40 -07:00
Xiang (Sean) ZhouandCopybara-Service d40df2edf2 chore: fix unit tests
PiperOrigin-RevId: 764107186
2025-05-27 23:07:29 -07:00
Xiang (Sean) ZhouandCopybara-Service 618c824994 feat: extract agent loading logic from fast_api.py to a separate AgentLoader class and support more agent definition folder/file structure.
Structures supported:
a) agents_dir/agent_name.py (with root_agent or agent.root_agent in it)
b) agents_dir/agent_name_folder/__init__.py (with root_agent or agent.root_agent in the package)
c) agents_dir/agent_name_folder/agent.py (where agent.py has root_agent)

PiperOrigin-RevId: 763943716
2025-05-27 14:14:51 -07:00
Xiang (Sean) ZhouandCopybara-Service 7c2df7e4f6 refactor: refactor evaluation to make cli module depends on evaluation module. Modules outside of cli module should not reference cli module
PiperOrigin-RevId: 763577749
2025-05-26 19:18:55 -07:00
Xiang (Sean) ZhouandCopybara-Service be0786ea88 refactor: rename agent_dir to agents_dir and rename app_id to app_name in fast_api.py to make it consistent among every endpoints
PiperOrigin-RevId: 763483339
2025-05-26 11:51:19 -07:00
Xiang (Sean) ZhouandCopybara-Service 694eca08e5 fix: fix bigquery credentials and bigquery tool to make it compatible with python 3.9 and make the credential serializable in session
PiperOrigin-RevId: 763332829
2025-05-26 01:58:52 -07:00
Xiang (Sean) ZhouandCopybara-Service 74a331f1fb chore: reformat the code with autoformat.sh
PiperOrigin-RevId: 763256626
2025-05-25 20:33:45 -07:00
Xiang (Sean) ZhouandCopybara-Service 98c8a71d67 refactor: rename parameter name and add type annotation of _get_agent_to_run method
PiperOrigin-RevId: 763234780
2025-05-25 18:56:16 -07:00
Xiang (Sean) ZhouandCopybara-Service b0fb3536e5 chore: cleanup unused import in cli module
PiperOrigin-RevId: 763233551
2025-05-25 18:50:31 -07:00
Xiang (Sean) ZhouandCopybara-Service a380d70ac7 chore: fix cli UT
PiperOrigin-RevId: 762734348
2025-05-24 00:11:32 -07:00
Xiang (Sean) ZhouandCopybara-Service a134ba6f7a feat: add toolset cleanup logic to run_cli
PiperOrigin-RevId: 762638389
2025-05-23 18:06:30 -07:00
Xiang (Sean) ZhouandCopybara-Service 756a326033 feat: add customized bigquer tool wrapper class to facilitate developer to handcraft bigquery api tool
PiperOrigin-RevId: 762626700
2025-05-23 17:23:41 -07:00
Xiang (Sean) ZhouandCopybara-Service 92c37496d3 refactor: simplify toolset cleanup codes and extract common cleanup codes to utils which could be utilized by cli or client codes that directly call runners
PiperOrigin-RevId: 762463028
2025-05-23 09:49:19 -07:00
Xiang (Sean) ZhouandCopybara-Service ff8a3c9b43 chore: reformat the codes using autoformat.sh
PiperOrigin-RevId: 762004002
2025-05-22 09:43:54 -07:00
Xiang (Sean) ZhouandCopybara-Service a2263b1808 refactor: refactor toolset to extract tool_filter logic to base class
PiperOrigin-RevId: 761828251
2025-05-21 23:54:33 -07:00
Xiang (Sean) ZhouandCopybara-Service 09cb128cf9 fix: fix function tool name parsing
to fix https://github.com/google/adk-python/issues/824

PiperOrigin-RevId: 761703269
2025-05-21 16:00:09 -07:00
Xiang (Sean) ZhouandCopybara-Service b2a2b11776 ADK changes
PiperOrigin-RevId: 761650284
2025-05-21 13:39:16 -07:00
Xiang (Sean) ZhouandCopybara-Service cbdb5fc507 chore: fix ut for fast api server
PiperOrigin-RevId: 761350248
2025-05-20 21:17:39 -07:00
Xiang (Sean) ZhouandCopybara-Service 76700d790b chore: add doc string for auto generated google api toolset
PiperOrigin-RevId: 761266870
2025-05-20 16:08:27 -07:00
Xiang (Sean) ZhouandCopybara-Service f67ccf32c3 fix: support Callable that has __call__ as coroutine function in FunctionTool
PiperOrigin-RevId: 760913537
2025-05-19 22:09:44 -07:00
Xiang (Sean) ZhouandCopybara-Service de7c9c6509 enhance toolbox toolset:
1. use async client
2. expose add headers and bind parameters functionality of toolbox tool

PiperOrigin-RevId: 760845486
2025-05-19 18:08:02 -07:00
Xiang (Sean) ZhouandCopybara-Service 9eef4464c6 fix: fix toolbox import path in sample toolbox agent
PiperOrigin-RevId: 760837908
2025-05-19 17:43:41 -07:00
Xiang (Sean) ZhouandCopybara-Service 74454170a3 refactor: refactor and refine LangChainTool
PiperOrigin-RevId: 760726719
2025-05-19 12:29:44 -07:00
Xiang (Sean) ZhouandCopybara-Service 3f117391a5 refactor: remote remote agent as there is no use case and it's not implemented properly
PiperOrigin-RevId: 760652423
2025-05-19 09:24:37 -07:00
Xiang (Sean) ZhouandCopybara-Service 3b5232c14f feat: add sample mcp agent that connects to mcp server via sse endpoint directly
PiperOrigin-RevId: 760388717
2025-05-18 17:00:03 -07:00
Xiang (Sean) ZhouandCopybara-Service 2ad1f79422 refactor: replace logger.warn which is deprecated with logger.warning
PiperOrigin-RevId: 760311574
2025-05-18 10:25:29 -07:00
Xiang (Sean) ZhouandCopybara-Service f83504dffb chore: add toolbox sample agent
PiperOrigin-RevId: 759746416
2025-05-16 14:01:53 -07:00
Xiang (Sean) ZhouandCopybara-Service 939b45a1c8 feat: add Toolbox Toolset
PiperOrigin-RevId: 759741922
2025-05-16 13:48:47 -07:00
Xiang (Sean) ZhouandCopybara-Service 015c667917 refactor: refactor base toolset to import from relative path
PiperOrigin-RevId: 759735824
2025-05-16 13:33:04 -07:00
Xiang (Sean) ZhouandCopybara-Service f384793649 fix: fix mcp agent filter logic
PiperOrigin-RevId: 759638254
2025-05-16 09:10:49 -07:00
Xiang (Sean) ZhouandCopybara-Service 05a853bc91 fix:fix mcp toolset close issue
PiperOrigin-RevId: 759636772
2025-05-16 09:05:56 -07:00
Xiang (Sean) ZhouandCopybara-Service 12507dc6cc fix: fix tool filtering in mcp toolset
PiperOrigin-RevId: 759632573
2025-05-16 08:53:21 -07:00
Xiang (Sean) ZhouandCopybara-Service 755ce74550 fix: stage correct ui files to adk
PiperOrigin-RevId: 759632290
2025-05-16 08:52:22 -07:00
Xiang (Sean) ZhouandCopybara-Service eb076a9ee8 fix: fix application integration toolset
PiperOrigin-RevId: 759450502
2025-05-15 22:23:49 -07:00
Xiang (Sean) ZhouandCopybara-Service fe5a5494e9 fix: fix oauth for dev UI
PiperOrigin-RevId: 759433563
2025-05-15 21:24:28 -07:00
Xiang (Sean) ZhouandCopybara-Service f0e11c72ab refactor: make readonly context optional in tool predicate
PiperOrigin-RevId: 759429453
2025-05-15 21:07:00 -07:00
Xiang (Sean) ZhouandCopybara-Service 770ecd8bb3 refactor: hide non-public field and cleanup for tools
PiperOrigin-RevId: 759400805
2025-05-15 19:24:10 -07:00
Xiang (Sean) ZhouandCopybara-Service cb000e5a94 refactor: remove async root_agent handling as it no longer needed for MCP toolset
PiperOrigin-RevId: 759364382
2025-05-15 17:12:57 -07:00
Xiang (Sean) ZhouandCopybara-Service fccd17df6f fix: adapt oauth calendar agent and bigquery agent to new Google API toolset interface
PiperOrigin-RevId: 759298612
2025-05-15 14:21:45 -07:00
Xiang (Sean) ZhouandCopybara-Service bdd678db31 refactor: refactor google api toolset to expose class instead of instance
PiperOrigin-RevId: 759289358
2025-05-15 13:59:57 -07:00
Xiang (Sean) ZhouandCopybara-Service 2214c1558f fix: fix the typo in MCPToolset when parsing the errlog argument to MCPSessionManager
PiperOrigin-RevId: 759018462
2025-05-15 00:31:18 -07:00
Xiang (Sean) ZhouandCopybara-Service 4a0751941b fix: fix application integration toolset
PiperOrigin-RevId: 759018346
2025-05-15 00:30:19 -07:00
Xiang (Sean) ZhouandCopybara-Service fc40226ec0 refactor: refactor mcp toolset to hide non-public field
PiperOrigin-RevId: 758494601
2025-05-13 21:06:03 -07:00
Xiang (Sean) ZhouandCopybara-Service 14cf910ce6 refactor: refactor application integration toolset to hide non-public field
PiperOrigin-RevId: 758469938
2025-05-13 19:28:01 -07:00
Xiang (Sean) ZhouandCopybara-Service 00e0035c8a refactor: refactor google api toolset to hide non-public field
PiperOrigin-RevId: 758469484
2025-05-13 19:26:05 -07:00
Xiang (Sean) ZhouandCopybara-Service da55b5a943 fix: update google api toolset to remove 'match' for python 3.9 compliant
PiperOrigin-RevId: 758438091
2025-05-13 17:21:05 -07:00
Xiang (Sean) ZhouandCopybara-Service 9647426500 refactor: refactor openapi toolset and tool parser to hide non public field
PiperOrigin-RevId: 758436303
2025-05-13 17:14:20 -07:00
Xiang (Sean) ZhouandCopybara-Service 931fb338f8 rename tool_set to toolset to uniform the naming
PiperOrigin-RevId: 758422703
2025-05-13 16:34:23 -07:00
Xiang (Sean) ZhouandCopybara-Service af978fe194 fix apihub toolset ut
PiperOrigin-RevId: 758400203
2025-05-13 15:32:28 -07:00
Xiang (Sean) ZhouandCopybara-Service f6bdd555fb refactor apihub toolset to hide non public field
PiperOrigin-RevId: 758378396
2025-05-13 14:34:04 -07:00
Xiang (Sean) ZhouandCopybara-Service dd8e9b71e5 fix application integration toolset unit tests
PiperOrigin-RevId: 758080966
2025-05-13 00:06:56 -07:00
Xiang (Sean) ZhouandCopybara-Service 2d84b13219 fix toolset codes to make it work with python 3.9
PiperOrigin-RevId: 757991503
2025-05-12 18:23:18 -07:00
Xiang (Sean) ZhouandCopybara-Service d35b99e6dd update mcp toolset and sample agent based on new tool_filter definition
PiperOrigin-RevId: 757969950
2025-05-12 17:05:24 -07:00
Xiang (Sean) ZhouandCopybara-Service 722028801a update application integration toolset agent based on new toolset interface
PiperOrigin-RevId: 757963384
2025-05-12 16:43:41 -07:00
Xiang (Sean) ZhouandCopybara-Service 57d1315533 adapt application integration toolset to new toolset interface
PiperOrigin-RevId: 757960706
2025-05-12 16:36:43 -07:00
Xiang (Sean) ZhouandCopybara-Service d19927b1bc update bigquery agent and oauth calendar agent according to new toolset interface
PiperOrigin-RevId: 757949732
2025-05-12 16:06:45 -07:00
Xiang (Sean) ZhouandCopybara-Service 6a04ff84ba adapt google api toolset and api hub toolset to new toolset interface
PiperOrigin-RevId: 757946541
2025-05-12 15:58:39 -07:00
Xiang (Sean) ZhouandCopybara-Service 5462862795 fix eval unittests failure
PiperOrigin-RevId: 757872670
2025-05-12 12:42:19 -07:00
Xiang (Sean) ZhouandCopybara-Service 166df01236 fix eval in api server and cli for below issue:
https://github.com/google/adk-python/issues/651

PiperOrigin-RevId: 756906937
2025-05-09 14:26:29 -07:00
Xiang (Sean) ZhouandCopybara-Service dc5c23c675 fix predicate class name in MCP toolset
PiperOrigin-RevId: 756761193
2025-05-09 07:39:08 -07:00
Xiang (Sean) ZhouandCopybara-Service 8c78945ec9 update mcp agent to use latest toolset interface
PiperOrigin-RevId: 756613972
2025-05-08 22:57:49 -07:00
Xiang (Sean) ZhouandCopybara-Service 7dffb59096 adapt MCP toolset to the latest Toolset interface
PiperOrigin-RevId: 756611140
2025-05-08 22:47:51 -07:00
Xiang (Sean) ZhouandCopybara-Service 4d7298e4f2 add toolset base class and allow llm agent to accept toolset as tools
PiperOrigin-RevId: 756605470
2025-05-08 22:27:52 -07:00
Xiang (Sean) ZhouandCopybara-Service e7d9cf359a make eval functions async
PiperOrigin-RevId: 756106627
2025-05-07 19:53:19 -07:00
Xiang (Sean) ZhouandCopybara-Service 5846b24b71 set param required tag to False by default
PiperOrigin-RevId: 754382785
2025-05-03 10:31:38 -07:00
Xiang (Sean) ZhouandCopybara-Service d387ab0c38 don't send content with empty text to LLM
PiperOrigin-RevId: 753869699
2025-05-01 23:01:50 -07:00
Xiang (Sean) ZhouandCopybara-Service dbbeb190b0 Add two adk run options:
--replay : a json file that contains the initial session state and user queries, adk run will create a new session based on the state and run the user queries against the session. Users cannot continue to interact with agent.

--resume : a json file that contains the previously saved session (by --save_session option), adk run will replay this session and then user can continue to interact with the agent.
PiperOrigin-RevId: 752923403
2025-04-29 16:23:10 -07:00
Xiang (Sean) ZhouandCopybara-Service 2ea4315e9f update the doc comments of requested_auth_configs to make it more precise.
PiperOrigin-RevId: 752861847
2025-04-29 13:33:37 -07:00
Xiang (Sean) ZhouandCopybara-Service 956fb912e8 feat(auth)!: expose access_token and refresh_token at top level of auth credentails
BREAKING CHANGE: `token` attribute of OAuth2Auth credentials used to be a dict containing both access_token and refresh_token, given that may cause confusions, now we replace it with access_token and refresh_token at top level of the auth credentials

PiperOrigin-RevId: 750346172
2025-04-22 15:23:21 -07:00