import { normalizeGoogleApiBaseUrl } from "../../infra/google-api-base-url.js";

import { streamWithPayloadPatch } from "../../llm/providers/stream-wrappers/stream-payload-utils.js";

import type { Model } from "../../llm/types.js";

import {

asDateTimestampMs,

isFutureDateTimestampMs,

resolveExpiresAtMsFromDurationMs,

} from "../../shared/number-coercion.js";

import { normalizeOptionalString } from "../../shared/string-coerce.js";

import { buildGuardedModelFetch } from "../provider-transport-fetch.js";

import type { StreamFn } from "../runtime/index.js";

if (!expireTime) {

return null;

}

const timestamp = Date.parse(expireTime);

return Number.isFinite(timestamp) ? timestamp : null;

return asDateTimestampMs(Date.parse(expireTime)) ?? null;

}

function convertManagedGoogleTools(tools: NonNullable<GooglePromptCacheContext["tools"]>) {

deps: GooglePromptCacheDeps,

): Promise<string | null> {

const baseUrl = normalizeGoogleApiBaseUrl(params.model.baseUrl);

const now = deps.now?.() ?? Date.now();

const now = asDateTimestampMs(deps.now?.() ?? Date.now());

if (now === undefined) {

return null;

}

const systemPromptDigest = digestSystemPrompt(params.systemPrompt);

const matchKey = buildGooglePromptCacheMatchKey({

provider: params.provider,

});

const latestEntry = readLatestGooglePromptCacheEntry(params.sessionManager, matchKey);

if (latestEntry?.status === "failed" && latestEntry.retryAfter > now) {

if (

latestEntry?.status === "failed" &&

isFutureDateTimestampMs(latestEntry.retryAfter, { nowMs: now })

) {

return null;

}

const fetchImpl = (deps.buildGuardedFetch ?? buildGuardedModelFetch)(params.model);

const refreshWindowMs = resolveGooglePromptCacheRefreshWindowMs(params.cacheRetention);

if (latestEntry?.status === "ready" && latestEntry.cachedContent) {

const expiresAt = parseExpireTimeMs(latestEntry.expireTime);

const isExpired = expiresAt !== null && expiresAt <= now;

const isExpired = expiresAt !== null && !isFutureDateTimestampMs(expiresAt, { nowMs: now });

if (!isExpired) {

const needsRefresh = expiresAt !== null && expiresAt - now <= refreshWindowMs;

if (!needsRefresh) {

systemPromptDigest,

cacheConfigDigest: params.cacheConfigDigest,

cacheRetention: params.cacheRetention,

retryAfter: now + GOOGLE_PROMPT_CACHE_RETRY_BACKOFF_MS,

retryAfter:

resolveExpiresAtMsFromDurationMs(GOOGLE_PROMPT_CACHE_RETRY_BACKOFF_MS, { nowMs: now }) ?? 0,

});

return null;

}