Merge pull request #405 from jklaise/api-tester-grpc-ambassador

Fix api-tester not working via GRPC and Ambassador
SeldonIO · Jan 24, 2019 · 224e22c · 224e22c
2 parents 04862f3 + 7ed6ab6
commit 224e22c
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/python/seldon_core/api_tester.py b/python/seldon_core/api_tester.py
@@ -297,7 +297,15 @@ def run_predict(args):
                 response = stub.Predict(
                     request=GRPC_request, metadata=metadata)
             else:
-                response = stub.Predict(request=GRPC_request)
+                # ambassador path
+                split = args.ambassador_path.split("/")
+                try:
+                    _, seldon, namespace, deploymentName = split
+                    metadata = [('seldon', deploymentName), ('namespace', namespace)]
+                except ValueError:
+                    _, seldon, deploymentName = split
+                    metadata = [('seldon', deploymentName)]
+                response = stub.Predict(request=GRPC_request, metadata=metadata)
 
             if args.prnt:
                 print("RECEIVED RESPONSE:")