romtuck/openclaw
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(agents): correct imports and API usage for cooldown check

Browse Source
This commit is contained in:
YiWang24 2026-01-26 03:53:09 -05:00 committed by Gustavo Madeira Santana
parent 23f65cc90b
commit 9227026af3
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/agents/model-fallback.ts
View File

@ -14,9 +14,9 @@ import {
resolveModelRefFromString, resolveModelRefFromString,
} from "./model-selection.js"; } from "./model-selection.js";
import type { FailoverReason } from "./pi-embedded-helpers.js"; import type { FailoverReason } from "./pi-embedded-helpers.js";
import { loadAuthProfileStore, isProfileInCooldown } from "./auth-profiles/usage.js"; import { isProfileInCooldown } from "./auth-profiles/usage.js";
import { loadAuthProfileStore } from "./auth-profiles/store.js";
import { resolveAuthProfileOrder } from "./auth-profiles/order.js"; import { resolveAuthProfileOrder } from "./auth-profiles/order.js";
import { getAgentAuthStoreDir } from "./agent-dirs.js";
type ModelCandidate = { type ModelCandidate = {
provider: string; provider: string;
@ -215,9 +215,7 @@ export async function runWithModelFallback<T>(params: {
fallbacksOverride: params.fallbacksOverride, fallbacksOverride: params.fallbacksOverride,
}); });
const authStore = params.cfg const authStore = params.cfg ? loadAuthProfileStore() : null;
? loadAuthProfileStore(getAgentAuthStoreDir(params.cfg, "main"))
: null;
const attempts: FallbackAttempt[] = []; const attempts: FallbackAttempt[] = [];
let lastError: unknown; let lastError: unknown;
@ -227,7 +225,11 @@ export async function runWithModelFallback<T>(params: {
// Skip candidates that are in cooldown // Skip candidates that are in cooldown
if (authStore) { if (authStore) {
const profileIds = resolveAuthProfileOrder(authStore, candidate.provider); const profileIds = resolveAuthProfileOrder({
cfg: params.cfg,
store: authStore,
provider: candidate.provider,
});
const isAnyProfileAvailable = profileIds.some((id) => !isProfileInCooldown(authStore, id)); const isAnyProfileAvailable = profileIds.some((id) => !isProfileInCooldown(authStore, id));
if (profileIds.length > 0 && !isAnyProfileAvailable) { if (profileIds.length > 0 && !isAnyProfileAvailable) {