[#263] add passing model_id and model_version as headers

legion-platform · Dec 27, 2018 · af816ec · af816ec
1 parent 320e389
commit af816ec
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/legion/legion/serving/pyserve.py b/legion/legion/serving/pyserve.py
@@ -122,7 +122,10 @@ def model_invoke(model_id, model_version, endpoint='default'):
 
     output = model.endpoints[endpoint].invoke(input_dict)
     response = legion.http.prepare_response(output)
+    response.headers[legion.containers.headers.MODEL_ID] = model_id
+    response.headers[legion.containers.headers.MODEL_VERSION] = model_version
     response.headers[legion.containers.headers.MODEL_ENDPOINT] = endpoint
+
     return response