30 lines
834 B
Python
30 lines
834 B
Python
from flask import Flask, request
|
||
from openai import OpenAI
|
||
from waitress import serve
|
||
|
||
app = Flask(__name__)
|
||
client = OpenAI(base_url="https://open.bigmodel.cn/api/paas/v4", 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 ''
|
||
print(f'input: {_input}')
|
||
# app.logger.info(f'input: {input}')
|
||
completion = client.chat.completions.create(
|
||
model="glm-4-flash",
|
||
messages=[
|
||
{"role": "user", "content": _input},
|
||
],
|
||
stream=False
|
||
)
|
||
_output = completion.choices[0].message.content
|
||
# app.logger.info(f'output: {output}')
|
||
return _output
|
||
|
||
|
||
serve(app, host='0.0.0.0', port=6721)
|