Add plugin_config for agent (#4848)

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Feb 6, 2024 · a087802 · a087802
1 parent 3180b23
commit a087802
Show file tree

Hide file tree

Showing 6 changed files with 18 additions and 22 deletions.
diff --git a/charts/flyte-core/README.md b/charts/flyte-core/README.md
@@ -189,6 +189,8 @@ helm install gateway bitnami/contour -n flyte
 | flyteadmin.serviceMonitor.scrapeTimeout | string | `"30s"` | Sets the timeout after which request to scrape metrics will time out |
 | flyteadmin.tolerations | list | `[]` | tolerations for Flyteadmin deployment |
 | flyteagent.enabled | bool | `false` |  |
+| flyteagent.plugin_config.plugins.agentService.defaultAgent.endpoint | string | `"dns:///flyteagent.flyte.svc.cluster.local:8000"` |  |
+| flyteagent.plugin_config.plugins.agentService.defaultAgent.insecure | bool | `true` |  |
 | flyteconsole.affinity | object | `{}` | affinity for Flyteconsole deployment |
 | flyteconsole.enabled | bool | `true` |  |
 | flyteconsole.ga.enabled | bool | `false` |  |

diff --git a/charts/flyte-core/templates/propeller/configmap.yaml b/charts/flyte-core/templates/propeller/configmap.yaml
@@ -55,3 +55,8 @@ data:
   task_logs.yaml: | {{ tpl (toYaml .) $ | nindent 4 }}
 {{- end }}
 {{- end }}
+{{- if .Values.flyteagent.enabled }}
+{{- with .Values.flyteagent.plugin_config }}
+  agent_service.yaml: | {{ tpl (toYaml .) $ | nindent 4 }}
+{{- end }}
+{{- end }}
diff --git a/charts/flyte-core/values.yaml b/charts/flyte-core/values.yaml
@@ -249,6 +249,12 @@ datacatalog:
 
 flyteagent:
   enabled: false
+  plugin_config:
+    plugins:
+      agentService:
+        defaultAgent:
+          endpoint: "dns:///flyteagent.flyte.svc.cluster.local:8000"
+          insecure: true
 
 #
 # FLYTEPROPELLER SETTINGS
@@ -752,11 +758,6 @@ configmap:
         #  DEFAULT_ENV_VAR: VALUE
         default-cpus: 100m
         default-memory: 100Mi
-      # Uncomment and modify to include configuration for Flyte Agent
-      # agent-service:
-      #   defaultGrpcEndpoint: flyteagent.flyte.svc.cluster.local:8000
-      #   supportedTaskTypes:
-      #     - bigquery_query_job_task
   remoteData:
     remoteData:
       region: "us-east-1"