From d8f08aacaeb52ae2db07faa6c451c6ed65de2818 Mon Sep 17 00:00:00 2001
From: JBD <jbd@golang.org>
Date: Sat, 10 Jan 2026 17:25:21 -0800
Subject: [PATCH] Always return State_STATE_FAILED when there is an error

---
 internal/server/server.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/internal/server/server.go b/internal/server/server.go
index c169dab..f9dba15 100644
--- a/internal/server/server.go
+++ b/internal/server/server.go
@@ -50,7 +50,6 @@ func (s *Server) TriggerSession(req *proto.TriggerSessionRequest, stream grpc.Se
 
 	// Trigger the session
 	if err := s.controller.TriggerSession(stream.Context(), sessionID, inputs, checkpointID); err != nil {
-		// Send failure response
 		stream.Send(&proto.TriggerSessionResponse{
 			SessionId: sessionID,
 			State:     proto.State_STATE_FAILED,
@@ -61,6 +60,10 @@ func (s *Server) TriggerSession(req *proto.TriggerSessionRequest, stream grpc.Se
 	// Get session state
 	session, err := s.controller.GetSession(sessionID)
 	if err != nil {
+		stream.Send(&proto.TriggerSessionResponse{
+			SessionId: sessionID,
+			State:     proto.State_STATE_FAILED,
+		})
 		return err
 	}