util.py

import requests

def get_client(host='127.0.0.1', port=8000):
    """Return a client wrapper for testing in same model_server 3.9 environment as server"""
    uri = f'http://{host}:{port}'

    def hit_endpoint(method, endpoint, params=None):
        if method == 'GET':
            resp = requests.get(uri + endpoint)
        elif method == 'PUT':
            resp = requests.put(uri + endpoint, params=params)
        else:
            raise Exception('Can only handle GET and PUT requests')
        if resp.status_code != 200:
            return {'status': resp.status_code, 'content': resp.text}
        else:
            return {'status': resp.status_code, 'content': resp.json()}
    return hit_endpoint