fix(executor): expire stop chunks without usage metadata

internal/runtime/executor/usage_helpers.go

@@ -416,28 +416,13 @@ func parseAntigravityStreamUsage(line []byte) (usage.Detail, bool) {
 	return detail, true
 }
 
-func jsonPayload(line []byte) []byte {
-	trimmed := bytes.TrimSpace(line)
-	if len(trimmed) == 0 {
-		return nil
-	}
-	if bytes.Equal(trimmed, []byte("[DONE]")) {
-		return nil
-	}
-	if bytes.HasPrefix(trimmed, []byte("event:")) {
-		return nil
-	}
-	if bytes.HasPrefix(trimmed, []byte("data:")) {
-		trimmed = bytes.TrimSpace(trimmed[len("data:"):])
-	}
-	if len(trimmed) == 0 || trimmed[0] != '{' {
-		return nil
-	}
-	return trimmed
-}
-
 var stopChunkWithoutUsage sync.Map
 
+func rememberStopWithoutUsage(traceID string) {
+	stopChunkWithoutUsage.Store(traceID, struct{}{})
+	time.AfterFunc(10*time.Minute, func() { stopChunkWithoutUsage.Delete(traceID) })
+}
+
 // FilterSSEUsageMetadata removes usageMetadata from SSE events that are not
 // terminal (finishReason != "stop"). Stop chunks are left untouched. This
 // function is shared between aistudio and antigravity executors.
@@ -462,15 +447,13 @@ func FilterSSEUsageMetadata(payload []byte) []byte {
 		rawJSON := bytes.TrimSpace(line[dataIdx+5:])
 		traceID := gjson.GetBytes(rawJSON, "traceId").String()
 		if isStopChunkWithoutUsage(rawJSON) && traceID != "" {
-			stopChunkWithoutUsage.Store(traceID, true)
+			rememberStopWithoutUsage(traceID)
 			continue
 		}
 		if traceID != "" {
-			if v, ok := stopChunkWithoutUsage.Load(traceID); ok {
+			if _, ok := stopChunkWithoutUsage.Load(traceID); ok && hasUsageMetadata(rawJSON) {
-				if keep, _ := v.(bool); keep && hasUsageMetadata(rawJSON) {
+				stopChunkWithoutUsage.Delete(traceID)
-					stopChunkWithoutUsage.Delete(traceID)
+				continue
-					continue
-				}
 			}
 		}
 
@@ -578,3 +561,23 @@ func isStopChunkWithoutUsage(jsonBytes []byte) bool {
 	}
 	return !hasUsageMetadata(jsonBytes)
 }
+
+func jsonPayload(line []byte) []byte {
+	trimmed := bytes.TrimSpace(line)
+	if len(trimmed) == 0 {
+		return nil
+	}
+	if bytes.Equal(trimmed, []byte("[DONE]")) {
+		return nil
+	}
+	if bytes.HasPrefix(trimmed, []byte("event:")) {
+		return nil
+	}
+	if bytes.HasPrefix(trimmed, []byte("data:")) {
+		trimmed = bytes.TrimSpace(trimmed[len("data:"):])
+	}
+	if len(trimmed) == 0 || trimmed[0] != '{' {
+		return nil
+	}
+	return trimmed
+}