263 lines
11 KiB
Python
263 lines
11 KiB
Python
import asyncio
|
||
import logging
|
||
|
||
from openai import AsyncOpenAI
|
||
from zai import ZhipuAiClient
|
||
|
||
from Config.Config import ZHIPU_API_KEY
|
||
|
||
# 设置日志
|
||
logger = logging.getLogger(__name__)
|
||
|
||
def create_llm_client() -> AsyncOpenAI:
|
||
"""初始化并返回异步OpenAI客户端"""
|
||
return AsyncOpenAI(
|
||
#api_key=LLM_API_KEY,
|
||
#base_url=LLM_BASE_URL
|
||
api_key="sk-f6da0c787eff4b0389e4ad03a35a911f",
|
||
base_url="https://dashscope.aliyuncs.com/compatible-mode/v1"
|
||
)
|
||
|
||
# 【5】如果在使用Segment创建两个点的线段时，需要指定一个标签名，然后使用ShowLabel隐藏这个标签名，注意使用的是ShowLabel而不是SetVisibleInView，而且只是Segment创建两个点的线段时才隐藏标签名
|
||
async def getGgbCommand(QvqResult):
|
||
"""异步版本的getGgb函数 - 支持流式输出"""
|
||
custom_prompt = """
|
||
# Role: GeoGebra 几何图形生成器
|
||
## Profile
|
||
- language: 中文
|
||
- description: 专门用于根据初中几何题的题干和几何图形的描述来生成Geogebra指令集的角色，确保生成的指令集严格遵循几何约束和动态坐标系构建规则，并具备错误熔断机制。
|
||
- background: 拥有深厚的几何学和计算机科学背景，熟悉Geogebra软件的操作和指令集编写。
|
||
- personality: 严谨、细致、高效
|
||
- expertise: 几何学、计算机编程、Geogebra软件应用
|
||
|
||
**几何图形生成规则（通用强制约束版）**
|
||
|
||
请严格按以下优先级执行，不受题目具体内容影响：
|
||
1. **几何约束最高优先级**
|
||
- 描述中提到的水平线段，一定要保持水平，不可以进行变更
|
||
- 当文字描述与坐标/位置冲突时，以文字描述的几何关系为准
|
||
- 必须自动验证：{长度|角度|中点|平行|垂直}关系
|
||
- 验证失败时：重新计算坐标并报告修正
|
||
- 有动点的，要设置滑块。没有动点的，不要设置滑块。
|
||
- 检查描述中提到的最长的边做为X轴，最长边的左端点做为原点
|
||
|
||
2. **动态坐标系构建**
|
||
```mermaid
|
||
graph TB
|
||
A[定位原点] --> B{原点坐标明确？}
|
||
B -->|是| C[直接采用]
|
||
B -->|否| D[根据相对关系推算]
|
||
E[建立坐标系] --> F{存在水平/垂直要求？}
|
||
F -->|是| G[调整坐标满足]
|
||
F -->|否| H[保持默认方向]
|
||
```
|
||
|
||
3、错误熔断机制
|
||
|
||
检测到矛盾 → 触发修正流程：
|
||
|
||
if not satisfy_constraint(description):
|
||
new_coords = recalculate_by_geometry()
|
||
log_adjustment(old_coords, new_coords, reason)
|
||
|
||
4、元素生成协议
|
||
|
||
顺序：固定点 → 固定线 → 动态点(初始位) → 动态元素
|
||
|
||
依赖检查：确保父元素先于子元素生成
|
||
|
||
5、输出结构（强制）
|
||
|
||
[修正报告] （若有调整）
|
||
[坐标摘要] （最终使用坐标）
|
||
[Geogebra指令] （完整可执行代码，以```geogebra 开头，以 ```结束。【注意】```geogebra 前面一定要加上 \n 换行！）
|
||
注意：
|
||
【1】GGB指令集中不要添加注释！不要添加除指令集外的其它描述性文字，需要可以直接批量执行的！
|
||
【2】仔细检查原图中是线段还是直线，要严格进行识别！
|
||
【3】仔细检查原图中的每一条边是否都正确进行了线段标识！
|
||
【4】不显示网格（使用ShowGrid(false)），不显示坐标轴（使用ShowAxes(false)）
|
||
【5】注意不要使用Perpendicular，需要使用PerpendicularLine来替代
|
||
【6】注意不要使用SetLabel，因为会提示“未知的指令”。
|
||
【7】注意不要使用Angle，无需标注角度。
|
||
【8】如果有滑块使用Slider
|
||
【9】如果没有动点就无需添加滑块。
|
||
【10】注意不要使用TangentPoint，因为会提示“未知的指令”。
|
||
|
||
6、题目的文字描述信息
|
||
```
|
||
{QvqResult}
|
||
```
|
||
"""
|
||
|
||
try:
|
||
# 创建异步OpenAI客户端
|
||
client = create_llm_client()
|
||
|
||
# 使用异步流式调用
|
||
completion = await client.chat.completions.create(
|
||
model="deepseek-r1-0528",
|
||
#model="deepseek-reasoner",
|
||
messages=[{"role": "user", "content": custom_prompt.replace("{QvqResult}", QvqResult)}],
|
||
stream=True
|
||
)
|
||
|
||
is_answering = False
|
||
async for chunk in completion:
|
||
delta = chunk.choices[0].delta
|
||
# 打印思考过程
|
||
if hasattr(delta, 'reasoning_content') and delta.reasoning_content != None:
|
||
yield delta.reasoning_content
|
||
else:
|
||
# 开始回复
|
||
if delta.content != "" and is_answering is False:
|
||
print("\n" + "=" * 20 + "完整回复" + "=" * 20 + "\n")
|
||
is_answering = True
|
||
# 打印回复过程
|
||
yield delta.content
|
||
except Exception as e:
|
||
print(f"异步getGgb调用错误: {str(e)}")
|
||
yield f"错误: {str(e)}"
|
||
|
||
|
||
async def translate_to_ggb(natural_language_cmd):
|
||
"""将自然语言指令翻译成GeoGebra指令集（异步版本）"""
|
||
# 精简的prompt，专注于单句指令翻译
|
||
custom_prompt = f"""
|
||
请将以下自然语言指令翻译为GeoGebra指令集：
|
||
"{natural_language_cmd}"
|
||
|
||
要求：
|
||
1. 仅返回GeoGebra指令，不包含任何解释性文字
|
||
2. 指令必须可直接在GeoGebra中执行
|
||
3. 保持指令简洁明确
|
||
4. 以```geogebra 开头，以 ```结束 【注意】```geogebra 前面一定要加上 \n 换行！
|
||
5. 用户提到的元素，比如点A,线段AB等，都视为已存在，不要重新创建
|
||
"""
|
||
|
||
try:
|
||
# 创建异步OpenAI客户端
|
||
client = create_llm_client()
|
||
|
||
# 使用异步流式调用
|
||
completion = await client.chat.completions.create(
|
||
model="deepseek-r1-0528",
|
||
messages=[{"role": "user", "content": custom_prompt}],
|
||
stream=True
|
||
)
|
||
|
||
is_answering = False
|
||
async for chunk in completion:
|
||
delta = chunk.choices[0].delta
|
||
# 打印思考过程
|
||
if hasattr(delta, 'reasoning_content') and delta.reasoning_content != None:
|
||
yield delta.reasoning_content
|
||
else:
|
||
# 开始回复
|
||
if delta.content != "" and is_answering is False:
|
||
print("\n" + "=" * 20 + "完整回复" + "=" * 20 + "\n")
|
||
is_answering = True
|
||
# 打印回复过程
|
||
yield delta.content
|
||
except Exception as e:
|
||
print(f"翻译指令错误: {str(e)}")
|
||
yield f"错误: {str(e)}"
|
||
|
||
|
||
# 示例用法
|
||
if __name__ == '__main__':
|
||
# file_name = "QvqResult_14.txt"
|
||
# with open(file_name, 'r', encoding='utf-8') as file:
|
||
# QvqResult = file.read() # 直接获取整个文件内容为字符串
|
||
#
|
||
# async def test():
|
||
# async for content in getGgbCommand(QvqResult):
|
||
# if content:
|
||
# print(content, end="", flush=True)
|
||
#
|
||
#
|
||
# asyncio.run(test())
|
||
|
||
|
||
async def test_translate():
|
||
async for content in translate_to_ggb("把A和B连接起来"):
|
||
if content:
|
||
print(content, end="", flush=True)
|
||
|
||
|
||
# 运行测试
|
||
asyncio.run(test_translate())
|
||
|
||
|
||
|
||
async def process_geometry_image(image_url: str):
|
||
"""
|
||
真正的流式处理几何题图片
|
||
"""
|
||
client = ZhipuAiClient(api_key=ZHIPU_API_KEY)
|
||
|
||
prompt = """你是"初中几何题-图像结构提取器"，专为初中生设计。任务只有一项：
|
||
看到几何题图片后，用中文输出"完整的题干原文 + 图形客观结构描述"，绝口不提解题思路或答案。
|
||
【返回格式】：标准markdown+latex格式
|
||
目标：
|
||
让后续 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点的下方
|
||
——请严格按以上格式输出，不要省略任何字段。"""
|
||
|
||
try:
|
||
messages = [{
|
||
"role": "user",
|
||
"content": [
|
||
{"type": "text", "text": prompt},
|
||
{"type": "image_url", "image_url": {"url": image_url}}
|
||
]
|
||
}]
|
||
|
||
# 使用智谱SDK的流式调用（同步调用包装为异步）
|
||
stream = await asyncio.to_thread(
|
||
client.chat.completions.create,
|
||
model="glm-4.1v-thinking-flash",
|
||
messages=messages,
|
||
stream=True
|
||
)
|
||
|
||
# 异步处理流式输出
|
||
for chunk in stream:
|
||
if chunk.choices and chunk.choices[0].delta.content is not None:
|
||
content = chunk.choices[0].delta.content
|
||
yield content
|
||
|
||
except Exception as e:
|
||
logger.error(f"流式处理图片时发生错误: {str(e)}")
|
||
yield ""
|
||
|