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)