Agent 网络流量快速实验工程
本工程用于快速完成以下闭环:
- 通过大模型生成任务行为画像(任务阶段、工具调用、拆分、消息大小、处理时间等);
- 将行为画像扩展为结构化 Agent 事件日志;
- 从训练日志估计状态转移、持续时间、消息大小、超时和分支参数;
- 使用双层状态机、节点队列和离散事件调度进行仿真;
- 自动完成预测验证、基线对比、压力实验和重试放大实验;
- 输出 CSV、JSON 和 PNG 图表。
快速开始
当前环境已具备主要依赖,直接运行:
cd agent_traffic_experiments
python run_pipeline.py
默认使用 local 模式,不需要 API Key。结果写入 outputs/latest/。
使用大模型生成行为画像
本工程兼容 OpenAI 风格的 /v1/chat/completions 接口。先设置环境变量:
$env:OPENAI_API_KEY="你的密钥"
$env:OPENAI_BASE_URL="https://api.openai.com/v1"
$env:OPENAI_MODEL="你要使用的模型名称"
python run_pipeline.py --generator llm
也可以使用其他兼容服务,只需修改 OPENAI_BASE_URL 和 OPENAI_MODEL。
大模型只负责生成少量、可解释的任务行为画像;本地生成器会校验画像并扩展成大量事件。这样比让大模型直接输出数万行日志更稳定、更便宜,也能保证消息 ID、父子关系、时间戳和流量统计一致。
如果 LLM 请求失败,程序默认自动回退到本地画像。使用 --no-fallback 可禁止回退。
常用命令
只生成数据:
python run_pipeline.py --steps generate
生成数据、估参并运行实验:
python run_pipeline.py --steps generate,estimate,experiment
使用已有数据重新实验:
python run_pipeline.py --steps estimate,experiment
指定配置和输出目录:
python run_pipeline.py --config configs/default.yaml --output outputs/run_001
输出说明
outputs/latest/
data/
behavior_profiles.json # LLM或本地生成的任务画像
train_events.csv # 参数估计数据
test_events.csv # 留出验证数据
task_truth.csv # 测试任务真实聚合值
parameters/
estimated_parameters.json # 从训练日志估计的模型参数
results/
validation_metrics.csv # 完整模型与基线误差
validation_predictions.csv # 逐场景预测值和真实值
stress_results.csv # 到达率压力实验
retry_results.csv # 超时与重试放大实验
summary.json # 实验摘要
figures/
prediction_vs_truth.png
model_comparison.png
stress_curves.png
retry_heatmap.png
实验内容
E1 预测验证
训练集用于估计参数,测试集作为“模拟实测值”。完整状态机模型预测测试场景的内部字节数、任务时延、消息数和工具调用数。
E2 基线对比
static_mean:固定平均流量放大倍数;no_context:不区分任务类型和任务阶段;full_model:按任务画像和估计参数运行状态机仿真。
E3 压力实验
逐渐提高外部到达率,观察吞吐量、平均/P95延迟、队列峰值、失败率和流量放大系数。
E4 重试放大实验
改变超时概率和最大重试次数,观察任务成功率、平均重试数和内部流量放大系数。
数据字段
事件日志主要字段包括:
timestamp、task_id、message_id、parent_message_id;task_type、task_phase、source、destination;state_before、state_after、state_duration_ms;message_type、message_size_bytes;queue_length、success、retry_count、is_external。
注意事项
- 当前数据是用于方法验证的模拟数据,不能冒充真实生产日志;
- LLM 画像应在报告中标注模型名称、生成时间和提示词版本;
- 正式参赛前,应尽量用少量真实 Agent 日志替换或校准模拟参数;
- 所有随机实验都由配置中的随机种子控制,便于复现。