add agents (#241)

* add agents

* templatize temporal workers

* refactor temporal out of workflows

* better interface

* fmt

* test agents enabled option

* agents enables workflows

* use temporalConfig helper

* fix label handling

* re-refactor workers

* fmt

* allow overriding worker affinity

* add usage note

* bump version

Author: lukeviens

charts/retool/Chart.yaml

@@ -2,7 +2,7 @@ apiVersion: v2
 name: retool
 description: A Helm chart for Kubernetes
 type: application
-version: 6.5.0
+version: 6.6.0
 maintainers:
   - name: Retool Engineering
     email: [email protected]

charts/retool/ci/test-agents-enabled-option.yaml

@@ -0,0 +1,2 @@
+agents:
+  enabled: true

charts/retool/templates/_helpers.tpl

@@ -125,6 +125,39 @@ app.kubernetes.io/instance: {{ .Release.Name }}
 telemetry.retool.com/service-name: code-executor
 {{- end }}
 
+{{/*
+Selector labels for agent worker. Note changes here will require manual
+deployment recreation and incur downtime, so should be avoided.
+*/}}
+{{- define "retool.agentWorker.selectorLabels" -}}
+retoolService: {{ include "retool.agentWorker.name" . }}
+{{- end }}
+
+{{/*
+Extra (non-selector) labels for agent worker.
+*/}}
+{{- define "retool.agentWorker.labels" -}}
+app.kubernetes.io/name: {{ include "retool.agentWorker.name" . }}
+app.kubernetes.io/instance: {{ .Release.Name }}
+telemetry.retool.com/service-name: agent-worker
+{{- end }}
+
+{{/*
+Selector labels for agent eval worker. Note changes here will require manual
+deployment recreation and incur downtime, so should be avoided.
+*/}}
+{{- define "retool.agentEvalWorker.selectorLabels" -}}
+retoolService: {{ include "retool.agentEvalWorker.name" . }}
+{{- end }}
+
+{{/*
+Extra (non-selector) labels for agent eval worker.
+*/}}
+{{- define "retool.agentEvalWorker.labels" -}}
+app.kubernetes.io/name: {{ include "retool.agentEvalWorker.name" . }}
+app.kubernetes.io/instance: {{ .Release.Name }}
+telemetry.retool.com/service-name: agent-eval-worker
+{{- end }}
 
 {{/*
 Create the name of the service account to use
@@ -237,6 +270,9 @@ Usage: (include "retool.workflows.enabled" .)
 {{- else -}}
   {{- $output = "" -}}
 {{- end -}}
+{{- if (eq (toString .Values.agents.enabled) "true") -}} {{/* workflows (backend) is required to use agents */}}
+  {{- $output = "1" -}}
+{{- end -}}
 {{- $output -}}
 {{- end -}}
 
@@ -269,13 +305,30 @@ Usage: (include "retool.codeExecutor.enabled" .)
 {{- $output -}}
 {{- end -}}
 
+{{/*
+Set agents enabled
+Usage: (include "retool.agents.enabled" .)
+*/}}
+{{- define "retool.agents.enabled" -}}
+{{- $output := "" -}}
+{{- if (eq (toString .Values.agents.enabled) "true") -}}
+  {{- $output = "1" -}}
+{{- end -}}
+{{- $output -}}
+{{- end -}}
+
+{{/* Global Temporal configuration */}}
+{{- define "retool.temporalConfig" -}}
+{{- .Values.workflows.temporal | default .Values.temporal | toYaml -}}
+{{- end -}}
 
 {{/*
 Set Temporal frontend host
 */}}
 {{- define "retool.temporal.host" -}}
-{{- if (.Values.workflows.temporal).enabled -}}
-{{- .Values.workflows.temporal.host | quote -}}
+{{- $temporalConfig := include "retool.temporalConfig" . | fromYaml -}}
+{{- if $temporalConfig.enabled -}}
+{{- $temporalConfig.host | quote -}}
 {{- else -}}
 {{- printf "%s-%s" (include "temporal.fullname" (index .Subcharts "retool-temporal-services-helm")) "frontend" -}}
 {{- end -}}
@@ -285,8 +338,9 @@ Set Temporal frontend host
 Set Temporal frontend port
 */}}
 {{- define "retool.temporal.port" -}}
-{{- if (.Values.workflows.temporal).enabled -}}
-{{- .Values.workflows.temporal.port | quote -}}
+{{- $temporalConfig := include "retool.temporalConfig" . | fromYaml -}}
+{{- if $temporalConfig.enabled -}}
+{{- $temporalConfig.port | quote -}}
 {{- else -}}
 {{- "7233" | quote -}}
 {{- end -}}
@@ -296,8 +350,9 @@ Set Temporal frontend port
 Set Temporal namespace
 */}}
 {{- define "retool.temporal.namespace" -}}
-{{- if (.Values.workflows.temporal).enabled -}}
-{{- .Values.workflows.temporal.namespace | quote -}}
+{{- $temporalConfig := include "retool.temporalConfig" . | fromYaml -}}
+{{- if $temporalConfig.enabled -}}
+{{- $temporalConfig.namespace | quote -}}
 {{- else -}}
 {{- "workflows" | quote -}}
 {{- end -}}
@@ -338,6 +393,19 @@ Set multiplayer service name
 {{ template "retool.fullname" . }}-multiplayer-ws
 {{- end -}}
 
+{{/*
+Set agent worker service name
+*/}}
+{{- define "retool.agentWorker.name" -}}
+{{ template "retool.fullname" . }}-agent-worker
+{{- end -}}
+
+{{/*
+Set agent eval worker service name
+*/}}
+{{- define "retool.agentEvalWorker.name" -}}
+{{ template "retool.fullname" . }}-agent-eval-worker
+{{- end -}}
 
 {{/*
 Set code executor image tag