From 25f2d6d78cbe5a8fa2fad1d2c2afb4f12ae6bc72 Mon Sep 17 00:00:00 2001
From: Michele Zanotti <m.zanotti@nebuly.ai>
Date: Tue, 22 Oct 2024 14:56:14 +0200
Subject: [PATCH] feat: update helm values templates

---
 main.tf                        |  4 +--
 templates/helm-values.tpl.yaml | 45 ++++++++++++++++------------------
 2 files changed, 23 insertions(+), 26 deletions(-)

diff --git a/main.tf b/main.tf
index 0c1e603..d06d3f7 100644
--- a/main.tf
+++ b/main.tf
@@ -435,8 +435,8 @@ locals {
       platform_domain        = var.platform_domain
       image_pull_secret_name = var.k8s_image_pull_secret_name
 
-      openai_endpoint               = var.openai_endpoint
-      openai_frustration_deployment = var.openai_gpt4_deployment_name
+      openai_endpoint         = var.openai_endpoint
+      openai_gpt4o_deployment = var.openai_gpt4_deployment_name
 
       secret_provider_class_name        = local.secret_provider_class_name
       secret_provider_class_secret_name = local.secret_provider_class_secret_name
diff --git a/templates/helm-values.tpl.yaml b/templates/helm-values.tpl.yaml
index dd67218..db733f6 100644
--- a/templates/helm-values.tpl.yaml
+++ b/templates/helm-values.tpl.yaml
@@ -141,29 +141,7 @@ ingestionWorker:
         volumeAttributes:
           secretProviderClass: ${secret_provider_class_name}
 
-  topicsClustering:
-    affinity:
-      nodeAffinity:
-        requiredDuringSchedulingIgnoredDuringExecution:
-          nodeSelectorTerms:
-            - matchExpressions:
-                - key: nebuly.com/accelerator
-                  operator: In
-                  values:
-                    - nvidia-tesla-t4
-
-  suggestionsGeneration:
-    affinity:
-      nodeAffinity:
-        requiredDuringSchedulingIgnoredDuringExecution:
-          nodeSelectorTerms:
-            - matchExpressions:
-                - key: nebuly.com/accelerator
-                  operator: In
-                  values:
-                    - nvidia-tesla-t4
-
-actionsProcessing:
+primaryProcessing:
   schedule: "0 23 * * *" # Every day at 11:00 PM, cluster's timezone
   resources:
     requests:
@@ -186,6 +164,25 @@ actionsProcessing:
       key: nvidia.com/gpu
       operator: Exists
 
+secondaryProcessing:
+  schedule: "0 2 * * *" # Every day at 2:00 AM, cluster's timezone
+
+  affinity:
+    nodeAffinity:
+      requiredDuringSchedulingIgnoredDuringExecution:
+        nodeSelectorTerms:
+          - matchExpressions:
+              - key: nebuly.com/accelerator
+                operator: In
+                values:
+                  - nvidia-l4
+
+  tolerations:
+    - effect: NoSchedule
+      key: nvidia.com/gpu
+      operator: Exists
+
+
 kafka:
   external: false
   zookeeper:
@@ -291,7 +288,7 @@ frontend:
 
 openAi:
   enabled: true
-  frustrationDetectionDeployment: ${openai_frustration_deployment}
+  gpt4oDeployment: ${openai_gpt4o_deployment}
   endpoint: ${openai_endpoint}
 
   existingSecret: