fix: agent helm chart

Author: Code2Life

charts/tensor-fusion/templates/controller-deployment.yaml

@@ -45,6 +45,19 @@ spec:
             - name: cert
               readOnly: true
               mountPath: /tmp/k8s-webhook-server/serving-certs
+        {{- if .Values.agentId }}
+        - name: cluster-agent
+          image: "{{ .Values.agent.image.repository }}:{{ .Values.agent.image.tag | default "latest" }}"
+          env:
+            - name: CLOUD_ENDPOINT
+              value: "{{ .Values.agent.cloudEndpoint }}"
+            - name: AGENT_ID
+              value: "{{ .Values.agent.agentId }}"
+            - name: ENROLL_TOKEN
+              value: "{{ .Values.agent.enrollToken }}"
+          resources:
+            {{- toYaml .Values.agent.resources | nindent 12 }}
+        {{- end }}
         - name: vector
           image: docker.io/timberio/vector:nightly-2025-01-07-debian
           env:

charts/tensor-fusion/templates/gpupool-sample.yaml

@@ -57,3 +57,21 @@ controller:
     secretName: tensor-fusion-webhook-secret
     patch:
       image: registry.k8s.io/ingress-nginx/kube-webhook-certgen:v1.5.0
+
+
+agent:
+  enrollToken: "token-from-cloud"
+  agentId: 'org-from-cloud:env'
+  cloudEndpoint: "wss://api.tensor-fusion.ai/_ws"
+  
+  image:
+    repository: tensorfusion/tensor-fusion-agent
+    tag: "latest"
+  
+  resources:
+    requests:
+      cpu: 50m
+      memory: 64Mi
+    limits:
+      cpu: 1000m
+      memory: 512Mi