Browse Source

第一个增强版上线

master
majiahui@haimaqingfan.com 2 years ago
parent
commit
928f84cb1e
  1. 6
      redis_check_uuid_mistral.py

6
redis_check_uuid_mistral.py

@ -33,6 +33,7 @@ redis_ = redis.Redis(connection_pool=pool, decode_responses=True)
db_key_query = 'query' db_key_query = 'query'
db_key_querying = 'querying' db_key_querying = 'querying'
db_key_error = 'error'
@app.route("/search", methods=["POST"]) @app.route("/search", methods=["POST"])
def handle_query(): def handle_query():
@ -47,6 +48,9 @@ def handle_query():
code = result_dict["status_code"] code = result_dict["status_code"]
texts = result_dict["texts"] texts = result_dict["texts"]
probabilities = result_dict["probabilities"] probabilities = result_dict["probabilities"]
if str(code) == 400:
redis_.rpush(db_key_error, json.dumps({"id": id_}))
return False
result_text = {'code': code, 'text': texts, 'probabilities': probabilities} result_text = {'code': code, 'text': texts, 'probabilities': probabilities}
else: else:
querying_list = list(redis_.smembers("querying")) querying_list = list(redis_.smembers("querying"))
@ -85,4 +89,4 @@ def handle_query():
if __name__ == "__main__": if __name__ == "__main__":
app.run(debug=False, host='0.0.0.0', port=14003) app.run(debug=False, host='0.0.0.0', port=14005)

Loading…
Cancel
Save