@prefactor/langchain 0.1.0

package/dist/index.cjs

@@ -0,0 +1,252 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __moduleCache = /* @__PURE__ */ new WeakMap;
+var __toCommonJS = (from) => {
+  var entry = __moduleCache.get(from), desc;
+  if (entry)
+    return entry;
+  entry = __defProp({}, "__esModule", { value: true });
+  if (from && typeof from === "object" || typeof from === "function")
+    __getOwnPropNames(from).map((key) => !__hasOwnProp.call(entry, key) && __defProp(entry, key, {
+      get: () => from[key],
+      enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable
+    }));
+  __moduleCache.set(from, entry);
+  return entry;
+};
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, {
+      get: all[name],
+      enumerable: true,
+      configurable: true,
+      set: (newValue) => all[name] = () => newValue
+    });
+};
+
+// packages/langchain/src/index.ts
+var exports_src = {};
+__export(exports_src, {
+  shutdown: () => shutdown,
+  init: () => init,
+  getTracer: () => getTracer,
+  extractTokenUsage: () => extractTokenUsage,
+  SpanType: () => import_core3.SpanType,
+  SpanStatus: () => import_core3.SpanStatus,
+  PrefactorMiddleware: () => PrefactorMiddleware
+});
+module.exports = __toCommonJS(exports_src);
+var import_core3 = require("@prefactor/core");
+
+// packages/langchain/src/init.ts
+var import_core2 = require("@prefactor/core");
+var import_pfid = require("@prefactor/pfid");
+var import_langchain = require("langchain");
+
+// packages/langchain/src/middleware.ts
+var import_core = require("@prefactor/core");
+
+// packages/langchain/src/metadata-extractor.ts
+function extractTokenUsage(response) {
+  try {
+    const tokenUsage = response?.token_usage ?? response?.usage;
+    if (tokenUsage) {
+      return {
+        promptTokens: tokenUsage.prompt_tokens ?? 0,
+        completionTokens: tokenUsage.completion_tokens ?? 0,
+        totalTokens: tokenUsage.total_tokens ?? 0
+      };
+    }
+    const usageMetadata = response?.usage_metadata;
+    if (usageMetadata) {
+      return {
+        promptTokens: usageMetadata.input_tokens ?? 0,
+        completionTokens: usageMetadata.output_tokens ?? 0,
+        totalTokens: usageMetadata.total_tokens ?? 0
+      };
+    }
+    const responseMetadata = response?.response_metadata;
+    if (responseMetadata?.token_usage) {
+      return {
+        promptTokens: responseMetadata.token_usage.prompt_tokens ?? 0,
+        completionTokens: responseMetadata.token_usage.completion_tokens ?? 0,
+        totalTokens: responseMetadata.token_usage.total_tokens ?? 0
+      };
+    }
+    return null;
+  } catch {
+    return null;
+  }
+}
+
+// packages/langchain/src/middleware.ts
+class PrefactorMiddleware {
+  tracer;
+  rootSpan = null;
+  constructor(tracer) {
+    this.tracer = tracer;
+  }
+  async beforeAgent(state) {
+    const parentSpan = import_core.SpanContext.getCurrent();
+    const messages = state?.messages ?? [];
+    this.tracer.startAgentInstance();
+    const span = this.tracer.startSpan({
+      name: "agent",
+      spanType: import_core.SpanType.AGENT,
+      inputs: { messages: messages.slice(-3).map((m) => String(m)) },
+      parentSpanId: parentSpan?.spanId,
+      traceId: parentSpan?.traceId
+    });
+    this.rootSpan = span;
+  }
+  async afterAgent(state) {
+    if (!this.rootSpan) {
+      return;
+    }
+    const messages = state?.messages ?? [];
+    this.tracer.endSpan(this.rootSpan, {
+      outputs: { messages: messages.slice(-3).map((m) => String(m)) }
+    });
+    this.tracer.finishAgentInstance();
+    import_core.SpanContext.clear();
+    this.rootSpan = null;
+  }
+  async wrapModelCall(request, handler) {
+    const parentSpan = import_core.SpanContext.getCurrent();
+    const span = this.tracer.startSpan({
+      name: this.extractModelName(request),
+      spanType: import_core.SpanType.LLM,
+      inputs: this.extractModelInputs(request),
+      parentSpanId: parentSpan?.spanId,
+      traceId: parentSpan?.traceId
+    });
+    try {
+      const response = await import_core.SpanContext.runAsync(span, async () => {
+        return handler(request);
+      });
+      const outputs = this.extractModelOutputs(response);
+      const tokenUsage = extractTokenUsage(response);
+      this.tracer.endSpan(span, { outputs, tokenUsage: tokenUsage ?? undefined });
+      return response;
+    } catch (error) {
+      this.tracer.endSpan(span, { error });
+      throw error;
+    }
+  }
+  async wrapToolCall(request, handler) {
+    const parentSpan = import_core.SpanContext.getCurrent();
+    const span = this.tracer.startSpan({
+      name: this.extractToolName(request),
+      spanType: import_core.SpanType.TOOL,
+      inputs: this.extractToolInputs(request),
+      parentSpanId: parentSpan?.spanId,
+      traceId: parentSpan?.traceId
+    });
+    try {
+      const response = await import_core.SpanContext.runAsync(span, async () => {
+        return handler(request);
+      });
+      this.tracer.endSpan(span, {
+        outputs: this.extractToolOutputs(response)
+      });
+      return response;
+    } catch (error) {
+      this.tracer.endSpan(span, { error });
+      throw error;
+    }
+  }
+  extractModelName(request) {
+    return request?.model ?? request?.modelName ?? "unknown";
+  }
+  extractModelInputs(request) {
+    const messages = request?.messages ?? [];
+    return { messages: messages.slice(-3).map((m) => String(m)) };
+  }
+  extractModelOutputs(response) {
+    const content = response?.content ?? response?.text ?? "";
+    return { content: String(content) };
+  }
+  extractToolName(request) {
+    return request?.name ?? request?.tool ?? "unknown";
+  }
+  extractToolInputs(request) {
+    return { input: request?.input ?? request?.args ?? {} };
+  }
+  extractToolOutputs(response) {
+    return { output: response?.output ?? response };
+  }
+}
+
+// packages/langchain/src/init.ts
+var logger = import_core2.getLogger("init");
+var globalTracer = null;
+var globalMiddleware = null;
+function init(config) {
+  import_core2.configureLogging();
+  const finalConfig = import_core2.createConfig(config);
+  logger.info("Initializing Prefactor SDK", { transport: finalConfig.transportType });
+  if (globalMiddleware !== null) {
+    return globalMiddleware;
+  }
+  let transport;
+  if (finalConfig.transportType === "stdio") {
+    transport = new import_core2.StdioTransport;
+  } else {
+    if (!finalConfig.httpConfig) {
+      throw new Error("HTTP transport requires httpConfig to be provided in configuration");
+    }
+    const httpConfig = import_core2.HttpTransportConfigSchema.parse(finalConfig.httpConfig);
+    transport = new import_core2.HttpTransport(httpConfig);
+  }
+  let partition;
+  if (finalConfig.httpConfig?.agentId) {
+    try {
+      partition = import_pfid.extractPartition(finalConfig.httpConfig.agentId);
+      logger.debug("Extracted partition from agent_id", { partition });
+    } catch (error) {
+      logger.warn("Failed to extract partition from agent_id, using random partition", { error });
+    }
+  }
+  globalTracer = new import_core2.Tracer(transport, partition);
+  const prefactorMiddleware = new PrefactorMiddleware(globalTracer);
+  const middleware = import_langchain.createMiddleware({
+    name: "prefactor",
+    wrapModelCall: async (request, handler) => {
+      return prefactorMiddleware.wrapModelCall(request, handler);
+    },
+    wrapToolCall: async (request, handler) => {
+      return prefactorMiddleware.wrapToolCall(request, handler);
+    },
+    beforeAgent: async (state) => {
+      await prefactorMiddleware.beforeAgent(state);
+    },
+    afterAgent: async (state) => {
+      await prefactorMiddleware.afterAgent(state);
+    }
+  });
+  globalMiddleware = middleware;
+  return middleware;
+}
+function getTracer() {
+  if (!globalTracer) {
+    init();
+  }
+  return globalTracer;
+}
+async function shutdown() {
+  if (globalTracer) {
+    logger.info("Shutting down Prefactor SDK");
+    await globalTracer.close();
+  }
+  globalTracer = null;
+  globalMiddleware = null;
+}
+process.on("beforeExit", () => {
+  shutdown().catch((error) => {
+    console.error("Error during Prefactor SDK shutdown:", error);
+  });
+});
+
+//# debugId=46B60F8DD9AA982064756E2164756E21

package/dist/index.cjs.map

@@ -0,0 +1,13 @@
+{
+  "version": 3,
+  "sources": ["../src/index.ts", "../src/init.ts", "../src/middleware.ts", "../src/metadata-extractor.ts"],
+  "sourcesContent": [
+    "// Main entry points\n\n// Convenience re-exports from core\nexport {\n  type Config,\n  type HttpTransportConfig,\n  type Span,\n  SpanStatus,\n  SpanType,\n} from '@prefactor/core';\nexport { getTracer, init, shutdown } from './init.js';\nexport { extractTokenUsage } from './metadata-extractor.js';\n// Middleware\nexport { PrefactorMiddleware } from './middleware.js';\n",
+    "import {\n  type Config,\n  configureLogging,\n  createConfig,\n  getLogger,\n  HttpTransport,\n  HttpTransportConfigSchema,\n  StdioTransport,\n  Tracer,\n  type Transport,\n} from '@prefactor/core';\nimport { extractPartition, type Partition } from '@prefactor/pfid';\nimport { type AgentMiddleware, createMiddleware } from 'langchain';\nimport { PrefactorMiddleware } from './middleware.js';\n\nconst logger = getLogger('init');\n\nlet globalTracer: Tracer | null = null;\nlet globalMiddleware: AgentMiddleware | null = null;\n\n/**\n * Initialize the Prefactor SDK and return middleware for LangChain.js\n *\n * This is the main entry point for the SDK. Call this function to create a middleware\n * instance that you can pass to your LangChain.js agents.\n *\n * @param config - Optional configuration object\n * @returns PrefactorMiddleware instance to use with LangChain.js agents\n *\n * @example\n * ```typescript\n * import { init } from '@prefactor/langchain';\n * import { createAgent } from 'langchain';\n *\n * // Initialize with defaults (stdio transport)\n * const middleware = init();\n *\n * // Or configure HTTP transport\n * const middleware = init({\n *   transportType: 'http',\n *   httpConfig: {\n *     apiUrl: 'https://api.prefactor.ai',\n *     apiToken: process.env.PREFACTOR_API_TOKEN!,\n *   }\n * });\n *\n * const agent = createAgent({\n *   model: 'claude-sonnet-4-5-20250929',\n *   tools: [myTool],\n *   middleware: [middleware],\n * });\n * ```\n */\nexport function init(config?: Partial<Config>): AgentMiddleware {\n  configureLogging();\n\n  const finalConfig = createConfig(config);\n  logger.info('Initializing Prefactor SDK', { transport: finalConfig.transportType });\n\n  if (globalMiddleware !== null) {\n    return globalMiddleware;\n  }\n\n  let transport: Transport;\n  if (finalConfig.transportType === 'stdio') {\n    transport = new StdioTransport();\n  } else {\n    if (!finalConfig.httpConfig) {\n      throw new Error('HTTP transport requires httpConfig to be provided in configuration');\n    }\n    // Parse httpConfig to apply defaults from schema\n    const httpConfig = HttpTransportConfigSchema.parse(finalConfig.httpConfig);\n    transport = new HttpTransport(httpConfig);\n  }\n\n  // Extract partition from agent_id if provided (for HTTP transport)\n  let partition: Partition | undefined;\n  if (finalConfig.httpConfig?.agentId) {\n    try {\n      partition = extractPartition(finalConfig.httpConfig.agentId);\n      logger.debug('Extracted partition from agent_id', { partition });\n    } catch (error) {\n      logger.warn('Failed to extract partition from agent_id, using random partition', { error });\n    }\n  }\n\n  globalTracer = new Tracer(transport, partition);\n  const prefactorMiddleware = new PrefactorMiddleware(globalTracer);\n\n  const middleware = createMiddleware({\n    name: 'prefactor',\n    // biome-ignore lint/suspicious/noExplicitAny: LangChain middleware hooks use dynamic types\n    wrapModelCall: async (request: any, handler: any) => {\n      return prefactorMiddleware.wrapModelCall(request, handler);\n    },\n    // biome-ignore lint/suspicious/noExplicitAny: LangChain middleware hooks use dynamic types\n    wrapToolCall: async (request: any, handler: any) => {\n      return prefactorMiddleware.wrapToolCall(request, handler);\n    },\n    // biome-ignore lint/suspicious/noExplicitAny: LangChain middleware hooks use dynamic types\n    beforeAgent: async (state: any) => {\n      await prefactorMiddleware.beforeAgent(state);\n    },\n    // biome-ignore lint/suspicious/noExplicitAny: LangChain middleware hooks use dynamic types\n    afterAgent: async (state: any) => {\n      await prefactorMiddleware.afterAgent(state);\n    },\n  });\n\n  globalMiddleware = middleware;\n  return middleware;\n}\n\n/**\n * Get the current tracer instance.\n *\n * If no tracer has been created yet, this will call init() with default configuration.\n *\n * @returns Tracer instance\n *\n * @example\n * ```typescript\n * import { getTracer } from '@prefactor/langchain';\n *\n * const tracer = getTracer();\n * const span = tracer.startSpan({\n *   name: 'custom-operation',\n *   spanType: SpanType.TOOL,\n *   inputs: { data: 'example' }\n * });\n * ```\n */\nexport function getTracer(): Tracer {\n  if (!globalTracer) {\n    init();\n  }\n  // Safe because init() always sets globalTracer\n  return globalTracer as Tracer;\n}\n\n/**\n * Shutdown the SDK and flush any pending spans.\n *\n * Call this before your application exits to ensure all spans are sent to the transport.\n * This is especially important for HTTP transport which has a queue of pending requests.\n *\n * @returns Promise that resolves when shutdown is complete\n *\n * @example\n * ```typescript\n * import { shutdown } from '@prefactor/langchain';\n *\n * process.on('SIGTERM', async () => {\n *   await shutdown();\n *   process.exit(0);\n * });\n * ```\n */\nexport async function shutdown(): Promise<void> {\n  if (globalTracer) {\n    logger.info('Shutting down Prefactor SDK');\n    await globalTracer.close();\n  }\n  globalTracer = null;\n  globalMiddleware = null;\n}\n\n// Automatic shutdown on process exit\nprocess.on('beforeExit', () => {\n  shutdown().catch((error) => {\n    console.error('Error during Prefactor SDK shutdown:', error);\n  });\n});\n",
+    "import { type Span, SpanContext, SpanType, type Tracer } from '@prefactor/core';\nimport { extractTokenUsage } from './metadata-extractor.js';\n\n/**\n * Prefactor middleware for LangChain.js agents.\n *\n * This middleware automatically traces LLM calls, tool executions, and agent workflows.\n * It integrates with LangChain.js middleware API to provide transparent instrumentation.\n *\n * Features:\n * - Automatic parent-child span relationships via context propagation\n * - Token usage extraction for LLM calls\n * - Error tracking and debugging\n * - Zero-overhead instrumentation (graceful failure)\n *\n * @example\n * ```typescript\n * import { init } from '@prefactor/langchain';\n * import { createReactAgent } from '@langchain/langgraph/prebuilt';\n *\n * const middleware = init();\n * const agent = createReactAgent({\n *   llm: model,\n *   tools: [myTool],\n *   middleware: [middleware],\n * });\n * ```\n */\nexport class PrefactorMiddleware {\n  private rootSpan: Span | null = null;\n\n  constructor(private tracer: Tracer) {}\n\n  /**\n   * Called before agent execution starts\n   *\n   * @param state - Agent state containing messages\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain state can be any structure\n  async beforeAgent(state: any): Promise<void> {\n    const parentSpan = SpanContext.getCurrent();\n    const messages = state?.messages ?? [];\n\n    this.tracer.startAgentInstance();\n\n    const span = this.tracer.startSpan({\n      name: 'agent',\n      spanType: SpanType.AGENT,\n      inputs: { messages: messages.slice(-3).map((m: unknown) => String(m)) },\n      parentSpanId: parentSpan?.spanId,\n      traceId: parentSpan?.traceId,\n    });\n\n    this.rootSpan = span;\n  }\n\n  /**\n   * Called after agent execution completes\n   *\n   * @param state - Agent state containing messages\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain state can be any structure\n  async afterAgent(state: any): Promise<void> {\n    if (!this.rootSpan) {\n      return;\n    }\n\n    const messages = state?.messages ?? [];\n    this.tracer.endSpan(this.rootSpan, {\n      outputs: { messages: messages.slice(-3).map((m: unknown) => String(m)) },\n    });\n\n    this.tracer.finishAgentInstance();\n    SpanContext.clear();\n    this.rootSpan = null;\n  }\n\n  /**\n   * Wrap a model call to trace LLM invocations\n   *\n   * @param request - Model invocation request\n   * @param handler - The actual model call function\n   * @returns Promise resolving to the model response\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain request/handler types are dynamic\n  async wrapModelCall<T>(request: any, handler: (req: any) => Promise<T>): Promise<T> {\n    const parentSpan = SpanContext.getCurrent();\n\n    const span = this.tracer.startSpan({\n      name: this.extractModelName(request),\n      spanType: SpanType.LLM,\n      inputs: this.extractModelInputs(request),\n      parentSpanId: parentSpan?.spanId,\n      traceId: parentSpan?.traceId,\n    });\n\n    try {\n      // CRITICAL: Wrap handler in context so child operations see this span\n      const response = await SpanContext.runAsync(span, async () => {\n        return handler(request);\n      });\n\n      const outputs = this.extractModelOutputs(response);\n      const tokenUsage = extractTokenUsage(response);\n\n      this.tracer.endSpan(span, { outputs, tokenUsage: tokenUsage ?? undefined });\n      return response;\n    } catch (error) {\n      this.tracer.endSpan(span, { error: error as Error });\n      throw error;\n    }\n  }\n\n  /**\n   * Wrap a tool call to trace tool executions\n   *\n   * @param request - Tool invocation request\n   * @param handler - The actual tool call function\n   * @returns Promise resolving to the tool response\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain request/handler types are dynamic\n  async wrapToolCall<T>(request: any, handler: (req: any) => Promise<T>): Promise<T> {\n    const parentSpan = SpanContext.getCurrent();\n\n    const span = this.tracer.startSpan({\n      name: this.extractToolName(request),\n      spanType: SpanType.TOOL,\n      inputs: this.extractToolInputs(request),\n      parentSpanId: parentSpan?.spanId,\n      traceId: parentSpan?.traceId,\n    });\n\n    try {\n      // CRITICAL: Wrap handler in context so child operations see this span\n      const response = await SpanContext.runAsync(span, async () => {\n        return handler(request);\n      });\n\n      this.tracer.endSpan(span, {\n        outputs: this.extractToolOutputs(response),\n      });\n      return response;\n    } catch (error) {\n      this.tracer.endSpan(span, { error: error as Error });\n      throw error;\n    }\n  }\n\n  /**\n   * Extract model name from request\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain request structure is dynamic\n  private extractModelName(request: any): string {\n    return request?.model ?? request?.modelName ?? 'unknown';\n  }\n\n  /**\n   * Extract model inputs from request\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain request structure is dynamic\n  private extractModelInputs(request: any): Record<string, unknown> {\n    const messages = request?.messages ?? [];\n    return { messages: messages.slice(-3).map((m: unknown) => String(m)) };\n  }\n\n  /**\n   * Extract model outputs from response\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain response structure is dynamic\n  private extractModelOutputs(response: any): Record<string, unknown> {\n    const content = response?.content ?? response?.text ?? '';\n    return { content: String(content) };\n  }\n\n  /**\n   * Extract tool name from request\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain request structure is dynamic\n  private extractToolName(request: any): string {\n    return request?.name ?? request?.tool ?? 'unknown';\n  }\n\n  /**\n   * Extract tool inputs from request\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain request structure is dynamic\n  private extractToolInputs(request: any): Record<string, unknown> {\n    return { input: request?.input ?? request?.args ?? {} };\n  }\n\n  /**\n   * Extract tool outputs from response\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain response structure is dynamic\n  private extractToolOutputs(response: any): Record<string, unknown> {\n    return { output: response?.output ?? response };\n  }\n}\n",
+    "import type { TokenUsage } from '@prefactor/core';\n\n/**\n * Extract token usage information from LLM responses.\n *\n * Handles multiple response formats from different LLM providers and LangChain versions.\n *\n * @param response - The LLM response object\n * @returns TokenUsage object or null if no usage data found\n *\n * @example\n * ```typescript\n * const response = await model.invoke(messages);\n * const tokenUsage = extractTokenUsage(response);\n * if (tokenUsage) {\n *   console.log(`Tokens used: ${tokenUsage.totalTokens}`);\n * }\n * ```\n */\n// biome-ignore lint/suspicious/noExplicitAny: LLM response structure varies by provider\nexport function extractTokenUsage(response: any): TokenUsage | null {\n  try {\n    // Try token_usage field (common format)\n    const tokenUsage = response?.token_usage ?? response?.usage;\n    if (tokenUsage) {\n      return {\n        promptTokens: tokenUsage.prompt_tokens ?? 0,\n        completionTokens: tokenUsage.completion_tokens ?? 0,\n        totalTokens: tokenUsage.total_tokens ?? 0,\n      };\n    }\n\n    // Try usage_metadata field (LangChain format)\n    const usageMetadata = response?.usage_metadata;\n    if (usageMetadata) {\n      return {\n        promptTokens: usageMetadata.input_tokens ?? 0,\n        completionTokens: usageMetadata.output_tokens ?? 0,\n        totalTokens: usageMetadata.total_tokens ?? 0,\n      };\n    }\n\n    // Try response_metadata.token_usage (nested format)\n    const responseMetadata = response?.response_metadata;\n    if (responseMetadata?.token_usage) {\n      return {\n        promptTokens: responseMetadata.token_usage.prompt_tokens ?? 0,\n        completionTokens: responseMetadata.token_usage.completion_tokens ?? 0,\n        totalTokens: responseMetadata.token_usage.total_tokens ?? 0,\n      };\n    }\n\n    return null;\n  } catch {\n    return null;\n  }\n}\n"
+  ],
+  "mappings": ";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AASO,IANP;;;ACOO,IAVP;AAWiD,IAAjD;AACuD,IAAvD;;;ACZ8D,IAA9D;;;ACoBO,SAAS,iBAAiB,CAAC,UAAkC;AAAA,EAClE,IAAI;AAAA,IAEF,MAAM,aAAa,UAAU,eAAe,UAAU;AAAA,IACtD,IAAI,YAAY;AAAA,MACd,OAAO;AAAA,QACL,cAAc,WAAW,iBAAiB;AAAA,QAC1C,kBAAkB,WAAW,qBAAqB;AAAA,QAClD,aAAa,WAAW,gBAAgB;AAAA,MAC1C;AAAA,IACF;AAAA,IAGA,MAAM,gBAAgB,UAAU;AAAA,IAChC,IAAI,eAAe;AAAA,MACjB,OAAO;AAAA,QACL,cAAc,cAAc,gBAAgB;AAAA,QAC5C,kBAAkB,cAAc,iBAAiB;AAAA,QACjD,aAAa,cAAc,gBAAgB;AAAA,MAC7C;AAAA,IACF;AAAA,IAGA,MAAM,mBAAmB,UAAU;AAAA,IACnC,IAAI,kBAAkB,aAAa;AAAA,MACjC,OAAO;AAAA,QACL,cAAc,iBAAiB,YAAY,iBAAiB;AAAA,QAC5D,kBAAkB,iBAAiB,YAAY,qBAAqB;AAAA,QACpE,aAAa,iBAAiB,YAAY,gBAAgB;AAAA,MAC5D;AAAA,IACF;AAAA,IAEA,OAAO;AAAA,IACP,MAAM;AAAA,IACN,OAAO;AAAA;AAAA;;;AD1BJ,MAAM,oBAAoB;AAAA,EAGX;AAAA,EAFZ,WAAwB;AAAA,EAEhC,WAAW,CAAS,QAAgB;AAAA,IAAhB;AAAA;AAAA,OAQd,YAAW,CAAC,OAA2B;AAAA,IAC3C,MAAM,aAAa,wBAAY,WAAW;AAAA,IAC1C,MAAM,WAAW,OAAO,YAAY,CAAC;AAAA,IAErC,KAAK,OAAO,mBAAmB;AAAA,IAE/B,MAAM,OAAO,KAAK,OAAO,UAAU;AAAA,MACjC,MAAM;AAAA,MACN,UAAU,qBAAS;AAAA,MACnB,QAAQ,EAAE,UAAU,SAAS,MAAM,EAAE,EAAE,IAAI,CAAC,MAAe,OAAO,CAAC,CAAC,EAAE;AAAA,MACtE,cAAc,YAAY;AAAA,MAC1B,SAAS,YAAY;AAAA,IACvB,CAAC;AAAA,IAED,KAAK,WAAW;AAAA;AAAA,OASZ,WAAU,CAAC,OAA2B;AAAA,IAC1C,IAAI,CAAC,KAAK,UAAU;AAAA,MAClB;AAAA,IACF;AAAA,IAEA,MAAM,WAAW,OAAO,YAAY,CAAC;AAAA,IACrC,KAAK,OAAO,QAAQ,KAAK,UAAU;AAAA,MACjC,SAAS,EAAE,UAAU,SAAS,MAAM,EAAE,EAAE,IAAI,CAAC,MAAe,OAAO,CAAC,CAAC,EAAE;AAAA,IACzE,CAAC;AAAA,IAED,KAAK,OAAO,oBAAoB;AAAA,IAChC,wBAAY,MAAM;AAAA,IAClB,KAAK,WAAW;AAAA;AAAA,OAWZ,cAAgB,CAAC,SAAc,SAA+C;AAAA,IAClF,MAAM,aAAa,wBAAY,WAAW;AAAA,IAE1C,MAAM,OAAO,KAAK,OAAO,UAAU;AAAA,MACjC,MAAM,KAAK,iBAAiB,OAAO;AAAA,MACnC,UAAU,qBAAS;AAAA,MACnB,QAAQ,KAAK,mBAAmB,OAAO;AAAA,MACvC,cAAc,YAAY;AAAA,MAC1B,SAAS,YAAY;AAAA,IACvB,CAAC;AAAA,IAED,IAAI;AAAA,MAEF,MAAM,WAAW,MAAM,wBAAY,SAAS,MAAM,YAAY;AAAA,QAC5D,OAAO,QAAQ,OAAO;AAAA,OACvB;AAAA,MAED,MAAM,UAAU,KAAK,oBAAoB,QAAQ;AAAA,MACjD,MAAM,aAAa,kBAAkB,QAAQ;AAAA,MAE7C,KAAK,OAAO,QAAQ,MAAM,EAAE,SAAS,YAAY,cAAc,UAAU,CAAC;AAAA,MAC1E,OAAO;AAAA,MACP,OAAO,OAAO;AAAA,MACd,KAAK,OAAO,QAAQ,MAAM,EAAE,MAAsB,CAAC;AAAA,MACnD,MAAM;AAAA;AAAA;AAAA,OAYJ,aAAe,CAAC,SAAc,SAA+C;AAAA,IACjF,MAAM,aAAa,wBAAY,WAAW;AAAA,IAE1C,MAAM,OAAO,KAAK,OAAO,UAAU;AAAA,MACjC,MAAM,KAAK,gBAAgB,OAAO;AAAA,MAClC,UAAU,qBAAS;AAAA,MACnB,QAAQ,KAAK,kBAAkB,OAAO;AAAA,MACtC,cAAc,YAAY;AAAA,MAC1B,SAAS,YAAY;AAAA,IACvB,CAAC;AAAA,IAED,IAAI;AAAA,MAEF,MAAM,WAAW,MAAM,wBAAY,SAAS,MAAM,YAAY;AAAA,QAC5D,OAAO,QAAQ,OAAO;AAAA,OACvB;AAAA,MAED,KAAK,OAAO,QAAQ,MAAM;AAAA,QACxB,SAAS,KAAK,mBAAmB,QAAQ;AAAA,MAC3C,CAAC;AAAA,MACD,OAAO;AAAA,MACP,OAAO,OAAO;AAAA,MACd,KAAK,OAAO,QAAQ,MAAM,EAAE,MAAsB,CAAC;AAAA,MACnD,MAAM;AAAA;AAAA;AAAA,EAQF,gBAAgB,CAAC,SAAsB;AAAA,IAC7C,OAAO,SAAS,SAAS,SAAS,aAAa;AAAA;AAAA,EAOzC,kBAAkB,CAAC,SAAuC;AAAA,IAChE,MAAM,WAAW,SAAS,YAAY,CAAC;AAAA,IACvC,OAAO,EAAE,UAAU,SAAS,MAAM,EAAE,EAAE,IAAI,CAAC,MAAe,OAAO,CAAC,CAAC,EAAE;AAAA;AAAA,EAO/D,mBAAmB,CAAC,UAAwC;AAAA,IAClE,MAAM,UAAU,UAAU,WAAW,UAAU,QAAQ;AAAA,IACvD,OAAO,EAAE,SAAS,OAAO,OAAO,EAAE;AAAA;AAAA,EAO5B,eAAe,CAAC,SAAsB;AAAA,IAC5C,OAAO,SAAS,QAAQ,SAAS,QAAQ;AAAA;AAAA,EAOnC,iBAAiB,CAAC,SAAuC;AAAA,IAC/D,OAAO,EAAE,OAAO,SAAS,SAAS,SAAS,QAAQ,CAAC,EAAE;AAAA;AAAA,EAOhD,kBAAkB,CAAC,UAAwC;AAAA,IACjE,OAAO,EAAE,QAAQ,UAAU,UAAU,SAAS;AAAA;AAElD;;;ADtLA,IAAM,SAAS,uBAAU,MAAM;AAE/B,IAAI,eAA8B;AAClC,IAAI,mBAA2C;AAmCxC,SAAS,IAAI,CAAC,QAA2C;AAAA,EAC9D,8BAAiB;AAAA,EAEjB,MAAM,cAAc,0BAAa,MAAM;AAAA,EACvC,OAAO,KAAK,8BAA8B,EAAE,WAAW,YAAY,cAAc,CAAC;AAAA,EAElF,IAAI,qBAAqB,MAAM;AAAA,IAC7B,OAAO;AAAA,EACT;AAAA,EAEA,IAAI;AAAA,EACJ,IAAI,YAAY,kBAAkB,SAAS;AAAA,IACzC,YAAY,IAAI;AAAA,EAClB,EAAO;AAAA,IACL,IAAI,CAAC,YAAY,YAAY;AAAA,MAC3B,MAAM,IAAI,MAAM,oEAAoE;AAAA,IACtF;AAAA,IAEA,MAAM,aAAa,uCAA0B,MAAM,YAAY,UAAU;AAAA,IACzE,YAAY,IAAI,2BAAc,UAAU;AAAA;AAAA,EAI1C,IAAI;AAAA,EACJ,IAAI,YAAY,YAAY,SAAS;AAAA,IACnC,IAAI;AAAA,MACF,YAAY,6BAAiB,YAAY,WAAW,OAAO;AAAA,MAC3D,OAAO,MAAM,qCAAqC,EAAE,UAAU,CAAC;AAAA,MAC/D,OAAO,OAAO;AAAA,MACd,OAAO,KAAK,qEAAqE,EAAE,MAAM,CAAC;AAAA;AAAA,EAE9F;AAAA,EAEA,eAAe,IAAI,oBAAO,WAAW,SAAS;AAAA,EAC9C,MAAM,sBAAsB,IAAI,oBAAoB,YAAY;AAAA,EAEhE,MAAM,aAAa,kCAAiB;AAAA,IAClC,MAAM;AAAA,IAEN,eAAe,OAAO,SAAc,YAAiB;AAAA,MACnD,OAAO,oBAAoB,cAAc,SAAS,OAAO;AAAA;AAAA,IAG3D,cAAc,OAAO,SAAc,YAAiB;AAAA,MAClD,OAAO,oBAAoB,aAAa,SAAS,OAAO;AAAA;AAAA,IAG1D,aAAa,OAAO,UAAe;AAAA,MACjC,MAAM,oBAAoB,YAAY,KAAK;AAAA;AAAA,IAG7C,YAAY,OAAO,UAAe;AAAA,MAChC,MAAM,oBAAoB,WAAW,KAAK;AAAA;AAAA,EAE9C,CAAC;AAAA,EAED,mBAAmB;AAAA,EACnB,OAAO;AAAA;AAsBF,SAAS,SAAS,GAAW;AAAA,EAClC,IAAI,CAAC,cAAc;AAAA,IACjB,KAAK;AAAA,EACP;AAAA,EAEA,OAAO;AAAA;AAqBT,eAAsB,QAAQ,GAAkB;AAAA,EAC9C,IAAI,cAAc;AAAA,IAChB,OAAO,KAAK,6BAA6B;AAAA,IACzC,MAAM,aAAa,MAAM;AAAA,EAC3B;AAAA,EACA,eAAe;AAAA,EACf,mBAAmB;AAAA;AAIrB,QAAQ,GAAG,cAAc,MAAM;AAAA,EAC7B,SAAS,EAAE,MAAM,CAAC,UAAU;AAAA,IAC1B,QAAQ,MAAM,wCAAwC,KAAK;AAAA,GAC5D;AAAA,CACF;",
+  "debugId": "46B60F8DD9AA982064756E2164756E21",
+  "names": []
+}

package/dist/index.d.ts

@@ -0,0 +1,5 @@
+export { type Config, type HttpTransportConfig, type Span, SpanStatus, SpanType, } from '@prefactor/core';
+export { getTracer, init, shutdown } from './init.js';
+export { extractTokenUsage } from './metadata-extractor.js';
+export { PrefactorMiddleware } from './middleware.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAGA,OAAO,EACL,KAAK,MAAM,EACX,KAAK,mBAAmB,EACxB,KAAK,IAAI,EACT,UAAU,EACV,QAAQ,GACT,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AACtD,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAE5D,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC"}

package/dist/index.js

@@ -0,0 +1,233 @@
+// packages/langchain/src/index.ts
+import {
+  SpanStatus,
+  SpanType as SpanType2
+} from "@prefactor/core";
+
+// packages/langchain/src/init.ts
+import {
+  configureLogging,
+  createConfig,
+  getLogger,
+  HttpTransport,
+  HttpTransportConfigSchema,
+  StdioTransport,
+  Tracer
+} from "@prefactor/core";
+import { extractPartition } from "@prefactor/pfid";
+import { createMiddleware } from "langchain";
+
+// packages/langchain/src/middleware.ts
+import { SpanContext, SpanType } from "@prefactor/core";
+
+// packages/langchain/src/metadata-extractor.ts
+function extractTokenUsage(response) {
+  try {
+    const tokenUsage = response?.token_usage ?? response?.usage;
+    if (tokenUsage) {
+      return {
+        promptTokens: tokenUsage.prompt_tokens ?? 0,
+        completionTokens: tokenUsage.completion_tokens ?? 0,
+        totalTokens: tokenUsage.total_tokens ?? 0
+      };
+    }
+    const usageMetadata = response?.usage_metadata;
+    if (usageMetadata) {
+      return {
+        promptTokens: usageMetadata.input_tokens ?? 0,
+        completionTokens: usageMetadata.output_tokens ?? 0,
+        totalTokens: usageMetadata.total_tokens ?? 0
+      };
+    }
+    const responseMetadata = response?.response_metadata;
+    if (responseMetadata?.token_usage) {
+      return {
+        promptTokens: responseMetadata.token_usage.prompt_tokens ?? 0,
+        completionTokens: responseMetadata.token_usage.completion_tokens ?? 0,
+        totalTokens: responseMetadata.token_usage.total_tokens ?? 0
+      };
+    }
+    return null;
+  } catch {
+    return null;
+  }
+}
+
+// packages/langchain/src/middleware.ts
+class PrefactorMiddleware {
+  tracer;
+  rootSpan = null;
+  constructor(tracer) {
+    this.tracer = tracer;
+  }
+  async beforeAgent(state) {
+    const parentSpan = SpanContext.getCurrent();
+    const messages = state?.messages ?? [];
+    this.tracer.startAgentInstance();
+    const span = this.tracer.startSpan({
+      name: "agent",
+      spanType: SpanType.AGENT,
+      inputs: { messages: messages.slice(-3).map((m) => String(m)) },
+      parentSpanId: parentSpan?.spanId,
+      traceId: parentSpan?.traceId
+    });
+    this.rootSpan = span;
+  }
+  async afterAgent(state) {
+    if (!this.rootSpan) {
+      return;
+    }
+    const messages = state?.messages ?? [];
+    this.tracer.endSpan(this.rootSpan, {
+      outputs: { messages: messages.slice(-3).map((m) => String(m)) }
+    });
+    this.tracer.finishAgentInstance();
+    SpanContext.clear();
+    this.rootSpan = null;
+  }
+  async wrapModelCall(request, handler) {
+    const parentSpan = SpanContext.getCurrent();
+    const span = this.tracer.startSpan({
+      name: this.extractModelName(request),
+      spanType: SpanType.LLM,
+      inputs: this.extractModelInputs(request),
+      parentSpanId: parentSpan?.spanId,
+      traceId: parentSpan?.traceId
+    });
+    try {
+      const response = await SpanContext.runAsync(span, async () => {
+        return handler(request);
+      });
+      const outputs = this.extractModelOutputs(response);
+      const tokenUsage = extractTokenUsage(response);
+      this.tracer.endSpan(span, { outputs, tokenUsage: tokenUsage ?? undefined });
+      return response;
+    } catch (error) {
+      this.tracer.endSpan(span, { error });
+      throw error;
+    }
+  }
+  async wrapToolCall(request, handler) {
+    const parentSpan = SpanContext.getCurrent();
+    const span = this.tracer.startSpan({
+      name: this.extractToolName(request),
+      spanType: SpanType.TOOL,
+      inputs: this.extractToolInputs(request),
+      parentSpanId: parentSpan?.spanId,
+      traceId: parentSpan?.traceId
+    });
+    try {
+      const response = await SpanContext.runAsync(span, async () => {
+        return handler(request);
+      });
+      this.tracer.endSpan(span, {
+        outputs: this.extractToolOutputs(response)
+      });
+      return response;
+    } catch (error) {
+      this.tracer.endSpan(span, { error });
+      throw error;
+    }
+  }
+  extractModelName(request) {
+    return request?.model ?? request?.modelName ?? "unknown";
+  }
+  extractModelInputs(request) {
+    const messages = request?.messages ?? [];
+    return { messages: messages.slice(-3).map((m) => String(m)) };
+  }
+  extractModelOutputs(response) {
+    const content = response?.content ?? response?.text ?? "";
+    return { content: String(content) };
+  }
+  extractToolName(request) {
+    return request?.name ?? request?.tool ?? "unknown";
+  }
+  extractToolInputs(request) {
+    return { input: request?.input ?? request?.args ?? {} };
+  }
+  extractToolOutputs(response) {
+    return { output: response?.output ?? response };
+  }
+}
+
+// packages/langchain/src/init.ts
+var logger = getLogger("init");
+var globalTracer = null;
+var globalMiddleware = null;
+function init(config) {
+  configureLogging();
+  const finalConfig = createConfig(config);
+  logger.info("Initializing Prefactor SDK", { transport: finalConfig.transportType });
+  if (globalMiddleware !== null) {
+    return globalMiddleware;
+  }
+  let transport;
+  if (finalConfig.transportType === "stdio") {
+    transport = new StdioTransport;
+  } else {
+    if (!finalConfig.httpConfig) {
+      throw new Error("HTTP transport requires httpConfig to be provided in configuration");
+    }
+    const httpConfig = HttpTransportConfigSchema.parse(finalConfig.httpConfig);
+    transport = new HttpTransport(httpConfig);
+  }
+  let partition;
+  if (finalConfig.httpConfig?.agentId) {
+    try {
+      partition = extractPartition(finalConfig.httpConfig.agentId);
+      logger.debug("Extracted partition from agent_id", { partition });
+    } catch (error) {
+      logger.warn("Failed to extract partition from agent_id, using random partition", { error });
+    }
+  }
+  globalTracer = new Tracer(transport, partition);
+  const prefactorMiddleware = new PrefactorMiddleware(globalTracer);
+  const middleware = createMiddleware({
+    name: "prefactor",
+    wrapModelCall: async (request, handler) => {
+      return prefactorMiddleware.wrapModelCall(request, handler);
+    },
+    wrapToolCall: async (request, handler) => {
+      return prefactorMiddleware.wrapToolCall(request, handler);
+    },
+    beforeAgent: async (state) => {
+      await prefactorMiddleware.beforeAgent(state);
+    },
+    afterAgent: async (state) => {
+      await prefactorMiddleware.afterAgent(state);
+    }
+  });
+  globalMiddleware = middleware;
+  return middleware;
+}
+function getTracer() {
+  if (!globalTracer) {
+    init();
+  }
+  return globalTracer;
+}
+async function shutdown() {
+  if (globalTracer) {
+    logger.info("Shutting down Prefactor SDK");
+    await globalTracer.close();
+  }
+  globalTracer = null;
+  globalMiddleware = null;
+}
+process.on("beforeExit", () => {
+  shutdown().catch((error) => {
+    console.error("Error during Prefactor SDK shutdown:", error);
+  });
+});
+export {
+  shutdown,
+  init,
+  getTracer,
+  extractTokenUsage,
+  SpanType2 as SpanType,
+  SpanStatus,
+  PrefactorMiddleware
+};
+
+//# debugId=B136D336DD9CEBB764756E2164756E21

package/dist/index.js.map

@@ -0,0 +1,13 @@
+{
+  "version": 3,
+  "sources": ["../src/index.ts", "../src/init.ts", "../src/middleware.ts", "../src/metadata-extractor.ts"],
+  "sourcesContent": [
+    "// Main entry points\n\n// Convenience re-exports from core\nexport {\n  type Config,\n  type HttpTransportConfig,\n  type Span,\n  SpanStatus,\n  SpanType,\n} from '@prefactor/core';\nexport { getTracer, init, shutdown } from './init.js';\nexport { extractTokenUsage } from './metadata-extractor.js';\n// Middleware\nexport { PrefactorMiddleware } from './middleware.js';\n",
+    "import {\n  type Config,\n  configureLogging,\n  createConfig,\n  getLogger,\n  HttpTransport,\n  HttpTransportConfigSchema,\n  StdioTransport,\n  Tracer,\n  type Transport,\n} from '@prefactor/core';\nimport { extractPartition, type Partition } from '@prefactor/pfid';\nimport { type AgentMiddleware, createMiddleware } from 'langchain';\nimport { PrefactorMiddleware } from './middleware.js';\n\nconst logger = getLogger('init');\n\nlet globalTracer: Tracer | null = null;\nlet globalMiddleware: AgentMiddleware | null = null;\n\n/**\n * Initialize the Prefactor SDK and return middleware for LangChain.js\n *\n * This is the main entry point for the SDK. Call this function to create a middleware\n * instance that you can pass to your LangChain.js agents.\n *\n * @param config - Optional configuration object\n * @returns PrefactorMiddleware instance to use with LangChain.js agents\n *\n * @example\n * ```typescript\n * import { init } from '@prefactor/langchain';\n * import { createAgent } from 'langchain';\n *\n * // Initialize with defaults (stdio transport)\n * const middleware = init();\n *\n * // Or configure HTTP transport\n * const middleware = init({\n *   transportType: 'http',\n *   httpConfig: {\n *     apiUrl: 'https://api.prefactor.ai',\n *     apiToken: process.env.PREFACTOR_API_TOKEN!,\n *   }\n * });\n *\n * const agent = createAgent({\n *   model: 'claude-sonnet-4-5-20250929',\n *   tools: [myTool],\n *   middleware: [middleware],\n * });\n * ```\n */\nexport function init(config?: Partial<Config>): AgentMiddleware {\n  configureLogging();\n\n  const finalConfig = createConfig(config);\n  logger.info('Initializing Prefactor SDK', { transport: finalConfig.transportType });\n\n  if (globalMiddleware !== null) {\n    return globalMiddleware;\n  }\n\n  let transport: Transport;\n  if (finalConfig.transportType === 'stdio') {\n    transport = new StdioTransport();\n  } else {\n    if (!finalConfig.httpConfig) {\n      throw new Error('HTTP transport requires httpConfig to be provided in configuration');\n    }\n    // Parse httpConfig to apply defaults from schema\n    const httpConfig = HttpTransportConfigSchema.parse(finalConfig.httpConfig);\n    transport = new HttpTransport(httpConfig);\n  }\n\n  // Extract partition from agent_id if provided (for HTTP transport)\n  let partition: Partition | undefined;\n  if (finalConfig.httpConfig?.agentId) {\n    try {\n      partition = extractPartition(finalConfig.httpConfig.agentId);\n      logger.debug('Extracted partition from agent_id', { partition });\n    } catch (error) {\n      logger.warn('Failed to extract partition from agent_id, using random partition', { error });\n    }\n  }\n\n  globalTracer = new Tracer(transport, partition);\n  const prefactorMiddleware = new PrefactorMiddleware(globalTracer);\n\n  const middleware = createMiddleware({\n    name: 'prefactor',\n    // biome-ignore lint/suspicious/noExplicitAny: LangChain middleware hooks use dynamic types\n    wrapModelCall: async (request: any, handler: any) => {\n      return prefactorMiddleware.wrapModelCall(request, handler);\n    },\n    // biome-ignore lint/suspicious/noExplicitAny: LangChain middleware hooks use dynamic types\n    wrapToolCall: async (request: any, handler: any) => {\n      return prefactorMiddleware.wrapToolCall(request, handler);\n    },\n    // biome-ignore lint/suspicious/noExplicitAny: LangChain middleware hooks use dynamic types\n    beforeAgent: async (state: any) => {\n      await prefactorMiddleware.beforeAgent(state);\n    },\n    // biome-ignore lint/suspicious/noExplicitAny: LangChain middleware hooks use dynamic types\n    afterAgent: async (state: any) => {\n      await prefactorMiddleware.afterAgent(state);\n    },\n  });\n\n  globalMiddleware = middleware;\n  return middleware;\n}\n\n/**\n * Get the current tracer instance.\n *\n * If no tracer has been created yet, this will call init() with default configuration.\n *\n * @returns Tracer instance\n *\n * @example\n * ```typescript\n * import { getTracer } from '@prefactor/langchain';\n *\n * const tracer = getTracer();\n * const span = tracer.startSpan({\n *   name: 'custom-operation',\n *   spanType: SpanType.TOOL,\n *   inputs: { data: 'example' }\n * });\n * ```\n */\nexport function getTracer(): Tracer {\n  if (!globalTracer) {\n    init();\n  }\n  // Safe because init() always sets globalTracer\n  return globalTracer as Tracer;\n}\n\n/**\n * Shutdown the SDK and flush any pending spans.\n *\n * Call this before your application exits to ensure all spans are sent to the transport.\n * This is especially important for HTTP transport which has a queue of pending requests.\n *\n * @returns Promise that resolves when shutdown is complete\n *\n * @example\n * ```typescript\n * import { shutdown } from '@prefactor/langchain';\n *\n * process.on('SIGTERM', async () => {\n *   await shutdown();\n *   process.exit(0);\n * });\n * ```\n */\nexport async function shutdown(): Promise<void> {\n  if (globalTracer) {\n    logger.info('Shutting down Prefactor SDK');\n    await globalTracer.close();\n  }\n  globalTracer = null;\n  globalMiddleware = null;\n}\n\n// Automatic shutdown on process exit\nprocess.on('beforeExit', () => {\n  shutdown().catch((error) => {\n    console.error('Error during Prefactor SDK shutdown:', error);\n  });\n});\n",
+    "import { type Span, SpanContext, SpanType, type Tracer } from '@prefactor/core';\nimport { extractTokenUsage } from './metadata-extractor.js';\n\n/**\n * Prefactor middleware for LangChain.js agents.\n *\n * This middleware automatically traces LLM calls, tool executions, and agent workflows.\n * It integrates with LangChain.js middleware API to provide transparent instrumentation.\n *\n * Features:\n * - Automatic parent-child span relationships via context propagation\n * - Token usage extraction for LLM calls\n * - Error tracking and debugging\n * - Zero-overhead instrumentation (graceful failure)\n *\n * @example\n * ```typescript\n * import { init } from '@prefactor/langchain';\n * import { createReactAgent } from '@langchain/langgraph/prebuilt';\n *\n * const middleware = init();\n * const agent = createReactAgent({\n *   llm: model,\n *   tools: [myTool],\n *   middleware: [middleware],\n * });\n * ```\n */\nexport class PrefactorMiddleware {\n  private rootSpan: Span | null = null;\n\n  constructor(private tracer: Tracer) {}\n\n  /**\n   * Called before agent execution starts\n   *\n   * @param state - Agent state containing messages\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain state can be any structure\n  async beforeAgent(state: any): Promise<void> {\n    const parentSpan = SpanContext.getCurrent();\n    const messages = state?.messages ?? [];\n\n    this.tracer.startAgentInstance();\n\n    const span = this.tracer.startSpan({\n      name: 'agent',\n      spanType: SpanType.AGENT,\n      inputs: { messages: messages.slice(-3).map((m: unknown) => String(m)) },\n      parentSpanId: parentSpan?.spanId,\n      traceId: parentSpan?.traceId,\n    });\n\n    this.rootSpan = span;\n  }\n\n  /**\n   * Called after agent execution completes\n   *\n   * @param state - Agent state containing messages\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain state can be any structure\n  async afterAgent(state: any): Promise<void> {\n    if (!this.rootSpan) {\n      return;\n    }\n\n    const messages = state?.messages ?? [];\n    this.tracer.endSpan(this.rootSpan, {\n      outputs: { messages: messages.slice(-3).map((m: unknown) => String(m)) },\n    });\n\n    this.tracer.finishAgentInstance();\n    SpanContext.clear();\n    this.rootSpan = null;\n  }\n\n  /**\n   * Wrap a model call to trace LLM invocations\n   *\n   * @param request - Model invocation request\n   * @param handler - The actual model call function\n   * @returns Promise resolving to the model response\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain request/handler types are dynamic\n  async wrapModelCall<T>(request: any, handler: (req: any) => Promise<T>): Promise<T> {\n    const parentSpan = SpanContext.getCurrent();\n\n    const span = this.tracer.startSpan({\n      name: this.extractModelName(request),\n      spanType: SpanType.LLM,\n      inputs: this.extractModelInputs(request),\n      parentSpanId: parentSpan?.spanId,\n      traceId: parentSpan?.traceId,\n    });\n\n    try {\n      // CRITICAL: Wrap handler in context so child operations see this span\n      const response = await SpanContext.runAsync(span, async () => {\n        return handler(request);\n      });\n\n      const outputs = this.extractModelOutputs(response);\n      const tokenUsage = extractTokenUsage(response);\n\n      this.tracer.endSpan(span, { outputs, tokenUsage: tokenUsage ?? undefined });\n      return response;\n    } catch (error) {\n      this.tracer.endSpan(span, { error: error as Error });\n      throw error;\n    }\n  }\n\n  /**\n   * Wrap a tool call to trace tool executions\n   *\n   * @param request - Tool invocation request\n   * @param handler - The actual tool call function\n   * @returns Promise resolving to the tool response\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain request/handler types are dynamic\n  async wrapToolCall<T>(request: any, handler: (req: any) => Promise<T>): Promise<T> {\n    const parentSpan = SpanContext.getCurrent();\n\n    const span = this.tracer.startSpan({\n      name: this.extractToolName(request),\n      spanType: SpanType.TOOL,\n      inputs: this.extractToolInputs(request),\n      parentSpanId: parentSpan?.spanId,\n      traceId: parentSpan?.traceId,\n    });\n\n    try {\n      // CRITICAL: Wrap handler in context so child operations see this span\n      const response = await SpanContext.runAsync(span, async () => {\n        return handler(request);\n      });\n\n      this.tracer.endSpan(span, {\n        outputs: this.extractToolOutputs(response),\n      });\n      return response;\n    } catch (error) {\n      this.tracer.endSpan(span, { error: error as Error });\n      throw error;\n    }\n  }\n\n  /**\n   * Extract model name from request\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain request structure is dynamic\n  private extractModelName(request: any): string {\n    return request?.model ?? request?.modelName ?? 'unknown';\n  }\n\n  /**\n   * Extract model inputs from request\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain request structure is dynamic\n  private extractModelInputs(request: any): Record<string, unknown> {\n    const messages = request?.messages ?? [];\n    return { messages: messages.slice(-3).map((m: unknown) => String(m)) };\n  }\n\n  /**\n   * Extract model outputs from response\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain response structure is dynamic\n  private extractModelOutputs(response: any): Record<string, unknown> {\n    const content = response?.content ?? response?.text ?? '';\n    return { content: String(content) };\n  }\n\n  /**\n   * Extract tool name from request\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain request structure is dynamic\n  private extractToolName(request: any): string {\n    return request?.name ?? request?.tool ?? 'unknown';\n  }\n\n  /**\n   * Extract tool inputs from request\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain request structure is dynamic\n  private extractToolInputs(request: any): Record<string, unknown> {\n    return { input: request?.input ?? request?.args ?? {} };\n  }\n\n  /**\n   * Extract tool outputs from response\n   */\n  // biome-ignore lint/suspicious/noExplicitAny: LangChain response structure is dynamic\n  private extractToolOutputs(response: any): Record<string, unknown> {\n    return { output: response?.output ?? response };\n  }\n}\n",
+    "import type { TokenUsage } from '@prefactor/core';\n\n/**\n * Extract token usage information from LLM responses.\n *\n * Handles multiple response formats from different LLM providers and LangChain versions.\n *\n * @param response - The LLM response object\n * @returns TokenUsage object or null if no usage data found\n *\n * @example\n * ```typescript\n * const response = await model.invoke(messages);\n * const tokenUsage = extractTokenUsage(response);\n * if (tokenUsage) {\n *   console.log(`Tokens used: ${tokenUsage.totalTokens}`);\n * }\n * ```\n */\n// biome-ignore lint/suspicious/noExplicitAny: LLM response structure varies by provider\nexport function extractTokenUsage(response: any): TokenUsage | null {\n  try {\n    // Try token_usage field (common format)\n    const tokenUsage = response?.token_usage ?? response?.usage;\n    if (tokenUsage) {\n      return {\n        promptTokens: tokenUsage.prompt_tokens ?? 0,\n        completionTokens: tokenUsage.completion_tokens ?? 0,\n        totalTokens: tokenUsage.total_tokens ?? 0,\n      };\n    }\n\n    // Try usage_metadata field (LangChain format)\n    const usageMetadata = response?.usage_metadata;\n    if (usageMetadata) {\n      return {\n        promptTokens: usageMetadata.input_tokens ?? 0,\n        completionTokens: usageMetadata.output_tokens ?? 0,\n        totalTokens: usageMetadata.total_tokens ?? 0,\n      };\n    }\n\n    // Try response_metadata.token_usage (nested format)\n    const responseMetadata = response?.response_metadata;\n    if (responseMetadata?.token_usage) {\n      return {\n        promptTokens: responseMetadata.token_usage.prompt_tokens ?? 0,\n        completionTokens: responseMetadata.token_usage.completion_tokens ?? 0,\n        totalTokens: responseMetadata.token_usage.total_tokens ?? 0,\n      };\n    }\n\n    return null;\n  } catch {\n    return null;\n  }\n}\n"
+  ],
+  "mappings": ";AAGA;AAAA;AAAA,cAKE;AAAA;;;ACRF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAWA;AACA;;;ACZA;;;ACoBO,SAAS,iBAAiB,CAAC,UAAkC;AAAA,EAClE,IAAI;AAAA,IAEF,MAAM,aAAa,UAAU,eAAe,UAAU;AAAA,IACtD,IAAI,YAAY;AAAA,MACd,OAAO;AAAA,QACL,cAAc,WAAW,iBAAiB;AAAA,QAC1C,kBAAkB,WAAW,qBAAqB;AAAA,QAClD,aAAa,WAAW,gBAAgB;AAAA,MAC1C;AAAA,IACF;AAAA,IAGA,MAAM,gBAAgB,UAAU;AAAA,IAChC,IAAI,eAAe;AAAA,MACjB,OAAO;AAAA,QACL,cAAc,cAAc,gBAAgB;AAAA,QAC5C,kBAAkB,cAAc,iBAAiB;AAAA,QACjD,aAAa,cAAc,gBAAgB;AAAA,MAC7C;AAAA,IACF;AAAA,IAGA,MAAM,mBAAmB,UAAU;AAAA,IACnC,IAAI,kBAAkB,aAAa;AAAA,MACjC,OAAO;AAAA,QACL,cAAc,iBAAiB,YAAY,iBAAiB;AAAA,QAC5D,kBAAkB,iBAAiB,YAAY,qBAAqB;AAAA,QACpE,aAAa,iBAAiB,YAAY,gBAAgB;AAAA,MAC5D;AAAA,IACF;AAAA,IAEA,OAAO;AAAA,IACP,MAAM;AAAA,IACN,OAAO;AAAA;AAAA;;;AD1BJ,MAAM,oBAAoB;AAAA,EAGX;AAAA,EAFZ,WAAwB;AAAA,EAEhC,WAAW,CAAS,QAAgB;AAAA,IAAhB;AAAA;AAAA,OAQd,YAAW,CAAC,OAA2B;AAAA,IAC3C,MAAM,aAAa,YAAY,WAAW;AAAA,IAC1C,MAAM,WAAW,OAAO,YAAY,CAAC;AAAA,IAErC,KAAK,OAAO,mBAAmB;AAAA,IAE/B,MAAM,OAAO,KAAK,OAAO,UAAU;AAAA,MACjC,MAAM;AAAA,MACN,UAAU,SAAS;AAAA,MACnB,QAAQ,EAAE,UAAU,SAAS,MAAM,EAAE,EAAE,IAAI,CAAC,MAAe,OAAO,CAAC,CAAC,EAAE;AAAA,MACtE,cAAc,YAAY;AAAA,MAC1B,SAAS,YAAY;AAAA,IACvB,CAAC;AAAA,IAED,KAAK,WAAW;AAAA;AAAA,OASZ,WAAU,CAAC,OAA2B;AAAA,IAC1C,IAAI,CAAC,KAAK,UAAU;AAAA,MAClB;AAAA,IACF;AAAA,IAEA,MAAM,WAAW,OAAO,YAAY,CAAC;AAAA,IACrC,KAAK,OAAO,QAAQ,KAAK,UAAU;AAAA,MACjC,SAAS,EAAE,UAAU,SAAS,MAAM,EAAE,EAAE,IAAI,CAAC,MAAe,OAAO,CAAC,CAAC,EAAE;AAAA,IACzE,CAAC;AAAA,IAED,KAAK,OAAO,oBAAoB;AAAA,IAChC,YAAY,MAAM;AAAA,IAClB,KAAK,WAAW;AAAA;AAAA,OAWZ,cAAgB,CAAC,SAAc,SAA+C;AAAA,IAClF,MAAM,aAAa,YAAY,WAAW;AAAA,IAE1C,MAAM,OAAO,KAAK,OAAO,UAAU;AAAA,MACjC,MAAM,KAAK,iBAAiB,OAAO;AAAA,MACnC,UAAU,SAAS;AAAA,MACnB,QAAQ,KAAK,mBAAmB,OAAO;AAAA,MACvC,cAAc,YAAY;AAAA,MAC1B,SAAS,YAAY;AAAA,IACvB,CAAC;AAAA,IAED,IAAI;AAAA,MAEF,MAAM,WAAW,MAAM,YAAY,SAAS,MAAM,YAAY;AAAA,QAC5D,OAAO,QAAQ,OAAO;AAAA,OACvB;AAAA,MAED,MAAM,UAAU,KAAK,oBAAoB,QAAQ;AAAA,MACjD,MAAM,aAAa,kBAAkB,QAAQ;AAAA,MAE7C,KAAK,OAAO,QAAQ,MAAM,EAAE,SAAS,YAAY,cAAc,UAAU,CAAC;AAAA,MAC1E,OAAO;AAAA,MACP,OAAO,OAAO;AAAA,MACd,KAAK,OAAO,QAAQ,MAAM,EAAE,MAAsB,CAAC;AAAA,MACnD,MAAM;AAAA;AAAA;AAAA,OAYJ,aAAe,CAAC,SAAc,SAA+C;AAAA,IACjF,MAAM,aAAa,YAAY,WAAW;AAAA,IAE1C,MAAM,OAAO,KAAK,OAAO,UAAU;AAAA,MACjC,MAAM,KAAK,gBAAgB,OAAO;AAAA,MAClC,UAAU,SAAS;AAAA,MACnB,QAAQ,KAAK,kBAAkB,OAAO;AAAA,MACtC,cAAc,YAAY;AAAA,MAC1B,SAAS,YAAY;AAAA,IACvB,CAAC;AAAA,IAED,IAAI;AAAA,MAEF,MAAM,WAAW,MAAM,YAAY,SAAS,MAAM,YAAY;AAAA,QAC5D,OAAO,QAAQ,OAAO;AAAA,OACvB;AAAA,MAED,KAAK,OAAO,QAAQ,MAAM;AAAA,QACxB,SAAS,KAAK,mBAAmB,QAAQ;AAAA,MAC3C,CAAC;AAAA,MACD,OAAO;AAAA,MACP,OAAO,OAAO;AAAA,MACd,KAAK,OAAO,QAAQ,MAAM,EAAE,MAAsB,CAAC;AAAA,MACnD,MAAM;AAAA;AAAA;AAAA,EAQF,gBAAgB,CAAC,SAAsB;AAAA,IAC7C,OAAO,SAAS,SAAS,SAAS,aAAa;AAAA;AAAA,EAOzC,kBAAkB,CAAC,SAAuC;AAAA,IAChE,MAAM,WAAW,SAAS,YAAY,CAAC;AAAA,IACvC,OAAO,EAAE,UAAU,SAAS,MAAM,EAAE,EAAE,IAAI,CAAC,MAAe,OAAO,CAAC,CAAC,EAAE;AAAA;AAAA,EAO/D,mBAAmB,CAAC,UAAwC;AAAA,IAClE,MAAM,UAAU,UAAU,WAAW,UAAU,QAAQ;AAAA,IACvD,OAAO,EAAE,SAAS,OAAO,OAAO,EAAE;AAAA;AAAA,EAO5B,eAAe,CAAC,SAAsB;AAAA,IAC5C,OAAO,SAAS,QAAQ,SAAS,QAAQ;AAAA;AAAA,EAOnC,iBAAiB,CAAC,SAAuC;AAAA,IAC/D,OAAO,EAAE,OAAO,SAAS,SAAS,SAAS,QAAQ,CAAC,EAAE;AAAA;AAAA,EAOhD,kBAAkB,CAAC,UAAwC;AAAA,IACjE,OAAO,EAAE,QAAQ,UAAU,UAAU,SAAS;AAAA;AAElD;;;ADtLA,IAAM,SAAS,UAAU,MAAM;AAE/B,IAAI,eAA8B;AAClC,IAAI,mBAA2C;AAmCxC,SAAS,IAAI,CAAC,QAA2C;AAAA,EAC9D,iBAAiB;AAAA,EAEjB,MAAM,cAAc,aAAa,MAAM;AAAA,EACvC,OAAO,KAAK,8BAA8B,EAAE,WAAW,YAAY,cAAc,CAAC;AAAA,EAElF,IAAI,qBAAqB,MAAM;AAAA,IAC7B,OAAO;AAAA,EACT;AAAA,EAEA,IAAI;AAAA,EACJ,IAAI,YAAY,kBAAkB,SAAS;AAAA,IACzC,YAAY,IAAI;AAAA,EAClB,EAAO;AAAA,IACL,IAAI,CAAC,YAAY,YAAY;AAAA,MAC3B,MAAM,IAAI,MAAM,oEAAoE;AAAA,IACtF;AAAA,IAEA,MAAM,aAAa,0BAA0B,MAAM,YAAY,UAAU;AAAA,IACzE,YAAY,IAAI,cAAc,UAAU;AAAA;AAAA,EAI1C,IAAI;AAAA,EACJ,IAAI,YAAY,YAAY,SAAS;AAAA,IACnC,IAAI;AAAA,MACF,YAAY,iBAAiB,YAAY,WAAW,OAAO;AAAA,MAC3D,OAAO,MAAM,qCAAqC,EAAE,UAAU,CAAC;AAAA,MAC/D,OAAO,OAAO;AAAA,MACd,OAAO,KAAK,qEAAqE,EAAE,MAAM,CAAC;AAAA;AAAA,EAE9F;AAAA,EAEA,eAAe,IAAI,OAAO,WAAW,SAAS;AAAA,EAC9C,MAAM,sBAAsB,IAAI,oBAAoB,YAAY;AAAA,EAEhE,MAAM,aAAa,iBAAiB;AAAA,IAClC,MAAM;AAAA,IAEN,eAAe,OAAO,SAAc,YAAiB;AAAA,MACnD,OAAO,oBAAoB,cAAc,SAAS,OAAO;AAAA;AAAA,IAG3D,cAAc,OAAO,SAAc,YAAiB;AAAA,MAClD,OAAO,oBAAoB,aAAa,SAAS,OAAO;AAAA;AAAA,IAG1D,aAAa,OAAO,UAAe;AAAA,MACjC,MAAM,oBAAoB,YAAY,KAAK;AAAA;AAAA,IAG7C,YAAY,OAAO,UAAe;AAAA,MAChC,MAAM,oBAAoB,WAAW,KAAK;AAAA;AAAA,EAE9C,CAAC;AAAA,EAED,mBAAmB;AAAA,EACnB,OAAO;AAAA;AAsBF,SAAS,SAAS,GAAW;AAAA,EAClC,IAAI,CAAC,cAAc;AAAA,IACjB,KAAK;AAAA,EACP;AAAA,EAEA,OAAO;AAAA;AAqBT,eAAsB,QAAQ,GAAkB;AAAA,EAC9C,IAAI,cAAc;AAAA,IAChB,OAAO,KAAK,6BAA6B;AAAA,IACzC,MAAM,aAAa,MAAM;AAAA,EAC3B;AAAA,EACA,eAAe;AAAA,EACf,mBAAmB;AAAA;AAIrB,QAAQ,GAAG,cAAc,MAAM;AAAA,EAC7B,SAAS,EAAE,MAAM,CAAC,UAAU;AAAA,IAC1B,QAAQ,MAAM,wCAAwC,KAAK;AAAA,GAC5D;AAAA,CACF;",
+  "debugId": "B136D336DD9CEBB764756E2164756E21",
+  "names": []
+}

package/dist/init.d.ts

@@ -0,0 +1,76 @@
+import { type Config, Tracer } from '@prefactor/core';
+import { type AgentMiddleware } from 'langchain';
+/**
+ * Initialize the Prefactor SDK and return middleware for LangChain.js
+ *
+ * This is the main entry point for the SDK. Call this function to create a middleware
+ * instance that you can pass to your LangChain.js agents.
+ *
+ * @param config - Optional configuration object
+ * @returns PrefactorMiddleware instance to use with LangChain.js agents
+ *
+ * @example
+ * ```typescript
+ * import { init } from '@prefactor/langchain';
+ * import { createAgent } from 'langchain';
+ *
+ * // Initialize with defaults (stdio transport)
+ * const middleware = init();
+ *
+ * // Or configure HTTP transport
+ * const middleware = init({
+ *   transportType: 'http',
+ *   httpConfig: {
+ *     apiUrl: 'https://api.prefactor.ai',
+ *     apiToken: process.env.PREFACTOR_API_TOKEN!,
+ *   }
+ * });
+ *
+ * const agent = createAgent({
+ *   model: 'claude-sonnet-4-5-20250929',
+ *   tools: [myTool],
+ *   middleware: [middleware],
+ * });
+ * ```
+ */
+export declare function init(config?: Partial<Config>): AgentMiddleware;
+/**
+ * Get the current tracer instance.
+ *
+ * If no tracer has been created yet, this will call init() with default configuration.
+ *
+ * @returns Tracer instance
+ *
+ * @example
+ * ```typescript
+ * import { getTracer } from '@prefactor/langchain';
+ *
+ * const tracer = getTracer();
+ * const span = tracer.startSpan({
+ *   name: 'custom-operation',
+ *   spanType: SpanType.TOOL,
+ *   inputs: { data: 'example' }
+ * });
+ * ```
+ */
+export declare function getTracer(): Tracer;
+/**
+ * Shutdown the SDK and flush any pending spans.
+ *
+ * Call this before your application exits to ensure all spans are sent to the transport.
+ * This is especially important for HTTP transport which has a queue of pending requests.
+ *
+ * @returns Promise that resolves when shutdown is complete
+ *
+ * @example
+ * ```typescript
+ * import { shutdown } from '@prefactor/langchain';
+ *
+ * process.on('SIGTERM', async () => {
+ *   await shutdown();
+ *   process.exit(0);
+ * });
+ * ```
+ */
+export declare function shutdown(): Promise<void>;
+//# sourceMappingURL=init.d.ts.map

package/dist/init.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"init.d.ts","sourceRoot":"","sources":["../src/init.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,MAAM,EAOX,MAAM,EAEP,MAAM,iBAAiB,CAAC;AAEzB,OAAO,EAAE,KAAK,eAAe,EAAoB,MAAM,WAAW,CAAC;AAQnE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCG;AACH,wBAAgB,IAAI,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,GAAG,eAAe,CA0D9D;AAED;;;;;;;;;;;;;;;;;;GAkBG;AACH,wBAAgB,SAAS,IAAI,MAAM,CAMlC;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAsB,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC,CAO9C"}

package/dist/init.js

@@ -0,0 +1,153 @@
+import { configureLogging, createConfig, getLogger, HttpTransport, HttpTransportConfigSchema, StdioTransport, Tracer, } from '@prefactor/core';
+import { extractPartition } from '@prefactor/pfid';
+import { createMiddleware } from 'langchain';
+import { PrefactorMiddleware } from './middleware.js';
+const logger = getLogger('init');
+let globalTracer = null;
+let globalMiddleware = null;
+/**
+ * Initialize the Prefactor SDK and return middleware for LangChain.js
+ *
+ * This is the main entry point for the SDK. Call this function to create a middleware
+ * instance that you can pass to your LangChain.js agents.
+ *
+ * @param config - Optional configuration object
+ * @returns PrefactorMiddleware instance to use with LangChain.js agents
+ *
+ * @example
+ * ```typescript
+ * import { init } from '@prefactor/langchain';
+ * import { createAgent } from 'langchain';
+ *
+ * // Initialize with defaults (stdio transport)
+ * const middleware = init();
+ *
+ * // Or configure HTTP transport
+ * const middleware = init({
+ *   transportType: 'http',
+ *   httpConfig: {
+ *     apiUrl: 'https://api.prefactor.ai',
+ *     apiToken: process.env.PREFACTOR_API_TOKEN!,
+ *   }
+ * });
+ *
+ * const agent = createAgent({
+ *   model: 'claude-sonnet-4-5-20250929',
+ *   tools: [myTool],
+ *   middleware: [middleware],
+ * });
+ * ```
+ */
+export function init(config) {
+    configureLogging();
+    const finalConfig = createConfig(config);
+    logger.info('Initializing Prefactor SDK', { transport: finalConfig.transportType });
+    if (globalMiddleware !== null) {
+        return globalMiddleware;
+    }
+    let transport;
+    if (finalConfig.transportType === 'stdio') {
+        transport = new StdioTransport();
+    }
+    else {
+        if (!finalConfig.httpConfig) {
+            throw new Error('HTTP transport requires httpConfig to be provided in configuration');
+        }
+        // Parse httpConfig to apply defaults from schema
+        const httpConfig = HttpTransportConfigSchema.parse(finalConfig.httpConfig);
+        transport = new HttpTransport(httpConfig);
+    }
+    // Extract partition from agent_id if provided (for HTTP transport)
+    let partition;
+    if (finalConfig.httpConfig?.agentId) {
+        try {
+            partition = extractPartition(finalConfig.httpConfig.agentId);
+            logger.debug('Extracted partition from agent_id', { partition });
+        }
+        catch (error) {
+            logger.warn('Failed to extract partition from agent_id, using random partition', { error });
+        }
+    }
+    globalTracer = new Tracer(transport, partition);
+    const prefactorMiddleware = new PrefactorMiddleware(globalTracer);
+    const middleware = createMiddleware({
+        name: 'prefactor',
+        // biome-ignore lint/suspicious/noExplicitAny: LangChain middleware hooks use dynamic types
+        wrapModelCall: async (request, handler) => {
+            return prefactorMiddleware.wrapModelCall(request, handler);
+        },
+        // biome-ignore lint/suspicious/noExplicitAny: LangChain middleware hooks use dynamic types
+        wrapToolCall: async (request, handler) => {
+            return prefactorMiddleware.wrapToolCall(request, handler);
+        },
+        // biome-ignore lint/suspicious/noExplicitAny: LangChain middleware hooks use dynamic types
+        beforeAgent: async (state) => {
+            await prefactorMiddleware.beforeAgent(state);
+        },
+        // biome-ignore lint/suspicious/noExplicitAny: LangChain middleware hooks use dynamic types
+        afterAgent: async (state) => {
+            await prefactorMiddleware.afterAgent(state);
+        },
+    });
+    globalMiddleware = middleware;
+    return middleware;
+}
+/**
+ * Get the current tracer instance.
+ *
+ * If no tracer has been created yet, this will call init() with default configuration.
+ *
+ * @returns Tracer instance
+ *
+ * @example
+ * ```typescript
+ * import { getTracer } from '@prefactor/langchain';
+ *
+ * const tracer = getTracer();
+ * const span = tracer.startSpan({
+ *   name: 'custom-operation',
+ *   spanType: SpanType.TOOL,
+ *   inputs: { data: 'example' }
+ * });
+ * ```
+ */
+export function getTracer() {
+    if (!globalTracer) {
+        init();
+    }
+    // Safe because init() always sets globalTracer
+    return globalTracer;
+}
+/**
+ * Shutdown the SDK and flush any pending spans.
+ *
+ * Call this before your application exits to ensure all spans are sent to the transport.
+ * This is especially important for HTTP transport which has a queue of pending requests.
+ *
+ * @returns Promise that resolves when shutdown is complete
+ *
+ * @example
+ * ```typescript
+ * import { shutdown } from '@prefactor/langchain';
+ *
+ * process.on('SIGTERM', async () => {
+ *   await shutdown();
+ *   process.exit(0);
+ * });
+ * ```
+ */
+export async function shutdown() {
+    if (globalTracer) {
+        logger.info('Shutting down Prefactor SDK');
+        await globalTracer.close();
+    }
+    globalTracer = null;
+    globalMiddleware = null;
+}
+// Automatic shutdown on process exit
+process.on('beforeExit', () => {
+    shutdown().catch((error) => {
+        console.error('Error during Prefactor SDK shutdown:', error);
+    });
+});
+//# sourceMappingURL=init.js.map

package/dist/init.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"init.js","sourceRoot":"","sources":["../src/init.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,gBAAgB,EAChB,YAAY,EACZ,SAAS,EACT,aAAa,EACb,yBAAyB,EACzB,cAAc,EACd,MAAM,GAEP,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,gBAAgB,EAAkB,MAAM,iBAAiB,CAAC;AACnE,OAAO,EAAwB,gBAAgB,EAAE,MAAM,WAAW,CAAC;AACnE,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AAEtD,MAAM,MAAM,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;AAEjC,IAAI,YAAY,GAAkB,IAAI,CAAC;AACvC,IAAI,gBAAgB,GAA2B,IAAI,CAAC;AAEpD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCG;AACH,MAAM,UAAU,IAAI,CAAC,MAAwB;IAC3C,gBAAgB,EAAE,CAAC;IAEnB,MAAM,WAAW,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;IACzC,MAAM,CAAC,IAAI,CAAC,4BAA4B,EAAE,EAAE,SAAS,EAAE,WAAW,CAAC,aAAa,EAAE,CAAC,CAAC;IAEpF,IAAI,gBAAgB,KAAK,IAAI,EAAE,CAAC;QAC9B,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAED,IAAI,SAAoB,CAAC;IACzB,IAAI,WAAW,CAAC,aAAa,KAAK,OAAO,EAAE,CAAC;QAC1C,SAAS,GAAG,IAAI,cAAc,EAAE,CAAC;IACnC,CAAC;SAAM,CAAC;QACN,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,oEAAoE,CAAC,CAAC;QACxF,CAAC;QACD,iDAAiD;QACjD,MAAM,UAAU,GAAG,yBAAyB,CAAC,KAAK,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;QAC3E,SAAS,GAAG,IAAI,aAAa,CAAC,UAAU,CAAC,CAAC;IAC5C,CAAC;IAED,mEAAmE;IACnE,IAAI,SAAgC,CAAC;IACrC,IAAI,WAAW,CAAC,UAAU,EAAE,OAAO,EAAE,CAAC;QACpC,IAAI,CAAC;YACH,SAAS,GAAG,gBAAgB,CAAC,WAAW,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;YAC7D,MAAM,CAAC,KAAK,CAAC,mCAAmC,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC;QACnE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,IAAI,CAAC,mEAAmE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC9F,CAAC;IACH,CAAC;IAED,YAAY,GAAG,IAAI,MAAM,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;IAChD,MAAM,mBAAmB,GAAG,IAAI,mBAAmB,CAAC,YAAY,CAAC,CAAC;IAElE,MAAM,UAAU,GAAG,gBAAgB,CAAC;QAClC,IAAI,EAAE,WAAW;QACjB,2FAA2F;QAC3F,aAAa,EAAE,KAAK,EAAE,OAAY,EAAE,OAAY,EAAE,EAAE;YAClD,OAAO,mBAAmB,CAAC,aAAa,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC7D,CAAC;QACD,2FAA2F;QAC3F,YAAY,EAAE,KAAK,EAAE,OAAY,EAAE,OAAY,EAAE,EAAE;YACjD,OAAO,mBAAmB,CAAC,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC5D,CAAC;QACD,2FAA2F;QAC3F,WAAW,EAAE,KAAK,EAAE,KAAU,EAAE,EAAE;YAChC,MAAM,mBAAmB,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAC/C,CAAC;QACD,2FAA2F;QAC3F,UAAU,EAAE,KAAK,EAAE,KAAU,EAAE,EAAE;YAC/B,MAAM,mBAAmB,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAC9C,CAAC;KACF,CAAC,CAAC;IAEH,gBAAgB,GAAG,UAAU,CAAC;IAC9B,OAAO,UAAU,CAAC;AACpB,CAAC;AAED;;;;;;;;;;;;;;;;;;GAkBG;AACH,MAAM,UAAU,SAAS;IACvB,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,IAAI,EAAE,CAAC;IACT,CAAC;IACD,+CAA+C;IAC/C,OAAO,YAAsB,CAAC;AAChC,CAAC;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,CAAC,KAAK,UAAU,QAAQ;IAC5B,IAAI,YAAY,EAAE,CAAC;QACjB,MAAM,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;QAC3C,MAAM,YAAY,CAAC,KAAK,EAAE,CAAC;IAC7B,CAAC;IACD,YAAY,GAAG,IAAI,CAAC;IACpB,gBAAgB,GAAG,IAAI,CAAC;AAC1B,CAAC;AAED,qCAAqC;AACrC,OAAO,CAAC,EAAE,CAAC,YAAY,EAAE,GAAG,EAAE;IAC5B,QAAQ,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;QACzB,OAAO,CAAC,KAAK,CAAC,sCAAsC,EAAE,KAAK,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/metadata-extractor.d.ts

@@ -0,0 +1,20 @@
+import type { TokenUsage } from '@prefactor/core';
+/**
+ * Extract token usage information from LLM responses.
+ *
+ * Handles multiple response formats from different LLM providers and LangChain versions.
+ *
+ * @param response - The LLM response object
+ * @returns TokenUsage object or null if no usage data found
+ *
+ * @example
+ * ```typescript
+ * const response = await model.invoke(messages);
+ * const tokenUsage = extractTokenUsage(response);
+ * if (tokenUsage) {
+ *   console.log(`Tokens used: ${tokenUsage.totalTokens}`);
+ * }
+ * ```
+ */
+export declare function extractTokenUsage(response: any): TokenUsage | null;
+//# sourceMappingURL=metadata-extractor.d.ts.map

package/dist/metadata-extractor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"metadata-extractor.d.ts","sourceRoot":"","sources":["../src/metadata-extractor.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAElD;;;;;;;;;;;;;;;;GAgBG;AAEH,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,GAAG,GAAG,UAAU,GAAG,IAAI,CAoClE"}

package/dist/metadata-extractor.js

@@ -0,0 +1,54 @@
+/**
+ * Extract token usage information from LLM responses.
+ *
+ * Handles multiple response formats from different LLM providers and LangChain versions.
+ *
+ * @param response - The LLM response object
+ * @returns TokenUsage object or null if no usage data found
+ *
+ * @example
+ * ```typescript
+ * const response = await model.invoke(messages);
+ * const tokenUsage = extractTokenUsage(response);
+ * if (tokenUsage) {
+ *   console.log(`Tokens used: ${tokenUsage.totalTokens}`);
+ * }
+ * ```
+ */
+// biome-ignore lint/suspicious/noExplicitAny: LLM response structure varies by provider
+export function extractTokenUsage(response) {
+    try {
+        // Try token_usage field (common format)
+        const tokenUsage = response?.token_usage ?? response?.usage;
+        if (tokenUsage) {
+            return {
+                promptTokens: tokenUsage.prompt_tokens ?? 0,
+                completionTokens: tokenUsage.completion_tokens ?? 0,
+                totalTokens: tokenUsage.total_tokens ?? 0,
+            };
+        }
+        // Try usage_metadata field (LangChain format)
+        const usageMetadata = response?.usage_metadata;
+        if (usageMetadata) {
+            return {
+                promptTokens: usageMetadata.input_tokens ?? 0,
+                completionTokens: usageMetadata.output_tokens ?? 0,
+                totalTokens: usageMetadata.total_tokens ?? 0,
+            };
+        }
+        // Try response_metadata.token_usage (nested format)
+        const responseMetadata = response?.response_metadata;
+        if (responseMetadata?.token_usage) {
+            return {
+                promptTokens: responseMetadata.token_usage.prompt_tokens ?? 0,
+                completionTokens: responseMetadata.token_usage.completion_tokens ?? 0,
+                totalTokens: responseMetadata.token_usage.total_tokens ?? 0,
+            };
+        }
+        return null;
+    }
+    catch {
+        return null;
+    }
+}
+//# sourceMappingURL=metadata-extractor.js.map

package/dist/metadata-extractor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"metadata-extractor.js","sourceRoot":"","sources":["../src/metadata-extractor.ts"],"names":[],"mappings":"AAEA;;;;;;;;;;;;;;;;GAgBG;AACH,wFAAwF;AACxF,MAAM,UAAU,iBAAiB,CAAC,QAAa;IAC7C,IAAI,CAAC;QACH,wCAAwC;QACxC,MAAM,UAAU,GAAG,QAAQ,EAAE,WAAW,IAAI,QAAQ,EAAE,KAAK,CAAC;QAC5D,IAAI,UAAU,EAAE,CAAC;YACf,OAAO;gBACL,YAAY,EAAE,UAAU,CAAC,aAAa,IAAI,CAAC;gBAC3C,gBAAgB,EAAE,UAAU,CAAC,iBAAiB,IAAI,CAAC;gBACnD,WAAW,EAAE,UAAU,CAAC,YAAY,IAAI,CAAC;aAC1C,CAAC;QACJ,CAAC;QAED,8CAA8C;QAC9C,MAAM,aAAa,GAAG,QAAQ,EAAE,cAAc,CAAC;QAC/C,IAAI,aAAa,EAAE,CAAC;YAClB,OAAO;gBACL,YAAY,EAAE,aAAa,CAAC,YAAY,IAAI,CAAC;gBAC7C,gBAAgB,EAAE,aAAa,CAAC,aAAa,IAAI,CAAC;gBAClD,WAAW,EAAE,aAAa,CAAC,YAAY,IAAI,CAAC;aAC7C,CAAC;QACJ,CAAC;QAED,oDAAoD;QACpD,MAAM,gBAAgB,GAAG,QAAQ,EAAE,iBAAiB,CAAC;QACrD,IAAI,gBAAgB,EAAE,WAAW,EAAE,CAAC;YAClC,OAAO;gBACL,YAAY,EAAE,gBAAgB,CAAC,WAAW,CAAC,aAAa,IAAI,CAAC;gBAC7D,gBAAgB,EAAE,gBAAgB,CAAC,WAAW,CAAC,iBAAiB,IAAI,CAAC;gBACrE,WAAW,EAAE,gBAAgB,CAAC,WAAW,CAAC,YAAY,IAAI,CAAC;aAC5D,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC"}

package/dist/middleware.d.ts

@@ -0,0 +1,84 @@
+import { type Tracer } from '@prefactor/core';
+/**
+ * Prefactor middleware for LangChain.js agents.
+ *
+ * This middleware automatically traces LLM calls, tool executions, and agent workflows.
+ * It integrates with LangChain.js middleware API to provide transparent instrumentation.
+ *
+ * Features:
+ * - Automatic parent-child span relationships via context propagation
+ * - Token usage extraction for LLM calls
+ * - Error tracking and debugging
+ * - Zero-overhead instrumentation (graceful failure)
+ *
+ * @example
+ * ```typescript
+ * import { init } from '@prefactor/langchain';
+ * import { createReactAgent } from '@langchain/langgraph/prebuilt';
+ *
+ * const middleware = init();
+ * const agent = createReactAgent({
+ *   llm: model,
+ *   tools: [myTool],
+ *   middleware: [middleware],
+ * });
+ * ```
+ */
+export declare class PrefactorMiddleware {
+    private tracer;
+    private rootSpan;
+    constructor(tracer: Tracer);
+    /**
+     * Called before agent execution starts
+     *
+     * @param state - Agent state containing messages
+     */
+    beforeAgent(state: any): Promise<void>;
+    /**
+     * Called after agent execution completes
+     *
+     * @param state - Agent state containing messages
+     */
+    afterAgent(state: any): Promise<void>;
+    /**
+     * Wrap a model call to trace LLM invocations
+     *
+     * @param request - Model invocation request
+     * @param handler - The actual model call function
+     * @returns Promise resolving to the model response
+     */
+    wrapModelCall<T>(request: any, handler: (req: any) => Promise<T>): Promise<T>;
+    /**
+     * Wrap a tool call to trace tool executions
+     *
+     * @param request - Tool invocation request
+     * @param handler - The actual tool call function
+     * @returns Promise resolving to the tool response
+     */
+    wrapToolCall<T>(request: any, handler: (req: any) => Promise<T>): Promise<T>;
+    /**
+     * Extract model name from request
+     */
+    private extractModelName;
+    /**
+     * Extract model inputs from request
+     */
+    private extractModelInputs;
+    /**
+     * Extract model outputs from response
+     */
+    private extractModelOutputs;
+    /**
+     * Extract tool name from request
+     */
+    private extractToolName;
+    /**
+     * Extract tool inputs from request
+     */
+    private extractToolInputs;
+    /**
+     * Extract tool outputs from response
+     */
+    private extractToolOutputs;
+}
+//# sourceMappingURL=middleware.d.ts.map

package/dist/middleware.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.d.ts","sourceRoot":"","sources":["../src/middleware.ts"],"names":[],"mappings":"AAAA,OAAO,EAAoC,KAAK,MAAM,EAAE,MAAM,iBAAiB,CAAC;AAGhF;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,qBAAa,mBAAmB;IAGlB,OAAO,CAAC,MAAM;IAF1B,OAAO,CAAC,QAAQ,CAAqB;gBAEjB,MAAM,EAAE,MAAM;IAElC;;;;OAIG;IAEG,WAAW,CAAC,KAAK,EAAE,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC;IAiB5C;;;;OAIG;IAEG,UAAU,CAAC,KAAK,EAAE,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC;IAe3C;;;;;;OAMG;IAEG,aAAa,CAAC,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,GAAG,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC;IA4BnF;;;;;;OAMG;IAEG,YAAY,CAAC,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,GAAG,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC;IA2BlF;;OAEG;IAEH,OAAO,CAAC,gBAAgB;IAIxB;;OAEG;IAEH,OAAO,CAAC,kBAAkB;IAK1B;;OAEG;IAEH,OAAO,CAAC,mBAAmB;IAK3B;;OAEG;IAEH,OAAO,CAAC,eAAe;IAIvB;;OAEG;IAEH,OAAO,CAAC,iBAAiB;IAIzB;;OAEG;IAEH,OAAO,CAAC,kBAAkB;CAG3B"}

package/dist/middleware.js

@@ -0,0 +1,180 @@
+import { SpanContext, SpanType } from '@prefactor/core';
+import { extractTokenUsage } from './metadata-extractor.js';
+/**
+ * Prefactor middleware for LangChain.js agents.
+ *
+ * This middleware automatically traces LLM calls, tool executions, and agent workflows.
+ * It integrates with LangChain.js middleware API to provide transparent instrumentation.
+ *
+ * Features:
+ * - Automatic parent-child span relationships via context propagation
+ * - Token usage extraction for LLM calls
+ * - Error tracking and debugging
+ * - Zero-overhead instrumentation (graceful failure)
+ *
+ * @example
+ * ```typescript
+ * import { init } from '@prefactor/langchain';
+ * import { createReactAgent } from '@langchain/langgraph/prebuilt';
+ *
+ * const middleware = init();
+ * const agent = createReactAgent({
+ *   llm: model,
+ *   tools: [myTool],
+ *   middleware: [middleware],
+ * });
+ * ```
+ */
+export class PrefactorMiddleware {
+    tracer;
+    rootSpan = null;
+    constructor(tracer) {
+        this.tracer = tracer;
+    }
+    /**
+     * Called before agent execution starts
+     *
+     * @param state - Agent state containing messages
+     */
+    // biome-ignore lint/suspicious/noExplicitAny: LangChain state can be any structure
+    async beforeAgent(state) {
+        const parentSpan = SpanContext.getCurrent();
+        const messages = state?.messages ?? [];
+        this.tracer.startAgentInstance();
+        const span = this.tracer.startSpan({
+            name: 'agent',
+            spanType: SpanType.AGENT,
+            inputs: { messages: messages.slice(-3).map((m) => String(m)) },
+            parentSpanId: parentSpan?.spanId,
+            traceId: parentSpan?.traceId,
+        });
+        this.rootSpan = span;
+    }
+    /**
+     * Called after agent execution completes
+     *
+     * @param state - Agent state containing messages
+     */
+    // biome-ignore lint/suspicious/noExplicitAny: LangChain state can be any structure
+    async afterAgent(state) {
+        if (!this.rootSpan) {
+            return;
+        }
+        const messages = state?.messages ?? [];
+        this.tracer.endSpan(this.rootSpan, {
+            outputs: { messages: messages.slice(-3).map((m) => String(m)) },
+        });
+        this.tracer.finishAgentInstance();
+        SpanContext.clear();
+        this.rootSpan = null;
+    }
+    /**
+     * Wrap a model call to trace LLM invocations
+     *
+     * @param request - Model invocation request
+     * @param handler - The actual model call function
+     * @returns Promise resolving to the model response
+     */
+    // biome-ignore lint/suspicious/noExplicitAny: LangChain request/handler types are dynamic
+    async wrapModelCall(request, handler) {
+        const parentSpan = SpanContext.getCurrent();
+        const span = this.tracer.startSpan({
+            name: this.extractModelName(request),
+            spanType: SpanType.LLM,
+            inputs: this.extractModelInputs(request),
+            parentSpanId: parentSpan?.spanId,
+            traceId: parentSpan?.traceId,
+        });
+        try {
+            // CRITICAL: Wrap handler in context so child operations see this span
+            const response = await SpanContext.runAsync(span, async () => {
+                return handler(request);
+            });
+            const outputs = this.extractModelOutputs(response);
+            const tokenUsage = extractTokenUsage(response);
+            this.tracer.endSpan(span, { outputs, tokenUsage: tokenUsage ?? undefined });
+            return response;
+        }
+        catch (error) {
+            this.tracer.endSpan(span, { error: error });
+            throw error;
+        }
+    }
+    /**
+     * Wrap a tool call to trace tool executions
+     *
+     * @param request - Tool invocation request
+     * @param handler - The actual tool call function
+     * @returns Promise resolving to the tool response
+     */
+    // biome-ignore lint/suspicious/noExplicitAny: LangChain request/handler types are dynamic
+    async wrapToolCall(request, handler) {
+        const parentSpan = SpanContext.getCurrent();
+        const span = this.tracer.startSpan({
+            name: this.extractToolName(request),
+            spanType: SpanType.TOOL,
+            inputs: this.extractToolInputs(request),
+            parentSpanId: parentSpan?.spanId,
+            traceId: parentSpan?.traceId,
+        });
+        try {
+            // CRITICAL: Wrap handler in context so child operations see this span
+            const response = await SpanContext.runAsync(span, async () => {
+                return handler(request);
+            });
+            this.tracer.endSpan(span, {
+                outputs: this.extractToolOutputs(response),
+            });
+            return response;
+        }
+        catch (error) {
+            this.tracer.endSpan(span, { error: error });
+            throw error;
+        }
+    }
+    /**
+     * Extract model name from request
+     */
+    // biome-ignore lint/suspicious/noExplicitAny: LangChain request structure is dynamic
+    extractModelName(request) {
+        return request?.model ?? request?.modelName ?? 'unknown';
+    }
+    /**
+     * Extract model inputs from request
+     */
+    // biome-ignore lint/suspicious/noExplicitAny: LangChain request structure is dynamic
+    extractModelInputs(request) {
+        const messages = request?.messages ?? [];
+        return { messages: messages.slice(-3).map((m) => String(m)) };
+    }
+    /**
+     * Extract model outputs from response
+     */
+    // biome-ignore lint/suspicious/noExplicitAny: LangChain response structure is dynamic
+    extractModelOutputs(response) {
+        const content = response?.content ?? response?.text ?? '';
+        return { content: String(content) };
+    }
+    /**
+     * Extract tool name from request
+     */
+    // biome-ignore lint/suspicious/noExplicitAny: LangChain request structure is dynamic
+    extractToolName(request) {
+        return request?.name ?? request?.tool ?? 'unknown';
+    }
+    /**
+     * Extract tool inputs from request
+     */
+    // biome-ignore lint/suspicious/noExplicitAny: LangChain request structure is dynamic
+    extractToolInputs(request) {
+        return { input: request?.input ?? request?.args ?? {} };
+    }
+    /**
+     * Extract tool outputs from response
+     */
+    // biome-ignore lint/suspicious/noExplicitAny: LangChain response structure is dynamic
+    extractToolOutputs(response) {
+        return { output: response?.output ?? response };
+    }
+}
+//# sourceMappingURL=middleware.js.map

package/dist/middleware.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.js","sourceRoot":"","sources":["../src/middleware.ts"],"names":[],"mappings":"AAAA,OAAO,EAAa,WAAW,EAAE,QAAQ,EAAe,MAAM,iBAAiB,CAAC;AAChF,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAE5D;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,MAAM,OAAO,mBAAmB;IAGV;IAFZ,QAAQ,GAAgB,IAAI,CAAC;IAErC,YAAoB,MAAc;QAAd,WAAM,GAAN,MAAM,CAAQ;IAAG,CAAC;IAEtC;;;;OAIG;IACH,mFAAmF;IACnF,KAAK,CAAC,WAAW,CAAC,KAAU;QAC1B,MAAM,UAAU,GAAG,WAAW,CAAC,UAAU,EAAE,CAAC;QAC5C,MAAM,QAAQ,GAAG,KAAK,EAAE,QAAQ,IAAI,EAAE,CAAC;QAEvC,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;QAEjC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;YACjC,IAAI,EAAE,OAAO;YACb,QAAQ,EAAE,QAAQ,CAAC,KAAK;YACxB,MAAM,EAAE,EAAE,QAAQ,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE;YACvE,YAAY,EAAE,UAAU,EAAE,MAAM;YAChC,OAAO,EAAE,UAAU,EAAE,OAAO;SAC7B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;IACvB,CAAC;IAED;;;;OAIG;IACH,mFAAmF;IACnF,KAAK,CAAC,UAAU,CAAC,KAAU;QACzB,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QAED,MAAM,QAAQ,GAAG,KAAK,EAAE,QAAQ,IAAI,EAAE,CAAC;QACvC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE;YACjC,OAAO,EAAE,EAAE,QAAQ,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE;SACzE,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC;QAClC,WAAW,CAAC,KAAK,EAAE,CAAC;QACpB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;IACvB,CAAC;IAED;;;;;;OAMG;IACH,0FAA0F;IAC1F,KAAK,CAAC,aAAa,CAAI,OAAY,EAAE,OAAiC;QACpE,MAAM,UAAU,GAAG,WAAW,CAAC,UAAU,EAAE,CAAC;QAE5C,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;YACjC,IAAI,EAAE,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC;YACpC,QAAQ,EAAE,QAAQ,CAAC,GAAG;YACtB,MAAM,EAAE,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC;YACxC,YAAY,EAAE,UAAU,EAAE,MAAM;YAChC,OAAO,EAAE,UAAU,EAAE,OAAO;SAC7B,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,sEAAsE;YACtE,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,QAAQ,CAAC,IAAI,EAAE,KAAK,IAAI,EAAE;gBAC3D,OAAO,OAAO,CAAC,OAAO,CAAC,CAAC;YAC1B,CAAC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;YACnD,MAAM,UAAU,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAE/C,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,IAAI,SAAS,EAAE,CAAC,CAAC;YAC5E,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,KAAc,EAAE,CAAC,CAAC;YACrD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;;;OAMG;IACH,0FAA0F;IAC1F,KAAK,CAAC,YAAY,CAAI,OAAY,EAAE,OAAiC;QACnE,MAAM,UAAU,GAAG,WAAW,CAAC,UAAU,EAAE,CAAC;QAE5C,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;YACjC,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC;YACnC,QAAQ,EAAE,QAAQ,CAAC,IAAI;YACvB,MAAM,EAAE,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC;YACvC,YAAY,EAAE,UAAU,EAAE,MAAM;YAChC,OAAO,EAAE,UAAU,EAAE,OAAO;SAC7B,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,sEAAsE;YACtE,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,QAAQ,CAAC,IAAI,EAAE,KAAK,IAAI,EAAE;gBAC3D,OAAO,OAAO,CAAC,OAAO,CAAC,CAAC;YAC1B,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE;gBACxB,OAAO,EAAE,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC;aAC3C,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,KAAc,EAAE,CAAC,CAAC;YACrD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,qFAAqF;IAC7E,gBAAgB,CAAC,OAAY;QACnC,OAAO,OAAO,EAAE,KAAK,IAAI,OAAO,EAAE,SAAS,IAAI,SAAS,CAAC;IAC3D,CAAC;IAED;;OAEG;IACH,qFAAqF;IAC7E,kBAAkB,CAAC,OAAY;QACrC,MAAM,QAAQ,GAAG,OAAO,EAAE,QAAQ,IAAI,EAAE,CAAC;QACzC,OAAO,EAAE,QAAQ,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACzE,CAAC;IAED;;OAEG;IACH,sFAAsF;IAC9E,mBAAmB,CAAC,QAAa;QACvC,MAAM,OAAO,GAAG,QAAQ,EAAE,OAAO,IAAI,QAAQ,EAAE,IAAI,IAAI,EAAE,CAAC;QAC1D,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IACtC,CAAC;IAED;;OAEG;IACH,qFAAqF;IAC7E,eAAe,CAAC,OAAY;QAClC,OAAO,OAAO,EAAE,IAAI,IAAI,OAAO,EAAE,IAAI,IAAI,SAAS,CAAC;IACrD,CAAC;IAED;;OAEG;IACH,qFAAqF;IAC7E,iBAAiB,CAAC,OAAY;QACpC,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,IAAI,OAAO,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC;IAC1D,CAAC;IAED;;OAEG;IACH,sFAAsF;IAC9E,kBAAkB,CAAC,QAAa;QACtC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,IAAI,QAAQ,EAAE,CAAC;IAClD,CAAC;CACF"}

package/package.json

@@ -0,0 +1,40 @@
+{
+  "name": "@prefactor/langchain",
+  "version": "0.1.0",
+  "description": "LangChain.js integration for Prefactor observability",
+  "type": "module",
+  "main": "dist/index.cjs",
+  "module": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js",
+      "require": "./dist/index.cjs"
+    }
+  },
+  "files": [
+    "dist"
+  ],
+  "keywords": [
+    "prefactor",
+    "observability",
+    "tracing",
+    "langchain",
+    "llm",
+    "agent",
+    "monitoring"
+  ],
+  "author": "Prefactor",
+  "license": "MIT",
+  "dependencies": {
+    "@prefactor/core": "0.1.0",
+    "@prefactor/pfid": "^0.1.0"
+  },
+  "peerDependencies": {
+    "langchain": "^1.0.0"
+  },
+  "engines": {
+    "node": ">=24.0.0"
+  }
+}