You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
65 lines
2.9 KiB
65 lines
2.9 KiB
# coding=utf-8
|
|
|
|
import requests
|
|
import json
|
|
import urllib3
|
|
from Config import *
|
|
|
|
if __name__ == '__main__':
|
|
# 禁止警告
|
|
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
|
|
|
|
# 示例用户输入
|
|
#user_input = "画一个函数 y = x^2 的图像，并添加切线。只需要返回指令，其它信息不要返回！同时，不要输出```"
|
|
user_input = """
|
|
画一个函数 y = sin(x) 的图像，并在 x = π/2 处添加切线。同时，绘制该函数在区间 [0, 2π] 上的定积分区域，并标注积分值。
|
|
"""
|
|
# 设置请求头和数据
|
|
headers = {
|
|
'Content-Type': 'application/json',
|
|
'Authorization': 'Bearer ' + HW_API_KEY # 把yourApiKey替换成真实的API Key
|
|
}
|
|
data = {
|
|
"model": HW_MODEL_NAME,
|
|
"messages": [
|
|
{"role": "system", "content": "You are a helpful assistant that generates GeoGebra commands."},
|
|
{"role": "user", "content": user_input}
|
|
],
|
|
"stream": True, # 启用流式输出
|
|
"temperature": 1.0
|
|
}
|
|
|
|
# 初始化变量，用于保存完整的响应内容
|
|
full_response = ""
|
|
|
|
# 发送请求并处理流式响应
|
|
with requests.post(HW_API_URL, headers=headers, data=json.dumps(data), verify=False, stream=True) as resp:
|
|
if resp.status_code == 200:
|
|
for line in resp.iter_lines():
|
|
if line: # 过滤掉空行
|
|
decoded_line = line.decode('utf-8')
|
|
if decoded_line.startswith("data: "): # 处理以 "data: " 开头的行
|
|
json_str = decoded_line[len("data: "):] # 去掉 "data: " 前缀
|
|
if json_str == "[DONE]": # 流式响应结束
|
|
break
|
|
try:
|
|
# 解析 JSON 数据
|
|
json_data = json.loads(json_str)
|
|
# 提取流式输出的内容
|
|
if "choices" in json_data and len(json_data["choices"]) > 0:
|
|
# 检查是否有 "delta" 字段
|
|
delta = json_data["choices"][0].get("delta", {})
|
|
# 尝试从 "content" 或 "reasoning_content" 中提取内容
|
|
content = delta.get("content", delta.get("reasoning_content", ""))
|
|
if content:
|
|
# 实时逐字输出内容
|
|
print(content, end="", flush=True)
|
|
# 将内容累积到完整响应中
|
|
full_response += content
|
|
except json.JSONDecodeError:
|
|
pass # 忽略解析错误
|
|
else:
|
|
print("请求失败，状态码:", resp.status_code)
|
|
print("响应内容:", resp.text)
|
|
|