使用 Python 的 `requests` 库实现自动化发牌功能(qwen3工具调用)

发布于:2025-07-08 ⋅ 阅读:(14) ⋅ 点赞:(0)

通过 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"]}

扩展应用

  1. 动态调整牌数
    修改 data["messages"][1]["content"] 中的数字即可生成任意数量的牌(1~52)。

  2. 集成到游戏框架
    将此代码封装为函数,作为游戏初始化模块的一部分,实现自动洗牌、发牌流程。

  3. 错误处理优化
    增加异常捕获逻辑,应对网络中断或接口异常:

    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 服务,通过浏览器直接调用,进一步提升交互体验!


网站公告

今日签到

点亮在社区的每一天
去签到