http请求示例

def query_tf_serving(url, data, request_time, num=1):
"""
请求模型服务接口
Args:
url: 请求url
data: 请求数据，image base64字符串
request_time: 请求时间
num: 请求次数，最多重试5次

Returns:
    分类模型预测值

"""
logger.info(f"The {num} query to tensorflow serving. URL: {url}. request_time: {str(request_time)}")

if num >= 5:
    raise Exception(f"Query tensorflow serving retry over 5 times.")

header = {
    'user-agent': logTag.x_call_id,
    "Content-Type": "application/json",
    "X-Call-Id": logTag.x_call_id,
    "Request-Time": str(request_time)
}

try:
    res = requests.post(url=url, data=data, headers=header, timeout=TF_SERVING_TIMEOUT)

    if res.status_code != 200:
        num += 1
        logger.error(f"Query tensorflow serving failed. status_code: {res.status_code}")
        sleep(RETRY_SLEEP_TIME)
        return query_tf_serving(url, data, request_time, num)

    return json.loads(res.text)['predictions']
except Exception as e:
    num += 1
    logger.error(f"Query tensorflow serving failed. error_message: {e}")
    sleep(RETRY_SLEEP_TIME)
    return query_tf_serving(url, data, request_time, num)