feat: filter candidates by request resolution

apps/api/internal/store/runtime_types.go

@@ -14,6 +14,7 @@ var (
 type ModelCandidateUnavailableError struct {
 	Code    string
 	Message string
+	Details map[string]any
 }
 
 func (e *ModelCandidateUnavailableError) Error() string {
@@ -32,6 +33,14 @@ func ModelCandidateErrorCode(err error) string {
 	return "no_model_candidate"
 }
 
+func ModelCandidateErrorDetails(err error) map[string]any {
+	var candidateErr *ModelCandidateUnavailableError
+	if errors.As(err, &candidateErr) && len(candidateErr.Details) > 0 {
+		return candidateErr.Details
+	}
+	return nil
+}
+
 type RateLimitExceededError struct {
 	ScopeType     string
 	ScopeKey      string
@@ -247,6 +256,7 @@ type FinishTaskFailureInput struct {
 	TaskID             string
 	Code               string
 	Message            string
+	Result             map[string]any
 	RequestID          string
 	Metrics            map[string]any
 	ResponseStartedAt  time.Time

apps/api/internal/store/tasks_runtime.go

@@ -778,22 +778,24 @@ func taskBillingString(value any) string {
 
 func (s *Store) FinishTaskFailure(ctx context.Context, input FinishTaskFailureInput) (GatewayTask, error) {
 	metricsJSON, _ := json.Marshal(emptyObjectIfNil(input.Metrics))
+	resultJSON, _ := json.Marshal(emptyObjectIfNil(input.Result))
 	if _, err := s.pool.Exec(ctx, `
-UPDATE gateway_tasks
-SET status = 'failed',
-    error = NULLIF($2::text, ''),
-    error_code = NULLIF($3::text, ''),
-    error_message = NULLIF($2::text, ''),
-    request_id = NULLIF($4::text, ''),
-    metrics = $5::jsonb,
-    response_started_at = $6::timestamptz,
-    response_finished_at = $7::timestamptz,
-    response_duration_ms = $8,
-    locked_by = NULL,
-    locked_at = NULL,
-    heartbeat_at = NULL,
-    finished_at = now(),
-    updated_at = now()
+	UPDATE gateway_tasks
+	SET status = 'failed',
+	    error = NULLIF($2::text, ''),
+	    error_code = NULLIF($3::text, ''),
+	    error_message = NULLIF($2::text, ''),
+	    request_id = NULLIF($4::text, ''),
+	    metrics = $5::jsonb,
+	    response_started_at = $6::timestamptz,
+	    response_finished_at = $7::timestamptz,
+	    response_duration_ms = $8,
+	    result = $9::jsonb,
+	    locked_by = NULL,
+	    locked_at = NULL,
+	    heartbeat_at = NULL,
+	    finished_at = now(),
+	    updated_at = now()
 WHERE id = $1::uuid`,
 		input.TaskID,
 		input.Message,
@@ -803,6 +805,7 @@ WHERE id = $1::uuid`,
 		nullableTime(input.ResponseStartedAt),
 		nullableTime(input.ResponseFinishedAt),
 		input.ResponseDurationMS,
+		string(resultJSON),
 	); err != nil {
 		return GatewayTask{}, err
 	}