From 928f84cb1e2ee64942375a6b0f9cac1c8badf98f Mon Sep 17 00:00:00 2001 From: "majiahui@haimaqingfan.com" Date: Tue, 31 Oct 2023 15:49:23 +0800 Subject: [PATCH] =?UTF-8?q?=E7=AC=AC=E4=B8=80=E4=B8=AA=E5=A2=9E=E5=BC=BA?= =?UTF-8?q?=E7=89=88=E4=B8=8A=E7=BA=BF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- redis_check_uuid_mistral.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/redis_check_uuid_mistral.py b/redis_check_uuid_mistral.py index 8d864e8..babee0a 100644 --- a/redis_check_uuid_mistral.py +++ b/redis_check_uuid_mistral.py @@ -33,6 +33,7 @@ redis_ = redis.Redis(connection_pool=pool, decode_responses=True) db_key_query = 'query' db_key_querying = 'querying' +db_key_error = 'error' @app.route("/search", methods=["POST"]) def handle_query(): @@ -47,6 +48,9 @@ def handle_query(): code = result_dict["status_code"] texts = result_dict["texts"] 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} else: querying_list = list(redis_.smembers("querying")) @@ -85,4 +89,4 @@ def handle_query(): 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)