improve agent error message

Signed-off-by: Future Outlier <[email protected]>
flyteorg · Jan 5, 2024 · 7982102 · 7982102
1 parent cb37291
commit 7982102
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flyteplugins/go/tasks/plugins/webapi/agent/plugin.go b/flyteplugins/go/tasks/plugins/webapi/agent/plugin.go
@@ -366,10 +366,10 @@ func initializeAgentRegistry(cfg *Config, connectionCache map[*Agent]*grpc.Clien
 			}
 
 			if !ok {
-				return nil, fmt.Errorf("failed to list agent with a non-gRPC error : [%v]", err)
+				return nil, fmt.Errorf("failed to list agent: [%v] with a non-gRPC error: [%v]", agentDeployment, err)
 			}
 
-			return nil, fmt.Errorf("failed to list agent with error: [%v]", err)
+			return nil, fmt.Errorf("failed to list agent: [%v] with error: [%v]", agentDeployment, err)
 		}
 
 		agents := res.GetAgents()