@effect-uai/google 0.6.0 → 0.7.0

package/dist/codec.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"codec.mjs","names":["Arr"],"sources":["../src/codec.ts"],"sourcesContent":["import { Array as Arr, Encoding, Match, Option, Result, Schema, pipe } from \"effect\"\nimport type {\n  ContentBlock,\n  ToolCall,\n  InputImage,\n  HistoryItem,\n  Message,\n} from \"@effect-uai/core/Items\"\nimport type { ToolDescriptor } from \"@effect-uai/core/Tool\"\nimport type { Turn } from \"@effect-uai/core/Turn\"\n\n// ---------------------------------------------------------------------------\n// Wire schemas - the subset of Gemini's streamGenerateContent payload we\n// consume. Reference: https://ai.google.dev/api/generate-content\n// ---------------------------------------------------------------------------\n\nconst TextPart = Schema.Struct({\n  text: Schema.String,\n  /**\n   * Gemini's flag for chain-of-thought parts. When `true`, the part is\n   * reasoning trace, not the model's user-facing answer. Maps onto\n   * `reasoning_delta { kind: \"trace\" }` in the canonical view.\n   */\n  thought: Schema.optional(Schema.Boolean),\n})\n\n/**\n * Gemini function-call part. Args are delivered as a whole JSON *object*\n * in a single chunk (Gemini does not stream tool-call args). On Gemini 3\n * the response also carries an `id` per call, which we echo back on the\n * corresponding `functionResponse`.\n */\nconst FunctionCallPart = Schema.Struct({\n  functionCall: Schema.Struct({\n    id: Schema.optional(Schema.String),\n    name: Schema.String,\n    args: Schema.optional(Schema.Unknown),\n  }),\n})\n\nconst Part = Schema.Union([TextPart, FunctionCallPart])\ntype WireFunctionCallPart = typeof FunctionCallPart.Type\n\nconst Content = Schema.Struct({\n  role: Schema.optional(Schema.String),\n  parts: Schema.optional(Schema.Array(Part)),\n})\n\nconst Candidate = Schema.Struct({\n  content: Schema.optional(Content),\n  finishReason: Schema.optional(Schema.String),\n  index: Schema.optional(Schema.Number),\n})\n\nconst UsageMetadata = Schema.Struct({\n  promptTokenCount: Schema.optional(Schema.Number),\n  candidatesTokenCount: Schema.optional(Schema.Number),\n  totalTokenCount: Schema.optional(Schema.Number),\n  cachedContentTokenCount: Schema.optional(Schema.Number),\n  thoughtsTokenCount: Schema.optional(Schema.Number),\n})\n\nexport const WireChunk = Schema.Struct({\n  candidates: Schema.optional(Schema.Array(Candidate)),\n  usageMetadata: Schema.optional(UsageMetadata),\n})\nexport type WireChunk = typeof WireChunk.Type\n\n// ---------------------------------------------------------------------------\n// History → request body\n// ---------------------------------------------------------------------------\n\ntype RequestPart =\n  | { readonly text: string }\n  | { readonly inlineData: { readonly mimeType: string; readonly data: string } }\n  | { readonly functionCall: { readonly name: string; readonly args: unknown } }\n  | {\n      readonly functionResponse: {\n        readonly name: string\n        readonly response: Record<string, unknown>\n      }\n    }\n\ntype RequestContent = {\n  readonly role: \"user\" | \"model\"\n  readonly parts: ReadonlyArray<RequestPart>\n}\n\ntype RequestSystemInstruction = {\n  readonly parts: ReadonlyArray<{ readonly text: string }>\n}\n\n/** Gemini's tool declaration. We translate one `ToolDescriptor` per entry. */\ntype RequestTool = {\n  readonly functionDeclarations: ReadonlyArray<{\n    readonly name: string\n    readonly description: string\n    readonly parameters: Record<string, unknown>\n  }>\n}\n\ntype RequestToolConfig = {\n  readonly functionCallingConfig: {\n    readonly mode: \"AUTO\" | \"ANY\" | \"NONE\"\n    readonly allowedFunctionNames?: ReadonlyArray<string>\n  }\n}\n\nexport type ThinkingConfig = {\n  readonly thinkingBudget: number\n}\n\nexport type GenerationConfig = {\n  readonly temperature?: number\n  readonly maxOutputTokens?: number\n  readonly topP?: number\n  readonly thinkingConfig?: ThinkingConfig\n  /** Set together with `responseJsonSchema` to constrain output to JSON. */\n  readonly responseMimeType?: string\n  /** JSON Schema constraint on the output. */\n  readonly responseJsonSchema?: Record<string, unknown>\n}\n\nexport type RequestBody = {\n  readonly contents: ReadonlyArray<RequestContent>\n  readonly systemInstruction?: RequestSystemInstruction\n  readonly generationConfig?: GenerationConfig\n  readonly tools?: ReadonlyArray<RequestTool>\n  readonly toolConfig?: RequestToolConfig\n}\n\nconst blockText = Match.type<ContentBlock>().pipe(\n  Match.discriminatorsExhaustive(\"type\")({\n    input_text: (b) => b.text,\n    input_image: () => \"\",\n    output_text: (b) => b.text,\n    refusal: (b) => b.text,\n  }),\n)\n\nconst messageText = (message: Message): string => message.content.map(blockText).join(\"\")\n\n/**\n * Gemini's `inlineData` form expects a base64 payload. URL-form images\n * would need to go through Gemini's Files API (upload then `fileData`\n * with the returned URI); pre-uploading isn't free, so we skip those for\n * now and document as a follow-up.\n */\nconst imageSourceToParts = Match.type<InputImage[\"source\"]>().pipe(\n  Match.tag(\"url\", (): ReadonlyArray<RequestPart> => []),\n  Match.tag(\n    \"base64\",\n    (s): ReadonlyArray<RequestPart> => [{ inlineData: { mimeType: s.mimeType, data: s.base64 } }],\n  ),\n  Match.tag(\n    \"bytes\",\n    (s): ReadonlyArray<RequestPart> => [\n      { inlineData: { mimeType: s.mimeType, data: Encoding.encodeBase64(s.bytes) } },\n    ],\n  ),\n  Match.exhaustive,\n)\n\nconst blockToParts = Match.type<ContentBlock>().pipe(\n  Match.discriminatorsExhaustive(\"type\")({\n    input_text: (b): ReadonlyArray<RequestPart> => (b.text.length === 0 ? [] : [{ text: b.text }]),\n    input_image: (b): ReadonlyArray<RequestPart> => imageSourceToParts(b.source),\n    output_text: (b): ReadonlyArray<RequestPart> => (b.text.length === 0 ? [] : [{ text: b.text }]),\n    // Refusals are assistant-side content; they don't round-trip into Gemini's\n    // request body as parts. Skip.\n    refusal: (): ReadonlyArray<RequestPart> => [],\n  }),\n)\n\nconst messageToContent = (message: Message): Result.Result<RequestContent, void> => {\n  const parts = pipe(message.content, Arr.flatMap(blockToParts))\n  if (parts.length === 0) return Result.failVoid\n  return Match.value(message.role).pipe(\n    Match.when(\"user\", () => Result.succeed({ role: \"user\" as const, parts })),\n    Match.when(\"assistant\", () => Result.succeed({ role: \"model\" as const, parts })),\n    Match.when(\"system\", () => Result.failVoid),\n    Match.exhaustive,\n  )\n}\n\nconst systemMessageText = (message: Message): Result.Result<string, void> => {\n  if (message.role !== \"system\") return Result.failVoid\n  const text = messageText(message)\n  return text.length === 0 ? Result.failVoid : Result.succeed(text)\n}\n\nconst allMessages = (history: ReadonlyArray<HistoryItem>): ReadonlyArray<Message> =>\n  pipe(\n    history,\n    Arr.filterMap((item) => (item.type === \"message\" ? Result.succeed(item) : Result.failVoid)),\n  )\n\n// ---------------------------------------------------------------------------\n// Function-call round-trip\n//\n// `ToolCall.arguments` and `ToolCallOutput.output` are JSON-encoded\n// strings; Gemini's `functionCall.args` and `functionResponse.response`\n// expect parsed JSON *objects*. Decode via Schema:\n//   - object payload → use as-is\n//   - scalar / array / malformed → wrap as `{ output: <raw string> }` so the\n//     model still sees *some* response without crashing the request.\n// ---------------------------------------------------------------------------\n\nconst JsonObject = Schema.Record(Schema.String, Schema.Unknown)\nconst decodeJsonObject = Schema.decodeResult(Schema.fromJsonString(JsonObject))\n\nconst parsedJsonObject =\n  (fallback: (raw: string) => Record<string, unknown>) =>\n  (encoded: string): Record<string, unknown> =>\n    pipe(\n      decodeJsonObject(encoded),\n      Result.match({\n        onSuccess: (v) => v,\n        onFailure: () => fallback(encoded),\n      }),\n    )\n\nconst parsedArgs = parsedJsonObject(() => ({}))\nconst parsedResponse = parsedJsonObject((raw) => ({ output: raw }))\n\n/**\n * `ToolCallOutput` only carries `call_id`; Gemini's `functionResponse`\n * requires the declared function `name`. Resolve the name by scanning prior\n * `function_call` items in the history for a matching `call_id`. If we\n * cannot resolve, fall back to `call_id` as the name - imperfect but\n * preserves stream shape so the model sees *some* response.\n */\nconst isFunctionCallItem = (item: HistoryItem): item is ToolCall => item.type === \"function_call\"\n\nconst nameForCallId = (\n  history: ReadonlyArray<HistoryItem>,\n  call_id: string,\n): Option.Option<string> =>\n  pipe(\n    history,\n    Arr.findFirst((item): item is ToolCall => isFunctionCallItem(item) && item.call_id === call_id),\n    Option.map((f) => f.name),\n  )\n\n/**\n * Extract the Gemini-3 wire id we stashed in `providerData` on the way out.\n * Schema-driven so the shape lives in one place; failure → `Option.none()`.\n */\nconst ProviderDataWithGeminiId = Schema.Struct({\n  gemini: Schema.Struct({ id: Schema.String }),\n})\nconst decodeProviderId = Schema.decodeUnknownResult(ProviderDataWithGeminiId)\n\nconst providerIdFor = (item: ToolCall): Option.Option<string> =>\n  pipe(\n    decodeProviderId(item.providerData),\n    Result.match({\n      onSuccess: (d) => Option.some(d.gemini.id),\n      onFailure: () => Option.none<string>(),\n    }),\n  )\n\nconst itemToContent =\n  (history: ReadonlyArray<HistoryItem>) =>\n  (item: HistoryItem): Result.Result<RequestContent, void> =>\n    Match.value(item).pipe(\n      Match.discriminatorsExhaustive(\"type\")({\n        message: messageToContent,\n        function_call: (f) =>\n          Result.succeed({\n            role: \"model\" as const,\n            parts: [\n              {\n                functionCall: {\n                  ...Option.match(providerIdFor(f), {\n                    onSome: (id) => ({ id }),\n                    onNone: () => ({}),\n                  }),\n                  name: f.name,\n                  args: parsedArgs(f.arguments),\n                },\n              },\n            ],\n          }),\n        function_call_output: (o) =>\n          Result.succeed({\n            role: \"user\" as const,\n            parts: [\n              {\n                functionResponse: {\n                  name: Option.getOrElse(nameForCallId(history, o.call_id), () => o.call_id),\n                  response: parsedResponse(o.output),\n                },\n              },\n            ],\n          }),\n        reasoning: () => Result.failVoid,\n      }),\n    )\n\n// ---------------------------------------------------------------------------\n// Tool descriptors → Gemini `functionDeclarations`. Gemini accepts only a\n// strict OpenAPI 3.0 subset for `parameters`; strip JSON-Schema keys it\n// rejects (`$schema`, `$ref`, `additionalProperties`, `oneOf`,\n// `definitions`).\n// ---------------------------------------------------------------------------\n\nconst UNSUPPORTED_SCHEMA_KEYS: ReadonlySet<string> = new Set([\n  \"$schema\",\n  \"$ref\",\n  \"$defs\",\n  \"definitions\",\n  \"additionalProperties\",\n  \"oneOf\",\n])\n\nconst sanitizeSchema = (schema: unknown): unknown => {\n  if (Array.isArray(schema)) return schema.map(sanitizeSchema)\n  if (schema === null || typeof schema !== \"object\") return schema\n  return pipe(\n    Object.entries(schema as Record<string, unknown>),\n    Arr.filterMap(([k, v]) =>\n      UNSUPPORTED_SCHEMA_KEYS.has(k)\n        ? Result.failVoid\n        : Result.succeed([k, sanitizeSchema(v)] as const),\n    ),\n    Object.fromEntries,\n  )\n}\n\nconst toolDescriptorsToTools = (tools: ReadonlyArray<ToolDescriptor>): ReadonlyArray<RequestTool> =>\n  tools.length === 0\n    ? []\n    : [\n        {\n          functionDeclarations: tools.map((t) => ({\n            name: t.name,\n            description: t.description,\n            parameters: sanitizeSchema(t.inputSchema) as Record<string, unknown>,\n          })),\n        },\n      ]\n\nexport const buildRequestBody = (\n  history: ReadonlyArray<HistoryItem>,\n  generationConfig: Option.Option<GenerationConfig>,\n  tools: ReadonlyArray<ToolDescriptor> = [],\n): RequestBody => {\n  const systemTexts = pipe(allMessages(history), Arr.filterMap(systemMessageText))\n  const contents = pipe(history, Arr.filterMap(itemToContent(history)))\n  const requestTools = toolDescriptorsToTools(tools)\n  return {\n    contents,\n    ...(systemTexts.length > 0 && {\n      systemInstruction: { parts: [{ text: systemTexts.join(\"\\n\") }] },\n    }),\n    ...Option.match(generationConfig, {\n      onNone: () => ({}),\n      onSome: (cfg) => ({ generationConfig: cfg }),\n    }),\n    ...(requestTools.length > 0 && {\n      tools: requestTools,\n      toolConfig: { functionCallingConfig: { mode: \"AUTO\" as const } },\n    }),\n  }\n}\n\n// ---------------------------------------------------------------------------\n// Stream-level state - accumulate chunk text + final usage/finish.\n//\n// `Accumulator` is immutable; `ingestChunk` returns a fresh one per chunk.\n// Drive it via `Stream.mapAccum` in the consumer.\n// ---------------------------------------------------------------------------\n\nconst finishReasonToStop = (reason: Option.Option<string>): Turn[\"stop_reason\"] =>\n  Option.match(reason, {\n    onNone: () => \"stop\" as const,\n    onSome: (r) => (r === \"MAX_TOKENS\" ? (\"max_tokens\" as const) : (\"stop\" as const)),\n  })\n\nexport type AccumulatedFunctionCall = {\n  /** Synthesized id-or-name we surface as `call_id` on the canonical item. */\n  readonly callId: string\n  readonly name: string\n  /** Wire id from Gemini 3, when present - echoed back on `functionResponse`. */\n  readonly providerId: Option.Option<string>\n  /** Args as JSON-encoded string, mirroring `Items.ToolCall.arguments`. */\n  readonly arguments: string\n}\n\nexport type Accumulator = {\n  readonly text: string\n  readonly reasoning: string\n  readonly functionCalls: ReadonlyArray<AccumulatedFunctionCall>\n  readonly finishReason: Option.Option<string>\n  readonly usage: {\n    readonly input_tokens?: number\n    readonly output_tokens?: number\n    readonly total_tokens?: number\n    readonly input_tokens_details?: { readonly cached_tokens?: number }\n    readonly output_tokens_details?: { readonly reasoning_tokens?: number }\n  }\n}\n\nexport const emptyAccumulator: Accumulator = {\n  text: \"\",\n  reasoning: \"\",\n  functionCalls: [],\n  finishReason: Option.none(),\n  usage: {},\n}\n\n/**\n * One part's worth of streamable output. `text` and `reasoning` are\n * incremental string deltas; `function_call` arrives whole-in-one-chunk\n * (Gemini does not stream tool-call args).\n */\nexport type ChunkPart =\n  | { readonly kind: \"text\"; readonly text: string }\n  | { readonly kind: \"reasoning\"; readonly text: string }\n  | {\n      readonly kind: \"function_call\"\n      readonly id: Option.Option<string>\n      readonly name: string\n      readonly args: unknown\n    }\n\nexport type ChunkResult = {\n  readonly accumulator: Accumulator\n  readonly parts: ReadonlyArray<ChunkPart>\n  readonly finished: boolean\n}\n\nconst isTextPart = (p: typeof Part.Type): p is typeof TextPart.Type => \"text\" in p\n\nconst textToChunkParts = (p: typeof TextPart.Type): ReadonlyArray<ChunkPart> =>\n  p.text.length === 0 ? [] : [{ kind: p.thought === true ? \"reasoning\" : \"text\", text: p.text }]\n\nconst functionCallToChunkParts = (p: WireFunctionCallPart): ReadonlyArray<ChunkPart> => [\n  {\n    kind: \"function_call\",\n    id: Option.fromNullishOr(p.functionCall.id),\n    name: p.functionCall.name,\n    args: p.functionCall.args ?? {},\n  },\n]\n\nconst partToChunkParts = (p: typeof Part.Type): ReadonlyArray<ChunkPart> =>\n  isTextPart(p) ? textToChunkParts(p) : functionCallToChunkParts(p)\n\nconst chunkParts = (chunk: WireChunk): ReadonlyArray<ChunkPart> =>\n  pipe(chunk.candidates?.[0]?.content?.parts ?? [], Arr.flatMap(partToChunkParts))\n\nconst sumStrings = (parts: ReadonlyArray<ChunkPart>, kind: \"text\" | \"reasoning\"): string =>\n  pipe(\n    parts,\n    Arr.filterMap((p) => (p.kind === kind ? Result.succeed(p.text) : Result.failVoid)),\n  ).join(\"\")\n\nconst collectFunctionCalls = (\n  parts: ReadonlyArray<ChunkPart>,\n): ReadonlyArray<Extract<ChunkPart, { kind: \"function_call\" }>> =>\n  pipe(\n    parts,\n    Arr.filterMap((p) => (p.kind === \"function_call\" ? Result.succeed(p) : Result.failVoid)),\n  )\n\nconst mergeUsage = (\n  prev: Accumulator[\"usage\"],\n  next: WireChunk[\"usageMetadata\"],\n): Accumulator[\"usage\"] =>\n  next === undefined\n    ? prev\n    : {\n        ...prev,\n        ...(next.promptTokenCount !== undefined && { input_tokens: next.promptTokenCount }),\n        ...(next.candidatesTokenCount !== undefined && {\n          output_tokens: next.candidatesTokenCount,\n        }),\n        ...(next.totalTokenCount !== undefined && { total_tokens: next.totalTokenCount }),\n        ...(next.cachedContentTokenCount !== undefined && {\n          input_tokens_details: { cached_tokens: next.cachedContentTokenCount },\n        }),\n        ...(next.thoughtsTokenCount !== undefined && {\n          output_tokens_details: { reasoning_tokens: next.thoughtsTokenCount },\n        }),\n      }\n\n/**\n * Synthesize a stable `call_id` for a function call. Gemini 3 provides one\n * via `functionCall.id`; older models do not, so we fall back to\n * `<name>_<index>` based on prior calls' position in the accumulator.\n */\nconst synthesizeCallId = (\n  call: Extract<ChunkPart, { kind: \"function_call\" }>,\n  priorCalls: ReadonlyArray<AccumulatedFunctionCall>,\n): string =>\n  Option.match(call.id, {\n    onSome: (id) => id,\n    onNone: () => `${call.name}_${priorCalls.length}`,\n  })\n\nconst chunkCallToAccumulated = (\n  prior: ReadonlyArray<AccumulatedFunctionCall>,\n  call: Extract<ChunkPart, { kind: \"function_call\" }>,\n): AccumulatedFunctionCall => ({\n  callId: synthesizeCallId(call, prior),\n  name: call.name,\n  providerId: call.id,\n  // `functionCallToChunkParts` already replaced null/undefined with `{}`,\n  // so `call.args` is always a JSON-encodable value here.\n  arguments: JSON.stringify(call.args),\n})\n\nconst appendFunctionCalls = (\n  prior: ReadonlyArray<AccumulatedFunctionCall>,\n  fromChunk: ReadonlyArray<Extract<ChunkPart, { kind: \"function_call\" }>>,\n): ReadonlyArray<AccumulatedFunctionCall> =>\n  fromChunk.reduce<ReadonlyArray<AccumulatedFunctionCall>>(\n    (acc, call) => [...acc, chunkCallToAccumulated(acc, call)],\n    prior,\n  )\n\nexport const ingestChunk = (acc: Accumulator, chunk: WireChunk): ChunkResult => {\n  const parts = chunkParts(chunk)\n  const finishReason = Option.fromNullishOr(chunk.candidates?.[0]?.finishReason)\n  return {\n    parts,\n    finished: Option.isSome(finishReason),\n    accumulator: {\n      text: acc.text + sumStrings(parts, \"text\"),\n      reasoning: acc.reasoning + sumStrings(parts, \"reasoning\"),\n      functionCalls: appendFunctionCalls(acc.functionCalls, collectFunctionCalls(parts)),\n      finishReason: Option.orElse(finishReason, () => acc.finishReason),\n      usage: mergeUsage(acc.usage, chunk.usageMetadata),\n    },\n  }\n}\n\nconst reasoningItems = (acc: Accumulator): ReadonlyArray<HistoryItem> =>\n  acc.reasoning.length > 0 ? [{ type: \"reasoning\", summary: acc.reasoning }] : []\n\nconst assistantMessageItems = (acc: Accumulator): ReadonlyArray<HistoryItem> =>\n  acc.text.length === 0\n    ? []\n    : [\n        {\n          type: \"message\",\n          role: \"assistant\",\n          content: [{ type: \"output_text\", text: acc.text }],\n        },\n      ]\n\nconst functionCallItems = (acc: Accumulator): ReadonlyArray<HistoryItem> =>\n  pipe(\n    acc.functionCalls,\n    Arr.map((c) => ({\n      type: \"function_call\" as const,\n      call_id: c.callId,\n      name: c.name,\n      arguments: c.arguments,\n      ...(Option.isSome(c.providerId) && { providerData: { gemini: { id: c.providerId.value } } }),\n    })),\n  )\n\nexport const accumulatorToTurn = (acc: Accumulator): Turn => ({\n  stop_reason:\n    acc.functionCalls.length > 0 ? (\"tool_calls\" as const) : finishReasonToStop(acc.finishReason),\n  usage: { ...acc.usage },\n  items: [...reasoningItems(acc), ...assistantMessageItems(acc), ...functionCallItems(acc)],\n})\n"],"mappings":";;;;;;;;;;AAgBA,MAAM,WAAW,OAAO,OAAO;CAC7B,MAAM,OAAO;;;;;;CAMb,SAAS,OAAO,SAAS,OAAO,QAAQ;CACzC,CAAC;;;;;;;AAQF,MAAM,mBAAmB,OAAO,OAAO,EACrC,cAAc,OAAO,OAAO;CAC1B,IAAI,OAAO,SAAS,OAAO,OAAO;CAClC,MAAM,OAAO;CACb,MAAM,OAAO,SAAS,OAAO,QAAQ;CACtC,CAAC,EACH,CAAC;AAEF,MAAM,OAAO,OAAO,MAAM,CAAC,UAAU,iBAAiB,CAAC;AAGvD,MAAM,UAAU,OAAO,OAAO;CAC5B,MAAM,OAAO,SAAS,OAAO,OAAO;CACpC,OAAO,OAAO,SAAS,OAAO,MAAM,KAAK,CAAC;CAC3C,CAAC;AAEF,MAAM,YAAY,OAAO,OAAO;CAC9B,SAAS,OAAO,SAAS,QAAQ;CACjC,cAAc,OAAO,SAAS,OAAO,OAAO;CAC5C,OAAO,OAAO,SAAS,OAAO,OAAO;CACtC,CAAC;AAEF,MAAM,gBAAgB,OAAO,OAAO;CAClC,kBAAkB,OAAO,SAAS,OAAO,OAAO;CAChD,sBAAsB,OAAO,SAAS,OAAO,OAAO;CACpD,iBAAiB,OAAO,SAAS,OAAO,OAAO;CAC/C,yBAAyB,OAAO,SAAS,OAAO,OAAO;CACvD,oBAAoB,OAAO,SAAS,OAAO,OAAO;CACnD,CAAC;AAEF,MAAa,YAAY,OAAO,OAAO;CACrC,YAAY,OAAO,SAAS,OAAO,MAAM,UAAU,CAAC;CACpD,eAAe,OAAO,SAAS,cAAc;CAC9C,CAAC;AAkEF,MAAM,YAAY,MAAM,MAAoB,CAAC,KAC3C,MAAM,yBAAyB,OAAO,CAAC;CACrC,aAAa,MAAM,EAAE;CACrB,mBAAmB;CACnB,cAAc,MAAM,EAAE;CACtB,UAAU,MAAM,EAAE;CACnB,CAAC,CACH;AAED,MAAM,eAAe,YAA6B,QAAQ,QAAQ,IAAI,UAAU,CAAC,KAAK,GAAG;;;;;;;AAQzF,MAAM,qBAAqB,MAAM,MAA4B,CAAC,KAC5D,MAAM,IAAI,aAAyC,EAAE,CAAC,EACtD,MAAM,IACJ,WACC,MAAkC,CAAC,EAAE,YAAY;CAAE,UAAU,EAAE;CAAU,MAAM,EAAE;CAAQ,EAAE,CAAC,CAC9F,EACD,MAAM,IACJ,UACC,MAAkC,CACjC,EAAE,YAAY;CAAE,UAAU,EAAE;CAAU,MAAM,SAAS,aAAa,EAAE,MAAM;CAAE,EAAE,CAC/E,CACF,EACD,MAAM,WACP;AAED,MAAM,eAAe,MAAM,MAAoB,CAAC,KAC9C,MAAM,yBAAyB,OAAO,CAAC;CACrC,aAAa,MAAmC,EAAE,KAAK,WAAW,IAAI,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC;CAC7F,cAAc,MAAkC,mBAAmB,EAAE,OAAO;CAC5E,cAAc,MAAmC,EAAE,KAAK,WAAW,IAAI,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC;CAG9F,eAA2C,EAAE;CAC9C,CAAC,CACH;AAED,MAAM,oBAAoB,YAA0D;CAClF,MAAM,QAAQ,KAAK,QAAQ,SAASA,QAAI,QAAQ,aAAa,CAAC;AAC9D,KAAI,MAAM,WAAW,EAAG,QAAO,OAAO;AACtC,QAAO,MAAM,MAAM,QAAQ,KAAK,CAAC,KAC/B,MAAM,KAAK,cAAc,OAAO,QAAQ;EAAE,MAAM;EAAiB;EAAO,CAAC,CAAC,EAC1E,MAAM,KAAK,mBAAmB,OAAO,QAAQ;EAAE,MAAM;EAAkB;EAAO,CAAC,CAAC,EAChF,MAAM,KAAK,gBAAgB,OAAO,SAAS,EAC3C,MAAM,WACP;;AAGH,MAAM,qBAAqB,YAAkD;AAC3E,KAAI,QAAQ,SAAS,SAAU,QAAO,OAAO;CAC7C,MAAM,OAAO,YAAY,QAAQ;AACjC,QAAO,KAAK,WAAW,IAAI,OAAO,WAAW,OAAO,QAAQ,KAAK;;AAGnE,MAAM,eAAe,YACnB,KACE,SACAA,QAAI,WAAW,SAAU,KAAK,SAAS,YAAY,OAAO,QAAQ,KAAK,GAAG,OAAO,SAAU,CAC5F;AAaH,MAAM,aAAa,OAAO,OAAO,OAAO,QAAQ,OAAO,QAAQ;AAC/D,MAAM,mBAAmB,OAAO,aAAa,OAAO,eAAe,WAAW,CAAC;AAE/E,MAAM,oBACH,cACA,YACC,KACE,iBAAiB,QAAQ,EACzB,OAAO,MAAM;CACX,YAAY,MAAM;CAClB,iBAAiB,SAAS,QAAQ;CACnC,CAAC,CACH;AAEL,MAAM,aAAa,wBAAwB,EAAE,EAAE;AAC/C,MAAM,iBAAiB,kBAAkB,SAAS,EAAE,QAAQ,KAAK,EAAE;;;;;;;;AASnE,MAAM,sBAAsB,SAAwC,KAAK,SAAS;AAElF,MAAM,iBACJ,SACA,YAEA,KACE,SACAA,QAAI,WAAW,SAA2B,mBAAmB,KAAK,IAAI,KAAK,YAAY,QAAQ,EAC/F,OAAO,KAAK,MAAM,EAAE,KAAK,CAC1B;;;;;AAMH,MAAM,2BAA2B,OAAO,OAAO,EAC7C,QAAQ,OAAO,OAAO,EAAE,IAAI,OAAO,QAAQ,CAAC,EAC7C,CAAC;AACF,MAAM,mBAAmB,OAAO,oBAAoB,yBAAyB;AAE7E,MAAM,iBAAiB,SACrB,KACE,iBAAiB,KAAK,aAAa,EACnC,OAAO,MAAM;CACX,YAAY,MAAM,OAAO,KAAK,EAAE,OAAO,GAAG;CAC1C,iBAAiB,OAAO,MAAc;CACvC,CAAC,CACH;AAEH,MAAM,iBACH,aACA,SACC,MAAM,MAAM,KAAK,CAAC,KAChB,MAAM,yBAAyB,OAAO,CAAC;CACrC,SAAS;CACT,gBAAgB,MACd,OAAO,QAAQ;EACb,MAAM;EACN,OAAO,CACL,EACE,cAAc;GACZ,GAAG,OAAO,MAAM,cAAc,EAAE,EAAE;IAChC,SAAS,QAAQ,EAAE,IAAI;IACvB,eAAe,EAAE;IAClB,CAAC;GACF,MAAM,EAAE;GACR,MAAM,WAAW,EAAE,UAAU;GAC9B,EACF,CACF;EACF,CAAC;CACJ,uBAAuB,MACrB,OAAO,QAAQ;EACb,MAAM;EACN,OAAO,CACL,EACE,kBAAkB;GAChB,MAAM,OAAO,UAAU,cAAc,SAAS,EAAE,QAAQ,QAAQ,EAAE,QAAQ;GAC1E,UAAU,eAAe,EAAE,OAAO;GACnC,EACF,CACF;EACF,CAAC;CACJ,iBAAiB,OAAO;CACzB,CAAC,CACH;AASL,MAAM,0BAA+C,IAAI,IAAI;CAC3D;CACA;CACA;CACA;CACA;CACA;CACD,CAAC;AAEF,MAAM,kBAAkB,WAA6B;AACnD,KAAI,MAAM,QAAQ,OAAO,CAAE,QAAO,OAAO,IAAI,eAAe;AAC5D,KAAI,WAAW,QAAQ,OAAO,WAAW,SAAU,QAAO;AAC1D,QAAO,KACL,OAAO,QAAQ,OAAkC,EACjDA,QAAI,WAAW,CAAC,GAAG,OACjB,wBAAwB,IAAI,EAAE,GAC1B,OAAO,WACP,OAAO,QAAQ,CAAC,GAAG,eAAe,EAAE,CAAC,CAAU,CACpD,EACD,OAAO,YACR;;AAGH,MAAM,0BAA0B,UAC9B,MAAM,WAAW,IACb,EAAE,GACF,CACE,EACE,sBAAsB,MAAM,KAAK,OAAO;CACtC,MAAM,EAAE;CACR,aAAa,EAAE;CACf,YAAY,eAAe,EAAE,YAAY;CAC1C,EAAE,EACJ,CACF;AAEP,MAAa,oBACX,SACA,kBACA,QAAuC,EAAE,KACzB;CAChB,MAAM,cAAc,KAAK,YAAY,QAAQ,EAAEA,QAAI,UAAU,kBAAkB,CAAC;CAChF,MAAM,WAAW,KAAK,SAASA,QAAI,UAAU,cAAc,QAAQ,CAAC,CAAC;CACrE,MAAM,eAAe,uBAAuB,MAAM;AAClD,QAAO;EACL;EACA,GAAI,YAAY,SAAS,KAAK,EAC5B,mBAAmB,EAAE,OAAO,CAAC,EAAE,MAAM,YAAY,KAAK,KAAK,EAAE,CAAC,EAAE,EACjE;EACD,GAAG,OAAO,MAAM,kBAAkB;GAChC,eAAe,EAAE;GACjB,SAAS,SAAS,EAAE,kBAAkB,KAAK;GAC5C,CAAC;EACF,GAAI,aAAa,SAAS,KAAK;GAC7B,OAAO;GACP,YAAY,EAAE,uBAAuB,EAAE,MAAM,QAAiB,EAAE;GACjE;EACF;;AAUH,MAAM,sBAAsB,WAC1B,OAAO,MAAM,QAAQ;CACnB,cAAc;CACd,SAAS,MAAO,MAAM,eAAgB,eAA0B;CACjE,CAAC;AA0BJ,MAAa,mBAAgC;CAC3C,MAAM;CACN,WAAW;CACX,eAAe,EAAE;CACjB,cAAc,OAAO,MAAM;CAC3B,OAAO,EAAE;CACV;AAuBD,MAAM,cAAc,MAAmD,UAAU;AAEjF,MAAM,oBAAoB,MACxB,EAAE,KAAK,WAAW,IAAI,EAAE,GAAG,CAAC;CAAE,MAAM,EAAE,YAAY,OAAO,cAAc;CAAQ,MAAM,EAAE;CAAM,CAAC;AAEhG,MAAM,4BAA4B,MAAsD,CACtF;CACE,MAAM;CACN,IAAI,OAAO,cAAc,EAAE,aAAa,GAAG;CAC3C,MAAM,EAAE,aAAa;CACrB,MAAM,EAAE,aAAa,QAAQ,EAAE;CAChC,CACF;AAED,MAAM,oBAAoB,MACxB,WAAW,EAAE,GAAG,iBAAiB,EAAE,GAAG,yBAAyB,EAAE;AAEnE,MAAM,cAAc,UAClB,KAAK,MAAM,aAAa,IAAI,SAAS,SAAS,EAAE,EAAEA,QAAI,QAAQ,iBAAiB,CAAC;AAElF,MAAM,cAAc,OAAiC,SACnD,KACE,OACAA,QAAI,WAAW,MAAO,EAAE,SAAS,OAAO,OAAO,QAAQ,EAAE,KAAK,GAAG,OAAO,SAAU,CACnF,CAAC,KAAK,GAAG;AAEZ,MAAM,wBACJ,UAEA,KACE,OACAA,QAAI,WAAW,MAAO,EAAE,SAAS,kBAAkB,OAAO,QAAQ,EAAE,GAAG,OAAO,SAAU,CACzF;AAEH,MAAM,cACJ,MACA,SAEA,SAAS,KAAA,IACL,OACA;CACE,GAAG;CACH,GAAI,KAAK,qBAAqB,KAAA,KAAa,EAAE,cAAc,KAAK,kBAAkB;CAClF,GAAI,KAAK,yBAAyB,KAAA,KAAa,EAC7C,eAAe,KAAK,sBACrB;CACD,GAAI,KAAK,oBAAoB,KAAA,KAAa,EAAE,cAAc,KAAK,iBAAiB;CAChF,GAAI,KAAK,4BAA4B,KAAA,KAAa,EAChD,sBAAsB,EAAE,eAAe,KAAK,yBAAyB,EACtE;CACD,GAAI,KAAK,uBAAuB,KAAA,KAAa,EAC3C,uBAAuB,EAAE,kBAAkB,KAAK,oBAAoB,EACrE;CACF;;;;;;AAOP,MAAM,oBACJ,MACA,eAEA,OAAO,MAAM,KAAK,IAAI;CACpB,SAAS,OAAO;CAChB,cAAc,GAAG,KAAK,KAAK,GAAG,WAAW;CAC1C,CAAC;AAEJ,MAAM,0BACJ,OACA,UAC6B;CAC7B,QAAQ,iBAAiB,MAAM,MAAM;CACrC,MAAM,KAAK;CACX,YAAY,KAAK;CAGjB,WAAW,KAAK,UAAU,KAAK,KAAK;CACrC;AAED,MAAM,uBACJ,OACA,cAEA,UAAU,QACP,KAAK,SAAS,CAAC,GAAG,KAAK,uBAAuB,KAAK,KAAK,CAAC,EAC1D,MACD;AAEH,MAAa,eAAe,KAAkB,UAAkC;CAC9E,MAAM,QAAQ,WAAW,MAAM;CAC/B,MAAM,eAAe,OAAO,cAAc,MAAM,aAAa,IAAI,aAAa;AAC9E,QAAO;EACL;EACA,UAAU,OAAO,OAAO,aAAa;EACrC,aAAa;GACX,MAAM,IAAI,OAAO,WAAW,OAAO,OAAO;GAC1C,WAAW,IAAI,YAAY,WAAW,OAAO,YAAY;GACzD,eAAe,oBAAoB,IAAI,eAAe,qBAAqB,MAAM,CAAC;GAClF,cAAc,OAAO,OAAO,oBAAoB,IAAI,aAAa;GACjE,OAAO,WAAW,IAAI,OAAO,MAAM,cAAc;GAClD;EACF;;AAGH,MAAM,kBAAkB,QACtB,IAAI,UAAU,SAAS,IAAI,CAAC;CAAE,MAAM;CAAa,SAAS,IAAI;CAAW,CAAC,GAAG,EAAE;AAEjF,MAAM,yBAAyB,QAC7B,IAAI,KAAK,WAAW,IAChB,EAAE,GACF,CACE;CACE,MAAM;CACN,MAAM;CACN,SAAS,CAAC;EAAE,MAAM;EAAe,MAAM,IAAI;EAAM,CAAC;CACnD,CACF;AAEP,MAAM,qBAAqB,QACzB,KACE,IAAI,eACJA,QAAI,KAAK,OAAO;CACd,MAAM;CACN,SAAS,EAAE;CACX,MAAM,EAAE;CACR,WAAW,EAAE;CACb,GAAI,OAAO,OAAO,EAAE,WAAW,IAAI,EAAE,cAAc,EAAE,QAAQ,EAAE,IAAI,EAAE,WAAW,OAAO,EAAE,EAAE;CAC5F,EAAE,CACJ;AAEH,MAAa,qBAAqB,SAA4B;CAC5D,aACE,IAAI,cAAc,SAAS,IAAK,eAAyB,mBAAmB,IAAI,aAAa;CAC/F,OAAO,EAAE,GAAG,IAAI,OAAO;CACvB,OAAO;EAAC,GAAG,eAAe,IAAI;EAAE,GAAG,sBAAsB,IAAI;EAAE,GAAG,kBAAkB,IAAI;EAAC;CAC1F"}
+{"version":3,"file":"codec.mjs","names":["Arr"],"sources":["../src/codec.ts"],"sourcesContent":["import { Array as Arr, Encoding, Match, Option, Result, Schema, pipe } from \"effect\"\nimport type {\n  ContentBlock,\n  ToolCall,\n  InputImage,\n  HistoryItem,\n  Message,\n} from \"@effect-uai/core/Items\"\nimport type { ToolDescriptor } from \"@effect-uai/core/Tool\"\nimport type { Turn } from \"@effect-uai/core/Turn\"\nimport type { CommonRequest } from \"@effect-uai/core/LanguageModel\"\n\n// ---------------------------------------------------------------------------\n// Wire schemas - the subset of Gemini's streamGenerateContent payload we\n// consume. Reference: https://ai.google.dev/api/generate-content\n// ---------------------------------------------------------------------------\n\nconst TextPart = Schema.Struct({\n  text: Schema.String,\n  /**\n   * Gemini's flag for chain-of-thought parts. When `true`, the part is\n   * reasoning trace, not the model's user-facing answer. Maps onto\n   * `reasoning_delta { kind: \"trace\" }` in the canonical view.\n   */\n  thought: Schema.optional(Schema.Boolean),\n})\n\n/**\n * Gemini function-call part. Args are delivered as a whole JSON *object*\n * in a single chunk (Gemini does not stream tool-call args). On Gemini 3\n * the response also carries an `id` per call, which we echo back on the\n * corresponding `functionResponse`.\n */\nconst FunctionCallPart = Schema.Struct({\n  functionCall: Schema.Struct({\n    id: Schema.optional(Schema.String),\n    name: Schema.String,\n    args: Schema.optional(Schema.Unknown),\n  }),\n})\n\nconst Part = Schema.Union([TextPart, FunctionCallPart])\ntype WireFunctionCallPart = typeof FunctionCallPart.Type\n\nconst Content = Schema.Struct({\n  role: Schema.optional(Schema.String),\n  parts: Schema.optional(Schema.Array(Part)),\n})\n\nconst Candidate = Schema.Struct({\n  content: Schema.optional(Content),\n  finishReason: Schema.optional(Schema.String),\n  index: Schema.optional(Schema.Number),\n})\n\nconst UsageMetadata = Schema.Struct({\n  promptTokenCount: Schema.optional(Schema.Number),\n  candidatesTokenCount: Schema.optional(Schema.Number),\n  totalTokenCount: Schema.optional(Schema.Number),\n  cachedContentTokenCount: Schema.optional(Schema.Number),\n  thoughtsTokenCount: Schema.optional(Schema.Number),\n})\n\nexport const WireChunk = Schema.Struct({\n  candidates: Schema.optional(Schema.Array(Candidate)),\n  usageMetadata: Schema.optional(UsageMetadata),\n})\nexport type WireChunk = typeof WireChunk.Type\n\n// ---------------------------------------------------------------------------\n// History → request body\n// ---------------------------------------------------------------------------\n\ntype RequestPart =\n  | { readonly text: string }\n  | { readonly inlineData: { readonly mimeType: string; readonly data: string } }\n  | { readonly functionCall: { readonly name: string; readonly args: unknown } }\n  | {\n      readonly functionResponse: {\n        readonly name: string\n        readonly response: Record<string, unknown>\n      }\n    }\n\ntype RequestContent = {\n  readonly role: \"user\" | \"model\"\n  readonly parts: ReadonlyArray<RequestPart>\n}\n\ntype RequestSystemInstruction = {\n  readonly parts: ReadonlyArray<{ readonly text: string }>\n}\n\n/** Gemini's tool declaration. We translate one `ToolDescriptor` per entry. */\ntype RequestTool = {\n  readonly functionDeclarations: ReadonlyArray<{\n    readonly name: string\n    readonly description: string\n    readonly parameters: Record<string, unknown>\n  }>\n}\n\ntype RequestToolConfig = {\n  readonly functionCallingConfig: {\n    readonly mode: \"AUTO\" | \"ANY\" | \"NONE\"\n    readonly allowedFunctionNames?: ReadonlyArray<string>\n  }\n}\n\nexport type ThinkingConfig = {\n  readonly thinkingBudget: number\n}\n\nexport type GenerationConfig = {\n  readonly temperature?: number\n  readonly maxOutputTokens?: number\n  readonly topP?: number\n  readonly thinkingConfig?: ThinkingConfig\n  /** Set together with `responseJsonSchema` to constrain output to JSON. */\n  readonly responseMimeType?: string\n  /** JSON Schema constraint on the output. */\n  readonly responseJsonSchema?: Record<string, unknown>\n}\n\nexport type RequestBody = {\n  readonly contents: ReadonlyArray<RequestContent>\n  readonly systemInstruction?: RequestSystemInstruction\n  readonly generationConfig?: GenerationConfig\n  readonly tools?: ReadonlyArray<RequestTool>\n  readonly toolConfig?: RequestToolConfig\n}\n\nconst blockText = Match.type<ContentBlock>().pipe(\n  Match.discriminatorsExhaustive(\"type\")({\n    input_text: (b) => b.text,\n    input_image: () => \"\",\n    output_text: (b) => b.text,\n    refusal: (b) => b.text,\n  }),\n)\n\nconst messageText = (message: Message): string => message.content.map(blockText).join(\"\")\n\n/**\n * Gemini's `inlineData` form expects a base64 payload. URL-form images\n * would need to go through Gemini's Files API (upload then `fileData`\n * with the returned URI); pre-uploading isn't free, so we skip those for\n * now and document as a follow-up.\n */\nconst imageSourceToParts = Match.type<InputImage[\"source\"]>().pipe(\n  Match.tag(\"url\", (): ReadonlyArray<RequestPart> => []),\n  Match.tag(\n    \"base64\",\n    (s): ReadonlyArray<RequestPart> => [{ inlineData: { mimeType: s.mimeType, data: s.base64 } }],\n  ),\n  Match.tag(\n    \"bytes\",\n    (s): ReadonlyArray<RequestPart> => [\n      { inlineData: { mimeType: s.mimeType, data: Encoding.encodeBase64(s.bytes) } },\n    ],\n  ),\n  Match.exhaustive,\n)\n\nconst blockToParts = Match.type<ContentBlock>().pipe(\n  Match.discriminatorsExhaustive(\"type\")({\n    input_text: (b): ReadonlyArray<RequestPart> => (b.text.length === 0 ? [] : [{ text: b.text }]),\n    input_image: (b): ReadonlyArray<RequestPart> => imageSourceToParts(b.source),\n    output_text: (b): ReadonlyArray<RequestPart> => (b.text.length === 0 ? [] : [{ text: b.text }]),\n    // Refusals are assistant-side content; they don't round-trip into Gemini's\n    // request body as parts. Skip.\n    refusal: (): ReadonlyArray<RequestPart> => [],\n  }),\n)\n\nconst messageToContent = (message: Message): Result.Result<RequestContent, void> => {\n  const parts = pipe(message.content, Arr.flatMap(blockToParts))\n  if (parts.length === 0) return Result.failVoid\n  return Match.value(message.role).pipe(\n    Match.when(\"user\", () => Result.succeed({ role: \"user\" as const, parts })),\n    Match.when(\"assistant\", () => Result.succeed({ role: \"model\" as const, parts })),\n    Match.when(\"system\", () => Result.failVoid),\n    Match.exhaustive,\n  )\n}\n\nconst systemMessageText = (message: Message): Result.Result<string, void> => {\n  if (message.role !== \"system\") return Result.failVoid\n  const text = messageText(message)\n  return text.length === 0 ? Result.failVoid : Result.succeed(text)\n}\n\nconst allMessages = (history: ReadonlyArray<HistoryItem>): ReadonlyArray<Message> =>\n  pipe(\n    history,\n    Arr.filterMap((item) => (item.type === \"message\" ? Result.succeed(item) : Result.failVoid)),\n  )\n\n// ---------------------------------------------------------------------------\n// Function-call round-trip\n//\n// `ToolCall.arguments` and `ToolCallOutput.output` are JSON-encoded\n// strings; Gemini's `functionCall.args` and `functionResponse.response`\n// expect parsed JSON *objects*. Decode via Schema:\n//   - object payload → use as-is\n//   - scalar / array / malformed → wrap as `{ output: <raw string> }` so the\n//     model still sees *some* response without crashing the request.\n// ---------------------------------------------------------------------------\n\nconst JsonObject = Schema.Record(Schema.String, Schema.Unknown)\nconst decodeJsonObject = Schema.decodeResult(Schema.fromJsonString(JsonObject))\n\nconst parsedJsonObject =\n  (fallback: (raw: string) => Record<string, unknown>) =>\n  (encoded: string): Record<string, unknown> =>\n    pipe(\n      decodeJsonObject(encoded),\n      Result.match({\n        onSuccess: (v) => v,\n        onFailure: () => fallback(encoded),\n      }),\n    )\n\nconst parsedArgs = parsedJsonObject(() => ({}))\nconst parsedResponse = parsedJsonObject((raw) => ({ output: raw }))\n\n/**\n * `ToolCallOutput` only carries `call_id`; Gemini's `functionResponse`\n * requires the declared function `name`. Resolve the name by scanning prior\n * `function_call` items in the history for a matching `call_id`. If we\n * cannot resolve, fall back to `call_id` as the name - imperfect but\n * preserves stream shape so the model sees *some* response.\n */\nconst isFunctionCallItem = (item: HistoryItem): item is ToolCall => item.type === \"function_call\"\n\nconst nameForCallId = (\n  history: ReadonlyArray<HistoryItem>,\n  call_id: string,\n): Option.Option<string> =>\n  pipe(\n    history,\n    Arr.findFirst((item): item is ToolCall => isFunctionCallItem(item) && item.call_id === call_id),\n    Option.map((f) => f.name),\n  )\n\n/**\n * Extract the Gemini-3 wire id we stashed in `providerData` on the way out.\n * Schema-driven so the shape lives in one place; failure → `Option.none()`.\n */\nconst ProviderDataWithGeminiId = Schema.Struct({\n  gemini: Schema.Struct({ id: Schema.String }),\n})\nconst decodeProviderId = Schema.decodeUnknownResult(ProviderDataWithGeminiId)\n\nconst providerIdFor = (item: ToolCall): Option.Option<string> =>\n  pipe(\n    decodeProviderId(item.providerData),\n    Result.match({\n      onSuccess: (d) => Option.some(d.gemini.id),\n      onFailure: () => Option.none<string>(),\n    }),\n  )\n\nconst itemToContent =\n  (history: ReadonlyArray<HistoryItem>) =>\n  (item: HistoryItem): Result.Result<RequestContent, void> =>\n    Match.value(item).pipe(\n      Match.discriminatorsExhaustive(\"type\")({\n        message: messageToContent,\n        function_call: (f) =>\n          Result.succeed({\n            role: \"model\" as const,\n            parts: [\n              {\n                functionCall: {\n                  ...Option.match(providerIdFor(f), {\n                    onSome: (id) => ({ id }),\n                    onNone: () => ({}),\n                  }),\n                  name: f.name,\n                  args: parsedArgs(f.arguments),\n                },\n              },\n            ],\n          }),\n        function_call_output: (o) =>\n          Result.succeed({\n            role: \"user\" as const,\n            parts: [\n              {\n                functionResponse: {\n                  name: Option.getOrElse(nameForCallId(history, o.call_id), () => o.call_id),\n                  response: parsedResponse(o.output),\n                },\n              },\n            ],\n          }),\n        reasoning: () => Result.failVoid,\n      }),\n    )\n\n// ---------------------------------------------------------------------------\n// Tool descriptors → Gemini `functionDeclarations`. Gemini accepts only a\n// strict OpenAPI 3.0 subset for `parameters`; strip JSON-Schema keys it\n// rejects (`$schema`, `$ref`, `additionalProperties`, `oneOf`,\n// `definitions`).\n// ---------------------------------------------------------------------------\n\nconst UNSUPPORTED_SCHEMA_KEYS: ReadonlySet<string> = new Set([\n  \"$schema\",\n  \"$ref\",\n  \"$defs\",\n  \"definitions\",\n  \"additionalProperties\",\n  \"oneOf\",\n])\n\nconst sanitizeSchema = (schema: unknown): unknown => {\n  if (Array.isArray(schema)) return schema.map(sanitizeSchema)\n  if (schema === null || typeof schema !== \"object\") return schema\n  return pipe(\n    Object.entries(schema as Record<string, unknown>),\n    Arr.filterMap(([k, v]) =>\n      UNSUPPORTED_SCHEMA_KEYS.has(k)\n        ? Result.failVoid\n        : Result.succeed([k, sanitizeSchema(v)] as const),\n    ),\n    Object.fromEntries,\n  )\n}\n\nconst toolDescriptorsToTools = (tools: ReadonlyArray<ToolDescriptor>): ReadonlyArray<RequestTool> =>\n  tools.length === 0\n    ? []\n    : [\n        {\n          functionDeclarations: tools.map((t) => ({\n            name: t.name,\n            description: t.description,\n            parameters: sanitizeSchema(t.inputSchema) as Record<string, unknown>,\n          })),\n        },\n      ]\n\n/**\n * Map our cross-provider `toolChoice` onto Gemini's `functionCallingConfig`.\n * Gemini honours the same three modes plus a function allow-list, so the\n * choice is mapped, not dropped. `undefined` keeps the API default (`AUTO`).\n */\nconst toolChoiceToFunctionCallingConfig = (\n  toolChoice: CommonRequest[\"toolChoice\"],\n): RequestToolConfig[\"functionCallingConfig\"] =>\n  Match.value(toolChoice ?? \"auto\").pipe(\n    Match.when(\"auto\", () => ({ mode: \"AUTO\" as const })),\n    Match.when(\"required\", () => ({ mode: \"ANY\" as const })),\n    Match.when(\"none\", () => ({ mode: \"NONE\" as const })),\n    Match.orElse((tc) => ({ mode: \"ANY\" as const, allowedFunctionNames: [tc.name] })),\n  )\n\n/**\n * URL-form images need Gemini's Files API (upload, then `fileData`); we\n * don't pre-upload, so such a part would otherwise vanish from the request.\n * The caller fails the request with `Unsupported` instead of silently\n * dropping the image.\n */\nexport const hasUrlImageSource = (history: ReadonlyArray<HistoryItem>): boolean =>\n  history.some(\n    (item) =>\n      item.type === \"message\" &&\n      item.content.some((b) => b.type === \"input_image\" && b.source._tag === \"url\"),\n  )\n\nexport const buildRequestBody = (\n  history: ReadonlyArray<HistoryItem>,\n  generationConfig: Option.Option<GenerationConfig>,\n  tools: ReadonlyArray<ToolDescriptor> = [],\n  toolChoice: CommonRequest[\"toolChoice\"] = undefined,\n): RequestBody => {\n  const systemTexts = pipe(allMessages(history), Arr.filterMap(systemMessageText))\n  const contents = pipe(history, Arr.filterMap(itemToContent(history)))\n  const requestTools = toolDescriptorsToTools(tools)\n  return {\n    contents,\n    ...(systemTexts.length > 0 && {\n      systemInstruction: { parts: [{ text: systemTexts.join(\"\\n\") }] },\n    }),\n    ...Option.match(generationConfig, {\n      onNone: () => ({}),\n      onSome: (cfg) => ({ generationConfig: cfg }),\n    }),\n    ...(requestTools.length > 0 && {\n      tools: requestTools,\n      toolConfig: { functionCallingConfig: toolChoiceToFunctionCallingConfig(toolChoice) },\n    }),\n  }\n}\n\n// ---------------------------------------------------------------------------\n// Stream-level state - accumulate chunk text + final usage/finish.\n//\n// `Accumulator` is immutable; `ingestChunk` returns a fresh one per chunk.\n// Drive it via `Stream.mapAccum` in the consumer.\n// ---------------------------------------------------------------------------\n\nconst finishReasonToStop = (reason: Option.Option<string>): Turn[\"stop_reason\"] =>\n  Option.match(reason, {\n    onNone: () => \"stop\" as const,\n    onSome: (r) => (r === \"MAX_TOKENS\" ? (\"max_tokens\" as const) : (\"stop\" as const)),\n  })\n\nexport type AccumulatedFunctionCall = {\n  /** Synthesized id-or-name we surface as `call_id` on the canonical item. */\n  readonly callId: string\n  readonly name: string\n  /** Wire id from Gemini 3, when present - echoed back on `functionResponse`. */\n  readonly providerId: Option.Option<string>\n  /** Args as JSON-encoded string, mirroring `Items.ToolCall.arguments`. */\n  readonly arguments: string\n}\n\nexport type Accumulator = {\n  readonly text: string\n  readonly reasoning: string\n  readonly functionCalls: ReadonlyArray<AccumulatedFunctionCall>\n  readonly finishReason: Option.Option<string>\n  readonly usage: {\n    readonly input_tokens?: number\n    readonly output_tokens?: number\n    readonly total_tokens?: number\n    readonly input_tokens_details?: { readonly cached_tokens?: number }\n    readonly output_tokens_details?: { readonly reasoning_tokens?: number }\n  }\n}\n\nexport const emptyAccumulator: Accumulator = {\n  text: \"\",\n  reasoning: \"\",\n  functionCalls: [],\n  finishReason: Option.none(),\n  usage: {},\n}\n\n/**\n * One part's worth of streamable output. `text` and `reasoning` are\n * incremental string deltas; `function_call` arrives whole-in-one-chunk\n * (Gemini does not stream tool-call args).\n */\nexport type ChunkPart =\n  | { readonly kind: \"text\"; readonly text: string }\n  | { readonly kind: \"reasoning\"; readonly text: string }\n  | {\n      readonly kind: \"function_call\"\n      readonly id: Option.Option<string>\n      readonly name: string\n      readonly args: unknown\n    }\n\nexport type ChunkResult = {\n  readonly accumulator: Accumulator\n  readonly parts: ReadonlyArray<ChunkPart>\n  readonly finished: boolean\n}\n\nconst isTextPart = (p: typeof Part.Type): p is typeof TextPart.Type => \"text\" in p\n\nconst textToChunkParts = (p: typeof TextPart.Type): ReadonlyArray<ChunkPart> =>\n  p.text.length === 0 ? [] : [{ kind: p.thought === true ? \"reasoning\" : \"text\", text: p.text }]\n\nconst functionCallToChunkParts = (p: WireFunctionCallPart): ReadonlyArray<ChunkPart> => [\n  {\n    kind: \"function_call\",\n    id: Option.fromNullishOr(p.functionCall.id),\n    name: p.functionCall.name,\n    args: p.functionCall.args ?? {},\n  },\n]\n\nconst partToChunkParts = (p: typeof Part.Type): ReadonlyArray<ChunkPart> =>\n  isTextPart(p) ? textToChunkParts(p) : functionCallToChunkParts(p)\n\nconst chunkParts = (chunk: WireChunk): ReadonlyArray<ChunkPart> =>\n  pipe(chunk.candidates?.[0]?.content?.parts ?? [], Arr.flatMap(partToChunkParts))\n\nconst sumStrings = (parts: ReadonlyArray<ChunkPart>, kind: \"text\" | \"reasoning\"): string =>\n  pipe(\n    parts,\n    Arr.filterMap((p) => (p.kind === kind ? Result.succeed(p.text) : Result.failVoid)),\n  ).join(\"\")\n\nconst collectFunctionCalls = (\n  parts: ReadonlyArray<ChunkPart>,\n): ReadonlyArray<Extract<ChunkPart, { kind: \"function_call\" }>> =>\n  pipe(\n    parts,\n    Arr.filterMap((p) => (p.kind === \"function_call\" ? Result.succeed(p) : Result.failVoid)),\n  )\n\nconst mergeUsage = (\n  prev: Accumulator[\"usage\"],\n  next: WireChunk[\"usageMetadata\"],\n): Accumulator[\"usage\"] =>\n  next === undefined\n    ? prev\n    : {\n        ...prev,\n        ...(next.promptTokenCount !== undefined && { input_tokens: next.promptTokenCount }),\n        ...(next.candidatesTokenCount !== undefined && {\n          output_tokens: next.candidatesTokenCount,\n        }),\n        ...(next.totalTokenCount !== undefined && { total_tokens: next.totalTokenCount }),\n        ...(next.cachedContentTokenCount !== undefined && {\n          input_tokens_details: { cached_tokens: next.cachedContentTokenCount },\n        }),\n        ...(next.thoughtsTokenCount !== undefined && {\n          output_tokens_details: { reasoning_tokens: next.thoughtsTokenCount },\n        }),\n      }\n\n/**\n * Synthesize a stable `call_id` for a function call. Gemini 3 provides one\n * via `functionCall.id`; older models do not, so we fall back to\n * `<name>_<index>` based on prior calls' position in the accumulator.\n */\nconst synthesizeCallId = (\n  call: Extract<ChunkPart, { kind: \"function_call\" }>,\n  priorCalls: ReadonlyArray<AccumulatedFunctionCall>,\n): string =>\n  Option.match(call.id, {\n    onSome: (id) => id,\n    onNone: () => `${call.name}_${priorCalls.length}`,\n  })\n\nconst chunkCallToAccumulated = (\n  prior: ReadonlyArray<AccumulatedFunctionCall>,\n  call: Extract<ChunkPart, { kind: \"function_call\" }>,\n): AccumulatedFunctionCall => ({\n  callId: synthesizeCallId(call, prior),\n  name: call.name,\n  providerId: call.id,\n  // `functionCallToChunkParts` already replaced null/undefined with `{}`,\n  // so `call.args` is always a JSON-encodable value here.\n  arguments: JSON.stringify(call.args),\n})\n\nconst appendFunctionCalls = (\n  prior: ReadonlyArray<AccumulatedFunctionCall>,\n  fromChunk: ReadonlyArray<Extract<ChunkPart, { kind: \"function_call\" }>>,\n): ReadonlyArray<AccumulatedFunctionCall> =>\n  fromChunk.reduce<ReadonlyArray<AccumulatedFunctionCall>>(\n    (acc, call) => [...acc, chunkCallToAccumulated(acc, call)],\n    prior,\n  )\n\nexport const ingestChunk = (acc: Accumulator, chunk: WireChunk): ChunkResult => {\n  const parts = chunkParts(chunk)\n  const finishReason = Option.fromNullishOr(chunk.candidates?.[0]?.finishReason)\n  return {\n    parts,\n    finished: Option.isSome(finishReason),\n    accumulator: {\n      text: acc.text + sumStrings(parts, \"text\"),\n      reasoning: acc.reasoning + sumStrings(parts, \"reasoning\"),\n      functionCalls: appendFunctionCalls(acc.functionCalls, collectFunctionCalls(parts)),\n      finishReason: Option.orElse(finishReason, () => acc.finishReason),\n      usage: mergeUsage(acc.usage, chunk.usageMetadata),\n    },\n  }\n}\n\nconst reasoningItems = (acc: Accumulator): ReadonlyArray<HistoryItem> =>\n  acc.reasoning.length > 0 ? [{ type: \"reasoning\", summary: acc.reasoning }] : []\n\nconst assistantMessageItems = (acc: Accumulator): ReadonlyArray<HistoryItem> =>\n  acc.text.length === 0\n    ? []\n    : [\n        {\n          type: \"message\",\n          role: \"assistant\",\n          content: [{ type: \"output_text\", text: acc.text }],\n        },\n      ]\n\nconst functionCallItems = (acc: Accumulator): ReadonlyArray<HistoryItem> =>\n  pipe(\n    acc.functionCalls,\n    Arr.map((c) => ({\n      type: \"function_call\" as const,\n      call_id: c.callId,\n      name: c.name,\n      arguments: c.arguments,\n      ...(Option.isSome(c.providerId) && { providerData: { gemini: { id: c.providerId.value } } }),\n    })),\n  )\n\nexport const accumulatorToTurn = (acc: Accumulator): Turn => ({\n  stop_reason:\n    acc.functionCalls.length > 0 ? (\"tool_calls\" as const) : finishReasonToStop(acc.finishReason),\n  usage: { ...acc.usage },\n  items: [...reasoningItems(acc), ...assistantMessageItems(acc), ...functionCallItems(acc)],\n})\n"],"mappings":";;;;;;;;;;;AAiBA,MAAM,WAAW,OAAO,OAAO;CAC7B,MAAM,OAAO;;;;;;CAMb,SAAS,OAAO,SAAS,OAAO,QAAQ;CACzC,CAAC;;;;;;;AAQF,MAAM,mBAAmB,OAAO,OAAO,EACrC,cAAc,OAAO,OAAO;CAC1B,IAAI,OAAO,SAAS,OAAO,OAAO;CAClC,MAAM,OAAO;CACb,MAAM,OAAO,SAAS,OAAO,QAAQ;CACtC,CAAC,EACH,CAAC;AAEF,MAAM,OAAO,OAAO,MAAM,CAAC,UAAU,iBAAiB,CAAC;AAGvD,MAAM,UAAU,OAAO,OAAO;CAC5B,MAAM,OAAO,SAAS,OAAO,OAAO;CACpC,OAAO,OAAO,SAAS,OAAO,MAAM,KAAK,CAAC;CAC3C,CAAC;AAEF,MAAM,YAAY,OAAO,OAAO;CAC9B,SAAS,OAAO,SAAS,QAAQ;CACjC,cAAc,OAAO,SAAS,OAAO,OAAO;CAC5C,OAAO,OAAO,SAAS,OAAO,OAAO;CACtC,CAAC;AAEF,MAAM,gBAAgB,OAAO,OAAO;CAClC,kBAAkB,OAAO,SAAS,OAAO,OAAO;CAChD,sBAAsB,OAAO,SAAS,OAAO,OAAO;CACpD,iBAAiB,OAAO,SAAS,OAAO,OAAO;CAC/C,yBAAyB,OAAO,SAAS,OAAO,OAAO;CACvD,oBAAoB,OAAO,SAAS,OAAO,OAAO;CACnD,CAAC;AAEF,MAAa,YAAY,OAAO,OAAO;CACrC,YAAY,OAAO,SAAS,OAAO,MAAM,UAAU,CAAC;CACpD,eAAe,OAAO,SAAS,cAAc;CAC9C,CAAC;AAkEF,MAAM,YAAY,MAAM,MAAoB,CAAC,KAC3C,MAAM,yBAAyB,OAAO,CAAC;CACrC,aAAa,MAAM,EAAE;CACrB,mBAAmB;CACnB,cAAc,MAAM,EAAE;CACtB,UAAU,MAAM,EAAE;CACnB,CAAC,CACH;AAED,MAAM,eAAe,YAA6B,QAAQ,QAAQ,IAAI,UAAU,CAAC,KAAK,GAAG;;;;;;;AAQzF,MAAM,qBAAqB,MAAM,MAA4B,CAAC,KAC5D,MAAM,IAAI,aAAyC,EAAE,CAAC,EACtD,MAAM,IACJ,WACC,MAAkC,CAAC,EAAE,YAAY;CAAE,UAAU,EAAE;CAAU,MAAM,EAAE;CAAQ,EAAE,CAAC,CAC9F,EACD,MAAM,IACJ,UACC,MAAkC,CACjC,EAAE,YAAY;CAAE,UAAU,EAAE;CAAU,MAAM,SAAS,aAAa,EAAE,MAAM;CAAE,EAAE,CAC/E,CACF,EACD,MAAM,WACP;AAED,MAAM,eAAe,MAAM,MAAoB,CAAC,KAC9C,MAAM,yBAAyB,OAAO,CAAC;CACrC,aAAa,MAAmC,EAAE,KAAK,WAAW,IAAI,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC;CAC7F,cAAc,MAAkC,mBAAmB,EAAE,OAAO;CAC5E,cAAc,MAAmC,EAAE,KAAK,WAAW,IAAI,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC;CAG9F,eAA2C,EAAE;CAC9C,CAAC,CACH;AAED,MAAM,oBAAoB,YAA0D;CAClF,MAAM,QAAQ,KAAK,QAAQ,SAASA,QAAI,QAAQ,aAAa,CAAC;AAC9D,KAAI,MAAM,WAAW,EAAG,QAAO,OAAO;AACtC,QAAO,MAAM,MAAM,QAAQ,KAAK,CAAC,KAC/B,MAAM,KAAK,cAAc,OAAO,QAAQ;EAAE,MAAM;EAAiB;EAAO,CAAC,CAAC,EAC1E,MAAM,KAAK,mBAAmB,OAAO,QAAQ;EAAE,MAAM;EAAkB;EAAO,CAAC,CAAC,EAChF,MAAM,KAAK,gBAAgB,OAAO,SAAS,EAC3C,MAAM,WACP;;AAGH,MAAM,qBAAqB,YAAkD;AAC3E,KAAI,QAAQ,SAAS,SAAU,QAAO,OAAO;CAC7C,MAAM,OAAO,YAAY,QAAQ;AACjC,QAAO,KAAK,WAAW,IAAI,OAAO,WAAW,OAAO,QAAQ,KAAK;;AAGnE,MAAM,eAAe,YACnB,KACE,SACAA,QAAI,WAAW,SAAU,KAAK,SAAS,YAAY,OAAO,QAAQ,KAAK,GAAG,OAAO,SAAU,CAC5F;AAaH,MAAM,aAAa,OAAO,OAAO,OAAO,QAAQ,OAAO,QAAQ;AAC/D,MAAM,mBAAmB,OAAO,aAAa,OAAO,eAAe,WAAW,CAAC;AAE/E,MAAM,oBACH,cACA,YACC,KACE,iBAAiB,QAAQ,EACzB,OAAO,MAAM;CACX,YAAY,MAAM;CAClB,iBAAiB,SAAS,QAAQ;CACnC,CAAC,CACH;AAEL,MAAM,aAAa,wBAAwB,EAAE,EAAE;AAC/C,MAAM,iBAAiB,kBAAkB,SAAS,EAAE,QAAQ,KAAK,EAAE;;;;;;;;AASnE,MAAM,sBAAsB,SAAwC,KAAK,SAAS;AAElF,MAAM,iBACJ,SACA,YAEA,KACE,SACAA,QAAI,WAAW,SAA2B,mBAAmB,KAAK,IAAI,KAAK,YAAY,QAAQ,EAC/F,OAAO,KAAK,MAAM,EAAE,KAAK,CAC1B;;;;;AAMH,MAAM,2BAA2B,OAAO,OAAO,EAC7C,QAAQ,OAAO,OAAO,EAAE,IAAI,OAAO,QAAQ,CAAC,EAC7C,CAAC;AACF,MAAM,mBAAmB,OAAO,oBAAoB,yBAAyB;AAE7E,MAAM,iBAAiB,SACrB,KACE,iBAAiB,KAAK,aAAa,EACnC,OAAO,MAAM;CACX,YAAY,MAAM,OAAO,KAAK,EAAE,OAAO,GAAG;CAC1C,iBAAiB,OAAO,MAAc;CACvC,CAAC,CACH;AAEH,MAAM,iBACH,aACA,SACC,MAAM,MAAM,KAAK,CAAC,KAChB,MAAM,yBAAyB,OAAO,CAAC;CACrC,SAAS;CACT,gBAAgB,MACd,OAAO,QAAQ;EACb,MAAM;EACN,OAAO,CACL,EACE,cAAc;GACZ,GAAG,OAAO,MAAM,cAAc,EAAE,EAAE;IAChC,SAAS,QAAQ,EAAE,IAAI;IACvB,eAAe,EAAE;IAClB,CAAC;GACF,MAAM,EAAE;GACR,MAAM,WAAW,EAAE,UAAU;GAC9B,EACF,CACF;EACF,CAAC;CACJ,uBAAuB,MACrB,OAAO,QAAQ;EACb,MAAM;EACN,OAAO,CACL,EACE,kBAAkB;GAChB,MAAM,OAAO,UAAU,cAAc,SAAS,EAAE,QAAQ,QAAQ,EAAE,QAAQ;GAC1E,UAAU,eAAe,EAAE,OAAO;GACnC,EACF,CACF;EACF,CAAC;CACJ,iBAAiB,OAAO;CACzB,CAAC,CACH;AASL,MAAM,0BAA+C,IAAI,IAAI;CAC3D;CACA;CACA;CACA;CACA;CACA;CACD,CAAC;AAEF,MAAM,kBAAkB,WAA6B;AACnD,KAAI,MAAM,QAAQ,OAAO,CAAE,QAAO,OAAO,IAAI,eAAe;AAC5D,KAAI,WAAW,QAAQ,OAAO,WAAW,SAAU,QAAO;AAC1D,QAAO,KACL,OAAO,QAAQ,OAAkC,EACjDA,QAAI,WAAW,CAAC,GAAG,OACjB,wBAAwB,IAAI,EAAE,GAC1B,OAAO,WACP,OAAO,QAAQ,CAAC,GAAG,eAAe,EAAE,CAAC,CAAU,CACpD,EACD,OAAO,YACR;;AAGH,MAAM,0BAA0B,UAC9B,MAAM,WAAW,IACb,EAAE,GACF,CACE,EACE,sBAAsB,MAAM,KAAK,OAAO;CACtC,MAAM,EAAE;CACR,aAAa,EAAE;CACf,YAAY,eAAe,EAAE,YAAY;CAC1C,EAAE,EACJ,CACF;;;;;;AAOP,MAAM,qCACJ,eAEA,MAAM,MAAM,cAAc,OAAO,CAAC,KAChC,MAAM,KAAK,eAAe,EAAE,MAAM,QAAiB,EAAE,EACrD,MAAM,KAAK,mBAAmB,EAAE,MAAM,OAAgB,EAAE,EACxD,MAAM,KAAK,eAAe,EAAE,MAAM,QAAiB,EAAE,EACrD,MAAM,QAAQ,QAAQ;CAAE,MAAM;CAAgB,sBAAsB,CAAC,GAAG,KAAK;CAAE,EAAE,CAClF;;;;;;;AAQH,MAAa,qBAAqB,YAChC,QAAQ,MACL,SACC,KAAK,SAAS,aACd,KAAK,QAAQ,MAAM,MAAM,EAAE,SAAS,iBAAiB,EAAE,OAAO,SAAS,MAAM,CAChF;AAEH,MAAa,oBACX,SACA,kBACA,QAAuC,EAAE,EACzC,aAA0C,KAAA,MAC1B;CAChB,MAAM,cAAc,KAAK,YAAY,QAAQ,EAAEA,QAAI,UAAU,kBAAkB,CAAC;CAChF,MAAM,WAAW,KAAK,SAASA,QAAI,UAAU,cAAc,QAAQ,CAAC,CAAC;CACrE,MAAM,eAAe,uBAAuB,MAAM;AAClD,QAAO;EACL;EACA,GAAI,YAAY,SAAS,KAAK,EAC5B,mBAAmB,EAAE,OAAO,CAAC,EAAE,MAAM,YAAY,KAAK,KAAK,EAAE,CAAC,EAAE,EACjE;EACD,GAAG,OAAO,MAAM,kBAAkB;GAChC,eAAe,EAAE;GACjB,SAAS,SAAS,EAAE,kBAAkB,KAAK;GAC5C,CAAC;EACF,GAAI,aAAa,SAAS,KAAK;GAC7B,OAAO;GACP,YAAY,EAAE,uBAAuB,kCAAkC,WAAW,EAAE;GACrF;EACF;;AAUH,MAAM,sBAAsB,WAC1B,OAAO,MAAM,QAAQ;CACnB,cAAc;CACd,SAAS,MAAO,MAAM,eAAgB,eAA0B;CACjE,CAAC;AA0BJ,MAAa,mBAAgC;CAC3C,MAAM;CACN,WAAW;CACX,eAAe,EAAE;CACjB,cAAc,OAAO,MAAM;CAC3B,OAAO,EAAE;CACV;AAuBD,MAAM,cAAc,MAAmD,UAAU;AAEjF,MAAM,oBAAoB,MACxB,EAAE,KAAK,WAAW,IAAI,EAAE,GAAG,CAAC;CAAE,MAAM,EAAE,YAAY,OAAO,cAAc;CAAQ,MAAM,EAAE;CAAM,CAAC;AAEhG,MAAM,4BAA4B,MAAsD,CACtF;CACE,MAAM;CACN,IAAI,OAAO,cAAc,EAAE,aAAa,GAAG;CAC3C,MAAM,EAAE,aAAa;CACrB,MAAM,EAAE,aAAa,QAAQ,EAAE;CAChC,CACF;AAED,MAAM,oBAAoB,MACxB,WAAW,EAAE,GAAG,iBAAiB,EAAE,GAAG,yBAAyB,EAAE;AAEnE,MAAM,cAAc,UAClB,KAAK,MAAM,aAAa,IAAI,SAAS,SAAS,EAAE,EAAEA,QAAI,QAAQ,iBAAiB,CAAC;AAElF,MAAM,cAAc,OAAiC,SACnD,KACE,OACAA,QAAI,WAAW,MAAO,EAAE,SAAS,OAAO,OAAO,QAAQ,EAAE,KAAK,GAAG,OAAO,SAAU,CACnF,CAAC,KAAK,GAAG;AAEZ,MAAM,wBACJ,UAEA,KACE,OACAA,QAAI,WAAW,MAAO,EAAE,SAAS,kBAAkB,OAAO,QAAQ,EAAE,GAAG,OAAO,SAAU,CACzF;AAEH,MAAM,cACJ,MACA,SAEA,SAAS,KAAA,IACL,OACA;CACE,GAAG;CACH,GAAI,KAAK,qBAAqB,KAAA,KAAa,EAAE,cAAc,KAAK,kBAAkB;CAClF,GAAI,KAAK,yBAAyB,KAAA,KAAa,EAC7C,eAAe,KAAK,sBACrB;CACD,GAAI,KAAK,oBAAoB,KAAA,KAAa,EAAE,cAAc,KAAK,iBAAiB;CAChF,GAAI,KAAK,4BAA4B,KAAA,KAAa,EAChD,sBAAsB,EAAE,eAAe,KAAK,yBAAyB,EACtE;CACD,GAAI,KAAK,uBAAuB,KAAA,KAAa,EAC3C,uBAAuB,EAAE,kBAAkB,KAAK,oBAAoB,EACrE;CACF;;;;;;AAOP,MAAM,oBACJ,MACA,eAEA,OAAO,MAAM,KAAK,IAAI;CACpB,SAAS,OAAO;CAChB,cAAc,GAAG,KAAK,KAAK,GAAG,WAAW;CAC1C,CAAC;AAEJ,MAAM,0BACJ,OACA,UAC6B;CAC7B,QAAQ,iBAAiB,MAAM,MAAM;CACrC,MAAM,KAAK;CACX,YAAY,KAAK;CAGjB,WAAW,KAAK,UAAU,KAAK,KAAK;CACrC;AAED,MAAM,uBACJ,OACA,cAEA,UAAU,QACP,KAAK,SAAS,CAAC,GAAG,KAAK,uBAAuB,KAAK,KAAK,CAAC,EAC1D,MACD;AAEH,MAAa,eAAe,KAAkB,UAAkC;CAC9E,MAAM,QAAQ,WAAW,MAAM;CAC/B,MAAM,eAAe,OAAO,cAAc,MAAM,aAAa,IAAI,aAAa;AAC9E,QAAO;EACL;EACA,UAAU,OAAO,OAAO,aAAa;EACrC,aAAa;GACX,MAAM,IAAI,OAAO,WAAW,OAAO,OAAO;GAC1C,WAAW,IAAI,YAAY,WAAW,OAAO,YAAY;GACzD,eAAe,oBAAoB,IAAI,eAAe,qBAAqB,MAAM,CAAC;GAClF,cAAc,OAAO,OAAO,oBAAoB,IAAI,aAAa;GACjE,OAAO,WAAW,IAAI,OAAO,MAAM,cAAc;GAClD;EACF;;AAGH,MAAM,kBAAkB,QACtB,IAAI,UAAU,SAAS,IAAI,CAAC;CAAE,MAAM;CAAa,SAAS,IAAI;CAAW,CAAC,GAAG,EAAE;AAEjF,MAAM,yBAAyB,QAC7B,IAAI,KAAK,WAAW,IAChB,EAAE,GACF,CACE;CACE,MAAM;CACN,MAAM;CACN,SAAS,CAAC;EAAE,MAAM;EAAe,MAAM,IAAI;EAAM,CAAC;CACnD,CACF;AAEP,MAAM,qBAAqB,QACzB,KACE,IAAI,eACJA,QAAI,KAAK,OAAO;CACd,MAAM;CACN,SAAS,EAAE;CACX,MAAM,EAAE;CACR,WAAW,EAAE;CACb,GAAI,OAAO,OAAO,EAAE,WAAW,IAAI,EAAE,cAAc,EAAE,QAAQ,EAAE,IAAI,EAAE,WAAW,OAAO,EAAE,EAAE;CAC5F,EAAE,CACJ;AAEH,MAAa,qBAAqB,SAA4B;CAC5D,aACE,IAAI,cAAc,SAAS,IAAK,eAAyB,mBAAmB,IAAI,aAAa;CAC/F,OAAO,EAAE,GAAG,IAAI,OAAO;CACvB,OAAO;EAAC,GAAG,eAAe,IAAI;EAAE,GAAG,sBAAsB,IAAI;EAAE,GAAG,kBAAkB,IAAI;EAAC;CAC1F"}

package/dist/index.d.mts

@@ -1,8 +1,7 @@
 import { t as codec_d_exports } from "./codec.mjs";
-import { GeminiSttModel, GeminiTtsModel, GeminiVoiceName, GoogleEmbeddingModel, GoogleModel, LyriaModel } from "./models.mjs";
+import { GeminiTtsModel, GeminiVoiceName, GoogleEmbeddingModel, GoogleModel, LyriaModel } from "./models.mjs";
 import { t as Gemini_d_exports } from "./Gemini.mjs";
 import { t as GeminiEmbedding_d_exports } from "./GeminiEmbedding.mjs";
 import { t as GeminiSynthesizer_d_exports } from "./GeminiSynthesizer.mjs";
-import { t as GeminiTranscriber_d_exports } from "./GeminiTranscriber.mjs";
 import { t as LyriaGenerator_d_exports } from "./LyriaGenerator.mjs";
-export { Gemini_d_exports as Gemini, GeminiEmbedding_d_exports as GeminiEmbedding, GeminiSttModel, GeminiSynthesizer_d_exports as GeminiSynthesizer, GeminiTranscriber_d_exports as GeminiTranscriber, GeminiTtsModel, GeminiVoiceName, GoogleEmbeddingModel, GoogleModel, LyriaGenerator_d_exports as LyriaGenerator, LyriaModel, codec_d_exports as codec };
+export { Gemini_d_exports as Gemini, GeminiEmbedding_d_exports as GeminiEmbedding, GeminiSynthesizer_d_exports as GeminiSynthesizer, GeminiTtsModel, GeminiVoiceName, GoogleEmbeddingModel, GoogleModel, LyriaGenerator_d_exports as LyriaGenerator, LyriaModel, codec_d_exports as codec };

package/dist/index.mjs

@@ -2,7 +2,6 @@ import { t as codec_exports } from "./codec.mjs";
 import { t as Gemini_exports } from "./Gemini.mjs";
 import { t as GeminiEmbedding_exports } from "./GeminiEmbedding.mjs";
 import { t as GeminiSynthesizer_exports } from "./GeminiSynthesizer.mjs";
-import { t as GeminiTranscriber_exports } from "./GeminiTranscriber.mjs";
 import { t as LyriaGenerator_exports } from "./LyriaGenerator.mjs";
 import "./models.mjs";
-export { Gemini_exports as Gemini, GeminiEmbedding_exports as GeminiEmbedding, GeminiSynthesizer_exports as GeminiSynthesizer, GeminiTranscriber_exports as GeminiTranscriber, LyriaGenerator_exports as LyriaGenerator, codec_exports as codec };
+export { Gemini_exports as Gemini, GeminiEmbedding_exports as GeminiEmbedding, GeminiSynthesizer_exports as GeminiSynthesizer, LyriaGenerator_exports as LyriaGenerator, codec_exports as codec };

package/dist/models.d.mts

@@ -28,15 +28,6 @@ type GoogleModel = "gemini-3.1-pro-preview" | "gemini-3-flash-preview" | "gemini
  * Reference: https://ai.google.dev/gemini-api/docs/speech-generation
  */
 type GeminiTtsModel = "gemini-2.5-flash-preview-tts" | "gemini-2.5-pro-preview-tts" | "gemini-3.1-flash-tts-preview" | (string & {});
-/**
- * Known Gemini models that accept audio input via `inlineData` /
- * `fileData` on `generateContent`. Used by `GeminiTranscriber` —
- * transcription is prompt-driven on top of these models, not a
- * dedicated endpoint.
- *
- * Reference: https://ai.google.dev/gemini-api/docs/audio
- */
-type GeminiSttModel = "gemini-3-flash-preview" | "gemini-3.1-pro-preview" | "gemini-3.1-flash-lite-preview" | "gemini-2.5-flash" | "gemini-2.5-pro" | (string & {});
 /**
  * 30 prebuilt voice names for Gemini TTS. No custom-voice / cloning path
  * is exposed on this surface, so the tail is stock-only (no
@@ -57,5 +48,5 @@ type GeminiVoiceName = "Zephyr" | "Puck" | "Charon" | "Kore" | "Fenrir" | "Leda"
 type LyriaModel = "lyria-3-clip-preview" | "lyria-3-pro-preview" | (string & {});
 type GoogleEmbeddingModel = "gemini-embedding-2" | "gemini-embedding-001" | (string & {});
 //#endregion
-export { GeminiSttModel, GeminiTtsModel, GeminiVoiceName, GoogleEmbeddingModel, GoogleModel, LyriaModel };
+export { GeminiTtsModel, GeminiVoiceName, GoogleEmbeddingModel, GoogleModel, LyriaModel };
 //# sourceMappingURL=models.d.mts.map

package/dist/models.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"models.d.mts","names":[],"sources":["../src/models.ts"],"mappings":";;AAOA;;;;;AAgCA;KAhCY,WAAA;;;;AA+CZ;;;;;AAgBA;;;;;AAyCA;;;;;AAMA;;KA9EY,cAAA;;;;;;;;;KAeA,cAAA;;;;;;;;KAgBA,eAAA;;;;;;;;;;KAyCA,UAAA;AAAA,KAMA,oBAAA"}
+{"version":3,"file":"models.d.mts","names":[],"sources":["../src/models.ts"],"mappings":";;AAOA;;;;;AAgCA;KAhCY,WAAA;;;;AA8CZ;;;;;AAyCA;;;;;AAMA;;;;;;;KA7DY,cAAA;;;;;;;;KAcA,eAAA;;;;;;;;;;KAyCA,UAAA;AAAA,KAMA,oBAAA"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@effect-uai/google",
-  "version": "0.6.0",
+  "version": "0.7.0",
   "description": "Google Gemini provider for @effect-uai/core.",
   "keywords": [
     "ai",
@@ -46,10 +46,6 @@
       "types": "./dist/GeminiSynthesizer.d.mts",
       "import": "./dist/GeminiSynthesizer.mjs"
     },
-    "./GeminiTranscriber": {
-      "types": "./dist/GeminiTranscriber.d.mts",
-      "import": "./dist/GeminiTranscriber.mjs"
-    },
     "./LyriaGenerator": {
       "types": "./dist/LyriaGenerator.d.mts",
       "import": "./dist/LyriaGenerator.mjs"
@@ -61,7 +57,7 @@
   "devDependencies": {
     "effect": "4.0.0-beta.57",
     "typescript": "^6.0.3",
-    "@effect-uai/core": "0.6.0"
+    "@effect-uai/core": "0.7.0"
   },
   "peerDependencies": {
     "@effect-uai/core": ">=0.2.0 <1",

package/src/Gemini.ts

@@ -17,6 +17,7 @@ import {
   accumulatorToTurn,
   buildRequestBody,
   emptyAccumulator,
+  hasUrlImageSource,
   ingestChunk,
 } from "./codec.js"
 import { Match } from "effect"
@@ -162,9 +163,22 @@ const buildNativeStream = (cfg: Config) => {
     Stream.unwrap(
       Effect.gen(function* () {
         const client = yield* HttpClient.HttpClient
+        if (hasUrlImageSource(request.history)) {
+          return yield* new AiError.Unsupported({
+            provider: "gemini",
+            capability: "imageInput",
+            reason:
+              "Gemini needs URL images pre-uploaded via the Files API; pass the image as base64 or raw bytes instead.",
+          })
+        }
         const url = `${baseUrl}/models/${request.model}:streamGenerateContent?alt=sse`
         const generationConfig = buildGenerationConfig(request)
-        const body = buildRequestBody(request.history, generationConfig, request.tools ?? [])
+        const body = buildRequestBody(
+          request.history,
+          generationConfig,
+          request.tools ?? [],
+          request.toolChoice,
+        )
         const httpRequest = HttpClientRequest.post(url).pipe(
           HttpClientRequest.setHeader("x-goog-api-key", Redacted.value(cfg.apiKey)),
           HttpClientRequest.bodyJsonUnsafe(body),

package/src/GeminiEmbedding.ts

@@ -8,6 +8,7 @@ import type {
   Usage,
 } from "@effect-uai/core/Embedding"
 import {
+  assertEncoding,
   type CommonEmbedManyRequest,
   type CommonEmbedRequest,
   type EmbedEncoding,
@@ -348,20 +349,24 @@ export const layer = (
     Effect.map(
       make(cfg),
       (s): EmbeddingModelService => ({
-        // Gemini's endpoints only emit float32; the cast is sound for the
-        // 99% case. A caller who asks for `encoding: "int8"` via the
-        // generic tag gets the type they requested but the runtime returns
-        // float32 (Gemini ignores the field).
+        // Gemini's embed endpoints emit float32 only. Reject a non-float32
+        // `encoding` (bucket 1) so a float32 vector isn't handed back
+        // mislabeled as the requested type. `task` is left silent: it's a
+        // per-model gap (honored by gemini-embedding-001, ignored by
+        // gemini-embedding-2 server-side), and per capabilities.md §2.3 we
+        // don't maintain per-model tables; there's no wire error to translate.
         embed: <E extends EmbedEncoding | undefined = undefined>(
           req: Omit<CommonEmbedRequest, "encoding"> & { readonly encoding?: E },
-        ) => s.embed(req as GeminiEmbedRequest) as Effect.Effect<EmbedResponse<E>, AiError.AiError>,
+        ) =>
+          assertEncoding(req.encoding, ["float32"], "gemini").pipe(
+            Effect.andThen(s.embed(req as GeminiEmbedRequest)),
+          ) as Effect.Effect<EmbedResponse<E>, AiError.AiError>,
         embedMany: <E extends EmbedEncoding | undefined = undefined>(
           req: Omit<CommonEmbedManyRequest, "encoding"> & { readonly encoding?: E },
         ) =>
-          s.embedMany(req as GeminiEmbedManyRequest) as Effect.Effect<
-            EmbedManyResponse<E>,
-            AiError.AiError
-          >,
+          assertEncoding(req.encoding, ["float32"], "gemini").pipe(
+            Effect.andThen(s.embedMany(req as GeminiEmbedManyRequest)),
+          ) as Effect.Effect<EmbedManyResponse<E>, AiError.AiError>,
       }),
     ),
   )

package/src/GeminiSynthesizer.test.ts

@@ -1,20 +1,12 @@
-import { Effect, Layer, Redacted, Stream } from "effect"
-import { FetchHttpClient } from "effect/unstable/http"
-import { describe, expect, expectTypeOf, it } from "vitest"
-import type * as AiError from "@effect-uai/core/AiError"
-import type { AudioBlob } from "@effect-uai/core/Audio"
-import * as SpeechSynthesizer from "@effect-uai/core/SpeechSynthesizer"
+import { Result } from "effect"
+import { describe, expect, it } from "vitest"
 import * as GeminiSynthesizer from "./GeminiSynthesizer.js"
 
-const cfg: GeminiSynthesizer.Config = { apiKey: Redacted.make("test-key") }
-// FetchHttpClient is required for `make`, but these tests only exercise
-// the codec and the compile-time / runtime Unsupported branches — no
-// real HTTP call is made.
-const live = Layer.provide(GeminiSynthesizer.layer(cfg), FetchHttpClient.layer)
-
 describe("GeminiSynthesizer.realizeOutput", () => {
-  it("maps raw to native PCM 24 kHz mono with an identity wrapper", async () => {
-    const [format, wrap] = await Effect.runPromise(GeminiSynthesizer.realizeOutput("raw"))
+  it("maps raw to native PCM 24 kHz mono with an identity wrapper", () => {
+    const result = GeminiSynthesizer.realizeOutput("raw")
+    expect(Result.isSuccess(result)).toBe(true)
+    const [format, wrap] = Result.getOrThrow(result)
     expect(format).toEqual({
       container: "raw",
       encoding: "pcm_s16le",
@@ -25,8 +17,8 @@ describe("GeminiSynthesizer.realizeOutput", () => {
     expect(wrap(pcm)).toBe(pcm)
   })
 
-  it("maps wav to a RIFF/WAVE wrapper that prepends a 44-byte header", async () => {
-    const [format, wrap] = await Effect.runPromise(GeminiSynthesizer.realizeOutput("wav"))
+  it("maps wav to a RIFF/WAVE wrapper that prepends a 44-byte header", () => {
+    const [format, wrap] = Result.getOrThrow(GeminiSynthesizer.realizeOutput("wav"))
     expect(format.container).toBe("wav")
     const out = wrap(new Uint8Array([0, 1, 2, 3]))
     expect(out.length).toBe(44 + 4)
@@ -38,56 +30,10 @@ describe("GeminiSynthesizer.realizeOutput", () => {
     expect(ascii(36, 4)).toBe("data")
   })
 
-  it("fails Unsupported for mp3 / opus / aac / flac / ogg / webm", async () => {
-    const exit = await Effect.runPromiseExit(GeminiSynthesizer.realizeOutput("mp3"))
-    expect(exit._tag).toBe("Failure")
-    if (exit._tag === "Failure") {
-      expect(JSON.stringify(exit.cause)).toContain("Unsupported")
-      expect(JSON.stringify(exit.cause)).toContain("outputFormat")
-    }
-  })
-})
-
-describe("GeminiSynthesizer capability guards (runtime)", () => {
-  it("streamSynthesisFrom returns an Unsupported stream", async () => {
-    const program = GeminiSynthesizer.GeminiSynthesizer.use((s) =>
-      Stream.runDrain(
-        s.streamSynthesisFrom(Stream.fromIterable(["hi"]), {
-          model: "gemini-2.5-flash-preview-tts",
-          voiceId: "Kore",
-        }),
-      ),
-    )
-    const exit = await Effect.runPromiseExit(program.pipe(Effect.provide(live)))
-    expect(exit._tag).toBe("Failure")
-    if (exit._tag === "Failure") {
-      expect(JSON.stringify(exit.cause)).toContain("Unsupported")
-      expect(JSON.stringify(exit.cause)).toContain("streamSynthesisFrom")
-    }
-  })
-})
-
-describe("GeminiSynthesizer Layer (compile-time)", () => {
-  it("leaves `TtsIncrementalText` unsatisfied when using `streamSynthesisFrom` against this Layer", () => {
-    const tokens: Stream.Stream<string> = Stream.fromIterable(["a"])
-    const audio = tokens.pipe(
-      SpeechSynthesizer.streamSynthesisFrom({
-        model: "gemini-2.5-flash-preview-tts",
-        voiceId: "Kore",
-      }),
-    )
-    const provided = Stream.runDrain(audio).pipe(Effect.provide(live))
-    expectTypeOf(provided).toEqualTypeOf<
-      Effect.Effect<void, AiError.AiError, SpeechSynthesizer.TtsIncrementalText>
-    >()
-  })
-
-  it("sync `synthesize` requires no marker", () => {
-    const synth = SpeechSynthesizer.synthesize({
-      text: "hi",
-      model: "gemini-2.5-flash-preview-tts",
-      voiceId: "Kore",
-    }).pipe(Effect.provide(live))
-    expectTypeOf(synth).toEqualTypeOf<Effect.Effect<AudioBlob, AiError.AiError, never>>()
+  it("fails Unsupported for mp3 / opus / aac / flac / ogg / webm", () => {
+    const result = GeminiSynthesizer.realizeOutput("mp3")
+    expect(Result.isFailure(result)).toBe(true)
+    expect(JSON.stringify(result)).toContain("Unsupported")
+    expect(JSON.stringify(result)).toContain("outputFormat")
   })
 })

package/src/GeminiSynthesizer.ts

@@ -1,7 +1,8 @@
-import { Context, Effect, Layer, Match, Redacted, Schema, Stream } from "effect"
+import { Context, Effect, Layer, Match, Redacted, Result, Schema, Stream } from "effect"
 import { HttpClient, HttpClientRequest } from "effect/unstable/http"
 import * as AiError from "@effect-uai/core/AiError"
 import type { AudioBlob, AudioChunk, AudioFormat } from "@effect-uai/core/Audio"
+import * as Capabilities from "@effect-uai/core/Capabilities"
 import {
   type CommonSynthesizeRequest,
   SpeechSynthesizer,
@@ -51,19 +52,19 @@ const PCM_CHANNELS = 1
 
 export const realizeOutput = Match.type<AudioFormat["container"]>().pipe(
   Match.when("raw", () =>
-    Effect.succeed([
+    Result.succeed([
       { container: "raw", encoding: "pcm_s16le", sampleRate: PCM_RATE, channels: PCM_CHANNELS },
       (b: Uint8Array) => b,
     ] as const),
   ),
   Match.when("wav", () =>
-    Effect.succeed([
+    Result.succeed([
       { container: "wav", encoding: "pcm_s16le", sampleRate: PCM_RATE, channels: PCM_CHANNELS },
       (b: Uint8Array) => wrapPcmAsWav(b, PCM_RATE, PCM_CHANNELS),
     ] as const),
   ),
   Match.whenOr("mp3", "opus", "aac", "flac", "ogg", "webm", (c) =>
-    Effect.fail(
+    Result.fail(
       new AiError.Unsupported({
         provider: "gemini",
         capability: "outputFormat",
@@ -125,6 +126,31 @@ const baseUrl = (cfg: Config) => cfg.baseUrl ?? "https://generativelanguage.goog
 
 const synthesizeImpl = (cfg: Config) => (request: GeminiSynthesizeRequest) =>
   Effect.gen(function* () {
+    // Gemini `:generateContent` TTS has no phoneme field. Pronunciations
+    // are load-bearing (bucket 1), so reject rather than mispronounce.
+    if (request.pronunciations !== undefined && request.pronunciations.length > 0) {
+      return yield* Effect.fail(
+        new AiError.Unsupported({
+          provider: "gemini",
+          capability: "pronunciations",
+          reason:
+            "Gemini `:generateContent` TTS has no phoneme field. Use Cloud Text-to-Speech (Chirp 3 HD) or a provider with inline phonemes (Inworld) for pronunciation overrides.",
+        }),
+      )
+    }
+    // No speaking-rate or language parameters on this endpoint; bucket 2.
+    yield* Capabilities.warnDroppedWhen(request.speed, {
+      provider: "gemini",
+      capability: "speed",
+      field: "speed",
+      reason: "Gemini `:generateContent` TTS has no speaking-rate parameter.",
+    })
+    yield* Capabilities.warnDroppedWhen(request.languageCode, {
+      provider: "gemini",
+      capability: "languageCode",
+      field: "languageCode",
+      reason: "Gemini `:generateContent` TTS has no language parameter.",
+    })
     const client = yield* HttpClient.HttpClient
     const [format, wrap] = yield* realizeOutput(request.outputFormat?.container ?? "raw")
     const httpRequest = HttpClientRequest.post(

package/src/LyriaGenerator.test.ts

@@ -1,81 +1,17 @@
-import { Effect, Layer, Redacted, Stream } from "effect"
+import { Duration, Effect, Layer, Redacted, Stream } from "effect"
 import { FetchHttpClient } from "effect/unstable/http"
 import { describe, expect, expectTypeOf, it } from "vitest"
 import type * as AiError from "@effect-uai/core/AiError"
-import * as MusicGenerator from "@effect-uai/core/MusicGenerator"
+import type { GenerateResult } from "@effect-uai/core/Music"
 import { promptsInput } from "@effect-uai/core/Music"
+import * as MusicGenerator from "@effect-uai/core/MusicGenerator"
 import * as LyriaGenerator from "./LyriaGenerator.js"
 
 const cfg: LyriaGenerator.Config = { apiKey: Redacted.make("test-key") }
 // FetchHttpClient is required for `make`, but these tests only exercise the
-// compile-time gating and runtime Unsupported branches — no real HTTP is made.
+// compile-time gating and runtime Unsupported branches, no real HTTP is made.
 const live = Layer.provide(LyriaGenerator.layer(cfg), FetchHttpClient.layer)
 
-describe("buildPrompt", () => {
-  it("returns a plain string when prompts is a string", () => {
-    const out = LyriaGenerator.buildPrompt({
-      model: "lyria-3-clip-preview",
-      prompts: "upbeat indie pop",
-    })
-    expect(out).toBe("upbeat indie pop")
-  })
-
-  it("joins weighted prompts with weight annotations", () => {
-    const out = LyriaGenerator.buildPrompt({
-      model: "lyria-3-clip-preview",
-      prompts: [
-        { text: "minimal techno", weight: 1.0 },
-        { text: "1980s synthwave", weight: 0.3 },
-      ],
-    })
-    expect(out).toContain("minimal techno")
-    expect(out).toContain("1980s synthwave (weight 0.3)")
-  })
-
-  it("omits the weight annotation for weight === 1", () => {
-    const out = LyriaGenerator.buildPrompt({
-      model: "lyria-3-clip-preview",
-      prompts: [{ text: "ambient", weight: 1 }],
-    })
-    expect(out).toBe("ambient")
-  })
-
-  it("splices in bpm, scale, duration, and instrumental hints", () => {
-    const out = LyriaGenerator.buildPrompt({
-      model: "lyria-3-clip-preview",
-      prompts: "house",
-      bpm: 124,
-      scale: "C_MAJOR",
-      durationSeconds: 30,
-      instrumental: true,
-    })
-    expect(out).toContain("Instrumental only")
-    expect(out).toContain("BPM: 124")
-    expect(out).toContain("Key/scale: C_MAJOR")
-    expect(out).toContain("Target duration: 30s")
-  })
-
-  it("appends a lyrics block when lyrics is set and instrumental is not true", () => {
-    const out = LyriaGenerator.buildPrompt({
-      model: "lyria-3-clip-preview",
-      prompts: "indie rock",
-      lyrics: "[Verse]\nhello world",
-    })
-    expect(out).toContain("Lyrics:")
-    expect(out).toContain("[Verse]")
-  })
-
-  it("drops the lyrics block when instrumental is true", () => {
-    const out = LyriaGenerator.buildPrompt({
-      model: "lyria-3-clip-preview",
-      prompts: "indie rock",
-      lyrics: "[Verse]\nhello world",
-      instrumental: true,
-    })
-    expect(out).not.toContain("Lyrics:")
-  })
-})
-
 describe("containerToMimeType", () => {
   it("maps mp3 and wav", async () => {
     expect(await Effect.runPromise(LyriaGenerator.containerToMimeType("mp3"))).toBe("audio/mp3")
@@ -115,7 +51,7 @@ describe("LyriaGenerator capability guards (runtime)", () => {
       Stream.runDrain(
         s.streamGenerationFrom(Stream.fromIterable([promptsInput([{ text: "x" }])]), {
           model: "lyria-3-clip-preview",
-          prompts: "",
+          prompt: "",
         }),
       ),
     )
@@ -131,32 +67,52 @@ describe("LyriaGenerator capability guards (runtime)", () => {
 describe("LyriaGenerator Layer (compile-time)", () => {
   it("leaves `MusicInteractiveSession` unsatisfied when using `streamGenerationFrom` against this Layer", () => {
     const inputs = Stream.fromIterable([promptsInput([{ text: "techno" }])])
-    const audio = inputs.pipe(
+    const events = inputs.pipe(
       MusicGenerator.streamGenerationFrom({
         model: "lyria-3-clip-preview",
-        prompts: "",
+        prompt: "",
       }),
     )
-    const provided = Stream.runDrain(audio).pipe(Effect.provide(live))
+    const provided = Stream.runDrain(events).pipe(Effect.provide(live))
     expectTypeOf(provided).toEqualTypeOf<
       Effect.Effect<void, AiError.AiError, MusicGenerator.MusicInteractiveSession>
     >()
   })
 
-  it("sync `generate` and chunked `streamGeneration` require no marker", () => {
+  it("sync `generate` returns GenerateResult and requires no marker", () => {
     const gen = MusicGenerator.generate({
       model: "lyria-3-clip-preview",
-      prompts: "ambient",
+      prompt: "ambient",
     }).pipe(Effect.provide(live))
-    expectTypeOf(gen).toEqualTypeOf<
-      Effect.Effect<import("@effect-uai/core/Music").MusicResult, AiError.AiError, never>
-    >()
+    expectTypeOf(gen).toEqualTypeOf<Effect.Effect<GenerateResult, AiError.AiError, never>>()
+  })
+
+  it("chunked `streamGeneration` requires no marker", () => {
     const _stream = Stream.runCollect(
       MusicGenerator.streamGeneration({
         model: "lyria-3-clip-preview",
-        prompts: "ambient",
+        prompt: "ambient",
       }),
     ).pipe(Effect.provide(live))
     expect(_stream).toBeDefined()
   })
+
+  it("Lyria-typed request accepts no `instrumental` field", () => {
+    const _req: LyriaGenerator.LyriaGenerateRequest = {
+      model: "lyria-3-clip-preview",
+      prompt: "ambient",
+      // @ts-expect-error: instrumental was removed from LyriaGenerateRequest in v0.7
+      instrumental: true,
+    }
+    expect(_req).toBeDefined()
+  })
+
+  it("Common request supports `duration: Duration.Duration`", () => {
+    const _req: MusicGenerator.CommonGenerateMusicRequest = {
+      model: "lyria-3-clip-preview",
+      prompt: "ambient",
+      duration: Duration.seconds(30),
+    }
+    expect(_req).toBeDefined()
+  })
 })