feat: Add comprehensive test coverage and unified session management for agents #543
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
…for agents This commit introduces extensive test coverage for both ExploreAgent and DecomposeAgent, along with refactoring to ensure unified session management across all agent-based workflow operations. Changes: - Create DecomposeAgent class in dedicated file (decompose_agent.py) - Add 21 comprehensive tests for DecomposeAgent - Add 9 session management tests for ExploreAgent - Update workflow operations (decompose.py, exploration.py) to use unified session management with session_storage_path and include_session_in_prompt - Export DecomposeAgent from agents module Test Coverage (54 total tests, 100% passing): - DecomposeAgent: 21 tests covering initialization, decomposition methods, integration scenarios, error handling, and session management - ExploreAgent: 33 tests including new session management validation Session Management Features: - session_id: Conversation tracking across agent calls - session_storage_path: Persistent storage location configuration - include_session_in_prompt: Optional session context inclusion All agents and operations now consistently implement the unified session management approach for improved conversation persistence and tracking. 🤖 Generated with Claude Code Co-Authored-By: Claude <[email protected]>
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Summary
This PR adds comprehensive test coverage for the automation agents and ensures unified session management across all agent-based workflow operations.
Key Changes
1. DecomposeAgent Refactoring (
decompose_agent.py)DecomposeAgentclass to dedicated moduledecompose()method for task decomposition2. Test Coverage (54 total tests, 100% passing)
DecomposeAgent Tests (21 new tests):
ExploreAgent Tests (9 new session management tests):
3. Unified Session Management
Updated workflow operations to use consistent session configuration:
Parameters supported:
session_id- Track conversations across agent callssession_storage_path- Configure persistent storage locationinclude_session_in_prompt- Control session context inclusionFiles updated:
plugins/automation/workflows/steps/operations/decompose.pyplugins/automation/workflows/steps/operations/exploration.pyTest Results
Benefits
Testing
All tests pass successfully:
uv run pytest plugins/automation/tests/test_decompose_agent.py \ plugins/automation/tests/test_explore_agent.py -vRelated
🤖 Generated with Claude Code