diff --git a/service/codex_credential_refresh_task.go b/service/codex_credential_refresh_task.go index 627ab929..5cc53fea 100644 --- a/service/codex_credential_refresh_task.go +++ b/service/codex_credential_refresh_task.go @@ -28,6 +28,10 @@ var ( codexCredentialRefreshRunning atomic.Bool ) +func shouldAutoRefreshCodexChannelStatus(status int) bool { + return status == common.ChannelStatusEnabled || status == common.ChannelStatusAutoDisabled +} + func StartCodexCredentialAutoRefreshTask() { codexCredentialRefreshOnce.Do(func() { if !common.IsMasterNode { @@ -65,7 +69,11 @@ func runCodexCredentialAutoRefreshOnce() { var channels []*model.Channel err := model.DB. Select("id", "name", "key", "status", "channel_info"). - Where("type = ? AND status = 1", constant.ChannelTypeCodex). + Where("type = ? AND (status = ? OR status = ?)", + constant.ChannelTypeCodex, + common.ChannelStatusEnabled, + common.ChannelStatusAutoDisabled, + ). Order("id asc"). Limit(codexCredentialRefreshBatchSize). Offset(offset).