51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
import json
|
||
|
||
import requests
|
||
|
||
from Config.Config import GLM_API_KEY, GLM_MODEL_NAME, GLM_BASE_URL
|
||
|
||
url = GLM_BASE_URL
|
||
headers = {
|
||
"Authorization": "Bearer " + GLM_API_KEY,
|
||
"Content-Type": "application/json"
|
||
}
|
||
data = {
|
||
"model": GLM_MODEL_NAME,
|
||
"messages": [
|
||
{
|
||
"role": "user",
|
||
# "content": "做水煮鱼一般用什么鱼"
|
||
"content":
|
||
[
|
||
{
|
||
"type": "text",
|
||
"text": "请帮我解决这个题目，给出详细过程和答案。第4问请注意，F可能在三角形内部，也可能出现在BC的外侧，即F在三角形外两种情况，需要分类讨论作答。"
|
||
},
|
||
{
|
||
"type": "image_url",
|
||
"image_url": {
|
||
"url": "https://dsideal.obs.cn-north-1.myhuaweicloud.com/HuangHai/Backup/23.jpg"
|
||
}
|
||
}
|
||
]
|
||
}
|
||
],
|
||
"stream": True # 启用流式调用
|
||
}
|
||
|
||
with requests.post(url, headers=headers, json=data, stream=True) as response:
|
||
for chunk in response.iter_lines():
|
||
if chunk:
|
||
decoded = chunk.decode('utf-8')
|
||
if decoded.startswith('[DONE]'):
|
||
print("完成！")
|
||
break
|
||
try:
|
||
decoded = decoded[5:]
|
||
json_data = json.loads(decoded)
|
||
content = json_data["choices"][0]["delta"]['content']
|
||
if content and len(content) > 0:
|
||
print(content, end="")
|
||
except Exception as e:
|
||
print(e)
|