effect 4.0.0-beta.4 → 4.0.0-beta.41

package/src/unstable/ai/Model.ts

@@ -14,7 +14,7 @@
  *
  * declare const myAnthropicLayer: Layer.Layer<LanguageModel.LanguageModel>
  *
- * const anthropicModel = Model.make("anthropic", myAnthropicLayer)
+ * const anthropicModel = Model.make("anthropic", "claude-3-5-haiku", myAnthropicLayer)
  *
  * const program = Effect.gen(function*() {
  *   const response = yield* LanguageModel.generateText({
@@ -55,8 +55,13 @@ const TypeId = "~effect/ai/Model" as const
  */
 export interface Model<in out Provider, in out Provides, in out Requires>
   extends
-    Layer.Layer<Provides | ProviderName, never, Requires>,
-    Effect.Yieldable<Model<Provider, Provides, Requires>, Layer.Layer<Provides | ProviderName>, never, Requires>
+    Layer.Layer<Provides | ProviderName | ModelName, never, Requires>,
+    Effect.Yieldable<
+      Model<Provider, Provides, Requires>,
+      Layer.Layer<Provides | ProviderName | ModelName>,
+      never,
+      Requires
+    >
 {
   readonly [TypeId]: typeof TypeId
   /**
@@ -79,6 +84,20 @@ export class ProviderName extends ServiceMap.Service<ProviderName, string>()(
   "effect/unstable/ai/Model/ProviderName"
 ) {}
 
+/**
+ * Service tag that provides the current large language model name.
+ *
+ * This tag is automatically provided by Model instances and can be used to
+ * access the name of the model that is currently in use within a given Effect
+ * program.
+ *
+ * @since 4.0.0
+ * @category services
+ */
+export class ModelName extends ServiceMap.Service<ModelName, string>()(
+  "effect/unstable/ai/Model/ModelName"
+) {}
+
 const Proto = {
   ...YieldableProto,
   ...PipeInspectableProto,
@@ -112,11 +131,12 @@ const Proto = {
  *
  * declare const bedrockLayer: Layer.Layer<LanguageModel.LanguageModel>
  *
- * // Model automatically provides ProviderName service
+ * // Model automatically provides ProviderName and ModelName services
  * const checkProviderAndGenerate = Effect.gen(function*() {
  *   const provider = yield* Model.ProviderName
+ *   const modelName = yield* Model.ModelName
  *
- *   console.log(`Generating with: ${provider}`)
+ *   console.log(`Generating with: ${provider}/${modelName}`)
  *
  *   return yield* LanguageModel.generateText({
  *     prompt: `Hello from ${provider}!`
@@ -124,19 +144,23 @@ const Proto = {
  * })
  *
  * const program = checkProviderAndGenerate.pipe(
- *   Effect.provide(Model.make("amazon-bedrock", bedrockLayer))
+ *   Effect.provide(Model.make("amazon-bedrock", "claude-3-5-haiku", bedrockLayer))
  * )
- * // Will log: "Generating with: amazon-bedrock"
+ * // Will log: "Generating with: amazon-bedrock/claude-3-5-haiku"
  * ```
  *
  * @since 4.0.0
  * @category constructors
  */
-export const make = <const Provider extends string, Provides, Requires>(
+export const make = <const Provider extends string, const Name extends string, Provides, Requires>(
   /**
    * Provider identifier (e.g., "openai", "anthropic", "amazon-bedrock").
    */
   provider: Provider,
+  /**
+   * Model identifier (e.g., "gpt-5", "claude-3-5-haiku").
+   */
+  modelName: Name,
   /**
    * Layer that provides the AI services for this provider.
    */
@@ -145,5 +169,12 @@ export const make = <const Provider extends string, Provides, Requires>(
   Object.assign(
     Object.create(Proto),
     { provider },
-    Layer.merge(Layer.succeed(ProviderName)(provider), layer)
+    Layer.merge(
+      layer,
+      Layer.succeedServices(
+        ProviderName.serviceMap(provider).pipe(
+          ServiceMap.add(ModelName, modelName)
+        )
+      )
+    )
   )

package/src/unstable/ai/OpenAiStructuredOutput.ts

@@ -11,6 +11,7 @@ import * as Rec from "../../Record.ts"
 import * as Schema from "../../Schema.ts"
 import * as AST from "../../SchemaAST.ts"
 import * as Transformation from "../../SchemaTransformation.ts"
+import * as Tool from "./Tool.ts"
 
 /**
  * Transforms a `Schema.Codec` into a form compatible with OpenAI's
@@ -243,6 +244,9 @@ function recurOpenAI(ast: AST.AST): AST.AST {
         }
       } else if (ast.indexSignatures.length === 1 && ast.propertySignatures.length === 0) {
         const is = ast.indexSignatures[0]
+        if (Tool.isEmptyParamsRecord(is)) {
+          return ast
+        }
         // records are not supported by OpenAI, so we translate them to arrays of key-value pairs
         if (annotations !== undefined && typeof annotations.description === "string") {
           annotations.description = `${RECORD_DESCRIPTION}; ${annotations.description}`

package/src/unstable/ai/Prompt.ts

@@ -86,9 +86,9 @@ const constEmptyObject = () => ({})
  * @since 4.0.0
  * @category models
  */
-export const ProviderOptions: Schema.Record$<
+export const ProviderOptions: Schema.$Record<
   Schema.String,
-  Schema.NullOr<Schema.Codec<Schema.Json, Schema.Json>>
+  Schema.NullOr<Schema.Codec<Schema.Json>>
 > = Schema.Record(Schema.String, Schema.NullOr(Schema.Json))
 
 /**
@@ -309,9 +309,9 @@ export const TextPart: Schema.Struct<
     readonly text: Schema.String
     readonly "~effect/ai/Prompt/Part": Schema.withDecodingDefaultKey<Schema.Literal<"~effect/ai/Prompt/Part">>
     readonly options: Schema.withDecodingDefault<
-      Schema.Record$<
+      Schema.$Record<
         Schema.String,
-        Schema.NullOr<Schema.Codec<Schema.Json, Schema.Json>>
+        Schema.NullOr<Schema.Codec<Schema.Json>>
       >
     >
   }
@@ -389,9 +389,9 @@ export const ReasoningPart: Schema.Struct<{
   readonly text: Schema.String
   readonly "~effect/ai/Prompt/Part": Schema.withDecodingDefaultKey<Schema.Literal<"~effect/ai/Prompt/Part">>
   readonly options: Schema.withDecodingDefault<
-    Schema.Record$<
+    Schema.$Record<
       Schema.String,
-      Schema.NullOr<Schema.Codec<Schema.Json, Schema.Json>>
+      Schema.NullOr<Schema.Codec<Schema.Json>>
     >
   >
 }> = Schema.Struct({
@@ -497,9 +497,9 @@ export const FilePart: Schema.Struct<{
   readonly data: Schema.Union<readonly [Schema.String, Schema.Uint8Array, Schema.URL]>
   readonly "~effect/ai/Prompt/Part": Schema.withDecodingDefaultKey<Schema.Literal<"~effect/ai/Prompt/Part">>
   readonly options: Schema.withDecodingDefault<
-    Schema.Record$<
+    Schema.$Record<
       Schema.String,
-      Schema.NullOr<Schema.Codec<Schema.Json, Schema.Json>>
+      Schema.NullOr<Schema.Codec<Schema.Json>>
     >
   >
 }> = Schema.Struct({
@@ -607,9 +607,9 @@ export const ToolCallPart: Schema.Struct<{
   readonly providerExecuted: Schema.withDecodingDefault<Schema.Boolean>
   readonly "~effect/ai/Prompt/Part": Schema.withDecodingDefaultKey<Schema.Literal<"~effect/ai/Prompt/Part">>
   readonly options: Schema.withDecodingDefault<
-    Schema.Record$<
+    Schema.$Record<
       Schema.String,
-      Schema.NullOr<Schema.Codec<Schema.Json, Schema.Json>>
+      Schema.NullOr<Schema.Codec<Schema.Json>>
     >
   >
 }> = Schema.Struct({
@@ -723,9 +723,9 @@ export const ToolResultPart: Schema.Struct<{
   readonly result: Schema.Unknown
   readonly "~effect/ai/Prompt/Part": Schema.withDecodingDefaultKey<Schema.Literal<"~effect/ai/Prompt/Part">>
   readonly options: Schema.withDecodingDefault<
-    Schema.Record$<
+    Schema.$Record<
       Schema.String,
-      Schema.NullOr<Schema.Codec<Schema.Json, Schema.Json>>
+      Schema.NullOr<Schema.Codec<Schema.Json>>
     >
   >
 }> = Schema.Struct({
@@ -841,9 +841,9 @@ export const ToolApprovalResponsePart: Schema.Struct<{
   readonly reason: Schema.optional<Schema.String>
   readonly "~effect/ai/Prompt/Part": Schema.withDecodingDefaultKey<Schema.Literal<"~effect/ai/Prompt/Part">>
   readonly options: Schema.withDecodingDefault<
-    Schema.Record$<
+    Schema.$Record<
       Schema.String,
-      Schema.NullOr<Schema.Codec<Schema.Json, Schema.Json>>
+      Schema.NullOr<Schema.Codec<Schema.Json>>
     >
   >
 }> = Schema.Struct({
@@ -942,9 +942,9 @@ export const ToolApprovalRequestPart: Schema.Struct<{
   readonly toolCallId: Schema.String
   readonly "~effect/ai/Prompt/Part": Schema.withDecodingDefaultKey<Schema.Literal<"~effect/ai/Prompt/Part">>
   readonly options: Schema.withDecodingDefault<
-    Schema.Record$<
+    Schema.$Record<
       Schema.String,
-      Schema.NullOr<Schema.Codec<Schema.Json, Schema.Json>>
+      Schema.NullOr<Schema.Codec<Schema.Json>>
     >
   >
 }> = Schema.Struct({
@@ -1087,9 +1087,9 @@ export const ContentFromString: Schema.decodeTo<
         readonly text: Schema.String
         readonly "~effect/ai/Prompt/Part": Schema.withDecodingDefaultKey<Schema.Literal<"~effect/ai/Prompt/Part">>
         readonly options: Schema.withDecodingDefault<
-          Schema.Record$<
+          Schema.$Record<
             Schema.String,
-            Schema.NullOr<Schema.Codec<Schema.Json, Schema.Json>>
+            Schema.NullOr<Schema.Codec<Schema.Json>>
           >
         >
       }>
@@ -1166,9 +1166,9 @@ export const SystemMessage: Schema.Struct<{
   readonly content: Schema.String
   readonly "~effect/ai/Prompt/Message": Schema.withDecodingDefaultKey<Schema.Literal<"~effect/ai/Prompt/Message">>
   readonly options: Schema.withDecodingDefault<
-    Schema.Record$<
+    Schema.$Record<
       Schema.String,
-      Schema.NullOr<Schema.Codec<Schema.Json, Schema.Json>>
+      Schema.NullOr<Schema.Codec<Schema.Json>>
     >
   >
 }> = Schema.Struct({
@@ -1285,9 +1285,9 @@ export const UserMessage: Schema.Struct<{
               readonly text: Schema.String
               readonly "~effect/ai/Prompt/Part": Schema.withDecodingDefaultKey<Schema.Literal<"~effect/ai/Prompt/Part">>
               readonly options: Schema.withDecodingDefault<
-                Schema.Record$<
+                Schema.$Record<
                   Schema.String,
-                  Schema.NullOr<Schema.Codec<Schema.Json, Schema.Json>>
+                  Schema.NullOr<Schema.Codec<Schema.Json>>
                 >
               >
             }>
@@ -1297,7 +1297,7 @@ export const UserMessage: Schema.Struct<{
         never,
         never
       >,
-      Schema.Array$<
+      Schema.$Array<
         Schema.Union<
           readonly [
             Schema.Struct<{
@@ -1305,9 +1305,9 @@ export const UserMessage: Schema.Struct<{
               readonly text: Schema.String
               readonly "~effect/ai/Prompt/Part": Schema.withDecodingDefaultKey<Schema.Literal<"~effect/ai/Prompt/Part">>
               readonly options: Schema.withDecodingDefault<
-                Schema.Record$<
+                Schema.$Record<
                   Schema.String,
-                  Schema.NullOr<Schema.Codec<Schema.Json, Schema.Json>>
+                  Schema.NullOr<Schema.Codec<Schema.Json>>
                 >
               >
             }>,
@@ -1318,9 +1318,9 @@ export const UserMessage: Schema.Struct<{
               readonly data: Schema.Union<readonly [Schema.String, Schema.Uint8Array, Schema.URL]>
               readonly "~effect/ai/Prompt/Part": Schema.withDecodingDefaultKey<Schema.Literal<"~effect/ai/Prompt/Part">>
               readonly options: Schema.withDecodingDefault<
-                Schema.Record$<
+                Schema.$Record<
                   Schema.String,
-                  Schema.NullOr<Schema.Codec<Schema.Json, Schema.Json>>
+                  Schema.NullOr<Schema.Codec<Schema.Json>>
                 >
               >
             }>
@@ -1331,9 +1331,9 @@ export const UserMessage: Schema.Struct<{
   >
   readonly "~effect/ai/Prompt/Message": Schema.withDecodingDefaultKey<Schema.Literal<"~effect/ai/Prompt/Message">>
   readonly options: Schema.withDecodingDefault<
-    Schema.Record$<
+    Schema.$Record<
       Schema.String,
-      Schema.NullOr<Schema.Codec<Schema.Json, Schema.Json>>
+      Schema.NullOr<Schema.Codec<Schema.Json>>
     >
   >
 }> = Schema.Struct({
@@ -1470,9 +1470,9 @@ export const AssistantMessage: Schema.Struct<{
               readonly text: Schema.String
               readonly "~effect/ai/Prompt/Part": Schema.withDecodingDefaultKey<Schema.Literal<"~effect/ai/Prompt/Part">>
               readonly options: Schema.withDecodingDefault<
-                Schema.Record$<
+                Schema.$Record<
                   Schema.String,
-                  Schema.NullOr<Schema.Codec<Schema.Json, Schema.Json>>
+                  Schema.NullOr<Schema.Codec<Schema.Json>>
                 >
               >
             }>
@@ -1482,7 +1482,7 @@ export const AssistantMessage: Schema.Struct<{
         never,
         never
       >,
-      Schema.Array$<
+      Schema.$Array<
         Schema.Union<
           readonly [
             typeof TextPart,
@@ -1498,9 +1498,9 @@ export const AssistantMessage: Schema.Struct<{
   >
   readonly "~effect/ai/Prompt/Message": Schema.withDecodingDefaultKey<Schema.Literal<"~effect/ai/Prompt/Message">>
   readonly options: Schema.withDecodingDefault<
-    Schema.Record$<
+    Schema.$Record<
       Schema.String,
-      Schema.NullOr<Schema.Codec<Schema.Json, Schema.Json>>
+      Schema.NullOr<Schema.Codec<Schema.Json>>
     >
   >
 }> = Schema.Struct({
@@ -1613,14 +1613,14 @@ export interface ToolMessageOptions extends ProviderOptions {}
  */
 export const ToolMessage: Schema.Struct<{
   readonly role: Schema.Literal<"tool">
-  readonly content: Schema.Array$<
+  readonly content: Schema.$Array<
     Schema.Union<readonly [typeof ToolResultPart, typeof ToolApprovalResponsePart]>
   >
   readonly "~effect/ai/Prompt/Message": Schema.withDecodingDefaultKey<Schema.Literal<"~effect/ai/Prompt/Message">>
   readonly options: Schema.withDecodingDefault<
-    Schema.Record$<
+    Schema.$Record<
       Schema.String,
-      Schema.NullOr<Schema.Codec<Schema.Json, Schema.Json>>
+      Schema.NullOr<Schema.Codec<Schema.Json>>
     >
   >
 }> = Schema.Struct({

package/src/unstable/ai/Response.ts

@@ -453,9 +453,9 @@ export type ToolResultParts<Tools extends Record<string, Tool.Any>> = {
  * @since 1.0.0
  * @category schemas
  */
-export const ProviderMetadata: Schema.Record$<
+export const ProviderMetadata: Schema.$Record<
   Schema.String,
-  Schema.NullOr<Schema.Codec<Schema.Json, Schema.Json>>
+  Schema.NullOr<Schema.Codec<Schema.Json>>
 > = Schema.Record(Schema.String, Schema.NullOr(Schema.Json))
 
 /**
@@ -629,7 +629,7 @@ export const TextPart: Schema.Struct<{
   readonly text: Schema.String
   readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
   readonly metadata: Schema.withDecodingDefault<
-    Schema.Record$<Schema.String, Schema.Codec<Schema.Json, Schema.Json>>
+    Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
 }> = Schema.Struct({
   ...BasePart.fields,
@@ -689,7 +689,7 @@ export const TextStartPart: Schema.Struct<{
   readonly id: Schema.String
   readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
   readonly metadata: Schema.withDecodingDefault<
-    Schema.Record$<Schema.String, Schema.Codec<Schema.Json, Schema.Json>>
+    Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
 }> = Schema.Struct({
   ...BasePart.fields,
@@ -757,7 +757,7 @@ export const TextDeltaPart: Schema.Struct<{
   readonly delta: Schema.String
   readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
   readonly metadata: Schema.withDecodingDefault<
-    Schema.Record$<Schema.String, Schema.Codec<Schema.Json, Schema.Json>>
+    Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
 }> = Schema.Struct({
   ...BasePart.fields,
@@ -818,7 +818,7 @@ export const TextEndPart: Schema.Struct<{
   readonly id: Schema.String
   readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
   readonly metadata: Schema.withDecodingDefault<
-    Schema.Record$<Schema.String, Schema.Codec<Schema.Json, Schema.Json>>
+    Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
 }> = Schema.Struct({
   ...BasePart.fields,
@@ -889,7 +889,7 @@ export const ReasoningPart: Schema.Struct<{
   readonly text: Schema.String
   readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
   readonly metadata: Schema.withDecodingDefault<
-    Schema.Record$<Schema.String, Schema.Codec<Schema.Json, Schema.Json>>
+    Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
 }> = Schema.Struct({
   ...BasePart.fields,
@@ -949,7 +949,7 @@ export const ReasoningStartPart: Schema.Struct<{
   readonly id: Schema.String
   readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
   readonly metadata: Schema.withDecodingDefault<
-    Schema.Record$<Schema.String, Schema.Codec<Schema.Json, Schema.Json>>
+    Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
 }> = Schema.Struct({
   ...BasePart.fields,
@@ -1017,7 +1017,7 @@ export const ReasoningDeltaPart: Schema.Struct<{
   readonly delta: Schema.String
   readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
   readonly metadata: Schema.withDecodingDefault<
-    Schema.Record$<Schema.String, Schema.Codec<Schema.Json, Schema.Json>>
+    Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
 }> = Schema.Struct({
   ...BasePart.fields,
@@ -1078,7 +1078,7 @@ export const ReasoningEndPart: Schema.Struct<{
   readonly id: Schema.String
   readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
   readonly metadata: Schema.withDecodingDefault<
-    Schema.Record$<Schema.String, Schema.Codec<Schema.Json, Schema.Json>>
+    Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
 }> = Schema.Struct({
   ...BasePart.fields,
@@ -1159,7 +1159,7 @@ export const ToolParamsStartPart: Schema.Struct<{
   readonly providerExecuted: Schema.withDecodingDefaultKey<Schema.Boolean>
   readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
   readonly metadata: Schema.withDecodingDefault<
-    Schema.Record$<Schema.String, Schema.Codec<Schema.Json, Schema.Json>>
+    Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
 }> = Schema.Struct({
   ...BasePart.fields,
@@ -1234,7 +1234,7 @@ export const ToolParamsDeltaPart: Schema.Struct<{
   readonly delta: Schema.String
   readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
   readonly metadata: Schema.withDecodingDefault<
-    Schema.Record$<Schema.String, Schema.Codec<Schema.Json, Schema.Json>>
+    Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
 }> = Schema.Struct({
   ...BasePart.fields,
@@ -1299,7 +1299,7 @@ export const ToolParamsEndPart: Schema.Struct<{
   readonly id: Schema.String
   readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
   readonly metadata: Schema.withDecodingDefault<
-    Schema.Record$<Schema.String, Schema.Codec<Schema.Json, Schema.Json>>
+    Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
 }> = Schema.Struct({
   ...BasePart.fields,
@@ -1414,7 +1414,7 @@ export const ToolCallPart: <const Name extends string, Params extends Schema.Top
     readonly providerExecuted: Schema.withDecodingDefaultKey<Schema.Boolean>
     readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
     readonly metadata: Schema.withDecodingDefault<
-      Schema.Record$<Schema.String, Schema.Codec<Schema.Json, Schema.Json>>
+      Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
     >
   }
 > = <const Name extends string, Params extends Schema.Top>(
@@ -1618,9 +1618,9 @@ export const ToolResultPart: <const Name extends string, Success extends Schema.
       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$<
+      readonly metadata: Schema.$Record<
         Schema.String,
-        Schema.NullOr<Schema.Codec<Schema.Json, Schema.Json, never, never>>
+        Schema.NullOr<Schema.Codec<Schema.Json>>
       >
       readonly encodedResult: Schema.toEncoded<Schema.Union<readonly [Success, Failure]>>
       readonly preliminary: Schema.Boolean
@@ -1635,7 +1635,7 @@ export const ToolResultPart: <const Name extends string, Success extends Schema.
       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, Schema.Json, never, never>>>
+        Schema.$Record<Schema.String, Schema.NullOr<Schema.Codec<Schema.Json>>>
       >
       readonly preliminary: Schema.optional<Schema.Boolean>
       readonly id: Schema.String
@@ -1795,7 +1795,7 @@ export const ToolApprovalRequestPart: Schema.Struct<{
   readonly toolCallId: Schema.String
   readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
   readonly metadata: Schema.withDecodingDefault<
-    Schema.Record$<Schema.String, Schema.Codec<Schema.Json, Schema.Json>>
+    Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
 }> = Schema.Struct({
   ...BasePart.fields,
@@ -1888,7 +1888,7 @@ export const FilePart: Schema.Struct<{
   readonly data: Schema.Uint8ArrayFromBase64
   readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
   readonly metadata: Schema.withDecodingDefault<
-    Schema.Record$<Schema.String, Schema.Codec<Schema.Json, Schema.Json>>
+    Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
 }> = Schema.Struct({
   ...BasePart.fields,
@@ -1985,7 +1985,7 @@ export const DocumentSourcePart: Schema.Struct<{
   readonly fileName: Schema.optionalKey<Schema.String>
   readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
   readonly metadata: Schema.withDecodingDefault<
-    Schema.Record$<Schema.String, Schema.Codec<Schema.Json, Schema.Json>>
+    Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
 }> = Schema.Struct({
   ...BasePart.fields,
@@ -2076,7 +2076,7 @@ export const UrlSourcePart: Schema.Struct<{
   readonly title: Schema.String
   readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
   readonly metadata: Schema.withDecodingDefault<
-    Schema.Record$<Schema.String, Schema.Codec<Schema.Json, Schema.Json>>
+    Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
 }> = Schema.Struct({
   ...BasePart.fields,
@@ -2115,7 +2115,7 @@ export const UrlSourcePart: Schema.Struct<{
  * @category schemas
  */
 export const HttpRequestDetails = Schema.Struct({
-  method: Schema.Literals(["GET", "POST", "PATCH", "PUT", "DELETE", "HEAD", "OPTIONS"]),
+  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.UndefinedOr(Schema.String),
@@ -2257,7 +2257,7 @@ export const ResponseMetadataPart: Schema.Struct<{
   readonly request: Schema.UndefinedOr<typeof HttpRequestDetails>
   readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
   readonly metadata: Schema.withDecodingDefault<
-    Schema.Record$<Schema.String, Schema.Codec<Schema.Json, Schema.Json>>
+    Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
 }> = Schema.Struct({
   ...BasePart.fields,
@@ -2464,7 +2464,7 @@ export const FinishPart: Schema.Struct<{
   readonly response: Schema.UndefinedOr<typeof HttpResponseDetails>
   readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
   readonly metadata: Schema.withDecodingDefault<
-    Schema.Record$<Schema.String, Schema.Codec<Schema.Json, Schema.Json>>
+    Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
 }> = Schema.Struct({
   ...BasePart.fields,
@@ -2527,7 +2527,7 @@ export const ErrorPart: Schema.Struct<{
   readonly error: Schema.Unknown
   readonly "~effect/ai/Content/Part": Schema.withDecodingDefaultKey<Schema.tag<"~effect/ai/Content/Part">>
   readonly metadata: Schema.withDecodingDefault<
-    Schema.Record$<Schema.String, Schema.Codec<Schema.Json, Schema.Json>>
+    Schema.$Record<Schema.String, Schema.Codec<Schema.Json>>
   >
 }> = Schema.Struct({
   ...BasePart.fields,

package/src/unstable/ai/ResponseIdTracker.ts

@@ -0,0 +1,97 @@
+/**
+ * @since 4.0.0
+ */
+import * as Effect from "../../Effect.ts"
+import * as Option from "../../Option.ts"
+import * as ServiceMap from "../../ServiceMap.ts"
+import * as Prompt from "./Prompt.ts"
+
+/**
+ * @since 4.0.0
+ * @category models
+ */
+export interface PrepareResult {
+  readonly previousResponseId: string
+  readonly prompt: Prompt.Prompt
+}
+
+/**
+ * @since 4.0.0
+ * @category models
+ */
+export interface Service {
+  clearUnsafe(): void
+  markParts(parts: ReadonlyArray<object>, responseId: string): void
+  prepareUnsafe(prompt: Prompt.Prompt): Option.Option<PrepareResult>
+}
+
+/**
+ * @since 4.0.0
+ * @category Services
+ */
+export class ResponseIdTracker
+  extends ServiceMap.Service<ResponseIdTracker, Service>()("effect/ai/ResponseIdTracker")
+{}
+
+/**
+ * @since 4.0.0
+ * @category constructors
+ */
+export const make: Effect.Effect<Service> = Effect.sync(() => {
+  const sentParts = new Map<object, string>()
+
+  const none = () => {
+    sentParts.clear()
+    return Option.none<PrepareResult>()
+  }
+
+  return {
+    clearUnsafe() {
+      sentParts.clear()
+    },
+    markParts(parts, responseId) {
+      for (let i = 0; i < parts.length; i++) {
+        sentParts.set(parts[i], responseId)
+      }
+    },
+    prepareUnsafe(prompt) {
+      const messages = prompt.content
+
+      let anyTracked = false
+      for (let i = 0; i < messages.length; i++) {
+        if (sentParts.has(messages[i])) {
+          anyTracked = true
+          break
+        }
+      }
+      if (!anyTracked) return none()
+
+      let lastAssistantIndex = -1
+      for (let i = messages.length - 1; i >= 0; i--) {
+        if (messages[i].role === "assistant") {
+          lastAssistantIndex = i
+          break
+        }
+      }
+      if (lastAssistantIndex === -1) return none()
+
+      let responseId: string | undefined
+      for (let i = 0; i < lastAssistantIndex; i++) {
+        const id = sentParts.get(messages[i])
+        if (id === undefined) return none()
+        responseId = id
+      }
+      if (responseId === undefined) return none()
+
+      const partsAfterLastAssistant = messages.slice(lastAssistantIndex + 1)
+      if (partsAfterLastAssistant.length === 0) {
+        return none()
+      }
+
+      return Option.some({
+        previousResponseId: responseId,
+        prompt: Prompt.fromMessages(partsAfterLastAssistant)
+      })
+    }
+  }
+})