encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-07-09 18:19:28 -07:00
Code Issues Packages Projects Releases Wiki Activity
1,636 Commits 12 Branches 39 Tags
93aad611983dc1daf415d3a73105db45bbdd1988
Commit Graph
100 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
Shangjie ChenandCopybara-Service e218254495 feat: Add SqliteSessionService and a migration script to migrate existing DB using DatabaseSessionService to SqliteSessionService 
The new Sqlite version has fixed schema and use a single column to store Event data, this should avoid DB migration for future add to the Event object.

- This change introduces `SqliteSessionService`, an asynchronous session service using `aiosqlite` that stores event data as JSON within SQLite.
- A migration script, `migrate_from_sqlalchemy_sqlite.py`, is included to transition data from the older SQLAlchemy-based SQLite schema to this new format.
- The CLI service registry is updated to use SqliteSessionService for sqlite:// URIs.
- Throw error when user trying to access a legacy DB and advice the user to do the migration.

Co-authored-by: Shangjie Chen <deanchen@google.com>
PiperOrigin-RevId: 829971174
 2025-11-08 21:19:38 -08:00
Shangjie ChenandCopybara-Service f511352508 chore: Clarify the usage of agent_state 
Co-authored-by: Shangjie Chen <deanchen@google.com>
PiperOrigin-RevId: 828733706
 2025-11-05 19:30:58 -08:00
Shangjie ChenandCopybara-Service d6b928bdf7 chore: Returns agent state regardless if ctx.is_resumable 
This allows state to be passing across agents

Co-authored-by: Shangjie Chen <deanchen@google.com>
PiperOrigin-RevId: 828557989
 2025-11-05 11:31:26 -08:00
Shangjie ChenandCopybara-Service 9ab17f2afd fix: Yield the long running tool response before pausing execution 
PiperOrigin-RevId: 825056377
 2025-10-28 08:41:47 -07:00
Shangjie ChenandCopybara-Service e194ebb33c feat: Implement artifact_version related methods in GcsArtifactService 
PiperOrigin-RevId: 824646770
 2025-10-27 13:13:07 -07:00
Shangjie ChenandCopybara-Service f7e2a7a40e chore: Make VertexAiSessionService fully asynchronous 
This CL refactors VertexAiSessionService to use the asynchronous aio client for all Vertex AI API calls. This ensures that the service methods are non-blocking and can be used effectively in an asyncio environment.

PiperOrigin-RevId: 824573356
 2025-10-27 10:19:25 -07:00
Shangjie ChenandCopybara-Service 97a224fe46 chore: Clean up rag memory dependencies 
PiperOrigin-RevId: 823556599
 2025-10-24 09:19:49 -07:00
Shangjie ChenandCopybara-Service 391628fcdc feat: Add a service registry to provide a generic way to register custom service implementations to be used in FastAPI server 
To register a custom service:
- Create a factory function that takes a URI and returns an instance of your custom service. This function will parse any details it needs from the URI.
- Register your factory with the global service registry. You need to define a unique URI scheme for your service (e.g., custom).

PiperOrigin-RevId: 822310466
 2025-10-21 15:58:51 -07:00
Shangjie ChenandCopybara-Service 2a901d12f4 chore: Raise AlreadyExistsError when trying to create a resource with same ID 
Move the dedupe logic into session service so that the internal error can be surfaced to client

PiperOrigin-RevId: 822294430
 2025-10-21 15:11:33 -07:00
Shangjie ChenandCopybara-Service 6a5eac0bdc feat: Allow passing extra kwargs to create_session of VertexAiSessionService 
This can be used to set ttl and other configs.

PiperOrigin-RevId: 821782343
 2025-10-20 13:34:07 -07:00
Shangjie ChenandCopybara-Service d327538724 chore: Throw 409 already exist error when trying to create a session with existing id 
Resolves https://github.com/google/adk-python/issues/2950

PiperOrigin-RevId: 821362008
 2025-10-19 11:21:36 -07:00
Shangjie ChenandCopybara-Service 8b3ed059c2 chore: Refactor and fix state management in the session service 
Also refactoring the test cases to focus on the expected behaviors

PiperOrigin-RevId: 820734484
 2025-10-17 10:04:36 -07:00
Shangjie ChenandCopybara-Service 9dce06f9b0 feat: Add rewind_async to support rewinding the session to before a previous invocation 
PiperOrigin-RevId: 820552460
 2025-10-16 23:24:40 -07:00
Shangjie ChenandCopybara-Service 6ab1498aa0 fix: Add usage_metadata and citation_metadata to DatabaseSessionService 
PiperOrigin-RevId: 819900773
 2025-10-16 13:05:52 -07:00
Shangjie ChenandCopybara-Service a985cc38ec feat: Support return all sessions when user id is none 
PiperOrigin-RevId: 819884236
 2025-10-15 13:14:24 -07:00
Shangjie ChenandCopybara-Service 141318f775 feat: Support returning all sessions when user_id is none in the request 
resolves https://github.com/google/adk-python/issues/3154

PiperOrigin-RevId: 819401023
 2025-10-14 14:30:47 -07:00
Shangjie ChenandCopybara-Service 2c7a342593 feat: Add create_time and mime_type to ArtifactVersion 
PiperOrigin-RevId: 819372593
 2025-10-14 13:29:24 -07:00
Shangjie ChenandCopybara-Service bb1ea74924 chore: Delegate the agent state reset logic to LoopAgent 
This is so we don't need to worry about side effect of Loop in all agent type. Custom agent should do the same if there exists loop inside.

PiperOrigin-RevId: 818766305
 2025-10-13 11:53:59 -07:00
Shangjie ChenandCopybara-Service 75179243b4 chore: Update human_tool_confirmation agent to use resumability feature 
PiperOrigin-RevId: 816793131
 2025-10-08 11:35:13 -07:00
Shangjie ChenandCopybara-Service 636def3687 fix: Add AuthConfig json serialization in vertex ai session service 
Resolves https://github.com/google/adk-python/issues/2035

PiperOrigin-RevId: 816387628
 2025-10-07 14:47:28 -07:00
Shangjie ChenandCopybara-Service 90d4c19c51 feat: Migrate vertex ai session service to use agent engine sdk 
PiperOrigin-RevId: 816259798
 2025-10-07 09:58:27 -07:00
Shangjie ChenandCopybara-Service 97b950b36b feat: Migrate vertex_ai_session_service to use Agent Engine SDK 
PiperOrigin-RevId: 814948921
 2025-10-03 21:00:15 -07:00
Shangjie ChenandCopybara-Service b170a84279 chore: Handle exception in preload_memory_tool to not fail the llm request 
Resolves https://github.com/google/adk-python/issues/3069

PiperOrigin-RevId: 814391260
 2025-10-02 14:41:27 -07:00
Shangjie ChenandCopybara-Service 55bc985821 chore: Fix vertexai import rule of memory service for google internal dependencies 
PiperOrigin-RevId: 813941520
 2025-10-02 13:42:23 -07:00
Shangjie ChenandCopybara-Service 83fd045718 feat: Migrate VertexAiMemoryBankService to use Agent Engine SDK 
PiperOrigin-RevId: 813104746
 2025-09-29 23:14:50 -07:00
Shangjie ChenandCopybara-Service ce9c39f5a8 feat: Implement checkpoint and resume logic for LoopAgent 
PiperOrigin-RevId: 813096880
 2025-09-29 22:45:57 -07:00
Shangjie ChenandCopybara-Service 2f1040f296 feat: Implement checkpoint and resume logic for ParallelAgent 
PiperOrigin-RevId: 812658378
 2025-09-29 00:26:32 -07:00
Shangjie ChenandCopybara-Service 7b707cebea chore: Simplfiy the parallel agent py version handling logic 
PiperOrigin-RevId: 811992425
 2025-09-26 16:15:51 -07:00
Shangjie ChenandCopybara-Service 1ee01cc05a feat: Implement checkpoint and resume logic for SequentialAgent 
PiperOrigin-RevId: 811977004
 2025-09-26 15:26:42 -07:00
Shangjie ChenandCopybara-Service 2c752934a8 feat: Skip running a workflow agent if it has no sub-agents 
PiperOrigin-RevId: 811528166
 2025-09-25 15:39:38 -07:00
Shangjie ChenandCopybara-Service 8b081751ed feat: Add core checkpointing primitive for base agent 
PiperOrigin-RevId: 811458903
 2025-09-25 12:35:36 -07:00
Shangjie ChenandCopybara-Service 839d2e43bb feat: Define an AgentState to be used for resuming agent invocation 
PiperOrigin-RevId: 811414736
 2025-09-25 10:49:49 -07:00
Shangjie ChenandCopybara-Service 97cf724f14 chore: Release ADK 1.14.0 
PiperOrigin-RevId: 805506354
 2025-09-10 14:04:22 -07:00
Shangjie ChenandCopybara-Service 8452d2bcba feat: Allow function tool to pass in a callable to decide whether it need confirmation 
PiperOrigin-RevId: 804583437
 2025-09-08 14:16:59 -07:00
Shangjie ChenandCopybara-Service e63fe0c0eb fix: Fix pagination of list_sessions in VertexAiSessionService 
Resolves https://github.com/google/adk-python/issues/2860

PiperOrigin-RevId: 804511401
 2025-09-08 11:14:32 -07:00
Shangjie ChenandCopybara-Service cecf7e805d fix: Support saving text artifacts in GCS artifact service 
Resolves https://github.com/google/adk-python/issues/2775

PiperOrigin-RevId: 802850139
 2025-09-03 21:55:07 -07:00
Shangjie ChenandCopybara-Service fb009d8ea6 fix: Add custom_metadata to DatabaseSessionService 
Resolve https://github.com/google/adk-python/issues/2677

PiperOrigin-RevId: 802375768
 2025-09-02 20:09:46 -07:00
Shangjie ChenandCopybara-Service 219815d2d7 chore: Update create_session endpoint to use Request message as post body 
Also deprecate create_session_with_id which is not very RESTful.

PiperOrigin-RevId: 802357718
 2025-09-02 18:56:45 -07:00
Shangjie ChenandCopybara-Service 0c87907bcb fix: Enforce foreign key constraint for SQLite DB 
Resolves https://github.com/google/adk-python/issues/2752

PiperOrigin-RevId: 801106660
 2025-08-29 19:07:22 -07:00
Shangjie ChenandCopybara-Service a17bcbb2aa feat: Add a tool confirmation flow that can guard tool execution with explicit confirmation and custom input 
The existing `LongRunningTool` does not define a programmatic way to provide & validate structured input, also it relies on LLM to reason and parse the user's response.

For a quick start, annotate the function with `FunctionTool(my_function, require_confirmation=True)`. A more advanced flow is shown in the `human_tool_confirmation` sample.

The new flow is similar to the existing Auth flow:
- User request a tool confirmation by calling `tool_context.request_confirmation()` in the tool or `before_tool_callback`, or just using the `require_confirmation` shortcut in FunctionTool.
- User can provide custom validation logic before tool call proceeds.
- ADK creates corresponding RequestConfirmation FunctionCall Event to ask user for confirmation
- User needs to provide the expected tool confirmation to a RequestConfirmation FunctionResponse Event.
- ADK then checks the response and continues the tool call.

PiperOrigin-RevId: 801019917
 2025-08-29 13:56:54 -07:00
Shangjie ChenandCopybara-Service f03f167779 chore: Return explict None type for DELETE endpoints 
The HTTP response code will become 204 No Content instead of 200.

PiperOrigin-RevId: 796123628
 2025-08-17 09:58:40 -07:00
Shangjie ChenandCopybara-Service ecaa7b4c98 chore: Rename run related method and request to align with the conventions 
PiperOrigin-RevId: 795660109
 2025-08-15 16:48:01 -07:00
Shangjie ChenandCopybara-Service a2b7909fc3 fix: Fix the packaging version comparison logic in adk cli 
Resolves https://github.com/google/adk-python/issues/2503

PiperOrigin-RevId: 795519711
 2025-08-15 10:16:45 -07:00
Shangjie ChenandCopybara-Service bb3735c9ca chore: Remove logging that contains full event data from DatabaseSessionService 
Resolves https://github.com/google/adk-python/issues/2153

PiperOrigin-RevId: 794288546
 2025-08-12 15:19:52 -07:00
Shangjie ChenandCopybara-Service 88114d7c73 chore: Add docstring to clarify the behavior of preload memory tool 
PiperOrigin-RevId: 793840979
 2025-08-11 16:33:49 -07:00
Shangjie ChenandCopybara-Service 7f12387eb1 chore: Make all FastAPI endpoints async 
PiperOrigin-RevId: 792886431
 2025-08-08 21:58:34 -07:00
Shangjie ChenandCopybara-Service c323de5c69 chore: Group FastAPI endpoints with tags 
PiperOrigin-RevId: 792816574
 2025-08-08 17:20:46 -07:00
Shangjie ChenandCopybara-Service 20c30d5819 chore: Bump version number to 1.10.0 
PiperOrigin-RevId: 792206610
 2025-08-07 10:05:01 -07:00
Shangjie ChenandCopybara-Service 1cfe6e9ffe chore: Remove unnecessary flags 
PiperOrigin-RevId: 789379877
 2025-07-31 10:07:39 -07:00
Shangjie ChenandCopybara-Service 314d6a4f95 fix: Return session state in list_session API endpoint 
Resolves https://github.com/google/adk-python/issues/2193
Resolves https://github.com/google/adk-python/issues/781

PiperOrigin-RevId: 789143973
 2025-07-30 19:53:46 -07:00
Shangjie ChenandCopybara-Service 7d06fb735e chore: Move create_session log to where the session is actually created 
PiperOrigin-RevId: 789094066
 2025-07-30 16:46:25 -07:00
Shangjie ChenandCopybara-Service 11037fc133 chore: Filter event with only functions, thought_signatures when adding to memory 
PiperOrigin-RevId: 786825628
 2025-07-24 13:54:37 -07:00
Shangjie ChenandCopybara-Service f46396fa98 chore: Allow httpoptions override in VertexAiSessionService 
PiperOrigin-RevId: 785886180
 2025-07-22 09:36:17 -07:00
Shangjie ChenandCopybara-Service 2e778049d0 feat: Support passing fully qualified agent engine resource name when constructing session service and memory service 
Resolves https://github.com/google/adk-python/issues/1760

PiperOrigin-RevId: 784353411
 2025-07-17 16:19:55 -07:00
Shangjie ChenandCopybara-Service eccb484985 chore: Update label name for bot triaged issues 
Also assign agent engine related issue to yeesian as poc

PiperOrigin-RevId: 783941809
 2025-07-16 16:39:25 -07:00
Shangjie ChenandCopybara-Service dc414cb507 chore: Add docstring to clarify that inmemory service are not suitable for production 
environment

Resolves https://github.com/google/adk-python/issues/1569

PiperOrigin-RevId: 780013850
 2025-07-07 02:13:56 -07:00
Shangjie ChenandCopybara-Service a021222077 feat: Add custom_metadata to VertexAiSessionService event handling 
PiperOrigin-RevId: 779255526
 2025-07-04 12:38:15 -07:00
Shangjie ChenandCopybara-Service 3fa2ea7cb9 fix: Remove duplicate options from adk deploy 
Resolves https://github.com/google/adk-python/issues/1739

PiperOrigin-RevId: 778692588
 2025-07-02 17:27:45 -07:00
Shangjie ChenandCopybara-Service df141db60c chore: Add tenacity into project.toml 
PiperOrigin-RevId: 778535340
 2025-07-02 09:31:19 -07:00
Shangjie ChenandCopybara-Service 3d2f13cecd chore: Update the retry logic of create session polling 
This should slightly increase the timeout also reduce the polling frequency.

PiperOrigin-RevId: 778323416
 2025-07-01 21:21:57 -07:00
Shangjie ChenandCopybara-Service 17d6042995 fix: Make sure each partial event has different timestamp 
PiperOrigin-RevId: 778148175
 2025-07-01 12:36:33 -07:00
Shangjie ChenandCopybara-Service 88a4402d14 chore: Do not send api request when session does not have events 
PiperOrigin-RevId: 775423356
 2025-06-24 16:15:23 -07:00
Shangjie ChenandCopybara-Service abc89d2c81 feat: Add implementation of VertexAiMemoryBankService and support in FastAPI endpoint 
PiperOrigin-RevId: 775327151
 2025-06-24 11:57:03 -07:00
Shangjie ChenandCopybara-Service f033e405c1 chore: Clarify the behavior of Event.invocation_id 
PiperOrigin-RevId: 774827874
 2025-06-23 10:12:16 -07:00
Shangjie ChenandCopybara-Service 2fd8feb65d chore: Support allow_origins in cloud_run deployment 
Also reorganize the fast_api_common_options. This resolves https://github.com/google/adk-python/issues/1444.

PiperOrigin-RevId: 773890111
 2025-06-20 16:54:08 -07:00
Shangjie ChenandCopybara-Service 8677d5c8dc chore: bump version number to 1.4.2 
PiperOrigin-RevId: 773867075
 2025-06-20 15:41:50 -07:00
Shangjie ChenandCopybara-Service 4d72d31b13 fix: Add type checking to handle different response type of genai API client 
Fixes https://github.com/google/adk-python/issues/1514

PiperOrigin-RevId: 773838035
 2025-06-20 14:12:55 -07:00
Shangjie ChenandCopybara-Service 17beb32880 chore: Bump version number and update changelog for 1.4.1 release 
PiperOrigin-RevId: 773148349
 2025-06-18 19:08:43 -07:00
Shangjie ChenandCopybara-Service fc65873d7c chore: Set agent_engine_id in the service constructor, also use the agent_engine_id field instead of overriding app_name in FastAPI endpoint 
PiperOrigin-RevId: 770427903
 2025-06-11 19:46:25 -07:00
Shangjie ChenandCopybara-Service 416dc6feed feat: add memory_service option to CLI 
chore: consolidate ADK service CLI options

PiperOrigin-RevId: 769944881
 2025-06-10 21:22:14 -07:00
Shangjie ChenandCopybara-Service c6e1e82efb chore: Minor improvement to session service 
- Add missing override.
- Add warning to failed actions.
- Remove unused import.
- Remove unused fields.
- Add type checking.

PiperOrigin-RevId: 767209697
 2025-06-04 10:58:35 -07:00
Shangjie ChenandCopybara-Service e99f87de73 chore: Minor improvement to session service 
- Add missing override.
- Add warning to failed actions.
- Remove unused import.
- Remove unused fields.
- Add type checking.

PiperOrigin-RevId: 766882634
 2025-06-03 17:13:33 -07:00
Shangjie ChenandCopybara-Service 8d36dbda52 feat: Add an option to use gcs artifact service in adk web. 
Resolves https://github.com/google/adk-python/issues/309

PiperOrigin-RevId: 765772763
 2025-06-01 00:28:49 -07:00
Shangjie ChenandCopybara-Service 01965bdd74 chore:Extract common options for adk web and adk api_server. 
PiperOrigin-RevId: 765326316
 2025-05-30 13:58:22 -07:00
Shangjie ChenandCopybara-Service 65063023a5 fix: Continue fetching events if there are multiple pages. 
Fixes https://github.com/google/adk-python/issues/920

PiperOrigin-RevId: 764985371
 2025-05-29 19:42:50 -07:00
Shangjie ChenandCopybara-Service db4bc9809c fix: Filter memory by app_name and user_id. 
Fixing https://github.com/google/adk-python/issues/1019.

PiperOrigin-RevId: 764903966
 2025-05-29 15:15:59 -07:00
Shangjie ChenandCopybara-Service bf27f22a95 fix: Make GroundingMetadata JSON serializable. Also use the same logic to simplify content serialization. 
PiperOrigin-RevId: 764401248
 2025-05-28 13:50:25 -07:00
Shangjie ChenandCopybara-Service d83ee1a6f2 fix: Remove excessive db call to update state table. 
PiperOrigin-RevId: 764375522
 2025-05-28 12:46:13 -07:00
Shangjie ChenandCopybara-Service d212e50c10 feat:Make VertexAiSessionService true async. 
PiperOrigin-RevId: 762547133
 2025-05-23 13:31:38 -07:00
Shangjie ChenandCopybara-Service 53b14325ce fix: Use sync request method in VertexAiSessionService. The api_client has it own event loop management. 
PiperOrigin-RevId: 761250268
 2025-05-20 15:20:21 -07:00
Shangjie ChenandCopybara-Service 0497fc578c fix: fix return type of get_session. 
PiperOrigin-RevId: 760638917
 2025-05-19 08:46:54 -07:00
Shangjie ChenandCopybara-Service 5ee17a3268 feat!: remove close_session from session service as it is not implemented yet. 
PiperOrigin-RevId: 759705443
 2025-05-16 12:08:57 -07:00
Shangjie ChenandCopybara-Service 2f006264ce chore: fix python format. 
PiperOrigin-RevId: 759674648
 2025-05-16 10:49:06 -07:00
Shangjie ChenandCopybara-Service a71d9ea9a1 fix: fix broken tests. 
PiperOrigin-RevId: 759312577
 2025-05-15 14:57:04 -07:00
Shangjie ChenandCopybara-Service 5b3204c356 feat! Update session service interface to be async. 
Also keep the sync version in the InMemorySessionService as create_session_sync() as a temporary migration option.

PiperOrigin-RevId: 759224250
 2025-05-15 11:17:13 -07:00
Shangjie ChenandCopybara-Service 3ea9eca929 feat! Remove list_events from session service interface as it has many overlap with get_session. 
PiperOrigin-RevId: 759184310
 2025-05-15 09:41:16 -07:00
Shangjie ChenandCopybara-Service 293f406148 fix: fix breaking delete_session. 
PiperOrigin-RevId: 758862440
 2025-05-14 15:42:44 -07:00
Shangjie ChenandCopybara-Service bc43a1196a feat: add _sync implementation in the inmemory session sevice. 
PiperOrigin-RevId: 758832846
 2025-05-14 14:22:46 -07:00
Shangjie ChenandCopybara-Service c39f24f072 fix: Refine the confusing error message. 
PiperOrigin-RevId: 757841288
 2025-05-12 11:20:54 -07:00
Shangjie ChenandCopybara-Service 70a6bfae90 fix: Explictly throw execption when custom id is set for vertex ai session service. 
PiperOrigin-RevId: 757815469
 2025-05-12 10:19:25 -07:00
Shangjie ChenandCopybara-Service c4ecef54bc fix: Generate event id at beginning, and regen for every new yield event. This is because trace module requires an event id. 
PiperOrigin-RevId: 756905471
 2025-05-09 14:22:42 -07:00
Shangjie ChenandCopybara-Service 0299020cc4 ADK v0.5.0 release. 
PiperOrigin-RevId: 756410107
 2025-05-08 12:15:46 -07:00
Shangjie ChenandCopybara-Service 26f55fb5ab fix: Use new id for every event. 
Fixes #297

PiperOrigin-RevId: 755901205
 2025-05-07 10:07:38 -07:00
Shangjie ChenandCopybara-Service f96cdc675c fix: Use await to wait for the full list rather than async for which only works for async iterable. 
PiperOrigin-RevId: 755541004
 2025-05-06 15:10:14 -07:00
Shangjie ChenandCopybara-Service e2ba484b43 fix: add missing async change in code execution. 
PiperOrigin-RevId: 755429199
 2025-05-06 10:37:17 -07:00
Shangjie ChenandCopybara-Service 905c20dad6 ADK changes 
PiperOrigin-RevId: 755201925
 2025-05-05 21:58:47 -07:00
Shangjie ChenandCopybara-Service bcf1deb582 ADK changes 
PiperOrigin-RevId: 754131080
 2025-05-02 14:19:18 -07:00
Shangjie ChenandCopybara-Service 14933ba470 feat: Extract content encode/decode logic to a shared util and resolve issues with JSON serialization. 
feat: Update key length for DB table to avoid key too long issue in mysql

PiperOrigin-RevId: 753614879
 2025-05-01 09:13:28 -07:00
Shangjie ChenandCopybara-Service 2a9ddec7e3 Set the max size of strings in database columns. 
PiperOrigin-RevId: 752918808
 2025-04-29 16:07:45 -07:00
Shangjie ChenandCopybara-Service 9a9f7a3765 Update callback comments. 
PiperOrigin-RevId: 752162448
 2025-04-27 22:20:02 -07:00