-
Notifications
You must be signed in to change notification settings - Fork 1.1k
[ag-ui-adk] Fixes #907 Finegrained configuration of SessionManager #908
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[ag-ui-adk] Fixes #907 Finegrained configuration of SessionManager #908
Conversation
contextablemark
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@jplikesbikes Thank you for another great submission!
Besides the one comment, this one needs more tests, specifically :
-
There are tests for both flags false and both true in the "end of session" behavior, but needs to be tests for the two "mixed" combinations.
-
Additional cleanup loop tests for the same mixed settings as well as cleanup loop behavior with both set to false.
-
Test that flags make it through to agent._session_manager when either the direct
ADKAgentconstructor is used or whenADKAgent.from_appis used -
Edge case tests
- What happens when save session is true, but memory service is set to None?
- Verify that sessions with
pending_tool_callsare preserved even whendelete_session_on_cleanup=True
…on_to_memory_on_cleanup=True
…session_on_cleanup=True
contextablemark
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
lgtm... will run e2e tests and merge
|
E2E tests confirmed with #928 |
This pr implements issue #907 it separates the concerns of deleting the session from the adk SessionService and adding the session to the MemoryService during session clean up in the SessionManager
It splits the
auto_cleanupparameter into two separate parametersdelete_session_on_cleanupandsave_session_to_memory_on_cleanupThese parameters are exposed when creating the ADKAgent