99 lines
3.6 KiB
Python
99 lines
3.6 KiB
Python
|
from openai import OpenAI
|
|||
|
|
|
|||
|
|
# 初始化OpenAI客户端
|
|||
|
|
client = OpenAI(
|
|||
|
|
api_key="sk-f6da0c787eff4b0389e4ad03a35a911f",
|
|||
|
|
base_url="https://dashscope.aliyuncs.com/compatible-mode/v1"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
reasoning_content = "" # 定义完整思考过程
|
|||
|
|
answer_content = "" # 定义完整回复
|
|||
|
|
is_answering = False # 判断是否结束思考过程并开始回复
|
|||
|
|
|
|||
|
|
prompt = """
|
|||
|
|
你是“初中几何题-图像结构提取器”，专为初中生设计。任务只有一项：
|
|||
|
|
看到几何题图片后，用中文输出“完整的题干原文 + 图形客观结构描述”，绝口不提解题思路或答案。
|
|||
|
|
目标：
|
|||
|
|
让后续 GeoGebra 生成器仅凭你的输出即可复刻原图，同时让初中生一眼看懂图形。
|
|||
|
|
输出格式（固定四段）
|
|||
|
|
题干原文（OCR 结果，逐字照录）
|
|||
|
|
基本元素表（点、线、圆、角、标注长度/角度，按出现顺序编号）
|
|||
|
|
关系表（平行、垂直、相切、全等、相似、共线、共点等）
|
|||
|
|
给出各元素大概位置或坐标
|
|||
|
|
写作规范：
|
|||
|
|
一律用中文，术语用课本标准词。
|
|||
|
|
描述顺序：先整体外形（如“梯形 ABCD”），再局部细节（如“AB∥CD”）。
|
|||
|
|
长度、角度直接写数值并带单位；若图中仅给符号，则照抄符号。
|
|||
|
|
位置关系用“//、⊥、≅、∼、∈”等符号，后接括号说明。
|
|||
|
|
注意几何图形的形状和线段的长度
|
|||
|
|
不使用“可能、大约”等模糊词。
|
|||
|
|
不解释题意，不提示解法。
|
|||
|
|
示例模板：
|
|||
|
|
【题干原文】
|
|||
|
|
如图，梯形 ABCD 中，AB∥CD …（以下略）
|
|||
|
|
【基本元素表】
|
|||
|
|
点：A、B、C、D、E（对角线交点）
|
|||
|
|
线段：AB=6 cm，CD=10 cm，AD=4 cm，BC=4 cm，AC，BD
|
|||
|
|
角：∠DAB=60°，∠ABC=120°
|
|||
|
|
圆：无
|
|||
|
|
【关系表】
|
|||
|
|
AB // CD
|
|||
|
|
AD = BC
|
|||
|
|
AC 与 BD 交于 E
|
|||
|
|
∠DAB 与 ∠ABC 互补
|
|||
|
|
【元素位置】
|
|||
|
|
A点在左上，B点在右上，C点在A点和B点的下方
|
|||
|
|
——请严格按以上格式输出，不要省略任何字段。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
# 创建聊天完成请求
|
|||
|
|
completion = client.chat.completions.create(
|
|||
|
|
model="qvq-max-latest", # 此处以 qvq-max 为例，可按需更换模型名称
|
|||
|
|
messages=[
|
|||
|
|
{
|
|||
|
|
"role": "user",
|
|||
|
|
"content": [
|
|||
|
|
{
|
|||
|
|
"type": "image_url",
|
|||
|
|
"image_url": {
|
|||
|
|
"url": "https://dsideal.obs.cn-north-1.myhuaweicloud.com/wb/img10.jpg"
|
|||
|
|
},
|
|||
|
|
},
|
|||
|
|
{"type": "text",
|
|||
|
|
"text": prompt},
|
|||
|
|
],
|
|||
|
|
},
|
|||
|
|
],
|
|||
|
|
stream=True,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
print("\n" + "=" * 20 + "思考过程" + "=" * 20 + "\n")
|
|||
|
|
|
|||
|
|
for chunk in completion:
|
|||
|
|
# 如果chunk.choices为空，则打印usage
|
|||
|
|
if not chunk.choices:
|
|||
|
|
print("\nUsage:")
|
|||
|
|
print(chunk.usage)
|
|||
|
|
else:
|
|||
|
|
delta = chunk.choices[0].delta
|
|||
|
|
# 打印思考过程
|
|||
|
|
if hasattr(delta, 'reasoning_content') and delta.reasoning_content != None:
|
|||
|
|
print(delta.reasoning_content, end='', flush=True)
|
|||
|
|
reasoning_content += delta.reasoning_content
|
|||
|
|
else:
|
|||
|
|
# 开始回复
|
|||
|
|
if delta.content != "" and is_answering is False:
|
|||
|
|
print("\n" + "=" * 20 + "完整回复" + "=" * 20 + "\n")
|
|||
|
|
is_answering = True
|
|||
|
|
# 打印回复过程
|
|||
|
|
print(delta.content, end='', flush=True)
|
|||
|
|
answer_content += delta.content
|
|||
|
|
|
|||
|
|
# print("=" * 20 + "完整思考过程" + "=" * 20 + "\n")
|
|||
|
|
# print(reasoning_content)
|
|||
|
|
# print("=" * 20 + "完整回复" + "=" * 20 + "\n")
|
|||
|
|
# 保存成QvqResult.txt
|
|||
|
|
with open("QvqResult.txt", "w", encoding='utf-8') as f:
|
|||
|
|
f.write(answer_content)
|
|||
|
|
print("试题解析文本保存成功！")
|