多Agent联合分析Python-C跨语言漏洞

此项目使用多Agent系统分析PyCBench中的Python-C跨语言漏洞案例。

安装依赖

pip install -r requirements.txt

配置

创建.env文件，配置API：

Deepseek API

API_TYPE=deepseek
API_KEY=your_deepseek_api_key

Coze API

API_TYPE=coze
COZE_BASE_URL=https://api.coze.cn
API_KEY=your_coz_api_key
BOT_ID=your_bot_id
CONVERSATION_ID=your_conversation_id  # 可选，不填则自动创建会话

本地模型

API_TYPE=local
MODEL_NAME=microsoft/DialoGPT-medium  # 或其他Hugging Face模型

运行

python main.py

程序将遍历PyCBench中的所有案例，为每个案例生成analysis_results.json文件，包含多Agent的分析结果。

Agent角色

• C_Analyzer: 分析C代码中的漏洞
• Python_Analyzer: 分析Python代码中的漏洞
• CrossLang_Analyzer: 分析跨语言交互中的问题
• Coordinator: 整合所有分析结果