pip install locustlocust --version # 查看版本号
from locust import HttpUser, task, betweenclass OllamaUser(HttpUser):# 每1-3秒间隔启动下一个任务wait_time = between(1, 3)@taskdef generate_text(self):response = self.client.post("/api/generate",json={"model": "qwq:32b","prompt": "简单解释一下量子计算","stream": False},# timeout=60 # 超时)# 可选：检查超时或失败if response.status_code != 200:print(f"Request failed with status {response.status_code}")
$ locust -f test.py -P 1111
原文始发于微信公众号（生有可恋）：API 压测工具 Locust
免责声明:文章中涉及的程序(方法)可能带有攻击性，仅供安全研究与教学之用，读者将其信息做其他用途，由读者承担全部法律及连带责任，本站不承担任何法律及连带责任；如有问题可邮件联系(建议使用企业邮箱或有效邮箱,避免邮件被拦截，联系方式见首页)，望知悉。
- 左青龙
- 微信扫一扫
-
- 右白虎
- 微信扫一扫
-
评论