From a97a42265541482d2bdf5d7e0ffdf01b5c85a509 Mon Sep 17 00:00:00 2001
From: Future Outlier <eric901201@gmai.com>
Date: Thu, 21 Sep 2023 11:10:39 +0800
Subject: [PATCH] add task timeout

Signed-off-by: Future Outlier <eric901201@gmai.com>
---
 charts/flyte-binary/values.yaml                      | 3 +++
 docker/sandbox-bundled/manifests/complete-agent.yaml | 9 ++++++---
 docker/sandbox-bundled/manifests/complete.yaml       | 9 ++++++---
 docker/sandbox-bundled/manifests/dev.yaml            | 4 ++--
 4 files changed, 17 insertions(+), 8 deletions(-)

diff --git a/charts/flyte-binary/values.yaml b/charts/flyte-binary/values.yaml
index c893d8f3c2..d831815e5f 100644
--- a/charts/flyte-binary/values.yaml
+++ b/charts/flyte-binary/values.yaml
@@ -155,6 +155,9 @@ configuration:
     defaultAgent:
       endpoint: "dns:///flyteagent.flyte.svc.cluster.local:8000"
       insecure: true
+      timeouts:
+        GetTask: 10s
+      defaultTimeout: 10s
   #   Uncomment and modify to include configuration for Flyte Agent
   #   defaultGrpcEndpoint: agent-service.agent-namespace:8000
   # externalConfigMap Specify an existing, external ConfigMap to use as configuration for Flyte
diff --git a/docker/sandbox-bundled/manifests/complete-agent.yaml b/docker/sandbox-bundled/manifests/complete-agent.yaml
index f9f21bca90..2a593b20e2 100644
--- a/docker/sandbox-bundled/manifests/complete-agent.yaml
+++ b/docker/sandbox-bundled/manifests/complete-agent.yaml
@@ -478,8 +478,11 @@ data:
             stackdriver-enabled: false
       agent-service:
         defaultAgent:
+          defaultTimeout: 10s
           endpoint: dns:///flyteagent.flyte.svc.cluster.local:8000
           insecure: true
+          timeouts:
+            GetTask: 10s
   002-database.yaml: |
     database:
       postgres:
@@ -811,7 +814,7 @@ type: Opaque
 ---
 apiVersion: v1
 data:
-  haSharedSecret: OVMxRkQ3eTRSSFIzb1BwNw==
+  haSharedSecret: bG8wM0N5bko0eGtTeXR3VA==
   proxyPassword: ""
   proxyUsername: ""
 kind: Secret
@@ -1241,7 +1244,7 @@ spec:
     metadata:
       annotations:
         checksum/cluster-resource-templates: 6fd9b172465e3089fcc59f738b92b8dc4d8939360c19de8ee65f68b0e7422035
-        checksum/configuration: 454886716e03425f5b6980d658280a5c9d2dd9b73767ab8e080ba381b97efc95
+        checksum/configuration: d8bf73644712bd050aa462a0478b0c779bd6d170645efe98feb5ffcacf52010e
         checksum/configuration-secret: 09216ffaa3d29e14f88b1f30af580d02a2a5e014de4d750b7f275cc07ed4e914
       labels:
         app.kubernetes.io/component: flyte-binary
@@ -1404,7 +1407,7 @@ spec:
     metadata:
       annotations:
         checksum/config: 8f50e768255a87f078ba8b9879a0c174c3e045ffb46ac8723d2eedbe293c8d81
-        checksum/secret: 5bbe0cc7953f85ea716803f63f0621812aba5ca65b0cd4663650d9ee01071830
+        checksum/secret: 0e903dc801d2d56962db092f2a87ab3ac2ff7ad2c63c9010b0f4d2d7b62ee991
       labels:
         app: docker-registry
         release: flyte-sandbox
diff --git a/docker/sandbox-bundled/manifests/complete.yaml b/docker/sandbox-bundled/manifests/complete.yaml
index 1c263bfbe2..dc4fe87ffd 100644
--- a/docker/sandbox-bundled/manifests/complete.yaml
+++ b/docker/sandbox-bundled/manifests/complete.yaml
@@ -467,8 +467,11 @@ data:
             stackdriver-enabled: false
       agent-service:
         defaultAgent:
+          defaultTimeout: 10s
           endpoint: dns:///flyteagent.flyte.svc.cluster.local:8000
           insecure: true
+          timeouts:
+            GetTask: 10s
   002-database.yaml: |
     database:
       postgres:
@@ -800,7 +803,7 @@ type: Opaque
 ---
 apiVersion: v1
 data:
-  haSharedSecret: d1B5UTNPWVBoRjA4UGFhYw==
+  haSharedSecret: NGdxVlh5RUFDQ2VUdTlqNA==
   proxyPassword: ""
   proxyUsername: ""
 kind: Secret
@@ -1198,7 +1201,7 @@ spec:
     metadata:
       annotations:
         checksum/cluster-resource-templates: 6fd9b172465e3089fcc59f738b92b8dc4d8939360c19de8ee65f68b0e7422035
-        checksum/configuration: 454886716e03425f5b6980d658280a5c9d2dd9b73767ab8e080ba381b97efc95
+        checksum/configuration: d8bf73644712bd050aa462a0478b0c779bd6d170645efe98feb5ffcacf52010e
         checksum/configuration-secret: 09216ffaa3d29e14f88b1f30af580d02a2a5e014de4d750b7f275cc07ed4e914
       labels:
         app.kubernetes.io/component: flyte-binary
@@ -1361,7 +1364,7 @@ spec:
     metadata:
       annotations:
         checksum/config: 8f50e768255a87f078ba8b9879a0c174c3e045ffb46ac8723d2eedbe293c8d81
-        checksum/secret: 750125a1cf681cdd45d59c2ff0f2cfb69349e246452fff3387bb1d65f5706794
+        checksum/secret: 24742f705403480bc843cf5c699f70632a9699fbdf771ad34a1672a448ffc14d
       labels:
         app: docker-registry
         release: flyte-sandbox
diff --git a/docker/sandbox-bundled/manifests/dev.yaml b/docker/sandbox-bundled/manifests/dev.yaml
index b9f1cd9e51..0c2ad54483 100644
--- a/docker/sandbox-bundled/manifests/dev.yaml
+++ b/docker/sandbox-bundled/manifests/dev.yaml
@@ -499,7 +499,7 @@ metadata:
 ---
 apiVersion: v1
 data:
-  haSharedSecret: Y1dWb0I2NXpmSUxtZFJBRA==
+  haSharedSecret: VVVDd08zZmw3d3NkV3o3dA==
   proxyPassword: ""
   proxyUsername: ""
 kind: Secret
@@ -933,7 +933,7 @@ spec:
     metadata:
       annotations:
         checksum/config: 8f50e768255a87f078ba8b9879a0c174c3e045ffb46ac8723d2eedbe293c8d81
-        checksum/secret: 9cfe892587af60802fd5c44efb44f44fca5ca5f5081a3f04c1ba944029aca54e
+        checksum/secret: 91877d32644e12d6cfebcc9c65192104dac73b183b4235af08d041d93a160cec
       labels:
         app: docker-registry
         release: flyte-sandbox