From 8001876359477d365547e5d8b8e17eec77495002 Mon Sep 17 00:00:00 2001 From: deng Date: Sun, 26 Feb 2023 05:10:28 +0800 Subject: [PATCH] test rest api --- get_registered_model_via_rest_api.py | 30 ++++++++++++++++++++++++++++ start_mlflow_server.sh | 0 2 files changed, 30 insertions(+) create mode 100644 get_registered_model_via_rest_api.py mode change 100644 => 100755 start_mlflow_server.sh diff --git a/get_registered_model_via_rest_api.py b/get_registered_model_via_rest_api.py new file mode 100644 index 0000000..c5f6664 --- /dev/null +++ b/get_registered_model_via_rest_api.py @@ -0,0 +1,30 @@ +# get_registered_model_via_rest_api.py +# +# author: deng +# date : 20230224 + +import json +import requests + +def main(): + + registered_model_name = 'fortune_predict_model' + production_model_version = None + + query = {'name': registered_model_name} + res = requests.get('http://127.0.0.1:5000/api/2.0/mlflow/registered-models/get', params=query) + content = json.loads(res.text) + print(content) + + for model in content['registered_model']['latest_versions']: + + if model['current_stage'] == 'Production': + production_model_version = model['version'] + + if production_model_version is not None: + query = {'name': registered_model_name, 'version': production_model_version} + res = requests.get('http://127.0.0.1:5000/api/2.0/mlflow/model-versions/get-download-uri', params=query) + print(res.text) + +if __name__ == '__main__': + main() \ No newline at end of file diff --git a/start_mlflow_server.sh b/start_mlflow_server.sh old mode 100644 new mode 100755