Update Gitops template for RAG DB

templates/http/base/database-config.yaml

@@ -0,0 +1,7 @@
+apiVersion: v1
+kind: ConfigMap
+metadata:
+  name: {{values.name}}-database-config
+data:
+  VECTORDB_HOST: "{{values.name}}-database"
+  VECTORDB_PORT: "{{values.dbPort}}"

templates/http/base/deployment-database.yaml

@@ -0,0 +1,38 @@
+apiVersion: apps/v1
+kind: Deployment
+metadata:
+  labels: 
+    app.kubernetes.io/instance: {{values.name}}-database
+    app.kubernetes.io/managed-by: kustomize
+    app.kubernetes.io/name: {{values.name}}-database
+    app.kubernetes.io/part-of: {{values.name}}  
+  name: {{values.name}}-database
+spec:
+  replicas: 1
+  selector:
+    matchLabels:
+      app.kubernetes.io/instance: {{values.name}}-database
+  template:
+    metadata: 
+      labels:
+        app.kubernetes.io/instance: {{values.name}}-database
+    spec:
+      containers:
+      - image: {{values.dbContainer}}
+        name: database
+        args: [
+            "--workers",
+            "1", 
+            "--host", 
+            "0.0.0.0",
+            "--port",
+            "{{values.dbPort}}",
+            "--proxy-headers", 
+            "--log-config", 
+            "chromadb/log_config.yml", 
+            "--timeout-keep-alive", 
+            "30"]
+        ports:
+        - containerPort: {{values.dbPort}}
+        securityContext:
+          runAsNonRoot: true

templates/http/base/deployment-model-server.yaml

@@ -1,11 +1,6 @@
 apiVersion: apps/v1
 kind: Deployment
 metadata:
-  annotations:  
-    tad.gitops.set/image: ".spec.template.spec.containers[0].image"
-    tad.gitops.get/image: ".spec.template.spec.containers[0].image"
-    tad.gitops.set/replicas: ".spec.replicas"
-    tad.gitops.get/replicas: ".spec.replicas" 
   labels: 
     app.kubernetes.io/instance: {{values.name}}-model-server
     app.kubernetes.io/managed-by: kustomize

templates/http/base/deployment.yaml

@@ -28,7 +28,11 @@ spec:
         envFrom:
         - configMapRef:
             name: {{values.name}}-model-config
+        {%- if values.dbRequired %}
+        - configMapRef:
+            name: {{values.name}}-database-config
+        {%- endif %}
         ports:
         - containerPort: {{values.appPort}}
         securityContext:
-          runAsNonRoot: true
+          runAsNonRoot: true

templates/http/base/kustomization.yaml

@@ -19,6 +19,11 @@ resources:
 - route.yaml
 - service.yaml
 - model-config.yaml
+{%- if values.dbRequired %}
+- deployment-database.yaml
+- service-database.yaml
+- database-config.yaml
+{%- endif %}
 {%- if values.rhoaiSelected %}
 - rhoai/
-{%- endif %}
+{%- endif %}

templates/http/base/rhoai/initialize-dsp.yaml

@@ -97,6 +97,10 @@ spec:
                     envFrom:
                     - configMapRef:
                         name: {{values.name}}-model-config
+                    {%- if values.dbRequired %}
+                    - configMapRef:
+                        name: {{values.name}}-database-config
+                    {%- endif %}
                     ports:
                       - containerPort: 8888
                         name: notebook-port

templates/http/base/service-database.yaml

@@ -0,0 +1,15 @@
+apiVersion: v1
+kind: Service
+metadata: 
+  labels: 
+    app.kubernetes.io/instance: {{values.name}}-database
+    app.kubernetes.io/managed-by: kustomize
+    app.kubernetes.io/name: {{values.name}}-database 
+  name: {{values.name}}-database
+spec:
+  ports:
+  - port: {{values.dbPort}}
+    protocol: TCP
+    targetPort: {{values.dbPort}}
+  selector:
+    app.kubernetes.io/instance: {{values.name}}-database