@effect/ai 0.8.1 → 0.8.2

package/dist/esm/AiChat.js

@@ -19,111 +19,99 @@ export class AiChat extends /*#__PURE__*/Effect.Tag("@effect/ai/AiChat")() {}
  * @since 1.0.0
  * @category constructors
  */
-export const fromInput = input => Ref.make(AiInput.make(input)).pipe(Effect.bindTo("historyRef"), Effect.bind("completions", () => Completions), Effect.map(({
-  completions,
-  historyRef
-}) => new AiChatImpl(historyRef, completions)));
-class AiChatImpl {
-  historyRef;
-  completions;
-  semaphore = /*#__PURE__*/Effect.unsafeMakeSemaphore(1);
-  constructor(historyRef, completions) {
-    this.historyRef = historyRef;
-    this.completions = completions;
-  }
-  get history() {
-    return Ref.get(this.historyRef);
-  }
-  get export() {
-    return Ref.get(this.historyRef).pipe(Effect.flatMap(Schema.encode(AiInput.Schema)), Effect.orDie);
-  }
-  get exportJson() {
-    return Ref.get(this.historyRef).pipe(Effect.flatMap(Schema.encode(AiInput.SchemaJson)), Effect.orDie);
-  }
-  send(input) {
-    const newParts = AiInput.make(input);
-    return Ref.get(this.historyRef).pipe(Effect.flatMap(parts => {
-      const allParts = Chunk.appendAll(parts, newParts);
-      return this.completions.create(allParts).pipe(Effect.tap(response => {
-        const responseParts = AiInput.make(response);
-        return Ref.set(this.historyRef, Chunk.appendAll(allParts, responseParts));
+export const fromInput = /*#__PURE__*/Effect.fnUntraced(function* (input) {
+  const completions = yield* Completions;
+  const history = yield* Ref.make(AiInput.make(input));
+  const semaphore = yield* Effect.makeSemaphore(1);
+  return AiChat.of({
+    history: Ref.get(history),
+    export: Ref.get(history).pipe(Effect.flatMap(Schema.encode(AiInput.Schema)), Effect.orDie),
+    exportJson: Ref.get(history).pipe(Effect.flatMap(Schema.encode(AiInput.SchemaJson)), Effect.orDie),
+    send(input) {
+      const newParts = AiInput.make(input);
+      return Ref.get(history).pipe(Effect.flatMap(parts => {
+        const allParts = Chunk.appendAll(parts, newParts);
+        return completions.create(allParts).pipe(Effect.tap(response => {
+          const responseParts = AiInput.make(response);
+          return Ref.set(history, Chunk.appendAll(allParts, responseParts));
+        }));
+      }), semaphore.withPermits(1), Effect.withSpan("AiChat.send", {
+        attributes: {
+          input
+        },
+        captureStackTrace: false
       }));
-    }), this.semaphore.withPermits(1), Effect.withSpan("AiChat.send", {
-      attributes: {
-        input
-      },
-      captureStackTrace: false
-    }));
-  }
-  stream(input) {
-    return Stream.suspend(() => {
-      let combined = AiResponse.empty;
-      return Stream.fromChannel(Channel.acquireUseRelease(this.semaphore.take(1).pipe(Effect.zipRight(Ref.get(this.historyRef)), Effect.map(Chunk.appendAll(AiInput.make(input)))), parts => this.completions.stream(parts).pipe(Stream.map(chunk => {
-        combined = combined.concat(chunk);
-        return chunk;
-      }), Stream.toChannel), parts => Effect.zipRight(Ref.set(this.historyRef, Chunk.appendAll(parts, AiInput.make(combined))), this.semaphore.release(1))));
-    }).pipe(Stream.withSpan("AiChat.stream", {
-      attributes: {
-        input
-      },
-      captureStackTrace: false
-    }));
-  }
-  structured(schema, input) {
-    const newParts = AiInput.make(input);
-    return Ref.get(this.historyRef).pipe(Effect.flatMap(parts => {
-      const allParts = Chunk.appendAll(parts, newParts);
-      return this.completions.structured({
-        input: allParts,
-        schema
-      }).pipe(Effect.flatMap(response => {
-        const responseParts = AiInput.make(response);
-        return Effect.as(Ref.set(this.historyRef, Chunk.appendAll(allParts, responseParts)), response.unsafeValue);
+    },
+    stream(input) {
+      return Stream.suspend(() => {
+        let combined = AiResponse.empty;
+        return Stream.fromChannel(Channel.acquireUseRelease(semaphore.take(1).pipe(Effect.zipRight(Ref.get(history)), Effect.map(Chunk.appendAll(AiInput.make(input)))), parts => completions.stream(parts).pipe(Stream.map(chunk => {
+          combined = combined.concat(chunk);
+          return chunk;
+        }), Stream.toChannel), parts => Effect.zipRight(Ref.set(history, Chunk.appendAll(parts, AiInput.make(combined))), semaphore.release(1))));
+      }).pipe(Stream.withSpan("AiChat.stream", {
+        attributes: {
+          input
+        },
+        captureStackTrace: false
       }));
-    }), this.semaphore.withPermits(1), Effect.withSpan("AiChat.structured", {
-      attributes: {
-        input,
-        schema: schema._tag ?? schema.identifier
-      },
-      captureStackTrace: false
-    }));
-  }
-  toolkit(options) {
-    const newParts = AiInput.make(options.input);
-    return Ref.get(this.historyRef).pipe(Effect.flatMap(parts => {
-      const allParts = Chunk.appendAll(parts, newParts);
-      return this.completions.toolkit({
-        ...options,
-        input: allParts
-      }).pipe(Effect.tap(response => {
-        const responseParts = AiInput.make(response);
-        return Ref.set(this.historyRef, Chunk.appendAll(allParts, responseParts));
+    },
+    structured(options) {
+      const newParts = AiInput.make(options.input);
+      return Ref.get(history).pipe(Effect.flatMap(parts => {
+        const allParts = Chunk.appendAll(parts, newParts);
+        return completions.structured({
+          ...options,
+          input: allParts
+        }).pipe(Effect.flatMap(response => {
+          const responseParts = AiInput.make(response);
+          return Effect.as(Ref.set(history, Chunk.appendAll(allParts, responseParts)), response.unsafeValue);
+        }));
+      }), semaphore.withPermits(1), Effect.withSpan("AiChat.structured", {
+        attributes: {
+          input: options.input,
+          schema: "toolCallId" in options ? options.toolCallId : "_tag" in options.schema ? options.schema._tag : options.schema.identifier
+        },
+        captureStackTrace: false
       }));
-    }), this.semaphore.withPermits(1), Effect.withSpan("AiChat.toolkit", {
-      attributes: {
-        input: options.input
-      },
-      captureStackTrace: false
-    }));
-  }
-  toolkitStream(options) {
-    return Stream.suspend(() => {
-      let combined = WithResolved.empty;
-      return Stream.fromChannel(Channel.acquireUseRelease(this.semaphore.take(1).pipe(Effect.zipRight(Ref.get(this.historyRef)), Effect.map(Chunk.appendAll(AiInput.make(options.input)))), parts => this.completions.toolkitStream({
-        ...options,
-        input: parts
-      }).pipe(Stream.map(chunk => {
-        combined = combined.concat(chunk);
-        return chunk;
-      }), Stream.toChannel), parts => Effect.zipRight(Ref.set(this.historyRef, Chunk.appendAll(parts, AiInput.make(combined))), this.semaphore.release(1))));
-    }).pipe(Stream.withSpan("AiChat.toolkitStream", {
-      attributes: {
-        input: options.input
-      },
-      captureStackTrace: false
-    }));
-  }
-}
+    },
+    toolkit(options) {
+      const newParts = AiInput.make(options.input);
+      return Ref.get(history).pipe(Effect.flatMap(parts => {
+        const allParts = Chunk.appendAll(parts, newParts);
+        return completions.toolkit({
+          ...options,
+          input: allParts
+        }).pipe(Effect.tap(response => {
+          const responseParts = AiInput.make(response);
+          return Ref.set(history, Chunk.appendAll(allParts, responseParts));
+        }));
+      }), semaphore.withPermits(1), Effect.withSpan("AiChat.toolkit", {
+        attributes: {
+          input: options.input
+        },
+        captureStackTrace: false
+      }));
+    },
+    toolkitStream(options) {
+      return Stream.suspend(() => {
+        let combined = WithResolved.empty;
+        return Stream.fromChannel(Channel.acquireUseRelease(semaphore.take(1).pipe(Effect.zipRight(Ref.get(history)), Effect.map(Chunk.appendAll(AiInput.make(options.input)))), parts => completions.toolkitStream({
+          ...options,
+          input: parts
+        }).pipe(Stream.map(chunk => {
+          combined = combined.concat(chunk);
+          return chunk;
+        }), Stream.toChannel), parts => Effect.zipRight(Ref.set(history, Chunk.appendAll(parts, AiInput.make(combined))), semaphore.release(1))));
+      }).pipe(Stream.withSpan("AiChat.toolkitStream", {
+        attributes: {
+          input: options.input
+        },
+        captureStackTrace: false
+      }));
+    }
+  });
+});
 /**
  * @since 1.0.0
  * @category constructors

package/dist/esm/AiChat.js.map

@@ -1 +1 @@
-{"version":3,"file":"AiChat.js","names":["Channel","Chunk","Effect","Ref","Schema","Stream","AiInput","AiResponse","WithResolved","Completions","AiChat","Tag","fromInput","input","make","pipe","bindTo","bind","map","completions","historyRef","AiChatImpl","semaphore","unsafeMakeSemaphore","constructor","history","get","export","flatMap","encode","orDie","exportJson","SchemaJson","send","newParts","parts","allParts","appendAll","create","tap","response","responseParts","set","withPermits","withSpan","attributes","captureStackTrace","stream","suspend","combined","empty","fromChannel","acquireUseRelease","take","zipRight","chunk","concat","toChannel","release","structured","schema","as","unsafeValue","_tag","identifier","toolkit","options","toolkitStream","fromExport","data","decodeUnknown","fromJson","decode"],"sources":["../../src/AiChat.ts"],"sourcesContent":[null],"mappings":"AAAA;;;AAGA,OAAO,KAAKA,OAAO,MAAM,gBAAgB;AACzC,OAAO,KAAKC,KAAK,MAAM,cAAc;AACrC,OAAO,KAAKC,MAAM,MAAM,eAAe;AAEvC,OAAO,KAAKC,GAAG,MAAM,YAAY;AACjC,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,OAAO,KAAKC,MAAM,MAAM,eAAe;AAGvC,OAAO,KAAKC,OAAO,MAAM,cAAc;AACvC,SAASC,UAAU,EAAEC,YAAY,QAAQ,iBAAiB;AAE1D,SAASC,WAAW,QAAQ,kBAAkB;AAE9C;;;;AAIA,OAAM,MAAOC,MAAO,sBAAQR,MAAM,CAACS,GAAG,CAAC,mBAAmB,CAAC,EAGxD;AAgDH;;;;AAIA,OAAO,MAAMC,SAAS,GAAIC,KAAoB,IAC5CV,GAAG,CAACW,IAAI,CAACR,OAAO,CAACQ,IAAI,CAACD,KAAK,CAAC,CAAC,CAACE,IAAI,CAChCb,MAAM,CAACc,MAAM,CAAC,YAAY,CAAC,EAC3Bd,MAAM,CAACe,IAAI,CAAC,aAAa,EAAE,MAAMR,WAAW,CAAC,EAC7CP,MAAM,CAACgB,GAAG,CAAC,CAAC;EAAEC,WAAW;EAAEC;AAAU,CAAE,KAAK,IAAIC,UAAU,CAACD,UAAU,EAAED,WAAW,CAAC,CAAC,CACrF;AAEH,MAAME,UAAU;EAIHD,UAAA;EACAD,WAAA;EAJFG,SAAS,gBAAGpB,MAAM,CAACqB,mBAAmB,CAAC,CAAC,CAAC;EAElDC,YACWJ,UAAoC,EACpCD,WAAgC;IADhC,KAAAC,UAAU,GAAVA,UAAU;IACV,KAAAD,WAAW,GAAXA,WAAW;EACnB;EAEH,IAAIM,OAAOA,CAAA;IACT,OAAOtB,GAAG,CAACuB,GAAG,CAAC,IAAI,CAACN,UAAU,CAAC;EACjC;EAEA,IAAIO,MAAMA,CAAA;IACR,OAAOxB,GAAG,CAACuB,GAAG,CAAC,IAAI,CAACN,UAAU,CAAC,CAACL,IAAI,CAClCb,MAAM,CAAC0B,OAAO,CAACxB,MAAM,CAACyB,MAAM,CAACvB,OAAO,CAACF,MAAM,CAAC,CAAC,EAC7CF,MAAM,CAAC4B,KAAK,CACb;EACH;EAEA,IAAIC,UAAUA,CAAA;IACZ,OAAO5B,GAAG,CAACuB,GAAG,CAAC,IAAI,CAACN,UAAU,CAAC,CAACL,IAAI,CAClCb,MAAM,CAAC0B,OAAO,CAACxB,MAAM,CAACyB,MAAM,CAACvB,OAAO,CAAC0B,UAAU,CAAC,CAAC,EACjD9B,MAAM,CAAC4B,KAAK,CACb;EACH;EAEAG,IAAIA,CAACpB,KAAoB;IACvB,MAAMqB,QAAQ,GAAG5B,OAAO,CAACQ,IAAI,CAACD,KAAK,CAAC;IACpC,OAAOV,GAAG,CAACuB,GAAG,CAAC,IAAI,CAACN,UAAU,CAAC,CAACL,IAAI,CAClCb,MAAM,CAAC0B,OAAO,CAAEO,KAAK,IAAI;MACvB,MAAMC,QAAQ,GAAGnC,KAAK,CAACoC,SAAS,CAACF,KAAK,EAAED,QAAQ,CAAC;MACjD,OAAO,IAAI,CAACf,WAAW,CAACmB,MAAM,CAACF,QAAQ,CAAC,CAACrB,IAAI,CAC3Cb,MAAM,CAACqC,GAAG,CAAEC,QAAQ,IAAI;QACtB,MAAMC,aAAa,GAAGnC,OAAO,CAACQ,IAAI,CAAC0B,QAAQ,CAAC;QAC5C,OAAOrC,GAAG,CAACuC,GAAG,CAAC,IAAI,CAACtB,UAAU,EAAEnB,KAAK,CAACoC,SAAS,CAACD,QAAQ,EAAEK,aAAa,CAAC,CAAC;MAC3E,CAAC,CAAC,CACH;IACH,CAAC,CAAC,EACF,IAAI,CAACnB,SAAS,CAACqB,WAAW,CAAC,CAAC,CAAC,EAC7BzC,MAAM,CAAC0C,QAAQ,CAAC,aAAa,EAAE;MAAEC,UAAU,EAAE;QAAEhC;MAAK,CAAE;MAAEiC,iBAAiB,EAAE;IAAK,CAAE,CAAC,CACpF;EACH;EAEAC,MAAMA,CAAClC,KAAoB;IACzB,OAAOR,MAAM,CAAC2C,OAAO,CAAC,MAAK;MACzB,IAAIC,QAAQ,GAAG1C,UAAU,CAAC2C,KAAK;MAC/B,OAAO7C,MAAM,CAAC8C,WAAW,CAACnD,OAAO,CAACoD,iBAAiB,CACjD,IAAI,CAAC9B,SAAS,CAAC+B,IAAI,CAAC,CAAC,CAAC,CAACtC,IAAI,CACzBb,MAAM,CAACoD,QAAQ,CAACnD,GAAG,CAACuB,GAAG,CAAC,IAAI,CAACN,UAAU,CAAC,CAAC,EACzClB,MAAM,CAACgB,GAAG,CAACjB,KAAK,CAACoC,SAAS,CAAC/B,OAAO,CAACQ,IAAI,CAACD,KAAK,CAAC,CAAC,CAAC,CACjD,EACAsB,KAAK,IACJ,IAAI,CAAChB,WAAW,CAAC4B,MAAM,CAACZ,KAAK,CAAC,CAACpB,IAAI,CACjCV,MAAM,CAACa,GAAG,CAAEqC,KAAK,IAAI;QACnBN,QAAQ,GAAGA,QAAQ,CAACO,MAAM,CAACD,KAAK,CAAC;QACjC,OAAOA,KAAK;MACd,CAAC,CAAC,EACFlD,MAAM,CAACoD,SAAS,CACjB,EACFtB,KAAK,IACJjC,MAAM,CAACoD,QAAQ,CACbnD,GAAG,CAACuC,GAAG,CAAC,IAAI,CAACtB,UAAU,EAAEnB,KAAK,CAACoC,SAAS,CAACF,KAAK,EAAE7B,OAAO,CAACQ,IAAI,CAACmC,QAAQ,CAAC,CAAC,CAAC,EACxE,IAAI,CAAC3B,SAAS,CAACoC,OAAO,CAAC,CAAC,CAAC,CAC1B,CACJ,CAAC;IACJ,CAAC,CAAC,CAAC3C,IAAI,CAACV,MAAM,CAACuC,QAAQ,CAAC,eAAe,EAAE;MAAEC,UAAU,EAAE;QAAEhC;MAAK,CAAE;MAAEiC,iBAAiB,EAAE;IAAK,CAAE,CAAC,CAAC;EAChG;EAEAa,UAAUA,CACRC,MAA6C,EAC7C/C,KAAoB;IAEpB,MAAMqB,QAAQ,GAAG5B,OAAO,CAACQ,IAAI,CAACD,KAAK,CAAC;IACpC,OAAOV,GAAG,CAACuB,GAAG,CAAC,IAAI,CAACN,UAAU,CAAC,CAACL,IAAI,CAClCb,MAAM,CAAC0B,OAAO,CAAEO,KAAK,IAAI;MACvB,MAAMC,QAAQ,GAAGnC,KAAK,CAACoC,SAAS,CAACF,KAAK,EAAED,QAAQ,CAAC;MACjD,OAAO,IAAI,CAACf,WAAW,CAACwC,UAAU,CAAC;QACjC9C,KAAK,EAAEuB,QAAQ;QACfwB;OACD,CAAC,CAAC7C,IAAI,CACLb,MAAM,CAAC0B,OAAO,CAAEY,QAAQ,IAAI;QAC1B,MAAMC,aAAa,GAAGnC,OAAO,CAACQ,IAAI,CAAC0B,QAAQ,CAAC;QAC5C,OAAOtC,MAAM,CAAC2D,EAAE,CACd1D,GAAG,CAACuC,GAAG,CAAC,IAAI,CAACtB,UAAU,EAAEnB,KAAK,CAACoC,SAAS,CAACD,QAAQ,EAAEK,aAAa,CAAC,CAAC,EAClED,QAAQ,CAACsB,WAAW,CACrB;MACH,CAAC,CAAC,CACH;IACH,CAAC,CAAC,EACF,IAAI,CAACxC,SAAS,CAACqB,WAAW,CAAC,CAAC,CAAC,EAC7BzC,MAAM,CAAC0C,QAAQ,CAAC,mBAAmB,EAAE;MACnCC,UAAU,EAAE;QAAEhC,KAAK;QAAE+C,MAAM,EAAEA,MAAM,CAACG,IAAI,IAAIH,MAAM,CAACI;MAAU,CAAE;MAC/DlB,iBAAiB,EAAE;KACpB,CAAC,CACH;EACH;EAEAmB,OAAOA,CACLC,OAKC;IAMD,MAAMhC,QAAQ,GAAG5B,OAAO,CAACQ,IAAI,CAACoD,OAAO,CAACrD,KAAK,CAAC;IAC5C,OAAOV,GAAG,CAACuB,GAAG,CAAC,IAAI,CAACN,UAAU,CAAC,CAACL,IAAI,CAClCb,MAAM,CAAC0B,OAAO,CAAEO,KAAK,IAAI;MACvB,MAAMC,QAAQ,GAAGnC,KAAK,CAACoC,SAAS,CAACF,KAAK,EAAED,QAAQ,CAAC;MACjD,OAAO,IAAI,CAACf,WAAW,CAAC8C,OAAO,CAAC;QAC9B,GAAGC,OAAO;QACVrD,KAAK,EAAEuB;OACR,CAAC,CAACrB,IAAI,CACLb,MAAM,CAACqC,GAAG,CAAEC,QAAQ,IAAI;QACtB,MAAMC,aAAa,GAAGnC,OAAO,CAACQ,IAAI,CAAC0B,QAAQ,CAAC;QAC5C,OAAOrC,GAAG,CAACuC,GAAG,CAAC,IAAI,CAACtB,UAAU,EAAEnB,KAAK,CAACoC,SAAS,CAACD,QAAQ,EAAEK,aAAa,CAAC,CAAC;MAC3E,CAAC,CAAC,CACH;IACH,CAAC,CAAC,EACF,IAAI,CAACnB,SAAS,CAACqB,WAAW,CAAC,CAAC,CAAC,EAC7BzC,MAAM,CAAC0C,QAAQ,CAAC,gBAAgB,EAAE;MAAEC,UAAU,EAAE;QAAEhC,KAAK,EAAEqD,OAAO,CAACrD;MAAK,CAAE;MAAEiC,iBAAiB,EAAE;IAAK,CAAE,CAAC,CACtG;EACH;EAEAqB,aAAaA,CACXD,OAKC;IAMD,OAAO7D,MAAM,CAAC2C,OAAO,CAAC,MAAK;MACzB,IAAIC,QAAQ,GAAGzC,YAAY,CAAC0C,KAAoD;MAChF,OAAO7C,MAAM,CAAC8C,WAAW,CAACnD,OAAO,CAACoD,iBAAiB,CACjD,IAAI,CAAC9B,SAAS,CAAC+B,IAAI,CAAC,CAAC,CAAC,CAACtC,IAAI,CACzBb,MAAM,CAACoD,QAAQ,CAACnD,GAAG,CAACuB,GAAG,CAAC,IAAI,CAACN,UAAU,CAAC,CAAC,EACzClB,MAAM,CAACgB,GAAG,CAACjB,KAAK,CAACoC,SAAS,CAAC/B,OAAO,CAACQ,IAAI,CAACoD,OAAO,CAACrD,KAAK,CAAC,CAAC,CAAC,CACzD,EACAsB,KAAK,IACJ,IAAI,CAAChB,WAAW,CAACgD,aAAa,CAAC;QAC7B,GAAGD,OAAO;QACVrD,KAAK,EAAEsB;OACR,CAAC,CAACpB,IAAI,CACLV,MAAM,CAACa,GAAG,CAAEqC,KAAK,IAAI;QACnBN,QAAQ,GAAGA,QAAQ,CAACO,MAAM,CAACD,KAAK,CAAC;QACjC,OAAOA,KAAK;MACd,CAAC,CAAC,EACFlD,MAAM,CAACoD,SAAS,CACjB,EACFtB,KAAK,IACJjC,MAAM,CAACoD,QAAQ,CACbnD,GAAG,CAACuC,GAAG,CAAC,IAAI,CAACtB,UAAU,EAAEnB,KAAK,CAACoC,SAAS,CAACF,KAAK,EAAE7B,OAAO,CAACQ,IAAI,CAACmC,QAAQ,CAAC,CAAC,CAAC,EACxE,IAAI,CAAC3B,SAAS,CAACoC,OAAO,CAAC,CAAC,CAAC,CAC1B,CACJ,CAAC;IACJ,CAAC,CAAC,CAAC3C,IAAI,CAACV,MAAM,CAACuC,QAAQ,CAAC,sBAAsB,EAAE;MAAEC,UAAU,EAAE;QAAEhC,KAAK,EAAEqD,OAAO,CAACrD;MAAK,CAAE;MAAEiC,iBAAiB,EAAE;IAAK,CAAE,CAAC,CAAC;EACtH;;AAGF;;;;AAIA,OAAO,MAAMI,KAAK,gBAAsDtC,SAAS,CAACN,OAAO,CAAC4C,KAAK,CAAC;AAEhG;;;;AAIA,OAAO,MAAMkB,UAAU,GAAIC,IAAa,IACtCjE,MAAM,CAACkE,aAAa,CAAChE,OAAO,CAACF,MAAM,CAAC,CAACiE,IAAI,CAAC,CAACtD,IAAI,CAC7Cb,MAAM,CAAC0B,OAAO,CAAChB,SAAS,CAAC,CAC1B;AAEH;;;;AAIA,OAAO,MAAM2D,QAAQ,GAAIF,IAAY,IACnCjE,MAAM,CAACoE,MAAM,CAAClE,OAAO,CAAC0B,UAAU,CAAC,CAACqC,IAAI,CAAC,CAACtD,IAAI,CAC1Cb,MAAM,CAAC0B,OAAO,CAAChB,SAAS,CAAC,CAC1B","ignoreList":[]}
+{"version":3,"file":"AiChat.js","names":["Channel","Chunk","Effect","Ref","Schema","Stream","AiInput","AiResponse","WithResolved","Completions","AiChat","Tag","fromInput","fnUntraced","input","completions","history","make","semaphore","makeSemaphore","of","get","export","pipe","flatMap","encode","orDie","exportJson","SchemaJson","send","newParts","parts","allParts","appendAll","create","tap","response","responseParts","set","withPermits","withSpan","attributes","captureStackTrace","stream","suspend","combined","empty","fromChannel","acquireUseRelease","take","zipRight","map","chunk","concat","toChannel","release","structured","options","as","unsafeValue","schema","toolCallId","_tag","identifier","toolkit","toolkitStream","fromExport","data","decodeUnknown","fromJson","decode"],"sources":["../../src/AiChat.ts"],"sourcesContent":[null],"mappings":"AAAA;;;AAGA,OAAO,KAAKA,OAAO,MAAM,gBAAgB;AACzC,OAAO,KAAKC,KAAK,MAAM,cAAc;AACrC,OAAO,KAAKC,MAAM,MAAM,eAAe;AAEvC,OAAO,KAAKC,GAAG,MAAM,YAAY;AACjC,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,OAAO,KAAKC,MAAM,MAAM,eAAe;AAGvC,OAAO,KAAKC,OAAO,MAAM,cAAc;AACvC,SAASC,UAAU,EAAEC,YAAY,QAAQ,iBAAiB;AAE1D,SAASC,WAAW,QAAQ,kBAAkB;AAE9C;;;;AAIA,OAAM,MAAOC,MAAO,sBAAQR,MAAM,CAACS,GAAG,CAAC,mBAAmB,CAAC,EAGxD;AAuDH;;;;AAIA,OAAO,MAAMC,SAAS,gBAAGV,MAAM,CAACW,UAAU,CACxC,WAAUC,KAAoB;EAC5B,MAAMC,WAAW,GAAG,OAAON,WAAW;EACtC,MAAMO,OAAO,GAAG,OAAOb,GAAG,CAACc,IAAI,CAACX,OAAO,CAACW,IAAI,CAACH,KAAK,CAAC,CAAC;EACpD,MAAMI,SAAS,GAAG,OAAOhB,MAAM,CAACiB,aAAa,CAAC,CAAC,CAAC;EAEhD,OAAOT,MAAM,CAACU,EAAE,CAAC;IACfJ,OAAO,EAAEb,GAAG,CAACkB,GAAG,CAACL,OAAO,CAAC;IACzBM,MAAM,EAAEnB,GAAG,CAACkB,GAAG,CAACL,OAAO,CAAC,CAACO,IAAI,CAC3BrB,MAAM,CAACsB,OAAO,CAACpB,MAAM,CAACqB,MAAM,CAACnB,OAAO,CAACF,MAAM,CAAC,CAAC,EAC7CF,MAAM,CAACwB,KAAK,CACb;IACDC,UAAU,EAAExB,GAAG,CAACkB,GAAG,CAACL,OAAO,CAAC,CAACO,IAAI,CAC/BrB,MAAM,CAACsB,OAAO,CAACpB,MAAM,CAACqB,MAAM,CAACnB,OAAO,CAACsB,UAAU,CAAC,CAAC,EACjD1B,MAAM,CAACwB,KAAK,CACb;IACDG,IAAIA,CAACf,KAAK;MACR,MAAMgB,QAAQ,GAAGxB,OAAO,CAACW,IAAI,CAACH,KAAK,CAAC;MACpC,OAAOX,GAAG,CAACkB,GAAG,CAACL,OAAO,CAAC,CAACO,IAAI,CAC1BrB,MAAM,CAACsB,OAAO,CAAEO,KAAK,IAAI;QACvB,MAAMC,QAAQ,GAAG/B,KAAK,CAACgC,SAAS,CAACF,KAAK,EAAED,QAAQ,CAAC;QACjD,OAAOf,WAAW,CAACmB,MAAM,CAACF,QAAQ,CAAC,CAACT,IAAI,CACtCrB,MAAM,CAACiC,GAAG,CAAEC,QAAQ,IAAI;UACtB,MAAMC,aAAa,GAAG/B,OAAO,CAACW,IAAI,CAACmB,QAAQ,CAAC;UAC5C,OAAOjC,GAAG,CAACmC,GAAG,CAACtB,OAAO,EAAEf,KAAK,CAACgC,SAAS,CAACD,QAAQ,EAAEK,aAAa,CAAC,CAAC;QACnE,CAAC,CAAC,CACH;MACH,CAAC,CAAC,EACFnB,SAAS,CAACqB,WAAW,CAAC,CAAC,CAAC,EACxBrC,MAAM,CAACsC,QAAQ,CAAC,aAAa,EAAE;QAC7BC,UAAU,EAAE;UAAE3B;QAAK,CAAE;QACrB4B,iBAAiB,EAAE;OACpB,CAAC,CACH;IACH,CAAC;IACDC,MAAMA,CAAC7B,KAAK;MACV,OAAOT,MAAM,CAACuC,OAAO,CAAC,MAAK;QACzB,IAAIC,QAAQ,GAAGtC,UAAU,CAACuC,KAAK;QAC/B,OAAOzC,MAAM,CAAC0C,WAAW,CAAC/C,OAAO,CAACgD,iBAAiB,CACjD9B,SAAS,CAAC+B,IAAI,CAAC,CAAC,CAAC,CAAC1B,IAAI,CACpBrB,MAAM,CAACgD,QAAQ,CAAC/C,GAAG,CAACkB,GAAG,CAACL,OAAO,CAAC,CAAC,EACjCd,MAAM,CAACiD,GAAG,CAAClD,KAAK,CAACgC,SAAS,CAAC3B,OAAO,CAACW,IAAI,CAACH,KAAK,CAAC,CAAC,CAAC,CACjD,EACAiB,KAAK,IACJhB,WAAW,CAAC4B,MAAM,CAACZ,KAAK,CAAC,CAACR,IAAI,CAC5BlB,MAAM,CAAC8C,GAAG,CAAEC,KAAK,IAAI;UACnBP,QAAQ,GAAGA,QAAQ,CAACQ,MAAM,CAACD,KAAK,CAAC;UACjC,OAAOA,KAAK;QACd,CAAC,CAAC,EACF/C,MAAM,CAACiD,SAAS,CACjB,EACFvB,KAAK,IACJ7B,MAAM,CAACgD,QAAQ,CACb/C,GAAG,CAACmC,GAAG,CAACtB,OAAO,EAAEf,KAAK,CAACgC,SAAS,CAACF,KAAK,EAAEzB,OAAO,CAACW,IAAI,CAAC4B,QAAQ,CAAC,CAAC,CAAC,EAChE3B,SAAS,CAACqC,OAAO,CAAC,CAAC,CAAC,CACrB,CACJ,CAAC;MACJ,CAAC,CAAC,CAAChC,IAAI,CAAClB,MAAM,CAACmC,QAAQ,CAAC,eAAe,EAAE;QACvCC,UAAU,EAAE;UAAE3B;QAAK,CAAE;QACrB4B,iBAAiB,EAAE;OACpB,CAAC,CAAC;IACL,CAAC;IACDc,UAAUA,CAACC,OAAO;MAChB,MAAM3B,QAAQ,GAAGxB,OAAO,CAACW,IAAI,CAACwC,OAAO,CAAC3C,KAAK,CAAC;MAC5C,OAAOX,GAAG,CAACkB,GAAG,CAACL,OAAO,CAAC,CAACO,IAAI,CAC1BrB,MAAM,CAACsB,OAAO,CAAEO,KAAK,IAAI;QACvB,MAAMC,QAAQ,GAAG/B,KAAK,CAACgC,SAAS,CAACF,KAAK,EAAED,QAAQ,CAAC;QACjD,OAAOf,WAAW,CAACyC,UAAU,CAAC;UAC5B,GAAGC,OAAO;UACV3C,KAAK,EAAEkB;SACD,CAAC,CAACT,IAAI,CACZrB,MAAM,CAACsB,OAAO,CAAEY,QAAQ,IAAI;UAC1B,MAAMC,aAAa,GAAG/B,OAAO,CAACW,IAAI,CAACmB,QAAQ,CAAC;UAC5C,OAAOlC,MAAM,CAACwD,EAAE,CACdvD,GAAG,CAACmC,GAAG,CAACtB,OAAO,EAAEf,KAAK,CAACgC,SAAS,CAACD,QAAQ,EAAEK,aAAa,CAAC,CAAC,EAC1DD,QAAQ,CAACuB,WAAW,CACrB;QACH,CAAC,CAAC,CACH;MACH,CAAC,CAAC,EACFzC,SAAS,CAACqB,WAAW,CAAC,CAAC,CAAC,EACxBrC,MAAM,CAACsC,QAAQ,CAAC,mBAAmB,EAAE;QACnCC,UAAU,EAAE;UACV3B,KAAK,EAAE2C,OAAO,CAAC3C,KAAK;UACpB8C,MAAM,EAAE,YAAY,IAAIH,OAAO,GAC3BA,OAAO,CAACI,UAAU,GAClB,MAAM,IAAIJ,OAAO,CAACG,MAAM,GACxBH,OAAO,CAACG,MAAM,CAACE,IAAI,GACnBL,OAAO,CAACG,MAAM,CAACG;SACpB;QACDrB,iBAAiB,EAAE;OACpB,CAAC,CACH;IACH,CAAC;IACDsB,OAAOA,CAACP,OAAO;MACb,MAAM3B,QAAQ,GAAGxB,OAAO,CAACW,IAAI,CAACwC,OAAO,CAAC3C,KAAK,CAAC;MAC5C,OAAOX,GAAG,CAACkB,GAAG,CAACL,OAAO,CAAC,CAACO,IAAI,CAC1BrB,MAAM,CAACsB,OAAO,CAAEO,KAAK,IAAI;QACvB,MAAMC,QAAQ,GAAG/B,KAAK,CAACgC,SAAS,CAACF,KAAK,EAAED,QAAQ,CAAC;QACjD,OAAOf,WAAW,CAACiD,OAAO,CAAC;UACzB,GAAGP,OAAO;UACV3C,KAAK,EAAEkB;SACR,CAAC,CAACT,IAAI,CACLrB,MAAM,CAACiC,GAAG,CAAEC,QAAQ,IAAI;UACtB,MAAMC,aAAa,GAAG/B,OAAO,CAACW,IAAI,CAACmB,QAAQ,CAAC;UAC5C,OAAOjC,GAAG,CAACmC,GAAG,CAACtB,OAAO,EAAEf,KAAK,CAACgC,SAAS,CAACD,QAAQ,EAAEK,aAAa,CAAC,CAAC;QACnE,CAAC,CAAC,CACH;MACH,CAAC,CAAC,EACFnB,SAAS,CAACqB,WAAW,CAAC,CAAC,CAAC,EACxBrC,MAAM,CAACsC,QAAQ,CAAC,gBAAgB,EAAE;QAChCC,UAAU,EAAE;UAAE3B,KAAK,EAAE2C,OAAO,CAAC3C;QAAK,CAAE;QACpC4B,iBAAiB,EAAE;OACpB,CAAC,CACH;IACH,CAAC;IACDuB,aAAaA,CAAyCR,OAKrD;MAKC,OAAOpD,MAAM,CAACuC,OAAO,CAAC,MAAK;QACzB,IAAIC,QAAQ,GAAGrC,YAAY,CAACsC,KAAoD;QAChF,OAAOzC,MAAM,CAAC0C,WAAW,CAAC/C,OAAO,CAACgD,iBAAiB,CACjD9B,SAAS,CAAC+B,IAAI,CAAC,CAAC,CAAC,CAAC1B,IAAI,CACpBrB,MAAM,CAACgD,QAAQ,CAAC/C,GAAG,CAACkB,GAAG,CAACL,OAAO,CAAC,CAAC,EACjCd,MAAM,CAACiD,GAAG,CAAClD,KAAK,CAACgC,SAAS,CAAC3B,OAAO,CAACW,IAAI,CAACwC,OAAO,CAAC3C,KAAK,CAAC,CAAC,CAAC,CACzD,EACAiB,KAAK,IACJhB,WAAW,CAACkD,aAAa,CAAC;UACxB,GAAGR,OAAO;UACV3C,KAAK,EAAEiB;SACR,CAAC,CAACR,IAAI,CACLlB,MAAM,CAAC8C,GAAG,CAAEC,KAAK,IAAI;UACnBP,QAAQ,GAAGA,QAAQ,CAACQ,MAAM,CAACD,KAAK,CAAC;UACjC,OAAOA,KAAK;QACd,CAAC,CAAC,EACF/C,MAAM,CAACiD,SAAS,CACjB,EACFvB,KAAK,IACJ7B,MAAM,CAACgD,QAAQ,CACb/C,GAAG,CAACmC,GAAG,CAACtB,OAAO,EAAEf,KAAK,CAACgC,SAAS,CAACF,KAAK,EAAEzB,OAAO,CAACW,IAAI,CAAC4B,QAAQ,CAAC,CAAC,CAAC,EAChE3B,SAAS,CAACqC,OAAO,CAAC,CAAC,CAAC,CACrB,CACJ,CAAC;MACJ,CAAC,CAAC,CAAChC,IAAI,CAAClB,MAAM,CAACmC,QAAQ,CAAC,sBAAsB,EAAE;QAC9CC,UAAU,EAAE;UAAE3B,KAAK,EAAE2C,OAAO,CAAC3C;QAAK,CAAE;QACpC4B,iBAAiB,EAAE;OACpB,CAAC,CAAC;IACL;GACD,CAAC;AACJ,CAAC,CACF;AAED;;;;AAIA,OAAO,MAAMI,KAAK,gBAAsDlC,SAAS,CAACN,OAAO,CAACwC,KAAK,CAAC;AAEhG;;;;AAIA,OAAO,MAAMoB,UAAU,GAAIC,IAAa,IACtC/D,MAAM,CAACgE,aAAa,CAAC9D,OAAO,CAACF,MAAM,CAAC,CAAC+D,IAAI,CAAC,CAAC5C,IAAI,CAC7CrB,MAAM,CAACsB,OAAO,CAACZ,SAAS,CAAC,CAC1B;AAEH;;;;AAIA,OAAO,MAAMyD,QAAQ,GAAIF,IAAY,IACnC/D,MAAM,CAACkE,MAAM,CAAChE,OAAO,CAACsB,UAAU,CAAC,CAACuC,IAAI,CAAC,CAAC5C,IAAI,CAC1CrB,MAAM,CAACsB,OAAO,CAACZ,SAAS,CAAC,CAC1B","ignoreList":[]}

package/dist/esm/AiTelemetry.js

@@ -0,0 +1,42 @@
+/**
+ * @since 1.0.0
+ */
+import { dual } from "effect/Function";
+import * as Predicate from "effect/Predicate";
+import * as String from "effect/String";
+/**
+ * @since 1.0.0
+ * @category utilities
+ */
+export const addSpanAttributes = (keyPrefix, transformKey) => (span, attributes) => {
+  for (const [key, value] of Object.entries(attributes)) {
+    if (Predicate.isNotNullable(value)) {
+      span.attribute(`${keyPrefix}.${transformKey(key)}`, value);
+    }
+  }
+};
+const addSpanBaseAttributes = /*#__PURE__*/addSpanAttributes("gen_ai", String.camelToSnake);
+const addSpanOperationAttributes = /*#__PURE__*/addSpanAttributes("gen_ai.operation", String.camelToSnake);
+const addSpanRequestAttributes = /*#__PURE__*/addSpanAttributes("gen_ai.request", String.camelToSnake);
+const addSpanResponseAttributes = /*#__PURE__*/addSpanAttributes("gen_ai.response", String.camelToSnake);
+const addSpanTokenAttributes = /*#__PURE__*/addSpanAttributes("gen_ai.token", String.camelToSnake);
+const addSpanUsageAttributes = /*#__PURE__*/addSpanAttributes("gen_ai.usage", String.camelToSnake);
+/**
+ * Applies the specified GenAI telemetry attributes to the provided `Span`.
+ *
+ * **NOTE**: This method will mutate the `Span` **in-place**.
+ *
+ * @since 1.0.0
+ * @since utilities
+ */
+export const addGenAIAnnotations = /*#__PURE__*/dual(2, (span, options) => {
+  addSpanBaseAttributes(span, {
+    system: options.system
+  });
+  if (Predicate.isNotNullable(options.operation)) addSpanOperationAttributes(span, options.operation);
+  if (Predicate.isNotNullable(options.request)) addSpanRequestAttributes(span, options.request);
+  if (Predicate.isNotNullable(options.response)) addSpanResponseAttributes(span, options.response);
+  if (Predicate.isNotNullable(options.token)) addSpanTokenAttributes(span, options.token);
+  if (Predicate.isNotNullable(options.usage)) addSpanUsageAttributes(span, options.usage);
+});
+//# sourceMappingURL=AiTelemetry.js.map

package/dist/esm/AiTelemetry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AiTelemetry.js","names":["dual","Predicate","String","addSpanAttributes","keyPrefix","transformKey","span","attributes","key","value","Object","entries","isNotNullable","attribute","addSpanBaseAttributes","camelToSnake","addSpanOperationAttributes","addSpanRequestAttributes","addSpanResponseAttributes","addSpanTokenAttributes","addSpanUsageAttributes","addGenAIAnnotations","options","system","operation","request","response","token","usage"],"sources":["../../src/AiTelemetry.ts"],"sourcesContent":[null],"mappings":"AAAA;;;AAIA,SAASA,IAAI,QAAQ,iBAAiB;AACtC,OAAO,KAAKC,SAAS,MAAM,kBAAkB;AAC7C,OAAO,KAAKC,MAAM,MAAM,eAAe;AA6NvC;;;;AAIA,OAAO,MAAMC,iBAAiB,GAAGA,CAC/BC,SAAiB,EACjBC,YAAqC,KAEvC,CAAyCC,IAAU,EAAEC,UAAsB,KAAU;EACnF,KAAK,MAAM,CAACC,GAAG,EAAEC,KAAK,CAAC,IAAIC,MAAM,CAACC,OAAO,CAACJ,UAAU,CAAC,EAAE;IACrD,IAAIN,SAAS,CAACW,aAAa,CAACH,KAAK,CAAC,EAAE;MAClCH,IAAI,CAACO,SAAS,CAAC,GAAGT,SAAS,IAAIC,YAAY,CAACG,GAAG,CAAC,EAAE,EAAEC,KAAK,CAAC;IAC5D;EACF;AACF,CAAC;AAED,MAAMK,qBAAqB,gBAAGX,iBAAiB,CAAC,QAAQ,EAAED,MAAM,CAACa,YAAY,CAAuB;AACpG,MAAMC,0BAA0B,gBAAGb,iBAAiB,CAAC,kBAAkB,EAAED,MAAM,CAACa,YAAY,CAA4B;AACxH,MAAME,wBAAwB,gBAAGd,iBAAiB,CAAC,gBAAgB,EAAED,MAAM,CAACa,YAAY,CAA0B;AAClH,MAAMG,yBAAyB,gBAAGf,iBAAiB,CAAC,iBAAiB,EAAED,MAAM,CAACa,YAAY,CAA2B;AACrH,MAAMI,sBAAsB,gBAAGhB,iBAAiB,CAAC,cAAc,EAAED,MAAM,CAACa,YAAY,CAAwB;AAC5G,MAAMK,sBAAsB,gBAAGjB,iBAAiB,CAAC,cAAc,EAAED,MAAM,CAACa,YAAY,CAAwB;AAc5G;;;;;;;;AAQA,OAAO,MAAMM,mBAAmB,gBAAGrB,IAAI,CAmBrC,CAAC,EAAE,CAACM,IAAI,EAAEgB,OAAO,KAAI;EACrBR,qBAAqB,CAACR,IAAI,EAAE;IAAEiB,MAAM,EAAED,OAAO,CAACC;EAAM,CAAE,CAAC;EACvD,IAAItB,SAAS,CAACW,aAAa,CAACU,OAAO,CAACE,SAAS,CAAC,EAAER,0BAA0B,CAACV,IAAI,EAAEgB,OAAO,CAACE,SAAS,CAAC;EACnG,IAAIvB,SAAS,CAACW,aAAa,CAACU,OAAO,CAACG,OAAO,CAAC,EAAER,wBAAwB,CAACX,IAAI,EAAEgB,OAAO,CAACG,OAAO,CAAC;EAC7F,IAAIxB,SAAS,CAACW,aAAa,CAACU,OAAO,CAACI,QAAQ,CAAC,EAAER,yBAAyB,CAACZ,IAAI,EAAEgB,OAAO,CAACI,QAAQ,CAAC;EAChG,IAAIzB,SAAS,CAACW,aAAa,CAACU,OAAO,CAACK,KAAK,CAAC,EAAER,sBAAsB,CAACb,IAAI,EAAEgB,OAAO,CAACK,KAAK,CAAC;EACvF,IAAI1B,SAAS,CAACW,aAAa,CAACU,OAAO,CAACM,KAAK,CAAC,EAAER,sBAAsB,CAACd,IAAI,EAAEgB,OAAO,CAACM,KAAK,CAAC;AACzF,CAAC,CAAC","ignoreList":[]}

package/dist/esm/Completions.js

@@ -26,36 +26,43 @@ const constEmptyMap = /*#__PURE__*/new Map();
 export const make = options => Effect.map(Effect.serviceOption(AiInput.SystemInstruction), parentSystem => {
   return Completions.of({
     create(input) {
-      return Effect.serviceOption(AiInput.SystemInstruction).pipe(Effect.flatMap(system => options.create({
+      return Effect.useSpan("Completions.create", {
+        captureStackTrace: false
+      }, span => Effect.serviceOption(AiInput.SystemInstruction).pipe(Effect.flatMap(system => options.create({
         input: AiInput.make(input),
         system: Option.orElse(system, () => parentSystem),
         tools: [],
-        required: false
-      })), Effect.withSpan("Completions.create", {
-        captureStackTrace: false
-      }));
+        required: false,
+        span
+      }))));
     },
     stream(input_) {
       const input = AiInput.make(input_);
-      return Effect.serviceOption(AiInput.SystemInstruction).pipe(Effect.map(system => options.stream({
+      return Effect.makeSpanScoped("Completions.stream", {
+        captureStackTrace: false
+      }).pipe(Effect.zip(Effect.serviceOption(AiInput.SystemInstruction)), Effect.map(([span, system]) => options.stream({
         input: input,
         system: Option.orElse(system, () => parentSystem),
         tools: [],
-        required: false
-      })), Stream.unwrap, Stream.withSpan("Completions.stream", {
-        captureStackTrace: false
-      }));
+        required: false,
+        span
+      })), Stream.unwrapScoped);
     },
     structured(opts) {
       const input = AiInput.make(opts.input);
-      const schema = opts.schema;
-      const decode = Schema.decodeUnknown(schema);
-      const toolId = schema._tag ?? schema.identifier;
-      return Effect.serviceOption(AiInput.SystemInstruction).pipe(Effect.flatMap(system => options.create({
+      const decode = Schema.decodeUnknown(opts.schema);
+      const toolId = "toolCallId" in opts ? opts.toolCallId : "_tag" in opts.schema ? opts.schema._tag : opts.schema.identifier;
+      return Effect.useSpan("Completions.structured", {
+        attributes: {
+          toolId
+        },
+        captureStackTrace: false
+      }, span => Effect.serviceOption(AiInput.SystemInstruction).pipe(Effect.flatMap(system => options.create({
         input: input,
         system: Option.orElse(system, () => parentSystem),
-        tools: [convertTool(schema, true)],
-        required: true
+        tools: [convertTool(toolId, opts.schema, true)],
+        required: true,
+        span
       })), Effect.flatMap(response => Chunk.findFirst(response.parts, part => part._tag === "ToolCall" && part.name === toolId).pipe(Option.match({
         onNone: () => Effect.fail(new AiError({
           module: "Completions",
@@ -75,12 +82,7 @@ export const make = options => Effect.map(Effect.serviceOption(AiInput.SystemIns
             encoded: new Map([[toolCall.id, toolCall.params]])
           }))
         })
-      }), Effect.withSpan("Completions.structured", {
-        attributes: {
-          tool: toolId
-        },
-        captureStackTrace: false
-      }))));
+      })))));
     },
     toolkit({
       concurrency,
@@ -91,25 +93,26 @@ export const make = options => Effect.map(Effect.serviceOption(AiInput.SystemIns
       const input = AiInput.make(inputInput);
       const toolArr = [];
       for (const [, tool] of tools.toolkit.tools) {
-        toolArr.push(convertTool(tool));
+        toolArr.push(convertTool(tool._tag, tool));
       }
-      return Effect.serviceOption(AiInput.SystemInstruction).pipe(Effect.flatMap(system => options.create({
+      return Effect.useSpan("Completions.toolkit", {
+        attributes: {
+          concurrency,
+          required
+        },
+        captureStackTrace: false
+      }, span => Effect.serviceOption(AiInput.SystemInstruction).pipe(Effect.flatMap(system => options.create({
         input: input,
         system: Option.orElse(system, () => parentSystem),
         tools: toolArr,
-        required: required
+        required: required,
+        span
       })), Effect.flatMap(response => resolveParts({
         response,
         tools,
         concurrency,
         method: "toolkit"
-      })), Effect.withSpan("Completions.toolkit", {
-        captureStackTrace: false,
-        attributes: {
-          concurrency,
-          required
-        }
-      }));
+      }))));
     },
     toolkitStream({
       concurrency,
@@ -119,34 +122,35 @@ export const make = options => Effect.map(Effect.serviceOption(AiInput.SystemIns
     }) {
       const toolArr = [];
       for (const [, tool] of tools.toolkit.tools) {
-        toolArr.push(convertTool(tool));
+        toolArr.push(convertTool(tool._tag, tool));
       }
-      return Effect.serviceOption(AiInput.SystemInstruction).pipe(Effect.map(system => options.stream({
+      return Effect.makeSpanScoped("Completions.stream", {
+        captureStackTrace: false,
+        attributes: {
+          required,
+          concurrency
+        }
+      }).pipe(Effect.zip(Effect.serviceOption(AiInput.SystemInstruction)), Effect.map(([span, system]) => options.stream({
         input: AiInput.make(input),
         system: Option.orElse(system, () => parentSystem),
         tools: toolArr,
-        required: required
-      })), Stream.unwrap, Stream.mapEffect(chunk => resolveParts({
+        required: required,
+        span
+      })), Stream.unwrapScoped, Stream.mapEffect(chunk => resolveParts({
         response: chunk,
         tools,
         concurrency,
         method: "toolkitStream"
       }), {
         concurrency: "unbounded"
-      }), Stream.withSpan("Completions.toolkitStream", {
-        captureStackTrace: false,
-        attributes: {
-          concurrency,
-          required
-        }
       }));
     }
   });
 });
-const convertTool = (tool, structured = false) => ({
-  name: tool._tag ?? tool.identifier,
-  description: getDescription(tool.ast),
-  parameters: makeJsonSchema(tool.ast),
+const convertTool = (name, schema, structured = false) => ({
+  name,
+  description: getDescription(schema.ast),
+  parameters: makeJsonSchema(schema.ast),
   structured
 });
 const makeJsonSchema = ast => {

package/dist/esm/Completions.js.map

@@ -1 +1 @@
-{"version":3,"file":"Completions.js","names":["Chunk","Context","Effect","HashMap","JsonSchema","Option","Schema","AST","Stream","AiError","AiInput","WithResolved","Completions","Tag","constEmptyMap","Map","make","options","map","serviceOption","SystemInstruction","parentSystem","of","create","input","pipe","flatMap","system","orElse","tools","required","withSpan","captureStackTrace","stream","input_","unwrap","structured","opts","schema","decode","decodeUnknown","toolId","_tag","identifier","convertTool","response","findFirst","parts","part","name","match","onNone","fail","module","method","description","onSome","toolCall","matchEffect","params","onFailure","cause","onSuccess","resolved","succeed","id","encoded","attributes","tool","toolkit","concurrency","inputInput","toolArr","push","resolveParts","toolkitStream","mapEffect","chunk","getDescription","ast","parameters","makeJsonSchema","$defs","fromAST","definitions","topLevelReferenceStrategy","Object","keys","length","annotations","to","DescriptionAnnotationId","toolNames","toolParts","filter","isEmpty","annotateCurrentSpan","zipRight","forEach","unsafeGet","handler","handlers","decodeParams","encodeSuccess","encode","success","mapError","tap","value","encodedValue","set","discard","as"],"sources":["../../src/Completions.ts"],"sourcesContent":[null],"mappings":"AAAA;;;AAGA,OAAO,KAAKA,KAAK,MAAM,cAAc;AACrC,OAAO,KAAKC,OAAO,MAAM,gBAAgB;AACzC,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,OAAO,KAAKC,OAAO,MAAM,gBAAgB;AACzC,OAAO,KAAKC,UAAU,MAAM,mBAAmB;AAC/C,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,OAAO,KAAKC,GAAG,MAAM,kBAAkB;AACvC,OAAO,KAAKC,MAAM,MAAM,eAAe;AAEvC,SAASC,OAAO,QAAQ,cAAc;AAEtC,OAAO,KAAKC,OAAO,MAAM,cAAc;AAEvC,SAASC,YAAY,QAAQ,iBAAiB;AAG9C;;;;AAIA,OAAM,MAAOC,WAAY,sBAAQX,OAAO,CAACY,GAAG,CAAC,wBAAwB,CAAC,EAGnE;AAwDH,MAAMC,aAAa,gBAAG,IAAIC,GAAG,EAAgB;AAkB7C;;;;AAIA,OAAO,MAAMC,IAAI,GAAIC,OAuBpB,IACCf,MAAM,CAACgB,GAAG,CAAChB,MAAM,CAACiB,aAAa,CAACT,OAAO,CAACU,iBAAiB,CAAC,EAAGC,YAAY,IAAI;EAC3E,OAAOT,WAAW,CAACU,EAAE,CAAC;IACpBC,MAAMA,CAACC,KAAK;MACV,OAAOtB,MAAM,CAACiB,aAAa,CAACT,OAAO,CAACU,iBAAiB,CAAC,CAACK,IAAI,CACzDvB,MAAM,CAACwB,OAAO,CAAEC,MAAM,IACpBV,OAAO,CAACM,MAAM,CAAC;QACbC,KAAK,EAAEd,OAAO,CAACM,IAAI,CAACQ,KAAK,CAAiC;QAC1DG,MAAM,EAAEtB,MAAM,CAACuB,MAAM,CAACD,MAAM,EAAE,MAAMN,YAAY,CAAC;QACjDQ,KAAK,EAAE,EAAE;QACTC,QAAQ,EAAE;OACX,CAAC,CACH,EACD5B,MAAM,CAAC6B,QAAQ,CAAC,oBAAoB,EAAE;QAAEC,iBAAiB,EAAE;MAAK,CAAE,CAAC,CACpE;IACH,CAAC;IACDC,MAAMA,CAACC,MAAM;MACX,MAAMV,KAAK,GAAGd,OAAO,CAACM,IAAI,CAACkB,MAAM,CAAC;MAClC,OAAOhC,MAAM,CAACiB,aAAa,CAACT,OAAO,CAACU,iBAAiB,CAAC,CAACK,IAAI,CACzDvB,MAAM,CAACgB,GAAG,CAAES,MAAM,IAChBV,OAAO,CAACgB,MAAM,CAAC;QACbT,KAAK,EAAEA,KAAqC;QAC5CG,MAAM,EAAEtB,MAAM,CAACuB,MAAM,CAACD,MAAM,EAAE,MAAMN,YAAY,CAAC;QACjDQ,KAAK,EAAE,EAAE;QACTC,QAAQ,EAAE;OACX,CAAC,CACH,EACDtB,MAAM,CAAC2B,MAAM,EACb3B,MAAM,CAACuB,QAAQ,CAAC,oBAAoB,EAAE;QAAEC,iBAAiB,EAAE;MAAK,CAAE,CAAC,CACpE;IACH,CAAC;IACDI,UAAUA,CAACC,IAAI;MACb,MAAMb,KAAK,GAAGd,OAAO,CAACM,IAAI,CAACqB,IAAI,CAACb,KAAK,CAAC;MACtC,MAAMc,MAAM,GAAGD,IAAI,CAACC,MAAM;MAC1B,MAAMC,MAAM,GAAGjC,MAAM,CAACkC,aAAa,CAACF,MAAM,CAAC;MAC3C,MAAMG,MAAM,GAAGH,MAAM,CAACI,IAAI,IAAIJ,MAAM,CAACK,UAAU;MAC/C,OAAOzC,MAAM,CAACiB,aAAa,CAACT,OAAO,CAACU,iBAAiB,CAAC,CAACK,IAAI,CACzDvB,MAAM,CAACwB,OAAO,CAAEC,MAAM,IACpBV,OAAO,CAACM,MAAM,CAAC;QACbC,KAAK,EAAEA,KAAqC;QAC5CG,MAAM,EAAEtB,MAAM,CAACuB,MAAM,CAACD,MAAM,EAAE,MAAMN,YAAY,CAAC;QACjDQ,KAAK,EAAE,CAACe,WAAW,CAACN,MAAM,EAAE,IAAI,CAAC,CAAC;QAClCR,QAAQ,EAAE;OACX,CAAC,CACH,EACD5B,MAAM,CAACwB,OAAO,CAAEmB,QAAQ,IACtB7C,KAAK,CAAC8C,SAAS,CACbD,QAAQ,CAACE,KAAK,EACbC,IAAI,IAA2BA,IAAI,CAACN,IAAI,KAAK,UAAU,IAAIM,IAAI,CAACC,IAAI,KAAKR,MAAM,CACjF,CAAChB,IAAI,CACJpB,MAAM,CAAC6C,KAAK,CAAC;QACXC,MAAM,EAAEA,CAAA,KACNjD,MAAM,CAACkD,IAAI,CACT,IAAI3C,OAAO,CAAC;UACV4C,MAAM,EAAE,aAAa;UACrBC,MAAM,EAAE,YAAY;UACpBC,WAAW,EAAE,cAAcd,MAAM;SAClC,CAAC,CACH;QACHe,MAAM,EAAGC,QAAQ,IACfvD,MAAM,CAACwD,WAAW,CAACnB,MAAM,CAACkB,QAAQ,CAACE,MAAM,CAAC,EAAE;UAC1CC,SAAS,EAAGC,KAAK,IACf,IAAIpD,OAAO,CAAC;YACV4C,MAAM,EAAE,aAAa;YACrBC,MAAM,EAAE,YAAY;YACpBC,WAAW,EAAE,+BAA+Bd,MAAM,cAAc;YAChEoB;WACD,CAAC;UACJC,SAAS,EAAGC,QAAQ,IAClB7D,MAAM,CAAC8D,OAAO,CACZ,IAAIrD,YAAY,CAAC;YACfkC,QAAQ;YACRkB,QAAQ,EAAE,IAAIhD,GAAG,CAAC,CAAC,CAAC0C,QAAQ,CAACQ,EAAE,EAAEF,QAAQ,CAAC,CAAC,CAAC;YAC5CG,OAAO,EAAE,IAAInD,GAAG,CAAC,CAAC,CAAC0C,QAAQ,CAACQ,EAAE,EAAER,QAAQ,CAACE,MAAM,CAAC,CAAC;WAClD,CAAC;SAEP;OACJ,CAAC,EACFzD,MAAM,CAAC6B,QAAQ,CAAC,wBAAwB,EAAE;QACxCoC,UAAU,EAAE;UAAEC,IAAI,EAAE3B;QAAM,CAAE;QAC5BT,iBAAiB,EAAE;OACpB,CAAC,CACH,CACF,CACF;IACH,CAAC;IACDqC,OAAOA,CAAC;MAAEC,WAAW;MAAE9C,KAAK,EAAE+C,UAAU;MAAEzC,QAAQ,GAAG,KAAK;MAAED;IAAK,CAAE;MACjE,MAAML,KAAK,GAAGd,OAAO,CAACM,IAAI,CAACuD,UAAU,CAAC;MACtC,MAAMC,OAAO,GAKR,EAAE;MACP,KAAK,MAAM,GAAGJ,IAAI,CAAC,IAAIvC,KAAK,CAACwC,OAAO,CAACxC,KAAK,EAAE;QAC1C2C,OAAO,CAACC,IAAI,CAAC7B,WAAW,CAACwB,IAAW,CAAC,CAAC;MACxC;MACA,OAAOlE,MAAM,CAACiB,aAAa,CAACT,OAAO,CAACU,iBAAiB,CAAC,CAACK,IAAI,CACzDvB,MAAM,CAACwB,OAAO,CAAEC,MAAM,IACpBV,OAAO,CAACM,MAAM,CAAC;QACbC,KAAK,EAAEA,KAAqC;QAC5CG,MAAM,EAAEtB,MAAM,CAACuB,MAAM,CAACD,MAAM,EAAE,MAAMN,YAAY,CAAC;QACjDQ,KAAK,EAAE2C,OAAO;QACd1C,QAAQ,EAAEA;OACX,CAAC,CACH,EACD5B,MAAM,CAACwB,OAAO,CAAEmB,QAAQ,IAAK6B,YAAY,CAAC;QAAE7B,QAAQ;QAAEhB,KAAK;QAAEyC,WAAW;QAAEhB,MAAM,EAAE;MAAS,CAAE,CAAC,CAAC,EAC/FpD,MAAM,CAAC6B,QAAQ,CAAC,qBAAqB,EAAE;QACrCC,iBAAiB,EAAE,KAAK;QACxBmC,UAAU,EAAE;UACVG,WAAW;UACXxC;;OAEH,CAAC,CACI;IACV,CAAC;IACD6C,aAAaA,CAAC;MAAEL,WAAW;MAAE9C,KAAK;MAAEM,QAAQ,GAAG,KAAK;MAAED;IAAK,CAAE;MAC3D,MAAM2C,OAAO,GAKR,EAAE;MACP,KAAK,MAAM,GAAGJ,IAAI,CAAC,IAAIvC,KAAK,CAACwC,OAAO,CAACxC,KAAK,EAAE;QAC1C2C,OAAO,CAACC,IAAI,CAAC7B,WAAW,CAACwB,IAAW,CAAC,CAAC;MACxC;MACA,OAAOlE,MAAM,CAACiB,aAAa,CAACT,OAAO,CAACU,iBAAiB,CAAC,CAACK,IAAI,CACzDvB,MAAM,CAACgB,GAAG,CAAES,MAAM,IAChBV,OAAO,CAACgB,MAAM,CAAC;QACbT,KAAK,EAAEd,OAAO,CAACM,IAAI,CAACQ,KAAK,CAAiC;QAC1DG,MAAM,EAAEtB,MAAM,CAACuB,MAAM,CAACD,MAAM,EAAE,MAAMN,YAAY,CAAC;QACjDQ,KAAK,EAAE2C,OAAO;QACd1C,QAAQ,EAAEA;OACX,CAAC,CACH,EACDtB,MAAM,CAAC2B,MAAM,EACb3B,MAAM,CAACoE,SAAS,CACbC,KAAK,IAAKH,YAAY,CAAC;QAAE7B,QAAQ,EAAEgC,KAAK;QAAEhD,KAAK;QAAEyC,WAAW;QAAEhB,MAAM,EAAE;MAAe,CAAE,CAAC,EACzF;QAAEgB,WAAW,EAAE;MAAW,CAAE,CAC7B,EACD9D,MAAM,CAACuB,QAAQ,CAAC,2BAA2B,EAAE;QAC3CC,iBAAiB,EAAE,KAAK;QACxBmC,UAAU,EAAE;UACVG,WAAW;UACXxC;;OAEH,CAAC,CACI;IACV;GACD,CAAC;AACJ,CAAC,CAAC;AAEJ,MAAMc,WAAW,GAAGA,CAAUwB,IAA2C,EAAEhC,UAAU,GAAG,KAAK,MAAM;EACjGa,IAAI,EAAEmB,IAAI,CAAC1B,IAAI,IAAI0B,IAAI,CAACzB,UAAU;EAClCY,WAAW,EAAEuB,cAAc,CAACV,IAAI,CAACW,GAAG,CAAC;EACrCC,UAAU,EAAEC,cAAc,CAACb,IAAI,CAACW,GAAG,CAAC;EACpC3C;CACD,CAAC;AAEF,MAAM6C,cAAc,GAAIF,GAAY,IAA4B;EAC9D,MAAMG,KAAK,GAAG,EAAE;EAChB,MAAM5C,MAAM,GAAGlC,UAAU,CAAC+E,OAAO,CAACJ,GAAG,EAAE;IACrCK,WAAW,EAAEF,KAAK;IAClBG,yBAAyB,EAAE;GAC5B,CAAC;EACF,IAAIC,MAAM,CAACC,IAAI,CAACL,KAAK,CAAC,CAACM,MAAM,KAAK,CAAC,EAAE,OAAOlD,MAAM;EAChDA,MAAc,CAAC4C,KAAK,GAAGA,KAAK;EAC9B,OAAO5C,MAAM;AACf,CAAC;AAED,MAAMwC,cAAc,GAAIC,GAAY,IAAY;EAC9C,MAAMU,WAAW,GAAGV,GAAG,CAACrC,IAAI,KAAK,gBAAgB,GAC/C;IACE,GAAGqC,GAAG,CAACW,EAAE,CAACD,WAAW;IACrB,GAAGV,GAAG,CAACU;GACR,GACDV,GAAG,CAACU,WAAW;EACjB,OAAOlF,GAAG,CAACoF,uBAAuB,IAAIF,WAAW,GAAGA,WAAW,CAAClF,GAAG,CAACoF,uBAAuB,CAAW,GAAG,EAAE;AAC7G,CAAC;AAED,MAAMjB,YAAY,GAChBzD,OAKC,IACC;EACF,MAAM2E,SAAS,GAAkB,EAAE;EACnC,MAAMC,SAAS,GAAG7F,KAAK,CAAC8F,MAAM,CAC5B7E,OAAO,CAAC4B,QAAQ,CAACE,KAAK,EACrBC,IAAI,IAA0B;IAC7B,IAAIA,IAAI,CAACN,IAAI,KAAK,UAAU,EAAE;MAC5BkD,SAAS,CAACnB,IAAI,CAACzB,IAAI,CAACC,IAAI,CAAC;MACzB,OAAO,IAAI;IACb;IACA,OAAO,KAAK;EACd,CAAC,CACF;EACD,IAAIjD,KAAK,CAAC+F,OAAO,CAACF,SAAS,CAAC,EAAE;IAC5B,OAAO3F,MAAM,CAAC8D,OAAO,CACnB,IAAIrD,YAAY,CAAC;MACfkC,QAAQ,EAAE5B,OAAO,CAAC4B,QAAQ;MAC1BkB,QAAQ,EAAEjD,aAAa;MACvBoD,OAAO,EAAEpD;KACV,CAAC,CACH;EACH;EACA,MAAMiD,QAAQ,GAAG,IAAIhD,GAAG,EAA2C;EACnE,MAAMmD,OAAO,GAAG,IAAInD,GAAG,EAAuB;EAC9C,OAAOb,MAAM,CAAC8F,mBAAmB,CAAC,WAAW,EAAEJ,SAAS,CAAC,CAACnE,IAAI,CAC5DvB,MAAM,CAAC+F,QAAQ,CAAC/F,MAAM,CAACgG,OAAO,CAC5BL,SAAS,EACR7C,IAAI,IAAI;IACP,MAAMoB,IAAI,GAAGjE,OAAO,CAACgG,SAAS,CAAClF,OAAO,CAACY,KAAK,CAACwC,OAAO,CAACxC,KAAK,EAAEmB,IAAI,CAACC,IAAI,CAAC;IACtE,MAAMmD,OAAO,GAAGjG,OAAO,CAACgG,SAAS,CAAClF,OAAO,CAACY,KAAK,CAACwE,QAAQ,EAAErD,IAAI,CAACC,IAAI,CAAC;IACpE,MAAMqD,YAAY,GAAGhG,MAAM,CAACkC,aAAa,CAAC4B,IAAW,CAAC;IACtD,MAAMmC,aAAa,GAAGjG,MAAM,CAACkG,MAAM,CAACpC,IAAI,CAACqC,OAAO,CAAC;IACjD,OAAOH,YAAY,CAACtD,IAAI,CAACW,MAAM,CAAC,CAAClC,IAAI,CACnCvB,MAAM,CAACwG,QAAQ,CAAE7C,KAAK,IACpB,IAAIpD,OAAO,CAAC;MACV4C,MAAM,EAAE,aAAa;MACrBC,MAAM,EAAErC,OAAO,CAACqC,MAAM;MACtBC,WAAW,EAAE,+BAA+BP,IAAI,CAACC,IAAI,cAAc;MACnEY;KACD,CAAC,CACH,EACD3D,MAAM,CAACwB,OAAO,CAAC0E,OAAO,CAAC,EACvBlG,MAAM,CAACyG,GAAG,CAAEC,KAAK,IAAI;MACnB,OAAOL,aAAa,CAACK,KAAK,CAAC,CAACnF,IAAI,CAC9BvB,MAAM,CAACwG,QAAQ,CAAE7C,KAAK,IACpB,IAAIpD,OAAO,CAAC;QACV4C,MAAM,EAAE,aAAa;QACrBC,MAAM,EAAErC,OAAO,CAACqC,MAAM;QACtBC,WAAW,EAAE,+BAA+BP,IAAI,CAACC,IAAI,UAAU;QAC/DY;OACD,CAAC,CACH,EACD3D,MAAM,CAACgB,GAAG,CAAE2F,YAAY,IAAI;QAC1B9C,QAAQ,CAAC+C,GAAG,CAAC9D,IAAI,CAACiB,EAAE,EAAE2C,KAAK,CAAC;QAC5B1C,OAAO,CAAC4C,GAAG,CAAC9D,IAAI,CAACiB,EAAE,EAAE4C,YAAY,CAAC;MACpC,CAAC,CAAC,CACH;IACH,CAAC,CAAC,CACH;EACH,CAAC,EACD;IAAEvC,WAAW,EAAErD,OAAO,CAACqD,WAAW;IAAEyC,OAAO,EAAE;EAAI,CAAE,CACpD,CAAC,EACF7G,MAAM,CAAC8G,EAAE,CAAC,IAAIrG,YAAY,CAAC;IAAEkC,QAAQ,EAAE5B,OAAO,CAAC4B,QAAQ;IAAEkB,QAAQ;IAAEG;EAAO,CAAE,CAAC,CAAC,CAC/E;AACH,CAAC","ignoreList":[]}
+{"version":3,"file":"Completions.js","names":["Chunk","Context","Effect","HashMap","JsonSchema","Option","Schema","AST","Stream","AiError","AiInput","WithResolved","Completions","Tag","constEmptyMap","Map","make","options","map","serviceOption","SystemInstruction","parentSystem","of","create","input","useSpan","captureStackTrace","span","pipe","flatMap","system","orElse","tools","required","stream","input_","makeSpanScoped","zip","unwrapScoped","structured","opts","decode","decodeUnknown","schema","toolId","toolCallId","_tag","identifier","attributes","convertTool","response","findFirst","parts","part","name","match","onNone","fail","module","method","description","onSome","toolCall","matchEffect","params","onFailure","cause","onSuccess","resolved","succeed","id","encoded","toolkit","concurrency","inputInput","toolArr","tool","push","resolveParts","toolkitStream","mapEffect","chunk","getDescription","ast","parameters","makeJsonSchema","$defs","fromAST","definitions","topLevelReferenceStrategy","Object","keys","length","annotations","to","DescriptionAnnotationId","toolNames","toolParts","filter","isEmpty","annotateCurrentSpan","zipRight","forEach","unsafeGet","handler","handlers","decodeParams","encodeSuccess","encode","success","mapError","tap","value","encodedValue","set","discard","as"],"sources":["../../src/Completions.ts"],"sourcesContent":[null],"mappings":"AAAA;;;AAGA,OAAO,KAAKA,KAAK,MAAM,cAAc;AACrC,OAAO,KAAKC,OAAO,MAAM,gBAAgB;AACzC,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,OAAO,KAAKC,OAAO,MAAM,gBAAgB;AACzC,OAAO,KAAKC,UAAU,MAAM,mBAAmB;AAC/C,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,OAAO,KAAKC,GAAG,MAAM,kBAAkB;AACvC,OAAO,KAAKC,MAAM,MAAM,eAAe;AAGvC,SAASC,OAAO,QAAQ,cAAc;AAEtC,OAAO,KAAKC,OAAO,MAAM,cAAc;AAEvC,SAASC,YAAY,QAAQ,iBAAiB;AAG9C;;;;AAIA,OAAM,MAAOC,WAAY,sBAAQX,OAAO,CAACY,GAAG,CAAC,wBAAwB,CAAC,EAGnE;AA0EH,MAAMC,aAAa,gBAAG,IAAIC,GAAG,EAAgB;AAkB7C;;;;AAIA,OAAO,MAAMC,IAAI,GAAIC,OAyBpB,IACCf,MAAM,CAACgB,GAAG,CAAChB,MAAM,CAACiB,aAAa,CAACT,OAAO,CAACU,iBAAiB,CAAC,EAAGC,YAAY,IAAI;EAC3E,OAAOT,WAAW,CAACU,EAAE,CAAC;IACpBC,MAAMA,CAACC,KAAK;MACV,OAAOtB,MAAM,CAACuB,OAAO,CACnB,oBAAoB,EACpB;QAAEC,iBAAiB,EAAE;MAAK,CAAE,EAC3BC,IAAI,IACHzB,MAAM,CAACiB,aAAa,CAACT,OAAO,CAACU,iBAAiB,CAAC,CAACQ,IAAI,CAClD1B,MAAM,CAAC2B,OAAO,CAAEC,MAAM,IACpBb,OAAO,CAACM,MAAM,CAAC;QACbC,KAAK,EAAEd,OAAO,CAACM,IAAI,CAACQ,KAAK,CAAiC;QAC1DM,MAAM,EAAEzB,MAAM,CAAC0B,MAAM,CAACD,MAAM,EAAE,MAAMT,YAAY,CAAC;QACjDW,KAAK,EAAE,EAAE;QACTC,QAAQ,EAAE,KAAK;QACfN;OACD,CAAC,CACH,CACF,CACJ;IACH,CAAC;IACDO,MAAMA,CAACC,MAAM;MACX,MAAMX,KAAK,GAAGd,OAAO,CAACM,IAAI,CAACmB,MAAM,CAAC;MAClC,OAAOjC,MAAM,CAACkC,cAAc,CAAC,oBAAoB,EAAE;QAAEV,iBAAiB,EAAE;MAAK,CAAE,CAAC,CAACE,IAAI,CACnF1B,MAAM,CAACmC,GAAG,CAACnC,MAAM,CAACiB,aAAa,CAACT,OAAO,CAACU,iBAAiB,CAAC,CAAC,EAC3DlB,MAAM,CAACgB,GAAG,CAAC,CAAC,CAACS,IAAI,EAAEG,MAAM,CAAC,KACxBb,OAAO,CAACiB,MAAM,CAAC;QACbV,KAAK,EAAEA,KAAqC;QAC5CM,MAAM,EAAEzB,MAAM,CAAC0B,MAAM,CAACD,MAAM,EAAE,MAAMT,YAAY,CAAC;QACjDW,KAAK,EAAE,EAAE;QACTC,QAAQ,EAAE,KAAK;QACfN;OACD,CAAC,CACH,EACDnB,MAAM,CAAC8B,YAAY,CACpB;IACH,CAAC;IACDC,UAAUA,CAACC,IAAI;MACb,MAAMhB,KAAK,GAAGd,OAAO,CAACM,IAAI,CAACwB,IAAI,CAAChB,KAAK,CAAC;MACtC,MAAMiB,MAAM,GAAGnC,MAAM,CAACoC,aAAa,CAACF,IAAI,CAACG,MAAM,CAAC;MAChD,MAAMC,MAAM,GAAG,YAAY,IAAIJ,IAAI,GAC/BA,IAAI,CAACK,UAAU,GACf,MAAM,IAAIL,IAAI,CAACG,MAAM,GACrBH,IAAI,CAACG,MAAM,CAACG,IAAI,GAChBN,IAAI,CAACG,MAAM,CAACI,UAAU;MAC1B,OAAO7C,MAAM,CAACuB,OAAO,CACnB,wBAAwB,EACxB;QAAEuB,UAAU,EAAE;UAAEJ;QAAM,CAAE;QAAElB,iBAAiB,EAAE;MAAK,CAAE,EACnDC,IAAI,IACHzB,MAAM,CAACiB,aAAa,CAACT,OAAO,CAACU,iBAAiB,CAAC,CAACQ,IAAI,CAClD1B,MAAM,CAAC2B,OAAO,CAAEC,MAAM,IACpBb,OAAO,CAACM,MAAM,CAAC;QACbC,KAAK,EAAEA,KAAqC;QAC5CM,MAAM,EAAEzB,MAAM,CAAC0B,MAAM,CAACD,MAAM,EAAE,MAAMT,YAAY,CAAC;QACjDW,KAAK,EAAE,CAACiB,WAAW,CAACL,MAAM,EAAEJ,IAAI,CAACG,MAAM,EAAE,IAAI,CAAC,CAAC;QAC/CV,QAAQ,EAAE,IAAI;QACdN;OACD,CAAC,CACH,EACDzB,MAAM,CAAC2B,OAAO,CAAEqB,QAAQ,IACtBlD,KAAK,CAACmD,SAAS,CACbD,QAAQ,CAACE,KAAK,EACbC,IAAI,IAA2BA,IAAI,CAACP,IAAI,KAAK,UAAU,IAAIO,IAAI,CAACC,IAAI,KAAKV,MAAM,CACjF,CAAChB,IAAI,CACJvB,MAAM,CAACkD,KAAK,CAAC;QACXC,MAAM,EAAEA,CAAA,KACNtD,MAAM,CAACuD,IAAI,CACT,IAAIhD,OAAO,CAAC;UACViD,MAAM,EAAE,aAAa;UACrBC,MAAM,EAAE,YAAY;UACpBC,WAAW,EAAE,cAAchB,MAAM;SAClC,CAAC,CACH;QACHiB,MAAM,EAAGC,QAAQ,IACf5D,MAAM,CAAC6D,WAAW,CAACtB,MAAM,CAACqB,QAAQ,CAACE,MAAM,CAAC,EAAE;UAC1CC,SAAS,EAAGC,KAAK,IACf,IAAIzD,OAAO,CAAC;YACViD,MAAM,EAAE,aAAa;YACrBC,MAAM,EAAE,YAAY;YACpBC,WAAW,EAAE,+BAA+BhB,MAAM,cAAc;YAChEsB;WACD,CAAC;UACJC,SAAS,EAAGC,QAAQ,IAClBlE,MAAM,CAACmE,OAAO,CACZ,IAAI1D,YAAY,CAAC;YACfuC,QAAQ;YACRkB,QAAQ,EAAE,IAAIrD,GAAG,CAAC,CAAC,CAAC+C,QAAQ,CAACQ,EAAE,EAAEF,QAAQ,CAAC,CAAC,CAAC;YAC5CG,OAAO,EAAE,IAAIxD,GAAG,CAAC,CAAC,CAAC+C,QAAQ,CAACQ,EAAE,EAAER,QAAQ,CAACE,MAAM,CAAC,CAAC;WAClD,CAAC;SAEP;OACJ,CAAC,CACH,CACF,CACF,CACJ;IACH,CAAC;IACDQ,OAAOA,CAAC;MAAEC,WAAW;MAAEjD,KAAK,EAAEkD,UAAU;MAAEzC,QAAQ,GAAG,KAAK;MAAED;IAAK,CAAE;MACjE,MAAMR,KAAK,GAAGd,OAAO,CAACM,IAAI,CAAC0D,UAAU,CAAC;MACtC,MAAMC,OAAO,GAKR,EAAE;MACP,KAAK,MAAM,GAAGC,IAAI,CAAC,IAAI5C,KAAK,CAACwC,OAAO,CAACxC,KAAK,EAAE;QAC1C2C,OAAO,CAACE,IAAI,CAAC5B,WAAW,CAAC2B,IAAI,CAAC9B,IAAI,EAAE8B,IAAW,CAAC,CAAC;MACnD;MACA,OAAO1E,MAAM,CAACuB,OAAO,CACnB,qBAAqB,EACrB;QAAEuB,UAAU,EAAE;UAAEyB,WAAW;UAAExC;QAAQ,CAAE;QAAEP,iBAAiB,EAAE;MAAK,CAAE,EAClEC,IAAI,IACHzB,MAAM,CAACiB,aAAa,CAACT,OAAO,CAACU,iBAAiB,CAAC,CAACQ,IAAI,CAClD1B,MAAM,CAAC2B,OAAO,CAAEC,MAAM,IACpBb,OAAO,CAACM,MAAM,CAAC;QACbC,KAAK,EAAEA,KAAqC;QAC5CM,MAAM,EAAEzB,MAAM,CAAC0B,MAAM,CAACD,MAAM,EAAE,MAAMT,YAAY,CAAC;QACjDW,KAAK,EAAE2C,OAAO;QACd1C,QAAQ,EAAEA,QAAe;QACzBN;OACD,CAAC,CACH,EACDzB,MAAM,CAAC2B,OAAO,CAAEqB,QAAQ,IAAK4B,YAAY,CAAC;QAAE5B,QAAQ;QAAElB,KAAK;QAAEyC,WAAW;QAAEd,MAAM,EAAE;MAAS,CAAE,CAAC,CAAC,CACzF,CACX;IACH,CAAC;IACDoB,aAAaA,CAAC;MAAEN,WAAW;MAAEjD,KAAK;MAAES,QAAQ,GAAG,KAAK;MAAED;IAAK,CAAE;MAC3D,MAAM2C,OAAO,GAKR,EAAE;MACP,KAAK,MAAM,GAAGC,IAAI,CAAC,IAAI5C,KAAK,CAACwC,OAAO,CAACxC,KAAK,EAAE;QAC1C2C,OAAO,CAACE,IAAI,CAAC5B,WAAW,CAAC2B,IAAI,CAAC9B,IAAI,EAAE8B,IAAW,CAAC,CAAC;MACnD;MACA,OAAO1E,MAAM,CAACkC,cAAc,CAAC,oBAAoB,EAAE;QACjDV,iBAAiB,EAAE,KAAK;QACxBsB,UAAU,EAAE;UAAEf,QAAQ;UAAEwC;QAAW;OACpC,CAAC,CAAC7C,IAAI,CACL1B,MAAM,CAACmC,GAAG,CAACnC,MAAM,CAACiB,aAAa,CAACT,OAAO,CAACU,iBAAiB,CAAC,CAAC,EAC3DlB,MAAM,CAACgB,GAAG,CAAC,CAAC,CAACS,IAAI,EAAEG,MAAM,CAAC,KACxBb,OAAO,CAACiB,MAAM,CAAC;QACbV,KAAK,EAAEd,OAAO,CAACM,IAAI,CAACQ,KAAK,CAAiC;QAC1DM,MAAM,EAAEzB,MAAM,CAAC0B,MAAM,CAACD,MAAM,EAAE,MAAMT,YAAY,CAAC;QACjDW,KAAK,EAAE2C,OAAO;QACd1C,QAAQ,EAAEA,QAAe;QACzBN;OACD,CAAC,CACH,EACDnB,MAAM,CAAC8B,YAAY,EACnB9B,MAAM,CAACwE,SAAS,CACbC,KAAK,IAAKH,YAAY,CAAC;QAAE5B,QAAQ,EAAE+B,KAAK;QAAEjD,KAAK;QAAEyC,WAAW;QAAEd,MAAM,EAAE;MAAe,CAAE,CAAC,EACzF;QAAEc,WAAW,EAAE;MAAW,CAAE,CAC7B,CACK;IACV;GACD,CAAC;AACJ,CAAC,CAAC;AAEJ,MAAMxB,WAAW,GAAGA,CAClBK,IAAY,EACZX,MAA8B,EAC9BJ,UAAU,GAAG,KAAK,MACd;EACJe,IAAI;EACJM,WAAW,EAAEsB,cAAc,CAACvC,MAAM,CAACwC,GAAG,CAAC;EACvCC,UAAU,EAAEC,cAAc,CAAC1C,MAAM,CAACwC,GAAG,CAAC;EACtC5C;CACD,CAAC;AAEF,MAAM8C,cAAc,GAAIF,GAAY,IAA4B;EAC9D,MAAMG,KAAK,GAAG,EAAE;EAChB,MAAM3C,MAAM,GAAGvC,UAAU,CAACmF,OAAO,CAACJ,GAAG,EAAE;IACrCK,WAAW,EAAEF,KAAK;IAClBG,yBAAyB,EAAE;GAC5B,CAAC;EACF,IAAIC,MAAM,CAACC,IAAI,CAACL,KAAK,CAAC,CAACM,MAAM,KAAK,CAAC,EAAE,OAAOjD,MAAM;EAChDA,MAAc,CAAC2C,KAAK,GAAGA,KAAK;EAC9B,OAAO3C,MAAM;AACf,CAAC;AAED,MAAMuC,cAAc,GAAIC,GAAY,IAAY;EAC9C,MAAMU,WAAW,GAAGV,GAAG,CAACrC,IAAI,KAAK,gBAAgB,GAC/C;IACE,GAAGqC,GAAG,CAACW,EAAE,CAACD,WAAW;IACrB,GAAGV,GAAG,CAACU;GACR,GACDV,GAAG,CAACU,WAAW;EACjB,OAAOtF,GAAG,CAACwF,uBAAuB,IAAIF,WAAW,GAAGA,WAAW,CAACtF,GAAG,CAACwF,uBAAuB,CAAW,GAAG,EAAE;AAC7G,CAAC;AAED,MAAMjB,YAAY,GAChB7D,OAKC,IACC;EACF,MAAM+E,SAAS,GAAkB,EAAE;EACnC,MAAMC,SAAS,GAAGjG,KAAK,CAACkG,MAAM,CAC5BjF,OAAO,CAACiC,QAAQ,CAACE,KAAK,EACrBC,IAAI,IAA0B;IAC7B,IAAIA,IAAI,CAACP,IAAI,KAAK,UAAU,EAAE;MAC5BkD,SAAS,CAACnB,IAAI,CAACxB,IAAI,CAACC,IAAI,CAAC;MACzB,OAAO,IAAI;IACb;IACA,OAAO,KAAK;EACd,CAAC,CACF;EACD,IAAItD,KAAK,CAACmG,OAAO,CAACF,SAAS,CAAC,EAAE;IAC5B,OAAO/F,MAAM,CAACmE,OAAO,CACnB,IAAI1D,YAAY,CAAC;MACfuC,QAAQ,EAAEjC,OAAO,CAACiC,QAAQ;MAC1BkB,QAAQ,EAAEtD,aAAa;MACvByD,OAAO,EAAEzD;KACV,CAAC,CACH;EACH;EACA,MAAMsD,QAAQ,GAAG,IAAIrD,GAAG,EAA2C;EACnE,MAAMwD,OAAO,GAAG,IAAIxD,GAAG,EAAuB;EAC9C,OAAOb,MAAM,CAACkG,mBAAmB,CAAC,WAAW,EAAEJ,SAAS,CAAC,CAACpE,IAAI,CAC5D1B,MAAM,CAACmG,QAAQ,CAACnG,MAAM,CAACoG,OAAO,CAC5BL,SAAS,EACR5C,IAAI,IAAI;IACP,MAAMuB,IAAI,GAAGzE,OAAO,CAACoG,SAAS,CAACtF,OAAO,CAACe,KAAK,CAACwC,OAAO,CAACxC,KAAK,EAAEqB,IAAI,CAACC,IAAI,CAAC;IACtE,MAAMkD,OAAO,GAAGrG,OAAO,CAACoG,SAAS,CAACtF,OAAO,CAACe,KAAK,CAACyE,QAAQ,EAAEpD,IAAI,CAACC,IAAI,CAAC;IACpE,MAAMoD,YAAY,GAAGpG,MAAM,CAACoC,aAAa,CAACkC,IAAW,CAAC;IACtD,MAAM+B,aAAa,GAAGrG,MAAM,CAACsG,MAAM,CAAChC,IAAI,CAACiC,OAAO,CAAC;IACjD,OAAOH,YAAY,CAACrD,IAAI,CAACW,MAAM,CAAC,CAACpC,IAAI,CACnC1B,MAAM,CAAC4G,QAAQ,CAAE5C,KAAK,IACpB,IAAIzD,OAAO,CAAC;MACViD,MAAM,EAAE,aAAa;MACrBC,MAAM,EAAE1C,OAAO,CAAC0C,MAAM;MACtBC,WAAW,EAAE,+BAA+BP,IAAI,CAACC,IAAI,cAAc;MACnEY;KACD,CAAC,CACH,EACDhE,MAAM,CAAC2B,OAAO,CAAC2E,OAAO,CAAC,EACvBtG,MAAM,CAAC6G,GAAG,CAAEC,KAAK,IAAI;MACnB,OAAOL,aAAa,CAACK,KAAK,CAAC,CAACpF,IAAI,CAC9B1B,MAAM,CAAC4G,QAAQ,CAAE5C,KAAK,IACpB,IAAIzD,OAAO,CAAC;QACViD,MAAM,EAAE,aAAa;QACrBC,MAAM,EAAE1C,OAAO,CAAC0C,MAAM;QACtBC,WAAW,EAAE,+BAA+BP,IAAI,CAACC,IAAI,UAAU;QAC/DY;OACD,CAAC,CACH,EACDhE,MAAM,CAACgB,GAAG,CAAE+F,YAAY,IAAI;QAC1B7C,QAAQ,CAAC8C,GAAG,CAAC7D,IAAI,CAACiB,EAAE,EAAE0C,KAAK,CAAC;QAC5BzC,OAAO,CAAC2C,GAAG,CAAC7D,IAAI,CAACiB,EAAE,EAAE2C,YAAY,CAAC;MACpC,CAAC,CAAC,CACH;IACH,CAAC,CAAC,CACH;EACH,CAAC,EACD;IAAExC,WAAW,EAAExD,OAAO,CAACwD,WAAW;IAAE0C,OAAO,EAAE;EAAI,CAAE,CACpD,CAAC,EACFjH,MAAM,CAACkH,EAAE,CAAC,IAAIzG,YAAY,CAAC;IAAEuC,QAAQ,EAAEjC,OAAO,CAACiC,QAAQ;IAAEkB,QAAQ;IAAEG;EAAO,CAAE,CAAC,CAAC,CAC/E;AACH,CAAC","ignoreList":[]}

package/dist/esm/Embeddings.js

@@ -0,0 +1,80 @@
+/**
+ * @since 1.0.0
+ */
+import { dataLoader } from "@effect/experimental/RequestResolver";
+import * as Context from "effect/Context";
+import * as Effect from "effect/Effect";
+import { identity } from "effect/Function";
+import * as Option from "effect/Option";
+import * as Request from "effect/Request";
+import * as RequestResolver from "effect/RequestResolver";
+import * as Schema from "effect/Schema";
+import { AiError } from "./AiError.js";
+/**
+ * @since 1.0.0
+ * @category tags
+ */
+export class Embeddings extends /*#__PURE__*/Context.Tag("@effect/ai/Embeddings")() {}
+class EmbeddingRequest extends /*#__PURE__*/Schema.TaggedRequest()("EmbeddingRequest", {
+  failure: AiError,
+  success: /*#__PURE__*/Schema.mutable( /*#__PURE__*/Schema.Array(Schema.Number)),
+  payload: {
+    input: Schema.String
+  }
+}) {}
+const makeBatchedResolver = embedMany => RequestResolver.makeBatched(requests => embedMany(requests.map(request => request.input)).pipe(Effect.flatMap(Effect.forEach(({
+  embeddings,
+  index
+}) => Request.succeed(requests[index], embeddings), {
+  discard: true
+})), Effect.catchAll(error => Effect.forEach(requests, request => Request.fail(request, error), {
+  discard: true
+}))));
+/**
+ * @since 1.0.0
+ * @category constructors
+ */
+export const make = options => Effect.gen(function* () {
+  const cache = yield* Option.fromNullable(options.cache).pipe(Effect.flatMap(config => Request.makeCache(config)), Effect.optionFromOptional);
+  const resolver = makeBatchedResolver(options.embedMany).pipe(options.maxBatchSize ? RequestResolver.batchN(options.maxBatchSize) : identity);
+  function embed(input) {
+    const request = Effect.request(new EmbeddingRequest({
+      input
+    }), resolver);
+    return Option.match(cache, {
+      onNone: () => request,
+      onSome: cache => request.pipe(Effect.withRequestCaching(true), Effect.withRequestCache(cache))
+    }).pipe(Effect.withSpan("Embeddings.embed", {
+      captureStackTrace: false
+    }));
+  }
+  return Embeddings.of({
+    embed
+  });
+});
+/**
+ * Creates an `Embeddings` service which will aggregate all `embed` requests
+ * received during the specified `window` (up to a maximum of `maxBatchSize`
+ * requests, if specified) and execute them as a single batch.
+ *
+ * @since 1.0.0
+ * @category constructors
+ */
+export const makeDataLoader = options => Effect.gen(function* () {
+  const resolver = makeBatchedResolver(options.embedMany);
+  const resolverDelayed = yield* dataLoader(resolver, {
+    window: options.window,
+    maxBatchSize: options.maxBatchSize
+  });
+  function embed(input) {
+    return Effect.request(new EmbeddingRequest({
+      input
+    }), resolverDelayed).pipe(Effect.withSpan("Embeddings.embed", {
+      captureStackTrace: false
+    }));
+  }
+  return Embeddings.of({
+    embed
+  });
+});
+//# sourceMappingURL=Embeddings.js.map

package/dist/esm/Embeddings.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Embeddings.js","names":["dataLoader","Context","Effect","identity","Option","Request","RequestResolver","Schema","AiError","Embeddings","Tag","EmbeddingRequest","TaggedRequest","failure","success","mutable","Array","Number","payload","input","String","makeBatchedResolver","embedMany","makeBatched","requests","map","request","pipe","flatMap","forEach","embeddings","index","succeed","discard","catchAll","error","fail","make","options","gen","cache","fromNullable","config","makeCache","optionFromOptional","resolver","maxBatchSize","batchN","embed","match","onNone","onSome","withRequestCaching","withRequestCache","withSpan","captureStackTrace","of","makeDataLoader","resolverDelayed","window"],"sources":["../../src/Embeddings.ts"],"sourcesContent":[null],"mappings":"AAAA;;;AAGA,SAASA,UAAU,QAAQ,sCAAsC;AACjE,OAAO,KAAKC,OAAO,MAAM,gBAAgB;AAEzC,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,SAASC,QAAQ,QAAQ,iBAAiB;AAC1C,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,OAAO,KAAKC,OAAO,MAAM,gBAAgB;AACzC,OAAO,KAAKC,eAAe,MAAM,wBAAwB;AACzD,OAAO,KAAKC,MAAM,MAAM,eAAe;AACvC,SAASC,OAAO,QAAQ,cAAc;AAEtC;;;;AAIA,OAAM,MAAOC,UAAW,sBAAQR,OAAO,CAACS,GAAG,CAAC,uBAAuB,CAAC,EAGjE;AAyBH,MAAMC,gBAAiB,sBAAQJ,MAAM,CAACK,aAAa,EAAoB,CAAC,kBAAkB,EAAE;EAC1FC,OAAO,EAAEL,OAAO;EAChBM,OAAO,eAAEP,MAAM,CAACQ,OAAO,eAACR,MAAM,CAACS,KAAK,CAACT,MAAM,CAACU,MAAM,CAAC,CAAC;EACpDC,OAAO,EAAE;IAAEC,KAAK,EAAEZ,MAAM,CAACa;EAAM;CAChC,CAAC;AAEF,MAAMC,mBAAmB,GACvBC,SAA6F,IAE7FhB,eAAe,CAACiB,WAAW,CACxBC,QAAyC,IACxCF,SAAS,CAACE,QAAQ,CAACC,GAAG,CAAEC,OAAO,IAAKA,OAAO,CAACP,KAAK,CAAC,CAAC,CAACQ,IAAI,CACtDzB,MAAM,CAAC0B,OAAO,CACZ1B,MAAM,CAAC2B,OAAO,CACZ,CAAC;EAAEC,UAAU;EAAEC;AAAK,CAAE,KAAK1B,OAAO,CAAC2B,OAAO,CAACR,QAAQ,CAACO,KAAK,CAAC,EAAED,UAAU,CAAC,EACvE;EAAEG,OAAO,EAAE;AAAI,CAAE,CAClB,CACF,EACD/B,MAAM,CAACgC,QAAQ,CAAEC,KAAK,IACpBjC,MAAM,CAAC2B,OAAO,CACZL,QAAQ,EACPE,OAAO,IAAKrB,OAAO,CAAC+B,IAAI,CAACV,OAAO,EAAES,KAAK,CAAC,EACzC;EAAEF,OAAO,EAAE;AAAI,CAAE,CAClB,CACF,CACF,CACJ;AAEH;;;;AAIA,OAAO,MAAMI,IAAI,GAAIC,OAOpB,IACCpC,MAAM,CAACqC,GAAG,CAAC,aAAS;EAClB,MAAMC,KAAK,GAAG,OAAOpC,MAAM,CAACqC,YAAY,CAACH,OAAO,CAACE,KAAK,CAAC,CAACb,IAAI,CAC1DzB,MAAM,CAAC0B,OAAO,CAAEc,MAAM,IAAKrC,OAAO,CAACsC,SAAS,CAACD,MAAM,CAAC,CAAC,EACrDxC,MAAM,CAAC0C,kBAAkB,CAC1B;EAED,MAAMC,QAAQ,GAAGxB,mBAAmB,CAACiB,OAAO,CAAChB,SAAS,CAAC,CAACK,IAAI,CAC1DW,OAAO,CAACQ,YAAY,GAAGxC,eAAe,CAACyC,MAAM,CAACT,OAAO,CAACQ,YAAY,CAAC,GAAG3C,QAAQ,CAC/E;EAED,SAAS6C,KAAKA,CAAC7B,KAAa;IAC1B,MAAMO,OAAO,GAAGxB,MAAM,CAACwB,OAAO,CAAC,IAAIf,gBAAgB,CAAC;MAAEQ;IAAK,CAAE,CAAC,EAAE0B,QAAQ,CAAC;IACzE,OAAOzC,MAAM,CAAC6C,KAAK,CAACT,KAAK,EAAE;MACzBU,MAAM,EAAEA,CAAA,KAAMxB,OAAO;MACrByB,MAAM,EAAGX,KAAK,IACZd,OAAO,CAACC,IAAI,CACVzB,MAAM,CAACkD,kBAAkB,CAAC,IAAI,CAAC,EAC/BlD,MAAM,CAACmD,gBAAgB,CAACb,KAAK,CAAC;KAEnC,CAAC,CAACb,IAAI,CAACzB,MAAM,CAACoD,QAAQ,CAAC,kBAAkB,EAAE;MAAEC,iBAAiB,EAAE;IAAK,CAAE,CAAC,CAAC;EAC5E;EAEA,OAAO9C,UAAU,CAAC+C,EAAE,CAAC;IACnBR;GACD,CAAC;AACJ,CAAC,CAAC;AAEJ;;;;;;;;AAQA,OAAO,MAAMS,cAAc,GAAInB,OAI9B,IACCpC,MAAM,CAACqC,GAAG,CAAC,aAAS;EAClB,MAAMM,QAAQ,GAAGxB,mBAAmB,CAACiB,OAAO,CAAChB,SAAS,CAAC;EACvD,MAAMoC,eAAe,GAAG,OAAO1D,UAAU,CAAC6C,QAAQ,EAAE;IAClDc,MAAM,EAAErB,OAAO,CAACqB,MAAM;IACtBb,YAAY,EAAER,OAAO,CAACQ;GACvB,CAAC;EAEF,SAASE,KAAKA,CAAC7B,KAAa;IAC1B,OAAOjB,MAAM,CAACwB,OAAO,CAAC,IAAIf,gBAAgB,CAAC;MAAEQ;IAAK,CAAE,CAAC,EAAEuC,eAAe,CAAC,CAAC/B,IAAI,CAC1EzB,MAAM,CAACoD,QAAQ,CAAC,kBAAkB,EAAE;MAAEC,iBAAiB,EAAE;IAAK,CAAE,CAAC,CAClE;EACH;EAEA,OAAO9C,UAAU,CAAC+C,EAAE,CAAC;IACnBR;GACD,CAAC;AACJ,CAAC,CAAC","ignoreList":[]}