From c754d9af7af3aec1c6088ac8fe22682ea126dbc6 Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin.bharambe@gmail.com>
Date: Mon, 2 Jun 2025 16:31:24 -0700
Subject: [PATCH] fixes

---
 .../k8s/ingress-k8s.yaml.template             |  7 ++++-
 .../distributions/k8s/stack-k8s.yaml.template | 15 ++--------
 .../distributions/k8s/ui-k8s.yaml.template    | 29 ++++++-------------
 3 files changed, 17 insertions(+), 34 deletions(-)

diff --git a/docs/source/distributions/k8s/ingress-k8s.yaml.template b/docs/source/distributions/k8s/ingress-k8s.yaml.template
index 1cd95b5e5..9ebe86b69 100644
--- a/docs/source/distributions/k8s/ingress-k8s.yaml.template
+++ b/docs/source/distributions/k8s/ingress-k8s.yaml.template
@@ -7,6 +7,11 @@ spec:
   selector:
     app.kubernetes.io/name: llama-stack
   ports:
-    - port: 8321
+    - name: llama-stack-api
+      port: 8321
       targetPort: 8321
       protocol: TCP
+    - name: llama-stack-ui
+      port: 8322
+      targetPort: 8322
+      protocol: TCP
diff --git a/docs/source/distributions/k8s/stack-k8s.yaml.template b/docs/source/distributions/k8s/stack-k8s.yaml.template
index 1e51e611b..1cfc63ef5 100644
--- a/docs/source/distributions/k8s/stack-k8s.yaml.template
+++ b/docs/source/distributions/k8s/stack-k8s.yaml.template
@@ -18,10 +18,12 @@ spec:
   selector:
     matchLabels:
       app.kubernetes.io/name: llama-stack
+      app.kubernetes.io/component: server
   template:
     metadata:
       labels:
         app.kubernetes.io/name: llama-stack
+        app.kubernetes.io/component: server
     spec:
       containers:
       - name: llama-stack
@@ -65,16 +67,3 @@ spec:
       - name: llama-config
         configMap:
           name: llama-stack-config
----
-apiVersion: v1
-kind: Service
-metadata:
-  name: llama-stack-service
-spec:
-  selector:
-    app.kubernetes.io/name: llama-stack
-  ports:
-  - protocol: TCP
-    port: 8321
-    targetPort: 8321
-  type: ClusterIP
diff --git a/docs/source/distributions/k8s/ui-k8s.yaml.template b/docs/source/distributions/k8s/ui-k8s.yaml.template
index ca0d6050d..79a094d42 100644
--- a/docs/source/distributions/k8s/ui-k8s.yaml.template
+++ b/docs/source/distributions/k8s/ui-k8s.yaml.template
@@ -3,16 +3,19 @@ kind: Deployment
 metadata:
   name: llama-stack-ui
   labels:
-    app: llama-stack-ui
+    app.kubernetes.io/name: llama-stack
+    app.kubernetes.io/component: ui
 spec:
   replicas: 1
   selector:
     matchLabels:
-      app: llama-stack-ui
+      app.kubernetes.io/name: llama-stack
+      app.kubernetes.io/component: ui
   template:
     metadata:
       labels:
-        app: llama-stack-ui
+        app.kubernetes.io/name: llama-stack
+        app.kubernetes.io/component: ui
     spec:
       containers:
       - name: llama-stack-ui
@@ -21,6 +24,8 @@ spec:
         env:
         - name: NEXT_PUBLIC_LLAMA_STACK_BASE_URL
           value: ${NEXT_PUBLIC_LLAMA_STACK_BASE_URL}
+        - name: LLAMA_STACK_UI_PORT
+          value: "8322"
         args:
           - -c
           - |
@@ -51,23 +56,7 @@ spec:
             npx next --version || echo "Next.js not found, checking node_modules..."
             ls -la node_modules/.bin/ | grep next || echo "No next binary found"
 
-            npm run dev --port 8322
+            npm run dev
         ports:
         - containerPort: 8322
         workingDir: /app
-
----
-apiVersion: v1
-kind: Service
-metadata:
-  name: llama-stack-ui-service
-  labels:
-    app: llama-stack-ui
-spec:
-  selector:
-    app: llama-stack-ui
-  ports:
-  - port: 8322
-    targetPort: 8322
-    protocol: TCP
-  type: ClusterIP