forked from Gitlink/gitlink_help_center
33 lines
999 B
Python
33 lines
999 B
Python
import os
|
|
import json
|
|
import requests
|
|
|
|
from pdfminer.high_level import extract_text
|
|
import re
|
|
def send_pdf_and_get_response(message):
|
|
# Moonshot API 密钥
|
|
MOONSHOT_API_KEY = "fea58b082997a69f199264ce8221430a.zC0hu8lce49CNBPJ"
|
|
|
|
# 构造请求的 headers
|
|
headers = {
|
|
'Content-Type': 'application/json',
|
|
'Authorization': f'Bearer {MOONSHOT_API_KEY}',
|
|
}
|
|
# 构造请求数据
|
|
data = {
|
|
"model": "glm-4", # Moonshot 模型
|
|
"messages": [
|
|
{"role": "user", "content":message}
|
|
],
|
|
"temperature": 0.01, # 温度参数，控制助手回复的多样性
|
|
}
|
|
|
|
# 发送 POST 请求
|
|
response = requests.post('https://open.bigmodel.cn/api/paas/v4/chat/completions', headers=headers, data=json.dumps(data))
|
|
response_json = response.json()
|
|
|
|
# 提取助手的回复
|
|
assistant_response = response_json#['choices'][0]['message']['content']
|
|
|
|
return assistant_response
|
|
print(send_pdf_and_get_response(1)) |