初始化提交
This commit is contained in:
26
ai-server.py
Normal file
26
ai-server.py
Normal file
@@ -0,0 +1,26 @@
|
||||
from flask import Flask, request
|
||||
from zhipuai import ZhipuAI
|
||||
|
||||
app = Flask(__name__)
|
||||
client = ZhipuAI(api_key="d1e97306540d12bb2f834be961fcacb1.SNBShlCxWYJCx0qZ")
|
||||
|
||||
|
||||
@app.route('/ai_text', methods=['POST'])
|
||||
def ai_text():
|
||||
input = request.get_data(as_text=True)
|
||||
# 如果input值为空,则返回空字符串
|
||||
if not input:
|
||||
return ''
|
||||
completion = client.chat.completions.create(
|
||||
model="glm-4-flash",
|
||||
messages=[
|
||||
{"role": "user", "content": input},
|
||||
],
|
||||
)
|
||||
return completion.choices[0].message.content
|
||||
|
||||
|
||||
# app.run(host='0.0.0.0', port=6721)
|
||||
from waitress import serve
|
||||
|
||||
serve(app, host='0.0.0.0', port=6721)
|
||||
Reference in New Issue
Block a user