diff --git a/workspace-server/internal/handlers/audit.go b/workspace-server/internal/handlers/audit.go index f2cc5a39c..859f146d8 100644 --- a/workspace-server/internal/handlers/audit.go +++ b/workspace-server/internal/handlers/audit.go @@ -252,6 +252,9 @@ func scanAuditRows(rows *sql.Rows) ([]auditEventRow, error) { } result = append(result, ev) } + if err := rows.Err(); err != nil { + return nil, err + } return result, nil } diff --git a/workspace-server/internal/handlers/llm_billing_mode.go b/workspace-server/internal/handlers/llm_billing_mode.go index af8aafadf..1576ea548 100644 --- a/workspace-server/internal/handlers/llm_billing_mode.go +++ b/workspace-server/internal/handlers/llm_billing_mode.go @@ -377,6 +377,9 @@ func readWorkspaceDeriveInputs(ctx context.Context, workspaceID string) (runtime availableAuthEnv = append(availableAuthEnv, k) } } + if err := rows.Err(); err != nil { + log.Printf("llm_billing_mode: rows iteration error for %s: %v (deriving with partial model/auth-env)", workspaceID, err) + } return runtime, model, availableAuthEnv }