@opencode/ai 0.0.0-dev-20546 → 0.0.0-dev-20549

package/README.md

@@ -106,6 +106,41 @@ for await (const event of ai.llm.stream(ai.llm.request(input))) {
 await ai.dispose()
 ```
 
+## Venice AI
+
+`Venice` provides native Chat Completions with streaming tools and reasoning. `model` and `chat`
+select the same API; credentials default to `VENICE_API_KEY`.
+
+```ts
+import { LLM } from "@opencode/ai"
+import { Venice } from "@opencode/ai/providers"
+import { Effect } from "effect"
+
+const program = Effect.gen(function* () {
+  const response = yield* LLM.generate({
+    model: Venice.configure({ apiKey: process.env.VENICE_API_KEY }).chat("qwen3-6-27b"),
+    prompt: "Explain this design.",
+    providerOptions: {
+      reasoningEffort: "high",
+      veniceParameters: { includeVeniceSystemPrompt: false },
+    },
+  })
+  console.log(response.text)
+})
+```
+
+Effort lowers to `reasoning.effort`; `reasoning.enabled` and `reasoning.summary` are also available.
+Supported effort levels and toggles depend on the selected model. Omitted controls preserve its defaults.
+Venice's added system prompt is disabled by default, matching the previous OpenCode Venice SDK behavior.
+
+Replay complete `response.message` values to retain signed/encrypted reasoning and Gemini thought
+signatures, including per-tool signatures. Venice's encrypted scalar trailers are excluded from visible
+reasoning but retained in provider metadata for replay. Cache affinity uses `promptCacheKey`, and cache-write
+usage reads Venice's `cache_creation_input_tokens` field.
+
+The native package entrypoint is `@opencode/ai/providers/venice`. Image generation, embeddings, Responses, and
+client-side E2EE are not implemented by this provider.
+
 ## Experimental evaluation
 
 Evaluation models compare shared state with typed choice, score, and boolean questions. The API is

package/dist/protocols/openai-chat.d.ts

@@ -18,6 +18,19 @@ declare const ExtraContent: Schema.Struct<{
         readonly thought_signature: Schema.String;
     }>;
 }>;
+export declare const OpenAIChatAssistantToolCall: Schema.Struct<{
+    readonly id: Schema.String;
+    readonly type: Schema.tag<"function">;
+    readonly function: Schema.Struct<{
+        readonly name: Schema.String;
+        readonly arguments: Schema.String;
+    }>;
+    readonly extra_content: Schema.optional<Schema.Struct<{
+        readonly google: Schema.Struct<{
+            readonly thought_signature: Schema.String;
+        }>;
+    }>>;
+}>;
 declare const ReasoningDetail: Schema.Union<readonly [Schema.StructWithRest<Schema.Struct<{
     readonly id: Schema.optional<Schema.NullOr<Schema.String>>;
     readonly format: Schema.optional<Schema.String>;
@@ -49,6 +62,88 @@ declare const ReasoningDetail: Schema.Union<readonly [Schema.StructWithRest<Sche
     readonly index: Schema.optional<Schema.Number>;
 }>]>;
 type ReasoningDetail = Schema.Schema.Type<typeof ReasoningDetail>;
+export declare const OpenAIChatMessage: Schema.toTaggedUnion<"role", readonly [Schema.Struct<{
+    readonly role: Schema.Literal<"system">;
+    readonly content: Schema.Union<readonly [Schema.String, Schema.$Array<Schema.Union<readonly [Schema.Struct<{
+        readonly type: Schema.Literal<"text">;
+        readonly text: Schema.String;
+        readonly cache_control: Schema.optional<Schema.Struct<{
+            readonly type: Schema.Literal<"ephemeral">;
+            readonly ttl: Schema.optional<Schema.String>;
+        }>>;
+    }>, Schema.Struct<{
+        readonly type: Schema.Literal<"image_url">;
+        readonly image_url: Schema.Struct<{
+            readonly url: Schema.String;
+        }>;
+    }>, Schema.Struct<{
+        readonly type: Schema.Literal<"file">;
+        readonly file: Schema.Struct<{
+            readonly filename: Schema.String;
+            readonly file_data: Schema.String;
+        }>;
+    }>]>>]>;
+}>, Schema.Struct<{
+    readonly role: Schema.Literal<"user">;
+    readonly content: Schema.Union<readonly [Schema.String, Schema.$Array<Schema.Union<readonly [Schema.Struct<{
+        readonly type: Schema.Literal<"text">;
+        readonly text: Schema.String;
+        readonly cache_control: Schema.optional<Schema.Struct<{
+            readonly type: Schema.Literal<"ephemeral">;
+            readonly ttl: Schema.optional<Schema.String>;
+        }>>;
+    }>, Schema.Struct<{
+        readonly type: Schema.Literal<"image_url">;
+        readonly image_url: Schema.Struct<{
+            readonly url: Schema.String;
+        }>;
+    }>, Schema.Struct<{
+        readonly type: Schema.Literal<"file">;
+        readonly file: Schema.Struct<{
+            readonly filename: Schema.String;
+            readonly file_data: Schema.String;
+        }>;
+    }>]>>]>;
+}>, Schema.StructWithRest<Schema.Struct<{
+    readonly role: Schema.Literal<"assistant">;
+    readonly content: Schema.NullOr<Schema.Union<readonly [Schema.String, Schema.$Array<Schema.Struct<{
+        readonly type: Schema.Literal<"text">;
+        readonly text: Schema.String;
+        readonly cache_control: Schema.optional<Schema.Struct<{
+            readonly type: Schema.Literal<"ephemeral">;
+            readonly ttl: Schema.optional<Schema.String>;
+        }>>;
+    }>>]>>;
+    readonly tool_calls: Schema.optional<Schema.$Array<Schema.Struct<{
+        readonly id: Schema.String;
+        readonly type: Schema.tag<"function">;
+        readonly function: Schema.Struct<{
+            readonly name: Schema.String;
+            readonly arguments: Schema.String;
+        }>;
+        readonly extra_content: Schema.optional<Schema.Struct<{
+            readonly google: Schema.Struct<{
+                readonly thought_signature: Schema.String;
+            }>;
+        }>>;
+    }>>>;
+    readonly reasoning_content: Schema.optional<Schema.String>;
+    readonly reasoning: Schema.optional<Schema.String>;
+    readonly reasoning_text: Schema.optional<Schema.String>;
+    readonly reasoning_details: Schema.optional<Schema.Unknown>;
+}>, readonly [Schema.$Record<Schema.String, Schema.Unknown>]>, Schema.Struct<{
+    readonly role: Schema.Literal<"tool">;
+    readonly tool_call_id: Schema.String;
+    readonly content: Schema.Union<readonly [Schema.String, Schema.$Array<Schema.Struct<{
+        readonly type: Schema.Literal<"text">;
+        readonly text: Schema.String;
+        readonly cache_control: Schema.optional<Schema.Struct<{
+            readonly type: Schema.Literal<"ephemeral">;
+            readonly ttl: Schema.optional<Schema.String>;
+        }>>;
+    }>>]>;
+}>]>;
+type OpenAIChatMessage = Schema.Schema.Type<typeof OpenAIChatMessage>;
 export declare const bodyFields: {
     model: Schema.String;
     messages: Schema.$Array<Schema.toTaggedUnion<"role", readonly [Schema.Struct<{
@@ -288,6 +383,89 @@ declare const OpenAIChatBody: Schema.Struct<{
     stop: Schema.optional<Schema.$Array<Schema.String>>;
 }>;
 export type OpenAIChatBody = Schema.Schema.Type<typeof OpenAIChatBody>;
+export declare const OpenAIChatUsage: Schema.StructWithRest<Schema.Struct<{
+    readonly prompt_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+    readonly completion_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+    readonly total_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+    readonly cached_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+    readonly prompt_cache_hit_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+    readonly cache_read_input_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+    readonly cache_created_input_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+    readonly prompt_tokens_details: Schema.optional<Schema.NullOr<Schema.StructWithRest<Schema.Struct<{
+        readonly cached_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+        readonly cache_write_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+    }>, readonly [Schema.$Record<Schema.String, Schema.Unknown>]>>>;
+    readonly completion_tokens_details: Schema.optional<Schema.NullOr<Schema.StructWithRest<Schema.Struct<{
+        readonly reasoning_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+        readonly accepted_prediction_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+        readonly rejected_prediction_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+    }>, readonly [Schema.$Record<Schema.String, Schema.Unknown>]>>>;
+}>, readonly [Schema.$Record<Schema.String, Schema.Unknown>]>;
+export declare const OpenAIChatToolCallDelta: Schema.Struct<{
+    readonly index: Schema.optional<Schema.NullOr<Schema.Number>>;
+    readonly id: Schema.optional<Schema.NullOr<Schema.String>>;
+    readonly function: Schema.optional<Schema.NullOr<Schema.Struct<{
+        readonly name: Schema.optional<Schema.NullOr<Schema.String>>;
+        readonly arguments: Schema.optional<Schema.NullOr<Schema.String>>;
+    }>>>;
+    readonly extra_content: Schema.optional<Schema.NullOr<Schema.Unknown>>;
+}>;
+export declare const OpenAIChatDelta: Schema.StructWithRest<Schema.Struct<{
+    readonly content: Schema.optional<Schema.NullOr<Schema.String>>;
+    readonly refusal: Schema.optional<Schema.NullOr<Schema.String>>;
+    readonly reasoning_content: Schema.optional<Schema.NullOr<Schema.String>>;
+    readonly reasoning: Schema.optional<Schema.NullOr<Schema.String>>;
+    readonly reasoning_text: Schema.optional<Schema.NullOr<Schema.String>>;
+    readonly reasoning_details: Schema.optional<Schema.NullOr<Schema.Unknown>>;
+    readonly tool_calls: Schema.optional<Schema.NullOr<Schema.$Array<Schema.Struct<{
+        readonly index: Schema.optional<Schema.NullOr<Schema.Number>>;
+        readonly id: Schema.optional<Schema.NullOr<Schema.String>>;
+        readonly function: Schema.optional<Schema.NullOr<Schema.Struct<{
+            readonly name: Schema.optional<Schema.NullOr<Schema.String>>;
+            readonly arguments: Schema.optional<Schema.NullOr<Schema.String>>;
+        }>>>;
+        readonly extra_content: Schema.optional<Schema.NullOr<Schema.Unknown>>;
+    }>>>>;
+}>, readonly [Schema.$Record<Schema.String, Schema.Unknown>]>;
+export declare const OpenAIChatChoice: Schema.StructWithRest<Schema.Struct<{
+    readonly delta: Schema.optional<Schema.NullOr<Schema.StructWithRest<Schema.Struct<{
+        readonly content: Schema.optional<Schema.NullOr<Schema.String>>;
+        readonly refusal: Schema.optional<Schema.NullOr<Schema.String>>;
+        readonly reasoning_content: Schema.optional<Schema.NullOr<Schema.String>>;
+        readonly reasoning: Schema.optional<Schema.NullOr<Schema.String>>;
+        readonly reasoning_text: Schema.optional<Schema.NullOr<Schema.String>>;
+        readonly reasoning_details: Schema.optional<Schema.NullOr<Schema.Unknown>>;
+        readonly tool_calls: Schema.optional<Schema.NullOr<Schema.$Array<Schema.Struct<{
+            readonly index: Schema.optional<Schema.NullOr<Schema.Number>>;
+            readonly id: Schema.optional<Schema.NullOr<Schema.String>>;
+            readonly function: Schema.optional<Schema.NullOr<Schema.Struct<{
+                readonly name: Schema.optional<Schema.NullOr<Schema.String>>;
+                readonly arguments: Schema.optional<Schema.NullOr<Schema.String>>;
+            }>>>;
+            readonly extra_content: Schema.optional<Schema.NullOr<Schema.Unknown>>;
+        }>>>>;
+    }>, readonly [Schema.$Record<Schema.String, Schema.Unknown>]>>>;
+    readonly finish_reason: Schema.optional<Schema.NullOr<Schema.String>>;
+    readonly native_finish_reason: Schema.optional<Schema.NullOr<Schema.String>>;
+    readonly usage: Schema.optional<Schema.NullOr<Schema.StructWithRest<Schema.Struct<{
+        readonly prompt_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+        readonly completion_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+        readonly total_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+        readonly cached_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+        readonly prompt_cache_hit_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+        readonly cache_read_input_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+        readonly cache_created_input_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+        readonly prompt_tokens_details: Schema.optional<Schema.NullOr<Schema.StructWithRest<Schema.Struct<{
+            readonly cached_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+            readonly cache_write_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+        }>, readonly [Schema.$Record<Schema.String, Schema.Unknown>]>>>;
+        readonly completion_tokens_details: Schema.optional<Schema.NullOr<Schema.StructWithRest<Schema.Struct<{
+            readonly reasoning_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+            readonly accepted_prediction_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+            readonly rejected_prediction_tokens: Schema.optional<Schema.NullOr<Schema.Number>>;
+        }>, readonly [Schema.$Record<Schema.String, Schema.Unknown>]>>>;
+    }>, readonly [Schema.$Record<Schema.String, Schema.Unknown>]>>>;
+}>, readonly [Schema.$Record<Schema.String, Schema.Unknown>]>;
 export declare const OpenAIChatEvent: Schema.StructWithRest<Schema.Struct<{
     readonly choices: Schema.optional<Schema.NullOr<Schema.$Array<Schema.StructWithRest<Schema.Struct<{
         readonly delta: Schema.optional<Schema.NullOr<Schema.StructWithRest<Schema.Struct<{
@@ -379,6 +557,8 @@ export interface ParserState {
 interface LoweringOptions {
     readonly cacheControl?: (cache: CacheHint | undefined) => OpenAIChatCacheControl | undefined;
     readonly toolCallID?: (id: string) => string;
+    /** Project provider-specific fields from the exact source, even when other messages are dropped during lowering. */
+    readonly assistant?: (source: LLMRequest["messages"][number], message: OpenAIChatMessage) => OpenAIChatMessage;
 }
 export declare const fromRequest: (request: LLMRequest<import("../schema/options.js").LanguageModel<{
     readonly [x: string]: unknown;
@@ -623,6 +803,222 @@ export declare const fromRequest: (request: LLMRequest<import("../schema/options
     } | undefined;
     stream: true;
 }, AIError, never>;
+export declare const finishEvents: (state: ParserState) => Effect.Effect<(({
+    readonly type: "compaction";
+    readonly provider: string & import("effect/Brand").Brand<"AI.ProviderID">;
+    readonly id?: string | undefined;
+    readonly text?: string | null | undefined;
+    readonly encrypted?: string | undefined;
+} & ({
+    readonly encrypted: string;
+    readonly text?: never;
+} | {
+    readonly text: string | null;
+    readonly encrypted?: never;
+})) | {
+    readonly type: "step-start";
+    readonly index: number;
+} | {
+    readonly id: string;
+    readonly type: "text-start";
+    readonly providerMetadata?: {
+        readonly [x: string]: {
+            readonly [x: string]: unknown;
+        };
+    } | undefined;
+} | {
+    readonly id: string;
+    readonly type: "text-delta";
+    readonly text: string;
+    readonly providerMetadata?: {
+        readonly [x: string]: {
+            readonly [x: string]: unknown;
+        };
+    } | undefined;
+} | {
+    readonly id: string;
+    readonly type: "text-end";
+    readonly text?: string | undefined;
+    readonly providerMetadata?: {
+        readonly [x: string]: {
+            readonly [x: string]: unknown;
+        };
+    } | undefined;
+} | {
+    readonly id: string;
+    readonly type: "reasoning-start";
+    readonly providerMetadata?: {
+        readonly [x: string]: {
+            readonly [x: string]: unknown;
+        };
+    } | undefined;
+} | {
+    readonly id: string;
+    readonly type: "reasoning-delta";
+    readonly text: string;
+    readonly providerMetadata?: {
+        readonly [x: string]: {
+            readonly [x: string]: unknown;
+        };
+    } | undefined;
+} | {
+    readonly id: string;
+    readonly type: "reasoning-end";
+    readonly text?: string | undefined;
+    readonly providerMetadata?: {
+        readonly [x: string]: {
+            readonly [x: string]: unknown;
+        };
+    } | undefined;
+} | {
+    readonly id: string;
+    readonly type: "tool-input-start";
+    readonly name: string;
+    readonly providerMetadata?: {
+        readonly [x: string]: {
+            readonly [x: string]: unknown;
+        };
+    } | undefined;
+    readonly namespace?: string | undefined;
+    readonly providerExecuted?: boolean | undefined;
+} | {
+    readonly id: string;
+    readonly type: "tool-input-delta";
+    readonly name: string;
+    readonly text: string;
+    readonly input?: unknown;
+    readonly namespace?: string | undefined;
+} | {
+    readonly id: string;
+    readonly type: "tool-input-end";
+    readonly name: string;
+    readonly providerMetadata?: {
+        readonly [x: string]: {
+            readonly [x: string]: unknown;
+        };
+    } | undefined;
+    readonly namespace?: string | undefined;
+} | {
+    readonly id: string;
+    readonly type: "tool-input-error";
+    readonly name: string;
+    readonly raw: string;
+    readonly namespace?: string | undefined;
+} | {
+    readonly id: string;
+    readonly type: "tool-call";
+    readonly name: string;
+    readonly input: unknown;
+    readonly providerMetadata?: {
+        readonly [x: string]: {
+            readonly [x: string]: unknown;
+        };
+    } | undefined;
+    readonly namespace?: string | undefined;
+    readonly providerExecuted?: boolean | undefined;
+} | {
+    readonly id: string;
+    readonly type: "tool-result";
+    readonly name: string;
+    readonly result: {
+        readonly type: "json";
+        readonly value: unknown;
+    } | {
+        readonly type: "text";
+        readonly value: unknown;
+    } | {
+        readonly type: "error";
+        readonly value: unknown;
+    } | {
+        readonly type: "content";
+        readonly value: readonly ({
+            readonly type: "text";
+            readonly text: string;
+        } | {
+            readonly type: "file";
+            readonly uri: string;
+            readonly mime: string;
+            readonly name?: string | undefined;
+        })[];
+    };
+    readonly output?: {
+        readonly structured: unknown;
+        readonly content: readonly ({
+            readonly type: "text";
+            readonly text: string;
+        } | {
+            readonly type: "file";
+            readonly uri: string;
+            readonly mime: string;
+            readonly name?: string | undefined;
+        })[];
+    } | undefined;
+    readonly providerMetadata?: {
+        readonly [x: string]: {
+            readonly [x: string]: unknown;
+        };
+    } | undefined;
+    readonly namespace?: string | undefined;
+    readonly providerExecuted?: boolean | undefined;
+} | {
+    readonly id: string;
+    readonly type: "tool-error";
+    readonly message: string;
+    readonly name: string;
+    readonly error?: unknown;
+    readonly providerMetadata?: {
+        readonly [x: string]: {
+            readonly [x: string]: unknown;
+        };
+    } | undefined;
+    readonly namespace?: string | undefined;
+} | {
+    readonly type: "media";
+    readonly media: import("../media.js").Asset;
+    readonly providerMetadata?: {
+        readonly [x: string]: {
+            readonly [x: string]: unknown;
+        };
+    } | undefined;
+} | {
+    readonly type: "step-finish";
+    readonly reason: {
+        readonly normalized: "length" | "stop" | "tool-calls" | "content-filter" | "error" | "unknown";
+        readonly raw?: string | undefined;
+        readonly category?: string | undefined;
+        readonly explanation?: string | undefined;
+    };
+    readonly index: number;
+    readonly providerMetadata?: {
+        readonly [x: string]: {
+            readonly [x: string]: unknown;
+        };
+    } | undefined;
+    readonly usage?: Usage | undefined;
+} | {
+    readonly type: "finish";
+    readonly reason: {
+        readonly normalized: "length" | "stop" | "tool-calls" | "content-filter" | "error" | "unknown";
+        readonly raw?: string | undefined;
+        readonly category?: string | undefined;
+        readonly explanation?: string | undefined;
+    };
+    readonly providerMetadata?: {
+        readonly [x: string]: {
+            readonly [x: string]: unknown;
+        };
+    } | undefined;
+    readonly usage?: Usage | undefined;
+} | {
+    readonly type: "provider-error";
+    readonly message: string;
+    readonly classification?: "context-overflow" | "payload-too-large" | undefined;
+    readonly providerMetadata?: {
+        readonly [x: string]: {
+            readonly [x: string]: unknown;
+        };
+    } | undefined;
+})[], AIError, never>;
 /**
  * The OpenAI Chat protocol — request body construction, body schema, and the
  * streaming-event state machine. Reused by every route that speaks OpenAI Chat

package/dist/protocols/openai-chat.js

@@ -43,7 +43,7 @@ const ExtraContent = Schema.Struct({
     google: Schema.Struct({ thought_signature: Schema.String }),
 });
 const decodeExtraContent = (value) => Option.getOrUndefined(Schema.decodeUnknownOption(ExtraContent)(value));
-const OpenAIChatAssistantToolCall = Schema.Struct({
+export const OpenAIChatAssistantToolCall = Schema.Struct({
     id: Schema.String,
     type: Schema.tag("function"),
     function: Schema.Struct({
@@ -94,7 +94,7 @@ const OpenAIChatUserContent = Schema.Union([
         file: Schema.Struct({ filename: Schema.String, file_data: Schema.String }),
     }),
 ]);
-const OpenAIChatMessage = Schema.Union([
+export const OpenAIChatMessage = Schema.Union([
     Schema.Struct({
         role: Schema.Literal("system"),
         content: Schema.Union([Schema.String, Schema.Array(OpenAIChatUserContent)]),
@@ -152,7 +152,7 @@ const OpenAIChatBody = Schema.Struct(bodyFields);
 // The event schema is one decoded SSE `data:` payload. `Framing.sse` splits the
 // byte stream into strings, then `Protocol.jsonEvent` decodes each string into
 // this provider-native event shape.
-const OpenAIChatUsage = Schema.StructWithRest(Schema.Struct({
+export const OpenAIChatUsage = Schema.StructWithRest(Schema.Struct({
     prompt_tokens: optionalNull(Schema.Number),
     completion_tokens: optionalNull(Schema.Number),
     total_tokens: optionalNull(Schema.Number),
@@ -175,13 +175,13 @@ const OpenAIChatToolCallDeltaFunction = Schema.Struct({
     name: optionalNull(Schema.String),
     arguments: optionalNull(Schema.String),
 });
-const OpenAIChatToolCallDelta = Schema.Struct({
+export const OpenAIChatToolCallDelta = Schema.Struct({
     index: optionalNull(Schema.Number),
     id: optionalNull(Schema.String),
     function: optionalNull(OpenAIChatToolCallDeltaFunction),
     extra_content: optionalNull(Schema.Unknown),
 });
-const OpenAIChatDelta = Schema.StructWithRest(Schema.Struct({
+export const OpenAIChatDelta = Schema.StructWithRest(Schema.Struct({
     content: optionalNull(Schema.String),
     refusal: optionalNull(Schema.String),
     reasoning_content: optionalNull(Schema.String),
@@ -190,7 +190,7 @@ const OpenAIChatDelta = Schema.StructWithRest(Schema.Struct({
     reasoning_details: optionalNull(Schema.Unknown),
     tool_calls: optionalNull(Schema.Array(OpenAIChatToolCallDelta)),
 }), [Schema.Record(Schema.String, Schema.Unknown)]);
-const OpenAIChatChoice = Schema.StructWithRest(Schema.Struct({
+export const OpenAIChatChoice = Schema.StructWithRest(Schema.Struct({
     delta: optionalNull(OpenAIChatDelta),
     finish_reason: optionalNull(Schema.String),
     native_finish_reason: optionalNull(Schema.String),
@@ -386,8 +386,10 @@ const toolMessage = (toolCallID, text, cacheControl) => ({
 const lowerMessage = Effect.fnUntraced(function* (message, reasoningField, requireReasoning, options) {
     if (message.role === "user")
         return [yield* lowerUserMessage(message, options)];
-    if (message.role === "assistant")
-        return [yield* lowerAssistantMessage(message, reasoningField, requireReasoning, options)];
+    if (message.role === "assistant") {
+        const lowered = yield* lowerAssistantMessage(message, reasoningField, requireReasoning, options);
+        return [options.assistant?.(message, lowered) ?? lowered];
+    }
     return (yield* lowerToolMessages(message, options)).messages;
 });
 const lowerMessages = Effect.fnUntraced(function* (request, options) {
@@ -981,7 +983,7 @@ const step = (state, event) => Effect.gen(function* () {
         events,
     ];
 });
-const finishEvents = Effect.fnUntraced(function* (state) {
+export const finishEvents = Effect.fnUntraced(function* (state) {
     if (state.finishReason === undefined && state.requireFinishReason)
         return yield* new AIError({
             reason: new InvalidProviderOutputError({