import { isAnthropicBillingError, isApiKeyRateLimitError } from "./live-auth-keys.js";

import { isModelNotFoundErrorMessage } from "./live-model-errors.js";

export type LiveProviderDriftReason =

type LiveProviderDriftReason =

| "auth"

| "billing"

| "model-not-found"

| "timeout";

/** A normalized reason for skipping or soft-failing live provider drift. */

export type LiveProviderDriftDecision = {

type LiveProviderDriftDecision = {

label: string;

reason: LiveProviderDriftReason;

};

/** Classifier options that control which live-provider drift reasons are allowed. */

export type LiveProviderDriftOptions = {

type LiveProviderDriftOptions = {

allowAuth?: boolean;

allowBilling?: boolean;

allowModelNotFound?: boolean;

};

/** Converts arbitrary thrown values into text for provider drift matchers. */

export function liveProviderErrorText(error: unknown): string {

function liveProviderErrorText(error: unknown): string {

return error instanceof Error ? `${error.name}: ${error.message}` : String(error);

}

}

/** Returns whether an error is expected live timeout drift. */

export function isLiveTimeoutDrift(error: unknown): boolean {

function isLiveTimeoutDrift(error: unknown): boolean {

return isTimeoutErrorMessage(liveProviderErrorText(error));

}

/** Returns whether an error is expected live missing-model drift. */

export function isLiveModelNotFoundDrift(error: unknown): boolean {

function isLiveModelNotFoundDrift(error: unknown): boolean {

return isModelNotFoundErrorMessage(liveProviderErrorText(error));

}

});

}

/** Returns whether a task is an active session-scoped media generation task. */

export function isActiveMediaGenerationTask(params: {

task: TaskRecord;

taskKind: string;

}): boolean {

return (

params.task.runtime === "cli" &&

params.task.scopeKind === "session" &&

params.task.taskKind === params.taskKind &&

(params.task.status === "queued" || params.task.status === "running")

);

}

/** Records a just-started media task so duplicate guards work before persistence. */

export function recordRecentMediaGenerationTaskStartForSession(params: {

sessionKey?: string;

}

/** Extracts a provider id from a media task source id with the given prefix. */

export function getMediaGenerationTaskProviderId(

function getMediaGenerationTaskProviderId(

task: TaskRecord,

sourcePrefix: string,

): string | undefined {