@usagetap/sdk 0.4.0 → 0.6.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (39) hide show
  1. package/README.md +26 -10
  2. package/dist/adapters/openai.d.cts +135 -2
  3. package/dist/adapters/openai.d.ts +135 -2
  4. package/dist/adapters/{openai.js → openai.mjs} +2 -2
  5. package/dist/adapters/openai.mjs.map +1 -0
  6. package/dist/adapters/openrouter.d.cts +2 -1
  7. package/dist/adapters/openrouter.d.ts +2 -1
  8. package/dist/adapters/{openrouter.js → openrouter.mjs} +2 -2
  9. package/dist/adapters/openrouter.mjs.map +1 -0
  10. package/dist/{openai-Ch7hN2vD.d.cts → client-DAY6cR0C.d.cts} +1 -134
  11. package/dist/{openai-Ch7hN2vD.d.ts → client-DAY6cR0C.d.ts} +1 -134
  12. package/dist/express/index.cjs +691 -0
  13. package/dist/express/index.cjs.map +1 -0
  14. package/dist/express/index.d.cts +93 -0
  15. package/dist/express/index.d.ts +93 -0
  16. package/dist/express/index.mjs +688 -0
  17. package/dist/express/index.mjs.map +1 -0
  18. package/dist/index.cjs +1 -778
  19. package/dist/index.cjs.map +1 -1
  20. package/dist/index.d.cts +3 -94
  21. package/dist/index.d.ts +3 -94
  22. package/dist/{index.js → index.mjs} +4 -773
  23. package/dist/index.mjs.map +1 -0
  24. package/dist/openai/index.cjs +775 -0
  25. package/dist/openai/index.cjs.map +1 -0
  26. package/dist/openai/index.d.cts +4 -0
  27. package/dist/openai/index.d.ts +4 -0
  28. package/dist/openai/index.mjs +768 -0
  29. package/dist/openai/index.mjs.map +1 -0
  30. package/dist/react/index.cjs.map +1 -1
  31. package/dist/react/index.d.cts +1 -1
  32. package/dist/react/index.d.ts +1 -1
  33. package/dist/react/{index.js → index.mjs} +2 -2
  34. package/dist/react/index.mjs.map +1 -0
  35. package/package.json +23 -1
  36. package/dist/adapters/openai.js.map +0 -1
  37. package/dist/adapters/openrouter.js.map +0 -1
  38. package/dist/index.js.map +0 -1
  39. package/dist/react/index.js.map +0 -1
package/dist/index.cjs CHANGED
@@ -127,7 +127,7 @@ var IDEMPOTENCY_HEADER = "idempotency-key";
127
127
  var SDK_HEADER = "x-usage-sdk";
128
128
  var USER_AGENT = "UsageTapClient";
129
129
  var CANONICAL_MEDIA_TYPE = "application/vnd.usagetap.v1+json";
130
- var SDK_VERSION = "0.4.0" ;
130
+ var SDK_VERSION = "0.6.0" ;
131
131
  var HAS_WINDOW = typeof globalThis !== "undefined" && typeof globalThis.window !== "undefined";
132
132
  var UsageTapClient = class {
133
133
  apiKey;
@@ -639,742 +639,6 @@ function wrapEndCallError(error, correlationId) {
639
639
  );
640
640
  }
641
641
 
642
- // src/adapters/openai.ts
643
- function createOpenAIAdapter(init) {
644
- const { client, usageTap } = init;
645
- return {
646
- async invoke(params) {
647
- const result = await usageTap.withUsage(
648
- params.begin,
649
- async (ctx) => {
650
- const response = await params.call(client, {
651
- hints: ctx.begin.data.vendorHints,
652
- begin: ctx.begin
653
- });
654
- tryInferUsage(response, ctx.begin.data.vendorHints, params.extractUsage, ctx);
655
- return {
656
- data: response,
657
- begin: ctx.begin
658
- };
659
- },
660
- params.withUsageOptions
661
- );
662
- return result;
663
- },
664
- async invokeStream(params) {
665
- const result = await usageTap.withUsage(
666
- params.begin,
667
- async (ctx) => {
668
- const { stream, onComplete } = await params.call(client, {
669
- hints: ctx.begin.data.vendorHints,
670
- begin: ctx.begin
671
- });
672
- const wrapped = wrapStreamForUsageTap(stream, async () => {
673
- if (!onComplete) return;
674
- try {
675
- const maybeUsage = await onComplete();
676
- if (maybeUsage) {
677
- ctx.setUsage(maybeUsage);
678
- }
679
- } catch (error) {
680
- ctx.setError({
681
- code: "USAGE_FINALIZE_ERROR",
682
- message: error instanceof Error ? error.message : String(error)
683
- });
684
- throw error;
685
- }
686
- }, ctx);
687
- const finalize = async () => {
688
- await wrapped.__usageTapFinalize?.();
689
- };
690
- return {
691
- stream: wrapped,
692
- begin: ctx.begin,
693
- finalize
694
- };
695
- },
696
- params.withUsageOptions
697
- );
698
- return result;
699
- }
700
- };
701
- }
702
- function toNextResponse(stream, options = {}) {
703
- const mode = options.mode ?? "text";
704
- const headers = new Headers(options.headers ?? {});
705
- if (mode === "sse") {
706
- headers.set("content-type", "text/event-stream; charset=utf-8");
707
- headers.set("cache-control", "no-cache, no-transform");
708
- headers.set("connection", "keep-alive");
709
- headers.set("x-accel-buffering", "no");
710
- } else {
711
- headers.set("content-type", options.contentType ?? "text/plain; charset=utf-8");
712
- }
713
- const encoder = new TextEncoder();
714
- let iterator;
715
- const body = new ReadableStream({
716
- async start(controller) {
717
- try {
718
- const getIterator = stream[Symbol.asyncIterator];
719
- if (typeof getIterator !== "function") {
720
- controller.close();
721
- return;
722
- }
723
- iterator = getIterator.call(stream);
724
- while (true) {
725
- const result = await iterator.next();
726
- if (result.done) {
727
- break;
728
- }
729
- const text = chunkToText(result.value);
730
- if (!text) {
731
- continue;
732
- }
733
- if (mode === "sse") {
734
- controller.enqueue(encoder.encode(formatSsePayload(text, options.sse)));
735
- } else {
736
- controller.enqueue(encoder.encode(text));
737
- }
738
- }
739
- controller.close();
740
- } catch (error) {
741
- controller.error(error);
742
- } finally {
743
- await stream.__usageTapFinalize?.();
744
- }
745
- },
746
- async cancel() {
747
- if (!iterator) {
748
- const getIterator = stream[Symbol.asyncIterator];
749
- if (typeof getIterator === "function") {
750
- iterator = getIterator.call(stream);
751
- }
752
- }
753
- if (iterator && typeof iterator.return === "function") {
754
- await iterator.return();
755
- }
756
- await stream.__usageTapFinalize?.();
757
- }
758
- });
759
- return new Response(body, { headers });
760
- }
761
- async function pipeToResponse(stream, res, options = {}) {
762
- const mode = options.mode ?? "text";
763
- if (mode === "sse") {
764
- setHeaderIfPossible(res, "Content-Type", "text/event-stream; charset=utf-8");
765
- setHeaderIfPossible(res, "Cache-Control", "no-cache, no-transform");
766
- setHeaderIfPossible(res, "Connection", "keep-alive");
767
- setHeaderIfPossible(res, "X-Accel-Buffering", "no");
768
- } else {
769
- setHeaderIfPossible(res, "Content-Type", options.contentType ?? "text/plain; charset=utf-8");
770
- }
771
- const encoder = new TextEncoder();
772
- const iterator = stream[Symbol.asyncIterator]();
773
- try {
774
- while (true) {
775
- const result = await iterator.next();
776
- if (result.done) {
777
- break;
778
- }
779
- const text = chunkToText(result.value);
780
- if (!text) {
781
- continue;
782
- }
783
- const payload = mode === "sse" ? formatSsePayload(text, options.sse) : text;
784
- res.write(Buffer.from(encoder.encode(payload)));
785
- res.flush?.();
786
- }
787
- } finally {
788
- res.end();
789
- await stream.__usageTapFinalize?.();
790
- }
791
- }
792
- var USAGETAP_CORRELATION_HEADER = "x-usage-correlation-id";
793
- function wrapOpenAI(client, usageTap, options = {}) {
794
- if (!client) {
795
- throw new UsageTapError("USAGETAP_BAD_REQUEST", "wrapOpenAI requires an OpenAI client instance");
796
- }
797
- const defaultContext = options.defaultContext;
798
- const applyVendorHints = options.applyVendorHints !== false;
799
- const proxiedChat = client.chat ? createChatProxy(client.chat, usageTap, defaultContext, applyVendorHints) : void 0;
800
- const proxiedResponses = typeof client.responses !== "undefined" ? createResponsesProxy(client.responses, usageTap, defaultContext, applyVendorHints) : void 0;
801
- const handler = {
802
- get(target, prop, receiver) {
803
- if (prop === "chat" && proxiedChat) {
804
- return proxiedChat;
805
- }
806
- if (prop === "responses" && typeof target.responses !== "undefined") {
807
- return proxiedResponses ?? Reflect.get(target, prop, receiver);
808
- }
809
- if (prop === "toNextResponse") {
810
- return toNextResponse;
811
- }
812
- if (prop === "pipeToResponse") {
813
- return pipeToResponse;
814
- }
815
- if (prop === "unwrap") {
816
- return () => target;
817
- }
818
- return Reflect.get(target, prop, receiver);
819
- }
820
- };
821
- return new Proxy(client, handler);
822
- }
823
- function streamOpenAIRoute(usageTap, openai, options) {
824
- if (!options?.getRequest) {
825
- throw new UsageTapError("USAGETAP_BAD_REQUEST", "streamOpenAIRoute requires a getRequest function");
826
- }
827
- const wrapConfig = options.wrapOptions || options.defaultContext ? {
828
- ...options.wrapOptions ?? {},
829
- defaultContext: options.defaultContext ?? options.wrapOptions?.defaultContext
830
- } : void 0;
831
- const wrappedClient = wrapConfig ? wrapOpenAI(openai, usageTap, wrapConfig) : wrapOpenAI(openai, usageTap);
832
- return async (req) => {
833
- const requestConfig = await options.getRequest(req);
834
- const mergedParams = {
835
- ...requestConfig.params,
836
- stream: true
837
- };
838
- const callOptions = {};
839
- if (requestConfig.usageTap) {
840
- callOptions.usageTap = requestConfig.usageTap;
841
- }
842
- if (requestConfig.withUsage) {
843
- callOptions.withUsage = requestConfig.withUsage;
844
- }
845
- const stream = await wrappedClient.chat.completions.create(
846
- mergedParams,
847
- Object.keys(callOptions).length ? callOptions : void 0
848
- );
849
- const baseResponse = toNextResponse(stream, {
850
- mode: options.stream?.mode ?? "sse",
851
- headers: options.stream?.headers
852
- });
853
- const init = options.stream?.responseInit;
854
- if (!init) {
855
- return baseResponse;
856
- }
857
- const mergedHeaders = new Headers(baseResponse.headers);
858
- if (init.headers) {
859
- const extra = normalizeHeaders(init.headers);
860
- for (const [key, value] of Object.entries(extra)) {
861
- mergedHeaders.set(key, value);
862
- }
863
- }
864
- return new Response(baseResponse.body, {
865
- status: init.status ?? baseResponse.status,
866
- statusText: init.statusText ?? baseResponse.statusText,
867
- headers: mergedHeaders
868
- });
869
- };
870
- }
871
- function createChatProxy(resource, usageTap, defaultContext, applyVendorHints) {
872
- const completions = createChatCompletionsProxy(
873
- resource.completions,
874
- usageTap,
875
- defaultContext,
876
- applyVendorHints
877
- );
878
- const handler = {
879
- get(target, prop, receiver) {
880
- if (prop === "completions") {
881
- return completions;
882
- }
883
- return Reflect.get(target, prop, receiver);
884
- }
885
- };
886
- return new Proxy(resource, handler);
887
- }
888
- function createResponsesProxy(resource, usageTap, defaultContext, applyVendorHints) {
889
- if (!resource || typeof resource !== "object") {
890
- return void 0;
891
- }
892
- if (!("create" in resource) || typeof resource.create !== "function") {
893
- return resource;
894
- }
895
- const originalCreate = resource.create.bind(resource);
896
- const wrappedCreate = (params, options) => {
897
- const { requestOptions, usageContext, withUsage: withUsage2 } = splitUsageOptions(options);
898
- const beginRequest = resolveBeginRequest(defaultContext, usageContext);
899
- const wantsStream = isStreamingRequest(params);
900
- return usageTap.withUsage(beginRequest, (ctx) => {
901
- const finalParams = applyVendorHints ? applyResponsesVendorHints(params, ctx.begin.data.vendorHints) : params;
902
- const request = attachCorrelationHeader(requestOptions, ctx.begin.correlationId);
903
- if (wantsStream) {
904
- const apiPromise2 = originalCreate(finalParams, request);
905
- const wrappedPromise2 = transformApiPromise(apiPromise2, (rawStream) => {
906
- ensureAsyncIterable(rawStream, "responses.create");
907
- const wrappedStream = wrapStreamForUsageTap(rawStream, async () => {
908
- const usage = await extractUsageFromStream(rawStream, ctx.begin.data.vendorHints);
909
- if (usage) {
910
- ctx.setUsage(usage);
911
- }
912
- }, ctx);
913
- return wrappedStream;
914
- });
915
- return wrappedPromise2;
916
- }
917
- const apiPromise = originalCreate(finalParams, request);
918
- const wrappedPromise = transformApiPromise(apiPromise, (response) => {
919
- tryInferUsage(response, ctx.begin.data.vendorHints, void 0, ctx);
920
- return response;
921
- });
922
- return wrappedPromise;
923
- }, withUsage2);
924
- };
925
- const handler = {
926
- get(target, prop, receiver) {
927
- if (prop === "create") {
928
- return wrappedCreate;
929
- }
930
- return Reflect.get(target, prop, receiver);
931
- }
932
- };
933
- return new Proxy(resource, handler);
934
- }
935
- function createChatCompletionsProxy(resource, usageTap, defaultContext, applyVendorHints) {
936
- const originalCreate = resource.create.bind(resource);
937
- const streamCandidate = resource.stream;
938
- const originalStream = typeof streamCandidate === "function" ? streamCandidate.bind(resource) : void 0;
939
- const wrappedCreate = (params, options) => {
940
- const { requestOptions, usageContext, withUsage: withUsage2 } = splitUsageOptions(options);
941
- const beginRequest = resolveBeginRequest(defaultContext, usageContext);
942
- const wantsStream = isStreamingRequest(params);
943
- return usageTap.withUsage(beginRequest, (ctx) => {
944
- const finalParams = applyVendorHints ? applyChatVendorHints(params, ctx.begin.data.vendorHints) : params;
945
- const request = attachCorrelationHeader(requestOptions, ctx.begin.correlationId);
946
- if (wantsStream) {
947
- const apiPromise2 = originalCreate(finalParams, request);
948
- const wrappedPromise2 = transformApiPromise(apiPromise2, (rawStream) => {
949
- ensureAsyncIterable(rawStream, "chat.completions.create");
950
- const wrappedStream2 = wrapStreamForUsageTap(rawStream, async () => {
951
- const usage = await extractUsageFromStream(rawStream, ctx.begin.data.vendorHints);
952
- if (usage) {
953
- ctx.setUsage(usage);
954
- }
955
- }, ctx);
956
- return wrappedStream2;
957
- });
958
- return wrappedPromise2;
959
- }
960
- const apiPromise = originalCreate(finalParams, request);
961
- const wrappedPromise = transformApiPromise(apiPromise, (response) => {
962
- tryInferUsage(response, ctx.begin.data.vendorHints, void 0, ctx);
963
- return response;
964
- });
965
- return wrappedPromise;
966
- }, withUsage2);
967
- };
968
- const wrappedStream = originalStream ? (params, options) => {
969
- const { requestOptions, usageContext, withUsage: withUsage2 } = splitUsageOptions(options);
970
- const beginRequest = resolveBeginRequest(defaultContext, usageContext);
971
- return usageTap.withUsage(beginRequest, (ctx) => {
972
- const finalParams = applyVendorHints ? applyChatVendorHints(params, ctx.begin.data.vendorHints) : params;
973
- const request = attachCorrelationHeader(requestOptions, ctx.begin.correlationId);
974
- const apiPromise = originalStream(finalParams, request);
975
- const wrappedPromise = transformApiPromise(apiPromise, (rawStream) => {
976
- ensureAsyncIterable(rawStream, "chat.completions.stream");
977
- const wrappedStreamInner = wrapStreamForUsageTap(rawStream, async () => {
978
- const usage = await extractUsageFromStream(rawStream, ctx.begin.data.vendorHints);
979
- if (usage) {
980
- ctx.setUsage(usage);
981
- }
982
- }, ctx);
983
- return wrappedStreamInner;
984
- });
985
- return wrappedPromise;
986
- }, withUsage2);
987
- } : void 0;
988
- const handler = {
989
- get(target, prop, receiver) {
990
- if (prop === "create") {
991
- return wrappedCreate;
992
- }
993
- if (prop === "stream" && wrappedStream) {
994
- return wrappedStream;
995
- }
996
- return Reflect.get(target, prop, receiver);
997
- }
998
- };
999
- return new Proxy(resource, handler);
1000
- }
1001
- function splitUsageOptions(options) {
1002
- if (!options || typeof options !== "object") {
1003
- return {};
1004
- }
1005
- const { usageTap, withUsage: withUsage2, ...rest } = options;
1006
- const requestOptions = Object.keys(rest).length ? cloneRequestOptions(rest) : void 0;
1007
- return {
1008
- requestOptions,
1009
- usageContext: usageTap,
1010
- withUsage: withUsage2
1011
- };
1012
- }
1013
- function resolveBeginRequest(defaults, override) {
1014
- const base = defaults ?? {};
1015
- const current = override ?? {};
1016
- const customerId = current.customerId ?? base.customerId;
1017
- if (!customerId) {
1018
- throw new UsageTapError(
1019
- "USAGETAP_BAD_REQUEST",
1020
- "wrapOpenAI requires usageTap.customerId (provide defaultContext or options.usageTap)"
1021
- );
1022
- }
1023
- const tags = mergeTags(base.tags, current.tags);
1024
- const begin = { customerId };
1025
- const requested = current.requested ?? base.requested;
1026
- if (requested) begin.requested = requested;
1027
- const feature = current.feature ?? base.feature;
1028
- if (feature) begin.feature = feature;
1029
- const idempotency = current.idempotency ?? base.idempotency;
1030
- if (idempotency) begin.idempotency = idempotency;
1031
- const customerName = current.customerName ?? base.customerName;
1032
- if (customerName) begin.customerName = customerName;
1033
- const customerEmail = current.customerEmail ?? base.customerEmail;
1034
- if (customerEmail) begin.customerEmail = customerEmail;
1035
- if (tags?.length) {
1036
- begin.tags = tags;
1037
- }
1038
- return begin;
1039
- }
1040
- function transformApiPromise(apiPromise, onResolve) {
1041
- const resolvedPromise = Promise.resolve(apiPromise).then(onResolve);
1042
- if (isObjectRecord(apiPromise)) {
1043
- const proto = Object.getPrototypeOf(apiPromise);
1044
- if (proto) {
1045
- Object.setPrototypeOf(resolvedPromise, proto);
1046
- }
1047
- for (const key of Reflect.ownKeys(apiPromise)) {
1048
- if (key === "then" || key === "catch" || key === "finally") {
1049
- continue;
1050
- }
1051
- try {
1052
- const descriptor = Object.getOwnPropertyDescriptor(apiPromise, key);
1053
- if (descriptor) {
1054
- Reflect.defineProperty(resolvedPromise, key, descriptor);
1055
- }
1056
- } catch {
1057
- }
1058
- }
1059
- }
1060
- return resolvedPromise;
1061
- }
1062
- function isObjectRecord(value) {
1063
- return typeof value === "object" && value !== null;
1064
- }
1065
- function cloneRecord(value) {
1066
- return isObjectRecord(value) ? { ...value } : {};
1067
- }
1068
- function isStringTuple(value) {
1069
- return Array.isArray(value) && value.length >= 2 && typeof value[0] === "string" && typeof value[1] === "string";
1070
- }
1071
- function cloneRequestOptions(source) {
1072
- const clone = { ...source };
1073
- if ("headers" in clone) {
1074
- clone.headers = normalizeHeaders(clone.headers);
1075
- }
1076
- return clone;
1077
- }
1078
- function attachCorrelationHeader(options, correlationId) {
1079
- const normalized = normalizeHeaders(options?.headers);
1080
- if (correlationId && !normalized[USAGETAP_CORRELATION_HEADER]) {
1081
- normalized[USAGETAP_CORRELATION_HEADER] = correlationId;
1082
- }
1083
- if (!options) {
1084
- return Object.keys(normalized).length ? { headers: normalized } : void 0;
1085
- }
1086
- const next = { ...options };
1087
- if (Object.keys(normalized).length) {
1088
- next.headers = normalized;
1089
- }
1090
- return next;
1091
- }
1092
- function normalizeHeaders(headers) {
1093
- if (!headers) {
1094
- return {};
1095
- }
1096
- if (headers instanceof Headers) {
1097
- const result = {};
1098
- headers.forEach((value, key) => {
1099
- result[key.toLowerCase()] = value;
1100
- });
1101
- return result;
1102
- }
1103
- if (Array.isArray(headers)) {
1104
- const result = {};
1105
- for (const entry of headers) {
1106
- if (!isStringTuple(entry)) {
1107
- continue;
1108
- }
1109
- const [key, value] = entry;
1110
- result[key.toLowerCase()] = value;
1111
- }
1112
- return result;
1113
- }
1114
- if (isObjectRecord(headers)) {
1115
- const result = {};
1116
- const record = headers;
1117
- for (const key of Object.keys(record)) {
1118
- const value = record[key];
1119
- if (value !== void 0 && value !== null) {
1120
- result[key.toLowerCase()] = String(value);
1121
- }
1122
- }
1123
- return result;
1124
- }
1125
- return {};
1126
- }
1127
- function mergeTags(a, b) {
1128
- const values = [...a ?? [], ...b ?? []].map((value) => typeof value === "string" ? value.trim() : "").filter(Boolean);
1129
- if (!values.length) {
1130
- return void 0;
1131
- }
1132
- return dedupeStrings2(values);
1133
- }
1134
- function dedupeStrings2(values) {
1135
- return Array.from(new Set(values));
1136
- }
1137
- function isStreamingRequest(params) {
1138
- if (!params || typeof params !== "object") {
1139
- return false;
1140
- }
1141
- const stream = params.stream;
1142
- if (typeof stream === "boolean") {
1143
- return stream;
1144
- }
1145
- return stream != null;
1146
- }
1147
- function applyChatVendorHints(params, hints) {
1148
- if (!hints) {
1149
- return params;
1150
- }
1151
- const next = cloneRecord(params);
1152
- if (hints.preferredModel && (next.model === void 0 || next.model === null)) {
1153
- next.model = hints.preferredModel;
1154
- }
1155
- if (typeof hints.maxResponseTokens === "number" && next.max_tokens == null) {
1156
- next.max_tokens = hints.maxResponseTokens;
1157
- }
1158
- if (typeof hints.maxInputTokens === "number" && next.max_input_tokens == null) {
1159
- next.max_input_tokens = hints.maxInputTokens;
1160
- }
1161
- return next;
1162
- }
1163
- function applyResponsesVendorHints(params, hints) {
1164
- if (!hints) {
1165
- return params;
1166
- }
1167
- const next = cloneRecord(params);
1168
- if (hints.preferredModel && (next.model === void 0 || next.model === null)) {
1169
- next.model = hints.preferredModel;
1170
- }
1171
- if (typeof hints.maxResponseTokens === "number" && next.max_output_tokens == null) {
1172
- next.max_output_tokens = hints.maxResponseTokens;
1173
- }
1174
- return next;
1175
- }
1176
- async function extractUsageFromStream(stream, hints) {
1177
- const finalPayload = await resolveStreamFinalPayload(stream);
1178
- if (!finalPayload) {
1179
- return void 0;
1180
- }
1181
- return inferUsageFromResponse(finalPayload, hints);
1182
- }
1183
- async function resolveStreamFinalPayload(stream) {
1184
- if (!stream || typeof stream !== "object") {
1185
- return void 0;
1186
- }
1187
- const candidate = stream;
1188
- if (typeof candidate.finalChatCompletion === "function") {
1189
- return candidate.finalChatCompletion();
1190
- }
1191
- if (typeof candidate.finalResponse === "function") {
1192
- return candidate.finalResponse();
1193
- }
1194
- if (typeof candidate.finalCompletion === "function") {
1195
- return candidate.finalCompletion();
1196
- }
1197
- if (typeof candidate.finalContent === "function") {
1198
- return candidate.finalContent();
1199
- }
1200
- return void 0;
1201
- }
1202
- function ensureAsyncIterable(value, label) {
1203
- if (!value || typeof value !== "object" || typeof value[Symbol.asyncIterator] !== "function") {
1204
- throw new UsageTapError(
1205
- "USAGETAP_BAD_REQUEST",
1206
- `${label} expected an async iterable stream but received ${typeof value}`
1207
- );
1208
- }
1209
- }
1210
- function chunkToText(chunk) {
1211
- if (chunk === void 0 || chunk === null) {
1212
- return "";
1213
- }
1214
- if (typeof chunk === "string") {
1215
- return chunk;
1216
- }
1217
- if (typeof chunk === "object") {
1218
- const candidate = chunk;
1219
- const delta = candidate.choices?.[0]?.delta;
1220
- const content = delta?.content ?? candidate.content;
1221
- if (typeof content === "string") {
1222
- return content;
1223
- }
1224
- if (Array.isArray(content)) {
1225
- return content.map((entry) => {
1226
- if (!entry) return "";
1227
- if (typeof entry === "string") return entry;
1228
- if (typeof entry.text === "string") return entry.text;
1229
- return "";
1230
- }).join("");
1231
- }
1232
- }
1233
- return String(chunk);
1234
- }
1235
- function formatSsePayload(text, options) {
1236
- if (!text) {
1237
- return "";
1238
- }
1239
- const lines = text.split(/\r?\n/);
1240
- const eventLine = options?.event ? `event: ${options.event}
1241
- ` : "";
1242
- const retryLine = options?.retry ? `retry: ${options.retry}
1243
- ` : "";
1244
- const dataLines = lines.map((line) => `data: ${line}`).join("\n");
1245
- return `${eventLine}${retryLine}${dataLines}
1246
-
1247
- `;
1248
- }
1249
- function setHeaderIfPossible(res, key, value) {
1250
- if (typeof res.setHeader === "function" && res.headersSent !== true) {
1251
- res.setHeader(key, value);
1252
- }
1253
- }
1254
- function tryInferUsage(response, hints, extractor, ctx) {
1255
- const explicit = extractor?.(response);
1256
- const inferred = explicit ?? inferUsageFromResponse(response, hints);
1257
- if (inferred) {
1258
- ctx.setUsage(inferred);
1259
- }
1260
- }
1261
- function inferUsageFromResponse(response, hints) {
1262
- if (!response || typeof response !== "object") {
1263
- return void 0;
1264
- }
1265
- const candidate = response;
1266
- if (!candidate.usage) {
1267
- return void 0;
1268
- }
1269
- return {
1270
- modelUsed: candidate.model ?? hints?.preferredModel,
1271
- inputTokens: candidate.usage.prompt_tokens,
1272
- responseTokens: candidate.usage.completion_tokens,
1273
- cachedTokens: candidate.usage.cached_tokens
1274
- };
1275
- }
1276
- function wrapStreamForUsageTap(source, finalize, ctx) {
1277
- const getIterator = source[Symbol.asyncIterator];
1278
- if (typeof getIterator !== "function") {
1279
- throw new TypeError("Stream is not async iterable");
1280
- }
1281
- const iterator = getIterator.call(source);
1282
- let completed = false;
1283
- const invokeFinalize = async () => {
1284
- if (completed) return;
1285
- completed = true;
1286
- try {
1287
- await finalize();
1288
- } catch (error) {
1289
- ctx.setError({
1290
- code: "USAGE_FINALIZE_ERROR",
1291
- message: error instanceof Error ? error.message : String(error)
1292
- });
1293
- throw error;
1294
- }
1295
- };
1296
- const prototype = Object.getPrototypeOf(source) ?? Object.prototype;
1297
- const wrapped = Object.create(prototype);
1298
- for (const key of Reflect.ownKeys(source)) {
1299
- try {
1300
- const descriptor = Object.getOwnPropertyDescriptor(source, key);
1301
- if (descriptor) {
1302
- Object.defineProperty(wrapped, key, descriptor);
1303
- }
1304
- } catch {
1305
- }
1306
- }
1307
- Object.defineProperty(wrapped, Symbol.asyncIterator, {
1308
- value() {
1309
- return this;
1310
- },
1311
- configurable: true
1312
- });
1313
- Object.defineProperty(wrapped, "next", {
1314
- value: async (...args) => {
1315
- try {
1316
- const result = await iterator.next(...args);
1317
- if (result.done) {
1318
- await invokeFinalize();
1319
- }
1320
- return result;
1321
- } catch (error) {
1322
- await invokeFinalize().catch(() => void 0);
1323
- throw error;
1324
- }
1325
- },
1326
- configurable: true,
1327
- writable: true
1328
- });
1329
- Object.defineProperty(wrapped, "return", {
1330
- value: async (value) => {
1331
- if (typeof iterator.return === "function") {
1332
- const rawResult = await iterator.return(value);
1333
- if (!isIteratorResult(rawResult)) {
1334
- throw new TypeError("Iterator.return() returned an invalid result");
1335
- }
1336
- await invokeFinalize();
1337
- return rawResult;
1338
- }
1339
- await invokeFinalize();
1340
- return { done: true, value };
1341
- },
1342
- configurable: true,
1343
- writable: true
1344
- });
1345
- Object.defineProperty(wrapped, "throw", {
1346
- value: async (error) => {
1347
- if (typeof iterator.throw === "function") {
1348
- const rawResult = await iterator.throw(error);
1349
- if (!isIteratorResult(rawResult)) {
1350
- throw new TypeError("Iterator.throw() returned an invalid result");
1351
- }
1352
- await invokeFinalize();
1353
- return rawResult;
1354
- }
1355
- await invokeFinalize();
1356
- throw error;
1357
- },
1358
- configurable: true,
1359
- writable: true
1360
- });
1361
- Object.defineProperty(wrapped, "__usageTapFinalize", {
1362
- value: async () => {
1363
- await invokeFinalize();
1364
- },
1365
- configurable: true
1366
- });
1367
- return wrapped;
1368
- }
1369
- function isIteratorResult(value) {
1370
- return isObjectRecord(value) && "done" in value;
1371
- }
1372
-
1373
- // src/adapters/openrouter.ts
1374
- function createOpenRouterAdapter(init) {
1375
- return createOpenAIAdapter(init);
1376
- }
1377
-
1378
642
  // src/adapters/fetch-wrapper.ts
1379
643
  function isJsonRecord(value) {
1380
644
  return typeof value === "object" && value !== null && !Array.isArray(value);
@@ -1618,51 +882,10 @@ async function finalizeCall(callState, usageTap, error, usage) {
1618
882
  }
1619
883
  }
1620
884
 
1621
- // src/adapters/express-middleware.ts
1622
- function withUsageMiddleware(usageTap, options) {
1623
- return async (req, res, next) => {
1624
- try {
1625
- const customerId = await options.getCustomerId(req);
1626
- const feature = options.getFeature ? await options.getFeature(req) : void 0;
1627
- const tags = options.getTags ? await options.getTags(req) : void 0;
1628
- const baseContext = {
1629
- customerId,
1630
- feature,
1631
- tags,
1632
- ...options.defaultContext
1633
- };
1634
- req.usageTap = {
1635
- openai: (client, contextOverride, wrapOptions) => {
1636
- const mergedContext = { ...baseContext, ...contextOverride };
1637
- return wrapOpenAI(client, usageTap, {
1638
- ...wrapOptions,
1639
- defaultContext: mergedContext
1640
- });
1641
- },
1642
- pipeToResponse
1643
- };
1644
- next();
1645
- } catch (error) {
1646
- next(error);
1647
- }
1648
- };
1649
- }
1650
- function withUsage(usageTap, getCustomerId) {
1651
- return withUsageMiddleware(usageTap, { getCustomerId });
1652
- }
1653
-
1654
885
  exports.UsageTapClient = UsageTapClient;
1655
886
  exports.UsageTapError = UsageTapError;
1656
887
  exports.createIdempotencyKey = createIdempotencyKey;
1657
- exports.createOpenAIAdapter = createOpenAIAdapter;
1658
- exports.createOpenRouterAdapter = createOpenRouterAdapter;
1659
888
  exports.isUsageTapError = isUsageTapError;
1660
- exports.pipeToResponse = pipeToResponse;
1661
- exports.streamOpenAIRoute = streamOpenAIRoute;
1662
- exports.toNextResponse = toNextResponse;
1663
- exports.withUsage = withUsage;
1664
- exports.withUsageMiddleware = withUsageMiddleware;
1665
889
  exports.wrapFetch = wrapFetch;
1666
- exports.wrapOpenAI = wrapOpenAI;
1667
890
  //# sourceMappingURL=index.cjs.map
1668
891
  //# sourceMappingURL=index.cjs.map