import { ModelToolCallRequest } from "../contracts/model-tool-call-request.type.mjs";
import { FinishReason } from "../contracts/finish-reason.type.mjs";
import { ModelCapabilities } from "../contracts/model.contract.mjs";
import { ImageModelPricing } from "../contracts/image-model.contract.mjs";
import { SpeechModelPricing } from "../contracts/speech-model.contract.mjs";
import { TranscriptionModelPricing } from "../contracts/transcription-model.contract.mjs";
import { MockImageResponse } from "./mock-image-model.mjs";
import { MockSpeechResponse } from "./mock-speech-model.mjs";
import { MockTranscriptionResponse } from "./mock-transcription-model.mjs";

//#region ../ai/src/mock/mock-config.type.d.ts
/**
 * Token counts a scripted `MockModelResponse` may declare.
 *
 * Deliberately NOT `Usage`. `Usage` is an emitted result — every field
 * on it is authoritative and `total` is always present. This is script
 * *input*, and `MockModel.buildResponse()` only honors `input`,
 * `output` and `cachedTokens`; it always recomputes `total` as
 * `input + output`, so a scripted `total` can never disagree with the
 * numbers it is derived from. `reasoningTokens`, `cacheWriteTokens`
 * and `cost` are omitted because the mock does not forward them —
 * declaring one here would be silently dropped.
 */
type MockUsage = {
  /** Prompt tokens the scripted response reports. */input: number; /** Completion tokens the scripted response reports. */
  output: number;
  /**
   * Ignored — `MockModel` always recomputes `total` as
   * `input + output`. Accepted so existing fixtures that spell it out
   * still compile.
   */
  total?: number; /** Subset of `input` served from the provider's prompt cache. */
  cachedTokens?: number;
};
/**
 * Configuration for a single mock model response.
 * Responses are consumed in order — last one repeats if list is exhausted.
 */
type MockModelResponse = {
  content: string;
  finishReason?: FinishReason;
  usage?: MockUsage;
  toolCalls?: ModelToolCallRequest[];
  /**
   * Exact chunk boundaries `stream()` should emit for `content`.
   * Omitted = the mock splits `content` on whitespace. Use this when a
   * test asserts on delta boundaries themselves (partial JSON, prose
   * fragments) rather than on the assembled text.
   */
  deltas?: string[]; /** Simulate a delay in ms before resolving */
  delay?: number; /** Throw this error instead of returning a response */
  error?: Error;
};
type MockSDKConfig = {
  /** Responses to return in sequence. Last one repeats when exhausted. */responses?: MockModelResponse[]; /** Default model name reported by mock models */
  defaultModelName?: string;
  /**
   * Capability flags reported by every model this mock SDK creates.
   * Tests use this to exercise capability-gated agent behavior (e.g.
   * vision attachments, native structured output) without standing up
   * a real provider.
   */
  capabilities?: ModelCapabilities;
  /**
   * Scripted responses for `image()` models this mock SDK creates,
   * consumed in sequence (last repeats when exhausted). Omitted = a
   * single default 1×1 PNG.
   */
  imageResponses?: MockImageResponse[];
  /**
   * Pricing stamped on every `image()` model this mock SDK creates, so
   * tests can exercise the per-image / per-token cost rollup.
   */
  imagePricing?: ImageModelPricing; /** Scripted responses for `speech()` (TTS) models. Omitted = a default audio blob. */
  speechResponses?: MockSpeechResponse[]; /** Pricing stamped on every `speech()` model — exercise the per-character / per-token rollup. */
  speechPricing?: SpeechModelPricing; /** Scripted responses for `transcribe()` (STT) models. Omitted = a default transcript. */
  transcriptionResponses?: MockTranscriptionResponse[]; /** Pricing stamped on every `transcribe()` model — exercise the per-minute / per-token rollup. */
  transcriptionPricing?: TranscriptionModelPricing;
};
//#endregion
export { MockModelResponse, MockSDKConfig, MockUsage };
//# sourceMappingURL=mock-config.type.d.mts.map