opik 0.2.0 → 0.2.1
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.
- package/dist/{Client-DilyWTdT.d.cts → Client-W52L9RIs.d.cts} +2 -0
- package/dist/{Client-DilyWTdT.d.ts → Client-W52L9RIs.d.ts} +2 -0
- package/dist/Node18UniversalStreamWrapper-GZXI7BSF.js +1 -0
- package/dist/NodePre18StreamWrapper-TZYQXHQT.js +1 -0
- package/dist/UndiciStreamWrapper-4QYH36IP.js +1 -0
- package/dist/chunk-POLPT3OJ.js +4 -0
- package/dist/chunk-WOT6VMZA.js +1 -0
- package/dist/index.cjs +4 -13021
- package/dist/index.d.cts +5 -3
- package/dist/index.d.ts +5 -3
- package/dist/index.js +1 -4
- package/dist/vercel/index.cjs +4 -13037
- package/dist/vercel/index.d.cts +1 -1
- package/dist/vercel/index.d.ts +1 -1
- package/dist/vercel/index.js +1 -207
- package/package.json +7 -2
- package/dist/Node18UniversalStreamWrapper-CE7OKAXD.js +0 -217
- package/dist/Node18UniversalStreamWrapper-CE7OKAXD.js.map +0 -1
- package/dist/NodePre18StreamWrapper-TYJO3GAZ.js +0 -89
- package/dist/NodePre18StreamWrapper-TYJO3GAZ.js.map +0 -1
- package/dist/UndiciStreamWrapper-UCC32KBP.js +0 -205
- package/dist/UndiciStreamWrapper-UCC32KBP.js.map +0 -1
- package/dist/chunk-PZ5AY32C.js +0 -9
- package/dist/chunk-PZ5AY32C.js.map +0 -1
- package/dist/chunk-ZB4ZJRLY.js +0 -12479
- package/dist/chunk-ZB4ZJRLY.js.map +0 -1
- package/dist/index.cjs.map +0 -1
- package/dist/index.js.map +0 -1
- package/dist/vercel/index.cjs.map +0 -1
- package/dist/vercel/index.js.map +0 -1
package/dist/vercel/index.d.cts
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { O as OpikClient, S as Span, T as Trace } from '../Client-
|
|
1
|
+
import { O as OpikClient, S as Span, T as Trace } from '../Client-W52L9RIs.cjs';
|
|
2
2
|
import { AttributeValue, Tracer } from '@opentelemetry/api';
|
|
3
3
|
import { NodeSDKConfiguration } from '@opentelemetry/sdk-node';
|
|
4
4
|
import 'stream';
|
package/dist/vercel/index.d.ts
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { O as OpikClient, S as Span, T as Trace } from '../Client-
|
|
1
|
+
import { O as OpikClient, S as Span, T as Trace } from '../Client-W52L9RIs.js';
|
|
2
2
|
import { AttributeValue, Tracer } from '@opentelemetry/api';
|
|
3
3
|
import { NodeSDKConfiguration } from '@opentelemetry/sdk-node';
|
|
4
4
|
import 'stream';
|
package/dist/vercel/index.js
CHANGED
|
@@ -1,207 +1 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import '../chunk-PZ5AY32C.js';
|
|
3
|
-
|
|
4
|
-
// node_modules/@opentelemetry/core/build/esm/ExportResult.js
|
|
5
|
-
var ExportResultCode;
|
|
6
|
-
(function(ExportResultCode2) {
|
|
7
|
-
ExportResultCode2[ExportResultCode2["SUCCESS"] = 0] = "SUCCESS";
|
|
8
|
-
ExportResultCode2[ExportResultCode2["FAILED"] = 1] = "FAILED";
|
|
9
|
-
})(ExportResultCode || (ExportResultCode = {}));
|
|
10
|
-
|
|
11
|
-
// src/opik/integrations/vercel/exporter.ts
|
|
12
|
-
var aiSDKClient = new OpikClient();
|
|
13
|
-
var OpikExporter = class {
|
|
14
|
-
constructor({ client = aiSDKClient } = {}) {
|
|
15
|
-
// <otelTraceId, opikTrace>
|
|
16
|
-
this.traces = /* @__PURE__ */ new Map();
|
|
17
|
-
// <otelSpanId, opikSpan>
|
|
18
|
-
this.spans = /* @__PURE__ */ new Map();
|
|
19
|
-
this.getSpanInput = (otelSpan) => {
|
|
20
|
-
let input = {};
|
|
21
|
-
const { attributes } = otelSpan;
|
|
22
|
-
const attributeKeys = Object.keys(attributes);
|
|
23
|
-
attributeKeys.forEach((key) => {
|
|
24
|
-
if (key === "ai.prompt" || key === "gen_ai.request") {
|
|
25
|
-
const parsedValue = tryParseJSON(attributes[key]);
|
|
26
|
-
if (parsedValue) {
|
|
27
|
-
input = { ...input, ...parsedValue };
|
|
28
|
-
}
|
|
29
|
-
}
|
|
30
|
-
if (key.startsWith("ai.prompt.")) {
|
|
31
|
-
const promptKey = key.replace("ai.prompt.", "");
|
|
32
|
-
input[promptKey] = safeParseJson(attributes[key]);
|
|
33
|
-
}
|
|
34
|
-
if (key.startsWith("gen_ai.request.")) {
|
|
35
|
-
const promptKey = key.replace("gen_ai.request.", "");
|
|
36
|
-
input[promptKey] = safeParseJson(attributes[key]);
|
|
37
|
-
}
|
|
38
|
-
});
|
|
39
|
-
if (Object.keys(input).length > 0) {
|
|
40
|
-
return input;
|
|
41
|
-
}
|
|
42
|
-
if ("ai.toolCall.name" in attributes) {
|
|
43
|
-
input.toolName = attributes["ai.toolCall.name"];
|
|
44
|
-
}
|
|
45
|
-
if ("ai.toolCall.args" in attributes) {
|
|
46
|
-
input.args = attributes["ai.toolCall.args"];
|
|
47
|
-
}
|
|
48
|
-
return input;
|
|
49
|
-
};
|
|
50
|
-
this.getSpanOutput = (otelSpan) => {
|
|
51
|
-
const { attributes } = otelSpan;
|
|
52
|
-
if (attributes["ai.response.text"]) {
|
|
53
|
-
return { text: attributes["ai.response.text"] };
|
|
54
|
-
}
|
|
55
|
-
if (attributes["ai.toolCall.result"]) {
|
|
56
|
-
return { result: attributes["ai.toolCall.result"] };
|
|
57
|
-
}
|
|
58
|
-
if (attributes["ai.response.toolCalls"]) {
|
|
59
|
-
return { toolCalls: safeParseJson(attributes["ai.response.toolCalls"]) };
|
|
60
|
-
}
|
|
61
|
-
return {};
|
|
62
|
-
};
|
|
63
|
-
this.getSpanMetadata = (otelSpan) => {
|
|
64
|
-
const { attributes } = otelSpan;
|
|
65
|
-
const metadata = {};
|
|
66
|
-
if (attributes["gen_ai.response.model"]) {
|
|
67
|
-
metadata.model = attributes["gen_ai.response.model"];
|
|
68
|
-
}
|
|
69
|
-
if (attributes["gen_ai.system"]) {
|
|
70
|
-
metadata.system = attributes["gen_ai.system"];
|
|
71
|
-
}
|
|
72
|
-
return metadata;
|
|
73
|
-
};
|
|
74
|
-
this.getSpanUsage = (otelSpan) => {
|
|
75
|
-
const { attributes } = otelSpan;
|
|
76
|
-
const usage = {};
|
|
77
|
-
if ("ai.usage.promptTokens" in attributes) {
|
|
78
|
-
usage.prompt_tokens = attributes["ai.usage.promptTokens"];
|
|
79
|
-
}
|
|
80
|
-
if ("gen_ai.usage.input_tokens" in attributes) {
|
|
81
|
-
usage.prompt_tokens = attributes["gen_ai.usage.input_tokens"];
|
|
82
|
-
}
|
|
83
|
-
if ("ai.usage.completionTokens" in attributes) {
|
|
84
|
-
usage.completion_tokens = attributes["ai.usage.completionTokens"];
|
|
85
|
-
}
|
|
86
|
-
if ("gen_ai.usage.output_tokens" in attributes) {
|
|
87
|
-
usage.completion_tokens = attributes["gen_ai.usage.output_tokens"];
|
|
88
|
-
}
|
|
89
|
-
if ("prompt_tokens" in usage || "completion_tokens" in usage) {
|
|
90
|
-
usage.total_tokens = (usage.prompt_tokens || 0) + (usage.completion_tokens || 0);
|
|
91
|
-
}
|
|
92
|
-
return usage;
|
|
93
|
-
};
|
|
94
|
-
this.processSpan = ({
|
|
95
|
-
otelSpan,
|
|
96
|
-
parentSpan,
|
|
97
|
-
trace
|
|
98
|
-
}) => {
|
|
99
|
-
return trace.span({
|
|
100
|
-
name: otelSpan.name,
|
|
101
|
-
startTime: new Date(hrTimeToMilliseconds(otelSpan.startTime)),
|
|
102
|
-
endTime: new Date(hrTimeToMilliseconds(otelSpan.endTime)),
|
|
103
|
-
parentSpanId: parentSpan == null ? void 0 : parentSpan.data.id,
|
|
104
|
-
input: this.getSpanInput(otelSpan),
|
|
105
|
-
output: this.getSpanOutput(otelSpan),
|
|
106
|
-
metadata: this.getSpanMetadata(otelSpan),
|
|
107
|
-
usage: this.getSpanUsage(otelSpan),
|
|
108
|
-
type: "llm"
|
|
109
|
-
});
|
|
110
|
-
};
|
|
111
|
-
this.shutdown = async () => {
|
|
112
|
-
await this.client.flush();
|
|
113
|
-
};
|
|
114
|
-
this.forceFlush = async () => {
|
|
115
|
-
await this.client.flush();
|
|
116
|
-
};
|
|
117
|
-
this.export = (otelSpans, resultCallback) => {
|
|
118
|
-
const spanGroups = groupAndSortOtelSpans(otelSpans);
|
|
119
|
-
Object.entries(spanGroups).forEach(([otelTraceId, otelSpans2]) => {
|
|
120
|
-
var _a, _b;
|
|
121
|
-
const [rootOtelSpan, ...otherOtelSpans] = otelSpans2;
|
|
122
|
-
const trace = this.client.trace({
|
|
123
|
-
startTime: new Date(hrTimeToMilliseconds(rootOtelSpan.startTime)),
|
|
124
|
-
endTime: new Date(hrTimeToMilliseconds(rootOtelSpan.endTime)),
|
|
125
|
-
name: (_b = (_a = rootOtelSpan.attributes["ai.telemetry.metadata.traceName"]) == null ? void 0 : _a.toString()) != null ? _b : rootOtelSpan.name,
|
|
126
|
-
input: this.getSpanInput(rootOtelSpan),
|
|
127
|
-
output: this.getSpanOutput(rootOtelSpan),
|
|
128
|
-
metadata: this.getSpanMetadata(rootOtelSpan),
|
|
129
|
-
usage: this.getSpanUsage(rootOtelSpan)
|
|
130
|
-
});
|
|
131
|
-
this.traces.set(otelTraceId, trace);
|
|
132
|
-
otherOtelSpans.forEach((otelSpan) => {
|
|
133
|
-
var _a2;
|
|
134
|
-
const parentSpan = this.spans.get((_a2 = otelSpan.parentSpanId) != null ? _a2 : "");
|
|
135
|
-
const span = this.processSpan({ parentSpan, otelSpan, trace });
|
|
136
|
-
this.spans.set(otelSpan.spanContext().spanId, span);
|
|
137
|
-
});
|
|
138
|
-
});
|
|
139
|
-
resultCallback({ code: ExportResultCode.SUCCESS });
|
|
140
|
-
};
|
|
141
|
-
this.client = client;
|
|
142
|
-
}
|
|
143
|
-
static getSettings(settings) {
|
|
144
|
-
var _a, _b, _c;
|
|
145
|
-
const metadata = { ...settings.metadata };
|
|
146
|
-
if (settings.name) {
|
|
147
|
-
metadata.traceName = settings.name;
|
|
148
|
-
}
|
|
149
|
-
return {
|
|
150
|
-
isEnabled: (_a = settings.isEnabled) != null ? _a : true,
|
|
151
|
-
recordInputs: (_b = settings.recordInputs) != null ? _b : true,
|
|
152
|
-
recordOutputs: (_c = settings.recordOutputs) != null ? _c : true,
|
|
153
|
-
functionId: settings.functionId,
|
|
154
|
-
metadata
|
|
155
|
-
};
|
|
156
|
-
}
|
|
157
|
-
};
|
|
158
|
-
function groupAndSortOtelSpans(otelSpans) {
|
|
159
|
-
const spanGroupsByTraceId = {};
|
|
160
|
-
otelSpans.forEach((otelSpan) => {
|
|
161
|
-
const context = otelSpan.spanContext();
|
|
162
|
-
if (!spanGroupsByTraceId[context.traceId]) {
|
|
163
|
-
spanGroupsByTraceId[context.traceId] = [];
|
|
164
|
-
}
|
|
165
|
-
spanGroupsByTraceId[context.traceId].push(otelSpan);
|
|
166
|
-
});
|
|
167
|
-
Object.values(spanGroupsByTraceId).forEach((otelSpans2) => {
|
|
168
|
-
otelSpans2.sort((a, b) => {
|
|
169
|
-
if (!a.parentSpanId && b.parentSpanId) return -1;
|
|
170
|
-
if (a.parentSpanId && !b.parentSpanId) return 1;
|
|
171
|
-
if (a.parentSpanId && b.parentSpanId) {
|
|
172
|
-
if (a.parentSpanId === b.spanContext().spanId) return 1;
|
|
173
|
-
if (b.parentSpanId === a.spanContext().spanId) return -1;
|
|
174
|
-
}
|
|
175
|
-
return 0;
|
|
176
|
-
});
|
|
177
|
-
});
|
|
178
|
-
return spanGroupsByTraceId;
|
|
179
|
-
}
|
|
180
|
-
function hrTimeToMilliseconds(hrTime) {
|
|
181
|
-
return hrTime[0] * 1e3 + hrTime[1] / 1e6;
|
|
182
|
-
}
|
|
183
|
-
function safeParseJson(value) {
|
|
184
|
-
try {
|
|
185
|
-
return JSON.parse(value);
|
|
186
|
-
} catch {
|
|
187
|
-
return value;
|
|
188
|
-
}
|
|
189
|
-
}
|
|
190
|
-
function tryParseJSON(input) {
|
|
191
|
-
if (typeof input !== "string") {
|
|
192
|
-
return void 0;
|
|
193
|
-
}
|
|
194
|
-
try {
|
|
195
|
-
const parsed = JSON.parse(input);
|
|
196
|
-
if (parsed !== null && typeof parsed === "object" && !Array.isArray(parsed)) {
|
|
197
|
-
return parsed;
|
|
198
|
-
}
|
|
199
|
-
} catch {
|
|
200
|
-
return void 0;
|
|
201
|
-
}
|
|
202
|
-
return void 0;
|
|
203
|
-
}
|
|
204
|
-
|
|
205
|
-
export { OpikExporter };
|
|
206
|
-
//# sourceMappingURL=index.js.map
|
|
207
|
-
//# sourceMappingURL=index.js.map
|
|
1
|
+
import {b}from'../chunk-POLPT3OJ.js';import'../chunk-WOT6VMZA.js';var s;(function(a){a[a.SUCCESS=0]="SUCCESS",a[a.FAILED=1]="FAILED";})(s||(s={}));var I=new b,c=class{constructor({client:r=I}={}){this.traces=new Map;this.spans=new Map;this.getSpanInput=r=>{let e={},{attributes:t}=r;return Object.keys(t).forEach(o=>{if(o==="ai.prompt"||o==="gen_ai.request"){let n=h(t[o]);n&&(e={...e,...n});}if(o.startsWith("ai.prompt.")){let n=o.replace("ai.prompt.","");e[n]=m(t[o]);}if(o.startsWith("gen_ai.request.")){let n=o.replace("gen_ai.request.","");e[n]=m(t[o]);}}),Object.keys(e).length>0||("ai.toolCall.name"in t&&(e.toolName=t["ai.toolCall.name"]),"ai.toolCall.args"in t&&(e.args=t["ai.toolCall.args"])),e};this.getSpanOutput=r=>{let{attributes:e}=r;return e["ai.response.text"]?{text:e["ai.response.text"]}:e["ai.toolCall.result"]?{result:e["ai.toolCall.result"]}:e["ai.response.toolCalls"]?{toolCalls:m(e["ai.response.toolCalls"])}:{}};this.getSpanMetadata=r=>{let{attributes:e}=r,t={};return e["gen_ai.response.model"]&&(t.model=e["gen_ai.response.model"]),e["gen_ai.system"]&&(t.system=e["gen_ai.system"]),t};this.getSpanUsage=r=>{let{attributes:e}=r,t={};return "ai.usage.promptTokens"in e&&(t.prompt_tokens=e["ai.usage.promptTokens"]),"gen_ai.usage.input_tokens"in e&&(t.prompt_tokens=e["gen_ai.usage.input_tokens"]),"ai.usage.completionTokens"in e&&(t.completion_tokens=e["ai.usage.completionTokens"]),"gen_ai.usage.output_tokens"in e&&(t.completion_tokens=e["gen_ai.usage.output_tokens"]),("prompt_tokens"in t||"completion_tokens"in t)&&(t.total_tokens=(t.prompt_tokens||0)+(t.completion_tokens||0)),t};this.processSpan=({otelSpan:r,parentSpan:e,trace:t})=>t.span({name:r.name,startTime:new Date(p(r.startTime)),endTime:new Date(p(r.endTime)),parentSpanId:e==null?void 0:e.data.id,input:this.getSpanInput(r),output:this.getSpanOutput(r),metadata:this.getSpanMetadata(r),usage:this.getSpanUsage(r),type:"llm"});this.shutdown=async()=>{await this.client.flush();};this.forceFlush=async()=>{await this.client.flush();};this.export=(r,e)=>{let t=_(r);Object.entries(t).forEach(([i,o])=>{var d,g;let[n,...S]=o,l=this.client.trace({startTime:new Date(p(n.startTime)),endTime:new Date(p(n.endTime)),name:(g=(d=n.attributes["ai.telemetry.metadata.traceName"])==null?void 0:d.toString())!=null?g:n.name,input:this.getSpanInput(n),output:this.getSpanOutput(n),metadata:this.getSpanMetadata(n),usage:this.getSpanUsage(n)});this.traces.set(i,l),S.forEach(u=>{var f;let x=this.spans.get((f=u.parentSpanId)!=null?f:""),E=this.processSpan({parentSpan:x,otelSpan:u,trace:l});this.spans.set(u.spanContext().spanId,E);});}),e({code:s.SUCCESS});};this.client=r;}static getSettings(r){var t,i,o;let e={...r.metadata};return r.name&&(e.traceName=r.name),{isEnabled:(t=r.isEnabled)!=null?t:true,recordInputs:(i=r.recordInputs)!=null?i:true,recordOutputs:(o=r.recordOutputs)!=null?o:true,functionId:r.functionId,metadata:e}}};function _(a){let r={};return a.forEach(e=>{let t=e.spanContext();r[t.traceId]||(r[t.traceId]=[]),r[t.traceId].push(e);}),Object.values(r).forEach(e=>{e.sort((t,i)=>{if(!t.parentSpanId&&i.parentSpanId)return -1;if(t.parentSpanId&&!i.parentSpanId)return 1;if(t.parentSpanId&&i.parentSpanId){if(t.parentSpanId===i.spanContext().spanId)return 1;if(i.parentSpanId===t.spanContext().spanId)return -1}return 0});}),r}function p(a){return a[0]*1e3+a[1]/1e6}function m(a){try{return JSON.parse(a)}catch{return a}}function h(a){if(typeof a=="string")try{let r=JSON.parse(a);if(r!==null&&typeof r=="object"&&!Array.isArray(r))return r}catch{return}}export{c as OpikExporter};
|
package/package.json
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "opik",
|
|
3
3
|
"description": "Opik TypeScript and JavaScript SDK",
|
|
4
|
-
"version": "0.2.
|
|
4
|
+
"version": "0.2.1",
|
|
5
5
|
"repository": {
|
|
6
6
|
"type": "git",
|
|
7
7
|
"url": "git+https://github.com/comet-ml/opik.git",
|
|
@@ -73,7 +73,7 @@
|
|
|
73
73
|
"eslint": "^9.20.0",
|
|
74
74
|
"globals": "^15.14.0",
|
|
75
75
|
"prettier": "^3.4.1",
|
|
76
|
-
"tsup": "^8.3.
|
|
76
|
+
"tsup": "^8.3.6",
|
|
77
77
|
"typescript": "^5.7.2",
|
|
78
78
|
"typescript-eslint": "^8.24.0",
|
|
79
79
|
"vite-tsconfig-paths": "^5.1.3",
|
|
@@ -83,10 +83,15 @@
|
|
|
83
83
|
"engines": {
|
|
84
84
|
"node": ">=18"
|
|
85
85
|
},
|
|
86
|
+
"overrides": {
|
|
87
|
+
"esbuild": "^0.25.0"
|
|
88
|
+
},
|
|
86
89
|
"dependencies": {
|
|
90
|
+
"ansi-escapes": "^7.0.0",
|
|
87
91
|
"node-fetch": "^3.3.2",
|
|
88
92
|
"qs": "^6.13.1",
|
|
89
93
|
"readable-stream": "^4.7.0",
|
|
94
|
+
"tslog": "^4.9.3",
|
|
90
95
|
"url-join": "^5.0.0",
|
|
91
96
|
"uuid": "^11.0.3"
|
|
92
97
|
}
|
|
@@ -1,217 +0,0 @@
|
|
|
1
|
-
import './chunk-PZ5AY32C.js';
|
|
2
|
-
|
|
3
|
-
// src/opik/rest_api/core/fetcher/stream-wrappers/Node18UniversalStreamWrapper.ts
|
|
4
|
-
var Node18UniversalStreamWrapper = class _Node18UniversalStreamWrapper {
|
|
5
|
-
constructor(readableStream) {
|
|
6
|
-
this.readableStream = readableStream;
|
|
7
|
-
this.reader = this.readableStream.getReader();
|
|
8
|
-
this.events = {
|
|
9
|
-
data: [],
|
|
10
|
-
end: [],
|
|
11
|
-
error: [],
|
|
12
|
-
readable: [],
|
|
13
|
-
close: [],
|
|
14
|
-
pause: [],
|
|
15
|
-
resume: []
|
|
16
|
-
};
|
|
17
|
-
this.paused = false;
|
|
18
|
-
this.resumeCallback = null;
|
|
19
|
-
this.encoding = null;
|
|
20
|
-
}
|
|
21
|
-
on(event, callback) {
|
|
22
|
-
var _a;
|
|
23
|
-
(_a = this.events[event]) == null ? void 0 : _a.push(callback);
|
|
24
|
-
}
|
|
25
|
-
off(event, callback) {
|
|
26
|
-
var _a;
|
|
27
|
-
this.events[event] = (_a = this.events[event]) == null ? void 0 : _a.filter((cb) => cb !== callback);
|
|
28
|
-
}
|
|
29
|
-
pipe(dest) {
|
|
30
|
-
this.on("data", async (chunk) => {
|
|
31
|
-
if (dest instanceof _Node18UniversalStreamWrapper) {
|
|
32
|
-
dest._write(chunk);
|
|
33
|
-
} else if (dest instanceof WritableStream) {
|
|
34
|
-
const writer = dest.getWriter();
|
|
35
|
-
writer.write(chunk).then(() => writer.releaseLock());
|
|
36
|
-
} else {
|
|
37
|
-
dest.write(chunk);
|
|
38
|
-
}
|
|
39
|
-
});
|
|
40
|
-
this.on("end", async () => {
|
|
41
|
-
if (dest instanceof _Node18UniversalStreamWrapper) {
|
|
42
|
-
dest._end();
|
|
43
|
-
} else if (dest instanceof WritableStream) {
|
|
44
|
-
const writer = dest.getWriter();
|
|
45
|
-
writer.close();
|
|
46
|
-
} else {
|
|
47
|
-
dest.end();
|
|
48
|
-
}
|
|
49
|
-
});
|
|
50
|
-
this.on("error", async (error) => {
|
|
51
|
-
if (dest instanceof _Node18UniversalStreamWrapper) {
|
|
52
|
-
dest._error(error);
|
|
53
|
-
} else if (dest instanceof WritableStream) {
|
|
54
|
-
const writer = dest.getWriter();
|
|
55
|
-
writer.abort(error);
|
|
56
|
-
} else {
|
|
57
|
-
dest.destroy(error);
|
|
58
|
-
}
|
|
59
|
-
});
|
|
60
|
-
this._startReading();
|
|
61
|
-
return dest;
|
|
62
|
-
}
|
|
63
|
-
pipeTo(dest) {
|
|
64
|
-
return this.pipe(dest);
|
|
65
|
-
}
|
|
66
|
-
unpipe(dest) {
|
|
67
|
-
this.off("data", async (chunk) => {
|
|
68
|
-
if (dest instanceof _Node18UniversalStreamWrapper) {
|
|
69
|
-
dest._write(chunk);
|
|
70
|
-
} else if (dest instanceof WritableStream) {
|
|
71
|
-
const writer = dest.getWriter();
|
|
72
|
-
writer.write(chunk).then(() => writer.releaseLock());
|
|
73
|
-
} else {
|
|
74
|
-
dest.write(chunk);
|
|
75
|
-
}
|
|
76
|
-
});
|
|
77
|
-
this.off("end", async () => {
|
|
78
|
-
if (dest instanceof _Node18UniversalStreamWrapper) {
|
|
79
|
-
dest._end();
|
|
80
|
-
} else if (dest instanceof WritableStream) {
|
|
81
|
-
const writer = dest.getWriter();
|
|
82
|
-
writer.close();
|
|
83
|
-
} else {
|
|
84
|
-
dest.end();
|
|
85
|
-
}
|
|
86
|
-
});
|
|
87
|
-
this.off("error", async (error) => {
|
|
88
|
-
if (dest instanceof _Node18UniversalStreamWrapper) {
|
|
89
|
-
dest._error(error);
|
|
90
|
-
} else if (dest instanceof WritableStream) {
|
|
91
|
-
const writer = dest.getWriter();
|
|
92
|
-
writer.abort(error);
|
|
93
|
-
} else {
|
|
94
|
-
dest.destroy(error);
|
|
95
|
-
}
|
|
96
|
-
});
|
|
97
|
-
}
|
|
98
|
-
destroy(error) {
|
|
99
|
-
this.reader.cancel(error).then(() => {
|
|
100
|
-
this._emit("close");
|
|
101
|
-
}).catch((err) => {
|
|
102
|
-
this._emit("error", err);
|
|
103
|
-
});
|
|
104
|
-
}
|
|
105
|
-
pause() {
|
|
106
|
-
this.paused = true;
|
|
107
|
-
this._emit("pause");
|
|
108
|
-
}
|
|
109
|
-
resume() {
|
|
110
|
-
if (this.paused) {
|
|
111
|
-
this.paused = false;
|
|
112
|
-
this._emit("resume");
|
|
113
|
-
if (this.resumeCallback) {
|
|
114
|
-
this.resumeCallback();
|
|
115
|
-
this.resumeCallback = null;
|
|
116
|
-
}
|
|
117
|
-
}
|
|
118
|
-
}
|
|
119
|
-
get isPaused() {
|
|
120
|
-
return this.paused;
|
|
121
|
-
}
|
|
122
|
-
async read() {
|
|
123
|
-
if (this.paused) {
|
|
124
|
-
await new Promise((resolve) => {
|
|
125
|
-
this.resumeCallback = resolve;
|
|
126
|
-
});
|
|
127
|
-
}
|
|
128
|
-
const { done, value } = await this.reader.read();
|
|
129
|
-
if (done) {
|
|
130
|
-
return void 0;
|
|
131
|
-
}
|
|
132
|
-
return value;
|
|
133
|
-
}
|
|
134
|
-
setEncoding(encoding) {
|
|
135
|
-
this.encoding = encoding;
|
|
136
|
-
}
|
|
137
|
-
async text() {
|
|
138
|
-
const chunks = [];
|
|
139
|
-
while (true) {
|
|
140
|
-
const { done, value } = await this.reader.read();
|
|
141
|
-
if (done) {
|
|
142
|
-
break;
|
|
143
|
-
}
|
|
144
|
-
if (value) {
|
|
145
|
-
chunks.push(value);
|
|
146
|
-
}
|
|
147
|
-
}
|
|
148
|
-
const decoder = new TextDecoder(this.encoding || "utf-8");
|
|
149
|
-
return decoder.decode(await new Blob(chunks).arrayBuffer());
|
|
150
|
-
}
|
|
151
|
-
async json() {
|
|
152
|
-
const text = await this.text();
|
|
153
|
-
return JSON.parse(text);
|
|
154
|
-
}
|
|
155
|
-
_write(chunk) {
|
|
156
|
-
this._emit("data", chunk);
|
|
157
|
-
}
|
|
158
|
-
_end() {
|
|
159
|
-
this._emit("end");
|
|
160
|
-
}
|
|
161
|
-
_error(error) {
|
|
162
|
-
this._emit("error", error);
|
|
163
|
-
}
|
|
164
|
-
_emit(event, data) {
|
|
165
|
-
if (this.events[event]) {
|
|
166
|
-
for (const callback of this.events[event] || []) {
|
|
167
|
-
callback(data);
|
|
168
|
-
}
|
|
169
|
-
}
|
|
170
|
-
}
|
|
171
|
-
async _startReading() {
|
|
172
|
-
try {
|
|
173
|
-
this._emit("readable");
|
|
174
|
-
while (true) {
|
|
175
|
-
if (this.paused) {
|
|
176
|
-
await new Promise((resolve) => {
|
|
177
|
-
this.resumeCallback = resolve;
|
|
178
|
-
});
|
|
179
|
-
}
|
|
180
|
-
const { done, value } = await this.reader.read();
|
|
181
|
-
if (done) {
|
|
182
|
-
this._emit("end");
|
|
183
|
-
this._emit("close");
|
|
184
|
-
break;
|
|
185
|
-
}
|
|
186
|
-
if (value) {
|
|
187
|
-
this._emit("data", value);
|
|
188
|
-
}
|
|
189
|
-
}
|
|
190
|
-
} catch (error) {
|
|
191
|
-
this._emit("error", error);
|
|
192
|
-
}
|
|
193
|
-
}
|
|
194
|
-
[Symbol.asyncIterator]() {
|
|
195
|
-
return {
|
|
196
|
-
next: async () => {
|
|
197
|
-
if (this.paused) {
|
|
198
|
-
await new Promise((resolve) => {
|
|
199
|
-
this.resumeCallback = resolve;
|
|
200
|
-
});
|
|
201
|
-
}
|
|
202
|
-
const { done, value } = await this.reader.read();
|
|
203
|
-
if (done) {
|
|
204
|
-
return { done: true, value: void 0 };
|
|
205
|
-
}
|
|
206
|
-
return { done: false, value };
|
|
207
|
-
},
|
|
208
|
-
[Symbol.asyncIterator]() {
|
|
209
|
-
return this;
|
|
210
|
-
}
|
|
211
|
-
};
|
|
212
|
-
}
|
|
213
|
-
};
|
|
214
|
-
|
|
215
|
-
export { Node18UniversalStreamWrapper };
|
|
216
|
-
//# sourceMappingURL=Node18UniversalStreamWrapper-CE7OKAXD.js.map
|
|
217
|
-
//# sourceMappingURL=Node18UniversalStreamWrapper-CE7OKAXD.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"sources":["../src/opik/rest_api/core/fetcher/stream-wrappers/Node18UniversalStreamWrapper.ts"],"names":[],"mappings":";;;AAIa,IAAA,4BAAA,GAAN,MAAM,6BAGb,CAAA;AAAA,EAQI,YAAY,cAA4C,EAAA;AACpD,IAAA,IAAA,CAAK,cAAiB,GAAA,cAAA;AACtB,IAAK,IAAA,CAAA,MAAA,GAAS,IAAK,CAAA,cAAA,CAAe,SAAU,EAAA;AAC5C,IAAA,IAAA,CAAK,MAAS,GAAA;AAAA,MACV,MAAM,EAAC;AAAA,MACP,KAAK,EAAC;AAAA,MACN,OAAO,EAAC;AAAA,MACR,UAAU,EAAC;AAAA,MACX,OAAO,EAAC;AAAA,MACR,OAAO,EAAC;AAAA,MACR,QAAQ;AAAC,KACb;AACA,IAAA,IAAA,CAAK,MAAS,GAAA,KAAA;AACd,IAAA,IAAA,CAAK,cAAiB,GAAA,IAAA;AACtB,IAAA,IAAA,CAAK,QAAW,GAAA,IAAA;AAAA;AACpB,EAEO,EAAA,CAAG,OAAe,QAA+B,EAAA;AAhC5D,IAAA,IAAA,EAAA;AAiCQ,IAAA,CAAA,EAAA,GAAA,IAAA,CAAK,MAAO,CAAA,KAAK,CAAjB,KAAA,IAAA,GAAA,KAAA,CAAA,GAAA,EAAA,CAAoB,IAAK,CAAA,QAAA,CAAA;AAAA;AAC7B,EAEO,GAAA,CAAI,OAAe,QAA+B,EAAA;AApC7D,IAAA,IAAA,EAAA;AAqCQ,IAAK,IAAA,CAAA,MAAA,CAAO,KAAK,CAAA,GAAA,CAAI,EAAK,GAAA,IAAA,CAAA,MAAA,CAAO,KAAK,CAAA,KAAjB,IAAoB,GAAA,KAAA,CAAA,GAAA,EAAA,CAAA,MAAA,CAAO,CAAC,EAAA,KAAO,EAAO,KAAA,QAAA,CAAA;AAAA;AACnE,EAEO,KACH,IACgF,EAAA;AAChF,IAAK,IAAA,CAAA,EAAA,CAAG,MAAQ,EAAA,OAAO,KAAU,KAAA;AAC7B,MAAA,IAAI,gBAAgB,6BAA8B,EAAA;AAC9C,QAAA,IAAA,CAAK,OAAO,KAAK,CAAA;AAAA,OACrB,MAAA,IAAW,gBAAgB,cAAgB,EAAA;AACvC,QAAM,MAAA,MAAA,GAAS,KAAK,SAAU,EAAA;AAC9B,QAAA,MAAA,CAAO,MAAM,KAAK,CAAA,CAAE,KAAK,MAAM,MAAA,CAAO,aAAa,CAAA;AAAA,OAChD,MAAA;AACH,QAAA,IAAA,CAAK,MAAM,KAAK,CAAA;AAAA;AACpB,KACH,CAAA;AAED,IAAK,IAAA,CAAA,EAAA,CAAG,OAAO,YAAY;AACvB,MAAA,IAAI,gBAAgB,6BAA8B,EAAA;AAC9C,QAAA,IAAA,CAAK,IAAK,EAAA;AAAA,OACd,MAAA,IAAW,gBAAgB,cAAgB,EAAA;AACvC,QAAM,MAAA,MAAA,GAAS,KAAK,SAAU,EAAA;AAC9B,QAAA,MAAA,CAAO,KAAM,EAAA;AAAA,OACV,MAAA;AACH,QAAA,IAAA,CAAK,GAAI,EAAA;AAAA;AACb,KACH,CAAA;AAED,IAAK,IAAA,CAAA,EAAA,CAAG,OAAS,EAAA,OAAO,KAAU,KAAA;AAC9B,MAAA,IAAI,gBAAgB,6BAA8B,EAAA;AAC9C,QAAA,IAAA,CAAK,OAAO,KAAK,CAAA;AAAA,OACrB,MAAA,IAAW,gBAAgB,cAAgB,EAAA;AACvC,QAAM,MAAA,MAAA,GAAS,KAAK,SAAU,EAAA;AAC9B,QAAA,MAAA,CAAO,MAAM,KAAK,CAAA;AAAA,OACf,MAAA;AACH,QAAA,IAAA,CAAK,QAAQ,KAAK,CAAA;AAAA;AACtB,KACH,CAAA;AAED,IAAA,IAAA,CAAK,aAAc,EAAA;AAEnB,IAAO,OAAA,IAAA;AAAA;AACX,EAEO,OACH,IACgF,EAAA;AAChF,IAAO,OAAA,IAAA,CAAK,KAAK,IAAI,CAAA;AAAA;AACzB,EAEO,OAAO,IAA8F,EAAA;AACxG,IAAK,IAAA,CAAA,GAAA,CAAI,MAAQ,EAAA,OAAO,KAAU,KAAA;AAC9B,MAAA,IAAI,gBAAgB,6BAA8B,EAAA;AAC9C,QAAA,IAAA,CAAK,OAAO,KAAK,CAAA;AAAA,OACrB,MAAA,IAAW,gBAAgB,cAAgB,EAAA;AACvC,QAAM,MAAA,MAAA,GAAS,KAAK,SAAU,EAAA;AAC9B,QAAA,MAAA,CAAO,MAAM,KAAK,CAAA,CAAE,KAAK,MAAM,MAAA,CAAO,aAAa,CAAA;AAAA,OAChD,MAAA;AACH,QAAA,IAAA,CAAK,MAAM,KAAK,CAAA;AAAA;AACpB,KACH,CAAA;AAED,IAAK,IAAA,CAAA,GAAA,CAAI,OAAO,YAAY;AACxB,MAAA,IAAI,gBAAgB,6BAA8B,EAAA;AAC9C,QAAA,IAAA,CAAK,IAAK,EAAA;AAAA,OACd,MAAA,IAAW,gBAAgB,cAAgB,EAAA;AACvC,QAAM,MAAA,MAAA,GAAS,KAAK,SAAU,EAAA;AAC9B,QAAA,MAAA,CAAO,KAAM,EAAA;AAAA,OACV,MAAA;AACH,QAAA,IAAA,CAAK,GAAI,EAAA;AAAA;AACb,KACH,CAAA;AAED,IAAK,IAAA,CAAA,GAAA,CAAI,OAAS,EAAA,OAAO,KAAU,KAAA;AAC/B,MAAA,IAAI,gBAAgB,6BAA8B,EAAA;AAC9C,QAAA,IAAA,CAAK,OAAO,KAAK,CAAA;AAAA,OACrB,MAAA,IAAW,gBAAgB,cAAgB,EAAA;AACvC,QAAM,MAAA,MAAA,GAAS,KAAK,SAAU,EAAA;AAC9B,QAAA,MAAA,CAAO,MAAM,KAAK,CAAA;AAAA,OACf,MAAA;AACH,QAAA,IAAA,CAAK,QAAQ,KAAK,CAAA;AAAA;AACtB,KACH,CAAA;AAAA;AACL,EAEO,QAAQ,KAAqB,EAAA;AAChC,IAAA,IAAA,CAAK,MACA,CAAA,MAAA,CAAO,KAAK,CAAA,CACZ,KAAK,MAAM;AACR,MAAA,IAAA,CAAK,MAAM,OAAO,CAAA;AAAA,KACrB,CAAA,CACA,KAAM,CAAA,CAAC,GAAQ,KAAA;AACZ,MAAK,IAAA,CAAA,KAAA,CAAM,SAAS,GAAG,CAAA;AAAA,KAC1B,CAAA;AAAA;AACT,EAEO,KAAc,GAAA;AACjB,IAAA,IAAA,CAAK,MAAS,GAAA,IAAA;AACd,IAAA,IAAA,CAAK,MAAM,OAAO,CAAA;AAAA;AACtB,EAEO,MAAe,GAAA;AAClB,IAAA,IAAI,KAAK,MAAQ,EAAA;AACb,MAAA,IAAA,CAAK,MAAS,GAAA,KAAA;AACd,MAAA,IAAA,CAAK,MAAM,QAAQ,CAAA;AACnB,MAAA,IAAI,KAAK,cAAgB,EAAA;AACrB,QAAA,IAAA,CAAK,cAAe,EAAA;AACpB,QAAA,IAAA,CAAK,cAAiB,GAAA,IAAA;AAAA;AAC1B;AACJ;AACJ,EAEA,IAAW,QAAoB,GAAA;AAC3B,IAAA,OAAO,IAAK,CAAA,MAAA;AAAA;AAChB,EAEA,MAAa,IAAwC,GAAA;AACjD,IAAA,IAAI,KAAK,MAAQ,EAAA;AACb,MAAM,MAAA,IAAI,OAAQ,CAAA,CAAC,OAAY,KAAA;AAC3B,QAAA,IAAA,CAAK,cAAiB,GAAA,OAAA;AAAA,OACzB,CAAA;AAAA;AAEL,IAAA,MAAM,EAAE,IAAM,EAAA,KAAA,KAAU,MAAM,IAAA,CAAK,OAAO,IAAK,EAAA;AAE/C,IAAA,IAAI,IAAM,EAAA;AACN,MAAO,OAAA,KAAA,CAAA;AAAA;AAEX,IAAO,OAAA,KAAA;AAAA;AACX,EAEO,YAAY,QAAwB,EAAA;AACvC,IAAA,IAAA,CAAK,QAAW,GAAA,QAAA;AAAA;AACpB,EAEA,MAAa,IAAwB,GAAA;AACjC,IAAA,MAAM,SAAuB,EAAC;AAE9B,IAAA,OAAO,IAAM,EAAA;AACT,MAAA,MAAM,EAAE,IAAM,EAAA,KAAA,KAAU,MAAM,IAAA,CAAK,OAAO,IAAK,EAAA;AAC/C,MAAA,IAAI,IAAM,EAAA;AACN,QAAA;AAAA;AAEJ,MAAA,IAAI,KAAO,EAAA;AACP,QAAA,MAAA,CAAO,KAAK,KAAK,CAAA;AAAA;AACrB;AAGJ,IAAA,MAAM,OAAU,GAAA,IAAI,WAAY,CAAA,IAAA,CAAK,YAAY,OAAO,CAAA;AACxD,IAAO,OAAA,OAAA,CAAQ,OAAO,MAAM,IAAI,KAAK,MAAM,CAAA,CAAE,aAAa,CAAA;AAAA;AAC9D,EAEA,MAAa,IAAsB,GAAA;AAC/B,IAAM,MAAA,IAAA,GAAO,MAAM,IAAA,CAAK,IAAK,EAAA;AAC7B,IAAO,OAAA,IAAA,CAAK,MAAM,IAAI,CAAA;AAAA;AAC1B,EAEQ,OAAO,KAAyB,EAAA;AACpC,IAAK,IAAA,CAAA,KAAA,CAAM,QAAQ,KAAK,CAAA;AAAA;AAC5B,EAEQ,IAAa,GAAA;AACjB,IAAA,IAAA,CAAK,MAAM,KAAK,CAAA;AAAA;AACpB,EAEQ,OAAO,KAAkB,EAAA;AAC7B,IAAK,IAAA,CAAA,KAAA,CAAM,SAAS,KAAK,CAAA;AAAA;AAC7B,EAEQ,KAAA,CAAM,OAAe,IAAkB,EAAA;AAC3C,IAAI,IAAA,IAAA,CAAK,MAAO,CAAA,KAAK,CAAG,EAAA;AACpB,MAAA,KAAA,MAAW,YAAY,IAAK,CAAA,MAAA,CAAO,KAAK,CAAA,IAAK,EAAI,EAAA;AAC7C,QAAA,QAAA,CAAS,IAAI,CAAA;AAAA;AACjB;AACJ;AACJ,EAEA,MAAc,aAA+B,GAAA;AACzC,IAAI,IAAA;AACA,MAAA,IAAA,CAAK,MAAM,UAAU,CAAA;AACrB,MAAA,OAAO,IAAM,EAAA;AACT,QAAA,IAAI,KAAK,MAAQ,EAAA;AACb,UAAM,MAAA,IAAI,OAAQ,CAAA,CAAC,OAAY,KAAA;AAC3B,YAAA,IAAA,CAAK,cAAiB,GAAA,OAAA;AAAA,WACzB,CAAA;AAAA;AAEL,QAAA,MAAM,EAAE,IAAM,EAAA,KAAA,KAAU,MAAM,IAAA,CAAK,OAAO,IAAK,EAAA;AAC/C,QAAA,IAAI,IAAM,EAAA;AACN,UAAA,IAAA,CAAK,MAAM,KAAK,CAAA;AAChB,UAAA,IAAA,CAAK,MAAM,OAAO,CAAA;AAClB,UAAA;AAAA;AAEJ,QAAA,IAAI,KAAO,EAAA;AACP,UAAK,IAAA,CAAA,KAAA,CAAM,QAAQ,KAAK,CAAA;AAAA;AAC5B;AACJ,aACK,KAAO,EAAA;AACZ,MAAK,IAAA,CAAA,KAAA,CAAM,SAAS,KAAK,CAAA;AAAA;AAC7B;AACJ,EAEA,CAAC,MAAO,CAAA,aAAa,CAAuC,GAAA;AACxD,IAAO,OAAA;AAAA,MACH,MAAM,YAAY;AACd,QAAA,IAAI,KAAK,MAAQ,EAAA;AACb,UAAM,MAAA,IAAI,OAAQ,CAAA,CAAC,OAAY,KAAA;AAC3B,YAAA,IAAA,CAAK,cAAiB,GAAA,OAAA;AAAA,WACzB,CAAA;AAAA;AAEL,QAAA,MAAM,EAAE,IAAM,EAAA,KAAA,KAAU,MAAM,IAAA,CAAK,OAAO,IAAK,EAAA;AAC/C,QAAA,IAAI,IAAM,EAAA;AACN,UAAA,OAAO,EAAE,IAAA,EAAM,IAAM,EAAA,KAAA,EAAO,KAAU,CAAA,EAAA;AAAA;AAE1C,QAAO,OAAA,EAAE,IAAM,EAAA,KAAA,EAAO,KAAM,EAAA;AAAA,OAChC;AAAA,MACA,CAAC,MAAO,CAAA,aAAa,CAAI,GAAA;AACrB,QAAO,OAAA,IAAA;AAAA;AACX,KACJ;AAAA;AAER","file":"Node18UniversalStreamWrapper-CE7OKAXD.js","sourcesContent":["import type { Writable } from \"readable-stream\";\n\nimport { EventCallback, StreamWrapper } from \"./chooseStreamWrapper\";\n\nexport class Node18UniversalStreamWrapper<ReadFormat extends Uint8Array | Uint16Array | Uint32Array>\n implements\n StreamWrapper<Node18UniversalStreamWrapper<ReadFormat> | Writable | WritableStream<ReadFormat>, ReadFormat>\n{\n private readableStream: ReadableStream<ReadFormat>;\n private reader: ReadableStreamDefaultReader<ReadFormat>;\n private events: Record<string, EventCallback[] | undefined>;\n private paused: boolean;\n private resumeCallback: ((value?: unknown) => void) | null;\n private encoding: string | null;\n\n constructor(readableStream: ReadableStream<ReadFormat>) {\n this.readableStream = readableStream;\n this.reader = this.readableStream.getReader();\n this.events = {\n data: [],\n end: [],\n error: [],\n readable: [],\n close: [],\n pause: [],\n resume: [],\n };\n this.paused = false;\n this.resumeCallback = null;\n this.encoding = null;\n }\n\n public on(event: string, callback: EventCallback): void {\n this.events[event]?.push(callback);\n }\n\n public off(event: string, callback: EventCallback): void {\n this.events[event] = this.events[event]?.filter((cb) => cb !== callback);\n }\n\n public pipe(\n dest: Node18UniversalStreamWrapper<ReadFormat> | Writable | WritableStream<ReadFormat>,\n ): Node18UniversalStreamWrapper<ReadFormat> | Writable | WritableStream<ReadFormat> {\n this.on(\"data\", async (chunk) => {\n if (dest instanceof Node18UniversalStreamWrapper) {\n dest._write(chunk);\n } else if (dest instanceof WritableStream) {\n const writer = dest.getWriter();\n writer.write(chunk).then(() => writer.releaseLock());\n } else {\n dest.write(chunk);\n }\n });\n\n this.on(\"end\", async () => {\n if (dest instanceof Node18UniversalStreamWrapper) {\n dest._end();\n } else if (dest instanceof WritableStream) {\n const writer = dest.getWriter();\n writer.close();\n } else {\n dest.end();\n }\n });\n\n this.on(\"error\", async (error) => {\n if (dest instanceof Node18UniversalStreamWrapper) {\n dest._error(error);\n } else if (dest instanceof WritableStream) {\n const writer = dest.getWriter();\n writer.abort(error);\n } else {\n dest.destroy(error);\n }\n });\n\n this._startReading();\n\n return dest;\n }\n\n public pipeTo(\n dest: Node18UniversalStreamWrapper<ReadFormat> | Writable | WritableStream<ReadFormat>,\n ): Node18UniversalStreamWrapper<ReadFormat> | Writable | WritableStream<ReadFormat> {\n return this.pipe(dest);\n }\n\n public unpipe(dest: Node18UniversalStreamWrapper<ReadFormat> | Writable | WritableStream<ReadFormat>): void {\n this.off(\"data\", async (chunk) => {\n if (dest instanceof Node18UniversalStreamWrapper) {\n dest._write(chunk);\n } else if (dest instanceof WritableStream) {\n const writer = dest.getWriter();\n writer.write(chunk).then(() => writer.releaseLock());\n } else {\n dest.write(chunk);\n }\n });\n\n this.off(\"end\", async () => {\n if (dest instanceof Node18UniversalStreamWrapper) {\n dest._end();\n } else if (dest instanceof WritableStream) {\n const writer = dest.getWriter();\n writer.close();\n } else {\n dest.end();\n }\n });\n\n this.off(\"error\", async (error) => {\n if (dest instanceof Node18UniversalStreamWrapper) {\n dest._error(error);\n } else if (dest instanceof WritableStream) {\n const writer = dest.getWriter();\n writer.abort(error);\n } else {\n dest.destroy(error);\n }\n });\n }\n\n public destroy(error?: Error): void {\n this.reader\n .cancel(error)\n .then(() => {\n this._emit(\"close\");\n })\n .catch((err) => {\n this._emit(\"error\", err);\n });\n }\n\n public pause(): void {\n this.paused = true;\n this._emit(\"pause\");\n }\n\n public resume(): void {\n if (this.paused) {\n this.paused = false;\n this._emit(\"resume\");\n if (this.resumeCallback) {\n this.resumeCallback();\n this.resumeCallback = null;\n }\n }\n }\n\n public get isPaused(): boolean {\n return this.paused;\n }\n\n public async read(): Promise<ReadFormat | undefined> {\n if (this.paused) {\n await new Promise((resolve) => {\n this.resumeCallback = resolve;\n });\n }\n const { done, value } = await this.reader.read();\n\n if (done) {\n return undefined;\n }\n return value;\n }\n\n public setEncoding(encoding: string): void {\n this.encoding = encoding;\n }\n\n public async text(): Promise<string> {\n const chunks: ReadFormat[] = [];\n\n while (true) {\n const { done, value } = await this.reader.read();\n if (done) {\n break;\n }\n if (value) {\n chunks.push(value);\n }\n }\n\n const decoder = new TextDecoder(this.encoding || \"utf-8\");\n return decoder.decode(await new Blob(chunks).arrayBuffer());\n }\n\n public async json<T>(): Promise<T> {\n const text = await this.text();\n return JSON.parse(text);\n }\n\n private _write(chunk: ReadFormat): void {\n this._emit(\"data\", chunk);\n }\n\n private _end(): void {\n this._emit(\"end\");\n }\n\n private _error(error: any): void {\n this._emit(\"error\", error);\n }\n\n private _emit(event: string, data?: any): void {\n if (this.events[event]) {\n for (const callback of this.events[event] || []) {\n callback(data);\n }\n }\n }\n\n private async _startReading(): Promise<void> {\n try {\n this._emit(\"readable\");\n while (true) {\n if (this.paused) {\n await new Promise((resolve) => {\n this.resumeCallback = resolve;\n });\n }\n const { done, value } = await this.reader.read();\n if (done) {\n this._emit(\"end\");\n this._emit(\"close\");\n break;\n }\n if (value) {\n this._emit(\"data\", value);\n }\n }\n } catch (error) {\n this._emit(\"error\", error);\n }\n }\n\n [Symbol.asyncIterator](): AsyncIterableIterator<ReadFormat> {\n return {\n next: async () => {\n if (this.paused) {\n await new Promise((resolve) => {\n this.resumeCallback = resolve;\n });\n }\n const { done, value } = await this.reader.read();\n if (done) {\n return { done: true, value: undefined };\n }\n return { done: false, value };\n },\n [Symbol.asyncIterator]() {\n return this;\n },\n };\n }\n}\n"]}
|
|
@@ -1,89 +0,0 @@
|
|
|
1
|
-
import './chunk-PZ5AY32C.js';
|
|
2
|
-
|
|
3
|
-
// src/opik/rest_api/core/fetcher/stream-wrappers/NodePre18StreamWrapper.ts
|
|
4
|
-
var NodePre18StreamWrapper = class {
|
|
5
|
-
constructor(readableStream) {
|
|
6
|
-
this.readableStream = readableStream;
|
|
7
|
-
}
|
|
8
|
-
on(event, callback) {
|
|
9
|
-
this.readableStream.on(event, callback);
|
|
10
|
-
}
|
|
11
|
-
off(event, callback) {
|
|
12
|
-
this.readableStream.off(event, callback);
|
|
13
|
-
}
|
|
14
|
-
pipe(dest) {
|
|
15
|
-
this.readableStream.pipe(dest);
|
|
16
|
-
return dest;
|
|
17
|
-
}
|
|
18
|
-
pipeTo(dest) {
|
|
19
|
-
return this.pipe(dest);
|
|
20
|
-
}
|
|
21
|
-
unpipe(dest) {
|
|
22
|
-
if (dest) {
|
|
23
|
-
this.readableStream.unpipe(dest);
|
|
24
|
-
} else {
|
|
25
|
-
this.readableStream.unpipe();
|
|
26
|
-
}
|
|
27
|
-
}
|
|
28
|
-
destroy(error) {
|
|
29
|
-
this.readableStream.destroy(error);
|
|
30
|
-
}
|
|
31
|
-
pause() {
|
|
32
|
-
this.readableStream.pause();
|
|
33
|
-
}
|
|
34
|
-
resume() {
|
|
35
|
-
this.readableStream.resume();
|
|
36
|
-
}
|
|
37
|
-
get isPaused() {
|
|
38
|
-
return this.readableStream.isPaused();
|
|
39
|
-
}
|
|
40
|
-
async read() {
|
|
41
|
-
return new Promise((resolve, reject) => {
|
|
42
|
-
const chunk = this.readableStream.read();
|
|
43
|
-
if (chunk) {
|
|
44
|
-
resolve(chunk);
|
|
45
|
-
} else {
|
|
46
|
-
this.readableStream.once("readable", () => {
|
|
47
|
-
const chunk2 = this.readableStream.read();
|
|
48
|
-
resolve(chunk2);
|
|
49
|
-
});
|
|
50
|
-
this.readableStream.once("error", reject);
|
|
51
|
-
}
|
|
52
|
-
});
|
|
53
|
-
}
|
|
54
|
-
setEncoding(encoding) {
|
|
55
|
-
this.readableStream.setEncoding(encoding);
|
|
56
|
-
this.encoding = encoding;
|
|
57
|
-
}
|
|
58
|
-
async text() {
|
|
59
|
-
const chunks = [];
|
|
60
|
-
const encoder = new TextEncoder();
|
|
61
|
-
this.readableStream.setEncoding(this.encoding || "utf-8");
|
|
62
|
-
for await (const chunk of this.readableStream) {
|
|
63
|
-
chunks.push(encoder.encode(chunk));
|
|
64
|
-
}
|
|
65
|
-
const decoder = new TextDecoder(this.encoding || "utf-8");
|
|
66
|
-
return decoder.decode(Buffer.concat(chunks));
|
|
67
|
-
}
|
|
68
|
-
async json() {
|
|
69
|
-
const text = await this.text();
|
|
70
|
-
return JSON.parse(text);
|
|
71
|
-
}
|
|
72
|
-
[Symbol.asyncIterator]() {
|
|
73
|
-
const readableStream = this.readableStream;
|
|
74
|
-
const iterator = readableStream[Symbol.asyncIterator]();
|
|
75
|
-
return {
|
|
76
|
-
async next() {
|
|
77
|
-
const { value, done } = await iterator.next();
|
|
78
|
-
return { value, done };
|
|
79
|
-
},
|
|
80
|
-
[Symbol.asyncIterator]() {
|
|
81
|
-
return this;
|
|
82
|
-
}
|
|
83
|
-
};
|
|
84
|
-
}
|
|
85
|
-
};
|
|
86
|
-
|
|
87
|
-
export { NodePre18StreamWrapper };
|
|
88
|
-
//# sourceMappingURL=NodePre18StreamWrapper-TYJO3GAZ.js.map
|
|
89
|
-
//# sourceMappingURL=NodePre18StreamWrapper-TYJO3GAZ.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"sources":["../src/opik/rest_api/core/fetcher/stream-wrappers/NodePre18StreamWrapper.ts"],"names":["chunk"],"mappings":";;;AAIO,IAAM,yBAAN,MAAwE;AAAA,EAI3E,YAAY,cAA0B,EAAA;AAClC,IAAA,IAAA,CAAK,cAAiB,GAAA,cAAA;AAAA;AAC1B,EAEO,EAAA,CAAG,OAAe,QAA+B,EAAA;AACpD,IAAK,IAAA,CAAA,cAAA,CAAe,EAAG,CAAA,KAAA,EAAO,QAAQ,CAAA;AAAA;AAC1C,EAEO,GAAA,CAAI,OAAe,QAA+B,EAAA;AACrD,IAAK,IAAA,CAAA,cAAA,CAAe,GAAI,CAAA,KAAA,EAAO,QAAQ,CAAA;AAAA;AAC3C,EAEO,KAAK,IAA0B,EAAA;AAClC,IAAK,IAAA,CAAA,cAAA,CAAe,KAAK,IAAI,CAAA;AAC7B,IAAO,OAAA,IAAA;AAAA;AACX,EAEO,OAAO,IAA0B,EAAA;AACpC,IAAO,OAAA,IAAA,CAAK,KAAK,IAAI,CAAA;AAAA;AACzB,EAEO,OAAO,IAAuB,EAAA;AACjC,IAAA,IAAI,IAAM,EAAA;AACN,MAAK,IAAA,CAAA,cAAA,CAAe,OAAO,IAAI,CAAA;AAAA,KAC5B,MAAA;AACH,MAAA,IAAA,CAAK,eAAe,MAAO,EAAA;AAAA;AAC/B;AACJ,EAEO,QAAQ,KAAqB,EAAA;AAChC,IAAK,IAAA,CAAA,cAAA,CAAe,QAAQ,KAAK,CAAA;AAAA;AACrC,EAEO,KAAc,GAAA;AACjB,IAAA,IAAA,CAAK,eAAe,KAAM,EAAA;AAAA;AAC9B,EAEO,MAAe,GAAA;AAClB,IAAA,IAAA,CAAK,eAAe,MAAO,EAAA;AAAA;AAC/B,EAEA,IAAW,QAAoB,GAAA;AAC3B,IAAO,OAAA,IAAA,CAAK,eAAe,QAAS,EAAA;AAAA;AACxC,EAEA,MAAa,IAAoC,GAAA;AAC7C,IAAA,OAAO,IAAI,OAAA,CAAQ,CAAC,OAAA,EAAS,MAAW,KAAA;AACpC,MAAM,MAAA,KAAA,GAAQ,IAAK,CAAA,cAAA,CAAe,IAAK,EAAA;AACvC,MAAA,IAAI,KAAO,EAAA;AACP,QAAA,OAAA,CAAQ,KAAK,CAAA;AAAA,OACV,MAAA;AACH,QAAK,IAAA,CAAA,cAAA,CAAe,IAAK,CAAA,UAAA,EAAY,MAAM;AACvC,UAAMA,MAAAA,MAAAA,GAAQ,IAAK,CAAA,cAAA,CAAe,IAAK,EAAA;AACvC,UAAA,OAAA,CAAQA,MAAK,CAAA;AAAA,SAChB,CAAA;AACD,QAAK,IAAA,CAAA,cAAA,CAAe,IAAK,CAAA,OAAA,EAAS,MAAM,CAAA;AAAA;AAC5C,KACH,CAAA;AAAA;AACL,EAEO,YAAY,QAAyB,EAAA;AACxC,IAAK,IAAA,CAAA,cAAA,CAAe,YAAY,QAA0B,CAAA;AAC1D,IAAA,IAAA,CAAK,QAAW,GAAA,QAAA;AAAA;AACpB,EAEA,MAAa,IAAwB,GAAA;AACjC,IAAA,MAAM,SAAuB,EAAC;AAC9B,IAAM,MAAA,OAAA,GAAU,IAAI,WAAY,EAAA;AAChC,IAAA,IAAA,CAAK,cAAe,CAAA,WAAA,CAAa,IAAK,CAAA,QAAA,IAAY,OAA0B,CAAA;AAE5E,IAAiB,WAAA,MAAA,KAAA,IAAS,KAAK,cAAgB,EAAA;AAC3C,MAAA,MAAA,CAAO,IAAK,CAAA,OAAA,CAAQ,MAAO,CAAA,KAAK,CAAC,CAAA;AAAA;AAGrC,IAAA,MAAM,OAAU,GAAA,IAAI,WAAY,CAAA,IAAA,CAAK,YAAY,OAAO,CAAA;AACxD,IAAA,OAAO,OAAQ,CAAA,MAAA,CAAO,MAAO,CAAA,MAAA,CAAO,MAAM,CAAC,CAAA;AAAA;AAC/C,EAEA,MAAa,IAAsB,GAAA;AAC/B,IAAM,MAAA,IAAA,GAAO,MAAM,IAAA,CAAK,IAAK,EAAA;AAC7B,IAAO,OAAA,IAAA,CAAK,MAAM,IAAI,CAAA;AAAA;AAC1B,EAEA,CAAQ,MAAO,CAAA,aAAa,CAAmC,GAAA;AAC3D,IAAA,MAAM,iBAAiB,IAAK,CAAA,cAAA;AAC5B,IAAA,MAAM,QAAW,GAAA,cAAA,CAAe,MAAO,CAAA,aAAa,CAAE,EAAA;AAGtD,IAAO,OAAA;AAAA,MACH,MAAM,IAAwC,GAAA;AAC1C,QAAA,MAAM,EAAE,KAAO,EAAA,IAAA,EAAS,GAAA,MAAM,SAAS,IAAK,EAAA;AAC5C,QAAO,OAAA,EAAE,OAAwB,IAAK,EAAA;AAAA,OAC1C;AAAA,MACA,CAAC,MAAO,CAAA,aAAa,CAAI,GAAA;AACrB,QAAO,OAAA,IAAA;AAAA;AACX,KACJ;AAAA;AAER","file":"NodePre18StreamWrapper-TYJO3GAZ.js","sourcesContent":["import type { Readable, Writable } from \"readable-stream\";\n\nimport { EventCallback, StreamWrapper } from \"./chooseStreamWrapper\";\n\nexport class NodePre18StreamWrapper implements StreamWrapper<Writable, Buffer> {\n private readableStream: Readable;\n private encoding: string | undefined;\n\n constructor(readableStream: Readable) {\n this.readableStream = readableStream;\n }\n\n public on(event: string, callback: EventCallback): void {\n this.readableStream.on(event, callback);\n }\n\n public off(event: string, callback: EventCallback): void {\n this.readableStream.off(event, callback);\n }\n\n public pipe(dest: Writable): Writable {\n this.readableStream.pipe(dest);\n return dest;\n }\n\n public pipeTo(dest: Writable): Writable {\n return this.pipe(dest);\n }\n\n public unpipe(dest?: Writable): void {\n if (dest) {\n this.readableStream.unpipe(dest);\n } else {\n this.readableStream.unpipe();\n }\n }\n\n public destroy(error?: Error): void {\n this.readableStream.destroy(error);\n }\n\n public pause(): void {\n this.readableStream.pause();\n }\n\n public resume(): void {\n this.readableStream.resume();\n }\n\n public get isPaused(): boolean {\n return this.readableStream.isPaused();\n }\n\n public async read(): Promise<Buffer | undefined> {\n return new Promise((resolve, reject) => {\n const chunk = this.readableStream.read();\n if (chunk) {\n resolve(chunk);\n } else {\n this.readableStream.once(\"readable\", () => {\n const chunk = this.readableStream.read();\n resolve(chunk);\n });\n this.readableStream.once(\"error\", reject);\n }\n });\n }\n\n public setEncoding(encoding?: string): void {\n this.readableStream.setEncoding(encoding as BufferEncoding);\n this.encoding = encoding;\n }\n\n public async text(): Promise<string> {\n const chunks: Uint8Array[] = [];\n const encoder = new TextEncoder();\n this.readableStream.setEncoding((this.encoding || \"utf-8\") as BufferEncoding);\n\n for await (const chunk of this.readableStream) {\n chunks.push(encoder.encode(chunk));\n }\n\n const decoder = new TextDecoder(this.encoding || \"utf-8\");\n return decoder.decode(Buffer.concat(chunks));\n }\n\n public async json<T>(): Promise<T> {\n const text = await this.text();\n return JSON.parse(text);\n }\n\n public [Symbol.asyncIterator](): AsyncIterableIterator<Buffer> {\n const readableStream = this.readableStream;\n const iterator = readableStream[Symbol.asyncIterator]();\n\n // Create and return an async iterator that yields buffers\n return {\n async next(): Promise<IteratorResult<Buffer>> {\n const { value, done } = await iterator.next();\n return { value: value as Buffer, done };\n },\n [Symbol.asyncIterator]() {\n return this;\n },\n };\n }\n}\n"]}
|