Agent-Wiz支持多种主流AI框架,并提供了扩展机制,方便开发者适配新的框架。
1. 支持的框架
目前,Agent-Wiz内置了对以下框架的支持:
- AutoGen:支持核心功能和AgentChat模块。
- CrewAI:专为团队协作智能体设计。
- LangGraph:处理有向图工作流的复杂逻辑。
- LlamaIndex:支持数据索引和查询智能体。
- n8n:解析自动化工作流。
- OpenAI Agents:兼容OpenAI的智能体API。
- Pydantic-AI:支持基于Pydantic的智能体。
- Swarm:专注于多智能体协作。
2. 扩展新框架
如果用户需要使用Agent-Wiz分析未被内置支持的框架,可以通过以下步骤扩展:
- 开发自定义解析器:Agent-Wiz提供了SDK接口,允许开发者编写针对特定框架的AST解析器。
- 集成解析器:将自定义解析器集成到Agent-Wiz的工具链中,即可在
extract
命令中通过--framework
选项调用新框架。
注意事项:
为了确保解析准确性和效率,建议新框架的代码结构清晰,并包含必要的注释。此外,自定义解析器的开发需要一定的Python和静态代码分析知识。
Agent-Wiz的开源特性使得扩展新框架成为可能,用户可以根据具体需求灵活适配。
本答案来源于文章《Agent-Wiz:分析AI智能体工作流程和安全风险》