@kortyx/hooks 0.13.0 → 0.15.0

package/dist/internal.d.ts

@@ -1,5 +1,6 @@
 import { ProviderModelRef, KortyxReasoningOptions, KortyxResponseFormat, KortyxUsage, KortyxFinishReason, KortyxProviderMetadata, KortyxWarning } from '@kortyx/providers';
-import { R as ReasonTraceAdapter } from './tracing-BG-j1cer.js';
+import { K as KortyxTraceMetadata, R as ReasonTraceAdapter } from './tracing-DK-T8zX5.js';
+export { a as KortyxTelemetryConfig, d as KortyxTraceAdapter } from './tracing-DK-T8zX5.js';
 
 interface RunReasonEngineArgs {
     model: ProviderModelRef;
@@ -21,6 +22,7 @@ interface RunReasonEngineArgs {
     id?: string | undefined;
     opId?: string | undefined;
     segmentId?: string | undefined;
+    telemetry?: KortyxTraceMetadata | undefined;
     reasonTrace?: ReasonTraceAdapter | undefined;
 }
 interface RunReasonEngineResult {
@@ -33,4 +35,4 @@ interface RunReasonEngineResult {
 }
 declare function runReasonEngine(args: RunReasonEngineArgs): Promise<RunReasonEngineResult>;
 
-export { ReasonTraceAdapter, type RunReasonEngineArgs, type RunReasonEngineResult, runReasonEngine };
+export { KortyxTraceMetadata, ReasonTraceAdapter, type RunReasonEngineArgs, type RunReasonEngineResult, runReasonEngine };

package/dist/internal.js

@@ -80,7 +80,15 @@ async function runReasonEngine(args) {
       modelId: args.model.modelId,
       stream,
       emit,
+      ...args.telemetry?.operation ? { operation: args.telemetry.operation } : {},
+      ...args.telemetry?.prompt?.name ? { promptName: args.telemetry.prompt.name } : {},
+      ...args.telemetry?.prompt?.version !== void 0 ? { promptVersion: args.telemetry.prompt.version } : {},
+      ...args.telemetry?.prompt?.type ? { promptType: args.telemetry.prompt.type } : {},
       ...typeof args.nodeId === "string" && args.nodeId.length > 0 ? { nodeId: args.nodeId } : {}
+    },
+    telemetry: {
+      ...args.telemetry ?? {},
+      input: args.telemetry?.input ?? messages
     }
   });
   const endTrace = (span, result) => {
@@ -91,6 +99,10 @@ async function runReasonEngine(args) {
       ...result.warnings !== void 0 ? { warnings: result.warnings } : {},
       attributes: {
         textLength: result.text.length
+      },
+      telemetry: {
+        ...args.telemetry ?? {},
+        output: args.telemetry?.output ?? result.text
       }
     });
   };

package/dist/internal.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/internal.ts","../src/reason-engine.ts"],"sourcesContent":["export type {\n  RunReasonEngineArgs,\n  RunReasonEngineResult,\n} from \"./reason-engine\";\nexport { runReasonEngine } from \"./reason-engine\";\nexport type { ReasonTraceAdapter } from \"./tracing\";\n","import { randomUUID } from \"node:crypto\";\nimport type {\n  KortyxFinishReason,\n  KortyxPromptMessage,\n  KortyxProviderMetadata,\n  KortyxReasoningOptions,\n  KortyxResponseFormat,\n  KortyxUsage,\n  KortyxWarning,\n  ProviderModelRef,\n} from \"@kortyx/providers\";\nimport type { ReasonTraceAdapter, ReasonTraceSpan } from \"./tracing\";\n\nexport interface RunReasonEngineArgs {\n  model: ProviderModelRef;\n  input: string;\n  system?: string | undefined;\n  temperature?: number | undefined;\n  maxOutputTokens?: number | undefined;\n  stopSequences?: string[] | undefined;\n  abortSignal?: AbortSignal | undefined;\n  reasoning?: KortyxReasoningOptions | undefined;\n  responseFormat?: KortyxResponseFormat | undefined;\n  providerOptions?: Record<string, unknown> | undefined;\n  defaultTemperature?: number | undefined;\n  stream?: boolean | undefined;\n  emit?: boolean | undefined;\n  nodeId?: string | undefined;\n  emitEvent?: ((event: string, payload: unknown) => void) | undefined;\n  onTextChunk?: ((text: string) => void) | undefined;\n  id?: string | undefined;\n  opId?: string | undefined;\n  segmentId?: string | undefined;\n  reasonTrace?: ReasonTraceAdapter | undefined;\n}\n\nexport interface RunReasonEngineResult {\n  text: string;\n  raw?: unknown;\n  usage?: KortyxUsage;\n  finishReason?: KortyxFinishReason;\n  providerMetadata?: KortyxProviderMetadata;\n  warnings?: KortyxWarning[];\n}\n\nconst toError = (value: unknown): Error =>\n  value instanceof Error ? value : new Error(String(value));\n\nconst createRuntimeId = (): string => {\n  try {\n    return randomUUID();\n  } catch {\n    return `${Date.now()}-${Math.random().toString(36).slice(2)}`;\n  }\n};\n\nconst emitNodeEvent = (\n  emitEvent: RunReasonEngineArgs[\"emitEvent\"],\n  nodeId: string | undefined,\n  event: string,\n  payload: Record<string, unknown>,\n): void => {\n  if (!emitEvent) return;\n  emitEvent(event, {\n    ...payload,\n    ...(typeof nodeId === \"string\" && nodeId.length > 0\n      ? { node: nodeId }\n      : {}),\n  });\n};\n\nexport async function runReasonEngine(\n  args: RunReasonEngineArgs,\n): Promise<RunReasonEngineResult> {\n  const stream = args.stream ?? args.model.options?.streaming ?? true;\n  const emit = args.emit ?? true;\n  const temperature =\n    args.temperature ??\n    args.model.options?.temperature ??\n    args.defaultTemperature;\n  const maxOutputTokens =\n    args.maxOutputTokens ?? args.model.options?.maxOutputTokens;\n  const stopSequences = args.stopSequences ?? args.model.options?.stopSequences;\n  const abortSignal = args.abortSignal ?? args.model.options?.abortSignal;\n  const reasoning = args.reasoning ?? args.model.options?.reasoning;\n  const responseFormat =\n    args.responseFormat ?? args.model.options?.responseFormat;\n  const providerOptions =\n    args.providerOptions ?? args.model.options?.providerOptions;\n\n  const model = args.model.provider.getModel(args.model.modelId, {\n    ...(temperature !== undefined ? { temperature } : {}),\n    streaming: stream,\n    ...(maxOutputTokens !== undefined ? { maxOutputTokens } : {}),\n    ...(stopSequences !== undefined ? { stopSequences } : {}),\n    ...(abortSignal !== undefined ? { abortSignal } : {}),\n    ...(reasoning !== undefined ? { reasoning } : {}),\n    ...(responseFormat !== undefined ? { responseFormat } : {}),\n    ...(providerOptions !== undefined ? { providerOptions } : {}),\n  });\n\n  const messages: KortyxPromptMessage[] = [];\n  if (typeof args.system === \"string\" && args.system.length > 0) {\n    messages.push({ role: \"system\", content: args.system });\n  }\n  messages.push({ role: \"user\", content: String(args.input ?? \"\") });\n\n  const opId =\n    typeof args.opId === \"string\" && args.opId.length > 0\n      ? args.opId\n      : createRuntimeId();\n  const segmentId =\n    typeof args.segmentId === \"string\" && args.segmentId.length > 0\n      ? args.segmentId\n      : createRuntimeId();\n\n  const commonMeta: Record<string, unknown> = {\n    ...(typeof args.id === \"string\" && args.id.length > 0\n      ? { id: args.id }\n      : {}),\n    opId,\n    segmentId,\n  };\n  const traceSpan = args.reasonTrace?.startSpan({\n    name: \"runReasonEngine\",\n    attributes: {\n      ...commonMeta,\n      providerId: args.model.provider.id,\n      modelId: args.model.modelId,\n      stream,\n      emit,\n      ...(typeof args.nodeId === \"string\" && args.nodeId.length > 0\n        ? { nodeId: args.nodeId }\n        : {}),\n    },\n  });\n\n  const endTrace = (\n    span: ReasonTraceSpan | undefined,\n    result: RunReasonEngineResult,\n  ): void => {\n    span?.end?.({\n      ...(result.usage !== undefined ? { usage: result.usage } : {}),\n      ...(result.finishReason !== undefined\n        ? { finishReason: result.finishReason }\n        : {}),\n      ...(result.providerMetadata !== undefined\n        ? { providerMetadata: result.providerMetadata }\n        : {}),\n      ...(result.warnings !== undefined ? { warnings: result.warnings } : {}),\n      attributes: {\n        textLength: result.text.length,\n      },\n    });\n  };\n\n  const failTrace = (\n    span: ReasonTraceSpan | undefined,\n    error: unknown,\n  ): void => {\n    span?.fail?.(error, {\n      attributes: {\n        providerId: args.model.provider.id,\n        modelId: args.model.modelId,\n      },\n    });\n  };\n\n  try {\n    if (stream) {\n      let final = \"\";\n      let raw: unknown;\n      let usage: KortyxUsage | undefined;\n      let finishReason: KortyxFinishReason | undefined;\n      let providerMetadata: KortyxProviderMetadata | undefined;\n      let warnings: KortyxWarning[] | undefined;\n\n      if (emit) {\n        emitNodeEvent(args.emitEvent, args.nodeId, \"text-start\", {\n          ...commonMeta,\n        });\n      }\n\n      const response = await model.stream(messages);\n      for await (const chunk of response) {\n        switch (chunk.type) {\n          case \"text-delta\": {\n            if (chunk.delta.length === 0) break;\n            final += chunk.delta;\n            args.onTextChunk?.(chunk.delta);\n            if (chunk.raw !== undefined) {\n              raw = chunk.raw;\n            }\n            if (chunk.providerMetadata !== undefined) {\n              providerMetadata = chunk.providerMetadata;\n            }\n            if (emit) {\n              emitNodeEvent(args.emitEvent, args.nodeId, \"text-delta\", {\n                ...commonMeta,\n                delta: chunk.delta,\n              });\n            }\n            break;\n          }\n          case \"finish\": {\n            if (chunk.raw !== undefined) {\n              raw = chunk.raw;\n            }\n            if (chunk.usage !== undefined) {\n              usage = chunk.usage;\n            }\n            if (chunk.finishReason !== undefined) {\n              finishReason = chunk.finishReason;\n            }\n            if (chunk.providerMetadata !== undefined) {\n              providerMetadata = chunk.providerMetadata;\n            }\n            if (chunk.warnings !== undefined) {\n              warnings = chunk.warnings;\n            }\n            break;\n          }\n          case \"raw\": {\n            raw = chunk.raw;\n            if (chunk.providerMetadata !== undefined) {\n              providerMetadata = chunk.providerMetadata;\n            }\n            break;\n          }\n          case \"error\": {\n            if (chunk.raw !== undefined) {\n              raw = chunk.raw;\n            }\n            if (chunk.providerMetadata !== undefined) {\n              providerMetadata = chunk.providerMetadata;\n            }\n            if (chunk.warnings !== undefined) {\n              warnings = chunk.warnings;\n            }\n            throw toError(chunk.error);\n          }\n        }\n      }\n\n      if (emit) {\n        emitNodeEvent(args.emitEvent, args.nodeId, \"text-end\", {\n          ...commonMeta,\n        });\n      }\n\n      const result = {\n        text: final,\n        ...(raw !== undefined ? { raw } : {}),\n        ...(usage !== undefined ? { usage } : {}),\n        ...(finishReason !== undefined ? { finishReason } : {}),\n        ...(providerMetadata !== undefined ? { providerMetadata } : {}),\n        ...(warnings !== undefined ? { warnings } : {}),\n      };\n      endTrace(traceSpan, result);\n      return result;\n    }\n\n    const response = await model.invoke(messages);\n\n    if (emit) {\n      emitNodeEvent(args.emitEvent, args.nodeId, \"text-start\", {\n        ...commonMeta,\n      });\n      if (response.content) {\n        emitNodeEvent(args.emitEvent, args.nodeId, \"text-delta\", {\n          ...commonMeta,\n          delta: response.content,\n        });\n      }\n      emitNodeEvent(args.emitEvent, args.nodeId, \"text-end\", {\n        ...commonMeta,\n      });\n    }\n\n    const result = {\n      text: response.content,\n      ...(response.raw !== undefined ? { raw: response.raw } : {}),\n      ...(response.usage !== undefined ? { usage: response.usage } : {}),\n      ...(response.finishReason !== undefined\n        ? { finishReason: response.finishReason }\n        : {}),\n      ...(response.providerMetadata !== undefined\n        ? { providerMetadata: response.providerMetadata }\n        : {}),\n      ...(response.warnings !== undefined\n        ? { warnings: response.warnings }\n        : {}),\n    };\n    endTrace(traceSpan, result);\n    return result;\n  } catch (error) {\n    failTrace(traceSpan, error);\n    throw error;\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,yBAA2B;AA6C3B,IAAM,UAAU,CAAC,UACf,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC;AAE1D,IAAM,kBAAkB,MAAc;AACpC,MAAI;AACF,eAAO,+BAAW;AAAA,EACpB,QAAQ;AACN,WAAO,GAAG,KAAK,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,SAAS,EAAE,EAAE,MAAM,CAAC,CAAC;AAAA,EAC7D;AACF;AAEA,IAAM,gBAAgB,CACpB,WACA,QACA,OACA,YACS;AACT,MAAI,CAAC,UAAW;AAChB,YAAU,OAAO;AAAA,IACf,GAAG;AAAA,IACH,GAAI,OAAO,WAAW,YAAY,OAAO,SAAS,IAC9C,EAAE,MAAM,OAAO,IACf,CAAC;AAAA,EACP,CAAC;AACH;AAEA,eAAsB,gBACpB,MACgC;AAChC,QAAM,SAAS,KAAK,UAAU,KAAK,MAAM,SAAS,aAAa;AAC/D,QAAM,OAAO,KAAK,QAAQ;AAC1B,QAAM,cACJ,KAAK,eACL,KAAK,MAAM,SAAS,eACpB,KAAK;AACP,QAAM,kBACJ,KAAK,mBAAmB,KAAK,MAAM,SAAS;AAC9C,QAAM,gBAAgB,KAAK,iBAAiB,KAAK,MAAM,SAAS;AAChE,QAAM,cAAc,KAAK,eAAe,KAAK,MAAM,SAAS;AAC5D,QAAM,YAAY,KAAK,aAAa,KAAK,MAAM,SAAS;AACxD,QAAM,iBACJ,KAAK,kBAAkB,KAAK,MAAM,SAAS;AAC7C,QAAM,kBACJ,KAAK,mBAAmB,KAAK,MAAM,SAAS;AAE9C,QAAM,QAAQ,KAAK,MAAM,SAAS,SAAS,KAAK,MAAM,SAAS;AAAA,IAC7D,GAAI,gBAAgB,SAAY,EAAE,YAAY,IAAI,CAAC;AAAA,IACnD,WAAW;AAAA,IACX,GAAI,oBAAoB,SAAY,EAAE,gBAAgB,IAAI,CAAC;AAAA,IAC3D,GAAI,kBAAkB,SAAY,EAAE,cAAc,IAAI,CAAC;AAAA,IACvD,GAAI,gBAAgB,SAAY,EAAE,YAAY,IAAI,CAAC;AAAA,IACnD,GAAI,cAAc,SAAY,EAAE,UAAU,IAAI,CAAC;AAAA,IAC/C,GAAI,mBAAmB,SAAY,EAAE,eAAe,IAAI,CAAC;AAAA,IACzD,GAAI,oBAAoB,SAAY,EAAE,gBAAgB,IAAI,CAAC;AAAA,EAC7D,CAAC;AAED,QAAM,WAAkC,CAAC;AACzC,MAAI,OAAO,KAAK,WAAW,YAAY,KAAK,OAAO,SAAS,GAAG;AAC7D,aAAS,KAAK,EAAE,MAAM,UAAU,SAAS,KAAK,OAAO,CAAC;AAAA,EACxD;AACA,WAAS,KAAK,EAAE,MAAM,QAAQ,SAAS,OAAO,KAAK,SAAS,EAAE,EAAE,CAAC;AAEjE,QAAM,OACJ,OAAO,KAAK,SAAS,YAAY,KAAK,KAAK,SAAS,IAChD,KAAK,OACL,gBAAgB;AACtB,QAAM,YACJ,OAAO,KAAK,cAAc,YAAY,KAAK,UAAU,SAAS,IAC1D,KAAK,YACL,gBAAgB;AAEtB,QAAM,aAAsC;AAAA,IAC1C,GAAI,OAAO,KAAK,OAAO,YAAY,KAAK,GAAG,SAAS,IAChD,EAAE,IAAI,KAAK,GAAG,IACd,CAAC;AAAA,IACL;AAAA,IACA;AAAA,EACF;AACA,QAAM,YAAY,KAAK,aAAa,UAAU;AAAA,IAC5C,MAAM;AAAA,IACN,YAAY;AAAA,MACV,GAAG;AAAA,MACH,YAAY,KAAK,MAAM,SAAS;AAAA,MAChC,SAAS,KAAK,MAAM;AAAA,MACpB;AAAA,MACA;AAAA,MACA,GAAI,OAAO,KAAK,WAAW,YAAY,KAAK,OAAO,SAAS,IACxD,EAAE,QAAQ,KAAK,OAAO,IACtB,CAAC;AAAA,IACP;AAAA,EACF,CAAC;AAED,QAAM,WAAW,CACf,MACA,WACS;AACT,UAAM,MAAM;AAAA,MACV,GAAI,OAAO,UAAU,SAAY,EAAE,OAAO,OAAO,MAAM,IAAI,CAAC;AAAA,MAC5D,GAAI,OAAO,iBAAiB,SACxB,EAAE,cAAc,OAAO,aAAa,IACpC,CAAC;AAAA,MACL,GAAI,OAAO,qBAAqB,SAC5B,EAAE,kBAAkB,OAAO,iBAAiB,IAC5C,CAAC;AAAA,MACL,GAAI,OAAO,aAAa,SAAY,EAAE,UAAU,OAAO,SAAS,IAAI,CAAC;AAAA,MACrE,YAAY;AAAA,QACV,YAAY,OAAO,KAAK;AAAA,MAC1B;AAAA,IACF,CAAC;AAAA,EACH;AAEA,QAAM,YAAY,CAChB,MACA,UACS;AACT,UAAM,OAAO,OAAO;AAAA,MAClB,YAAY;AAAA,QACV,YAAY,KAAK,MAAM,SAAS;AAAA,QAChC,SAAS,KAAK,MAAM;AAAA,MACtB;AAAA,IACF,CAAC;AAAA,EACH;AAEA,MAAI;AACF,QAAI,QAAQ;AACV,UAAI,QAAQ;AACZ,UAAI;AACJ,UAAI;AACJ,UAAI;AACJ,UAAI;AACJ,UAAI;AAEJ,UAAI,MAAM;AACR,sBAAc,KAAK,WAAW,KAAK,QAAQ,cAAc;AAAA,UACvD,GAAG;AAAA,QACL,CAAC;AAAA,MACH;AAEA,YAAMA,YAAW,MAAM,MAAM,OAAO,QAAQ;AAC5C,uBAAiB,SAASA,WAAU;AAClC,gBAAQ,MAAM,MAAM;AAAA,UAClB,KAAK,cAAc;AACjB,gBAAI,MAAM,MAAM,WAAW,EAAG;AAC9B,qBAAS,MAAM;AACf,iBAAK,cAAc,MAAM,KAAK;AAC9B,gBAAI,MAAM,QAAQ,QAAW;AAC3B,oBAAM,MAAM;AAAA,YACd;AACA,gBAAI,MAAM,qBAAqB,QAAW;AACxC,iCAAmB,MAAM;AAAA,YAC3B;AACA,gBAAI,MAAM;AACR,4BAAc,KAAK,WAAW,KAAK,QAAQ,cAAc;AAAA,gBACvD,GAAG;AAAA,gBACH,OAAO,MAAM;AAAA,cACf,CAAC;AAAA,YACH;AACA;AAAA,UACF;AAAA,UACA,KAAK,UAAU;AACb,gBAAI,MAAM,QAAQ,QAAW;AAC3B,oBAAM,MAAM;AAAA,YACd;AACA,gBAAI,MAAM,UAAU,QAAW;AAC7B,sBAAQ,MAAM;AAAA,YAChB;AACA,gBAAI,MAAM,iBAAiB,QAAW;AACpC,6BAAe,MAAM;AAAA,YACvB;AACA,gBAAI,MAAM,qBAAqB,QAAW;AACxC,iCAAmB,MAAM;AAAA,YAC3B;AACA,gBAAI,MAAM,aAAa,QAAW;AAChC,yBAAW,MAAM;AAAA,YACnB;AACA;AAAA,UACF;AAAA,UACA,KAAK,OAAO;AACV,kBAAM,MAAM;AACZ,gBAAI,MAAM,qBAAqB,QAAW;AACxC,iCAAmB,MAAM;AAAA,YAC3B;AACA;AAAA,UACF;AAAA,UACA,KAAK,SAAS;AACZ,gBAAI,MAAM,QAAQ,QAAW;AAC3B,oBAAM,MAAM;AAAA,YACd;AACA,gBAAI,MAAM,qBAAqB,QAAW;AACxC,iCAAmB,MAAM;AAAA,YAC3B;AACA,gBAAI,MAAM,aAAa,QAAW;AAChC,yBAAW,MAAM;AAAA,YACnB;AACA,kBAAM,QAAQ,MAAM,KAAK;AAAA,UAC3B;AAAA,QACF;AAAA,MACF;AAEA,UAAI,MAAM;AACR,sBAAc,KAAK,WAAW,KAAK,QAAQ,YAAY;AAAA,UACrD,GAAG;AAAA,QACL,CAAC;AAAA,MACH;AAEA,YAAMC,UAAS;AAAA,QACb,MAAM;AAAA,QACN,GAAI,QAAQ,SAAY,EAAE,IAAI,IAAI,CAAC;AAAA,QACnC,GAAI,UAAU,SAAY,EAAE,MAAM,IAAI,CAAC;AAAA,QACvC,GAAI,iBAAiB,SAAY,EAAE,aAAa,IAAI,CAAC;AAAA,QACrD,GAAI,qBAAqB,SAAY,EAAE,iBAAiB,IAAI,CAAC;AAAA,QAC7D,GAAI,aAAa,SAAY,EAAE,SAAS,IAAI,CAAC;AAAA,MAC/C;AACA,eAAS,WAAWA,OAAM;AAC1B,aAAOA;AAAA,IACT;AAEA,UAAM,WAAW,MAAM,MAAM,OAAO,QAAQ;AAE5C,QAAI,MAAM;AACR,oBAAc,KAAK,WAAW,KAAK,QAAQ,cAAc;AAAA,QACvD,GAAG;AAAA,MACL,CAAC;AACD,UAAI,SAAS,SAAS;AACpB,sBAAc,KAAK,WAAW,KAAK,QAAQ,cAAc;AAAA,UACvD,GAAG;AAAA,UACH,OAAO,SAAS;AAAA,QAClB,CAAC;AAAA,MACH;AACA,oBAAc,KAAK,WAAW,KAAK,QAAQ,YAAY;AAAA,QACrD,GAAG;AAAA,MACL,CAAC;AAAA,IACH;AAEA,UAAM,SAAS;AAAA,MACb,MAAM,SAAS;AAAA,MACf,GAAI,SAAS,QAAQ,SAAY,EAAE,KAAK,SAAS,IAAI,IAAI,CAAC;AAAA,MAC1D,GAAI,SAAS,UAAU,SAAY,EAAE,OAAO,SAAS,MAAM,IAAI,CAAC;AAAA,MAChE,GAAI,SAAS,iBAAiB,SAC1B,EAAE,cAAc,SAAS,aAAa,IACtC,CAAC;AAAA,MACL,GAAI,SAAS,qBAAqB,SAC9B,EAAE,kBAAkB,SAAS,iBAAiB,IAC9C,CAAC;AAAA,MACL,GAAI,SAAS,aAAa,SACtB,EAAE,UAAU,SAAS,SAAS,IAC9B,CAAC;AAAA,IACP;AACA,aAAS,WAAW,MAAM;AAC1B,WAAO;AAAA,EACT,SAAS,OAAO;AACd,cAAU,WAAW,KAAK;AAC1B,UAAM;AAAA,EACR;AACF;","names":["response","result"]}
+{"version":3,"sources":["../src/internal.ts","../src/reason-engine.ts"],"sourcesContent":["export type {\n  RunReasonEngineArgs,\n  RunReasonEngineResult,\n} from \"./reason-engine\";\nexport { runReasonEngine } from \"./reason-engine\";\nexport type {\n  KortyxTelemetryConfig,\n  KortyxTraceAdapter,\n  KortyxTraceMetadata,\n  ReasonTraceAdapter,\n} from \"./tracing\";\n","import { randomUUID } from \"node:crypto\";\nimport type {\n  KortyxFinishReason,\n  KortyxPromptMessage,\n  KortyxProviderMetadata,\n  KortyxReasoningOptions,\n  KortyxResponseFormat,\n  KortyxUsage,\n  KortyxWarning,\n  ProviderModelRef,\n} from \"@kortyx/providers\";\nimport type {\n  KortyxTraceMetadata,\n  ReasonTraceAdapter,\n  ReasonTraceSpan,\n} from \"./tracing\";\n\nexport interface RunReasonEngineArgs {\n  model: ProviderModelRef;\n  input: string;\n  system?: string | undefined;\n  temperature?: number | undefined;\n  maxOutputTokens?: number | undefined;\n  stopSequences?: string[] | undefined;\n  abortSignal?: AbortSignal | undefined;\n  reasoning?: KortyxReasoningOptions | undefined;\n  responseFormat?: KortyxResponseFormat | undefined;\n  providerOptions?: Record<string, unknown> | undefined;\n  defaultTemperature?: number | undefined;\n  stream?: boolean | undefined;\n  emit?: boolean | undefined;\n  nodeId?: string | undefined;\n  emitEvent?: ((event: string, payload: unknown) => void) | undefined;\n  onTextChunk?: ((text: string) => void) | undefined;\n  id?: string | undefined;\n  opId?: string | undefined;\n  segmentId?: string | undefined;\n  telemetry?: KortyxTraceMetadata | undefined;\n  reasonTrace?: ReasonTraceAdapter | undefined;\n}\n\nexport interface RunReasonEngineResult {\n  text: string;\n  raw?: unknown;\n  usage?: KortyxUsage;\n  finishReason?: KortyxFinishReason;\n  providerMetadata?: KortyxProviderMetadata;\n  warnings?: KortyxWarning[];\n}\n\nconst toError = (value: unknown): Error =>\n  value instanceof Error ? value : new Error(String(value));\n\nconst createRuntimeId = (): string => {\n  try {\n    return randomUUID();\n  } catch {\n    return `${Date.now()}-${Math.random().toString(36).slice(2)}`;\n  }\n};\n\nconst emitNodeEvent = (\n  emitEvent: RunReasonEngineArgs[\"emitEvent\"],\n  nodeId: string | undefined,\n  event: string,\n  payload: Record<string, unknown>,\n): void => {\n  if (!emitEvent) return;\n  emitEvent(event, {\n    ...payload,\n    ...(typeof nodeId === \"string\" && nodeId.length > 0\n      ? { node: nodeId }\n      : {}),\n  });\n};\n\nexport async function runReasonEngine(\n  args: RunReasonEngineArgs,\n): Promise<RunReasonEngineResult> {\n  const stream = args.stream ?? args.model.options?.streaming ?? true;\n  const emit = args.emit ?? true;\n  const temperature =\n    args.temperature ??\n    args.model.options?.temperature ??\n    args.defaultTemperature;\n  const maxOutputTokens =\n    args.maxOutputTokens ?? args.model.options?.maxOutputTokens;\n  const stopSequences = args.stopSequences ?? args.model.options?.stopSequences;\n  const abortSignal = args.abortSignal ?? args.model.options?.abortSignal;\n  const reasoning = args.reasoning ?? args.model.options?.reasoning;\n  const responseFormat =\n    args.responseFormat ?? args.model.options?.responseFormat;\n  const providerOptions =\n    args.providerOptions ?? args.model.options?.providerOptions;\n\n  const model = args.model.provider.getModel(args.model.modelId, {\n    ...(temperature !== undefined ? { temperature } : {}),\n    streaming: stream,\n    ...(maxOutputTokens !== undefined ? { maxOutputTokens } : {}),\n    ...(stopSequences !== undefined ? { stopSequences } : {}),\n    ...(abortSignal !== undefined ? { abortSignal } : {}),\n    ...(reasoning !== undefined ? { reasoning } : {}),\n    ...(responseFormat !== undefined ? { responseFormat } : {}),\n    ...(providerOptions !== undefined ? { providerOptions } : {}),\n  });\n\n  const messages: KortyxPromptMessage[] = [];\n  if (typeof args.system === \"string\" && args.system.length > 0) {\n    messages.push({ role: \"system\", content: args.system });\n  }\n  messages.push({ role: \"user\", content: String(args.input ?? \"\") });\n\n  const opId =\n    typeof args.opId === \"string\" && args.opId.length > 0\n      ? args.opId\n      : createRuntimeId();\n  const segmentId =\n    typeof args.segmentId === \"string\" && args.segmentId.length > 0\n      ? args.segmentId\n      : createRuntimeId();\n\n  const commonMeta: Record<string, unknown> = {\n    ...(typeof args.id === \"string\" && args.id.length > 0\n      ? { id: args.id }\n      : {}),\n    opId,\n    segmentId,\n  };\n  const traceSpan = args.reasonTrace?.startSpan({\n    name: \"runReasonEngine\",\n    attributes: {\n      ...commonMeta,\n      providerId: args.model.provider.id,\n      modelId: args.model.modelId,\n      stream,\n      emit,\n      ...(args.telemetry?.operation\n        ? { operation: args.telemetry.operation }\n        : {}),\n      ...(args.telemetry?.prompt?.name\n        ? { promptName: args.telemetry.prompt.name }\n        : {}),\n      ...(args.telemetry?.prompt?.version !== undefined\n        ? { promptVersion: args.telemetry.prompt.version }\n        : {}),\n      ...(args.telemetry?.prompt?.type\n        ? { promptType: args.telemetry.prompt.type }\n        : {}),\n      ...(typeof args.nodeId === \"string\" && args.nodeId.length > 0\n        ? { nodeId: args.nodeId }\n        : {}),\n    },\n    telemetry: {\n      ...(args.telemetry ?? {}),\n      input: args.telemetry?.input ?? messages,\n    },\n  });\n\n  const endTrace = (\n    span: ReasonTraceSpan | undefined,\n    result: RunReasonEngineResult,\n  ): void => {\n    span?.end?.({\n      ...(result.usage !== undefined ? { usage: result.usage } : {}),\n      ...(result.finishReason !== undefined\n        ? { finishReason: result.finishReason }\n        : {}),\n      ...(result.providerMetadata !== undefined\n        ? { providerMetadata: result.providerMetadata }\n        : {}),\n      ...(result.warnings !== undefined ? { warnings: result.warnings } : {}),\n      attributes: {\n        textLength: result.text.length,\n      },\n      telemetry: {\n        ...(args.telemetry ?? {}),\n        output: args.telemetry?.output ?? result.text,\n      },\n    });\n  };\n\n  const failTrace = (\n    span: ReasonTraceSpan | undefined,\n    error: unknown,\n  ): void => {\n    span?.fail?.(error, {\n      attributes: {\n        providerId: args.model.provider.id,\n        modelId: args.model.modelId,\n      },\n    });\n  };\n\n  try {\n    if (stream) {\n      let final = \"\";\n      let raw: unknown;\n      let usage: KortyxUsage | undefined;\n      let finishReason: KortyxFinishReason | undefined;\n      let providerMetadata: KortyxProviderMetadata | undefined;\n      let warnings: KortyxWarning[] | undefined;\n\n      if (emit) {\n        emitNodeEvent(args.emitEvent, args.nodeId, \"text-start\", {\n          ...commonMeta,\n        });\n      }\n\n      const response = await model.stream(messages);\n      for await (const chunk of response) {\n        switch (chunk.type) {\n          case \"text-delta\": {\n            if (chunk.delta.length === 0) break;\n            final += chunk.delta;\n            args.onTextChunk?.(chunk.delta);\n            if (chunk.raw !== undefined) {\n              raw = chunk.raw;\n            }\n            if (chunk.providerMetadata !== undefined) {\n              providerMetadata = chunk.providerMetadata;\n            }\n            if (emit) {\n              emitNodeEvent(args.emitEvent, args.nodeId, \"text-delta\", {\n                ...commonMeta,\n                delta: chunk.delta,\n              });\n            }\n            break;\n          }\n          case \"finish\": {\n            if (chunk.raw !== undefined) {\n              raw = chunk.raw;\n            }\n            if (chunk.usage !== undefined) {\n              usage = chunk.usage;\n            }\n            if (chunk.finishReason !== undefined) {\n              finishReason = chunk.finishReason;\n            }\n            if (chunk.providerMetadata !== undefined) {\n              providerMetadata = chunk.providerMetadata;\n            }\n            if (chunk.warnings !== undefined) {\n              warnings = chunk.warnings;\n            }\n            break;\n          }\n          case \"raw\": {\n            raw = chunk.raw;\n            if (chunk.providerMetadata !== undefined) {\n              providerMetadata = chunk.providerMetadata;\n            }\n            break;\n          }\n          case \"error\": {\n            if (chunk.raw !== undefined) {\n              raw = chunk.raw;\n            }\n            if (chunk.providerMetadata !== undefined) {\n              providerMetadata = chunk.providerMetadata;\n            }\n            if (chunk.warnings !== undefined) {\n              warnings = chunk.warnings;\n            }\n            throw toError(chunk.error);\n          }\n        }\n      }\n\n      if (emit) {\n        emitNodeEvent(args.emitEvent, args.nodeId, \"text-end\", {\n          ...commonMeta,\n        });\n      }\n\n      const result = {\n        text: final,\n        ...(raw !== undefined ? { raw } : {}),\n        ...(usage !== undefined ? { usage } : {}),\n        ...(finishReason !== undefined ? { finishReason } : {}),\n        ...(providerMetadata !== undefined ? { providerMetadata } : {}),\n        ...(warnings !== undefined ? { warnings } : {}),\n      };\n      endTrace(traceSpan, result);\n      return result;\n    }\n\n    const response = await model.invoke(messages);\n\n    if (emit) {\n      emitNodeEvent(args.emitEvent, args.nodeId, \"text-start\", {\n        ...commonMeta,\n      });\n      if (response.content) {\n        emitNodeEvent(args.emitEvent, args.nodeId, \"text-delta\", {\n          ...commonMeta,\n          delta: response.content,\n        });\n      }\n      emitNodeEvent(args.emitEvent, args.nodeId, \"text-end\", {\n        ...commonMeta,\n      });\n    }\n\n    const result = {\n      text: response.content,\n      ...(response.raw !== undefined ? { raw: response.raw } : {}),\n      ...(response.usage !== undefined ? { usage: response.usage } : {}),\n      ...(response.finishReason !== undefined\n        ? { finishReason: response.finishReason }\n        : {}),\n      ...(response.providerMetadata !== undefined\n        ? { providerMetadata: response.providerMetadata }\n        : {}),\n      ...(response.warnings !== undefined\n        ? { warnings: response.warnings }\n        : {}),\n    };\n    endTrace(traceSpan, result);\n    return result;\n  } catch (error) {\n    failTrace(traceSpan, error);\n    throw error;\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,yBAA2B;AAkD3B,IAAM,UAAU,CAAC,UACf,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC;AAE1D,IAAM,kBAAkB,MAAc;AACpC,MAAI;AACF,eAAO,+BAAW;AAAA,EACpB,QAAQ;AACN,WAAO,GAAG,KAAK,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,SAAS,EAAE,EAAE,MAAM,CAAC,CAAC;AAAA,EAC7D;AACF;AAEA,IAAM,gBAAgB,CACpB,WACA,QACA,OACA,YACS;AACT,MAAI,CAAC,UAAW;AAChB,YAAU,OAAO;AAAA,IACf,GAAG;AAAA,IACH,GAAI,OAAO,WAAW,YAAY,OAAO,SAAS,IAC9C,EAAE,MAAM,OAAO,IACf,CAAC;AAAA,EACP,CAAC;AACH;AAEA,eAAsB,gBACpB,MACgC;AAChC,QAAM,SAAS,KAAK,UAAU,KAAK,MAAM,SAAS,aAAa;AAC/D,QAAM,OAAO,KAAK,QAAQ;AAC1B,QAAM,cACJ,KAAK,eACL,KAAK,MAAM,SAAS,eACpB,KAAK;AACP,QAAM,kBACJ,KAAK,mBAAmB,KAAK,MAAM,SAAS;AAC9C,QAAM,gBAAgB,KAAK,iBAAiB,KAAK,MAAM,SAAS;AAChE,QAAM,cAAc,KAAK,eAAe,KAAK,MAAM,SAAS;AAC5D,QAAM,YAAY,KAAK,aAAa,KAAK,MAAM,SAAS;AACxD,QAAM,iBACJ,KAAK,kBAAkB,KAAK,MAAM,SAAS;AAC7C,QAAM,kBACJ,KAAK,mBAAmB,KAAK,MAAM,SAAS;AAE9C,QAAM,QAAQ,KAAK,MAAM,SAAS,SAAS,KAAK,MAAM,SAAS;AAAA,IAC7D,GAAI,gBAAgB,SAAY,EAAE,YAAY,IAAI,CAAC;AAAA,IACnD,WAAW;AAAA,IACX,GAAI,oBAAoB,SAAY,EAAE,gBAAgB,IAAI,CAAC;AAAA,IAC3D,GAAI,kBAAkB,SAAY,EAAE,cAAc,IAAI,CAAC;AAAA,IACvD,GAAI,gBAAgB,SAAY,EAAE,YAAY,IAAI,CAAC;AAAA,IACnD,GAAI,cAAc,SAAY,EAAE,UAAU,IAAI,CAAC;AAAA,IAC/C,GAAI,mBAAmB,SAAY,EAAE,eAAe,IAAI,CAAC;AAAA,IACzD,GAAI,oBAAoB,SAAY,EAAE,gBAAgB,IAAI,CAAC;AAAA,EAC7D,CAAC;AAED,QAAM,WAAkC,CAAC;AACzC,MAAI,OAAO,KAAK,WAAW,YAAY,KAAK,OAAO,SAAS,GAAG;AAC7D,aAAS,KAAK,EAAE,MAAM,UAAU,SAAS,KAAK,OAAO,CAAC;AAAA,EACxD;AACA,WAAS,KAAK,EAAE,MAAM,QAAQ,SAAS,OAAO,KAAK,SAAS,EAAE,EAAE,CAAC;AAEjE,QAAM,OACJ,OAAO,KAAK,SAAS,YAAY,KAAK,KAAK,SAAS,IAChD,KAAK,OACL,gBAAgB;AACtB,QAAM,YACJ,OAAO,KAAK,cAAc,YAAY,KAAK,UAAU,SAAS,IAC1D,KAAK,YACL,gBAAgB;AAEtB,QAAM,aAAsC;AAAA,IAC1C,GAAI,OAAO,KAAK,OAAO,YAAY,KAAK,GAAG,SAAS,IAChD,EAAE,IAAI,KAAK,GAAG,IACd,CAAC;AAAA,IACL;AAAA,IACA;AAAA,EACF;AACA,QAAM,YAAY,KAAK,aAAa,UAAU;AAAA,IAC5C,MAAM;AAAA,IACN,YAAY;AAAA,MACV,GAAG;AAAA,MACH,YAAY,KAAK,MAAM,SAAS;AAAA,MAChC,SAAS,KAAK,MAAM;AAAA,MACpB;AAAA,MACA;AAAA,MACA,GAAI,KAAK,WAAW,YAChB,EAAE,WAAW,KAAK,UAAU,UAAU,IACtC,CAAC;AAAA,MACL,GAAI,KAAK,WAAW,QAAQ,OACxB,EAAE,YAAY,KAAK,UAAU,OAAO,KAAK,IACzC,CAAC;AAAA,MACL,GAAI,KAAK,WAAW,QAAQ,YAAY,SACpC,EAAE,eAAe,KAAK,UAAU,OAAO,QAAQ,IAC/C,CAAC;AAAA,MACL,GAAI,KAAK,WAAW,QAAQ,OACxB,EAAE,YAAY,KAAK,UAAU,OAAO,KAAK,IACzC,CAAC;AAAA,MACL,GAAI,OAAO,KAAK,WAAW,YAAY,KAAK,OAAO,SAAS,IACxD,EAAE,QAAQ,KAAK,OAAO,IACtB,CAAC;AAAA,IACP;AAAA,IACA,WAAW;AAAA,MACT,GAAI,KAAK,aAAa,CAAC;AAAA,MACvB,OAAO,KAAK,WAAW,SAAS;AAAA,IAClC;AAAA,EACF,CAAC;AAED,QAAM,WAAW,CACf,MACA,WACS;AACT,UAAM,MAAM;AAAA,MACV,GAAI,OAAO,UAAU,SAAY,EAAE,OAAO,OAAO,MAAM,IAAI,CAAC;AAAA,MAC5D,GAAI,OAAO,iBAAiB,SACxB,EAAE,cAAc,OAAO,aAAa,IACpC,CAAC;AAAA,MACL,GAAI,OAAO,qBAAqB,SAC5B,EAAE,kBAAkB,OAAO,iBAAiB,IAC5C,CAAC;AAAA,MACL,GAAI,OAAO,aAAa,SAAY,EAAE,UAAU,OAAO,SAAS,IAAI,CAAC;AAAA,MACrE,YAAY;AAAA,QACV,YAAY,OAAO,KAAK;AAAA,MAC1B;AAAA,MACA,WAAW;AAAA,QACT,GAAI,KAAK,aAAa,CAAC;AAAA,QACvB,QAAQ,KAAK,WAAW,UAAU,OAAO;AAAA,MAC3C;AAAA,IACF,CAAC;AAAA,EACH;AAEA,QAAM,YAAY,CAChB,MACA,UACS;AACT,UAAM,OAAO,OAAO;AAAA,MAClB,YAAY;AAAA,QACV,YAAY,KAAK,MAAM,SAAS;AAAA,QAChC,SAAS,KAAK,MAAM;AAAA,MACtB;AAAA,IACF,CAAC;AAAA,EACH;AAEA,MAAI;AACF,QAAI,QAAQ;AACV,UAAI,QAAQ;AACZ,UAAI;AACJ,UAAI;AACJ,UAAI;AACJ,UAAI;AACJ,UAAI;AAEJ,UAAI,MAAM;AACR,sBAAc,KAAK,WAAW,KAAK,QAAQ,cAAc;AAAA,UACvD,GAAG;AAAA,QACL,CAAC;AAAA,MACH;AAEA,YAAMA,YAAW,MAAM,MAAM,OAAO,QAAQ;AAC5C,uBAAiB,SAASA,WAAU;AAClC,gBAAQ,MAAM,MAAM;AAAA,UAClB,KAAK,cAAc;AACjB,gBAAI,MAAM,MAAM,WAAW,EAAG;AAC9B,qBAAS,MAAM;AACf,iBAAK,cAAc,MAAM,KAAK;AAC9B,gBAAI,MAAM,QAAQ,QAAW;AAC3B,oBAAM,MAAM;AAAA,YACd;AACA,gBAAI,MAAM,qBAAqB,QAAW;AACxC,iCAAmB,MAAM;AAAA,YAC3B;AACA,gBAAI,MAAM;AACR,4BAAc,KAAK,WAAW,KAAK,QAAQ,cAAc;AAAA,gBACvD,GAAG;AAAA,gBACH,OAAO,MAAM;AAAA,cACf,CAAC;AAAA,YACH;AACA;AAAA,UACF;AAAA,UACA,KAAK,UAAU;AACb,gBAAI,MAAM,QAAQ,QAAW;AAC3B,oBAAM,MAAM;AAAA,YACd;AACA,gBAAI,MAAM,UAAU,QAAW;AAC7B,sBAAQ,MAAM;AAAA,YAChB;AACA,gBAAI,MAAM,iBAAiB,QAAW;AACpC,6BAAe,MAAM;AAAA,YACvB;AACA,gBAAI,MAAM,qBAAqB,QAAW;AACxC,iCAAmB,MAAM;AAAA,YAC3B;AACA,gBAAI,MAAM,aAAa,QAAW;AAChC,yBAAW,MAAM;AAAA,YACnB;AACA;AAAA,UACF;AAAA,UACA,KAAK,OAAO;AACV,kBAAM,MAAM;AACZ,gBAAI,MAAM,qBAAqB,QAAW;AACxC,iCAAmB,MAAM;AAAA,YAC3B;AACA;AAAA,UACF;AAAA,UACA,KAAK,SAAS;AACZ,gBAAI,MAAM,QAAQ,QAAW;AAC3B,oBAAM,MAAM;AAAA,YACd;AACA,gBAAI,MAAM,qBAAqB,QAAW;AACxC,iCAAmB,MAAM;AAAA,YAC3B;AACA,gBAAI,MAAM,aAAa,QAAW;AAChC,yBAAW,MAAM;AAAA,YACnB;AACA,kBAAM,QAAQ,MAAM,KAAK;AAAA,UAC3B;AAAA,QACF;AAAA,MACF;AAEA,UAAI,MAAM;AACR,sBAAc,KAAK,WAAW,KAAK,QAAQ,YAAY;AAAA,UACrD,GAAG;AAAA,QACL,CAAC;AAAA,MACH;AAEA,YAAMC,UAAS;AAAA,QACb,MAAM;AAAA,QACN,GAAI,QAAQ,SAAY,EAAE,IAAI,IAAI,CAAC;AAAA,QACnC,GAAI,UAAU,SAAY,EAAE,MAAM,IAAI,CAAC;AAAA,QACvC,GAAI,iBAAiB,SAAY,EAAE,aAAa,IAAI,CAAC;AAAA,QACrD,GAAI,qBAAqB,SAAY,EAAE,iBAAiB,IAAI,CAAC;AAAA,QAC7D,GAAI,aAAa,SAAY,EAAE,SAAS,IAAI,CAAC;AAAA,MAC/C;AACA,eAAS,WAAWA,OAAM;AAC1B,aAAOA;AAAA,IACT;AAEA,UAAM,WAAW,MAAM,MAAM,OAAO,QAAQ;AAE5C,QAAI,MAAM;AACR,oBAAc,KAAK,WAAW,KAAK,QAAQ,cAAc;AAAA,QACvD,GAAG;AAAA,MACL,CAAC;AACD,UAAI,SAAS,SAAS;AACpB,sBAAc,KAAK,WAAW,KAAK,QAAQ,cAAc;AAAA,UACvD,GAAG;AAAA,UACH,OAAO,SAAS;AAAA,QAClB,CAAC;AAAA,MACH;AACA,oBAAc,KAAK,WAAW,KAAK,QAAQ,YAAY;AAAA,QACrD,GAAG;AAAA,MACL,CAAC;AAAA,IACH;AAEA,UAAM,SAAS;AAAA,MACb,MAAM,SAAS;AAAA,MACf,GAAI,SAAS,QAAQ,SAAY,EAAE,KAAK,SAAS,IAAI,IAAI,CAAC;AAAA,MAC1D,GAAI,SAAS,UAAU,SAAY,EAAE,OAAO,SAAS,MAAM,IAAI,CAAC;AAAA,MAChE,GAAI,SAAS,iBAAiB,SAC1B,EAAE,cAAc,SAAS,aAAa,IACtC,CAAC;AAAA,MACL,GAAI,SAAS,qBAAqB,SAC9B,EAAE,kBAAkB,SAAS,iBAAiB,IAC9C,CAAC;AAAA,MACL,GAAI,SAAS,aAAa,SACtB,EAAE,UAAU,SAAS,SAAS,IAC9B,CAAC;AAAA,IACP;AACA,aAAS,WAAW,MAAM;AAC1B,WAAO;AAAA,EACT,SAAS,OAAO;AACd,cAAU,WAAW,KAAK;AAC1B,UAAM;AAAA,EACR;AACF;","names":["response","result"]}

package/dist/tracing-DK-T8zX5.d.ts

@@ -0,0 +1,55 @@
+import { KortyxUsage, KortyxFinishReason, KortyxProviderMetadata, KortyxWarning } from '@kortyx/providers';
+
+type ReasonTraceAttributes = Record<string, unknown>;
+type KortyxTelemetryPrompt = {
+    name?: string | undefined;
+    version?: string | number | undefined;
+    type?: "text" | "chat" | (string & {}) | undefined;
+    source?: string | undefined;
+    metadata?: unknown;
+};
+type KortyxTelemetryContentCapture = boolean | {
+    input?: boolean | undefined;
+    output?: boolean | undefined;
+};
+type KortyxTraceMetadata = {
+    operation?: string | undefined;
+    prompt?: KortyxTelemetryPrompt | undefined;
+    metadata?: Record<string, unknown> | undefined;
+    tags?: string[] | undefined;
+    input?: unknown;
+    output?: unknown;
+    captureContent?: KortyxTelemetryContentCapture | undefined;
+};
+interface ReasonTraceSpanStartArgs {
+    name: string;
+    attributes?: ReasonTraceAttributes;
+    telemetry?: KortyxTraceMetadata | undefined;
+}
+interface ReasonTraceSpanEndArgs {
+    attributes?: ReasonTraceAttributes;
+    telemetry?: KortyxTraceMetadata | undefined;
+    usage?: KortyxUsage;
+    finishReason?: KortyxFinishReason;
+    providerMetadata?: KortyxProviderMetadata;
+    warnings?: KortyxWarning[];
+}
+interface ReasonTraceSpan {
+    setAttributes?: (attributes: ReasonTraceAttributes) => void;
+    addEvent?: (name: string, attributes?: ReasonTraceAttributes) => void;
+    end?: (args?: ReasonTraceSpanEndArgs) => void;
+    fail?: (error: unknown, args?: ReasonTraceSpanEndArgs) => void;
+}
+interface ReasonTraceAdapter {
+    startSpan: (args: ReasonTraceSpanStartArgs) => ReasonTraceSpan | undefined;
+    withSpan?: <T>(args: ReasonTraceSpanStartArgs, fn: (span: ReasonTraceSpan) => T | Promise<T>) => Promise<T>;
+}
+type KortyxTraceAdapter = ReasonTraceAdapter;
+type KortyxTelemetryConfig = {
+    trace?: KortyxTraceAdapter | undefined;
+    metadata?: Record<string, unknown> | undefined;
+    tags?: string[] | undefined;
+    captureContent?: KortyxTelemetryContentCapture | undefined;
+};
+
+export type { KortyxTraceMetadata as K, ReasonTraceAdapter as R, KortyxTelemetryConfig as a, KortyxTelemetryContentCapture as b, KortyxTelemetryPrompt as c, KortyxTraceAdapter as d, ReasonTraceAttributes as e, ReasonTraceSpan as f, ReasonTraceSpanEndArgs as g, ReasonTraceSpanStartArgs as h };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kortyx/hooks",
-  "version": "0.13.0",
+  "version": "0.15.0",
   "private": false,
   "description": "Lifecycle hooks and extension points for Kortyx.",
   "keywords": [

package/dist/tracing-BG-j1cer.d.ts

@@ -1,25 +0,0 @@
-import { KortyxUsage, KortyxFinishReason, KortyxProviderMetadata, KortyxWarning } from '@kortyx/providers';
-
-type ReasonTraceAttributes = Record<string, unknown>;
-interface ReasonTraceSpanStartArgs {
-    name: "useReason" | "runReasonEngine";
-    attributes?: ReasonTraceAttributes;
-}
-interface ReasonTraceSpanEndArgs {
-    attributes?: ReasonTraceAttributes;
-    usage?: KortyxUsage;
-    finishReason?: KortyxFinishReason;
-    providerMetadata?: KortyxProviderMetadata;
-    warnings?: KortyxWarning[];
-}
-interface ReasonTraceSpan {
-    setAttributes?: (attributes: ReasonTraceAttributes) => void;
-    addEvent?: (name: string, attributes?: ReasonTraceAttributes) => void;
-    end?: (args?: ReasonTraceSpanEndArgs) => void;
-    fail?: (error: unknown, args?: ReasonTraceSpanEndArgs) => void;
-}
-interface ReasonTraceAdapter {
-    startSpan: (args: ReasonTraceSpanStartArgs) => ReasonTraceSpan | undefined;
-}
-
-export type { ReasonTraceAdapter as R, ReasonTraceAttributes as a, ReasonTraceSpan as b, ReasonTraceSpanEndArgs as c, ReasonTraceSpanStartArgs as d };