|
b0a2aa35d4
|
feat: implement named toolset groups for PraisonAI tool registry (fixes #1841) (#1850)
* feat: implement named toolset groups for PraisonAI tool registry (fixes #1841)
This implements named toolset groups functionality to address the lack of
organized tool collections in PraisonAI's tool registry system.
Core SDK Changes (praisonaiagents):
- Add toolsets.py with ToolsetSpec dataclass and ToolsetRegistry
- Implement 8 prebuilt toolsets: web, files, code, system, scraping, research, safe, development
- Support toolset composition via includes with circular dependency detection
- Add TTL availability caching (30s) to ToolRegistry.list_available_tools()
- Update Agent.__init__ to accept toolsets parameter with automatic tool resolution
- Export toolsets functionality in main package __init__.py
Wrapper Layer Changes (praisonai):
- Extend ToolResolver with resolve_toolsets() and resolve_tools_and_toolsets()
- Add --toolset CLI flag to 'praisonai run' and 'praisonai chat' commands
- Integrate toolset resolution in CLI execution pipeline
Key Features:
- Named toolset groups for organized tool management
- Progressive disclosure (safe → research → development)
- Least-privilege tool surfaces for restricted environments
- Performance improvements via TTL caching
- Full backward compatibility with existing tools parameter
- Thread-safe operations for multi-agent scenarios
Usage:
- Python: Agent(role='researcher', toolsets=['research'])
- CLI: praisonai run 'task' --toolset research
- YAML: toolsets: [research] in agent configuration
Co-authored-by: MervinPraison <[email protected]>
* fix: address critical issues found in code review
- Fix CLI tools write-back regression in main.py (silent tool dropping)
- Improve error handling in Agent toolset resolution (fail fast on invalid toolsets)
- Fix mutable state leakage in ToolsetRegistry (defensive copying)
- Fix cache collision in ToolRegistry (use registry keys not tool.name)
- Fix NameError in CLI commands (missing toolset parameters)
- Add deduplication in tool resolution
- Improve test reliability with cleanup and assertions
- Filter empty toolset names from CLI input
Co-authored-by: Mervin Praison <[email protected]>
* fix: resolve final registry.py merge conflicts
- Fix missed conflict in list_available_tools method
- Proper TTL caching using registry keys to avoid collisions
- Clean up clear() method to remove _functions reference
Co-authored-by: Mervin Praison <[email protected]>
* feat: complete named toolsets YAML integration and testing
Wire YAML support for toolsets:
- Add 'toolsets' to known_fields in agents_generator.py
- Update validate_yaml_tools to validate toolset references
- Update praisonai_adapter to pass toolsets from YAML to Agent
Enhance testing:
- Move test_toolsets.py to tests/unit/test_toolsets.py
- Convert to pytest conventions with proper cleanup
- Add real agentic tests requiring LLM calls per AGENTS.md
- Add comprehensive test coverage including YAML validation
Fix error handling:
- Remove ImportError swallowing in Agent.__init__ for fail-fast behavior
- Add clear error messages for invalid/missing toolsets
Add documentation and examples:
- Add examples/python/toolsets_example.py with comprehensive demos
- Add examples/yaml/toolsets_workflow.yaml showing YAML usage
- Include both simple usage and complex multi-agent workflows
All changes maintain backward compatibility and follow AGENTS.md guidelines.
🤖 Generated with [Claude Code](https://claude.ai/code)
Co-Authored-By: Claude <[email protected]>
---------
Co-authored-by: praisonai-triage-agent[bot] <272766704+praisonai-triage-agent[bot]@users.noreply.github.com>
Co-authored-by: MervinPraison <[email protected]>
Co-authored-by: Claude <[email protected]>
|
2026-06-12 14:00:39 +01:00 |
|