@aui-x/prism 0.0.1

package/dist/wrapper-7jRyp54U.js

@@ -0,0 +1,242 @@
+import { wrapLanguageModel } from "ai";
+
+//#region src/wrapper.ts
+function wrapModelWithMiddleware(tracer, model, opts) {
+	return wrapLanguageModel({
+		model,
+		middleware: createPrismMiddleware(tracer, opts)
+	});
+}
+function extractTextFromContent(content) {
+	return content.filter((c) => c.type === "text" && typeof c.text === "string").map((c) => c.text).join("");
+}
+function truncateInput(prompt, maxBytes = 32768) {
+	let json = JSON.stringify(prompt);
+	if (new TextEncoder().encode(json).length <= maxBytes) return prompt;
+	const arr = [...prompt];
+	while (arr.length > 1) {
+		arr.shift();
+		json = JSON.stringify(arr);
+		if (new TextEncoder().encode(json).length <= maxBytes) return arr;
+	}
+	return arr;
+}
+function extractModelConfig(params) {
+	const keys = [
+		"temperature",
+		"maxOutputTokens",
+		"topP",
+		"topK",
+		"frequencyPenalty",
+		"presencePenalty",
+		"stopSequences"
+	];
+	const config = {};
+	for (const key of keys) if (params[key] !== void 0) config[key] = params[key];
+	return Object.keys(config).length > 0 ? config : void 0;
+}
+function extractToolOutput(output) {
+	if (output && typeof output === "object" && "type" in output) {
+		const o = output;
+		if (o.type === "text" || o.type === "json") return o.value;
+		if (o.type === "denied") return `[denied] ${o.reason ?? ""}`;
+	}
+	return output;
+}
+function resolveToolResults(prompt, pendingToolSpans) {
+	if (pendingToolSpans.size === 0) return;
+	for (const msg of prompt) {
+		if (msg.role !== "tool") continue;
+		for (const part of msg.content) {
+			if (part.type !== "tool-result") continue;
+			const span = pendingToolSpans.get(part.toolCallId);
+			if (span) {
+				span.end({
+					output: extractToolOutput(part.output),
+					status: "completed"
+				});
+				pendingToolSpans.delete(part.toolCallId);
+			}
+		}
+	}
+}
+function createRootHandle(tracer, opts, modelId, provider, input, metadata) {
+	const parentTraceId = opts?.parentTraceId;
+	if (parentTraceId) return {
+		isSpanMode: true,
+		traceId: parentTraceId,
+		handle: tracer.startSpanOnTrace(parentTraceId, {
+			name: opts?.name ?? modelId,
+			type: "llm",
+			input,
+			model: modelId,
+			provider,
+			metadata
+		})
+	};
+	const handle = tracer.startTrace({
+		name: opts?.name ?? modelId,
+		model: modelId,
+		provider,
+		input,
+		metadata,
+		tags: opts?.tags,
+		endUserId: opts?.endUserId
+	});
+	return {
+		isSpanMode: false,
+		traceId: handle.traceId,
+		handle
+	};
+}
+function createToolSpan(tracer, root, toolCall) {
+	if (root.isSpanMode) return tracer.startSpanOnTrace(root.traceId, {
+		name: toolCall.toolName,
+		type: "tool",
+		input: toolCall.input,
+		parentSpanId: root.handle.spanId
+	});
+	return root.handle.startSpan({
+		name: toolCall.toolName,
+		type: "tool",
+		input: toolCall.input
+	});
+}
+function endRootHandle(root, result) {
+	root.handle.end({
+		output: result.output,
+		status: "completed",
+		totalTokens: result.inputTokens + result.outputTokens,
+		promptTokens: result.inputTokens,
+		completionTokens: result.outputTokens,
+		ttftMs: result.ttftMs
+	});
+}
+function cleanupOnError(root, pendingToolSpans, error) {
+	for (const span of pendingToolSpans.values()) span.end({ status: "error" });
+	pendingToolSpans.clear();
+	root.handle.end({
+		status: "error",
+		error: error instanceof Error ? error.message : String(error)
+	});
+}
+function buildMetadata(opts, params) {
+	const modelConfig = extractModelConfig(params);
+	if (modelConfig) return {
+		...opts?.metadata,
+		modelConfig
+	};
+	return opts?.metadata;
+}
+function createPrismMiddleware(tracer, opts) {
+	const pendingToolSpans = /* @__PURE__ */ new Map();
+	return {
+		specificationVersion: "v3",
+		wrapGenerate: async ({ doGenerate, params, model }) => {
+			const input = truncateInput(params.prompt);
+			const metadata = buildMetadata(opts, params);
+			resolveToolResults(params.prompt, pendingToolSpans);
+			const root = createRootHandle(tracer, opts, model.modelId, model.provider, input, metadata);
+			try {
+				const result = await doGenerate();
+				const content = result.content;
+				const output = extractTextFromContent(content);
+				const inputTokens = result.usage.inputTokens.total ?? 0;
+				const outputTokens = result.usage.outputTokens.total ?? 0;
+				for (const part of content) if (part.type === "tool-call" && part.toolCallId) {
+					const span = createToolSpan(tracer, root, {
+						id: part.toolCallId,
+						toolName: part.toolName ?? "tool",
+						input: part.input ?? ""
+					});
+					if (root.isSpanMode) pendingToolSpans.set(part.toolCallId, span);
+					else span.end({ status: "completed" });
+				}
+				endRootHandle(root, {
+					output,
+					inputTokens,
+					outputTokens
+				});
+				return result;
+			} catch (error) {
+				cleanupOnError(root, pendingToolSpans, error);
+				throw error;
+			}
+		},
+		wrapStream: async ({ doStream, params, model }) => {
+			const input = truncateInput(params.prompt);
+			const metadata = buildMetadata(opts, params);
+			resolveToolResults(params.prompt, pendingToolSpans);
+			const root = createRootHandle(tracer, opts, model.modelId, model.provider, input, metadata);
+			try {
+				const streamStart = Date.now();
+				const { stream, ...rest } = await doStream();
+				let output = "";
+				let inputTokens = 0;
+				let outputTokens = 0;
+				let ttftMs;
+				let firstChunk = true;
+				const toolCalls = [];
+				let currentToolInput = null;
+				const transformStream = new TransformStream({
+					transform(chunk, controller) {
+						switch (chunk.type) {
+							case "text-delta":
+								if (firstChunk) {
+									ttftMs = Date.now() - streamStart;
+									firstChunk = false;
+								}
+								output += chunk.delta;
+								break;
+							case "tool-input-start":
+								currentToolInput = {
+									id: chunk.id,
+									toolName: chunk.toolName,
+									input: ""
+								};
+								break;
+							case "tool-input-delta":
+								if (currentToolInput && currentToolInput.id === chunk.id) currentToolInput.input += chunk.delta;
+								break;
+							case "tool-input-end":
+								if (currentToolInput && currentToolInput.id === chunk.id) {
+									toolCalls.push(currentToolInput);
+									currentToolInput = null;
+								}
+								break;
+							case "finish":
+								inputTokens = chunk.usage.inputTokens.total ?? 0;
+								outputTokens = chunk.usage.outputTokens.total ?? 0;
+								break;
+						}
+						controller.enqueue(chunk);
+					},
+					flush() {
+						for (const tc of toolCalls) {
+							const span = createToolSpan(tracer, root, tc);
+							if (root.isSpanMode) pendingToolSpans.set(tc.id, span);
+							else span.end({ status: "completed" });
+						}
+						endRootHandle(root, {
+							output,
+							inputTokens,
+							outputTokens,
+							ttftMs
+						});
+					}
+				});
+				return {
+					stream: stream.pipeThrough(transformStream),
+					...rest
+				};
+			} catch (error) {
+				cleanupOnError(root, pendingToolSpans, error);
+				throw error;
+			}
+		}
+	};
+}
+
+//#endregion
+export { wrapModelWithMiddleware as t };
+//# sourceMappingURL=wrapper-7jRyp54U.js.map

package/dist/wrapper-7jRyp54U.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"wrapper-7jRyp54U.js","names":[],"sources":["../src/wrapper.ts"],"sourcesContent":["import type {\n  LanguageModelV3,\n  LanguageModelV3Middleware,\n  LanguageModelV3Prompt,\n} from \"@ai-sdk/provider\";\nimport { wrapLanguageModel } from \"ai\";\nimport type { AuixPrism, SpanHandle, TraceHandle } from \"./client\";\nimport type { AISDKPrismOptions } from \"./integrations/ai-sdk\";\n\ntype MiddlewareOptions = AISDKPrismOptions & { parentTraceId?: string };\n\nexport function wrapModelWithMiddleware(\n  tracer: AuixPrism,\n  model: LanguageModelV3,\n  opts?: MiddlewareOptions,\n): LanguageModelV3 {\n  return wrapLanguageModel({\n    model,\n    middleware: createPrismMiddleware(tracer, opts),\n  });\n}\n\ntype ContentPart = {\n  type: string;\n  text?: string;\n  toolName?: string;\n  toolCallId?: string;\n  input?: string;\n  [key: string]: unknown;\n};\n\nfunction extractTextFromContent(content: ContentPart[]): string {\n  return content\n    .filter((c) => c.type === \"text\" && typeof c.text === \"string\")\n    .map((c) => c.text)\n    .join(\"\");\n}\n\nfunction truncateInput(\n  prompt: LanguageModelV3Prompt,\n  maxBytes = 32768,\n): unknown {\n  let json = JSON.stringify(prompt);\n  if (new TextEncoder().encode(json).length <= maxBytes) {\n    return prompt;\n  }\n\n  const arr = [...prompt];\n  while (arr.length > 1) {\n    arr.shift();\n    json = JSON.stringify(arr);\n    if (new TextEncoder().encode(json).length <= maxBytes) {\n      return arr;\n    }\n  }\n  return arr;\n}\n\nfunction extractModelConfig(params: Record<string, unknown>) {\n  const keys = [\n    \"temperature\",\n    \"maxOutputTokens\",\n    \"topP\",\n    \"topK\",\n    \"frequencyPenalty\",\n    \"presencePenalty\",\n    \"stopSequences\",\n  ] as const;\n\n  const config: Record<string, unknown> = {};\n  for (const key of keys) {\n    if (params[key] !== undefined) {\n      config[key] = params[key];\n    }\n  }\n  return Object.keys(config).length > 0 ? config : undefined;\n}\n\nfunction extractToolOutput(output: unknown): unknown {\n  if (output && typeof output === \"object\" && \"type\" in output) {\n    const o = output as { type: string; value?: unknown; reason?: string };\n    if (o.type === \"text\" || o.type === \"json\") return o.value;\n    if (o.type === \"denied\") return `[denied] ${o.reason ?? \"\"}`;\n  }\n  return output;\n}\n\nfunction resolveToolResults(\n  prompt: LanguageModelV3Prompt,\n  pendingToolSpans: Map<string, SpanHandle>,\n): void {\n  if (pendingToolSpans.size === 0) return;\n\n  for (const msg of prompt) {\n    if (msg.role !== \"tool\") continue;\n    for (const part of msg.content) {\n      if (part.type !== \"tool-result\") continue;\n      const span = pendingToolSpans.get(part.toolCallId);\n      if (span) {\n        span.end({\n          output: extractToolOutput(part.output),\n          status: \"completed\",\n        });\n        pendingToolSpans.delete(part.toolCallId);\n      }\n    }\n  }\n}\n\ntype RootHandle =\n  | { isSpanMode: true; traceId: string; handle: SpanHandle }\n  | { isSpanMode: false; traceId: string; handle: TraceHandle };\n\nfunction createRootHandle(\n  tracer: AuixPrism,\n  opts: MiddlewareOptions | undefined,\n  modelId: string,\n  provider: string,\n  input: unknown,\n  metadata: Record<string, unknown> | undefined,\n): RootHandle {\n  const parentTraceId = opts?.parentTraceId;\n\n  if (parentTraceId) {\n    const handle = tracer.startSpanOnTrace(parentTraceId, {\n      name: opts?.name ?? modelId,\n      type: \"llm\",\n      input,\n      model: modelId,\n      provider,\n      metadata,\n    });\n    return { isSpanMode: true, traceId: parentTraceId, handle };\n  }\n\n  const handle = tracer.startTrace({\n    name: opts?.name ?? modelId,\n    model: modelId,\n    provider,\n    input,\n    metadata,\n    tags: opts?.tags,\n    endUserId: opts?.endUserId,\n  });\n  return { isSpanMode: false, traceId: handle.traceId, handle };\n}\n\nfunction createToolSpan(\n  tracer: AuixPrism,\n  root: RootHandle,\n  toolCall: { id: string; toolName: string; input: string },\n): SpanHandle {\n  if (root.isSpanMode) {\n    return tracer.startSpanOnTrace(root.traceId, {\n      name: toolCall.toolName,\n      type: \"tool\",\n      input: toolCall.input,\n      parentSpanId: root.handle.spanId,\n    });\n  }\n  return root.handle.startSpan({\n    name: toolCall.toolName,\n    type: \"tool\",\n    input: toolCall.input,\n  });\n}\n\nfunction endRootHandle(\n  root: RootHandle,\n  result: {\n    output: string;\n    inputTokens: number;\n    outputTokens: number;\n    ttftMs?: number;\n  },\n): void {\n  root.handle.end({\n    output: result.output,\n    status: \"completed\",\n    totalTokens: result.inputTokens + result.outputTokens,\n    promptTokens: result.inputTokens,\n    completionTokens: result.outputTokens,\n    ttftMs: result.ttftMs,\n  });\n}\n\nfunction cleanupOnError(\n  root: RootHandle,\n  pendingToolSpans: Map<string, SpanHandle>,\n  error: unknown,\n): void {\n  for (const span of pendingToolSpans.values()) {\n    span.end({ status: \"error\" });\n  }\n  pendingToolSpans.clear();\n\n  root.handle.end({\n    status: \"error\",\n    error: error instanceof Error ? error.message : String(error),\n  });\n}\n\nfunction buildMetadata(\n  opts: MiddlewareOptions | undefined,\n  params: Record<string, unknown>,\n): Record<string, unknown> | undefined {\n  const modelConfig = extractModelConfig(params);\n  if (modelConfig) return { ...opts?.metadata, modelConfig };\n  return opts?.metadata;\n}\n\nfunction createPrismMiddleware(\n  tracer: AuixPrism,\n  opts?: MiddlewareOptions,\n): LanguageModelV3Middleware {\n  const pendingToolSpans = new Map<string, SpanHandle>();\n\n  return {\n    specificationVersion: \"v3\",\n\n    wrapGenerate: async ({ doGenerate, params, model }) => {\n      const input = truncateInput(params.prompt);\n      const metadata = buildMetadata(opts, params);\n\n      resolveToolResults(params.prompt, pendingToolSpans);\n\n      const root = createRootHandle(\n        tracer,\n        opts,\n        model.modelId,\n        model.provider,\n        input,\n        metadata,\n      );\n\n      try {\n        const result = await doGenerate();\n        const content = result.content as ContentPart[];\n        const output = extractTextFromContent(content);\n        const inputTokens = result.usage.inputTokens.total ?? 0;\n        const outputTokens = result.usage.outputTokens.total ?? 0;\n\n        for (const part of content) {\n          if (part.type === \"tool-call\" && part.toolCallId) {\n            const span = createToolSpan(tracer, root, {\n              id: part.toolCallId,\n              toolName: part.toolName ?? \"tool\",\n              input: part.input ?? \"\",\n            });\n            if (root.isSpanMode) {\n              pendingToolSpans.set(part.toolCallId, span);\n            } else {\n              span.end({ status: \"completed\" });\n            }\n          }\n        }\n\n        endRootHandle(root, { output, inputTokens, outputTokens });\n        return result;\n      } catch (error) {\n        cleanupOnError(root, pendingToolSpans, error);\n        throw error;\n      }\n    },\n\n    wrapStream: async ({ doStream, params, model }) => {\n      const input = truncateInput(params.prompt);\n      const metadata = buildMetadata(opts, params);\n\n      resolveToolResults(params.prompt, pendingToolSpans);\n\n      const root = createRootHandle(\n        tracer,\n        opts,\n        model.modelId,\n        model.provider,\n        input,\n        metadata,\n      );\n\n      try {\n        const streamStart = Date.now();\n        const { stream, ...rest } = await doStream();\n\n        let output = \"\";\n        let inputTokens = 0;\n        let outputTokens = 0;\n        let ttftMs: number | undefined;\n        let firstChunk = true;\n        const toolCalls: Array<{\n          id: string;\n          toolName: string;\n          input: string;\n        }> = [];\n        let currentToolInput: {\n          id: string;\n          toolName: string;\n          input: string;\n        } | null = null;\n\n        const transformStream = new TransformStream({\n          transform(chunk, controller) {\n            switch (chunk.type) {\n              case \"text-delta\":\n                if (firstChunk) {\n                  ttftMs = Date.now() - streamStart;\n                  firstChunk = false;\n                }\n                output += chunk.delta;\n                break;\n              case \"tool-input-start\":\n                currentToolInput = {\n                  id: chunk.id,\n                  toolName: chunk.toolName,\n                  input: \"\",\n                };\n                break;\n              case \"tool-input-delta\":\n                if (currentToolInput && currentToolInput.id === chunk.id) {\n                  currentToolInput.input += chunk.delta;\n                }\n                break;\n              case \"tool-input-end\":\n                if (currentToolInput && currentToolInput.id === chunk.id) {\n                  toolCalls.push(currentToolInput);\n                  currentToolInput = null;\n                }\n                break;\n              case \"finish\":\n                inputTokens = chunk.usage.inputTokens.total ?? 0;\n                outputTokens = chunk.usage.outputTokens.total ?? 0;\n                break;\n            }\n            controller.enqueue(chunk);\n          },\n          flush() {\n            for (const tc of toolCalls) {\n              const span = createToolSpan(tracer, root, tc);\n              if (root.isSpanMode) {\n                pendingToolSpans.set(tc.id, span);\n              } else {\n                span.end({ status: \"completed\" });\n              }\n            }\n\n            endRootHandle(root, {\n              output,\n              inputTokens,\n              outputTokens,\n              ttftMs,\n            });\n          },\n        });\n\n        return {\n          stream: stream.pipeThrough(transformStream),\n          ...rest,\n        };\n      } catch (error) {\n        cleanupOnError(root, pendingToolSpans, error);\n        throw error;\n      }\n    },\n  };\n}\n"],"mappings":";;;AAWA,SAAgB,wBACd,QACA,OACA,MACiB;AACjB,QAAO,kBAAkB;EACvB;EACA,YAAY,sBAAsB,QAAQ,KAAK;EAChD,CAAC;;AAYJ,SAAS,uBAAuB,SAAgC;AAC9D,QAAO,QACJ,QAAQ,MAAM,EAAE,SAAS,UAAU,OAAO,EAAE,SAAS,SAAS,CAC9D,KAAK,MAAM,EAAE,KAAK,CAClB,KAAK,GAAG;;AAGb,SAAS,cACP,QACA,WAAW,OACF;CACT,IAAI,OAAO,KAAK,UAAU,OAAO;AACjC,KAAI,IAAI,aAAa,CAAC,OAAO,KAAK,CAAC,UAAU,SAC3C,QAAO;CAGT,MAAM,MAAM,CAAC,GAAG,OAAO;AACvB,QAAO,IAAI,SAAS,GAAG;AACrB,MAAI,OAAO;AACX,SAAO,KAAK,UAAU,IAAI;AAC1B,MAAI,IAAI,aAAa,CAAC,OAAO,KAAK,CAAC,UAAU,SAC3C,QAAO;;AAGX,QAAO;;AAGT,SAAS,mBAAmB,QAAiC;CAC3D,MAAM,OAAO;EACX;EACA;EACA;EACA;EACA;EACA;EACA;EACD;CAED,MAAM,SAAkC,EAAE;AAC1C,MAAK,MAAM,OAAO,KAChB,KAAI,OAAO,SAAS,OAClB,QAAO,OAAO,OAAO;AAGzB,QAAO,OAAO,KAAK,OAAO,CAAC,SAAS,IAAI,SAAS;;AAGnD,SAAS,kBAAkB,QAA0B;AACnD,KAAI,UAAU,OAAO,WAAW,YAAY,UAAU,QAAQ;EAC5D,MAAM,IAAI;AACV,MAAI,EAAE,SAAS,UAAU,EAAE,SAAS,OAAQ,QAAO,EAAE;AACrD,MAAI,EAAE,SAAS,SAAU,QAAO,YAAY,EAAE,UAAU;;AAE1D,QAAO;;AAGT,SAAS,mBACP,QACA,kBACM;AACN,KAAI,iBAAiB,SAAS,EAAG;AAEjC,MAAK,MAAM,OAAO,QAAQ;AACxB,MAAI,IAAI,SAAS,OAAQ;AACzB,OAAK,MAAM,QAAQ,IAAI,SAAS;AAC9B,OAAI,KAAK,SAAS,cAAe;GACjC,MAAM,OAAO,iBAAiB,IAAI,KAAK,WAAW;AAClD,OAAI,MAAM;AACR,SAAK,IAAI;KACP,QAAQ,kBAAkB,KAAK,OAAO;KACtC,QAAQ;KACT,CAAC;AACF,qBAAiB,OAAO,KAAK,WAAW;;;;;AAUhD,SAAS,iBACP,QACA,MACA,SACA,UACA,OACA,UACY;CACZ,MAAM,gBAAgB,MAAM;AAE5B,KAAI,cASF,QAAO;EAAE,YAAY;EAAM,SAAS;EAAe,QARpC,OAAO,iBAAiB,eAAe;GACpD,MAAM,MAAM,QAAQ;GACpB,MAAM;GACN;GACA,OAAO;GACP;GACA;GACD,CAAC;EACyD;CAG7D,MAAM,SAAS,OAAO,WAAW;EAC/B,MAAM,MAAM,QAAQ;EACpB,OAAO;EACP;EACA;EACA;EACA,MAAM,MAAM;EACZ,WAAW,MAAM;EAClB,CAAC;AACF,QAAO;EAAE,YAAY;EAAO,SAAS,OAAO;EAAS;EAAQ;;AAG/D,SAAS,eACP,QACA,MACA,UACY;AACZ,KAAI,KAAK,WACP,QAAO,OAAO,iBAAiB,KAAK,SAAS;EAC3C,MAAM,SAAS;EACf,MAAM;EACN,OAAO,SAAS;EAChB,cAAc,KAAK,OAAO;EAC3B,CAAC;AAEJ,QAAO,KAAK,OAAO,UAAU;EAC3B,MAAM,SAAS;EACf,MAAM;EACN,OAAO,SAAS;EACjB,CAAC;;AAGJ,SAAS,cACP,MACA,QAMM;AACN,MAAK,OAAO,IAAI;EACd,QAAQ,OAAO;EACf,QAAQ;EACR,aAAa,OAAO,cAAc,OAAO;EACzC,cAAc,OAAO;EACrB,kBAAkB,OAAO;EACzB,QAAQ,OAAO;EAChB,CAAC;;AAGJ,SAAS,eACP,MACA,kBACA,OACM;AACN,MAAK,MAAM,QAAQ,iBAAiB,QAAQ,CAC1C,MAAK,IAAI,EAAE,QAAQ,SAAS,CAAC;AAE/B,kBAAiB,OAAO;AAExB,MAAK,OAAO,IAAI;EACd,QAAQ;EACR,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;EAC9D,CAAC;;AAGJ,SAAS,cACP,MACA,QACqC;CACrC,MAAM,cAAc,mBAAmB,OAAO;AAC9C,KAAI,YAAa,QAAO;EAAE,GAAG,MAAM;EAAU;EAAa;AAC1D,QAAO,MAAM;;AAGf,SAAS,sBACP,QACA,MAC2B;CAC3B,MAAM,mCAAmB,IAAI,KAAyB;AAEtD,QAAO;EACL,sBAAsB;EAEtB,cAAc,OAAO,EAAE,YAAY,QAAQ,YAAY;GACrD,MAAM,QAAQ,cAAc,OAAO,OAAO;GAC1C,MAAM,WAAW,cAAc,MAAM,OAAO;AAE5C,sBAAmB,OAAO,QAAQ,iBAAiB;GAEnD,MAAM,OAAO,iBACX,QACA,MACA,MAAM,SACN,MAAM,UACN,OACA,SACD;AAED,OAAI;IACF,MAAM,SAAS,MAAM,YAAY;IACjC,MAAM,UAAU,OAAO;IACvB,MAAM,SAAS,uBAAuB,QAAQ;IAC9C,MAAM,cAAc,OAAO,MAAM,YAAY,SAAS;IACtD,MAAM,eAAe,OAAO,MAAM,aAAa,SAAS;AAExD,SAAK,MAAM,QAAQ,QACjB,KAAI,KAAK,SAAS,eAAe,KAAK,YAAY;KAChD,MAAM,OAAO,eAAe,QAAQ,MAAM;MACxC,IAAI,KAAK;MACT,UAAU,KAAK,YAAY;MAC3B,OAAO,KAAK,SAAS;MACtB,CAAC;AACF,SAAI,KAAK,WACP,kBAAiB,IAAI,KAAK,YAAY,KAAK;SAE3C,MAAK,IAAI,EAAE,QAAQ,aAAa,CAAC;;AAKvC,kBAAc,MAAM;KAAE;KAAQ;KAAa;KAAc,CAAC;AAC1D,WAAO;YACA,OAAO;AACd,mBAAe,MAAM,kBAAkB,MAAM;AAC7C,UAAM;;;EAIV,YAAY,OAAO,EAAE,UAAU,QAAQ,YAAY;GACjD,MAAM,QAAQ,cAAc,OAAO,OAAO;GAC1C,MAAM,WAAW,cAAc,MAAM,OAAO;AAE5C,sBAAmB,OAAO,QAAQ,iBAAiB;GAEnD,MAAM,OAAO,iBACX,QACA,MACA,MAAM,SACN,MAAM,UACN,OACA,SACD;AAED,OAAI;IACF,MAAM,cAAc,KAAK,KAAK;IAC9B,MAAM,EAAE,QAAQ,GAAG,SAAS,MAAM,UAAU;IAE5C,IAAI,SAAS;IACb,IAAI,cAAc;IAClB,IAAI,eAAe;IACnB,IAAI;IACJ,IAAI,aAAa;IACjB,MAAM,YAID,EAAE;IACP,IAAI,mBAIO;IAEX,MAAM,kBAAkB,IAAI,gBAAgB;KAC1C,UAAU,OAAO,YAAY;AAC3B,cAAQ,MAAM,MAAd;OACE,KAAK;AACH,YAAI,YAAY;AACd,kBAAS,KAAK,KAAK,GAAG;AACtB,sBAAa;;AAEf,kBAAU,MAAM;AAChB;OACF,KAAK;AACH,2BAAmB;SACjB,IAAI,MAAM;SACV,UAAU,MAAM;SAChB,OAAO;SACR;AACD;OACF,KAAK;AACH,YAAI,oBAAoB,iBAAiB,OAAO,MAAM,GACpD,kBAAiB,SAAS,MAAM;AAElC;OACF,KAAK;AACH,YAAI,oBAAoB,iBAAiB,OAAO,MAAM,IAAI;AACxD,mBAAU,KAAK,iBAAiB;AAChC,4BAAmB;;AAErB;OACF,KAAK;AACH,sBAAc,MAAM,MAAM,YAAY,SAAS;AAC/C,uBAAe,MAAM,MAAM,aAAa,SAAS;AACjD;;AAEJ,iBAAW,QAAQ,MAAM;;KAE3B,QAAQ;AACN,WAAK,MAAM,MAAM,WAAW;OAC1B,MAAM,OAAO,eAAe,QAAQ,MAAM,GAAG;AAC7C,WAAI,KAAK,WACP,kBAAiB,IAAI,GAAG,IAAI,KAAK;WAEjC,MAAK,IAAI,EAAE,QAAQ,aAAa,CAAC;;AAIrC,oBAAc,MAAM;OAClB;OACA;OACA;OACA;OACD,CAAC;;KAEL,CAAC;AAEF,WAAO;KACL,QAAQ,OAAO,YAAY,gBAAgB;KAC3C,GAAG;KACJ;YACM,OAAO;AACd,mBAAe,MAAM,kBAAkB,MAAM;AAC7C,UAAM;;;EAGX"}

package/dist/wrapper-ByspXfxS.cjs

@@ -0,0 +1,247 @@
+let ai = require("ai");
+
+//#region src/wrapper.ts
+function wrapModelWithMiddleware(tracer, model, opts) {
+	return (0, ai.wrapLanguageModel)({
+		model,
+		middleware: createPrismMiddleware(tracer, opts)
+	});
+}
+function extractTextFromContent(content) {
+	return content.filter((c) => c.type === "text" && typeof c.text === "string").map((c) => c.text).join("");
+}
+function truncateInput(prompt, maxBytes = 32768) {
+	let json = JSON.stringify(prompt);
+	if (new TextEncoder().encode(json).length <= maxBytes) return prompt;
+	const arr = [...prompt];
+	while (arr.length > 1) {
+		arr.shift();
+		json = JSON.stringify(arr);
+		if (new TextEncoder().encode(json).length <= maxBytes) return arr;
+	}
+	return arr;
+}
+function extractModelConfig(params) {
+	const keys = [
+		"temperature",
+		"maxOutputTokens",
+		"topP",
+		"topK",
+		"frequencyPenalty",
+		"presencePenalty",
+		"stopSequences"
+	];
+	const config = {};
+	for (const key of keys) if (params[key] !== void 0) config[key] = params[key];
+	return Object.keys(config).length > 0 ? config : void 0;
+}
+function extractToolOutput(output) {
+	if (output && typeof output === "object" && "type" in output) {
+		const o = output;
+		if (o.type === "text" || o.type === "json") return o.value;
+		if (o.type === "denied") return `[denied] ${o.reason ?? ""}`;
+	}
+	return output;
+}
+function resolveToolResults(prompt, pendingToolSpans) {
+	if (pendingToolSpans.size === 0) return;
+	for (const msg of prompt) {
+		if (msg.role !== "tool") continue;
+		for (const part of msg.content) {
+			if (part.type !== "tool-result") continue;
+			const span = pendingToolSpans.get(part.toolCallId);
+			if (span) {
+				span.end({
+					output: extractToolOutput(part.output),
+					status: "completed"
+				});
+				pendingToolSpans.delete(part.toolCallId);
+			}
+		}
+	}
+}
+function createRootHandle(tracer, opts, modelId, provider, input, metadata) {
+	const parentTraceId = opts?.parentTraceId;
+	if (parentTraceId) return {
+		isSpanMode: true,
+		traceId: parentTraceId,
+		handle: tracer.startSpanOnTrace(parentTraceId, {
+			name: opts?.name ?? modelId,
+			type: "llm",
+			input,
+			model: modelId,
+			provider,
+			metadata
+		})
+	};
+	const handle = tracer.startTrace({
+		name: opts?.name ?? modelId,
+		model: modelId,
+		provider,
+		input,
+		metadata,
+		tags: opts?.tags,
+		endUserId: opts?.endUserId
+	});
+	return {
+		isSpanMode: false,
+		traceId: handle.traceId,
+		handle
+	};
+}
+function createToolSpan(tracer, root, toolCall) {
+	if (root.isSpanMode) return tracer.startSpanOnTrace(root.traceId, {
+		name: toolCall.toolName,
+		type: "tool",
+		input: toolCall.input,
+		parentSpanId: root.handle.spanId
+	});
+	return root.handle.startSpan({
+		name: toolCall.toolName,
+		type: "tool",
+		input: toolCall.input
+	});
+}
+function endRootHandle(root, result) {
+	root.handle.end({
+		output: result.output,
+		status: "completed",
+		totalTokens: result.inputTokens + result.outputTokens,
+		promptTokens: result.inputTokens,
+		completionTokens: result.outputTokens,
+		ttftMs: result.ttftMs
+	});
+}
+function cleanupOnError(root, pendingToolSpans, error) {
+	for (const span of pendingToolSpans.values()) span.end({ status: "error" });
+	pendingToolSpans.clear();
+	root.handle.end({
+		status: "error",
+		error: error instanceof Error ? error.message : String(error)
+	});
+}
+function buildMetadata(opts, params) {
+	const modelConfig = extractModelConfig(params);
+	if (modelConfig) return {
+		...opts?.metadata,
+		modelConfig
+	};
+	return opts?.metadata;
+}
+function createPrismMiddleware(tracer, opts) {
+	const pendingToolSpans = /* @__PURE__ */ new Map();
+	return {
+		specificationVersion: "v3",
+		wrapGenerate: async ({ doGenerate, params, model }) => {
+			const input = truncateInput(params.prompt);
+			const metadata = buildMetadata(opts, params);
+			resolveToolResults(params.prompt, pendingToolSpans);
+			const root = createRootHandle(tracer, opts, model.modelId, model.provider, input, metadata);
+			try {
+				const result = await doGenerate();
+				const content = result.content;
+				const output = extractTextFromContent(content);
+				const inputTokens = result.usage.inputTokens.total ?? 0;
+				const outputTokens = result.usage.outputTokens.total ?? 0;
+				for (const part of content) if (part.type === "tool-call" && part.toolCallId) {
+					const span = createToolSpan(tracer, root, {
+						id: part.toolCallId,
+						toolName: part.toolName ?? "tool",
+						input: part.input ?? ""
+					});
+					if (root.isSpanMode) pendingToolSpans.set(part.toolCallId, span);
+					else span.end({ status: "completed" });
+				}
+				endRootHandle(root, {
+					output,
+					inputTokens,
+					outputTokens
+				});
+				return result;
+			} catch (error) {
+				cleanupOnError(root, pendingToolSpans, error);
+				throw error;
+			}
+		},
+		wrapStream: async ({ doStream, params, model }) => {
+			const input = truncateInput(params.prompt);
+			const metadata = buildMetadata(opts, params);
+			resolveToolResults(params.prompt, pendingToolSpans);
+			const root = createRootHandle(tracer, opts, model.modelId, model.provider, input, metadata);
+			try {
+				const streamStart = Date.now();
+				const { stream, ...rest } = await doStream();
+				let output = "";
+				let inputTokens = 0;
+				let outputTokens = 0;
+				let ttftMs;
+				let firstChunk = true;
+				const toolCalls = [];
+				let currentToolInput = null;
+				const transformStream = new TransformStream({
+					transform(chunk, controller) {
+						switch (chunk.type) {
+							case "text-delta":
+								if (firstChunk) {
+									ttftMs = Date.now() - streamStart;
+									firstChunk = false;
+								}
+								output += chunk.delta;
+								break;
+							case "tool-input-start":
+								currentToolInput = {
+									id: chunk.id,
+									toolName: chunk.toolName,
+									input: ""
+								};
+								break;
+							case "tool-input-delta":
+								if (currentToolInput && currentToolInput.id === chunk.id) currentToolInput.input += chunk.delta;
+								break;
+							case "tool-input-end":
+								if (currentToolInput && currentToolInput.id === chunk.id) {
+									toolCalls.push(currentToolInput);
+									currentToolInput = null;
+								}
+								break;
+							case "finish":
+								inputTokens = chunk.usage.inputTokens.total ?? 0;
+								outputTokens = chunk.usage.outputTokens.total ?? 0;
+								break;
+						}
+						controller.enqueue(chunk);
+					},
+					flush() {
+						for (const tc of toolCalls) {
+							const span = createToolSpan(tracer, root, tc);
+							if (root.isSpanMode) pendingToolSpans.set(tc.id, span);
+							else span.end({ status: "completed" });
+						}
+						endRootHandle(root, {
+							output,
+							inputTokens,
+							outputTokens,
+							ttftMs
+						});
+					}
+				});
+				return {
+					stream: stream.pipeThrough(transformStream),
+					...rest
+				};
+			} catch (error) {
+				cleanupOnError(root, pendingToolSpans, error);
+				throw error;
+			}
+		}
+	};
+}
+
+//#endregion
+Object.defineProperty(exports, 'wrapModelWithMiddleware', {
+  enumerable: true,
+  get: function () {
+    return wrapModelWithMiddleware;
+  }
+});
+//# sourceMappingURL=wrapper-ByspXfxS.cjs.map

package/dist/wrapper-ByspXfxS.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"wrapper-ByspXfxS.cjs","names":[],"sources":["../src/wrapper.ts"],"sourcesContent":["import type {\n  LanguageModelV3,\n  LanguageModelV3Middleware,\n  LanguageModelV3Prompt,\n} from \"@ai-sdk/provider\";\nimport { wrapLanguageModel } from \"ai\";\nimport type { AuixPrism, SpanHandle, TraceHandle } from \"./client\";\nimport type { AISDKPrismOptions } from \"./integrations/ai-sdk\";\n\ntype MiddlewareOptions = AISDKPrismOptions & { parentTraceId?: string };\n\nexport function wrapModelWithMiddleware(\n  tracer: AuixPrism,\n  model: LanguageModelV3,\n  opts?: MiddlewareOptions,\n): LanguageModelV3 {\n  return wrapLanguageModel({\n    model,\n    middleware: createPrismMiddleware(tracer, opts),\n  });\n}\n\ntype ContentPart = {\n  type: string;\n  text?: string;\n  toolName?: string;\n  toolCallId?: string;\n  input?: string;\n  [key: string]: unknown;\n};\n\nfunction extractTextFromContent(content: ContentPart[]): string {\n  return content\n    .filter((c) => c.type === \"text\" && typeof c.text === \"string\")\n    .map((c) => c.text)\n    .join(\"\");\n}\n\nfunction truncateInput(\n  prompt: LanguageModelV3Prompt,\n  maxBytes = 32768,\n): unknown {\n  let json = JSON.stringify(prompt);\n  if (new TextEncoder().encode(json).length <= maxBytes) {\n    return prompt;\n  }\n\n  const arr = [...prompt];\n  while (arr.length > 1) {\n    arr.shift();\n    json = JSON.stringify(arr);\n    if (new TextEncoder().encode(json).length <= maxBytes) {\n      return arr;\n    }\n  }\n  return arr;\n}\n\nfunction extractModelConfig(params: Record<string, unknown>) {\n  const keys = [\n    \"temperature\",\n    \"maxOutputTokens\",\n    \"topP\",\n    \"topK\",\n    \"frequencyPenalty\",\n    \"presencePenalty\",\n    \"stopSequences\",\n  ] as const;\n\n  const config: Record<string, unknown> = {};\n  for (const key of keys) {\n    if (params[key] !== undefined) {\n      config[key] = params[key];\n    }\n  }\n  return Object.keys(config).length > 0 ? config : undefined;\n}\n\nfunction extractToolOutput(output: unknown): unknown {\n  if (output && typeof output === \"object\" && \"type\" in output) {\n    const o = output as { type: string; value?: unknown; reason?: string };\n    if (o.type === \"text\" || o.type === \"json\") return o.value;\n    if (o.type === \"denied\") return `[denied] ${o.reason ?? \"\"}`;\n  }\n  return output;\n}\n\nfunction resolveToolResults(\n  prompt: LanguageModelV3Prompt,\n  pendingToolSpans: Map<string, SpanHandle>,\n): void {\n  if (pendingToolSpans.size === 0) return;\n\n  for (const msg of prompt) {\n    if (msg.role !== \"tool\") continue;\n    for (const part of msg.content) {\n      if (part.type !== \"tool-result\") continue;\n      const span = pendingToolSpans.get(part.toolCallId);\n      if (span) {\n        span.end({\n          output: extractToolOutput(part.output),\n          status: \"completed\",\n        });\n        pendingToolSpans.delete(part.toolCallId);\n      }\n    }\n  }\n}\n\ntype RootHandle =\n  | { isSpanMode: true; traceId: string; handle: SpanHandle }\n  | { isSpanMode: false; traceId: string; handle: TraceHandle };\n\nfunction createRootHandle(\n  tracer: AuixPrism,\n  opts: MiddlewareOptions | undefined,\n  modelId: string,\n  provider: string,\n  input: unknown,\n  metadata: Record<string, unknown> | undefined,\n): RootHandle {\n  const parentTraceId = opts?.parentTraceId;\n\n  if (parentTraceId) {\n    const handle = tracer.startSpanOnTrace(parentTraceId, {\n      name: opts?.name ?? modelId,\n      type: \"llm\",\n      input,\n      model: modelId,\n      provider,\n      metadata,\n    });\n    return { isSpanMode: true, traceId: parentTraceId, handle };\n  }\n\n  const handle = tracer.startTrace({\n    name: opts?.name ?? modelId,\n    model: modelId,\n    provider,\n    input,\n    metadata,\n    tags: opts?.tags,\n    endUserId: opts?.endUserId,\n  });\n  return { isSpanMode: false, traceId: handle.traceId, handle };\n}\n\nfunction createToolSpan(\n  tracer: AuixPrism,\n  root: RootHandle,\n  toolCall: { id: string; toolName: string; input: string },\n): SpanHandle {\n  if (root.isSpanMode) {\n    return tracer.startSpanOnTrace(root.traceId, {\n      name: toolCall.toolName,\n      type: \"tool\",\n      input: toolCall.input,\n      parentSpanId: root.handle.spanId,\n    });\n  }\n  return root.handle.startSpan({\n    name: toolCall.toolName,\n    type: \"tool\",\n    input: toolCall.input,\n  });\n}\n\nfunction endRootHandle(\n  root: RootHandle,\n  result: {\n    output: string;\n    inputTokens: number;\n    outputTokens: number;\n    ttftMs?: number;\n  },\n): void {\n  root.handle.end({\n    output: result.output,\n    status: \"completed\",\n    totalTokens: result.inputTokens + result.outputTokens,\n    promptTokens: result.inputTokens,\n    completionTokens: result.outputTokens,\n    ttftMs: result.ttftMs,\n  });\n}\n\nfunction cleanupOnError(\n  root: RootHandle,\n  pendingToolSpans: Map<string, SpanHandle>,\n  error: unknown,\n): void {\n  for (const span of pendingToolSpans.values()) {\n    span.end({ status: \"error\" });\n  }\n  pendingToolSpans.clear();\n\n  root.handle.end({\n    status: \"error\",\n    error: error instanceof Error ? error.message : String(error),\n  });\n}\n\nfunction buildMetadata(\n  opts: MiddlewareOptions | undefined,\n  params: Record<string, unknown>,\n): Record<string, unknown> | undefined {\n  const modelConfig = extractModelConfig(params);\n  if (modelConfig) return { ...opts?.metadata, modelConfig };\n  return opts?.metadata;\n}\n\nfunction createPrismMiddleware(\n  tracer: AuixPrism,\n  opts?: MiddlewareOptions,\n): LanguageModelV3Middleware {\n  const pendingToolSpans = new Map<string, SpanHandle>();\n\n  return {\n    specificationVersion: \"v3\",\n\n    wrapGenerate: async ({ doGenerate, params, model }) => {\n      const input = truncateInput(params.prompt);\n      const metadata = buildMetadata(opts, params);\n\n      resolveToolResults(params.prompt, pendingToolSpans);\n\n      const root = createRootHandle(\n        tracer,\n        opts,\n        model.modelId,\n        model.provider,\n        input,\n        metadata,\n      );\n\n      try {\n        const result = await doGenerate();\n        const content = result.content as ContentPart[];\n        const output = extractTextFromContent(content);\n        const inputTokens = result.usage.inputTokens.total ?? 0;\n        const outputTokens = result.usage.outputTokens.total ?? 0;\n\n        for (const part of content) {\n          if (part.type === \"tool-call\" && part.toolCallId) {\n            const span = createToolSpan(tracer, root, {\n              id: part.toolCallId,\n              toolName: part.toolName ?? \"tool\",\n              input: part.input ?? \"\",\n            });\n            if (root.isSpanMode) {\n              pendingToolSpans.set(part.toolCallId, span);\n            } else {\n              span.end({ status: \"completed\" });\n            }\n          }\n        }\n\n        endRootHandle(root, { output, inputTokens, outputTokens });\n        return result;\n      } catch (error) {\n        cleanupOnError(root, pendingToolSpans, error);\n        throw error;\n      }\n    },\n\n    wrapStream: async ({ doStream, params, model }) => {\n      const input = truncateInput(params.prompt);\n      const metadata = buildMetadata(opts, params);\n\n      resolveToolResults(params.prompt, pendingToolSpans);\n\n      const root = createRootHandle(\n        tracer,\n        opts,\n        model.modelId,\n        model.provider,\n        input,\n        metadata,\n      );\n\n      try {\n        const streamStart = Date.now();\n        const { stream, ...rest } = await doStream();\n\n        let output = \"\";\n        let inputTokens = 0;\n        let outputTokens = 0;\n        let ttftMs: number | undefined;\n        let firstChunk = true;\n        const toolCalls: Array<{\n          id: string;\n          toolName: string;\n          input: string;\n        }> = [];\n        let currentToolInput: {\n          id: string;\n          toolName: string;\n          input: string;\n        } | null = null;\n\n        const transformStream = new TransformStream({\n          transform(chunk, controller) {\n            switch (chunk.type) {\n              case \"text-delta\":\n                if (firstChunk) {\n                  ttftMs = Date.now() - streamStart;\n                  firstChunk = false;\n                }\n                output += chunk.delta;\n                break;\n              case \"tool-input-start\":\n                currentToolInput = {\n                  id: chunk.id,\n                  toolName: chunk.toolName,\n                  input: \"\",\n                };\n                break;\n              case \"tool-input-delta\":\n                if (currentToolInput && currentToolInput.id === chunk.id) {\n                  currentToolInput.input += chunk.delta;\n                }\n                break;\n              case \"tool-input-end\":\n                if (currentToolInput && currentToolInput.id === chunk.id) {\n                  toolCalls.push(currentToolInput);\n                  currentToolInput = null;\n                }\n                break;\n              case \"finish\":\n                inputTokens = chunk.usage.inputTokens.total ?? 0;\n                outputTokens = chunk.usage.outputTokens.total ?? 0;\n                break;\n            }\n            controller.enqueue(chunk);\n          },\n          flush() {\n            for (const tc of toolCalls) {\n              const span = createToolSpan(tracer, root, tc);\n              if (root.isSpanMode) {\n                pendingToolSpans.set(tc.id, span);\n              } else {\n                span.end({ status: \"completed\" });\n              }\n            }\n\n            endRootHandle(root, {\n              output,\n              inputTokens,\n              outputTokens,\n              ttftMs,\n            });\n          },\n        });\n\n        return {\n          stream: stream.pipeThrough(transformStream),\n          ...rest,\n        };\n      } catch (error) {\n        cleanupOnError(root, pendingToolSpans, error);\n        throw error;\n      }\n    },\n  };\n}\n"],"mappings":";;;AAWA,SAAgB,wBACd,QACA,OACA,MACiB;AACjB,kCAAyB;EACvB;EACA,YAAY,sBAAsB,QAAQ,KAAK;EAChD,CAAC;;AAYJ,SAAS,uBAAuB,SAAgC;AAC9D,QAAO,QACJ,QAAQ,MAAM,EAAE,SAAS,UAAU,OAAO,EAAE,SAAS,SAAS,CAC9D,KAAK,MAAM,EAAE,KAAK,CAClB,KAAK,GAAG;;AAGb,SAAS,cACP,QACA,WAAW,OACF;CACT,IAAI,OAAO,KAAK,UAAU,OAAO;AACjC,KAAI,IAAI,aAAa,CAAC,OAAO,KAAK,CAAC,UAAU,SAC3C,QAAO;CAGT,MAAM,MAAM,CAAC,GAAG,OAAO;AACvB,QAAO,IAAI,SAAS,GAAG;AACrB,MAAI,OAAO;AACX,SAAO,KAAK,UAAU,IAAI;AAC1B,MAAI,IAAI,aAAa,CAAC,OAAO,KAAK,CAAC,UAAU,SAC3C,QAAO;;AAGX,QAAO;;AAGT,SAAS,mBAAmB,QAAiC;CAC3D,MAAM,OAAO;EACX;EACA;EACA;EACA;EACA;EACA;EACA;EACD;CAED,MAAM,SAAkC,EAAE;AAC1C,MAAK,MAAM,OAAO,KAChB,KAAI,OAAO,SAAS,OAClB,QAAO,OAAO,OAAO;AAGzB,QAAO,OAAO,KAAK,OAAO,CAAC,SAAS,IAAI,SAAS;;AAGnD,SAAS,kBAAkB,QAA0B;AACnD,KAAI,UAAU,OAAO,WAAW,YAAY,UAAU,QAAQ;EAC5D,MAAM,IAAI;AACV,MAAI,EAAE,SAAS,UAAU,EAAE,SAAS,OAAQ,QAAO,EAAE;AACrD,MAAI,EAAE,SAAS,SAAU,QAAO,YAAY,EAAE,UAAU;;AAE1D,QAAO;;AAGT,SAAS,mBACP,QACA,kBACM;AACN,KAAI,iBAAiB,SAAS,EAAG;AAEjC,MAAK,MAAM,OAAO,QAAQ;AACxB,MAAI,IAAI,SAAS,OAAQ;AACzB,OAAK,MAAM,QAAQ,IAAI,SAAS;AAC9B,OAAI,KAAK,SAAS,cAAe;GACjC,MAAM,OAAO,iBAAiB,IAAI,KAAK,WAAW;AAClD,OAAI,MAAM;AACR,SAAK,IAAI;KACP,QAAQ,kBAAkB,KAAK,OAAO;KACtC,QAAQ;KACT,CAAC;AACF,qBAAiB,OAAO,KAAK,WAAW;;;;;AAUhD,SAAS,iBACP,QACA,MACA,SACA,UACA,OACA,UACY;CACZ,MAAM,gBAAgB,MAAM;AAE5B,KAAI,cASF,QAAO;EAAE,YAAY;EAAM,SAAS;EAAe,QARpC,OAAO,iBAAiB,eAAe;GACpD,MAAM,MAAM,QAAQ;GACpB,MAAM;GACN;GACA,OAAO;GACP;GACA;GACD,CAAC;EACyD;CAG7D,MAAM,SAAS,OAAO,WAAW;EAC/B,MAAM,MAAM,QAAQ;EACpB,OAAO;EACP;EACA;EACA;EACA,MAAM,MAAM;EACZ,WAAW,MAAM;EAClB,CAAC;AACF,QAAO;EAAE,YAAY;EAAO,SAAS,OAAO;EAAS;EAAQ;;AAG/D,SAAS,eACP,QACA,MACA,UACY;AACZ,KAAI,KAAK,WACP,QAAO,OAAO,iBAAiB,KAAK,SAAS;EAC3C,MAAM,SAAS;EACf,MAAM;EACN,OAAO,SAAS;EAChB,cAAc,KAAK,OAAO;EAC3B,CAAC;AAEJ,QAAO,KAAK,OAAO,UAAU;EAC3B,MAAM,SAAS;EACf,MAAM;EACN,OAAO,SAAS;EACjB,CAAC;;AAGJ,SAAS,cACP,MACA,QAMM;AACN,MAAK,OAAO,IAAI;EACd,QAAQ,OAAO;EACf,QAAQ;EACR,aAAa,OAAO,cAAc,OAAO;EACzC,cAAc,OAAO;EACrB,kBAAkB,OAAO;EACzB,QAAQ,OAAO;EAChB,CAAC;;AAGJ,SAAS,eACP,MACA,kBACA,OACM;AACN,MAAK,MAAM,QAAQ,iBAAiB,QAAQ,CAC1C,MAAK,IAAI,EAAE,QAAQ,SAAS,CAAC;AAE/B,kBAAiB,OAAO;AAExB,MAAK,OAAO,IAAI;EACd,QAAQ;EACR,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;EAC9D,CAAC;;AAGJ,SAAS,cACP,MACA,QACqC;CACrC,MAAM,cAAc,mBAAmB,OAAO;AAC9C,KAAI,YAAa,QAAO;EAAE,GAAG,MAAM;EAAU;EAAa;AAC1D,QAAO,MAAM;;AAGf,SAAS,sBACP,QACA,MAC2B;CAC3B,MAAM,mCAAmB,IAAI,KAAyB;AAEtD,QAAO;EACL,sBAAsB;EAEtB,cAAc,OAAO,EAAE,YAAY,QAAQ,YAAY;GACrD,MAAM,QAAQ,cAAc,OAAO,OAAO;GAC1C,MAAM,WAAW,cAAc,MAAM,OAAO;AAE5C,sBAAmB,OAAO,QAAQ,iBAAiB;GAEnD,MAAM,OAAO,iBACX,QACA,MACA,MAAM,SACN,MAAM,UACN,OACA,SACD;AAED,OAAI;IACF,MAAM,SAAS,MAAM,YAAY;IACjC,MAAM,UAAU,OAAO;IACvB,MAAM,SAAS,uBAAuB,QAAQ;IAC9C,MAAM,cAAc,OAAO,MAAM,YAAY,SAAS;IACtD,MAAM,eAAe,OAAO,MAAM,aAAa,SAAS;AAExD,SAAK,MAAM,QAAQ,QACjB,KAAI,KAAK,SAAS,eAAe,KAAK,YAAY;KAChD,MAAM,OAAO,eAAe,QAAQ,MAAM;MACxC,IAAI,KAAK;MACT,UAAU,KAAK,YAAY;MAC3B,OAAO,KAAK,SAAS;MACtB,CAAC;AACF,SAAI,KAAK,WACP,kBAAiB,IAAI,KAAK,YAAY,KAAK;SAE3C,MAAK,IAAI,EAAE,QAAQ,aAAa,CAAC;;AAKvC,kBAAc,MAAM;KAAE;KAAQ;KAAa;KAAc,CAAC;AAC1D,WAAO;YACA,OAAO;AACd,mBAAe,MAAM,kBAAkB,MAAM;AAC7C,UAAM;;;EAIV,YAAY,OAAO,EAAE,UAAU,QAAQ,YAAY;GACjD,MAAM,QAAQ,cAAc,OAAO,OAAO;GAC1C,MAAM,WAAW,cAAc,MAAM,OAAO;AAE5C,sBAAmB,OAAO,QAAQ,iBAAiB;GAEnD,MAAM,OAAO,iBACX,QACA,MACA,MAAM,SACN,MAAM,UACN,OACA,SACD;AAED,OAAI;IACF,MAAM,cAAc,KAAK,KAAK;IAC9B,MAAM,EAAE,QAAQ,GAAG,SAAS,MAAM,UAAU;IAE5C,IAAI,SAAS;IACb,IAAI,cAAc;IAClB,IAAI,eAAe;IACnB,IAAI;IACJ,IAAI,aAAa;IACjB,MAAM,YAID,EAAE;IACP,IAAI,mBAIO;IAEX,MAAM,kBAAkB,IAAI,gBAAgB;KAC1C,UAAU,OAAO,YAAY;AAC3B,cAAQ,MAAM,MAAd;OACE,KAAK;AACH,YAAI,YAAY;AACd,kBAAS,KAAK,KAAK,GAAG;AACtB,sBAAa;;AAEf,kBAAU,MAAM;AAChB;OACF,KAAK;AACH,2BAAmB;SACjB,IAAI,MAAM;SACV,UAAU,MAAM;SAChB,OAAO;SACR;AACD;OACF,KAAK;AACH,YAAI,oBAAoB,iBAAiB,OAAO,MAAM,GACpD,kBAAiB,SAAS,MAAM;AAElC;OACF,KAAK;AACH,YAAI,oBAAoB,iBAAiB,OAAO,MAAM,IAAI;AACxD,mBAAU,KAAK,iBAAiB;AAChC,4BAAmB;;AAErB;OACF,KAAK;AACH,sBAAc,MAAM,MAAM,YAAY,SAAS;AAC/C,uBAAe,MAAM,MAAM,aAAa,SAAS;AACjD;;AAEJ,iBAAW,QAAQ,MAAM;;KAE3B,QAAQ;AACN,WAAK,MAAM,MAAM,WAAW;OAC1B,MAAM,OAAO,eAAe,QAAQ,MAAM,GAAG;AAC7C,WAAI,KAAK,WACP,kBAAiB,IAAI,GAAG,IAAI,KAAK;WAEjC,MAAK,IAAI,EAAE,QAAQ,aAAa,CAAC;;AAIrC,oBAAc,MAAM;OAClB;OACA;OACA;OACA;OACD,CAAC;;KAEL,CAAC;AAEF,WAAO;KACL,QAAQ,OAAO,YAAY,gBAAgB;KAC3C,GAAG;KACJ;YACM,OAAO;AACd,mBAAe,MAAM,kBAAkB,MAAM;AAC7C,UAAM;;;EAGX"}

package/package.json

@@ -0,0 +1,103 @@
+{
+  "name": "@aui-x/prism",
+  "version": "0.0.1",
+  "description": "LLM tracing SDK for auix — AI SDK middleware + manual tracing",
+  "sideEffects": false,
+  "type": "module",
+  "exports": {
+    ".": {
+      "import": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.js"
+      },
+      "require": {
+        "types": "./dist/index.d.cts",
+        "default": "./dist/index.cjs"
+      }
+    },
+    "./core": {
+      "import": {
+        "types": "./dist/core.d.ts",
+        "default": "./dist/core.js"
+      },
+      "require": {
+        "types": "./dist/core.d.cts",
+        "default": "./dist/core.cjs"
+      }
+    },
+    "./integrations/ai-sdk": {
+      "import": {
+        "types": "./dist/integrations/ai-sdk.d.ts",
+        "default": "./dist/integrations/ai-sdk.js"
+      },
+      "require": {
+        "types": "./dist/integrations/ai-sdk.d.cts",
+        "default": "./dist/integrations/ai-sdk.cjs"
+      }
+    },
+    "./integrations/openai": {
+      "import": {
+        "types": "./dist/integrations/openai.d.ts",
+        "default": "./dist/integrations/openai.js"
+      },
+      "require": {
+        "types": "./dist/integrations/openai.d.cts",
+        "default": "./dist/integrations/openai.cjs"
+      }
+    },
+    "./integrations/anthropic": {
+      "import": {
+        "types": "./dist/integrations/anthropic.d.ts",
+        "default": "./dist/integrations/anthropic.js"
+      },
+      "require": {
+        "types": "./dist/integrations/anthropic.d.cts",
+        "default": "./dist/integrations/anthropic.cjs"
+      }
+    }
+  },
+  "main": "./dist/index.cjs",
+  "module": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "files": [
+    "dist",
+    "LICENSE",
+    "README.md"
+  ],
+  "scripts": {
+    "build": "tsdown",
+    "ts-check": "tsc --noEmit"
+  },
+  "license": "SEE LICENSE IN LICENSE",
+  "homepage": "https://auix.dev",
+  "keywords": [
+    "ai",
+    "tracing",
+    "observability",
+    "ai-sdk",
+    "llm"
+  ],
+  "peerDependencies": {
+    "@ai-sdk/provider": ">=2.0.0",
+    "ai": ">=6.0.0",
+    "openai": ">=4.0.0",
+    "@anthropic-ai/sdk": ">=0.30.0"
+  },
+  "peerDependenciesMeta": {
+    "@ai-sdk/provider": {
+      "optional": true
+    },
+    "openai": {
+      "optional": true
+    },
+    "@anthropic-ai/sdk": {
+      "optional": true
+    }
+  },
+  "devDependencies": {
+    "@ai-sdk/provider": "catalog:",
+    "@auix/tsconfig": "workspace:*",
+    "ai": "catalog:",
+    "tsdown": "catalog:"
+  }
+}