@effect/ai-openai 0.11.0 → 0.11.2

package/dist/cjs/OpenAiClient.js

@@ -42,59 +42,80 @@ const make = options => Effect.gen(function* () {
   const httpClientOk = HttpClient.filterStatusOk(httpClient);
   const client = Generated.make(httpClient);
   const streamRequest = request => httpClientOk.execute(request).pipe(Effect.map(r => r.stream), Stream.unwrapScoped, Stream.decodeText(), Stream.pipeThroughChannel(Sse.makeChannel()), Stream.takeWhile(event => event.data !== "[DONE]"), Stream.map(event => JSON.parse(event.data)));
-  const stream = request => streamRequest(HttpClientRequest.post("/chat/completions", {
-    body: HttpBody.unsafeJson({
-      ...request,
-      stream: true
-    })
-  })).pipe(Stream.mapAccum(new Map(), (acc, chunk) => {
-    const parts = [];
-    for (let i = 0; i < chunk.choices.length; i++) {
-      const choice = chunk.choices[i];
-      if ("content" in choice.delta && typeof choice.delta.content === "string") {
-        let part = acc.get(choice.index);
-        part = {
-          _tag: "Content",
-          content: choice.delta.content
-        };
-        acc.set(choice.index, part);
-        parts.push(part);
-      } else if ("tool_calls" in choice.delta && Array.isArray(choice.delta.tool_calls)) {
-        const parts = acc.get(choice.index) ?? [];
-        for (const toolCall of choice.delta.tool_calls) {
-          const part = parts[toolCall.index];
-          const toolPart = part?._tag === "ToolCall" ? {
-            ...part,
-            arguments: part.arguments + toolCall.function.arguments
-          } : {
-            _tag: "ToolCall",
-            ...toolCall,
-            ...toolCall.function,
-            role: choice.delta.role
-          };
-          parts[toolCall.index] = toolPart;
+  const stream = request => Stream.suspend(() => {
+    const finishReasons = [];
+    return streamRequest(HttpClientRequest.post("/chat/completions", {
+      body: HttpBody.unsafeJson({
+        ...request,
+        stream: true,
+        stream_options: {
+          include_usage: true
+        }
+      })
+    })).pipe(Stream.mapAccum(new Map(), (acc, chunk) => {
+      const parts = [];
+      if (chunk.usage !== null) {
+        parts.push({
+          _tag: "Usage",
+          id: chunk.id,
+          model: chunk.model,
+          inputTokens: chunk.usage.prompt_tokens,
+          outputTokens: chunk.usage.completion_tokens,
+          finishReasons,
+          systemFingerprint: chunk.system_fingerprint,
+          serviceTier: chunk.service_tier
+        });
+      }
+      for (let i = 0; i < chunk.choices.length; i++) {
+        const choice = chunk.choices[i];
+        if (choice.finish_reason !== null) {
+          finishReasons.push(choice.finish_reason);
         }
-        acc.set(choice.index, parts);
-      } else if (choice.finish_reason === "tool_calls") {
-        const toolParts = acc.get(choice.index);
-        for (const part of toolParts) {
-          try {
-            const args = JSON.parse(part.arguments);
-            parts.push({
+        if ("content" in choice.delta && typeof choice.delta.content === "string") {
+          let part = acc.get(choice.index);
+          part = {
+            _tag: "Content",
+            content: choice.delta.content
+          };
+          acc.set(choice.index, part);
+          parts.push(part);
+        } else if ("tool_calls" in choice.delta && Array.isArray(choice.delta.tool_calls)) {
+          const parts = acc.get(choice.index) ?? [];
+          for (const toolCall of choice.delta.tool_calls) {
+            const part = parts[toolCall.index];
+            const toolPart = part?._tag === "ToolCall" ? {
+              ...part,
+              arguments: part.arguments + toolCall.function.arguments
+            } : {
               _tag: "ToolCall",
-              id: part.id,
-              name: part.name,
-              arguments: args
-            });
-            // eslint-disable-next-line no-empty
-          } catch {}
+              ...toolCall,
+              ...toolCall.function,
+              role: choice.delta.role
+            };
+            parts[toolCall.index] = toolPart;
+          }
+          acc.set(choice.index, parts);
+        } else if (choice.finish_reason === "tool_calls") {
+          const toolParts = acc.get(choice.index);
+          for (const part of toolParts) {
+            try {
+              const args = JSON.parse(part.arguments);
+              parts.push({
+                _tag: "ToolCall",
+                id: part.id,
+                name: part.name,
+                arguments: args
+              });
+              // eslint-disable-next-line no-empty
+            } catch {}
+          }
         }
       }
-    }
-    return [acc, parts.length === 0 ? Option.none() : Option.some(new StreamChunk({
-      parts
-    }))];
-  }), Stream.filterMap(_Function.identity));
+      return [acc, parts.length === 0 ? Option.none() : Option.some(new StreamChunk({
+        parts
+      }))];
+    }), Stream.filterMap(_Function.identity));
+  });
   return OpenAiClient.of({
     client,
     streamRequest,
@@ -136,17 +157,24 @@ class StreamChunk extends Data.Class {
       });
     }
     const part = this.parts[0];
-    return part._tag === "Content" ? AiResponse.AiResponse.fromText({
-      role: AiRole.model,
-      content: part.content
-    }) : new AiResponse.AiResponse({
-      role: AiRole.model,
-      parts: Chunk.of(AiResponse.ToolCallPart.fromUnknown({
-        id: part.id,
-        name: part.name,
-        params: part.arguments
-      }))
-    });
+    switch (part._tag) {
+      case "Content":
+        return AiResponse.AiResponse.fromText({
+          role: AiRole.model,
+          content: part.content
+        });
+      case "ToolCall":
+        return new AiResponse.AiResponse({
+          role: AiRole.model,
+          parts: Chunk.of(AiResponse.ToolCallPart.fromUnknown({
+            id: part.id,
+            name: part.name,
+            params: part.arguments
+          }))
+        });
+      case "Usage":
+        return AiResponse.AiResponse.empty;
+    }
   }
 }
 exports.StreamChunk = StreamChunk;

package/dist/cjs/OpenAiClient.js.map

@@ -1 +1 @@
-{"version":3,"file":"OpenAiClient.js","names":["AiResponse","_interopRequireWildcard","require","AiRole","Sse","HttpBody","HttpClient","HttpClientRequest","Chunk","Config","Context","Data","Effect","_Function","Layer","Option","Redacted","Stream","Generated","_getRequireWildcardCache","e","WeakMap","r","t","__esModule","default","has","get","n","__proto__","a","Object","defineProperty","getOwnPropertyDescriptor","u","hasOwnProperty","call","i","set","OpenAiClient","Tag","exports","make","options","gen","httpClient","pipe","mapRequest","request","prependUrl","apiUrl","apiKey","bearerToken","identity","organizationId","undefined","setHeader","value","projectId","acceptJson","transformClient","httpClientOk","filterStatusOk","client","streamRequest","execute","map","stream","unwrapScoped","decodeText","pipeThroughChannel","makeChannel","takeWhile","event","data","JSON","parse","post","body","unsafeJson","mapAccum","Map","acc","chunk","parts","choices","length","choice","delta","content","part","index","_tag","push","Array","isArray","tool_calls","toolCall","toolPart","arguments","function","role","finish_reason","toolParts","args","id","name","none","some","StreamChunk","filterMap","of","layer","effect","layerConfig","unwrap","flatMap","Class","text","asAiResponse","fromText","model","ToolCallPart","fromUnknown","params"],"sources":["../../src/OpenAiClient.ts"],"sourcesContent":[null],"mappings":";;;;;;AAGA,IAAAA,UAAA,GAAAC,uBAAA,CAAAC,OAAA;AACA,IAAAC,MAAA,GAAAF,uBAAA,CAAAC,OAAA;AACA,IAAAE,GAAA,GAAAH,uBAAA,CAAAC,OAAA;AACA,IAAAG,QAAA,GAAAJ,uBAAA,CAAAC,OAAA;AACA,IAAAI,UAAA,GAAAL,uBAAA,CAAAC,OAAA;AAEA,IAAAK,iBAAA,GAAAN,uBAAA,CAAAC,OAAA;AACA,IAAAM,KAAA,GAAAP,uBAAA,CAAAC,OAAA;AACA,IAAAO,MAAA,GAAAR,uBAAA,CAAAC,OAAA;AAEA,IAAAQ,OAAA,GAAAT,uBAAA,CAAAC,OAAA;AACA,IAAAS,IAAA,GAAAV,uBAAA,CAAAC,OAAA;AACA,IAAAU,MAAA,GAAAX,uBAAA,CAAAC,OAAA;AACA,IAAAW,SAAA,GAAAX,OAAA;AACA,IAAAY,KAAA,GAAAb,uBAAA,CAAAC,OAAA;AACA,IAAAa,MAAA,GAAAd,uBAAA,CAAAC,OAAA;AACA,IAAAc,QAAA,GAAAf,uBAAA,CAAAC,OAAA;AACA,IAAAe,MAAA,GAAAhB,uBAAA,CAAAC,OAAA;AACA,IAAAgB,SAAA,GAAAjB,uBAAA,CAAAC,OAAA;AAA2C,SAAAiB,yBAAAC,CAAA,6BAAAC,OAAA,mBAAAC,CAAA,OAAAD,OAAA,IAAAE,CAAA,OAAAF,OAAA,YAAAF,wBAAA,YAAAA,CAAAC,CAAA,WAAAA,CAAA,GAAAG,CAAA,GAAAD,CAAA,KAAAF,CAAA;AAAA,SAAAnB,wBAAAmB,CAAA,EAAAE,CAAA,SAAAA,CAAA,IAAAF,CAAA,IAAAA,CAAA,CAAAI,UAAA,SAAAJ,CAAA,eAAAA,CAAA,uBAAAA,CAAA,yBAAAA,CAAA,WAAAK,OAAA,EAAAL,CAAA,QAAAG,CAAA,GAAAJ,wBAAA,CAAAG,CAAA,OAAAC,CAAA,IAAAA,CAAA,CAAAG,GAAA,CAAAN,CAAA,UAAAG,CAAA,CAAAI,GAAA,CAAAP,CAAA,OAAAQ,CAAA,KAAAC,SAAA,UAAAC,CAAA,GAAAC,MAAA,CAAAC,cAAA,IAAAD,MAAA,CAAAE,wBAAA,WAAAC,CAAA,IAAAd,CAAA,oBAAAc,CAAA,OAAAC,cAAA,CAAAC,IAAA,CAAAhB,CAAA,EAAAc,CAAA,SAAAG,CAAA,GAAAP,CAAA,GAAAC,MAAA,CAAAE,wBAAA,CAAAb,CAAA,EAAAc,CAAA,UAAAG,CAAA,KAAAA,CAAA,CAAAV,GAAA,IAAAU,CAAA,CAAAC,GAAA,IAAAP,MAAA,CAAAC,cAAA,CAAAJ,CAAA,EAAAM,CAAA,EAAAG,CAAA,IAAAT,CAAA,CAAAM,CAAA,IAAAd,CAAA,CAAAc,CAAA,YAAAN,CAAA,CAAAH,OAAA,GAAAL,CAAA,EAAAG,CAAA,IAAAA,CAAA,CAAAe,GAAA,CAAAlB,CAAA,EAAAQ,CAAA,GAAAA,CAAA;AArB3C;;;;AAuBA;;;;AAIM,MAAOW,YAAa,sBAAQ7B,OAAO,CAAC8B,GAAG,CAAC,gCAAgC,CAAC,EAG5E;AAsBH;;;;AAAAC,OAAA,CAAAF,YAAA,GAAAA,YAAA;AAIO,MAAMG,IAAI,GAAIC,OAMpB,IACC/B,MAAM,CAACgC,GAAG,CAAC,aAAS;EAClB,MAAMC,UAAU,GAAG,CAAC,OAAOvC,UAAU,CAACA,UAAU,EAAEwC,IAAI,CACpDxC,UAAU,CAACyC,UAAU,CAAEC,OAAO,IAC5BA,OAAO,CAACF,IAAI,CACVvC,iBAAiB,CAAC0C,UAAU,CAACN,OAAO,CAACO,MAAM,IAAI,2BAA2B,CAAC,EAC3EP,OAAO,CAACQ,MAAM,GAAG5C,iBAAiB,CAAC6C,WAAW,CAACT,OAAO,CAACQ,MAAM,CAAC,GAAGE,kBAAQ,EACzEV,OAAO,CAACW,cAAc,KAAKC,SAAS,GAChChD,iBAAiB,CAACiD,SAAS,CAAC,qBAAqB,EAAExC,QAAQ,CAACyC,KAAK,CAACd,OAAO,CAACW,cAAc,CAAC,CAAC,GAC1FD,kBAAQ,EACZV,OAAO,CAACe,SAAS,KAAKH,SAAS,GAC3BhD,iBAAiB,CAACiD,SAAS,CAAC,gBAAgB,EAAExC,QAAQ,CAACyC,KAAK,CAACd,OAAO,CAACe,SAAS,CAAC,CAAC,GAChFL,kBAAQ,EACZ9C,iBAAiB,CAACoD,UAAU,CAC7B,CACF,EACDhB,OAAO,CAACiB,eAAe,GAAGjB,OAAO,CAACiB,eAAe,GAAGP,kBAAQ,CAC7D;EACD,MAAMQ,YAAY,GAAGvD,UAAU,CAACwD,cAAc,CAACjB,UAAU,CAAC;EAC1D,MAAMkB,MAAM,GAAG7C,SAAS,CAACwB,IAAI,CAACG,UAAU,CAAC;EACzC,MAAMmB,aAAa,GAAiBhB,OAA4C,IAC9Ea,YAAY,CAACI,OAAO,CAACjB,OAAO,CAAC,CAACF,IAAI,CAChClC,MAAM,CAACsD,GAAG,CAAE5C,CAAC,IAAKA,CAAC,CAAC6C,MAAM,CAAC,EAC3BlD,MAAM,CAACmD,YAAY,EACnBnD,MAAM,CAACoD,UAAU,EAAE,EACnBpD,MAAM,CAACqD,kBAAkB,CAAClE,GAAG,CAACmE,WAAW,EAAE,CAAC,EAC5CtD,MAAM,CAACuD,SAAS,CAAEC,KAAK,IAAKA,KAAK,CAACC,IAAI,KAAK,QAAQ,CAAC,EACpDzD,MAAM,CAACiD,GAAG,CAAEO,KAAK,IAAKE,IAAI,CAACC,KAAK,CAACH,KAAK,CAACC,IAAI,CAAM,CAAC,CACnD;EACH,MAAMP,MAAM,GAAInB,OAAgC,IAC9CgB,aAAa,CAAqBzD,iBAAiB,CAACsE,IAAI,CAAC,mBAAmB,EAAE;IAC5EC,IAAI,EAAEzE,QAAQ,CAAC0E,UAAU,CAAC;MACxB,GAAG/B,OAAO;MACVmB,MAAM,EAAE;KACT;GACF,CAAC,CAAC,CAACrB,IAAI,CACN7B,MAAM,CAAC+D,QAAQ,CAAC,IAAIC,GAAG,EAA6C,EAAE,CAACC,GAAG,EAAEC,KAAK,KAAI;IACnF,MAAMC,KAAK,GAA2B,EAAE;IACxC,KAAK,IAAI/C,CAAC,GAAG,CAAC,EAAEA,CAAC,GAAG8C,KAAK,CAACE,OAAO,CAACC,MAAM,EAAEjD,CAAC,EAAE,EAAE;MAC7C,MAAMkD,MAAM,GAAGJ,KAAK,CAACE,OAAO,CAAChD,CAAC,CAAC;MAC/B,IAAI,SAAS,IAAIkD,MAAM,CAACC,KAAK,IAAI,OAAOD,MAAM,CAACC,KAAK,CAACC,OAAO,KAAK,QAAQ,EAAE;QACzE,IAAIC,IAAI,GAAGR,GAAG,CAACvD,GAAG,CAAC4D,MAAM,CAACI,KAAK,CAA4B;QAC3DD,IAAI,GAAG;UACLE,IAAI,EAAE,SAAS;UACfH,OAAO,EAAEF,MAAM,CAACC,KAAK,CAACC;SACvB;QACDP,GAAG,CAAC5C,GAAG,CAACiD,MAAM,CAACI,KAAK,EAAED,IAAI,CAAC;QAC3BN,KAAK,CAACS,IAAI,CAACH,IAAI,CAAC;MAClB,CAAC,MAAM,IAAI,YAAY,IAAIH,MAAM,CAACC,KAAK,IAAIM,KAAK,CAACC,OAAO,CAACR,MAAM,CAACC,KAAK,CAACQ,UAAU,CAAC,EAAE;QACjF,MAAMZ,KAAK,GAAIF,GAAG,CAACvD,GAAG,CAAC4D,MAAM,CAACI,KAAK,CAAC,IAAI,EAA0B;QAClE,KAAK,MAAMM,QAAQ,IAAIV,MAAM,CAACC,KAAK,CAACQ,UAAU,EAAE;UAC9C,MAAMN,IAAI,GAAGN,KAAK,CAACa,QAAQ,CAACN,KAAK,CAAC;UAClC,MAAMO,QAAQ,GAAGR,IAAI,EAAEE,IAAI,KAAK,UAAU,GACxC;YACE,GAAGF,IAAI;YACPS,SAAS,EAAET,IAAI,CAACS,SAAS,GAAGF,QAAQ,CAACG,QAAQ,CAACD;WAC/C,GACD;YACEP,IAAI,EAAE,UAAU;YAChB,GAAGK,QAAQ;YACX,GAAGA,QAAQ,CAACG,QAAQ;YACpBC,IAAI,EAAEd,MAAM,CAACC,KAAK,CAACa;WACb;UACVjB,KAAK,CAACa,QAAQ,CAACN,KAAK,CAAC,GAAGO,QAAQ;QAClC;QACAhB,GAAG,CAAC5C,GAAG,CAACiD,MAAM,CAACI,KAAK,EAAEP,KAAK,CAAC;MAC9B,CAAC,MAAM,IAAIG,MAAM,CAACe,aAAa,KAAK,YAAY,EAAE;QAChD,MAAMC,SAAS,GAAGrB,GAAG,CAACvD,GAAG,CAAC4D,MAAM,CAACI,KAAK,CAAwB;QAC9D,KAAK,MAAMD,IAAI,IAAIa,SAAS,EAAE;UAC5B,IAAI;YACF,MAAMC,IAAI,GAAG7B,IAAI,CAACC,KAAK,CAACc,IAAI,CAACS,SAAmB,CAAC;YACjDf,KAAK,CAACS,IAAI,CAAC;cACTD,IAAI,EAAE,UAAU;cAChBa,EAAE,EAAEf,IAAI,CAACe,EAAE;cACXC,IAAI,EAAEhB,IAAI,CAACgB,IAAI;cACfP,SAAS,EAAEK;aACZ,CAAC;YACF;UACF,CAAC,CAAC,MAAM,CAAC;QACX;MACF;IACF;IACA,OAAO,CAACtB,GAAG,EAAEE,KAAK,CAACE,MAAM,KAAK,CAAC,GAAGvE,MAAM,CAAC4F,IAAI,EAAE,GAAG5F,MAAM,CAAC6F,IAAI,CAAC,IAAIC,WAAW,CAAC;MAAEzB;IAAK,CAAE,CAAC,CAAC,CAAC;EAC5F,CAAC,CAAC,EACFnE,MAAM,CAAC6F,SAAS,CAACzD,kBAAQ,CAAC,CAC3B;EACH,OAAOd,YAAY,CAACwE,EAAE,CAAC;IAAEhD,MAAM;IAAEC,aAAa;IAAEG;EAAM,CAAE,CAAC;AAC3D,CAAC,CAAC;AAEJ;;;;AAAA1B,OAAA,CAAAC,IAAA,GAAAA,IAAA;AAIO,MAAMsE,KAAK,GAAIrE,OAMrB,IAA8D7B,KAAK,CAACmG,MAAM,CAAC1E,YAAY,EAAEG,IAAI,CAACC,OAAO,CAAC,CAAC;AAExG;;;;AAAAF,OAAA,CAAAuE,KAAA,GAAAA,KAAA;AAIO,MAAME,WAAW,GACtBvE,OAME,IAEFlC,MAAM,CAAC0G,MAAM,CAACxE,OAAO,CAAC,CAACG,IAAI,CACzBlC,MAAM,CAACwG,OAAO,CAAC1E,IAAI,CAAC,EACpB5B,KAAK,CAACmG,MAAM,CAAC1E,YAAY,CAAC,CAC3B;AAqDH;;;;AAAAE,OAAA,CAAAyE,WAAA,GAAAA,WAAA;AAIM,MAAOL,WAAY,SAAQlG,IAAI,CAAC0G,KAEpC;EACA;;;EAGA,IAAIC,IAAIA,CAAA;IACN,OAAO,IAAI,CAAClC,KAAK,CAAC,CAAC,CAAC,EAAEQ,IAAI,KAAK,SAAS,GAAG7E,MAAM,CAAC6F,IAAI,CAAC,IAAI,CAACxB,KAAK,CAAC,CAAC,CAAC,CAACK,OAAO,CAAC,GAAG1E,MAAM,CAAC4F,IAAI,EAAE;EAC/F;EACA;;;EAGA,IAAIY,YAAYA,CAAA;IACd,IAAI,IAAI,CAACnC,KAAK,CAACE,MAAM,KAAK,CAAC,EAAE;MAC3B,OAAOtF,UAAU,CAACA,UAAU,CAACwH,QAAQ,CAAC;QACpCnB,IAAI,EAAElG,MAAM,CAACsH,KAAK;QAClBhC,OAAO,EAAE;OACV,CAAC;IACJ;IACA,MAAMC,IAAI,GAAG,IAAI,CAACN,KAAK,CAAC,CAAC,CAAC;IAC1B,OAAOM,IAAI,CAACE,IAAI,KAAK,SAAS,GAC5B5F,UAAU,CAACA,UAAU,CAACwH,QAAQ,CAAC;MAC7BnB,IAAI,EAAElG,MAAM,CAACsH,KAAK;MAClBhC,OAAO,EAAEC,IAAI,CAACD;KACf,CAAC,GACF,IAAIzF,UAAU,CAACA,UAAU,CAAC;MACxBqG,IAAI,EAAElG,MAAM,CAACsH,KAAK;MAClBrC,KAAK,EAAE5E,KAAK,CAACuG,EAAE,CAAC/G,UAAU,CAAC0H,YAAY,CAACC,WAAW,CAAC;QAClDlB,EAAE,EAAEf,IAAI,CAACe,EAAE;QACXC,IAAI,EAAEhB,IAAI,CAACgB,IAAI;QACfkB,MAAM,EAAElC,IAAI,CAACS;OACd,CAAC;KACH,CAAC;EACN","ignoreList":[]}
+{"version":3,"file":"OpenAiClient.js","names":["AiResponse","_interopRequireWildcard","require","AiRole","Sse","HttpBody","HttpClient","HttpClientRequest","Chunk","Config","Context","Data","Effect","_Function","Layer","Option","Redacted","Stream","Generated","_getRequireWildcardCache","e","WeakMap","r","t","__esModule","default","has","get","n","__proto__","a","Object","defineProperty","getOwnPropertyDescriptor","u","hasOwnProperty","call","i","set","OpenAiClient","Tag","exports","make","options","gen","httpClient","pipe","mapRequest","request","prependUrl","apiUrl","apiKey","bearerToken","identity","organizationId","undefined","setHeader","value","projectId","acceptJson","transformClient","httpClientOk","filterStatusOk","client","streamRequest","execute","map","stream","unwrapScoped","decodeText","pipeThroughChannel","makeChannel","takeWhile","event","data","JSON","parse","suspend","finishReasons","post","body","unsafeJson","stream_options","include_usage","mapAccum","Map","acc","chunk","parts","usage","push","_tag","id","model","inputTokens","prompt_tokens","outputTokens","completion_tokens","systemFingerprint","system_fingerprint","serviceTier","service_tier","choices","length","choice","finish_reason","delta","content","part","index","Array","isArray","tool_calls","toolCall","toolPart","arguments","function","role","toolParts","args","name","none","some","StreamChunk","filterMap","of","layer","effect","layerConfig","unwrap","flatMap","Class","text","asAiResponse","fromText","ToolCallPart","fromUnknown","params","empty"],"sources":["../../src/OpenAiClient.ts"],"sourcesContent":[null],"mappings":";;;;;;AAGA,IAAAA,UAAA,GAAAC,uBAAA,CAAAC,OAAA;AACA,IAAAC,MAAA,GAAAF,uBAAA,CAAAC,OAAA;AACA,IAAAE,GAAA,GAAAH,uBAAA,CAAAC,OAAA;AACA,IAAAG,QAAA,GAAAJ,uBAAA,CAAAC,OAAA;AACA,IAAAI,UAAA,GAAAL,uBAAA,CAAAC,OAAA;AAEA,IAAAK,iBAAA,GAAAN,uBAAA,CAAAC,OAAA;AACA,IAAAM,KAAA,GAAAP,uBAAA,CAAAC,OAAA;AACA,IAAAO,MAAA,GAAAR,uBAAA,CAAAC,OAAA;AAEA,IAAAQ,OAAA,GAAAT,uBAAA,CAAAC,OAAA;AACA,IAAAS,IAAA,GAAAV,uBAAA,CAAAC,OAAA;AACA,IAAAU,MAAA,GAAAX,uBAAA,CAAAC,OAAA;AACA,IAAAW,SAAA,GAAAX,OAAA;AACA,IAAAY,KAAA,GAAAb,uBAAA,CAAAC,OAAA;AACA,IAAAa,MAAA,GAAAd,uBAAA,CAAAC,OAAA;AACA,IAAAc,QAAA,GAAAf,uBAAA,CAAAC,OAAA;AACA,IAAAe,MAAA,GAAAhB,uBAAA,CAAAC,OAAA;AACA,IAAAgB,SAAA,GAAAjB,uBAAA,CAAAC,OAAA;AAA2C,SAAAiB,yBAAAC,CAAA,6BAAAC,OAAA,mBAAAC,CAAA,OAAAD,OAAA,IAAAE,CAAA,OAAAF,OAAA,YAAAF,wBAAA,YAAAA,CAAAC,CAAA,WAAAA,CAAA,GAAAG,CAAA,GAAAD,CAAA,KAAAF,CAAA;AAAA,SAAAnB,wBAAAmB,CAAA,EAAAE,CAAA,SAAAA,CAAA,IAAAF,CAAA,IAAAA,CAAA,CAAAI,UAAA,SAAAJ,CAAA,eAAAA,CAAA,uBAAAA,CAAA,yBAAAA,CAAA,WAAAK,OAAA,EAAAL,CAAA,QAAAG,CAAA,GAAAJ,wBAAA,CAAAG,CAAA,OAAAC,CAAA,IAAAA,CAAA,CAAAG,GAAA,CAAAN,CAAA,UAAAG,CAAA,CAAAI,GAAA,CAAAP,CAAA,OAAAQ,CAAA,KAAAC,SAAA,UAAAC,CAAA,GAAAC,MAAA,CAAAC,cAAA,IAAAD,MAAA,CAAAE,wBAAA,WAAAC,CAAA,IAAAd,CAAA,oBAAAc,CAAA,OAAAC,cAAA,CAAAC,IAAA,CAAAhB,CAAA,EAAAc,CAAA,SAAAG,CAAA,GAAAP,CAAA,GAAAC,MAAA,CAAAE,wBAAA,CAAAb,CAAA,EAAAc,CAAA,UAAAG,CAAA,KAAAA,CAAA,CAAAV,GAAA,IAAAU,CAAA,CAAAC,GAAA,IAAAP,MAAA,CAAAC,cAAA,CAAAJ,CAAA,EAAAM,CAAA,EAAAG,CAAA,IAAAT,CAAA,CAAAM,CAAA,IAAAd,CAAA,CAAAc,CAAA,YAAAN,CAAA,CAAAH,OAAA,GAAAL,CAAA,EAAAG,CAAA,IAAAA,CAAA,CAAAe,GAAA,CAAAlB,CAAA,EAAAQ,CAAA,GAAAA,CAAA;AArB3C;;;;AAuBA;;;;AAIM,MAAOW,YAAa,sBAAQ7B,OAAO,CAAC8B,GAAG,CAAC,gCAAgC,CAAC,EAG5E;AAsBH;;;;AAAAC,OAAA,CAAAF,YAAA,GAAAA,YAAA;AAIO,MAAMG,IAAI,GAAIC,OAMpB,IACC/B,MAAM,CAACgC,GAAG,CAAC,aAAS;EAClB,MAAMC,UAAU,GAAG,CAAC,OAAOvC,UAAU,CAACA,UAAU,EAAEwC,IAAI,CACpDxC,UAAU,CAACyC,UAAU,CAAEC,OAAO,IAC5BA,OAAO,CAACF,IAAI,CACVvC,iBAAiB,CAAC0C,UAAU,CAACN,OAAO,CAACO,MAAM,IAAI,2BAA2B,CAAC,EAC3EP,OAAO,CAACQ,MAAM,GAAG5C,iBAAiB,CAAC6C,WAAW,CAACT,OAAO,CAACQ,MAAM,CAAC,GAAGE,kBAAQ,EACzEV,OAAO,CAACW,cAAc,KAAKC,SAAS,GAChChD,iBAAiB,CAACiD,SAAS,CAAC,qBAAqB,EAAExC,QAAQ,CAACyC,KAAK,CAACd,OAAO,CAACW,cAAc,CAAC,CAAC,GAC1FD,kBAAQ,EACZV,OAAO,CAACe,SAAS,KAAKH,SAAS,GAC3BhD,iBAAiB,CAACiD,SAAS,CAAC,gBAAgB,EAAExC,QAAQ,CAACyC,KAAK,CAACd,OAAO,CAACe,SAAS,CAAC,CAAC,GAChFL,kBAAQ,EACZ9C,iBAAiB,CAACoD,UAAU,CAC7B,CACF,EACDhB,OAAO,CAACiB,eAAe,GAAGjB,OAAO,CAACiB,eAAe,GAAGP,kBAAQ,CAC7D;EACD,MAAMQ,YAAY,GAAGvD,UAAU,CAACwD,cAAc,CAACjB,UAAU,CAAC;EAC1D,MAAMkB,MAAM,GAAG7C,SAAS,CAACwB,IAAI,CAACG,UAAU,CAAC;EACzC,MAAMmB,aAAa,GAAiBhB,OAA4C,IAC9Ea,YAAY,CAACI,OAAO,CAACjB,OAAO,CAAC,CAACF,IAAI,CAChClC,MAAM,CAACsD,GAAG,CAAE5C,CAAC,IAAKA,CAAC,CAAC6C,MAAM,CAAC,EAC3BlD,MAAM,CAACmD,YAAY,EACnBnD,MAAM,CAACoD,UAAU,EAAE,EACnBpD,MAAM,CAACqD,kBAAkB,CAAClE,GAAG,CAACmE,WAAW,EAAE,CAAC,EAC5CtD,MAAM,CAACuD,SAAS,CAAEC,KAAK,IAAKA,KAAK,CAACC,IAAI,KAAK,QAAQ,CAAC,EACpDzD,MAAM,CAACiD,GAAG,CAAEO,KAAK,IAAKE,IAAI,CAACC,KAAK,CAACH,KAAK,CAACC,IAAI,CAAM,CAAC,CACnD;EACH,MAAMP,MAAM,GAAInB,OAAgC,IAC9C/B,MAAM,CAAC4D,OAAO,CAAC,MAAK;IAClB,MAAMC,aAAa,GAAkB,EAAE;IACvC,OAAOd,aAAa,CAAqBzD,iBAAiB,CAACwE,IAAI,CAAC,mBAAmB,EAAE;MACnFC,IAAI,EAAE3E,QAAQ,CAAC4E,UAAU,CAAC;QACxB,GAAGjC,OAAO;QACVmB,MAAM,EAAE,IAAI;QACZe,cAAc,EAAE;UAAEC,aAAa,EAAE;QAAI;OACtC;KACF,CAAC,CAAC,CAACrC,IAAI,CACN7B,MAAM,CAACmE,QAAQ,CAAC,IAAIC,GAAG,EAA6C,EAAE,CAACC,GAAG,EAAEC,KAAK,KAAI;MACnF,MAAMC,KAAK,GAA2B,EAAE;MACxC,IAAID,KAAK,CAACE,KAAK,KAAK,IAAI,EAAE;QACxBD,KAAK,CAACE,IAAI,CAAC;UACTC,IAAI,EAAE,OAAO;UACbC,EAAE,EAAEL,KAAK,CAACK,EAAE;UACZC,KAAK,EAAEN,KAAK,CAACM,KAAK;UAClBC,WAAW,EAAEP,KAAK,CAACE,KAAK,CAACM,aAAa;UACtCC,YAAY,EAAET,KAAK,CAACE,KAAK,CAACQ,iBAAiB;UAC3CnB,aAAa;UACboB,iBAAiB,EAAEX,KAAK,CAACY,kBAAkB;UAC3CC,WAAW,EAAEb,KAAK,CAACc;SACpB,CAAC;MACJ;MACA,KAAK,IAAIhE,CAAC,GAAG,CAAC,EAAEA,CAAC,GAAGkD,KAAK,CAACe,OAAO,CAACC,MAAM,EAAElE,CAAC,EAAE,EAAE;QAC7C,MAAMmE,MAAM,GAAGjB,KAAK,CAACe,OAAO,CAACjE,CAAC,CAAC;QAC/B,IAAImE,MAAM,CAACC,aAAa,KAAK,IAAI,EAAE;UACjC3B,aAAa,CAACY,IAAI,CAACc,MAAM,CAACC,aAAa,CAAC;QAC1C;QACA,IAAI,SAAS,IAAID,MAAM,CAACE,KAAK,IAAI,OAAOF,MAAM,CAACE,KAAK,CAACC,OAAO,KAAK,QAAQ,EAAE;UACzE,IAAIC,IAAI,GAAGtB,GAAG,CAAC3D,GAAG,CAAC6E,MAAM,CAACK,KAAK,CAA4B;UAC3DD,IAAI,GAAG;YACLjB,IAAI,EAAE,SAAS;YACfgB,OAAO,EAAEH,MAAM,CAACE,KAAK,CAACC;WACvB;UACDrB,GAAG,CAAChD,GAAG,CAACkE,MAAM,CAACK,KAAK,EAAED,IAAI,CAAC;UAC3BpB,KAAK,CAACE,IAAI,CAACkB,IAAI,CAAC;QAClB,CAAC,MAAM,IAAI,YAAY,IAAIJ,MAAM,CAACE,KAAK,IAAII,KAAK,CAACC,OAAO,CAACP,MAAM,CAACE,KAAK,CAACM,UAAU,CAAC,EAAE;UACjF,MAAMxB,KAAK,GAAIF,GAAG,CAAC3D,GAAG,CAAC6E,MAAM,CAACK,KAAK,CAAC,IAAI,EAA0B;UAClE,KAAK,MAAMI,QAAQ,IAAIT,MAAM,CAACE,KAAK,CAACM,UAAU,EAAE;YAC9C,MAAMJ,IAAI,GAAGpB,KAAK,CAACyB,QAAQ,CAACJ,KAAK,CAAC;YAClC,MAAMK,QAAQ,GAAGN,IAAI,EAAEjB,IAAI,KAAK,UAAU,GACxC;cACE,GAAGiB,IAAI;cACPO,SAAS,EAAEP,IAAI,CAACO,SAAS,GAAGF,QAAQ,CAACG,QAAQ,CAACD;aAC/C,GACD;cACExB,IAAI,EAAE,UAAU;cAChB,GAAGsB,QAAQ;cACX,GAAGA,QAAQ,CAACG,QAAQ;cACpBC,IAAI,EAAEb,MAAM,CAACE,KAAK,CAACW;aACb;YACV7B,KAAK,CAACyB,QAAQ,CAACJ,KAAK,CAAC,GAAGK,QAAQ;UAClC;UACA5B,GAAG,CAAChD,GAAG,CAACkE,MAAM,CAACK,KAAK,EAAErB,KAAK,CAAC;QAC9B,CAAC,MAAM,IAAIgB,MAAM,CAACC,aAAa,KAAK,YAAY,EAAE;UAChD,MAAMa,SAAS,GAAGhC,GAAG,CAAC3D,GAAG,CAAC6E,MAAM,CAACK,KAAK,CAAwB;UAC9D,KAAK,MAAMD,IAAI,IAAIU,SAAS,EAAE;YAC5B,IAAI;cACF,MAAMC,IAAI,GAAG5C,IAAI,CAACC,KAAK,CAACgC,IAAI,CAACO,SAAmB,CAAC;cACjD3B,KAAK,CAACE,IAAI,CAAC;gBACTC,IAAI,EAAE,UAAU;gBAChBC,EAAE,EAAEgB,IAAI,CAAChB,EAAE;gBACX4B,IAAI,EAAEZ,IAAI,CAACY,IAAI;gBACfL,SAAS,EAAEI;eACZ,CAAC;cACF;YACF,CAAC,CAAC,MAAM,CAAC;UACX;QACF;MACF;MACA,OAAO,CACLjC,GAAG,EACHE,KAAK,CAACe,MAAM,KAAK,CAAC,GACdxF,MAAM,CAAC0G,IAAI,EAAE,GACb1G,MAAM,CAAC2G,IAAI,CAAC,IAAIC,WAAW,CAAC;QAAEnC;MAAK,CAAE,CAAC,CAAC,CAC5C;IACH,CAAC,CAAC,EACFvE,MAAM,CAAC2G,SAAS,CAACvE,kBAAQ,CAAC,CAC3B;EACH,CAAC,CAAC;EACJ,OAAOd,YAAY,CAACsF,EAAE,CAAC;IAAE9D,MAAM;IAAEC,aAAa;IAAEG;EAAM,CAAE,CAAC;AAC3D,CAAC,CAAC;AAEJ;;;;AAAA1B,OAAA,CAAAC,IAAA,GAAAA,IAAA;AAIO,MAAMoF,KAAK,GAAInF,OAMrB,IAA8D7B,KAAK,CAACiH,MAAM,CAACxF,YAAY,EAAEG,IAAI,CAACC,OAAO,CAAC,CAAC;AAExG;;;;AAAAF,OAAA,CAAAqF,KAAA,GAAAA,KAAA;AAIO,MAAME,WAAW,GACtBrF,OAME,IAEFlC,MAAM,CAACwH,MAAM,CAACtF,OAAO,CAAC,CAACG,IAAI,CACzBlC,MAAM,CAACsH,OAAO,CAACxF,IAAI,CAAC,EACpB5B,KAAK,CAACiH,MAAM,CAACxF,YAAY,CAAC,CAC3B;AA4DH;;;;AAAAE,OAAA,CAAAuF,WAAA,GAAAA,WAAA;AAIM,MAAOL,WAAY,SAAQhH,IAAI,CAACwH,KAEpC;EACA;;;EAGA,IAAIC,IAAIA,CAAA;IACN,OAAO,IAAI,CAAC5C,KAAK,CAAC,CAAC,CAAC,EAAEG,IAAI,KAAK,SAAS,GAAG5E,MAAM,CAAC2G,IAAI,CAAC,IAAI,CAAClC,KAAK,CAAC,CAAC,CAAC,CAACmB,OAAO,CAAC,GAAG5F,MAAM,CAAC0G,IAAI,EAAE;EAC/F;EACA;;;EAGA,IAAIY,YAAYA,CAAA;IACd,IAAI,IAAI,CAAC7C,KAAK,CAACe,MAAM,KAAK,CAAC,EAAE;MAC3B,OAAOvG,UAAU,CAACA,UAAU,CAACsI,QAAQ,CAAC;QACpCjB,IAAI,EAAElH,MAAM,CAAC0F,KAAK;QAClBc,OAAO,EAAE;OACV,CAAC;IACJ;IACA,MAAMC,IAAI,GAAG,IAAI,CAACpB,KAAK,CAAC,CAAC,CAAC;IAC1B,QAAQoB,IAAI,CAACjB,IAAI;MACf,KAAK,SAAS;QACZ,OAAO3F,UAAU,CAACA,UAAU,CAACsI,QAAQ,CAAC;UACpCjB,IAAI,EAAElH,MAAM,CAAC0F,KAAK;UAClBc,OAAO,EAAEC,IAAI,CAACD;SACf,CAAC;MACJ,KAAK,UAAU;QACb,OAAO,IAAI3G,UAAU,CAACA,UAAU,CAAC;UAC/BqH,IAAI,EAAElH,MAAM,CAAC0F,KAAK;UAClBL,KAAK,EAAEhF,KAAK,CAACqH,EAAE,CAAC7H,UAAU,CAACuI,YAAY,CAACC,WAAW,CAAC;YAClD5C,EAAE,EAAEgB,IAAI,CAAChB,EAAE;YACX4B,IAAI,EAAEZ,IAAI,CAACY,IAAI;YACfiB,MAAM,EAAE7B,IAAI,CAACO;WACd,CAAC;SACH,CAAC;MACJ,KAAK,OAAO;QACV,OAAOnH,UAAU,CAACA,UAAU,CAAC0I,KAAK;IACtC;EACF","ignoreList":[]}

package/dist/cjs/OpenAiCompletions.js

@@ -11,9 +11,11 @@ var Completions = _interopRequireWildcard(require("@effect/ai/Completions"));
 var Arr = _interopRequireWildcard(require("effect/Array"));
 var Effect = _interopRequireWildcard(require("effect/Effect"));
 var Layer = _interopRequireWildcard(require("effect/Layer"));
+var Predicate = _interopRequireWildcard(require("effect/Predicate"));
 var Stream = _interopRequireWildcard(require("effect/Stream"));
 var _OpenAiClient = require("./OpenAiClient.js");
 var _OpenAiConfig = require("./OpenAiConfig.js");
+var _OpenAiTelemetry = require("./OpenAiTelemetry.js");
 var OpenAiTokenizer = _interopRequireWildcard(require("./OpenAiTokenizer.js"));
 function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function (e) { return e ? t : r; })(e); }
 function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != typeof e && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && {}.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
@@ -63,21 +65,30 @@ const make = options => Effect.gen(function* () {
     }));
   };
   return yield* Completions.make({
-    create(options) {
-      return makeRequest(options).pipe(Effect.flatMap(client.client.createChatCompletion), Effect.catchAll(cause => Effect.fail(new _AiError.AiError({
+    create({
+      span,
+      ...options
+    }) {
+      return makeRequest(options).pipe(Effect.tap(request => annotateRequest(span, request)), Effect.flatMap(client.client.createChatCompletion), Effect.tap(response => annotateChatResponse(span, response)), Effect.flatMap(response => makeResponse(response, "create", options.tools.length === 1 && options.tools[0].structured ? options.tools[0] : undefined)), Effect.catchAll(cause => Effect.fail(new _AiError.AiError({
         module: "OpenAiCompletions",
         method: "create",
         description: "An error occurred",
         cause
-      }))), Effect.flatMap(response => makeResponse(response, "create", options.tools.length === 1 && options.tools[0].structured ? options.tools[0] : undefined)));
+      }))));
     },
-    stream(options) {
-      return makeRequest(options).pipe(Effect.map(client.stream), Stream.unwrap, Stream.catchAll(cause => Effect.fail(new _AiError.AiError({
+    stream({
+      span,
+      ...options
+    }) {
+      return makeRequest(options).pipe(Effect.tap(request => annotateRequest(span, request)), Effect.map(client.stream), Stream.unwrap, Stream.tap(response => {
+        annotateStreamResponse(span, response);
+        return Effect.void;
+      }), Stream.map(response => response.asAiResponse), Stream.catchAll(cause => Effect.fail(new _AiError.AiError({
         module: "OpenAiCompletions",
         method: "stream",
         description: "An error occurred",
         cause
-      }))), Stream.map(response => response.asAiResponse));
+      }))));
     }
   });
 });
@@ -224,4 +235,69 @@ const makeSystemMessage = content => {
   };
 };
 const safeName = name => name.replace(/[^a-zA-Z0-9_-]/g, "_").replace(/_+/, "_");
+const annotateRequest = (span, request) => {
+  (0, _OpenAiTelemetry.addGenAIAnnotations)(span, {
+    system: "openai",
+    operation: {
+      name: "chat"
+    },
+    request: {
+      model: request.model,
+      temperature: request.temperature,
+      topP: request.top_p,
+      maxTokens: request.max_tokens,
+      stopSequences: Arr.ensure(request.stop).filter(Predicate.isNotNullable),
+      frequencyPenalty: request.frequency_penalty,
+      presencePenalty: request.presence_penalty,
+      seed: request.seed
+    },
+    openai: {
+      request: {
+        responseFormat: request.response_format?.type,
+        serviceTier: request.service_tier
+      }
+    }
+  });
+};
+const annotateChatResponse = (span, response) => {
+  (0, _OpenAiTelemetry.addGenAIAnnotations)(span, {
+    response: {
+      id: response.id,
+      model: response.model,
+      finishReasons: response.choices.map(choice => choice.finish_reason)
+    },
+    usage: {
+      inputTokens: response.usage?.prompt_tokens,
+      outputTokens: response.usage?.completion_tokens
+    },
+    openai: {
+      response: {
+        systemFingerprint: response.system_fingerprint,
+        serviceTier: response.service_tier
+      }
+    }
+  });
+};
+const annotateStreamResponse = (span, response) => {
+  const usage = response.parts.find(part => part._tag === "Usage");
+  if (Predicate.isNotNullable(usage)) {
+    (0, _OpenAiTelemetry.addGenAIAnnotations)(span, {
+      response: {
+        id: usage.id,
+        model: usage.model,
+        finishReasons: usage.finishReasons
+      },
+      usage: {
+        inputTokens: usage.inputTokens,
+        outputTokens: usage.outputTokens
+      },
+      openai: {
+        response: {
+          systemFingerprint: usage.systemFingerprint,
+          serviceTier: usage.serviceTier
+        }
+      }
+    });
+  }
+};
 //# sourceMappingURL=OpenAiCompletions.js.map

package/dist/cjs/OpenAiCompletions.js.map

@@ -1 +1 @@
-{"version":3,"file":"OpenAiCompletions.js","names":["_AiError","require","AiResponse","_interopRequireWildcard","AiRole","Completions","Arr","Effect","Layer","Stream","_OpenAiClient","_OpenAiConfig","OpenAiTokenizer","_getRequireWildcardCache","e","WeakMap","r","t","__esModule","default","has","get","n","__proto__","a","Object","defineProperty","getOwnPropertyDescriptor","u","hasOwnProperty","call","i","set","make","options","gen","client","OpenAiClient","config","OpenAiConfig","getOrUndefined","makeRequest","input","required","system","tools","useStructured","length","structured","map","context","model","unsafeMap","key","messages","makeMessages","response_format","type","json_schema","strict","name","description","schema","parameters","undefined","tool","function","tool_choice","create","pipe","flatMap","createChatCompletion","catchAll","cause","fail","AiError","module","method","response","makeResponse","stream","unwrap","asAiResponse","layerCompletions","effect","exports","layer","merge","_tag","makeSystemMessage","value","message","push","convertMessage","formatRole","role","safeName","parts","toolCalls","flushContent","content","flushToolCalls","tool_calls","part","id","arguments","JSON","stringify","params","tool_call_id","toolCallId","makeContentPart","image_url","url","asDataUri","detail","quality","text","structuredTool","head","choices","mapError","choice","empty","withToolCallsJson","res","fromText","toolCall","succeed","replace"],"sources":["../../src/OpenAiCompletions.ts"],"sourcesContent":[null],"mappings":";;;;;;AAGA,IAAAA,QAAA,GAAAC,OAAA;AAEA,IAAAC,UAAA,GAAAC,uBAAA,CAAAF,OAAA;AACA,IAAAG,MAAA,GAAAD,uBAAA,CAAAF,OAAA;AACA,IAAAI,WAAA,GAAAF,uBAAA,CAAAF,OAAA;AAEA,IAAAK,GAAA,GAAAH,uBAAA,CAAAF,OAAA;AACA,IAAAM,MAAA,GAAAJ,uBAAA,CAAAF,OAAA;AACA,IAAAO,KAAA,GAAAL,uBAAA,CAAAF,OAAA;AAEA,IAAAQ,MAAA,GAAAN,uBAAA,CAAAF,OAAA;AAEA,IAAAS,aAAA,GAAAT,OAAA;AACA,IAAAU,aAAA,GAAAV,OAAA;AACA,IAAAW,eAAA,GAAAT,uBAAA,CAAAF,OAAA;AAAuD,SAAAY,yBAAAC,CAAA,6BAAAC,OAAA,mBAAAC,CAAA,OAAAD,OAAA,IAAAE,CAAA,OAAAF,OAAA,YAAAF,wBAAA,YAAAA,CAAAC,CAAA,WAAAA,CAAA,GAAAG,CAAA,GAAAD,CAAA,KAAAF,CAAA;AAAA,SAAAX,wBAAAW,CAAA,EAAAE,CAAA,SAAAA,CAAA,IAAAF,CAAA,IAAAA,CAAA,CAAAI,UAAA,SAAAJ,CAAA,eAAAA,CAAA,uBAAAA,CAAA,yBAAAA,CAAA,WAAAK,OAAA,EAAAL,CAAA,QAAAG,CAAA,GAAAJ,wBAAA,CAAAG,CAAA,OAAAC,CAAA,IAAAA,CAAA,CAAAG,GAAA,CAAAN,CAAA,UAAAG,CAAA,CAAAI,GAAA,CAAAP,CAAA,OAAAQ,CAAA,KAAAC,SAAA,UAAAC,CAAA,GAAAC,MAAA,CAAAC,cAAA,IAAAD,MAAA,CAAAE,wBAAA,WAAAC,CAAA,IAAAd,CAAA,oBAAAc,CAAA,OAAAC,cAAA,CAAAC,IAAA,CAAAhB,CAAA,EAAAc,CAAA,SAAAG,CAAA,GAAAP,CAAA,GAAAC,MAAA,CAAAE,wBAAA,CAAAb,CAAA,EAAAc,CAAA,UAAAG,CAAA,KAAAA,CAAA,CAAAV,GAAA,IAAAU,CAAA,CAAAC,GAAA,IAAAP,MAAA,CAAAC,cAAA,CAAAJ,CAAA,EAAAM,CAAA,EAAAG,CAAA,IAAAT,CAAA,CAAAM,CAAA,IAAAd,CAAA,CAAAc,CAAA,YAAAN,CAAA,CAAAH,OAAA,GAAAL,CAAA,EAAAG,CAAA,IAAAA,CAAA,CAAAe,GAAA,CAAAlB,CAAA,EAAAQ,CAAA,GAAAA,CAAA;AAjBvD;;;;AAmBA,MAAMW,IAAI,GAAIC,OAEb,IACC3B,MAAM,CAAC4B,GAAG,CAAC,aAAS;EAClB,MAAMC,MAAM,GAAG,OAAOC,0BAAY;EAClC,MAAMC,MAAM,GAAG,OAAOC,0BAAY,CAACC,cAAc;EAEjD,MAAMC,WAAW,GAAGA,CAAC;IAAEC,KAAK;IAAEC,QAAQ;IAAEC,MAAM;IAAEC;EAAK,CAAiC,KAAI;IACxF,MAAMC,aAAa,GAAGD,KAAK,CAACE,MAAM,KAAK,CAAC,IAAIF,KAAK,CAAC,CAAC,CAAC,CAACG,UAAU;IAC/D,OAAOzC,MAAM,CAAC0C,GAAG,CACf1C,MAAM,CAAC2C,OAAO,EAAS,EACtBA,OAAO,KAA4D;MAClEC,KAAK,EAAEjB,OAAO,CAACiB,KAAK;MACpB,GAAGb,MAAM;MACT,GAAGY,OAAO,CAACE,SAAS,CAAC/B,GAAG,CAACkB,0BAAY,CAACc,GAAG,CAAC;MAC1CC,QAAQ,EAAEC,YAAY,CAACb,KAAK,EAAEE,MAAM,CAAC;MACrCY,eAAe,EAAEV,aAAa,GAC5B;QACEW,IAAI,EAAE,aAAa;QACnBC,WAAW,EAAE;UACXC,MAAM,EAAE,IAAI;UACZC,IAAI,EAAEf,KAAK,CAAC,CAAC,CAAC,CAACe,IAAI;UACnBC,WAAW,EAAEhB,KAAK,CAAC,CAAC,CAAC,CAACgB,WAAW;UACjCC,MAAM,EAAEjB,KAAK,CAAC,CAAC,CAAC,CAACkB;;OAEpB,GACDC,SAAS;MACXnB,KAAK,EAAE,CAACC,aAAa,IAAID,KAAK,CAACE,MAAM,GAAG,CAAC,GACvCF,KAAK,CAACI,GAAG,CAAEgB,IAAI,KAAM;QACnBR,IAAI,EAAE,UAAU;QAChBS,QAAQ,EAAE;UACRN,IAAI,EAAEK,IAAI,CAACL,IAAI;UACfC,WAAW,EAAEI,IAAI,CAACJ,WAAW;UAC7BE,UAAU,EAAEE,IAAI,CAACF,UAAiB;UAClCJ,MAAM,EAAE;;OAEX,CAAC,CAAC,GACHK,SAAS;MACXG,WAAW,EAAE,CAACrB,aAAa,IAAID,KAAK,CAACE,MAAM,GAAG,CAAC,GAC7C,OAAOJ,QAAQ,KAAK,SAAS,GAAIA,QAAQ,GAAG,UAAU,GAAG,MAAM,GAAI;QACjEc,IAAI,EAAE,UAAU;QAChBS,QAAQ,EAAE;UAAEN,IAAI,EAAEjB;QAAQ;OAC3B,GACDqB;KACH,CAAC,CACH;EACH,CAAC;EAED,OAAO,OAAO3D,WAAW,CAAC4B,IAAI,CAAC;IAC7BmC,MAAMA,CAAClC,OAAO;MACZ,OAAOO,WAAW,CAACP,OAAO,CAAC,CAACmC,IAAI,CAC9B9D,MAAM,CAAC+D,OAAO,CAAClC,MAAM,CAACA,MAAM,CAACmC,oBAAoB,CAAC,EAClDhE,MAAM,CAACiE,QAAQ,CAAEC,KAAK,IACpBlE,MAAM,CAACmE,IAAI,CACT,IAAIC,gBAAO,CAAC;QACVC,MAAM,EAAE,mBAAmB;QAC3BC,MAAM,EAAE,QAAQ;QAChBhB,WAAW,EAAE,mBAAmB;QAChCY;OACD,CAAC,CACH,CACF,EACDlE,MAAM,CAAC+D,OAAO,CAAEQ,QAAQ,IACtBC,YAAY,CACVD,QAAQ,EACR,QAAQ,EACR5C,OAAO,CAACW,KAAK,CAACE,MAAM,KAAK,CAAC,IAAIb,OAAO,CAACW,KAAK,CAAC,CAAC,CAAC,CAACG,UAAU,GAAGd,OAAO,CAACW,KAAK,CAAC,CAAC,CAAC,GAAGmB,SAAS,CACzF,CACF,CACF;IACH,CAAC;IACDgB,MAAMA,CAAC9C,OAAO;MACZ,OAAOO,WAAW,CAACP,OAAO,CAAC,CAACmC,IAAI,CAC9B9D,MAAM,CAAC0C,GAAG,CAACb,MAAM,CAAC4C,MAAM,CAAC,EACzBvE,MAAM,CAACwE,MAAM,EACbxE,MAAM,CAAC+D,QAAQ,CAAEC,KAAK,IACpBlE,MAAM,CAACmE,IAAI,CACT,IAAIC,gBAAO,CAAC;QACVC,MAAM,EAAE,mBAAmB;QAC3BC,MAAM,EAAE,QAAQ;QAChBhB,WAAW,EAAE,mBAAmB;QAChCY;OACD,CAAC,CACH,CACF,EACDhE,MAAM,CAACwC,GAAG,CAAE6B,QAAQ,IAAKA,QAAQ,CAACI,YAAY,CAAC,CAChD;IACH;GACD,CAAC;AACJ,CAAC,CAAC;AAEJ;;;;AAIO,MAAMC,gBAAgB,GAAIjD,OAEhC,IAAgE1B,KAAK,CAAC4E,MAAM,CAAC/E,WAAW,CAACA,WAAW,EAAE4B,IAAI,CAACC,OAAO,CAAC,CAAC;AAErH;;;;AAAAmD,OAAA,CAAAF,gBAAA,GAAAA,gBAAA;AAIO,MAAMG,KAAK,GAAIpD,OAErB,IACC1B,KAAK,CAAC+E,KAAK,CAACJ,gBAAgB,CAACjD,OAAO,CAAC,EAAEtB,eAAe,CAAC0E,KAAK,CAACpD,OAAO,CAAC,CAAC;AAAAmD,OAAA,CAAAC,KAAA,GAAAA,KAAA;AAExE,MAAM/B,YAAY,GAAGA,CACnBb,KAAsB,EACtBE,MAA6B,KACuD;EACpF,MAAMU,QAAQ,GAAiEV,MAAM,CAAC4C,IAAI,KAAK,MAAM,GACnG,CAACC,iBAAiB,CAAC7C,MAAM,CAAC8C,KAAK,CAAC,CAAC,GACjC,EAAE;EACJ,KAAK,MAAMC,OAAO,IAAIjD,KAAK,EAAE;IAC3B;IACAY,QAAQ,CAACsC,IAAI,CAAC,GAAGC,cAAc,CAACF,OAAO,CAAC,CAAC;EAC3C;EACA,OAAOrC,QAAe;AACxB,CAAC;AAED,MAAMwC,UAAU,GAAIC,IAAmB,IAAI;EACzC,QAAQA,IAAI,CAACP,IAAI;IACf,KAAK,cAAc;MACjB,OAAO;QACLO,IAAI,EAAE,MAAM;QACZnC,IAAI,EAAEoC,QAAQ,CAACD,IAAI,CAACnC,IAAI;OAChB;IACZ,KAAK,MAAM;MACT,OAAO;QACLmC,IAAI,EAAE;OACE;IACZ,KAAK,OAAO;MACV,OAAO;QACLA,IAAI,EAAE;OACE;EACd;AACF,CAAC;AAED,MAAMF,cAAc,GAClBF,OAAwB,IACwC;EAChE,MAAMrC,QAAQ,GAAiE,EAAE;EACjF,IAAI2C,KAAK,GAAgF,EAAE;EAC3F,IAAIC,SAAS,GAAkE,EAAE;EACjF,SAASC,YAAYA,CAAA;IACnB,IAAIF,KAAK,CAAClD,MAAM,KAAK,CAAC,EAAE;IACxBO,QAAQ,CAACsC,IAAI,CAAC;MACZ,GAAGE,UAAU,CAACH,OAAO,CAACI,IAAI,CAAC;MAC3BK,OAAO,EAAEH;KACV,CAAC;IACFA,KAAK,GAAG,EAAE;EACZ;EACA,SAASI,cAAcA,CAAA;IACrB,IAAIH,SAAS,CAACnD,MAAM,KAAK,CAAC,EAAE;IAC5BO,QAAQ,CAACsC,IAAI,CAAC;MACZG,IAAI,EAAE,WAAW;MACjBK,OAAO,EAAE,IAAI;MACbE,UAAU,EAAEJ;KACb,CAAC;IACFA,SAAS,GAAG,EAAE;EAChB;EACA,KAAK,MAAMK,IAAI,IAAIZ,OAAO,CAACM,KAAK,EAAE;IAChC,IAAIM,IAAI,CAACf,IAAI,KAAK,UAAU,EAAE;MAC5BW,YAAY,EAAE;MACdD,SAAS,CAACN,IAAI,CAAC;QACbY,EAAE,EAAED,IAAI,CAACC,EAAE;QACX/C,IAAI,EAAE,UAAU;QAChBS,QAAQ,EAAE;UACRN,IAAI,EAAE2C,IAAI,CAAC3C,IAAI;UACf6C,SAAS,EAAEC,IAAI,CAACC,SAAS,CAACJ,IAAI,CAACK,MAAM;;OAExC,CAAC;IACJ,CAAC,MAAM,IAAIL,IAAI,CAACf,IAAI,KAAK,kBAAkB,EAAE;MAC3CW,YAAY,EAAE;MACdE,cAAc,EAAE;MAChB/C,QAAQ,CAACsC,IAAI,CAAC;QACZG,IAAI,EAAE,MAAM;QACZc,YAAY,EAAEN,IAAI,CAACO,UAAU;QAC7BV,OAAO,EAAEM,IAAI,CAACC,SAAS,CAACJ,IAAI,CAACb,KAAK;OACnC,CAAC;IACJ,CAAC,MAAM;MACLW,cAAc,EAAE;MAChBJ,KAAK,CAACL,IAAI,CAACmB,eAAe,CAACR,IAAI,CAAC,CAAC;IACnC;EACF;EACAJ,YAAY,EAAE;EACdE,cAAc,EAAE;EAChB,OAAO/C,QAAQ;AACjB,CAAC;AAED,MAAMyD,eAAe,GACnBR,IAAiE,IACO;EACxE,QAAQA,IAAI,CAACf,IAAI;IACf,KAAK,OAAO;MACV,OAAO;QACL/B,IAAI,EAAE,WAAW;QACjBuD,SAAS,EAAE;UACTC,GAAG,EAAEV,IAAI,CAACW,SAAS;UACnBC,MAAM,EAAEZ,IAAI,CAACa;;OAEhB;IACH,KAAK,UAAU;MACb,OAAO;QACL3D,IAAI,EAAE,WAAW;QACjBuD,SAAS,EAAE;UACTC,GAAG,EAAEV,IAAI,CAACU,GAAG;UACbE,MAAM,EAAEZ,IAAI,CAACa;;OAEhB;IACH,KAAK,MAAM;MACT,OAAO;QACL3D,IAAI,EAAE,MAAM;QACZ4D,IAAI,EAAEd,IAAI,CAACH;OACZ;EACL;AACF,CAAC;AAED,MAAMrB,YAAY,GAAGA,CACnBD,QAAgD,EAChDD,MAAc,EACdyC,cAGC,KAEDhH,GAAG,CAACiH,IAAI,CAACzC,QAAQ,CAAC0C,OAAO,CAAC,CAACnD,IAAI,CAC7B9D,MAAM,CAACkH,QAAQ,CAAC,MACd,IAAI9C,gBAAO,CAAC;EACVC,MAAM,EAAE,mBAAmB;EAC3BC,MAAM;EACNhB,WAAW,EAAE;CACd,CAAC,CACH,EACDtD,MAAM,CAAC+D,OAAO,CAAEoD,MAAM,IAAI;EACxB,IAAIJ,cAAc,EAAE;IAClB,OAAOpH,UAAU,CAACA,UAAU,CAACyH,KAAK,CAACC,iBAAiB,CAAC,CACnD;MACEpB,EAAE,EAAE1B,QAAQ,CAAC0B,EAAE;MACf5C,IAAI,EAAE0D,cAAc,CAAC1D,IAAI;MACzBgD,MAAM,EAAEc,MAAM,CAAC/B,OAAO,CAACS;KACxB,CACF,CAAC;EACJ;EAEA,MAAMyB,GAAG,GAAG,OAAOH,MAAM,CAAC/B,OAAO,CAACS,OAAO,KAAK,QAAQ,GACpDlG,UAAU,CAACA,UAAU,CAAC4H,QAAQ,CAAC;IAC7B/B,IAAI,EAAE3F,MAAM,CAAC+C,KAAK;IAClBiD,OAAO,EAAEsB,MAAM,CAAC/B,OAAO,CAACS;GACzB,CAAC,GACFlG,UAAU,CAACA,UAAU,CAACyH,KAAK;EAE7B,IAAID,MAAM,CAAC/B,OAAO,CAACW,UAAU,IAAIoB,MAAM,CAAC/B,OAAO,CAACW,UAAU,CAACvD,MAAM,GAAG,CAAC,EAAE;IACrE,OAAO8E,GAAG,CAACD,iBAAiB,CAACF,MAAM,CAAC/B,OAAO,CAACW,UAAU,CAACrD,GAAG,CAAE8E,QAAQ,KAAM;MACxEvB,EAAE,EAAEuB,QAAQ,CAACvB,EAAE;MACf5C,IAAI,EAAEmE,QAAQ,CAAC7D,QAAQ,CAACN,IAAI;MAC5BgD,MAAM,EAAEmB,QAAQ,CAAC7D,QAAQ,CAACuC;KAC3B,CAAC,CAAC,CAAC;EACN;EAEA,OAAOlG,MAAM,CAACyH,OAAO,CAACH,GAAG,CAAC;AAC5B,CAAC,CAAC,CACH;AAEH,MAAMpC,iBAAiB,GAAIW,OAAe,IAAiE;EACzG,OAAO;IACLL,IAAI,EAAE,QAAQ;IACdK;GACD;AACH,CAAC;AAED,MAAMJ,QAAQ,GAAIpC,IAAY,IAAKA,IAAI,CAACqE,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAACA,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC","ignoreList":[]}
+{"version":3,"file":"OpenAiCompletions.js","names":["_AiError","require","AiResponse","_interopRequireWildcard","AiRole","Completions","Arr","Effect","Layer","Predicate","Stream","_OpenAiClient","_OpenAiConfig","_OpenAiTelemetry","OpenAiTokenizer","_getRequireWildcardCache","e","WeakMap","r","t","__esModule","default","has","get","n","__proto__","a","Object","defineProperty","getOwnPropertyDescriptor","u","hasOwnProperty","call","i","set","make","options","gen","client","OpenAiClient","config","OpenAiConfig","getOrUndefined","makeRequest","input","required","system","tools","useStructured","length","structured","map","context","model","unsafeMap","key","messages","makeMessages","response_format","type","json_schema","strict","name","description","schema","parameters","undefined","tool","function","tool_choice","create","span","pipe","tap","request","annotateRequest","flatMap","createChatCompletion","response","annotateChatResponse","makeResponse","catchAll","cause","fail","AiError","module","method","stream","unwrap","annotateStreamResponse","void","asAiResponse","layerCompletions","effect","exports","layer","merge","_tag","makeSystemMessage","value","message","push","convertMessage","formatRole","role","safeName","parts","toolCalls","flushContent","content","flushToolCalls","tool_calls","part","id","arguments","JSON","stringify","params","tool_call_id","toolCallId","makeContentPart","image_url","url","asDataUri","detail","quality","text","structuredTool","head","choices","mapError","choice","empty","withToolCallsJson","res","fromText","toolCall","succeed","replace","addGenAIAnnotations","operation","temperature","topP","top_p","maxTokens","max_tokens","stopSequences","ensure","stop","filter","isNotNullable","frequencyPenalty","frequency_penalty","presencePenalty","presence_penalty","seed","openai","responseFormat","serviceTier","service_tier","finishReasons","finish_reason","usage","inputTokens","prompt_tokens","outputTokens","completion_tokens","systemFingerprint","system_fingerprint","find"],"sources":["../../src/OpenAiCompletions.ts"],"sourcesContent":[null],"mappings":";;;;;;AAGA,IAAAA,QAAA,GAAAC,OAAA;AAEA,IAAAC,UAAA,GAAAC,uBAAA,CAAAF,OAAA;AACA,IAAAG,MAAA,GAAAD,uBAAA,CAAAF,OAAA;AACA,IAAAI,WAAA,GAAAF,uBAAA,CAAAF,OAAA;AAEA,IAAAK,GAAA,GAAAH,uBAAA,CAAAF,OAAA;AACA,IAAAM,MAAA,GAAAJ,uBAAA,CAAAF,OAAA;AACA,IAAAO,KAAA,GAAAL,uBAAA,CAAAF,OAAA;AAEA,IAAAQ,SAAA,GAAAN,uBAAA,CAAAF,OAAA;AACA,IAAAS,MAAA,GAAAP,uBAAA,CAAAF,OAAA;AAIA,IAAAU,aAAA,GAAAV,OAAA;AACA,IAAAW,aAAA,GAAAX,OAAA;AACA,IAAAY,gBAAA,GAAAZ,OAAA;AACA,IAAAa,eAAA,GAAAX,uBAAA,CAAAF,OAAA;AAAuD,SAAAc,yBAAAC,CAAA,6BAAAC,OAAA,mBAAAC,CAAA,OAAAD,OAAA,IAAAE,CAAA,OAAAF,OAAA,YAAAF,wBAAA,YAAAA,CAAAC,CAAA,WAAAA,CAAA,GAAAG,CAAA,GAAAD,CAAA,KAAAF,CAAA;AAAA,SAAAb,wBAAAa,CAAA,EAAAE,CAAA,SAAAA,CAAA,IAAAF,CAAA,IAAAA,CAAA,CAAAI,UAAA,SAAAJ,CAAA,eAAAA,CAAA,uBAAAA,CAAA,yBAAAA,CAAA,WAAAK,OAAA,EAAAL,CAAA,QAAAG,CAAA,GAAAJ,wBAAA,CAAAG,CAAA,OAAAC,CAAA,IAAAA,CAAA,CAAAG,GAAA,CAAAN,CAAA,UAAAG,CAAA,CAAAI,GAAA,CAAAP,CAAA,OAAAQ,CAAA,KAAAC,SAAA,UAAAC,CAAA,GAAAC,MAAA,CAAAC,cAAA,IAAAD,MAAA,CAAAE,wBAAA,WAAAC,CAAA,IAAAd,CAAA,oBAAAc,CAAA,OAAAC,cAAA,CAAAC,IAAA,CAAAhB,CAAA,EAAAc,CAAA,SAAAG,CAAA,GAAAP,CAAA,GAAAC,MAAA,CAAAE,wBAAA,CAAAb,CAAA,EAAAc,CAAA,UAAAG,CAAA,KAAAA,CAAA,CAAAV,GAAA,IAAAU,CAAA,CAAAC,GAAA,IAAAP,MAAA,CAAAC,cAAA,CAAAJ,CAAA,EAAAM,CAAA,EAAAG,CAAA,IAAAT,CAAA,CAAAM,CAAA,IAAAd,CAAA,CAAAc,CAAA,YAAAN,CAAA,CAAAH,OAAA,GAAAL,CAAA,EAAAG,CAAA,IAAAA,CAAA,CAAAe,GAAA,CAAAlB,CAAA,EAAAQ,CAAA,GAAAA,CAAA;AArBvD;;;;AA6BA,MAAMW,IAAI,GAAIC,OAEb,IACC7B,MAAM,CAAC8B,GAAG,CAAC,aAAS;EAClB,MAAMC,MAAM,GAAG,OAAOC,0BAAY;EAClC,MAAMC,MAAM,GAAG,OAAOC,0BAAY,CAACC,cAAc;EAEjD,MAAMC,WAAW,GAAGA,CAAC;IAAEC,KAAK;IAAEC,QAAQ;IAAEC,MAAM;IAAEC;EAAK,CAAiC,KAAI;IACxF,MAAMC,aAAa,GAAGD,KAAK,CAACE,MAAM,KAAK,CAAC,IAAIF,KAAK,CAAC,CAAC,CAAC,CAACG,UAAU;IAC/D,OAAO3C,MAAM,CAAC4C,GAAG,CACf5C,MAAM,CAAC6C,OAAO,EAAS,EACtBA,OAAO,KAA4D;MAClEC,KAAK,EAAEjB,OAAO,CAACiB,KAAK;MACpB,GAAGb,MAAM;MACT,GAAGY,OAAO,CAACE,SAAS,CAAC/B,GAAG,CAACkB,0BAAY,CAACc,GAAG,CAAC;MAC1CC,QAAQ,EAAEC,YAAY,CAACb,KAAK,EAAEE,MAAM,CAAC;MACrCY,eAAe,EAAEV,aAAa,GAC5B;QACEW,IAAI,EAAE,aAAa;QACnBC,WAAW,EAAE;UACXC,MAAM,EAAE,IAAI;UACZC,IAAI,EAAEf,KAAK,CAAC,CAAC,CAAC,CAACe,IAAI;UACnBC,WAAW,EAAEhB,KAAK,CAAC,CAAC,CAAC,CAACgB,WAAW;UACjCC,MAAM,EAAEjB,KAAK,CAAC,CAAC,CAAC,CAACkB;;OAEpB,GACDC,SAAS;MACXnB,KAAK,EAAE,CAACC,aAAa,IAAID,KAAK,CAACE,MAAM,GAAG,CAAC,GACvCF,KAAK,CAACI,GAAG,CAAEgB,IAAI,KAAM;QACnBR,IAAI,EAAE,UAAU;QAChBS,QAAQ,EAAE;UACRN,IAAI,EAAEK,IAAI,CAACL,IAAI;UACfC,WAAW,EAAEI,IAAI,CAACJ,WAAW;UAC7BE,UAAU,EAAEE,IAAI,CAACF,UAAiB;UAClCJ,MAAM,EAAE;;OAEX,CAAC,CAAC,GACHK,SAAS;MACXG,WAAW,EAAE,CAACrB,aAAa,IAAID,KAAK,CAACE,MAAM,GAAG,CAAC,GAC7C,OAAOJ,QAAQ,KAAK,SAAS,GAAIA,QAAQ,GAAG,UAAU,GAAG,MAAM,GAAI;QACjEc,IAAI,EAAE,UAAU;QAChBS,QAAQ,EAAE;UAAEN,IAAI,EAAEjB;QAAQ;OAC3B,GACDqB;KACH,CAAC,CACH;EACH,CAAC;EAED,OAAO,OAAO7D,WAAW,CAAC8B,IAAI,CAAC;IAC7BmC,MAAMA,CAAC;MAAEC,IAAI;MAAE,GAAGnC;IAAO,CAAE;MACzB,OAAOO,WAAW,CAACP,OAAO,CAAC,CAACoC,IAAI,CAC9BjE,MAAM,CAACkE,GAAG,CAAEC,OAAO,IAAKC,eAAe,CAACJ,IAAI,EAAEG,OAAO,CAAC,CAAC,EACvDnE,MAAM,CAACqE,OAAO,CAACtC,MAAM,CAACA,MAAM,CAACuC,oBAAoB,CAAC,EAClDtE,MAAM,CAACkE,GAAG,CAAEK,QAAQ,IAAKC,oBAAoB,CAACR,IAAI,EAAEO,QAAQ,CAAC,CAAC,EAC9DvE,MAAM,CAACqE,OAAO,CAAEE,QAAQ,IACtBE,YAAY,CACVF,QAAQ,EACR,QAAQ,EACR1C,OAAO,CAACW,KAAK,CAACE,MAAM,KAAK,CAAC,IAAIb,OAAO,CAACW,KAAK,CAAC,CAAC,CAAC,CAACG,UAAU,GACrDd,OAAO,CAACW,KAAK,CAAC,CAAC,CAAC,GAChBmB,SAAS,CACd,CACF,EACD3D,MAAM,CAAC0E,QAAQ,CAAEC,KAAK,IACpB3E,MAAM,CAAC4E,IAAI,CACT,IAAIC,gBAAO,CAAC;QACVC,MAAM,EAAE,mBAAmB;QAC3BC,MAAM,EAAE,QAAQ;QAChBvB,WAAW,EAAE,mBAAmB;QAChCmB;OACD,CAAC,CACH,CACF,CACF;IACH,CAAC;IACDK,MAAMA,CAAC;MAAEhB,IAAI;MAAE,GAAGnC;IAAO,CAAE;MACzB,OAAOO,WAAW,CAACP,OAAO,CAAC,CAACoC,IAAI,CAC9BjE,MAAM,CAACkE,GAAG,CAAEC,OAAO,IAAKC,eAAe,CAACJ,IAAI,EAAEG,OAAO,CAAC,CAAC,EACvDnE,MAAM,CAAC4C,GAAG,CAACb,MAAM,CAACiD,MAAM,CAAC,EACzB7E,MAAM,CAAC8E,MAAM,EACb9E,MAAM,CAAC+D,GAAG,CAAEK,QAAQ,IAAI;QACtBW,sBAAsB,CAAClB,IAAI,EAAEO,QAAQ,CAAC;QACtC,OAAOvE,MAAM,CAACmF,IAAI;MACpB,CAAC,CAAC,EACFhF,MAAM,CAACyC,GAAG,CAAE2B,QAAQ,IAAKA,QAAQ,CAACa,YAAY,CAAC,EAC/CjF,MAAM,CAACuE,QAAQ,CAAEC,KAAK,IACpB3E,MAAM,CAAC4E,IAAI,CACT,IAAIC,gBAAO,CAAC;QACVC,MAAM,EAAE,mBAAmB;QAC3BC,MAAM,EAAE,QAAQ;QAChBvB,WAAW,EAAE,mBAAmB;QAChCmB;OACD,CAAC,CACH,CACF,CACF;IACH;GACD,CAAC;AACJ,CAAC,CAAC;AAEJ;;;;AAIO,MAAMU,gBAAgB,GAAIxD,OAEhC,IAAgE5B,KAAK,CAACqF,MAAM,CAACxF,WAAW,CAACA,WAAW,EAAE8B,IAAI,CAACC,OAAO,CAAC,CAAC;AAErH;;;;AAAA0D,OAAA,CAAAF,gBAAA,GAAAA,gBAAA;AAIO,MAAMG,KAAK,GAAI3D,OAErB,IACC5B,KAAK,CAACwF,KAAK,CAACJ,gBAAgB,CAACxD,OAAO,CAAC,EAAEtB,eAAe,CAACiF,KAAK,CAAC3D,OAAO,CAAC,CAAC;AAAA0D,OAAA,CAAAC,KAAA,GAAAA,KAAA;AAExE,MAAMtC,YAAY,GAAGA,CACnBb,KAAsB,EACtBE,MAA6B,KACuD;EACpF,MAAMU,QAAQ,GAAiEV,MAAM,CAACmD,IAAI,KAAK,MAAM,GACnG,CAACC,iBAAiB,CAACpD,MAAM,CAACqD,KAAK,CAAC,CAAC,GACjC,EAAE;EACJ,KAAK,MAAMC,OAAO,IAAIxD,KAAK,EAAE;IAC3B;IACAY,QAAQ,CAAC6C,IAAI,CAAC,GAAGC,cAAc,CAACF,OAAO,CAAC,CAAC;EAC3C;EACA,OAAO5C,QAAe;AACxB,CAAC;AAED,MAAM+C,UAAU,GAAIC,IAAmB,IAAI;EACzC,QAAQA,IAAI,CAACP,IAAI;IACf,KAAK,cAAc;MACjB,OAAO;QACLO,IAAI,EAAE,MAAM;QACZ1C,IAAI,EAAE2C,QAAQ,CAACD,IAAI,CAAC1C,IAAI;OAChB;IACZ,KAAK,MAAM;MACT,OAAO;QACL0C,IAAI,EAAE;OACE;IACZ,KAAK,OAAO;MACV,OAAO;QACLA,IAAI,EAAE;OACE;EACd;AACF,CAAC;AAED,MAAMF,cAAc,GAClBF,OAAwB,IACwC;EAChE,MAAM5C,QAAQ,GAAiE,EAAE;EACjF,IAAIkD,KAAK,GAAgF,EAAE;EAC3F,IAAIC,SAAS,GAAkE,EAAE;EACjF,SAASC,YAAYA,CAAA;IACnB,IAAIF,KAAK,CAACzD,MAAM,KAAK,CAAC,EAAE;IACxBO,QAAQ,CAAC6C,IAAI,CAAC;MACZ,GAAGE,UAAU,CAACH,OAAO,CAACI,IAAI,CAAC;MAC3BK,OAAO,EAAEH;KACV,CAAC;IACFA,KAAK,GAAG,EAAE;EACZ;EACA,SAASI,cAAcA,CAAA;IACrB,IAAIH,SAAS,CAAC1D,MAAM,KAAK,CAAC,EAAE;IAC5BO,QAAQ,CAAC6C,IAAI,CAAC;MACZG,IAAI,EAAE,WAAW;MACjBK,OAAO,EAAE,IAAI;MACbE,UAAU,EAAEJ;KACb,CAAC;IACFA,SAAS,GAAG,EAAE;EAChB;EACA,KAAK,MAAMK,IAAI,IAAIZ,OAAO,CAACM,KAAK,EAAE;IAChC,IAAIM,IAAI,CAACf,IAAI,KAAK,UAAU,EAAE;MAC5BW,YAAY,EAAE;MACdD,SAAS,CAACN,IAAI,CAAC;QACbY,EAAE,EAAED,IAAI,CAACC,EAAE;QACXtD,IAAI,EAAE,UAAU;QAChBS,QAAQ,EAAE;UACRN,IAAI,EAAEkD,IAAI,CAAClD,IAAI;UACfoD,SAAS,EAAEC,IAAI,CAACC,SAAS,CAACJ,IAAI,CAACK,MAAM;;OAExC,CAAC;IACJ,CAAC,MAAM,IAAIL,IAAI,CAACf,IAAI,KAAK,kBAAkB,EAAE;MAC3CW,YAAY,EAAE;MACdE,cAAc,EAAE;MAChBtD,QAAQ,CAAC6C,IAAI,CAAC;QACZG,IAAI,EAAE,MAAM;QACZc,YAAY,EAAEN,IAAI,CAACO,UAAU;QAC7BV,OAAO,EAAEM,IAAI,CAACC,SAAS,CAACJ,IAAI,CAACb,KAAK;OACnC,CAAC;IACJ,CAAC,MAAM;MACLW,cAAc,EAAE;MAChBJ,KAAK,CAACL,IAAI,CAACmB,eAAe,CAACR,IAAI,CAAC,CAAC;IACnC;EACF;EACAJ,YAAY,EAAE;EACdE,cAAc,EAAE;EAChB,OAAOtD,QAAQ;AACjB,CAAC;AAED,MAAMgE,eAAe,GACnBR,IAAiE,IACO;EACxE,QAAQA,IAAI,CAACf,IAAI;IACf,KAAK,OAAO;MACV,OAAO;QACLtC,IAAI,EAAE,WAAW;QACjB8D,SAAS,EAAE;UACTC,GAAG,EAAEV,IAAI,CAACW,SAAS;UACnBC,MAAM,EAAEZ,IAAI,CAACa;;OAEhB;IACH,KAAK,UAAU;MACb,OAAO;QACLlE,IAAI,EAAE,WAAW;QACjB8D,SAAS,EAAE;UACTC,GAAG,EAAEV,IAAI,CAACU,GAAG;UACbE,MAAM,EAAEZ,IAAI,CAACa;;OAEhB;IACH,KAAK,MAAM;MACT,OAAO;QACLlE,IAAI,EAAE,MAAM;QACZmE,IAAI,EAAEd,IAAI,CAACH;OACZ;EACL;AACF,CAAC;AAED,MAAM7B,YAAY,GAAGA,CACnBF,QAAgD,EAChDQ,MAAc,EACdyC,cAGC,KAEDzH,GAAG,CAAC0H,IAAI,CAAClD,QAAQ,CAACmD,OAAO,CAAC,CAACzD,IAAI,CAC7BjE,MAAM,CAAC2H,QAAQ,CAAC,MACd,IAAI9C,gBAAO,CAAC;EACVC,MAAM,EAAE,mBAAmB;EAC3BC,MAAM;EACNvB,WAAW,EAAE;CACd,CAAC,CACH,EACDxD,MAAM,CAACqE,OAAO,CAAEuD,MAAM,IAAI;EACxB,IAAIJ,cAAc,EAAE;IAClB,OAAO7H,UAAU,CAACA,UAAU,CAACkI,KAAK,CAACC,iBAAiB,CAAC,CACnD;MACEpB,EAAE,EAAEnC,QAAQ,CAACmC,EAAE;MACfnD,IAAI,EAAEiE,cAAc,CAACjE,IAAI;MACzBuD,MAAM,EAAEc,MAAM,CAAC/B,OAAO,CAACS;KACxB,CACF,CAAC;EACJ;EAEA,MAAMyB,GAAG,GAAG,OAAOH,MAAM,CAAC/B,OAAO,CAACS,OAAO,KAAK,QAAQ,GACpD3G,UAAU,CAACA,UAAU,CAACqI,QAAQ,CAAC;IAC7B/B,IAAI,EAAEpG,MAAM,CAACiD,KAAK;IAClBwD,OAAO,EAAEsB,MAAM,CAAC/B,OAAO,CAACS;GACzB,CAAC,GACF3G,UAAU,CAACA,UAAU,CAACkI,KAAK;EAE7B,IAAID,MAAM,CAAC/B,OAAO,CAACW,UAAU,IAAIoB,MAAM,CAAC/B,OAAO,CAACW,UAAU,CAAC9D,MAAM,GAAG,CAAC,EAAE;IACrE,OAAOqF,GAAG,CAACD,iBAAiB,CAACF,MAAM,CAAC/B,OAAO,CAACW,UAAU,CAAC5D,GAAG,CAAEqF,QAAQ,KAAM;MACxEvB,EAAE,EAAEuB,QAAQ,CAACvB,EAAE;MACfnD,IAAI,EAAE0E,QAAQ,CAACpE,QAAQ,CAACN,IAAI;MAC5BuD,MAAM,EAAEmB,QAAQ,CAACpE,QAAQ,CAAC8C;KAC3B,CAAC,CAAC,CAAC;EACN;EAEA,OAAO3G,MAAM,CAACkI,OAAO,CAACH,GAAG,CAAC;AAC5B,CAAC,CAAC,CACH;AAEH,MAAMpC,iBAAiB,GAAIW,OAAe,IAAiE;EACzG,OAAO;IACLL,IAAI,EAAE,QAAQ;IACdK;GACD;AACH,CAAC;AAED,MAAMJ,QAAQ,GAAI3C,IAAY,IAAKA,IAAI,CAAC4E,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAACA,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC;AAE1F,MAAM/D,eAAe,GAAGA,CACtBJ,IAAU,EACVG,OAA6D,KACrD;EACR,IAAAiE,oCAAmB,EAACpE,IAAI,EAAE;IACxBzB,MAAM,EAAE,QAAQ;IAChB8F,SAAS,EAAE;MAAE9E,IAAI,EAAE;IAAM,CAAE;IAC3BY,OAAO,EAAE;MACPrB,KAAK,EAAEqB,OAAO,CAACrB,KAAK;MACpBwF,WAAW,EAAEnE,OAAO,CAACmE,WAAW;MAChCC,IAAI,EAAEpE,OAAO,CAACqE,KAAK;MACnBC,SAAS,EAAEtE,OAAO,CAACuE,UAAU;MAC7BC,aAAa,EAAE5I,GAAG,CAAC6I,MAAM,CAACzE,OAAO,CAAC0E,IAAI,CAAC,CAACC,MAAM,CAAC5I,SAAS,CAAC6I,aAAa,CAAC;MACvEC,gBAAgB,EAAE7E,OAAO,CAAC8E,iBAAiB;MAC3CC,eAAe,EAAE/E,OAAO,CAACgF,gBAAgB;MACzCC,IAAI,EAAEjF,OAAO,CAACiF;KACf;IACDC,MAAM,EAAE;MACNlF,OAAO,EAAE;QACPmF,cAAc,EAAEnF,OAAO,CAAChB,eAAe,EAAEC,IAAI;QAC7CmG,WAAW,EAAEpF,OAAO,CAACqF;;;GAG1B,CAAC;AACJ,CAAC;AAED,MAAMhF,oBAAoB,GAAGA,CAC3BR,IAAU,EACVO,QAAgD,KACxC;EACR,IAAA6D,oCAAmB,EAACpE,IAAI,EAAE;IACxBO,QAAQ,EAAE;MACRmC,EAAE,EAAEnC,QAAQ,CAACmC,EAAE;MACf5D,KAAK,EAAEyB,QAAQ,CAACzB,KAAK;MACrB2G,aAAa,EAAElF,QAAQ,CAACmD,OAAO,CAAC9E,GAAG,CAAEgF,MAAM,IAAKA,MAAM,CAAC8B,aAAa;KACrE;IACDC,KAAK,EAAE;MACLC,WAAW,EAAErF,QAAQ,CAACoF,KAAK,EAAEE,aAAa;MAC1CC,YAAY,EAAEvF,QAAQ,CAACoF,KAAK,EAAEI;KAC/B;IACDV,MAAM,EAAE;MACN9E,QAAQ,EAAE;QACRyF,iBAAiB,EAAEzF,QAAQ,CAAC0F,kBAAkB;QAC9CV,WAAW,EAAEhF,QAAQ,CAACiF;;;GAG3B,CAAC;AACJ,CAAC;AAED,MAAMtE,sBAAsB,GAAGA,CAC7BlB,IAAU,EACVO,QAAqB,KACnB;EACF,MAAMoF,KAAK,GAAGpF,QAAQ,CAAC4B,KAAK,CAAC+D,IAAI,CAAEzD,IAAI,IAAKA,IAAI,CAACf,IAAI,KAAK,OAAO,CAAC;EAClE,IAAIxF,SAAS,CAAC6I,aAAa,CAACY,KAAK,CAAC,EAAE;IAClC,IAAAvB,oCAAmB,EAACpE,IAAI,EAAE;MACxBO,QAAQ,EAAE;QACRmC,EAAE,EAAEiD,KAAK,CAACjD,EAAE;QACZ5D,KAAK,EAAE6G,KAAK,CAAC7G,KAAK;QAClB2G,aAAa,EAAEE,KAAK,CAACF;OACtB;MACDE,KAAK,EAAE;QACLC,WAAW,EAAED,KAAK,CAACC,WAAW;QAC9BE,YAAY,EAAEH,KAAK,CAACG;OACrB;MACDT,MAAM,EAAE;QACN9E,QAAQ,EAAE;UACRyF,iBAAiB,EAAEL,KAAK,CAACK,iBAAiB;UAC1CT,WAAW,EAAEI,KAAK,CAACJ;;;KAGxB,CAAC;EACJ;AACF,CAAC","ignoreList":[]}

package/dist/cjs/OpenAiEmbeddings.js

@@ -0,0 +1,95 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.layerDataLoader = exports.layer = exports.OpenAiEmbeddingsConfig = void 0;
+var _AiError = require("@effect/ai/AiError");
+var Embeddings = _interopRequireWildcard(require("@effect/ai/Embeddings"));
+var Context = _interopRequireWildcard(require("effect/Context"));
+var Effect = _interopRequireWildcard(require("effect/Effect"));
+var Layer = _interopRequireWildcard(require("effect/Layer"));
+var _OpenAiClient = require("./OpenAiClient.js");
+function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function (e) { return e ? t : r; })(e); }
+function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != typeof e && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && {}.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
+/**
+ * @since 1.0.0
+ */
+
+/**
+ * @since 1.0.0
+ * @category tags
+ */
+class OpenAiEmbeddingsConfig extends /*#__PURE__*/Context.Tag("@effect/ai-openai/OpenAiEmbeddings/Config")() {
+  /**
+   * @since 1.0.0
+   */
+  static getOrUndefined = /*#__PURE__*/Effect.map( /*#__PURE__*/Effect.context(), context => context.unsafeMap.get(OpenAiEmbeddingsConfig.key));
+}
+exports.OpenAiEmbeddingsConfig = OpenAiEmbeddingsConfig;
+const makeRequest = (client, input, parentConfig, options) => Effect.context().pipe(Effect.flatMap(context => {
+  const localConfig = context.unsafeMap.get(OpenAiEmbeddingsConfig.key);
+  return client.client.createEmbedding({
+    input,
+    model: options.model,
+    ...parentConfig,
+    ...localConfig
+  });
+}), Effect.map(response => response.data.map(({
+  embedding,
+  index
+}) => ({
+  embeddings: embedding,
+  index
+}))), Effect.mapError(cause => {
+  const common = {
+    module: "OpenAiEmbeddings",
+    method: "embed",
+    cause
+  };
+  if (cause._tag === "ParseError") {
+    return new _AiError.AiError({
+      description: "Malformed input detected in request",
+      ...common
+    });
+  }
+  return new _AiError.AiError({
+    description: "An error occurred with the OpenAI API",
+    ...common
+  });
+}));
+const make = /*#__PURE__*/Effect.fnUntraced(function* (options) {
+  const client = yield* _OpenAiClient.OpenAiClient;
+  const parentConfig = yield* OpenAiEmbeddingsConfig.getOrUndefined;
+  return yield* Embeddings.make({
+    cache: options.cache,
+    maxBatchSize: options.maxBatchSize ?? 2048,
+    embedMany(input) {
+      return makeRequest(client, input, parentConfig, options);
+    }
+  });
+});
+const makeDataLoader = /*#__PURE__*/Effect.fnUntraced(function* (options) {
+  const client = yield* _OpenAiClient.OpenAiClient;
+  const parentConfig = yield* OpenAiEmbeddingsConfig.getOrUndefined;
+  return yield* Embeddings.makeDataLoader({
+    window: options.window,
+    maxBatchSize: options.maxBatchSize ?? 2048,
+    embedMany(input) {
+      return makeRequest(client, input, parentConfig, options);
+    }
+  });
+});
+/**
+ * @since 1.0.0
+ * @category layers
+ */
+const layer = options => Layer.effect(Embeddings.Embeddings, make(options));
+/**
+ * @since 1.0.0
+ * @category layers
+ */
+exports.layer = layer;
+const layerDataLoader = options => Layer.scoped(Embeddings.Embeddings, makeDataLoader(options));
+exports.layerDataLoader = layerDataLoader;
+//# sourceMappingURL=OpenAiEmbeddings.js.map

package/dist/cjs/OpenAiEmbeddings.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"OpenAiEmbeddings.js","names":["_AiError","require","Embeddings","_interopRequireWildcard","Context","Effect","Layer","_OpenAiClient","_getRequireWildcardCache","e","WeakMap","r","t","__esModule","default","has","get","n","__proto__","a","Object","defineProperty","getOwnPropertyDescriptor","u","hasOwnProperty","call","i","set","OpenAiEmbeddingsConfig","Tag","getOrUndefined","map","context","unsafeMap","key","makeRequest","client","input","parentConfig","options","pipe","flatMap","localConfig","createEmbedding","model","response","data","embedding","index","embeddings","mapError","cause","common","module","method","_tag","AiError","description","make","fnUntraced","OpenAiClient","cache","maxBatchSize","embedMany","makeDataLoader","window","layer","effect","exports","layerDataLoader","scoped"],"sources":["../../src/OpenAiEmbeddings.ts"],"sourcesContent":[null],"mappings":";;;;;;AAGA,IAAAA,QAAA,GAAAC,OAAA;AACA,IAAAC,UAAA,GAAAC,uBAAA,CAAAF,OAAA;AACA,IAAAG,OAAA,GAAAD,uBAAA,CAAAF,OAAA;AAEA,IAAAI,MAAA,GAAAF,uBAAA,CAAAF,OAAA;AACA,IAAAK,KAAA,GAAAH,uBAAA,CAAAF,OAAA;AAGA,IAAAM,aAAA,GAAAN,OAAA;AAAgD,SAAAO,yBAAAC,CAAA,6BAAAC,OAAA,mBAAAC,CAAA,OAAAD,OAAA,IAAAE,CAAA,OAAAF,OAAA,YAAAF,wBAAA,YAAAA,CAAAC,CAAA,WAAAA,CAAA,GAAAG,CAAA,GAAAD,CAAA,KAAAF,CAAA;AAAA,SAAAN,wBAAAM,CAAA,EAAAE,CAAA,SAAAA,CAAA,IAAAF,CAAA,IAAAA,CAAA,CAAAI,UAAA,SAAAJ,CAAA,eAAAA,CAAA,uBAAAA,CAAA,yBAAAA,CAAA,WAAAK,OAAA,EAAAL,CAAA,QAAAG,CAAA,GAAAJ,wBAAA,CAAAG,CAAA,OAAAC,CAAA,IAAAA,CAAA,CAAAG,GAAA,CAAAN,CAAA,UAAAG,CAAA,CAAAI,GAAA,CAAAP,CAAA,OAAAQ,CAAA,KAAAC,SAAA,UAAAC,CAAA,GAAAC,MAAA,CAAAC,cAAA,IAAAD,MAAA,CAAAE,wBAAA,WAAAC,CAAA,IAAAd,CAAA,oBAAAc,CAAA,OAAAC,cAAA,CAAAC,IAAA,CAAAhB,CAAA,EAAAc,CAAA,SAAAG,CAAA,GAAAP,CAAA,GAAAC,MAAA,CAAAE,wBAAA,CAAAb,CAAA,EAAAc,CAAA,UAAAG,CAAA,KAAAA,CAAA,CAAAV,GAAA,IAAAU,CAAA,CAAAC,GAAA,IAAAP,MAAA,CAAAC,cAAA,CAAAJ,CAAA,EAAAM,CAAA,EAAAG,CAAA,IAAAT,CAAA,CAAAM,CAAA,IAAAd,CAAA,CAAAc,CAAA,YAAAN,CAAA,CAAAH,OAAA,GAAAL,CAAA,EAAAG,CAAA,IAAAA,CAAA,CAAAe,GAAA,CAAAlB,CAAA,EAAAQ,CAAA,GAAAA,CAAA;AAXhD;;;;AAmBA;;;;AAIM,MAAOW,sBAAuB,sBAAQxB,OAAO,CAACyB,GAAG,CAAC,2CAA2C,CAAC,EAUjG;EACD;;;EAGA,OAAgBC,cAAc,gBAAqEzB,MAAM,CAAC0B,GAAG,eAC3G1B,MAAM,CAAC2B,OAAO,EAAS,EACtBA,OAAO,IAAKA,OAAO,CAACC,SAAS,CAACjB,GAAG,CAACY,sBAAsB,CAACM,GAAG,CAAC,CAC/D;;;AAGH,MAAMC,WAAW,GAAGA,CAClBC,MAA4B,EAC5BC,KAA4B,EAC5BC,YAA+D,EAC/DC,OAOC,KAEDlC,MAAM,CAAC2B,OAAO,EAAS,CAACQ,IAAI,CAC1BnC,MAAM,CAACoC,OAAO,CAAET,OAAO,IAAI;EACzB,MAAMU,WAAW,GAAGV,OAAO,CAACC,SAAS,CAACjB,GAAG,CAACY,sBAAsB,CAACM,GAAG,CAAC;EACrE,OAAOE,MAAM,CAACA,MAAM,CAACO,eAAe,CAAC;IACnCN,KAAK;IACLO,KAAK,EAAEL,OAAO,CAACK,KAAK;IACpB,GAAGN,YAAY;IACf,GAAGI;GACJ,CAAC;AACJ,CAAC,CAAC,EACFrC,MAAM,CAAC0B,GAAG,CAAEc,QAAQ,IAClBA,QAAQ,CAACC,IAAI,CAACf,GAAG,CAAC,CAAC;EAAEgB,SAAS;EAAEC;AAAK,CAAE,MAAM;EAC3CC,UAAU,EAAEF,SAA0B;EACtCC;CACD,CAAC,CAAC,CACJ,EACD3C,MAAM,CAAC6C,QAAQ,CAAEC,KAAK,IAAI;EACxB,MAAMC,MAAM,GAAG;IACbC,MAAM,EAAE,kBAAkB;IAC1BC,MAAM,EAAE,OAAO;IACfH;GACD;EACD,IAAIA,KAAK,CAACI,IAAI,KAAK,YAAY,EAAE;IAC/B,OAAO,IAAIC,gBAAO,CAAC;MACjBC,WAAW,EAAE,qCAAqC;MAClD,GAAGL;KACJ,CAAC;EACJ;EACA,OAAO,IAAII,gBAAO,CAAC;IACjBC,WAAW,EAAE,uCAAuC;IACpD,GAAGL;GACJ,CAAC;AACJ,CAAC,CAAC,CACH;AAEH,MAAMM,IAAI,gBAAGrD,MAAM,CAACsD,UAAU,CAAC,WAAUpB,OAOxC;EACC,MAAMH,MAAM,GAAG,OAAOwB,0BAAY;EAClC,MAAMtB,YAAY,GAAG,OAAOV,sBAAsB,CAACE,cAAc;EACjE,OAAO,OAAO5B,UAAU,CAACwD,IAAI,CAAC;IAC5BG,KAAK,EAAEtB,OAAO,CAACsB,KAAK;IACpBC,YAAY,EAAEvB,OAAO,CAACuB,YAAY,IAAI,IAAI;IAC1CC,SAASA,CAAC1B,KAAK;MACb,OAAOF,WAAW,CAACC,MAAM,EAAEC,KAAK,EAAEC,YAAY,EAAEC,OAAO,CAAC;IAC1D;GACD,CAAC;AACJ,CAAC,CAAC;AAEF,MAAMyB,cAAc,gBAAG3D,MAAM,CAACsD,UAAU,CAAC,WAAUpB,OAIlD;EACC,MAAMH,MAAM,GAAG,OAAOwB,0BAAY;EAClC,MAAMtB,YAAY,GAAG,OAAOV,sBAAsB,CAACE,cAAc;EACjE,OAAO,OAAO5B,UAAU,CAAC8D,cAAc,CAAC;IACtCC,MAAM,EAAE1B,OAAO,CAAC0B,MAAM;IACtBH,YAAY,EAAEvB,OAAO,CAACuB,YAAY,IAAI,IAAI;IAC1CC,SAASA,CAAC1B,KAAK;MACb,OAAOF,WAAW,CAACC,MAAM,EAAEC,KAAK,EAAEC,YAAY,EAAEC,OAAO,CAAC;IAC1D;GACD,CAAC;AACJ,CAAC,CAAC;AAEF;;;;AAIO,MAAM2B,KAAK,GAAI3B,OAOrB,IAA8DjC,KAAK,CAAC6D,MAAM,CAACjE,UAAU,CAACA,UAAU,EAAEwD,IAAI,CAACnB,OAAO,CAAC,CAAC;AAEjH;;;;AAAA6B,OAAA,CAAAF,KAAA,GAAAA,KAAA;AAIO,MAAMG,eAAe,GAAI9B,OAI/B,IACCjC,KAAK,CAACgE,MAAM,CAACpE,UAAU,CAACA,UAAU,EAAE8D,cAAc,CAACzB,OAAO,CAAC,CAAC;AAAA6B,OAAA,CAAAC,eAAA,GAAAA,eAAA","ignoreList":[]}

package/dist/cjs/OpenAiTelemetry.js

@@ -0,0 +1,39 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.addGenAIAnnotations = void 0;
+var AiTelemetry = _interopRequireWildcard(require("@effect/ai/AiTelemetry"));
+var _Function = require("effect/Function");
+var Predicate = _interopRequireWildcard(require("effect/Predicate"));
+var String = _interopRequireWildcard(require("effect/String"));
+function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function (e) { return e ? t : r; })(e); }
+function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != typeof e && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && {}.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
+/**
+ * @since 1.0.0
+ */
+
+const addOpenAiRequestAttributes = /*#__PURE__*/AiTelemetry.addSpanAttributes("gen_ai.openai.request", String.camelToSnake);
+const addOpenAiResponseAttributes = /*#__PURE__*/AiTelemetry.addSpanAttributes("gen_ai.openai.response", String.camelToSnake);
+/**
+ * Applies the specified OpenAi GenAI telemetry attributes to the provided
+ * `Span`.
+ *
+ * **NOTE**: This method will mutate the `Span` **in-place**.
+ *
+ * @since 1.0.0
+ * @since utilities
+ */
+const addGenAIAnnotations = exports.addGenAIAnnotations = /*#__PURE__*/(0, _Function.dual)(2, (span, options) => {
+  AiTelemetry.addGenAIAnnotations(span, options);
+  if (Predicate.isNotNullable(options.openai)) {
+    if (Predicate.isNotNullable(options.openai.request)) {
+      addOpenAiRequestAttributes(span, options.openai.request);
+    }
+    if (Predicate.isNotNullable(options.openai.response)) {
+      addOpenAiResponseAttributes(span, options.openai.response);
+    }
+  }
+});
+//# sourceMappingURL=OpenAiTelemetry.js.map

package/dist/cjs/OpenAiTelemetry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"OpenAiTelemetry.js","names":["AiTelemetry","_interopRequireWildcard","require","_Function","Predicate","String","_getRequireWildcardCache","e","WeakMap","r","t","__esModule","default","has","get","n","__proto__","a","Object","defineProperty","getOwnPropertyDescriptor","u","hasOwnProperty","call","i","set","addOpenAiRequestAttributes","addSpanAttributes","camelToSnake","addOpenAiResponseAttributes","addGenAIAnnotations","exports","dual","span","options","isNotNullable","openai","request","response"],"sources":["../../src/OpenAiTelemetry.ts"],"sourcesContent":[null],"mappings":";;;;;;AAGA,IAAAA,WAAA,GAAAC,uBAAA,CAAAC,OAAA;AACA,IAAAC,SAAA,GAAAD,OAAA;AACA,IAAAE,SAAA,GAAAH,uBAAA,CAAAC,OAAA;AACA,IAAAG,MAAA,GAAAJ,uBAAA,CAAAC,OAAA;AAAuC,SAAAI,yBAAAC,CAAA,6BAAAC,OAAA,mBAAAC,CAAA,OAAAD,OAAA,IAAAE,CAAA,OAAAF,OAAA,YAAAF,wBAAA,YAAAA,CAAAC,CAAA,WAAAA,CAAA,GAAAG,CAAA,GAAAD,CAAA,KAAAF,CAAA;AAAA,SAAAN,wBAAAM,CAAA,EAAAE,CAAA,SAAAA,CAAA,IAAAF,CAAA,IAAAA,CAAA,CAAAI,UAAA,SAAAJ,CAAA,eAAAA,CAAA,uBAAAA,CAAA,yBAAAA,CAAA,WAAAK,OAAA,EAAAL,CAAA,QAAAG,CAAA,GAAAJ,wBAAA,CAAAG,CAAA,OAAAC,CAAA,IAAAA,CAAA,CAAAG,GAAA,CAAAN,CAAA,UAAAG,CAAA,CAAAI,GAAA,CAAAP,CAAA,OAAAQ,CAAA,KAAAC,SAAA,UAAAC,CAAA,GAAAC,MAAA,CAAAC,cAAA,IAAAD,MAAA,CAAAE,wBAAA,WAAAC,CAAA,IAAAd,CAAA,oBAAAc,CAAA,OAAAC,cAAA,CAAAC,IAAA,CAAAhB,CAAA,EAAAc,CAAA,SAAAG,CAAA,GAAAP,CAAA,GAAAC,MAAA,CAAAE,wBAAA,CAAAb,CAAA,EAAAc,CAAA,UAAAG,CAAA,KAAAA,CAAA,CAAAV,GAAA,IAAAU,CAAA,CAAAC,GAAA,IAAAP,MAAA,CAAAC,cAAA,CAAAJ,CAAA,EAAAM,CAAA,EAAAG,CAAA,IAAAT,CAAA,CAAAM,CAAA,IAAAd,CAAA,CAAAc,CAAA,YAAAN,CAAA,CAAAH,OAAA,GAAAL,CAAA,EAAAG,CAAA,IAAAA,CAAA,CAAAe,GAAA,CAAAlB,CAAA,EAAAQ,CAAA,GAAAA,CAAA;AANvC;;;;AA+GA,MAAMW,0BAA0B,gBAAG1B,WAAW,CAAC2B,iBAAiB,CAAC,uBAAuB,EAAEtB,MAAM,CAACuB,YAAY,CAE5G;AACD,MAAMC,2BAA2B,gBAAG7B,WAAW,CAAC2B,iBAAiB,CAAC,wBAAwB,EAAEtB,MAAM,CAACuB,YAAY,CAE9G;AAED;;;;;;;;;AASO,MAAME,mBAAmB,GAAAC,OAAA,CAAAD,mBAAA,gBAAG,IAAAE,cAAI,EAqBrC,CAAC,EAAE,CAACC,IAAI,EAAEC,OAAO,KAAI;EACrBlC,WAAW,CAAC8B,mBAAmB,CAACG,IAAI,EAAEC,OAAO,CAAC;EAC9C,IAAI9B,SAAS,CAAC+B,aAAa,CAACD,OAAO,CAACE,MAAM,CAAC,EAAE;IAC3C,IAAIhC,SAAS,CAAC+B,aAAa,CAACD,OAAO,CAACE,MAAM,CAACC,OAAO,CAAC,EAAE;MACnDX,0BAA0B,CAACO,IAAI,EAAEC,OAAO,CAACE,MAAM,CAACC,OAAO,CAAC;IAC1D;IACA,IAAIjC,SAAS,CAAC+B,aAAa,CAACD,OAAO,CAACE,MAAM,CAACE,QAAQ,CAAC,EAAE;MACpDT,2BAA2B,CAACI,IAAI,EAAEC,OAAO,CAACE,MAAM,CAACE,QAAQ,CAAC;IAC5D;EACF;AACF,CAAC,CAAC","ignoreList":[]}

package/dist/cjs/index.js

@@ -3,7 +3,7 @@
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
-exports.OpenAiTokenizer = exports.OpenAiConfig = exports.OpenAiCompletions = exports.OpenAiClient = exports.Generated = void 0;
+exports.OpenAiTokenizer = exports.OpenAiTelemetry = exports.OpenAiEmbeddings = exports.OpenAiConfig = exports.OpenAiCompletions = exports.OpenAiClient = exports.Generated = void 0;
 var _Generated = _interopRequireWildcard(require("./Generated.js"));
 exports.Generated = _Generated;
 var _OpenAiClient = _interopRequireWildcard(require("./OpenAiClient.js"));
@@ -12,6 +12,10 @@ var _OpenAiCompletions = _interopRequireWildcard(require("./OpenAiCompletions.js
 exports.OpenAiCompletions = _OpenAiCompletions;
 var _OpenAiConfig = _interopRequireWildcard(require("./OpenAiConfig.js"));
 exports.OpenAiConfig = _OpenAiConfig;
+var _OpenAiEmbeddings = _interopRequireWildcard(require("./OpenAiEmbeddings.js"));
+exports.OpenAiEmbeddings = _OpenAiEmbeddings;
+var _OpenAiTelemetry = _interopRequireWildcard(require("./OpenAiTelemetry.js"));
+exports.OpenAiTelemetry = _OpenAiTelemetry;
 var _OpenAiTokenizer = _interopRequireWildcard(require("./OpenAiTokenizer.js"));
 exports.OpenAiTokenizer = _OpenAiTokenizer;
 function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function (e) { return e ? t : r; })(e); }