effect 4.0.0-rc.112 → 4.0.0-rc.114

package/src/unstable/ai/Response.ts

@@ -16,10 +16,11 @@ import { identity } from "../../Function.ts"
 import * as Predicate from "../../Predicate.ts"
 import * as Schema from "../../Schema.ts"
 import * as SchemaTransformation from "../../SchemaTransformation.ts"
-import type * as Tool from "./Tool.ts"
+import * as AiError from "./AiError.ts"
+import * as Tool from "./Tool.ts"
 import type * as Toolkit from "./Toolkit.ts"
 
-const PartTypeId = "~effect/ai/Content/Part" as const
+const PartTypeId = "~effect/ai/Response/Part" as const
 
 // =============================================================================
 // All Parts
@@ -185,7 +186,7 @@ export const AllParts = <T extends Toolkit.Any | Toolkit.WithHandler<any>>(
   const toolResults: Array<Schema.Top> = []
   for (const tool of Object.values(toolkit.tools as Record<string, Tool.Any>)) {
     const toolCall = ToolCallPart(tool.name, tool.parametersSchema)
-    const toolResult = ToolResultPart(tool.name, tool.successSchema, tool.failureSchema)
+    const toolResult = ToolResultPart(tool.name, tool.successSchema, Tool.failureResultSchema(tool))
     toolCalls.push(toolCall)
     toolResults.push(toolResult)
   }
@@ -226,11 +227,11 @@ export const AllParts = <T extends Toolkit.Any | Toolkit.WithHandler<any>>(
  */
 export type Part<
   Tools extends Record<string, Tool.Any>,
-  EncodedToolParameters extends boolean = false
+  ParametersMode extends ToolParametersMode = "decoded"
 > =
   | TextPart
   | ReasoningPart
-  | ToolCallParts<Tools, EncodedToolParameters>
+  | ToolCallParts<Tools, ParametersMode>
   | ToolResultParts<Tools>
   | ToolApprovalRequestPart
   | FilePart
@@ -277,7 +278,7 @@ export const Part = <T extends Toolkit.Any | Toolkit.WithHandler<any>>(
   const toolResults: Array<Schema.Top> = []
   for (const tool of Object.values(toolkit.tools as Record<string, Tool.Any>)) {
     const toolCall = ToolCallPart(tool.name, tool.parametersSchema)
-    const toolResult = ToolResultPart(tool.name, tool.successSchema, tool.failureSchema)
+    const toolResult = ToolResultPart(tool.name, tool.successSchema, Tool.failureResultSchema(tool))
     toolCalls.push(toolCall)
     toolResults.push(toolResult)
   }
@@ -307,7 +308,7 @@ export const Part = <T extends Toolkit.Any | Toolkit.WithHandler<any>>(
  */
 export type StreamPart<
   Tools extends Record<string, Tool.Any>,
-  EncodedToolParameters extends boolean = false
+  ParametersMode extends ToolParametersMode = "decoded"
 > =
   | TextStartPart
   | TextDeltaPart
@@ -318,7 +319,7 @@ export type StreamPart<
   | ToolParamsStartPart
   | ToolParamsDeltaPart
   | ToolParamsEndPart
-  | ToolCallParts<Tools, EncodedToolParameters>
+  | ToolCallParts<Tools, ParametersMode>
   | ToolResultParts<Tools>
   | ToolApprovalRequestPart
   | FilePart
@@ -372,7 +373,7 @@ export const StreamPart = <T extends Toolkit.Any | Toolkit.WithHandler<any>>(
   const toolResults: Array<Schema.Top> = []
   for (const tool of Object.values(toolkit.tools as Record<string, Tool.Any>)) {
     const toolCall = ToolCallPart(tool.name, tool.parametersSchema)
-    const toolResult = ToolResultPart(tool.name, tool.successSchema, tool.failureSchema)
+    const toolResult = ToolResultPart(tool.name, tool.successSchema, Tool.failureResultSchema(tool))
     toolCalls.push(toolCall)
     toolResults.push(toolResult)
   }
@@ -410,14 +411,28 @@ export const StreamPart = <T extends Toolkit.Any | Toolkit.WithHandler<any>>(
  */
 export type ToolCallParts<
   Tools extends Record<string, Tool.Any>,
-  EncodedParameters extends boolean = false
-> = {
-  [Name in keyof Tools]: Name extends string ? ToolCallPart<
-      Name,
-      EncodedParameters extends true ? Tool.ParametersEncoded<Tools[Name]> : Tool.Parameters<Tools[Name]>
-    >
-    : never
-}[keyof Tools]
+  ParametersMode extends ToolParametersMode = "decoded"
+> = ToolCallPartForName<Tools, ParametersMode, keyof Tools>
+
+/**
+ * Controls whether tool parameters are decoded, encoded, or opaque.
+ *
+ * @category utility types
+ * @since 4.0.0
+ */
+export type ToolParametersMode = "decoded" | "encoded" | "opaque"
+
+type ToolCallPartForName<
+  Tools extends Record<string, Tool.Any>,
+  ParametersMode extends ToolParametersMode,
+  Name extends keyof Tools
+> = Name extends string ? ToolCallPart<
+    Name,
+    ParametersMode extends "encoded" ? Tool.ParametersEncoded<Tools[Name]>
+      : ParametersMode extends "opaque" ? unknown
+      : Tool.Parameters<Tools[Name]>
+  >
+  : never
 
 /**
  * Utility type that extracts tool result parts from a set of tools.
@@ -425,11 +440,12 @@ export type ToolCallParts<
  * @category utility types
  * @since 4.0.0
  */
-export type ToolResultParts<Tools extends Record<string, Tool.Any>> = {
-  [Name in keyof Tools]: Name extends string
-    ? ToolResultPart<Name, Tool.Success<Tools[Name]>, Tool.FailureResult<Tools[Name]>>
-    : never
-}[keyof Tools]
+export type ToolResultParts<Tools extends Record<string, Tool.Any>> = ToolResultPartForName<Tools, keyof Tools>
+
+type ToolResultPartForName<
+  Tools extends Record<string, Tool.Any>,
+  Name extends keyof Tools
+> = Name extends string ? ToolResultPart<Name, Tool.Success<Tools[Name]>, Tool.FailureResult<Tools[Name]>> : never
 
 // =============================================================================
 // Base Part
@@ -622,7 +638,7 @@ export interface TextPartMetadata extends ProviderMetadata {}
 export const TextPart: Schema.Struct<{
   readonly type: Schema.tag<"text">
   readonly text: Schema.String
-  readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
+  readonly "~effect/ai/Response/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Response/Part">>
   readonly metadata: Schema.withDecodingDefault<
     Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
@@ -681,7 +697,7 @@ export interface TextStartPartMetadata extends ProviderMetadata {}
 export const TextStartPart: Schema.Struct<{
   readonly type: Schema.tag<"text-start">
   readonly id: Schema.String
-  readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
+  readonly "~effect/ai/Response/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Response/Part">>
   readonly metadata: Schema.withDecodingDefault<
     Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
@@ -749,7 +765,7 @@ export const TextDeltaPart: Schema.Struct<{
   readonly type: Schema.tag<"text-delta">
   readonly id: Schema.String
   readonly delta: Schema.String
-  readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
+  readonly "~effect/ai/Response/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Response/Part">>
   readonly metadata: Schema.withDecodingDefault<
     Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
@@ -808,7 +824,7 @@ export interface TextEndPartMetadata extends ProviderMetadata {}
 export const TextEndPart: Schema.Struct<{
   readonly type: Schema.tag<"text-end">
   readonly id: Schema.String
-  readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
+  readonly "~effect/ai/Response/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Response/Part">>
   readonly metadata: Schema.withDecodingDefault<
     Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
@@ -880,7 +896,7 @@ export interface ReasoningPartMetadata extends ProviderMetadata {}
 export const ReasoningPart: Schema.Struct<{
   readonly type: Schema.tag<"reasoning">
   readonly text: Schema.String
-  readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
+  readonly "~effect/ai/Response/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Response/Part">>
   readonly metadata: Schema.withDecodingDefault<
     Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
@@ -939,7 +955,7 @@ export interface ReasoningStartPartMetadata extends ProviderMetadata {}
 export const ReasoningStartPart: Schema.Struct<{
   readonly type: Schema.tag<"reasoning-start">
   readonly id: Schema.String
-  readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
+  readonly "~effect/ai/Response/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Response/Part">>
   readonly metadata: Schema.withDecodingDefault<
     Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
@@ -1007,7 +1023,7 @@ export const ReasoningDeltaPart: Schema.Struct<{
   readonly type: Schema.tag<"reasoning-delta">
   readonly id: Schema.String
   readonly delta: Schema.String
-  readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
+  readonly "~effect/ai/Response/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Response/Part">>
   readonly metadata: Schema.withDecodingDefault<
     Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
@@ -1066,7 +1082,7 @@ export interface ReasoningEndPartMetadata extends ProviderMetadata {}
 export const ReasoningEndPart: Schema.Struct<{
   readonly type: Schema.tag<"reasoning-end">
   readonly id: Schema.String
-  readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
+  readonly "~effect/ai/Response/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Response/Part">>
   readonly metadata: Schema.withDecodingDefault<
     Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
@@ -1149,7 +1165,7 @@ export const ToolParamsStartPart: Schema.Struct<{
   readonly id: Schema.String
   readonly name: Schema.String
   readonly providerExecuted: Schema.withDecodingDefaultKey<Schema.Boolean>
-  readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
+  readonly "~effect/ai/Response/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Response/Part">>
   readonly metadata: Schema.withDecodingDefault<
     Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
@@ -1226,7 +1242,7 @@ export const ToolParamsDeltaPart: Schema.Struct<{
   readonly type: Schema.tag<"tool-params-delta">
   readonly id: Schema.String
   readonly delta: Schema.String
-  readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
+  readonly "~effect/ai/Response/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Response/Part">>
   readonly metadata: Schema.withDecodingDefault<
     Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
@@ -1293,7 +1309,7 @@ export interface ToolParamsEndPartMetadata extends ProviderMetadata {}
 export const ToolParamsEndPart: Schema.Struct<{
   readonly type: Schema.tag<"tool-params-end">
   readonly id: Schema.String
-  readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
+  readonly "~effect/ai/Response/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Response/Part">>
   readonly metadata: Schema.withDecodingDefault<
     Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
@@ -1410,7 +1426,7 @@ export const ToolCallPart: <const Name extends string, Params extends Schema.Con
     readonly name: Schema.Literal<Name>
     readonly params: Params
     readonly providerExecuted: Schema.withDecodingDefaultKey<Schema.Boolean>
-    readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
+    readonly "~effect/ai/Response/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Response/Part">>
     readonly metadata: Schema.withDecodingDefault<
       Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
     >
@@ -1614,7 +1630,7 @@ export interface ToolResultPartMetadata extends ProviderMetadata {}
  * @category schemas
  * @since 4.0.0
  */
-export const ToolResultPart: <
+export const ToolResultPart = <
   const Name extends string,
   Success extends Schema.Constraint,
   Failure extends Schema.Constraint
@@ -1622,89 +1638,54 @@ export const ToolResultPart: <
   name: Name,
   success: Success,
   failure: Failure
-) => Schema.decodeTo<
-  Schema.Struct<
-    {
-      readonly "~effect/ai/Content/Part": Schema.Literal<"~effect/ai/Content/Part">
-      readonly result: Schema.Union<readonly [Success, Failure]>
-      readonly providerExecuted: Schema.Boolean
-      readonly metadata: Schema.$Record<
-        Schema.String,
-        Schema.NullOr<Schema.Codec<Schema.Json>>
-      >
-      readonly encodedResult: Schema.toEncoded<Schema.Union<readonly [Success, Failure]>>
-      readonly preliminary: Schema.Boolean
-      readonly id: Schema.String
-      readonly type: Schema.Literal<"tool-result">
-      readonly isFailure: Schema.Boolean
-      readonly name: Schema.Literal<Name>
-    }
-  >,
-  Schema.Struct<
-    {
-      readonly result: Schema.toEncoded<Schema.Union<readonly [Success, Failure]>>
-      readonly providerExecuted: Schema.optional<Schema.Boolean>
-      readonly metadata: Schema.optional<
-        Schema.$Record<Schema.String, Schema.NullOr<Schema.Codec<Schema.Json>>>
-      >
-      readonly preliminary: Schema.optional<Schema.Boolean>
-      readonly id: Schema.String
-      readonly type: Schema.Literal<"tool-result">
-      readonly isFailure: Schema.Boolean
-      readonly name: Schema.Literal<Name>
+): Schema.Codec<
+  ToolResultPart<Name, Success["Type"], Failure["Type"]>,
+  ToolResultPartEncoded,
+  Success["DecodingServices"] | Failure["DecodingServices"],
+  Success["EncodingServices"] | Failure["EncodingServices"]
+> => {
+  const makeSchema = (result: Schema.Constraint, isFailure: boolean) => {
+    const encodedResult = Schema.toEncoded(result)
+    const Common = {
+      id: Schema.String,
+      type: Schema.Literal("tool-result"),
+      isFailure: Schema.Literal(isFailure),
+      name: Schema.Literal(name)
     }
-  >
-> = <
-  const Name extends string,
-  Success extends Schema.Constraint,
-  Failure extends Schema.Constraint
->(
-  name: Name,
-  success: Success,
-  failure: Failure
-) => {
-  const ResultSchema = Schema.Union([success, failure])
-  const Common = {
-    id: Schema.String,
-    type: Schema.Literal("tool-result"),
-    isFailure: Schema.Boolean,
-    name: Schema.Literal(name)
+    const Decoded = Schema.Struct({
+      ...Common,
+      [PartTypeId]: Schema.Literal(PartTypeId),
+      result,
+      providerExecuted: Schema.Boolean,
+      metadata: ProviderMetadata,
+      encodedResult,
+      preliminary: Schema.Boolean
+    })
+    const Encoded = Schema.Struct({
+      ...Common,
+      result: encodedResult,
+      providerExecuted: Schema.optional(Schema.Boolean),
+      metadata: Schema.optional(ProviderMetadata),
+      preliminary: Schema.optional(Schema.Boolean)
+    })
+    return Decoded.pipe(Schema.encodeTo(
+      Encoded,
+      SchemaTransformation.transform({
+        decode: (encoded) => ({
+          ...encoded,
+          [PartTypeId]: PartTypeId,
+          providerExecuted: encoded.providerExecuted ?? false,
+          metadata: encoded.metadata ?? {},
+          encodedResult: encoded.result,
+          preliminary: encoded.preliminary ?? false
+        }),
+        encode: identity
+      })
+    ))
   }
-  const Decoded = Schema.Struct({
-    ...Common,
-    [PartTypeId]: Schema.Literal(PartTypeId),
-    result: ResultSchema,
-    providerExecuted: Schema.Boolean,
-    metadata: ProviderMetadata,
-    encodedResult: Schema.toEncoded(ResultSchema),
-    preliminary: Schema.Boolean
-  })
-  const Encoded = Schema.Struct({
-    ...Common,
-    result: Schema.toEncoded(ResultSchema),
-    providerExecuted: Schema.optional(Schema.Boolean),
-    metadata: Schema.optional(ProviderMetadata),
-    preliminary: Schema.optional(Schema.Boolean)
+  return Schema.Union([makeSchema(success, false), makeSchema(failure, true)]).annotate({
+    identifier: `ToolResultPart(${name})`
   })
-  return Decoded.pipe(Schema.encodeTo(
-    Encoded,
-    SchemaTransformation.transform({
-      decode: (encoded) => ({
-        ...encoded,
-        [PartTypeId]: PartTypeId,
-        providerExecuted: encoded.providerExecuted ?? false,
-        metadata: encoded.metadata ?? {},
-        encodedResult: encoded.result,
-        preliminary: encoded.preliminary ?? false
-      }),
-      encode: identity
-    })
-  )).annotate({ identifier: `ToolResultPart(${name})` }) satisfies Schema.Codec<
-    ToolResultPart<Name, Success["Type"], Failure["Type"]>,
-    ToolResultPartEncoded,
-    Success["EncodingServices"] | Failure["EncodingServices"],
-    Success["DecodingServices"] | Failure["DecodingServices"]
-  >
 }
 
 /**
@@ -1807,7 +1788,7 @@ export const ToolApprovalRequestPart: Schema.Struct<{
   readonly type: Schema.tag<"tool-approval-request">
   readonly approvalId: Schema.String
   readonly toolCallId: Schema.String
-  readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
+  readonly "~effect/ai/Response/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Response/Part">>
   readonly metadata: Schema.withDecodingDefault<
     Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
@@ -1909,7 +1890,7 @@ export const FilePart: Schema.Struct<{
   readonly type: Schema.tag<"file">
   readonly mediaType: Schema.String
   readonly data: Schema.Uint8ArrayFromBase64
-  readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
+  readonly "~effect/ai/Response/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Response/Part">>
   readonly metadata: Schema.withDecodingDefault<
     Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
@@ -2019,7 +2000,7 @@ export const DocumentSourcePart: Schema.Struct<{
   readonly mediaType: Schema.String
   readonly title: Schema.String
   readonly fileName: Schema.optionalKey<Schema.String>
-  readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
+  readonly "~effect/ai/Response/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Response/Part">>
   readonly metadata: Schema.withDecodingDefault<
     Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
@@ -2109,7 +2090,7 @@ export const UrlSourcePart: Schema.Struct<{
   readonly id: Schema.String
   readonly url: Schema.URLFromString
   readonly title: Schema.String
-  readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
+  readonly "~effect/ai/Response/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Response/Part">>
   readonly metadata: Schema.withDecodingDefault<
     Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
@@ -2127,83 +2108,20 @@ export const UrlSourcePart: Schema.Struct<{
 // =============================================================================
 
 /**
- * Schema for HTTP request details associated with an AI response.
- *
- * **Details**
- *
- * Captures comprehensive information about the HTTP request made to the
- * AI provider, enabling inspection of request metadata for debugging and
- * observability purposes.
- *
- * **Example** (Describing an HTTP request)
- *
- * ```ts import.meta.vitest
- * import type { Response } from "effect/unstable/ai"
- *
- * const requestDetails: typeof Response.HttpRequestDetails.Type = {
- *   method: "POST",
- *   url: "https://api.openai.com/v1/responses",
- *   urlParams: [],
- *   hash: undefined,
- *   headers: { "Content-Type": "application/json" }
- * }
- * const result = [requestDetails.method, requestDetails.urlParams] // => ["POST", []]
- * ```
+ * Alias of {@link AiError.HttpRequestDetails}.
  *
  * @category schemas
  * @since 4.0.0
  */
-export const HttpRequestDetails = Schema.Struct({
-  method: Schema.Literals(["GET", "POST", "PATCH", "PUT", "DELETE", "HEAD", "OPTIONS", "TRACE"]),
-  url: Schema.String,
-  urlParams: Schema.Array(Schema.Tuple([Schema.String, Schema.String])),
-  hash: Schema.optional(Schema.String),
-  headers: Schema.Record(
-    Schema.String,
-    Schema.Union([
-      Schema.String,
-      Schema.Redacted(Schema.String)
-    ])
-  )
-}).annotate({ identifier: "HttpRequestDetails" })
+export const HttpRequestDetails = AiError.HttpRequestDetails
 
 /**
- * Schema for HTTP response details associated with an AI response.
- *
- * **Details**
- *
- * Captures essential information about the HTTP response received from
- * the AI provider, including status codes and headers for debugging and
- * observability purposes.
- *
- * **Example** (Describing an HTTP response)
- *
- * ```ts import.meta.vitest
- * import type { Response } from "effect/unstable/ai"
- *
- * const responseDetails: typeof Response.HttpResponseDetails.Type = {
- *   status: 200,
- *   headers: {
- *     "Content-Type": "application/json",
- *     "X-Request-Id": "req_abc123"
- *   }
- * }
- * const result = [responseDetails.status, responseDetails.headers["X-Request-Id"]] // => [200, "req_abc123"]
- * ```
+ * Alias of {@link AiError.HttpResponseDetails}.
  *
  * @category schemas
  * @since 4.0.0
  */
-export const HttpResponseDetails = Schema.Struct({
-  status: Schema.Int,
-  headers: Schema.Record(
-    Schema.String,
-    Schema.Union([
-      Schema.String,
-      Schema.Redacted(Schema.String)
-    ])
-  )
-}).annotate({ identifier: "HttpResponseDetails" })
+export const HttpResponseDetails = AiError.HttpResponseDetails
 
 // =============================================================================
 // Response Metadata Part
@@ -2300,7 +2218,7 @@ export const ResponseMetadataPart: Schema.Struct<{
   readonly modelId: Schema.optional<Schema.String>
   readonly timestamp: Schema.optional<Schema.DateTimeUtcFromString>
   readonly request: Schema.optional<typeof HttpRequestDetails>
-  readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
+  readonly "~effect/ai/Response/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Response/Part">>
   readonly metadata: Schema.withDecodingDefault<
     Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
@@ -2526,7 +2444,7 @@ export const FinishPart: Schema.Struct<{
   ]>
   readonly usage: typeof Usage
   readonly response: Schema.optional<typeof HttpResponseDetails>
-  readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
+  readonly "~effect/ai/Response/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Response/Part">>
   readonly metadata: Schema.withDecodingDefault<
     Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
@@ -2601,7 +2519,7 @@ export interface ErrorPartMetadata extends ProviderMetadata {}
 export const ErrorPart: Schema.Struct<{
   readonly type: Schema.tag<"error">
   readonly error: Schema.Unknown
-  readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
+  readonly "~effect/ai/Response/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Response/Part">>
   readonly metadata: Schema.withDecodingDefault<
     Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >

package/src/unstable/ai/Tokenizer.ts

@@ -165,7 +165,6 @@ const truncate = (
   maxTokens: number
 ): Effect.Effect<Prompt.Prompt, AiError.AiError> =>
   Effect.suspend(() => {
-    let count = 0
     let inputMessages = self.content
     let outputMessages: Array<Prompt.Message> = []
     const loop: Effect.Effect<Prompt.Prompt, AiError.AiError> = Effect.suspend(() => {
@@ -174,12 +173,12 @@ const truncate = (
         return Effect.succeed(Prompt.fromMessages(outputMessages))
       }
       inputMessages = inputMessages.slice(0, inputMessages.length - 1)
-      return Effect.flatMap(tokenize(Prompt.fromMessages([message])), (tokens) => {
-        count += tokens.length
-        if (count > maxTokens) {
+      const candidate = [message, ...outputMessages]
+      return Effect.flatMap(tokenize(Prompt.fromMessages(candidate)), (tokens) => {
+        if (tokens.length > maxTokens) {
           return Effect.succeed(Prompt.fromMessages(outputMessages))
         }
-        outputMessages = [message, ...outputMessages]
+        outputMessages = candidate
         return loop
       })
     })