From dee2528ae370c52ff87f5d7f7858d62d60220ba4 Mon Sep 17 00:00:00 2001
From: Lun-Kai Hsu <lunkai@google.com>
Date: Wed, 6 Jun 2018 15:25:26 -0700
Subject: [PATCH] add ambassador route for katib (#939)

---
 kubeflow/katib/modeldb.libsonnet      | 14 ++++++++++++++
 kubeflow/katib/prototypes/all.jsonnet |  8 +++++---
 2 files changed, 19 insertions(+), 3 deletions(-)

diff --git a/kubeflow/katib/modeldb.libsonnet b/kubeflow/katib/modeldb.libsonnet
index 9b76d02297d..e554f8e2678 100644
--- a/kubeflow/katib/modeldb.libsonnet
+++ b/kubeflow/katib/modeldb.libsonnet
@@ -154,6 +154,20 @@
         },
         name: "modeldb-frontend",
         namespace: namespace,
+        annotations: {
+          "getambassador.io/config":
+            std.join("\n", [
+              "---",
+              "apiVersion: ambassador/v0",
+              "kind:  Mapping",
+              "name: modeldb-mapping",
+              "prefix: /modeldb/",
+              "rewrite: /",
+              "method: GET",
+              "service: " + "modeldb-frontend." + namespace + ":3000",
+            ]),
+        },  //annotations
+
       },
       spec: {
         ports: [
diff --git a/kubeflow/katib/prototypes/all.jsonnet b/kubeflow/katib/prototypes/all.jsonnet
index 05dca809209..83ec2250b4a 100644
--- a/kubeflow/katib/prototypes/all.jsonnet
+++ b/kubeflow/katib/prototypes/all.jsonnet
@@ -19,6 +19,8 @@ local suggestion = import "kubeflow/katib/suggestion.libsonnet";
 
 local namespace = env.namespace;
 
-std.prune(k.core.v1.list.new(vizier.all(params, namespace)))
-+ std.prune(k.core.v1.list.new(modeldb.all(params, namespace)))
-+ std.prune(k.core.v1.list.new(suggestion.all(params, namespace)))
+std.prune(
+  k.core.v1.list.new(vizier.all(params, namespace))
+  + k.core.v1.list.new(suggestion.all(params, namespace))
+  + k.core.v1.list.new(modeldb.all(params, namespace))
+)