from fastapi import FastAPI, Request
import uvicorn, json, datetime
import requests
import sbert
import sbert_max_picture
import time
from flask import Flask, request, jsonify
app = Flask(__name__)
@app.route('/smart_companion', methods=['POST'])
def smart_companion_api():
# print('my_api')
# 获取 POST 请求的 JSON 数据
data = request.get_json()
cus_q = data["cus_q"]
session = data["session"]
# if session[-1] in ('a','b'): #灰度10%
result = sbert.main_sbert(cus_q, session)
# answer = chat_chain.invoke({'query':question}).content
# else:
# result = sbert_max_picture.main(cus_q, session)
# end_time = time.time()
# print("total_time:",end_time - start_time)
# 返回结果
return jsonify(result)
if __name__ == "__main__":
# uvicorn.run(app='main:app', host='0.0.0.0', port=5000, workers=1, log_level="info")
app.run('0.0.0.0', port=5000, threaded=True)#,,threaded=False,processes=20,port=400
threaded=True是开启了多线程,具体线程数量和执行由nohup gunicorn --threads 8 -b 0.0.0.0:5000 main:app &
执行