通过 HTTP 请求调用本地大模型生成扑克牌
背景介绍
在游戏开发或自动化测试中,快速生成随机扑克牌是一个常见需求。本文通过 Python 的 requests
库,向本地部署的大模型接口发送 HTTP 请求,实现自动化发牌功能。代码简洁高效,适合初学者快速上手。
核心代码解析
以下代码展示了如何通过 POST 请求调用本地模型接口生成 5 张扑克牌:
import requests
# 配置目标地址和请求头
url = "http://localhost:1234/v1/chat/completions"
headers = {
"Content-Type": "application/json"
}
# 定义请求体数据
data = {
"stream": False,
"detail": False,
"model": "qwen3-0.6b",
"tool_choice": "auto",
"tools": [
{
"type": "function",
"function": {
"name": "get_poker_cards",
"description": "生成指定数量的扑克牌",
"parameters": {
"type": "object",
"properties": {
"num": {
"type": "integer",
"description": "需要生成的扑克牌数量",
"minimum": 1,
"maximum": 52
}
},
"required": ["num"]
}
}
}
],
"messages": [
{
"role": "system",
"content": "你是一个专业的发牌助手,必须使用get_poker_cards工具来处理发牌请求"
},
{
"role": "user",
"content": "请发5张牌"
}
]
}
# 发送请求并打印结果
response = requests.post(url, headers=headers, json=data)
print("Status Code:", response.status_code)
print("Response Body:", response.text)
代码详解
1. 接口配置
url
: 本地模型服务的 API 地址(需提前部署)。headers
: 指定请求内容类型为 JSON。
2. 请求体参数
tools
: 定义可用工具get_poker_cards
,限制生成牌数范围为 1~52。messages
: 模拟用户对话,系统提示强制使用工具处理发牌请求。
3. 发送请求
- 使用
requests.post()
发送 POST 请求,传递 JSON 格式的数据。 - 打印响应状态码和结果(如生成的牌面信息)。
运行结果示例
假设接口正常返回,输出可能如下:
Status Code: 200
Response Body: {"cards": ["♠️8", "♥️Q", "♦️3", "♣️A", "♠️2"]}
扩展应用
动态调整牌数
修改data["messages"][1]["content"]
中的数字即可生成任意数量的牌(1~52)。集成到游戏框架
将此代码封装为函数,作为游戏初始化模块的一部分,实现自动洗牌、发牌流程。错误处理优化
增加异常捕获逻辑,应对网络中断或接口异常:try: response = requests.post(url, headers=headers, json=data, timeout=5) response.raise_for_status() # 抛出 HTTP 错误 except requests.exceptions.RequestException as e: print("请求失败:", e)
总结
通过 requests
库与本地模型接口的交互,我们实现了高效的扑克牌生成逻辑。这种方法不仅适用于游戏开发,还可拓展到自动化测试、数据生成等领域。结合 requests
的简洁性和 HTTP 接口的灵活性,开发者能快速构建实用工具链。
实践建议:尝试将此功能封装为 Web 服务,通过浏览器直接调用,进一步提升交互体验!