Python对接微信AI-Welm

刚刚看到新闻发布的Welm - 微信AI产品，夏柔迫不及待的搜索了一下关于welm的域名，值得高兴的是还有很多后缀没注册，所以现在的welm.cc属于夏柔的～

当然这一好消息必须先通知群友们，所以很多welm的域名都归我们群友啦～

废话不多说，上代码：

import requests
import json

def query_api(prompt, model, max_tokens, temperature, top_p, top_k, n, echo, stop, token):
    """
    查询API接口
    """
    url = 'https://welm.weixin.qq.com/v1/completions'
    headers = {
        'Content-Type': 'application/json',
        'Authorization': f'Bearer {token}'
    }
    data = {
        "prompt": prompt,
        "model": model,
        "max_tokens": max_tokens,
        "temperature": temperature,
        "top_p": top_p,
        "top_k": top_k,
        "n": n,
        "echo": echo,
        "stop": stop
    }

    try:
        response = requests.post(url, headers=headers, data=json.dumps(data))
        response.raise_for_status()
    except requests.exceptions.HTTPError as err:
        print(f"请求失败: {err}")
        return None

    result = json.loads(response.text)
    return result

if __name__ == '__main__':
    prompt = "测试"
    model = "xl"
    max_tokens = 16
    temperature = 0.0
    top_p = 0.0
    top_k = 10
    n = 1
    echo = False
    stop = ",，.。"
    token = "你的token"

    result = query_api(prompt, model, max_tokens, temperature, top_p, top_k, n, echo, stop, token)
    if result:
        print(result)

第45行换成你们在welm官网申请的key