mirror of
https://github.com/Kakune55/PyGetGPT.git
synced 2025-06-29 00:08:04 +08:00
102 lines
2.8 KiB
Python
102 lines
2.8 KiB
Python
import flask , requests , json , pymysql
|
|
|
|
# 设置请求的目标URL
|
|
url = "https://dashscope.aliyuncs.com/api/v1/services/aigc/text-generation/generation" # 替换为你的API端点URL
|
|
|
|
header = {
|
|
"Content-Type":"application/json",
|
|
"Authorization":"Bearer sk-69129a5d7fc6468a9f6f30d6935254c6"
|
|
|
|
}
|
|
|
|
def request_TY(userkey,prompt,history = ""):
|
|
# 设置请求数据
|
|
if history == "":
|
|
data = {
|
|
"model": "qwen-turbo",
|
|
"input":{
|
|
"prompt":f"{prompt}"
|
|
}
|
|
}
|
|
# 发送POST请求
|
|
response = json.loads(requests.post(url, json=data ,headers=header).text)
|
|
print(response)
|
|
|
|
out = {
|
|
"code":200,
|
|
"output":f"{response['output']['text']}"
|
|
}
|
|
reduce_value(userkey, response["usage"]["total_tokens"])
|
|
return out
|
|
|
|
def userSurplus(userkey):
|
|
#打开数据库连接
|
|
db = pymysql.connect(host='www.kakuweb.top',
|
|
port=33306,
|
|
user='root',
|
|
password='KAku3.14159..',
|
|
database='AIapi')
|
|
# 使用 cursor() 方法创建一个游标对象 cursor
|
|
cursor = db.cursor()
|
|
|
|
# 使用 execute() 方法执行 SQL 查询
|
|
cursor.execute(f"SELECT surplus FROM usersurplus WHERE userkey = '{userkey}';")
|
|
# 使用 fetchone() 方法获取单条数据.
|
|
data = cursor.fetchone()
|
|
if data != None:
|
|
return data[0]
|
|
return -99999
|
|
|
|
def reduce_value(userkey, value):
|
|
#打开数据库连接
|
|
db = pymysql.connect(host='www.kakuweb.top',
|
|
port=33306,
|
|
user='root',
|
|
password='KAku3.14159..',
|
|
database='AIapi')
|
|
# 使用 cursor() 方法创建一个游标对象 cursor
|
|
cursor = db.cursor()
|
|
|
|
# 执行 SQL 查询以获取当前值
|
|
cursor.execute(f"SELECT surplus FROM usersurplus WHERE userkey = '{userkey}';")
|
|
current_value = cursor.fetchone()[0]
|
|
|
|
# 如果没有找到用户，则返回错误信息
|
|
if current_value is None:
|
|
db.close()
|
|
return -1
|
|
|
|
# 计算新的值
|
|
new_value = current_value - value
|
|
|
|
# 更新数据库中的值
|
|
cursor.execute(f"UPDATE usersurplus SET surplus={new_value} WHERE userkey='{userkey}'")
|
|
|
|
# 提交事务
|
|
db.commit()
|
|
|
|
# 关闭连接
|
|
db.close()
|
|
|
|
# 返回新值
|
|
return 0
|
|
|
|
|
|
|
|
|
|
app = flask.Flask(__name__)
|
|
|
|
@app.route('/api/user', methods=['POST'])
|
|
def post_data():
|
|
userrequest = flask.request.json
|
|
surplus = userSurplus(userrequest['userkey'])
|
|
if surplus > 0:
|
|
return request_TY(userrequest['userkey'],userrequest['prompt'])
|
|
elif surplus == -99999:
|
|
return {"code":403,"output":"No User"}
|
|
elif surplus <= 0:
|
|
return {"code":403,"output":"No prompt Tokens"}
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run(debug=True) |