Fix Nuclio function invocations when deployed via the Helm chart

The `CVAT_NUCLIO_FUNCTION_NAMESPACE` needs to be defined consistently in order for Nuclio integration to work. Currently, it's set to `cvat` for the main CVAT server process, but not for any other CVAT process (which means it defaults to `nuclio` in those processes). Since it's the annotation worker process that actually invokes the Nuclio functions, the invocation fails. Fix it by synchronizing the Nuclio environment variables across all backend deployments. Technically, I think only the server and annotation worker deployments need these variables, but since they're accessed by `cvat/settings/base.py` in every process that loads Django, define them everywhere to be sure.
cvat-ai · Mar 23, 2023 · 945f358 · 945f358
1 parent 97461ea
commit 945f358
Show file tree

Hide file tree

Showing 7 changed files with 18 additions and 32 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -27,6 +27,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
   (<https://github.com/opencv/cvat/pull/5873>)
 - Width and height in CVAT dataset format mask annotations (<https://github.com/opencv/cvat/pull/5905>)
 - Empty list of export formats for a project without tasks (<https://github.com/opencv/cvat/pull/5899>)
+- Nuclio function invocations when deployed via the Helm chart
+  (<https://github.com/opencv/cvat/issues/5626>)
 
 ### Security
 - TDB

diff --git a/helm-chart/templates/_helpers.tpl b/helm-chart/templates/_helpers.tpl
@@ -60,3 +60,14 @@ Create the name of the service account to use
 {{- default "default" .Values.serviceAccount.name }}
 {{- end }}
 {{- end }}
+
+{{- define "cvat.nuclioEnv" }}
+{{- if .Values.nuclio.enabled }}
+- name: CVAT_SERVERLESS
+  value: "1"
+- name: CVAT_NUCLIO_HOST
+  value: "{{ .Release.Name }}-nuclio-dashboard"
+- name: CVAT_NUCLIO_FUNCTION_NAMESPACE
+  value: "{{ .Release.Namespace }}"
+{{- end }}
+{{- end }}
diff --git a/helm-chart/templates/cvat_backend/server/deployment.yml b/helm-chart/templates/cvat_backend/server/deployment.yml
@@ -97,14 +97,7 @@ spec:
               secretKeyRef:
                 name: "{{ tpl (.Values.postgresql.secret.name) . }}"
                 key: password
-          {{- if .Values.nuclio.enabled }}
-          - name: CVAT_SERVERLESS
-            value: "1"
-          - name: CVAT_NUCLIO_HOST
-            value: "{{ .Release.Name }}-nuclio-dashboard"
-          - name: CVAT_NUCLIO_FUNCTION_NAMESPACE
-            value: "{{ .Release.Namespace }}"
-          {{- end }}
+          {{ include "cvat.nuclioEnv" . | indent 10 }}
           {{- if .Values.analytics.enabled}}
           - name: CVAT_ANALYTICS
             value: "1"

diff --git a/helm-chart/templates/cvat_backend/utils/deployment.yml b/helm-chart/templates/cvat_backend/utils/deployment.yml
@@ -91,12 +91,7 @@ spec:
               secretKeyRef:
                 name: "{{ tpl (.Values.postgresql.secret.name) . }}"
                 key: password
-          {{- if .Values.nuclio }}
-          - name: CVAT_SERVERLESS
-            value: "1"
-          - name: CVAT_NUCLIO_HOST
-            value: "{{ .Release.Name }}-nuclio-dashboard"
-          {{- end }}
+          {{ include "cvat.nuclioEnv" . | indent 10 }}
           {{- if .Values.analytics.enabled}}
           - name: CVAT_ANALYTICS
             value: "1"

diff --git a/helm-chart/templates/cvat_backend/worker_annotation/deployment.yml b/helm-chart/templates/cvat_backend/worker_annotation/deployment.yml
@@ -91,12 +91,7 @@ spec:
               secretKeyRef:
                 name: "{{ tpl (.Values.postgresql.secret.name) . }}"
                 key: password
-          {{- if .Values.nuclio }}
-          - name: CVAT_SERVERLESS
-            value: "1"
-          - name: CVAT_NUCLIO_HOST
-            value: "{{ .Release.Name }}-nuclio-dashboard"
-          {{- end }}
+          {{ include "cvat.nuclioEnv" . | indent 10 }}
           {{- if .Values.analytics.enabled}}
           - name: CVAT_ANALYTICS
             value: "1"

diff --git a/helm-chart/templates/cvat_backend/worker_export/deployment.yml b/helm-chart/templates/cvat_backend/worker_export/deployment.yml
@@ -91,12 +91,7 @@ spec:
               secretKeyRef:
                 name: "{{ tpl (.Values.postgresql.secret.name) . }}"
                 key: password
-          {{- if .Values.nuclio }}
-          - name: CVAT_SERVERLESS
-            value: "1"
-          - name: CVAT_NUCLIO_HOST
-            value: "{{ .Release.Name }}-nuclio-dashboard"
-          {{- end }}
+          {{ include "cvat.nuclioEnv" . | indent 10 }}
           {{- if .Values.analytics.enabled}}
           - name: CVAT_ANALYTICS
             value: "1"

diff --git a/helm-chart/templates/cvat_backend/worker_import/deployment.yml b/helm-chart/templates/cvat_backend/worker_import/deployment.yml
@@ -91,12 +91,7 @@ spec:
               secretKeyRef:
                 name: "{{ tpl (.Values.postgresql.secret.name) . }}"
                 key: password
-          {{- if .Values.nuclio }}
-          - name: CVAT_SERVERLESS
-            value: "1"
-          - name: CVAT_NUCLIO_HOST
-            value: "{{ .Release.Name }}-nuclio-dashboard"
-          {{- end }}
+          {{ include "cvat.nuclioEnv" . | indent 10 }}
           {{- if .Values.analytics.enabled}}
           - name: CVAT_ANALYTICS
             value: "1"