opik 0.1.1 → 0.1.2
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-VaP_If_d.d.cts → Client-DilyWTdT.d.cts} +59 -11
- package/dist/{Client-VaP_If_d.d.ts → Client-DilyWTdT.d.ts} +59 -11
- package/dist/{chunk-OC5DO255.js → chunk-ZB4ZJRLY.js} +99 -4
- package/dist/chunk-ZB4ZJRLY.js.map +1 -0
- package/dist/index.cjs +97 -2
- package/dist/index.cjs.map +1 -1
- package/dist/index.d.cts +2 -2
- package/dist/index.d.ts +2 -2
- package/dist/index.js +1 -1
- package/dist/vercel/index.cjs +98 -3
- package/dist/vercel/index.cjs.map +1 -1
- package/dist/vercel/index.d.cts +1 -1
- package/dist/vercel/index.d.ts +1 -1
- package/dist/vercel/index.js +3 -3
- package/dist/vercel/index.js.map +1 -1
- package/package.json +9 -3
- package/dist/chunk-OC5DO255.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-DilyWTdT.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-DilyWTdT.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,4 +1,4 @@
|
|
|
1
|
-
import { OpikClient } from '../chunk-
|
|
1
|
+
import { OpikClient } from '../chunk-ZB4ZJRLY.js';
|
|
2
2
|
import '../chunk-PZ5AY32C.js';
|
|
3
3
|
|
|
4
4
|
// node_modules/@opentelemetry/core/build/esm/ExportResult.js
|
|
@@ -183,7 +183,7 @@ function hrTimeToMilliseconds(hrTime) {
|
|
|
183
183
|
function safeParseJson(value) {
|
|
184
184
|
try {
|
|
185
185
|
return JSON.parse(value);
|
|
186
|
-
} catch
|
|
186
|
+
} catch {
|
|
187
187
|
return value;
|
|
188
188
|
}
|
|
189
189
|
}
|
|
@@ -196,7 +196,7 @@ function tryParseJSON(input) {
|
|
|
196
196
|
if (parsed !== null && typeof parsed === "object" && !Array.isArray(parsed)) {
|
|
197
197
|
return parsed;
|
|
198
198
|
}
|
|
199
|
-
} catch
|
|
199
|
+
} catch {
|
|
200
200
|
return void 0;
|
|
201
201
|
}
|
|
202
202
|
return void 0;
|
package/dist/vercel/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../../node_modules/@opentelemetry/core/src/ExportResult.ts","../../src/opik/integrations/vercel/exporter.ts"],"names":["ExportResultCode","otelSpans","_a"],"mappings":";;;;AAqBA,IAAY,gBAAA;AAAZ,CAAA,SAAYA,iBAAgB,EAAA;AAC1B,EAAAA,iBAAAA,CAAAA,iBAAAA,CAAA,SAAA,CAAA,GAAA,CAAA,CAAA,GAAA,SAAA;AACA,EAAAA,iBAAAA,CAAAA,iBAAAA,CAAA,QAAA,CAAA,GAAA,CAAA,CAAA,GAAA,QAAA;AACF,CAHY,EAAA,gBAAA,KAAA,gBAAgB,GAAA,EAAA,CAAA,CAAA;;;ACE5B,IAAM,WAAA,GAAc,IAAI,UAAK,EAAA;AAEtB,IAAM,eAAN,MAA2C;AAAA,EAOhD,YAAY,EAAE,MAAA,GAAS,WAAY,EAAA,GAAuB,EAAI,EAAA;AAL9D;AAAA,IAAQ,IAAA,CAAA,MAAA,uBAAa,GAAmB,EAAA;AAExC;AAAA,IAAQ,IAAA,CAAA,KAAA,uBAAY,GAAkB,EAAA;AAOtC,IAAQ,IAAA,CAAA,YAAA,GAAe,CAAC,QAAoD,KAAA;AAC1E,MAAA,IAAI,QAAiC,EAAC;AACtC,MAAM,MAAA,EAAE,YAAe,GAAA,QAAA;AACvB,MAAM,MAAA,aAAA,GAAgB,MAAO,CAAA,IAAA,CAAK,UAAU,CAAA;AAE5C,MAAc,aAAA,CAAA,OAAA,CAAQ,CAAC,GAAQ,KAAA;AAC7B,QAAI,IAAA,GAAA,KAAQ,WAAe,IAAA,GAAA,KAAQ,gBAAkB,EAAA;AACnD,UAAA,MAAM,WAAc,GAAA,YAAA,CAAa,UAAW,CAAA,GAAG,CAAC,CAAA;AAEhD,UAAA,IAAI,WAAa,EAAA;AACf,YAAA,KAAA,GAAQ,EAAE,GAAG,KAAO,EAAA,GAAG,WAAY,EAAA;AAAA;AACrC;AAGF,QAAI,IAAA,GAAA,CAAI,UAAW,CAAA,YAAY,CAAG,EAAA;AAChC,UAAA,MAAM,SAAY,GAAA,GAAA,CAAI,OAAQ,CAAA,YAAA,EAAc,EAAE,CAAA;AAE9C,UAAA,KAAA,CAAM,SAAS,CAAA,GAAI,aAAc,CAAA,UAAA,CAAW,GAAG,CAAC,CAAA;AAAA;AAGlD,QAAI,IAAA,GAAA,CAAI,UAAW,CAAA,iBAAiB,CAAG,EAAA;AACrC,UAAA,MAAM,SAAY,GAAA,GAAA,CAAI,OAAQ,CAAA,iBAAA,EAAmB,EAAE,CAAA;AAEnD,UAAA,KAAA,CAAM,SAAS,CAAA,GAAI,aAAc,CAAA,UAAA,CAAW,GAAG,CAAC,CAAA;AAAA;AAClD,OACD,CAAA;AAED,MAAA,IAAI,MAAO,CAAA,IAAA,CAAK,KAAK,CAAA,CAAE,SAAS,CAAG,EAAA;AACjC,QAAO,OAAA,KAAA;AAAA;AAGT,MAAA,IAAI,sBAAsB,UAAY,EAAA;AACpC,QAAM,KAAA,CAAA,QAAA,GAAW,WAAW,kBAAkB,CAAA;AAAA;AAGhD,MAAA,IAAI,sBAAsB,UAAY,EAAA;AACpC,QAAM,KAAA,CAAA,IAAA,GAAO,WAAW,kBAAkB,CAAA;AAAA;AAG5C,MAAO,OAAA,KAAA;AAAA,KACT;AAEA,IAAQ,IAAA,CAAA,aAAA,GAAgB,CAAC,QAAoD,KAAA;AAC3E,MAAM,MAAA,EAAE,YAAe,GAAA,QAAA;AAEvB,MAAI,IAAA,UAAA,CAAW,kBAAkB,CAAG,EAAA;AAClC,QAAA,OAAO,EAAE,IAAA,EAAM,UAAW,CAAA,kBAAkB,CAAE,EAAA;AAAA;AAGhD,MAAI,IAAA,UAAA,CAAW,oBAAoB,CAAG,EAAA;AACpC,QAAA,OAAO,EAAE,MAAA,EAAQ,UAAW,CAAA,oBAAoB,CAAE,EAAA;AAAA;AAGpD,MAAI,IAAA,UAAA,CAAW,uBAAuB,CAAG,EAAA;AACvC,QAAA,OAAO,EAAE,SAAW,EAAA,aAAA,CAAc,UAAW,CAAA,uBAAuB,CAAC,CAAE,EAAA;AAAA;AAGzE,MAAA,OAAO,EAAC;AAAA,KACV;AAEA,IAAQ,IAAA,CAAA,eAAA,GAAkB,CACxB,QAC4B,KAAA;AAC5B,MAAM,MAAA,EAAE,YAAe,GAAA,QAAA;AACvB,MAAA,MAAM,WAAoC,EAAC;AAE3C,MAAI,IAAA,UAAA,CAAW,uBAAuB,CAAG,EAAA;AACvC,QAAS,QAAA,CAAA,KAAA,GAAQ,WAAW,uBAAuB,CAAA;AAAA;AAGrD,MAAI,IAAA,UAAA,CAAW,eAAe,CAAG,EAAA;AAC/B,QAAS,QAAA,CAAA,MAAA,GAAS,WAAW,eAAe,CAAA;AAAA;AAG9C,MAAO,OAAA,QAAA;AAAA,KACT;AAEA,IAAQ,IAAA,CAAA,YAAA,GAAe,CAAC,QAAmD,KAAA;AACzE,MAAM,MAAA,EAAE,YAAe,GAAA,QAAA;AACvB,MAAA,MAAM,QAAgC,EAAC;AAGvC,MAAA,IAAI,2BAA2B,UAAY,EAAA;AACzC,QAAM,KAAA,CAAA,aAAA,GAAgB,WAAW,uBAAuB,CAAA;AAAA;AAE1D,MAAA,IAAI,+BAA+B,UAAY,EAAA;AAC7C,QAAM,KAAA,CAAA,aAAA,GAAgB,WAAW,2BAA2B,CAAA;AAAA;AAI9D,MAAA,IAAI,+BAA+B,UAAY,EAAA;AAC7C,QAAM,KAAA,CAAA,iBAAA,GAAoB,WACxB,2BACF,CAAA;AAAA;AAEF,MAAA,IAAI,gCAAgC,UAAY,EAAA;AAC9C,QAAM,KAAA,CAAA,iBAAA,GAAoB,WACxB,4BACF,CAAA;AAAA;AAGF,MAAI,IAAA,eAAA,IAAmB,KAAS,IAAA,mBAAA,IAAuB,KAAO,EAAA;AAC5D,QAAA,KAAA,CAAM,YACH,GAAA,CAAA,KAAA,CAAM,aAAiB,IAAA,CAAA,KAAM,MAAM,iBAAqB,IAAA,CAAA,CAAA;AAAA;AAG7D,MAAO,OAAA,KAAA;AAAA,KACT;AAEA,IAAA,IAAA,CAAA,WAAA,GAAc,CAAC;AAAA,MACb,QAAA;AAAA,MACA,UAAA;AAAA,MACA;AAAA,KAKU,KAAA;AACV,MAAA,OAAO,MAAM,IAAK,CAAA;AAAA,QAChB,MAAM,QAAS,CAAA,IAAA;AAAA,QACf,WAAW,IAAI,IAAA,CAAK,oBAAqB,CAAA,QAAA,CAAS,SAAS,CAAC,CAAA;AAAA,QAC5D,SAAS,IAAI,IAAA,CAAK,oBAAqB,CAAA,QAAA,CAAS,OAAO,CAAC,CAAA;AAAA,QACxD,YAAA,EAAc,yCAAY,IAAK,CAAA,EAAA;AAAA,QAC/B,KAAA,EAAO,IAAK,CAAA,YAAA,CAAa,QAAQ,CAAA;AAAA,QACjC,MAAA,EAAQ,IAAK,CAAA,aAAA,CAAc,QAAQ,CAAA;AAAA,QACnC,QAAA,EAAU,IAAK,CAAA,eAAA,CAAgB,QAAQ,CAAA;AAAA,QACvC,KAAA,EAAO,IAAK,CAAA,YAAA,CAAa,QAAQ,CAAA;AAAA,QACjC,IAAM,EAAA;AAAA,OACP,CAAA;AAAA,KACH;AAEA,IAAA,IAAA,CAAA,QAAA,GAAW,YAA2B;AACpC,MAAM,MAAA,IAAA,CAAK,OAAO,KAAM,EAAA;AAAA,KAC1B;AAEA,IAAA,IAAA,CAAA,UAAA,GAAa,YAA2B;AACtC,MAAM,MAAA,IAAA,CAAK,OAAO,KAAM,EAAA;AAAA,KAC1B;AAEA,IAAyB,IAAA,CAAA,MAAA,GAAA,CAAC,WAAW,cAAmB,KAAA;AACtD,MAAM,MAAA,UAAA,GAAa,sBAAsB,SAAS,CAAA;AAElD,MAAO,MAAA,CAAA,OAAA,CAAQ,UAAU,CAAE,CAAA,OAAA,CAAQ,CAAC,CAAC,WAAA,EAAaC,UAAS,CAAM,KAAA;AAlLrE,QAAA,IAAA,EAAA,EAAA,EAAA;AAmLM,QAAA,MAAM,CAAC,YAAA,EAAc,GAAG,cAAc,CAAIA,GAAAA,UAAAA;AAC1C,QAAM,MAAA,KAAA,GAAQ,IAAK,CAAA,MAAA,CAAO,KAAM,CAAA;AAAA,UAC9B,WAAW,IAAI,IAAA,CAAK,oBAAqB,CAAA,YAAA,CAAa,SAAS,CAAC,CAAA;AAAA,UAChE,SAAS,IAAI,IAAA,CAAK,oBAAqB,CAAA,YAAA,CAAa,OAAO,CAAC,CAAA;AAAA,UAC5D,IAAA,EAAA,CACE,wBAAa,UACX,CAAA,iCACF,MAFA,IAEG,GAAA,KAAA,CAAA,GAAA,EAAA,CAAA,QAAA,EAAA,KAFH,YAEiB,YAAa,CAAA,IAAA;AAAA,UAChC,KAAA,EAAO,IAAK,CAAA,YAAA,CAAa,YAAY,CAAA;AAAA,UACrC,MAAA,EAAQ,IAAK,CAAA,aAAA,CAAc,YAAY,CAAA;AAAA,UACvC,QAAA,EAAU,IAAK,CAAA,eAAA,CAAgB,YAAY,CAAA;AAAA,UAC3C,KAAA,EAAO,IAAK,CAAA,YAAA,CAAa,YAAY;AAAA,SACtC,CAAA;AAED,QAAK,IAAA,CAAA,MAAA,CAAO,GAAI,CAAA,WAAA,EAAa,KAAK,CAAA;AAElC,QAAe,cAAA,CAAA,OAAA,CAAQ,CAAC,QAAa,KAAA;AAnM3C,UAAAC,IAAAA,GAAAA;AAoMQ,UAAM,MAAA,UAAA,GAAa,KAAK,KAAM,CAAA,GAAA,CAAA,CAAIA,MAAA,QAAS,CAAA,YAAA,KAAT,IAAAA,GAAAA,GAAAA,GAAyB,EAAE,CAAA;AAC7D,UAAA,MAAM,OAAO,IAAK,CAAA,WAAA,CAAY,EAAE,UAAY,EAAA,QAAA,EAAU,OAAO,CAAA;AAE7D,UAAA,IAAA,CAAK,MAAM,GAAI,CAAA,QAAA,CAAS,WAAY,EAAA,CAAE,QAAQ,IAAI,CAAA;AAAA,SACnD,CAAA;AAAA,OACF,CAAA;AAED,MAAA,cAAA,CAAe,EAAE,IAAA,EAAM,gBAAiB,CAAA,OAAA,EAAS,CAAA;AAAA,KACnD;AA3KE,IAAA,IAAA,CAAK,MAAS,GAAA,MAAA;AAAA;AAChB,EA4KA,OAAO,YAAY,QAAmD,EAAA;AA9MxE,IAAA,IAAA,EAAA,EAAA,EAAA,EAAA,EAAA;AA+MI,IAAA,MAAM,QAAW,GAAA,EAAE,GAAG,QAAA,CAAS,QAAS,EAAA;AAExC,IAAA,IAAI,SAAS,IAAM,EAAA;AACjB,MAAA,QAAA,CAAS,YAAY,QAAS,CAAA,IAAA;AAAA;AAGhC,IAAO,OAAA;AAAA,MACL,SAAA,EAAA,CAAW,EAAS,GAAA,QAAA,CAAA,SAAA,KAAT,IAAsB,GAAA,EAAA,GAAA,IAAA;AAAA,MACjC,YAAA,EAAA,CAAc,EAAS,GAAA,QAAA,CAAA,YAAA,KAAT,IAAyB,GAAA,EAAA,GAAA,IAAA;AAAA,MACvC,aAAA,EAAA,CAAe,EAAS,GAAA,QAAA,CAAA,aAAA,KAAT,IAA0B,GAAA,EAAA,GAAA,IAAA;AAAA,MACzC,YAAY,QAAS,CAAA,UAAA;AAAA,MACrB;AAAA,KACF;AAAA;AAEJ;AAEA,SAAS,sBACP,SACgC,EAAA;AAChC,EAAA,MAAM,sBAAsD,EAAC;AAE7D,EAAU,SAAA,CAAA,OAAA,CAAQ,CAAC,QAAa,KAAA;AAC9B,IAAM,MAAA,OAAA,GAAU,SAAS,WAAY,EAAA;AAErC,IAAA,IAAI,CAAC,mBAAA,CAAoB,OAAQ,CAAA,OAAO,CAAG,EAAA;AACzC,MAAoB,mBAAA,CAAA,OAAA,CAAQ,OAAO,CAAA,GAAI,EAAC;AAAA;AAG1C,IAAA,mBAAA,CAAoB,OAAQ,CAAA,OAAO,CAAE,CAAA,IAAA,CAAK,QAAQ,CAAA;AAAA,GACnD,CAAA;AAED,EAAA,MAAA,CAAO,MAAO,CAAA,mBAAmB,CAAE,CAAA,OAAA,CAAQ,CAACD,UAAc,KAAA;AACxD,IAAAA,UAAU,CAAA,IAAA,CAAK,CAAC,CAAA,EAAG,CAAM,KAAA;AAEvB,MAAA,IAAI,CAAC,CAAA,CAAE,YAAgB,IAAA,CAAA,CAAE,cAAqB,OAAA,CAAA,CAAA;AAC9C,MAAA,IAAI,CAAE,CAAA,YAAA,IAAgB,CAAC,CAAA,CAAE,cAAqB,OAAA,CAAA;AAG9C,MAAI,IAAA,CAAA,CAAE,YAAgB,IAAA,CAAA,CAAE,YAAc,EAAA;AACpC,QAAA,IAAI,EAAE,YAAiB,KAAA,CAAA,CAAE,WAAY,EAAA,CAAE,QAAe,OAAA,CAAA;AACtD,QAAA,IAAI,EAAE,YAAiB,KAAA,CAAA,CAAE,WAAY,EAAA,CAAE,QAAe,OAAA,CAAA,CAAA;AAAA;AAIxD,MAAO,OAAA,CAAA;AAAA,KACR,CAAA;AAAA,GACF,CAAA;AAED,EAAO,OAAA,mBAAA;AACT;AAGA,SAAS,qBAAqB,MAA0B,EAAA;AACtD,EAAA,OAAO,OAAO,CAAC,CAAA,GAAI,GAAM,GAAA,MAAA,CAAO,CAAC,CAAI,GAAA,GAAA;AACvC;AAEA,SAAS,cAAc,KAAyB,EAAA;AAC9C,EAAI,IAAA;AACF,IAAO,OAAA,IAAA,CAAK,MAAM,KAAe,CAAA;AAAA,WAC1B,CAAG,EAAA;AACV,IAAO,OAAA,KAAA;AAAA;AAEX;AAEA,SAAS,aAAa,KAAqD,EAAA;AACzE,EAAI,IAAA,OAAO,UAAU,QAAU,EAAA;AAC7B,IAAO,OAAA,KAAA,CAAA;AAAA;AAGT,EAAI,IAAA;AACF,IAAM,MAAA,MAAA,GAAS,IAAK,CAAA,KAAA,CAAM,KAAK,CAAA;AAE/B,IACE,IAAA,MAAA,KAAW,QACX,OAAO,MAAA,KAAW,YAClB,CAAC,KAAA,CAAM,OAAQ,CAAA,MAAM,CACrB,EAAA;AACA,MAAO,OAAA,MAAA;AAAA;AACT,WACO,KAAO,EAAA;AACd,IAAO,OAAA,KAAA,CAAA;AAAA;AAGT,EAAO,OAAA,KAAA,CAAA;AACT","file":"index.js","sourcesContent":["/*\n * Copyright The OpenTelemetry Authors\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * https://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nexport interface ExportResult {\n code: ExportResultCode;\n error?: Error;\n}\n\nexport enum ExportResultCode {\n SUCCESS,\n FAILED,\n}\n","import { Opik } from \"opik\";\nimport type { Span, Trace } from \"opik\";\nimport { AttributeValue, Tracer } from \"@opentelemetry/api\";\nimport { ExportResultCode } from \"@opentelemetry/core\";\nimport { NodeSDKConfiguration } from \"@opentelemetry/sdk-node\";\n\ntype SpanExporter = NodeSDKConfiguration[\"traceExporter\"];\ntype ExportFunction = SpanExporter[\"export\"];\ntype ReadableSpan = Parameters<ExportFunction>[0][0];\n\ntype TelemetrySettings = {\n isEnabled?: boolean;\n recordInputs?: boolean;\n recordOutputs?: boolean;\n functionId?: string;\n metadata?: Record<string, AttributeValue>;\n tracer?: Tracer;\n};\n\ntype OpikExporterSettings = TelemetrySettings & {\n name?: string;\n};\n\nconst aiSDKClient = new Opik();\n\nexport class OpikExporter implements SpanExporter {\n // <otelTraceId, opikTrace>\n private traces = new Map<string, Trace>();\n // <otelSpanId, opikSpan>\n private spans = new Map<string, Span>();\n private client: Opik;\n\n constructor({ client = aiSDKClient }: { client?: Opik } = {}) {\n this.client = client;\n }\n\n private getSpanInput = (otelSpan: ReadableSpan): Record<string, unknown> => {\n let input: Record<string, unknown> = {};\n const { attributes } = otelSpan;\n const attributeKeys = Object.keys(attributes);\n\n attributeKeys.forEach((key) => {\n if (key === \"ai.prompt\" || key === \"gen_ai.request\") {\n const parsedValue = tryParseJSON(attributes[key]);\n\n if (parsedValue) {\n input = { ...input, ...parsedValue };\n }\n }\n\n if (key.startsWith(\"ai.prompt.\")) {\n const promptKey = key.replace(\"ai.prompt.\", \"\");\n\n input[promptKey] = safeParseJson(attributes[key]);\n }\n\n if (key.startsWith(\"gen_ai.request.\")) {\n const promptKey = key.replace(\"gen_ai.request.\", \"\");\n\n input[promptKey] = safeParseJson(attributes[key]);\n }\n });\n\n if (Object.keys(input).length > 0) {\n return input;\n }\n\n if (\"ai.toolCall.name\" in attributes) {\n input.toolName = attributes[\"ai.toolCall.name\"];\n }\n\n if (\"ai.toolCall.args\" in attributes) {\n input.args = attributes[\"ai.toolCall.args\"];\n }\n\n return input;\n };\n\n private getSpanOutput = (otelSpan: ReadableSpan): Record<string, unknown> => {\n const { attributes } = otelSpan;\n\n if (attributes[\"ai.response.text\"]) {\n return { text: attributes[\"ai.response.text\"] };\n }\n\n if (attributes[\"ai.toolCall.result\"]) {\n return { result: attributes[\"ai.toolCall.result\"] };\n }\n\n if (attributes[\"ai.response.toolCalls\"]) {\n return { toolCalls: safeParseJson(attributes[\"ai.response.toolCalls\"]) };\n }\n\n return {};\n };\n\n private getSpanMetadata = (\n otelSpan: ReadableSpan\n ): Record<string, unknown> => {\n const { attributes } = otelSpan;\n const metadata: Record<string, unknown> = {};\n\n if (attributes[\"gen_ai.response.model\"]) {\n metadata.model = attributes[\"gen_ai.response.model\"];\n }\n\n if (attributes[\"gen_ai.system\"]) {\n metadata.system = attributes[\"gen_ai.system\"];\n }\n\n return metadata;\n };\n\n private getSpanUsage = (otelSpan: ReadableSpan): Record<string, number> => {\n const { attributes } = otelSpan;\n const usage: Record<string, number> = {};\n\n // prompt tokens\n if (\"ai.usage.promptTokens\" in attributes) {\n usage.prompt_tokens = attributes[\"ai.usage.promptTokens\"] as number;\n }\n if (\"gen_ai.usage.input_tokens\" in attributes) {\n usage.prompt_tokens = attributes[\"gen_ai.usage.input_tokens\"] as number;\n }\n\n // completion tokens\n if (\"ai.usage.completionTokens\" in attributes) {\n usage.completion_tokens = attributes[\n \"ai.usage.completionTokens\"\n ] as number;\n }\n if (\"gen_ai.usage.output_tokens\" in attributes) {\n usage.completion_tokens = attributes[\n \"gen_ai.usage.output_tokens\"\n ] as number;\n }\n\n if (\"prompt_tokens\" in usage || \"completion_tokens\" in usage) {\n usage.total_tokens =\n (usage.prompt_tokens || 0) + (usage.completion_tokens || 0);\n }\n\n return usage;\n };\n\n processSpan = ({\n otelSpan,\n parentSpan,\n trace,\n }: {\n otelSpan: ReadableSpan;\n parentSpan?: Span;\n trace: Trace;\n }): Span => {\n return trace.span({\n name: otelSpan.name,\n startTime: new Date(hrTimeToMilliseconds(otelSpan.startTime)),\n endTime: new Date(hrTimeToMilliseconds(otelSpan.endTime)),\n parentSpanId: parentSpan?.data.id,\n input: this.getSpanInput(otelSpan),\n output: this.getSpanOutput(otelSpan),\n metadata: this.getSpanMetadata(otelSpan),\n usage: this.getSpanUsage(otelSpan),\n type: \"llm\",\n });\n };\n\n shutdown = async (): Promise<void> => {\n await this.client.flush();\n };\n\n forceFlush = async (): Promise<void> => {\n await this.client.flush();\n };\n\n export: ExportFunction = (otelSpans, resultCallback) => {\n const spanGroups = groupAndSortOtelSpans(otelSpans);\n\n Object.entries(spanGroups).forEach(([otelTraceId, otelSpans]) => {\n const [rootOtelSpan, ...otherOtelSpans] = otelSpans;\n const trace = this.client.trace({\n startTime: new Date(hrTimeToMilliseconds(rootOtelSpan.startTime)),\n endTime: new Date(hrTimeToMilliseconds(rootOtelSpan.endTime)),\n name:\n rootOtelSpan.attributes[\n \"ai.telemetry.metadata.traceName\"\n ]?.toString() ?? rootOtelSpan.name,\n input: this.getSpanInput(rootOtelSpan),\n output: this.getSpanOutput(rootOtelSpan),\n metadata: this.getSpanMetadata(rootOtelSpan),\n usage: this.getSpanUsage(rootOtelSpan),\n });\n\n this.traces.set(otelTraceId, trace);\n\n otherOtelSpans.forEach((otelSpan) => {\n const parentSpan = this.spans.get(otelSpan.parentSpanId ?? \"\");\n const span = this.processSpan({ parentSpan, otelSpan, trace });\n\n this.spans.set(otelSpan.spanContext().spanId, span);\n });\n });\n\n resultCallback({ code: ExportResultCode.SUCCESS });\n };\n\n static getSettings(settings: OpikExporterSettings): TelemetrySettings {\n const metadata = { ...settings.metadata };\n\n if (settings.name) {\n metadata.traceName = settings.name;\n }\n\n return {\n isEnabled: settings.isEnabled ?? true,\n recordInputs: settings.recordInputs ?? true,\n recordOutputs: settings.recordOutputs ?? true,\n functionId: settings.functionId,\n metadata,\n };\n }\n}\n\nfunction groupAndSortOtelSpans(\n otelSpans: ReadableSpan[]\n): Record<string, ReadableSpan[]> {\n const spanGroupsByTraceId: Record<string, ReadableSpan[]> = {};\n\n otelSpans.forEach((otelSpan) => {\n const context = otelSpan.spanContext();\n\n if (!spanGroupsByTraceId[context.traceId]) {\n spanGroupsByTraceId[context.traceId] = [];\n }\n\n spanGroupsByTraceId[context.traceId].push(otelSpan);\n });\n\n Object.values(spanGroupsByTraceId).forEach((otelSpans) => {\n otelSpans.sort((a, b) => {\n // Put spans without parent first\n if (!a.parentSpanId && b.parentSpanId) return -1;\n if (a.parentSpanId && !b.parentSpanId) return 1;\n\n // If both have parents, check if one is parent of the other\n if (a.parentSpanId && b.parentSpanId) {\n if (a.parentSpanId === b.spanContext().spanId) return 1;\n if (b.parentSpanId === a.spanContext().spanId) return -1;\n }\n\n // If no parent relationship, maintain relative order\n return 0;\n });\n });\n\n return spanGroupsByTraceId;\n}\n\n// Convert hrTime ([seconds, nanoseconds]) to milliseconds\nfunction hrTimeToMilliseconds(hrTime: [number, number]) {\n return hrTime[0] * 1e3 + hrTime[1] / 1e6;\n}\n\nfunction safeParseJson(value: unknown): unknown {\n try {\n return JSON.parse(value as string) as Record<string, unknown>;\n } catch (e) {\n return value;\n }\n}\n\nfunction tryParseJSON(input: unknown): Record<string, unknown> | undefined {\n if (typeof input !== \"string\") {\n return undefined;\n }\n\n try {\n const parsed = JSON.parse(input);\n\n if (\n parsed !== null &&\n typeof parsed === \"object\" &&\n !Array.isArray(parsed)\n ) {\n return parsed as Record<string, unknown>;\n }\n } catch (error) {\n return undefined;\n }\n\n return undefined;\n}\n"]}
|
|
1
|
+
{"version":3,"sources":["../../node_modules/@opentelemetry/core/src/ExportResult.ts","../../src/opik/integrations/vercel/exporter.ts"],"names":["ExportResultCode","otelSpans","_a"],"mappings":";;;;AAqBA,IAAY,gBAAA;AAAZ,CAAA,SAAYA,iBAAgB,EAAA;AAC1B,EAAAA,iBAAAA,CAAAA,iBAAAA,CAAA,SAAA,CAAA,GAAA,CAAA,CAAA,GAAA,SAAA;AACA,EAAAA,iBAAAA,CAAAA,iBAAAA,CAAA,QAAA,CAAA,GAAA,CAAA,CAAA,GAAA,QAAA;AACF,CAHY,EAAA,gBAAA,KAAA,gBAAgB,GAAA,EAAA,CAAA,CAAA;;;ACE5B,IAAM,WAAA,GAAc,IAAI,UAAK,EAAA;AAEtB,IAAM,eAAN,MAA2C;AAAA,EAOhD,YAAY,EAAE,MAAA,GAAS,WAAY,EAAA,GAAuB,EAAI,EAAA;AAL9D;AAAA,IAAQ,IAAA,CAAA,MAAA,uBAAa,GAAmB,EAAA;AAExC;AAAA,IAAQ,IAAA,CAAA,KAAA,uBAAY,GAAkB,EAAA;AAOtC,IAAQ,IAAA,CAAA,YAAA,GAAe,CAAC,QAAoD,KAAA;AAC1E,MAAA,IAAI,QAAiC,EAAC;AACtC,MAAM,MAAA,EAAE,YAAe,GAAA,QAAA;AACvB,MAAM,MAAA,aAAA,GAAgB,MAAO,CAAA,IAAA,CAAK,UAAU,CAAA;AAE5C,MAAc,aAAA,CAAA,OAAA,CAAQ,CAAC,GAAQ,KAAA;AAC7B,QAAI,IAAA,GAAA,KAAQ,WAAe,IAAA,GAAA,KAAQ,gBAAkB,EAAA;AACnD,UAAA,MAAM,WAAc,GAAA,YAAA,CAAa,UAAW,CAAA,GAAG,CAAC,CAAA;AAEhD,UAAA,IAAI,WAAa,EAAA;AACf,YAAA,KAAA,GAAQ,EAAE,GAAG,KAAO,EAAA,GAAG,WAAY,EAAA;AAAA;AACrC;AAGF,QAAI,IAAA,GAAA,CAAI,UAAW,CAAA,YAAY,CAAG,EAAA;AAChC,UAAA,MAAM,SAAY,GAAA,GAAA,CAAI,OAAQ,CAAA,YAAA,EAAc,EAAE,CAAA;AAE9C,UAAA,KAAA,CAAM,SAAS,CAAA,GAAI,aAAc,CAAA,UAAA,CAAW,GAAG,CAAC,CAAA;AAAA;AAGlD,QAAI,IAAA,GAAA,CAAI,UAAW,CAAA,iBAAiB,CAAG,EAAA;AACrC,UAAA,MAAM,SAAY,GAAA,GAAA,CAAI,OAAQ,CAAA,iBAAA,EAAmB,EAAE,CAAA;AAEnD,UAAA,KAAA,CAAM,SAAS,CAAA,GAAI,aAAc,CAAA,UAAA,CAAW,GAAG,CAAC,CAAA;AAAA;AAClD,OACD,CAAA;AAED,MAAA,IAAI,MAAO,CAAA,IAAA,CAAK,KAAK,CAAA,CAAE,SAAS,CAAG,EAAA;AACjC,QAAO,OAAA,KAAA;AAAA;AAGT,MAAA,IAAI,sBAAsB,UAAY,EAAA;AACpC,QAAM,KAAA,CAAA,QAAA,GAAW,WAAW,kBAAkB,CAAA;AAAA;AAGhD,MAAA,IAAI,sBAAsB,UAAY,EAAA;AACpC,QAAM,KAAA,CAAA,IAAA,GAAO,WAAW,kBAAkB,CAAA;AAAA;AAG5C,MAAO,OAAA,KAAA;AAAA,KACT;AAEA,IAAQ,IAAA,CAAA,aAAA,GAAgB,CAAC,QAAoD,KAAA;AAC3E,MAAM,MAAA,EAAE,YAAe,GAAA,QAAA;AAEvB,MAAI,IAAA,UAAA,CAAW,kBAAkB,CAAG,EAAA;AAClC,QAAA,OAAO,EAAE,IAAA,EAAM,UAAW,CAAA,kBAAkB,CAAE,EAAA;AAAA;AAGhD,MAAI,IAAA,UAAA,CAAW,oBAAoB,CAAG,EAAA;AACpC,QAAA,OAAO,EAAE,MAAA,EAAQ,UAAW,CAAA,oBAAoB,CAAE,EAAA;AAAA;AAGpD,MAAI,IAAA,UAAA,CAAW,uBAAuB,CAAG,EAAA;AACvC,QAAA,OAAO,EAAE,SAAW,EAAA,aAAA,CAAc,UAAW,CAAA,uBAAuB,CAAC,CAAE,EAAA;AAAA;AAGzE,MAAA,OAAO,EAAC;AAAA,KACV;AAEA,IAAQ,IAAA,CAAA,eAAA,GAAkB,CACxB,QAC4B,KAAA;AAC5B,MAAM,MAAA,EAAE,YAAe,GAAA,QAAA;AACvB,MAAA,MAAM,WAAoC,EAAC;AAE3C,MAAI,IAAA,UAAA,CAAW,uBAAuB,CAAG,EAAA;AACvC,QAAS,QAAA,CAAA,KAAA,GAAQ,WAAW,uBAAuB,CAAA;AAAA;AAGrD,MAAI,IAAA,UAAA,CAAW,eAAe,CAAG,EAAA;AAC/B,QAAS,QAAA,CAAA,MAAA,GAAS,WAAW,eAAe,CAAA;AAAA;AAG9C,MAAO,OAAA,QAAA;AAAA,KACT;AAEA,IAAQ,IAAA,CAAA,YAAA,GAAe,CAAC,QAAmD,KAAA;AACzE,MAAM,MAAA,EAAE,YAAe,GAAA,QAAA;AACvB,MAAA,MAAM,QAAgC,EAAC;AAGvC,MAAA,IAAI,2BAA2B,UAAY,EAAA;AACzC,QAAM,KAAA,CAAA,aAAA,GAAgB,WAAW,uBAAuB,CAAA;AAAA;AAE1D,MAAA,IAAI,+BAA+B,UAAY,EAAA;AAC7C,QAAM,KAAA,CAAA,aAAA,GAAgB,WAAW,2BAA2B,CAAA;AAAA;AAI9D,MAAA,IAAI,+BAA+B,UAAY,EAAA;AAC7C,QAAM,KAAA,CAAA,iBAAA,GAAoB,WACxB,2BACF,CAAA;AAAA;AAEF,MAAA,IAAI,gCAAgC,UAAY,EAAA;AAC9C,QAAM,KAAA,CAAA,iBAAA,GAAoB,WACxB,4BACF,CAAA;AAAA;AAGF,MAAI,IAAA,eAAA,IAAmB,KAAS,IAAA,mBAAA,IAAuB,KAAO,EAAA;AAC5D,QAAA,KAAA,CAAM,YACH,GAAA,CAAA,KAAA,CAAM,aAAiB,IAAA,CAAA,KAAM,MAAM,iBAAqB,IAAA,CAAA,CAAA;AAAA;AAG7D,MAAO,OAAA,KAAA;AAAA,KACT;AAEA,IAAA,IAAA,CAAA,WAAA,GAAc,CAAC;AAAA,MACb,QAAA;AAAA,MACA,UAAA;AAAA,MACA;AAAA,KAKU,KAAA;AACV,MAAA,OAAO,MAAM,IAAK,CAAA;AAAA,QAChB,MAAM,QAAS,CAAA,IAAA;AAAA,QACf,WAAW,IAAI,IAAA,CAAK,oBAAqB,CAAA,QAAA,CAAS,SAAS,CAAC,CAAA;AAAA,QAC5D,SAAS,IAAI,IAAA,CAAK,oBAAqB,CAAA,QAAA,CAAS,OAAO,CAAC,CAAA;AAAA,QACxD,YAAA,EAAc,yCAAY,IAAK,CAAA,EAAA;AAAA,QAC/B,KAAA,EAAO,IAAK,CAAA,YAAA,CAAa,QAAQ,CAAA;AAAA,QACjC,MAAA,EAAQ,IAAK,CAAA,aAAA,CAAc,QAAQ,CAAA;AAAA,QACnC,QAAA,EAAU,IAAK,CAAA,eAAA,CAAgB,QAAQ,CAAA;AAAA,QACvC,KAAA,EAAO,IAAK,CAAA,YAAA,CAAa,QAAQ,CAAA;AAAA,QACjC,IAAM,EAAA;AAAA,OACP,CAAA;AAAA,KACH;AAEA,IAAA,IAAA,CAAA,QAAA,GAAW,YAA2B;AACpC,MAAM,MAAA,IAAA,CAAK,OAAO,KAAM,EAAA;AAAA,KAC1B;AAEA,IAAA,IAAA,CAAA,UAAA,GAAa,YAA2B;AACtC,MAAM,MAAA,IAAA,CAAK,OAAO,KAAM,EAAA;AAAA,KAC1B;AAEA,IAAyB,IAAA,CAAA,MAAA,GAAA,CAAC,WAAW,cAAmB,KAAA;AACtD,MAAM,MAAA,UAAA,GAAa,sBAAsB,SAAS,CAAA;AAElD,MAAO,MAAA,CAAA,OAAA,CAAQ,UAAU,CAAE,CAAA,OAAA,CAAQ,CAAC,CAAC,WAAA,EAAaC,UAAS,CAAM,KAAA;AAlLrE,QAAA,IAAA,EAAA,EAAA,EAAA;AAmLM,QAAA,MAAM,CAAC,YAAA,EAAc,GAAG,cAAc,CAAIA,GAAAA,UAAAA;AAC1C,QAAM,MAAA,KAAA,GAAQ,IAAK,CAAA,MAAA,CAAO,KAAM,CAAA;AAAA,UAC9B,WAAW,IAAI,IAAA,CAAK,oBAAqB,CAAA,YAAA,CAAa,SAAS,CAAC,CAAA;AAAA,UAChE,SAAS,IAAI,IAAA,CAAK,oBAAqB,CAAA,YAAA,CAAa,OAAO,CAAC,CAAA;AAAA,UAC5D,IAAA,EAAA,CACE,wBAAa,UACX,CAAA,iCACF,MAFA,IAEG,GAAA,KAAA,CAAA,GAAA,EAAA,CAAA,QAAA,EAAA,KAFH,YAEiB,YAAa,CAAA,IAAA;AAAA,UAChC,KAAA,EAAO,IAAK,CAAA,YAAA,CAAa,YAAY,CAAA;AAAA,UACrC,MAAA,EAAQ,IAAK,CAAA,aAAA,CAAc,YAAY,CAAA;AAAA,UACvC,QAAA,EAAU,IAAK,CAAA,eAAA,CAAgB,YAAY,CAAA;AAAA,UAC3C,KAAA,EAAO,IAAK,CAAA,YAAA,CAAa,YAAY;AAAA,SACtC,CAAA;AAED,QAAK,IAAA,CAAA,MAAA,CAAO,GAAI,CAAA,WAAA,EAAa,KAAK,CAAA;AAElC,QAAe,cAAA,CAAA,OAAA,CAAQ,CAAC,QAAa,KAAA;AAnM3C,UAAAC,IAAAA,GAAAA;AAoMQ,UAAM,MAAA,UAAA,GAAa,KAAK,KAAM,CAAA,GAAA,CAAA,CAAIA,MAAA,QAAS,CAAA,YAAA,KAAT,IAAAA,GAAAA,GAAAA,GAAyB,EAAE,CAAA;AAC7D,UAAA,MAAM,OAAO,IAAK,CAAA,WAAA,CAAY,EAAE,UAAY,EAAA,QAAA,EAAU,OAAO,CAAA;AAE7D,UAAA,IAAA,CAAK,MAAM,GAAI,CAAA,QAAA,CAAS,WAAY,EAAA,CAAE,QAAQ,IAAI,CAAA;AAAA,SACnD,CAAA;AAAA,OACF,CAAA;AAED,MAAA,cAAA,CAAe,EAAE,IAAA,EAAM,gBAAiB,CAAA,OAAA,EAAS,CAAA;AAAA,KACnD;AA3KE,IAAA,IAAA,CAAK,MAAS,GAAA,MAAA;AAAA;AAChB,EA4KA,OAAO,YAAY,QAAmD,EAAA;AA9MxE,IAAA,IAAA,EAAA,EAAA,EAAA,EAAA,EAAA;AA+MI,IAAA,MAAM,QAAW,GAAA,EAAE,GAAG,QAAA,CAAS,QAAS,EAAA;AAExC,IAAA,IAAI,SAAS,IAAM,EAAA;AACjB,MAAA,QAAA,CAAS,YAAY,QAAS,CAAA,IAAA;AAAA;AAGhC,IAAO,OAAA;AAAA,MACL,SAAA,EAAA,CAAW,EAAS,GAAA,QAAA,CAAA,SAAA,KAAT,IAAsB,GAAA,EAAA,GAAA,IAAA;AAAA,MACjC,YAAA,EAAA,CAAc,EAAS,GAAA,QAAA,CAAA,YAAA,KAAT,IAAyB,GAAA,EAAA,GAAA,IAAA;AAAA,MACvC,aAAA,EAAA,CAAe,EAAS,GAAA,QAAA,CAAA,aAAA,KAAT,IAA0B,GAAA,EAAA,GAAA,IAAA;AAAA,MACzC,YAAY,QAAS,CAAA,UAAA;AAAA,MACrB;AAAA,KACF;AAAA;AAEJ;AAEA,SAAS,sBACP,SACgC,EAAA;AAChC,EAAA,MAAM,sBAAsD,EAAC;AAE7D,EAAU,SAAA,CAAA,OAAA,CAAQ,CAAC,QAAa,KAAA;AAC9B,IAAM,MAAA,OAAA,GAAU,SAAS,WAAY,EAAA;AAErC,IAAA,IAAI,CAAC,mBAAA,CAAoB,OAAQ,CAAA,OAAO,CAAG,EAAA;AACzC,MAAoB,mBAAA,CAAA,OAAA,CAAQ,OAAO,CAAA,GAAI,EAAC;AAAA;AAG1C,IAAA,mBAAA,CAAoB,OAAQ,CAAA,OAAO,CAAE,CAAA,IAAA,CAAK,QAAQ,CAAA;AAAA,GACnD,CAAA;AAED,EAAA,MAAA,CAAO,MAAO,CAAA,mBAAmB,CAAE,CAAA,OAAA,CAAQ,CAACD,UAAc,KAAA;AACxD,IAAAA,UAAU,CAAA,IAAA,CAAK,CAAC,CAAA,EAAG,CAAM,KAAA;AAEvB,MAAA,IAAI,CAAC,CAAA,CAAE,YAAgB,IAAA,CAAA,CAAE,cAAqB,OAAA,CAAA,CAAA;AAC9C,MAAA,IAAI,CAAE,CAAA,YAAA,IAAgB,CAAC,CAAA,CAAE,cAAqB,OAAA,CAAA;AAG9C,MAAI,IAAA,CAAA,CAAE,YAAgB,IAAA,CAAA,CAAE,YAAc,EAAA;AACpC,QAAA,IAAI,EAAE,YAAiB,KAAA,CAAA,CAAE,WAAY,EAAA,CAAE,QAAe,OAAA,CAAA;AACtD,QAAA,IAAI,EAAE,YAAiB,KAAA,CAAA,CAAE,WAAY,EAAA,CAAE,QAAe,OAAA,CAAA,CAAA;AAAA;AAIxD,MAAO,OAAA,CAAA;AAAA,KACR,CAAA;AAAA,GACF,CAAA;AAED,EAAO,OAAA,mBAAA;AACT;AAGA,SAAS,qBAAqB,MAA0B,EAAA;AACtD,EAAA,OAAO,OAAO,CAAC,CAAA,GAAI,GAAM,GAAA,MAAA,CAAO,CAAC,CAAI,GAAA,GAAA;AACvC;AAEA,SAAS,cAAc,KAAyB,EAAA;AAC9C,EAAI,IAAA;AACF,IAAO,OAAA,IAAA,CAAK,MAAM,KAAe,CAAA;AAAA,GAC3B,CAAA,MAAA;AACN,IAAO,OAAA,KAAA;AAAA;AAEX;AAEA,SAAS,aAAa,KAAqD,EAAA;AACzE,EAAI,IAAA,OAAO,UAAU,QAAU,EAAA;AAC7B,IAAO,OAAA,KAAA,CAAA;AAAA;AAGT,EAAI,IAAA;AACF,IAAM,MAAA,MAAA,GAAS,IAAK,CAAA,KAAA,CAAM,KAAK,CAAA;AAE/B,IACE,IAAA,MAAA,KAAW,QACX,OAAO,MAAA,KAAW,YAClB,CAAC,KAAA,CAAM,OAAQ,CAAA,MAAM,CACrB,EAAA;AACA,MAAO,OAAA,MAAA;AAAA;AACT,GACM,CAAA,MAAA;AACN,IAAO,OAAA,KAAA,CAAA;AAAA;AAGT,EAAO,OAAA,KAAA,CAAA;AACT","file":"index.js","sourcesContent":["/*\n * Copyright The OpenTelemetry Authors\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n * https://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nexport interface ExportResult {\n code: ExportResultCode;\n error?: Error;\n}\n\nexport enum ExportResultCode {\n SUCCESS,\n FAILED,\n}\n","import { Opik } from \"opik\";\nimport type { Span, Trace } from \"opik\";\nimport { AttributeValue, Tracer } from \"@opentelemetry/api\";\nimport { ExportResultCode } from \"@opentelemetry/core\";\nimport { NodeSDKConfiguration } from \"@opentelemetry/sdk-node\";\n\ntype SpanExporter = NodeSDKConfiguration[\"traceExporter\"];\ntype ExportFunction = SpanExporter[\"export\"];\ntype ReadableSpan = Parameters<ExportFunction>[0][0];\n\ntype TelemetrySettings = {\n isEnabled?: boolean;\n recordInputs?: boolean;\n recordOutputs?: boolean;\n functionId?: string;\n metadata?: Record<string, AttributeValue>;\n tracer?: Tracer;\n};\n\ntype OpikExporterSettings = TelemetrySettings & {\n name?: string;\n};\n\nconst aiSDKClient = new Opik();\n\nexport class OpikExporter implements SpanExporter {\n // <otelTraceId, opikTrace>\n private traces = new Map<string, Trace>();\n // <otelSpanId, opikSpan>\n private spans = new Map<string, Span>();\n private client: Opik;\n\n constructor({ client = aiSDKClient }: { client?: Opik } = {}) {\n this.client = client;\n }\n\n private getSpanInput = (otelSpan: ReadableSpan): Record<string, unknown> => {\n let input: Record<string, unknown> = {};\n const { attributes } = otelSpan;\n const attributeKeys = Object.keys(attributes);\n\n attributeKeys.forEach((key) => {\n if (key === \"ai.prompt\" || key === \"gen_ai.request\") {\n const parsedValue = tryParseJSON(attributes[key]);\n\n if (parsedValue) {\n input = { ...input, ...parsedValue };\n }\n }\n\n if (key.startsWith(\"ai.prompt.\")) {\n const promptKey = key.replace(\"ai.prompt.\", \"\");\n\n input[promptKey] = safeParseJson(attributes[key]);\n }\n\n if (key.startsWith(\"gen_ai.request.\")) {\n const promptKey = key.replace(\"gen_ai.request.\", \"\");\n\n input[promptKey] = safeParseJson(attributes[key]);\n }\n });\n\n if (Object.keys(input).length > 0) {\n return input;\n }\n\n if (\"ai.toolCall.name\" in attributes) {\n input.toolName = attributes[\"ai.toolCall.name\"];\n }\n\n if (\"ai.toolCall.args\" in attributes) {\n input.args = attributes[\"ai.toolCall.args\"];\n }\n\n return input;\n };\n\n private getSpanOutput = (otelSpan: ReadableSpan): Record<string, unknown> => {\n const { attributes } = otelSpan;\n\n if (attributes[\"ai.response.text\"]) {\n return { text: attributes[\"ai.response.text\"] };\n }\n\n if (attributes[\"ai.toolCall.result\"]) {\n return { result: attributes[\"ai.toolCall.result\"] };\n }\n\n if (attributes[\"ai.response.toolCalls\"]) {\n return { toolCalls: safeParseJson(attributes[\"ai.response.toolCalls\"]) };\n }\n\n return {};\n };\n\n private getSpanMetadata = (\n otelSpan: ReadableSpan\n ): Record<string, unknown> => {\n const { attributes } = otelSpan;\n const metadata: Record<string, unknown> = {};\n\n if (attributes[\"gen_ai.response.model\"]) {\n metadata.model = attributes[\"gen_ai.response.model\"];\n }\n\n if (attributes[\"gen_ai.system\"]) {\n metadata.system = attributes[\"gen_ai.system\"];\n }\n\n return metadata;\n };\n\n private getSpanUsage = (otelSpan: ReadableSpan): Record<string, number> => {\n const { attributes } = otelSpan;\n const usage: Record<string, number> = {};\n\n // prompt tokens\n if (\"ai.usage.promptTokens\" in attributes) {\n usage.prompt_tokens = attributes[\"ai.usage.promptTokens\"] as number;\n }\n if (\"gen_ai.usage.input_tokens\" in attributes) {\n usage.prompt_tokens = attributes[\"gen_ai.usage.input_tokens\"] as number;\n }\n\n // completion tokens\n if (\"ai.usage.completionTokens\" in attributes) {\n usage.completion_tokens = attributes[\n \"ai.usage.completionTokens\"\n ] as number;\n }\n if (\"gen_ai.usage.output_tokens\" in attributes) {\n usage.completion_tokens = attributes[\n \"gen_ai.usage.output_tokens\"\n ] as number;\n }\n\n if (\"prompt_tokens\" in usage || \"completion_tokens\" in usage) {\n usage.total_tokens =\n (usage.prompt_tokens || 0) + (usage.completion_tokens || 0);\n }\n\n return usage;\n };\n\n processSpan = ({\n otelSpan,\n parentSpan,\n trace,\n }: {\n otelSpan: ReadableSpan;\n parentSpan?: Span;\n trace: Trace;\n }): Span => {\n return trace.span({\n name: otelSpan.name,\n startTime: new Date(hrTimeToMilliseconds(otelSpan.startTime)),\n endTime: new Date(hrTimeToMilliseconds(otelSpan.endTime)),\n parentSpanId: parentSpan?.data.id,\n input: this.getSpanInput(otelSpan),\n output: this.getSpanOutput(otelSpan),\n metadata: this.getSpanMetadata(otelSpan),\n usage: this.getSpanUsage(otelSpan),\n type: \"llm\",\n });\n };\n\n shutdown = async (): Promise<void> => {\n await this.client.flush();\n };\n\n forceFlush = async (): Promise<void> => {\n await this.client.flush();\n };\n\n export: ExportFunction = (otelSpans, resultCallback) => {\n const spanGroups = groupAndSortOtelSpans(otelSpans);\n\n Object.entries(spanGroups).forEach(([otelTraceId, otelSpans]) => {\n const [rootOtelSpan, ...otherOtelSpans] = otelSpans;\n const trace = this.client.trace({\n startTime: new Date(hrTimeToMilliseconds(rootOtelSpan.startTime)),\n endTime: new Date(hrTimeToMilliseconds(rootOtelSpan.endTime)),\n name:\n rootOtelSpan.attributes[\n \"ai.telemetry.metadata.traceName\"\n ]?.toString() ?? rootOtelSpan.name,\n input: this.getSpanInput(rootOtelSpan),\n output: this.getSpanOutput(rootOtelSpan),\n metadata: this.getSpanMetadata(rootOtelSpan),\n usage: this.getSpanUsage(rootOtelSpan),\n });\n\n this.traces.set(otelTraceId, trace);\n\n otherOtelSpans.forEach((otelSpan) => {\n const parentSpan = this.spans.get(otelSpan.parentSpanId ?? \"\");\n const span = this.processSpan({ parentSpan, otelSpan, trace });\n\n this.spans.set(otelSpan.spanContext().spanId, span);\n });\n });\n\n resultCallback({ code: ExportResultCode.SUCCESS });\n };\n\n static getSettings(settings: OpikExporterSettings): TelemetrySettings {\n const metadata = { ...settings.metadata };\n\n if (settings.name) {\n metadata.traceName = settings.name;\n }\n\n return {\n isEnabled: settings.isEnabled ?? true,\n recordInputs: settings.recordInputs ?? true,\n recordOutputs: settings.recordOutputs ?? true,\n functionId: settings.functionId,\n metadata,\n };\n }\n}\n\nfunction groupAndSortOtelSpans(\n otelSpans: ReadableSpan[]\n): Record<string, ReadableSpan[]> {\n const spanGroupsByTraceId: Record<string, ReadableSpan[]> = {};\n\n otelSpans.forEach((otelSpan) => {\n const context = otelSpan.spanContext();\n\n if (!spanGroupsByTraceId[context.traceId]) {\n spanGroupsByTraceId[context.traceId] = [];\n }\n\n spanGroupsByTraceId[context.traceId].push(otelSpan);\n });\n\n Object.values(spanGroupsByTraceId).forEach((otelSpans) => {\n otelSpans.sort((a, b) => {\n // Put spans without parent first\n if (!a.parentSpanId && b.parentSpanId) return -1;\n if (a.parentSpanId && !b.parentSpanId) return 1;\n\n // If both have parents, check if one is parent of the other\n if (a.parentSpanId && b.parentSpanId) {\n if (a.parentSpanId === b.spanContext().spanId) return 1;\n if (b.parentSpanId === a.spanContext().spanId) return -1;\n }\n\n // If no parent relationship, maintain relative order\n return 0;\n });\n });\n\n return spanGroupsByTraceId;\n}\n\n// Convert hrTime ([seconds, nanoseconds]) to milliseconds\nfunction hrTimeToMilliseconds(hrTime: [number, number]) {\n return hrTime[0] * 1e3 + hrTime[1] / 1e6;\n}\n\nfunction safeParseJson(value: unknown): unknown {\n try {\n return JSON.parse(value as string) as Record<string, unknown>;\n } catch {\n return value;\n }\n}\n\nfunction tryParseJSON(input: unknown): Record<string, unknown> | undefined {\n if (typeof input !== \"string\") {\n return undefined;\n }\n\n try {\n const parsed = JSON.parse(input);\n\n if (\n parsed !== null &&\n typeof parsed === \"object\" &&\n !Array.isArray(parsed)\n ) {\n return parsed as Record<string, unknown>;\n }\n } catch {\n return undefined;\n }\n\n return undefined;\n}\n"]}
|
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.1.
|
|
4
|
+
"version": "0.1.2",
|
|
5
5
|
"repository": {
|
|
6
6
|
"type": "git",
|
|
7
7
|
"url": "git+https://github.com/comet-ml/opik.git",
|
|
@@ -51,26 +51,31 @@
|
|
|
51
51
|
"scripts": {
|
|
52
52
|
"build": "tsup",
|
|
53
53
|
"watch": "tsup --watch",
|
|
54
|
-
"lint": "eslint '
|
|
54
|
+
"lint": "eslint '**/*.{ts,tsx}'",
|
|
55
|
+
"typecheck": "tsc --noEmit",
|
|
55
56
|
"format": "prettier --write 'src/**/*.{ts,tsx,js,jsx,json,md}'",
|
|
56
57
|
"test": "vitest"
|
|
57
58
|
},
|
|
58
59
|
"devDependencies": {
|
|
59
60
|
"@ai-sdk/openai": "^1.1.9",
|
|
61
|
+
"@eslint/js": "^9.20.0",
|
|
60
62
|
"@opentelemetry/api": "^1.9.0",
|
|
61
63
|
"@opentelemetry/auto-instrumentations-node": "^0.56.0",
|
|
62
64
|
"@opentelemetry/sdk-node": "^0.57.1",
|
|
63
65
|
"@types/node-fetch": "^2.6.12",
|
|
64
66
|
"@types/qs": "^6.9.17",
|
|
67
|
+
"@types/readable-stream": "^4.0.18",
|
|
65
68
|
"@types/url-join": "^4.0.3",
|
|
66
69
|
"@types/uuid": "^10.0.0",
|
|
67
70
|
"@typescript-eslint/eslint-plugin": "^8.16.0",
|
|
68
71
|
"@typescript-eslint/parser": "^8.16.0",
|
|
69
72
|
"ai": "^4.1.34",
|
|
70
|
-
"eslint": "^9.
|
|
73
|
+
"eslint": "^9.20.0",
|
|
74
|
+
"globals": "^15.14.0",
|
|
71
75
|
"prettier": "^3.4.1",
|
|
72
76
|
"tsup": "^8.3.5",
|
|
73
77
|
"typescript": "^5.7.2",
|
|
78
|
+
"typescript-eslint": "^8.24.0",
|
|
74
79
|
"vite-tsconfig-paths": "^5.1.3",
|
|
75
80
|
"vitest": "^3.0.5",
|
|
76
81
|
"zod": "^3.24.1"
|
|
@@ -81,6 +86,7 @@
|
|
|
81
86
|
"dependencies": {
|
|
82
87
|
"node-fetch": "^3.3.2",
|
|
83
88
|
"qs": "^6.13.1",
|
|
89
|
+
"readable-stream": "^4.7.0",
|
|
84
90
|
"url-join": "^5.0.0",
|
|
85
91
|
"uuid": "^11.0.3"
|
|
86
92
|
}
|