@prefactor/langchain 0.1.1 → 0.2.1

package/README.md

@@ -10,7 +10,7 @@ npm install @prefactor/langchain
 bun add @prefactor/langchain
 ```
 
-Most users should install `@prefactor/sdk` instead, which bundles both `@prefactor/core` and `@prefactor/langchain`.
+Note: This package requires `@prefactor/core` as a peer dependency, which will be installed automatically.
 
 ## Peer Dependencies
 
@@ -78,6 +78,7 @@ For convenience, common types are re-exported:
 ```typescript
 import {
   type Config,
+  type CoreRuntime,
   type HttpTransportConfig,
   type Span,
   SpanStatus,
@@ -93,6 +94,11 @@ import {
 - `PREFACTOR_API_URL`: API endpoint for HTTP transport
 - `PREFACTOR_API_TOKEN`: Authentication token
 - `PREFACTOR_SAMPLE_RATE`: Sampling rate 0.0-1.0 (default: `1.0`)
+- `PREFACTOR_CAPTURE_INPUTS`: Capture span inputs (default: `true`)
+- `PREFACTOR_CAPTURE_OUTPUTS`: Capture span outputs (default: `true`)
+- `PREFACTOR_MAX_INPUT_LENGTH`: Max input string length (default: `10000`)
+- `PREFACTOR_MAX_OUTPUT_LENGTH`: Max output string length (default: `10000`)
+- `PREFACTOR_LOG_LEVEL`: `"debug"` | `"info"` | `"warn"` | `"error"` (default: `"info")`
 
 ### Programmatic Configuration
 
@@ -109,7 +115,9 @@ const middleware = init({
     apiUrl: 'https://api.prefactor.ai',
     apiToken: process.env.PREFACTOR_API_TOKEN!,
     agentId: 'my-agent',
-    agentVersion: '1.0.0',
+    agentIdentifier: '1.0.0',
+    agentName: 'My Agent',
+    agentDescription: 'An agent description',
   },
 });
 ```

package/dist/index.cjs

@@ -29,6 +29,7 @@ var __export = (target, all) => {
 // packages/langchain/src/index.ts
 var exports_src = {};
 __export(exports_src, {
+  withSpan: () => withSpan,
   shutdown: () => shutdown,
   init: () => init,
   getTracer: () => getTracer,
@@ -42,7 +43,6 @@ 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
@@ -84,22 +84,24 @@ function extractTokenUsage(response) {
 // packages/langchain/src/middleware.ts
 class PrefactorMiddleware {
   tracer;
+  agentManager;
+  agentInfo;
   rootSpan = null;
-  constructor(tracer) {
+  constructor(tracer, agentManager, agentInfo) {
     this.tracer = tracer;
+    this.agentManager = agentManager;
+    this.agentInfo = agentInfo;
   }
   async beforeAgent(state) {
-    const parentSpan = import_core.SpanContext.getCurrent();
     const messages = state?.messages ?? [];
-    this.tracer.startAgentInstance();
+    this.agentManager.startInstance(this.agentInfo);
     const span = this.tracer.startSpan({
-      name: "agent",
+      name: "langchain:agent",
       spanType: import_core.SpanType.AGENT,
-      inputs: { messages: messages.slice(-3).map((m) => String(m)) },
-      parentSpanId: parentSpan?.spanId,
-      traceId: parentSpan?.traceId
+      inputs: { messages: import_core.serializeValue(messages.slice(-3)) }
     });
     this.rootSpan = span;
+    import_core.SpanContext.enter(span);
   }
   async afterAgent(state) {
     if (!this.rootSpan) {
@@ -107,20 +109,17 @@ class PrefactorMiddleware {
     }
     const messages = state?.messages ?? [];
     this.tracer.endSpan(this.rootSpan, {
-      outputs: { messages: messages.slice(-3).map((m) => String(m)) }
+      outputs: { messages: import_core.serializeValue(messages.slice(-3)) }
     });
-    this.tracer.finishAgentInstance();
-    import_core.SpanContext.clear();
+    this.agentManager.finishInstance();
+    import_core.SpanContext.exit();
     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
+      inputs: this.extractModelInputs(request)
     });
     try {
       const response = await import_core.SpanContext.runAsync(span, async () => {
@@ -136,13 +135,10 @@ class PrefactorMiddleware {
     }
   }
   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
+      inputs: this.extractToolInputs(request)
     });
     try {
       const response = await import_core.SpanContext.runAsync(span, async () => {
@@ -158,15 +154,31 @@ class PrefactorMiddleware {
     }
   }
   extractModelName(request) {
-    return request?.model ?? request?.modelName ?? "unknown";
+    const candidate = request?.model ?? request?.modelName;
+    if (typeof candidate === "string") {
+      return candidate;
+    }
+    if (candidate && typeof candidate === "object") {
+      const modelObject = candidate;
+      if (Array.isArray(modelObject.id) && modelObject.id.every((item) => typeof item === "string")) {
+        return modelObject.id.join(".");
+      }
+      if (typeof modelObject.modelName === "string") {
+        return modelObject.modelName;
+      }
+      if (typeof modelObject.name === "string") {
+        return modelObject.name;
+      }
+    }
+    return "unknown";
   }
   extractModelInputs(request) {
     const messages = request?.messages ?? [];
-    return { messages: messages.slice(-3).map((m) => String(m)) };
+    return { messages: import_core.serializeValue(messages.slice(-3)) };
   }
   extractModelOutputs(response) {
     const content = response?.content ?? response?.text ?? "";
-    return { content: String(content) };
+    return { content: import_core.serializeValue(content) };
   }
   extractToolName(request) {
     return request?.name ?? request?.tool ?? "unknown";
@@ -181,36 +193,67 @@ class PrefactorMiddleware {
 
 // packages/langchain/src/init.ts
 var logger = import_core2.getLogger("init");
+var DEFAULT_LANGCHAIN_AGENT_SCHEMA = {
+  external_identifier: "prefactor",
+  span_schemas: {
+    agent: { type: "object", additionalProperties: true },
+    llm: { type: "object", additionalProperties: true },
+    tool: { type: "object", additionalProperties: true },
+    chain: { type: "object", additionalProperties: true }
+  }
+};
+var globalCore = null;
 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 });
+  let configWithHttp = config;
+  const transportType = config?.transportType ?? process.env.PREFACTOR_TRANSPORT ?? "http";
+  if (transportType === "http" && !config?.httpConfig) {
+    const apiUrl = process.env.PREFACTOR_API_URL;
+    const apiToken = process.env.PREFACTOR_API_TOKEN;
+    if (!apiUrl || !apiToken) {
+      throw new Error("HTTP transport requires PREFACTOR_API_URL and PREFACTOR_API_TOKEN environment variables, " + "or httpConfig to be provided in configuration");
+    }
+    configWithHttp = {
+      ...config,
+      transportType: "http",
+      httpConfig: {
+        apiUrl,
+        apiToken,
+        agentId: process.env.PREFACTOR_AGENT_ID,
+        agentName: process.env.PREFACTOR_AGENT_NAME,
+        agentIdentifier: process.env.PREFACTOR_AGENT_IDENTIFIER || "1.0.0",
+        agentSchema: DEFAULT_LANGCHAIN_AGENT_SCHEMA
+      }
+    };
+  } else if (transportType === "http" && config?.httpConfig && !config.httpConfig.agentSchema) {
+    configWithHttp = {
+      ...config,
+      httpConfig: {
+        ...config.httpConfig,
+        agentSchema: DEFAULT_LANGCHAIN_AGENT_SCHEMA
+      }
+    };
+  }
+  const finalConfig = import_core2.createConfig(configWithHttp);
   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 });
-    }
+  const core = import_core2.createCore(finalConfig);
+  globalCore = core;
+  globalTracer = core.tracer;
+  const httpConfig = finalConfig.httpConfig;
+  if (httpConfig?.agentSchema) {
+    core.agentManager.registerSchema(httpConfig.agentSchema);
   }
-  globalTracer = new import_core2.Tracer(transport, partition);
-  const prefactorMiddleware = new PrefactorMiddleware(globalTracer);
+  const agentInfo = finalConfig.httpConfig ? {
+    agentId: finalConfig.httpConfig.agentId,
+    agentIdentifier: finalConfig.httpConfig.agentIdentifier,
+    agentName: finalConfig.httpConfig.agentName,
+    agentDescription: finalConfig.httpConfig.agentDescription
+  } : undefined;
+  const prefactorMiddleware = new PrefactorMiddleware(core.tracer, core.agentManager, agentInfo);
   const middleware = import_langchain.createMiddleware({
     name: "prefactor",
     wrapModelCall: async (request, handler) => {
@@ -235,11 +278,25 @@ function getTracer() {
   }
   return globalTracer;
 }
+async function withSpan(options, fn) {
+  const tracer = getTracer();
+  const span = tracer.startSpan(options);
+  try {
+    const result = await import_core2.SpanContext.runAsync(span, async () => await fn());
+    tracer.endSpan(span);
+    return result;
+  } catch (error) {
+    const normalizedError = error instanceof Error ? error : new Error(String(error));
+    tracer.endSpan(span, { error: normalizedError });
+    throw error;
+  }
+}
 async function shutdown() {
-  if (globalTracer) {
+  if (globalCore) {
     logger.info("Shutting down Prefactor SDK");
-    await globalTracer.close();
+    await globalCore.shutdown();
   }
+  globalCore = null;
   globalTracer = null;
   globalMiddleware = null;
 }
@@ -249,4 +306,4 @@ process.on("beforeExit", () => {
   });
 });
 
-//# debugId=46B60F8DD9AA982064756E2164756E21
+//# debugId=534355E26FF7C82564756E2164756E21

package/dist/index.cjs.map

@@ -2,12 +2,12 @@
   "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",
+    "// Main entry points\n\n// Convenience re-exports from core\nexport {\n  type Config,\n  type CoreRuntime,\n  type HttpTransportConfig,\n  type Span,\n  SpanStatus,\n  SpanType,\n} from '@prefactor/core';\nexport { getTracer, init, shutdown, withSpan } from './init.js';\nexport { extractTokenUsage } from './metadata-extractor.js';\n// Middleware\nexport { PrefactorMiddleware } from './middleware.js';\n",
+    "import {\n  type Config,\n  type CoreRuntime,\n  configureLogging,\n  createConfig,\n  createCore,\n  getLogger,\n  SpanContext,\n  type Tracer,\n} from '@prefactor/core';\nimport { type AgentMiddleware, createMiddleware } from 'langchain';\nimport { PrefactorMiddleware } from './middleware.js';\n\nconst logger = getLogger('init');\n\nconst DEFAULT_LANGCHAIN_AGENT_SCHEMA = {\n  external_identifier: 'prefactor',\n  span_schemas: {\n    agent: { type: 'object', additionalProperties: true },\n    llm: { type: 'object', additionalProperties: true },\n    tool: { type: 'object', additionalProperties: true },\n    chain: { type: 'object', additionalProperties: true },\n  },\n} as const;\n\nlet globalCore: CoreRuntime | null = null;\nlet globalTracer: Tracer | null = null;\nlet globalMiddleware: AgentMiddleware | null = null;\n\nexport type ManualSpanOptions = {\n  name: string;\n  spanType: string;\n  inputs: Record<string, unknown>;\n  metadata?: Record<string, unknown>;\n};\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 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 *     agentIdentifier: 'my-langchain-agent',\n *   },\n * });\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 *     agentIdentifier: 'my-langchain-agent', // Required\n *     agentId: 'legacy-agent-id', // Optional legacy identifier\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  let configWithHttp = config;\n  const transportType = config?.transportType ?? process.env.PREFACTOR_TRANSPORT ?? 'http';\n\n  if (transportType === 'http' && !config?.httpConfig) {\n    const apiUrl = process.env.PREFACTOR_API_URL;\n    const apiToken = process.env.PREFACTOR_API_TOKEN;\n\n    if (!apiUrl || !apiToken) {\n      throw new Error(\n        'HTTP transport requires PREFACTOR_API_URL and PREFACTOR_API_TOKEN environment variables, ' +\n          'or httpConfig to be provided in configuration'\n      );\n    }\n\n    configWithHttp = {\n      ...config,\n      transportType: 'http',\n      httpConfig: {\n        apiUrl,\n        apiToken,\n        agentId: process.env.PREFACTOR_AGENT_ID,\n        agentName: process.env.PREFACTOR_AGENT_NAME,\n        agentIdentifier: process.env.PREFACTOR_AGENT_IDENTIFIER || '1.0.0',\n        agentSchema: DEFAULT_LANGCHAIN_AGENT_SCHEMA,\n      },\n    };\n  } else if (transportType === 'http' && config?.httpConfig && !config.httpConfig.agentSchema) {\n    configWithHttp = {\n      ...config,\n      httpConfig: {\n        ...config.httpConfig,\n        agentSchema: DEFAULT_LANGCHAIN_AGENT_SCHEMA,\n      },\n    };\n  }\n\n  const finalConfig = createConfig(configWithHttp);\n\n  if (globalMiddleware !== null) {\n    return globalMiddleware;\n  }\n\n  const core = createCore(finalConfig);\n  globalCore = core;\n  globalTracer = core.tracer;\n\n  const httpConfig = finalConfig.httpConfig;\n  if (httpConfig?.agentSchema) {\n    core.agentManager.registerSchema(httpConfig.agentSchema);\n  }\n\n  const agentInfo = finalConfig.httpConfig\n    ? {\n        agentId: finalConfig.httpConfig.agentId,\n        agentIdentifier: finalConfig.httpConfig.agentIdentifier,\n        agentName: finalConfig.httpConfig.agentName,\n        agentDescription: finalConfig.httpConfig.agentDescription,\n      }\n    : undefined;\n\n  const prefactorMiddleware = new PrefactorMiddleware(core.tracer, core.agentManager, agentInfo);\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\nexport async function withSpan<T>(\n  options: ManualSpanOptions,\n  fn: () => Promise<T> | T\n): Promise<T> {\n  const tracer = getTracer();\n  const span = tracer.startSpan(options);\n\n  try {\n    const result = await SpanContext.runAsync(span, async () => await fn());\n    tracer.endSpan(span);\n    return result;\n  } catch (error) {\n    const normalizedError = error instanceof Error ? error : new Error(String(error));\n    tracer.endSpan(span, { error: normalizedError });\n    throw error;\n  }\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 (globalCore) {\n    logger.info('Shutting down Prefactor SDK');\n    await globalCore.shutdown();\n  }\n  globalCore = null;\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 {\n  type AgentInstanceManager,\n  SpanContext,\n  SpanType,\n  serializeValue,\n  type Tracer,\n} 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: ReturnType<Tracer['startSpan']> | null = null;\n\n  constructor(\n    private tracer: Tracer,\n    private agentManager: AgentInstanceManager,\n    private agentInfo?: Parameters<AgentInstanceManager['startInstance']>[0]\n  ) {}\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 messages = state?.messages ?? [];\n\n    this.agentManager.startInstance(this.agentInfo);\n\n    const span = this.tracer.startSpan({\n      name: 'langchain:agent',\n      spanType: SpanType.AGENT,\n      inputs: { messages: serializeValue(messages.slice(-3)) },\n    });\n\n    this.rootSpan = span;\n    SpanContext.enter(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: serializeValue(messages.slice(-3)) },\n    });\n\n    this.agentManager.finishInstance();\n    SpanContext.exit();\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 span = this.tracer.startSpan({\n      name: this.extractModelName(request),\n      spanType: SpanType.LLM,\n      inputs: this.extractModelInputs(request),\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 span = this.tracer.startSpan({\n      name: this.extractToolName(request),\n      spanType: SpanType.TOOL,\n      inputs: this.extractToolInputs(request),\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    const candidate = request?.model ?? request?.modelName;\n\n    if (typeof candidate === 'string') {\n      return candidate;\n    }\n\n    if (candidate && typeof candidate === 'object') {\n      const modelObject = candidate as Record<string, unknown>;\n      if (\n        Array.isArray(modelObject.id) &&\n        modelObject.id.every((item) => typeof item === 'string')\n      ) {\n        return (modelObject.id as string[]).join('.');\n      }\n\n      if (typeof modelObject.modelName === 'string') {\n        return modelObject.modelName;\n      }\n\n      if (typeof modelObject.name === 'string') {\n        return modelObject.name;\n      }\n    }\n\n    return '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: serializeValue(messages.slice(-3)) };\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: serializeValue(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",
+  "mappings": ";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAUO,IAPP;;;ACMO,IATP;AAUuD,IAAvD;;;ACJO,IANP;;;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;;;ADpBJ,MAAM,oBAAoB;AAAA,EAIrB;AAAA,EACA;AAAA,EACA;AAAA,EALF,WAAmD;AAAA,EAE3D,WAAW,CACD,QACA,cACA,WACR;AAAA,IAHQ;AAAA,IACA;AAAA,IACA;AAAA;AAAA,OASJ,YAAW,CAAC,OAA2B;AAAA,IAC3C,MAAM,WAAW,OAAO,YAAY,CAAC;AAAA,IAErC,KAAK,aAAa,cAAc,KAAK,SAAS;AAAA,IAE9C,MAAM,OAAO,KAAK,OAAO,UAAU;AAAA,MACjC,MAAM;AAAA,MACN,UAAU,qBAAS;AAAA,MACnB,QAAQ,EAAE,UAAU,2BAAe,SAAS,MAAM,EAAE,CAAC,EAAE;AAAA,IACzD,CAAC;AAAA,IAED,KAAK,WAAW;AAAA,IAChB,wBAAY,MAAM,IAAI;AAAA;AAAA,OASlB,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,2BAAe,SAAS,MAAM,EAAE,CAAC,EAAE;AAAA,IAC1D,CAAC;AAAA,IAED,KAAK,aAAa,eAAe;AAAA,IACjC,wBAAY,KAAK;AAAA,IACjB,KAAK,WAAW;AAAA;AAAA,OAWZ,cAAgB,CAAC,SAAc,SAA+C;AAAA,IAClF,MAAM,OAAO,KAAK,OAAO,UAAU;AAAA,MACjC,MAAM,KAAK,iBAAiB,OAAO;AAAA,MACnC,UAAU,qBAAS;AAAA,MACnB,QAAQ,KAAK,mBAAmB,OAAO;AAAA,IACzC,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,OAAO,KAAK,OAAO,UAAU;AAAA,MACjC,MAAM,KAAK,gBAAgB,OAAO;AAAA,MAClC,UAAU,qBAAS;AAAA,MACnB,QAAQ,KAAK,kBAAkB,OAAO;AAAA,IACxC,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,MAAM,YAAY,SAAS,SAAS,SAAS;AAAA,IAE7C,IAAI,OAAO,cAAc,UAAU;AAAA,MACjC,OAAO;AAAA,IACT;AAAA,IAEA,IAAI,aAAa,OAAO,cAAc,UAAU;AAAA,MAC9C,MAAM,cAAc;AAAA,MACpB,IACE,MAAM,QAAQ,YAAY,EAAE,KAC5B,YAAY,GAAG,MAAM,CAAC,SAAS,OAAO,SAAS,QAAQ,GACvD;AAAA,QACA,OAAQ,YAAY,GAAgB,KAAK,GAAG;AAAA,MAC9C;AAAA,MAEA,IAAI,OAAO,YAAY,cAAc,UAAU;AAAA,QAC7C,OAAO,YAAY;AAAA,MACrB;AAAA,MAEA,IAAI,OAAO,YAAY,SAAS,UAAU;AAAA,QACxC,OAAO,YAAY;AAAA,MACrB;AAAA,IACF;AAAA,IAEA,OAAO;AAAA;AAAA,EAOD,kBAAkB,CAAC,SAAuC;AAAA,IAChE,MAAM,WAAW,SAAS,YAAY,CAAC;AAAA,IACvC,OAAO,EAAE,UAAU,2BAAe,SAAS,MAAM,EAAE,CAAC,EAAE;AAAA;AAAA,EAOhD,mBAAmB,CAAC,UAAwC;AAAA,IAClE,MAAM,UAAU,UAAU,WAAW,UAAU,QAAQ;AAAA,IACvD,OAAO,EAAE,SAAS,2BAAe,OAAO,EAAE;AAAA;AAAA,EAOpC,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;;;ADhNA,IAAM,SAAS,uBAAU,MAAM;AAE/B,IAAM,iCAAiC;AAAA,EACrC,qBAAqB;AAAA,EACrB,cAAc;AAAA,IACZ,OAAO,EAAE,MAAM,UAAU,sBAAsB,KAAK;AAAA,IACpD,KAAK,EAAE,MAAM,UAAU,sBAAsB,KAAK;AAAA,IAClD,MAAM,EAAE,MAAM,UAAU,sBAAsB,KAAK;AAAA,IACnD,OAAO,EAAE,MAAM,UAAU,sBAAsB,KAAK;AAAA,EACtD;AACF;AAEA,IAAI,aAAiC;AACrC,IAAI,eAA8B;AAClC,IAAI,mBAA2C;AAmDxC,SAAS,IAAI,CAAC,QAA2C;AAAA,EAC9D,8BAAiB;AAAA,EAEjB,IAAI,iBAAiB;AAAA,EACrB,MAAM,gBAAgB,QAAQ,iBAAiB,QAAQ,IAAI,uBAAuB;AAAA,EAElF,IAAI,kBAAkB,UAAU,CAAC,QAAQ,YAAY;AAAA,IACnD,MAAM,SAAS,QAAQ,IAAI;AAAA,IAC3B,MAAM,WAAW,QAAQ,IAAI;AAAA,IAE7B,IAAI,CAAC,UAAU,CAAC,UAAU;AAAA,MACxB,MAAM,IAAI,MACR,8FACE,+CACJ;AAAA,IACF;AAAA,IAEA,iBAAiB;AAAA,SACZ;AAAA,MACH,eAAe;AAAA,MACf,YAAY;AAAA,QACV;AAAA,QACA;AAAA,QACA,SAAS,QAAQ,IAAI;AAAA,QACrB,WAAW,QAAQ,IAAI;AAAA,QACvB,iBAAiB,QAAQ,IAAI,8BAA8B;AAAA,QAC3D,aAAa;AAAA,MACf;AAAA,IACF;AAAA,EACF,EAAO,SAAI,kBAAkB,UAAU,QAAQ,cAAc,CAAC,OAAO,WAAW,aAAa;AAAA,IAC3F,iBAAiB;AAAA,SACZ;AAAA,MACH,YAAY;AAAA,WACP,OAAO;AAAA,QACV,aAAa;AAAA,MACf;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,cAAc,0BAAa,cAAc;AAAA,EAE/C,IAAI,qBAAqB,MAAM;AAAA,IAC7B,OAAO;AAAA,EACT;AAAA,EAEA,MAAM,OAAO,wBAAW,WAAW;AAAA,EACnC,aAAa;AAAA,EACb,eAAe,KAAK;AAAA,EAEpB,MAAM,aAAa,YAAY;AAAA,EAC/B,IAAI,YAAY,aAAa;AAAA,IAC3B,KAAK,aAAa,eAAe,WAAW,WAAW;AAAA,EACzD;AAAA,EAEA,MAAM,YAAY,YAAY,aAC1B;AAAA,IACE,SAAS,YAAY,WAAW;AAAA,IAChC,iBAAiB,YAAY,WAAW;AAAA,IACxC,WAAW,YAAY,WAAW;AAAA,IAClC,kBAAkB,YAAY,WAAW;AAAA,EAC3C,IACA;AAAA,EAEJ,MAAM,sBAAsB,IAAI,oBAAoB,KAAK,QAAQ,KAAK,cAAc,SAAS;AAAA,EAE7F,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;AAGT,eAAsB,QAAW,CAC/B,SACA,IACY;AAAA,EACZ,MAAM,SAAS,UAAU;AAAA,EACzB,MAAM,OAAO,OAAO,UAAU,OAAO;AAAA,EAErC,IAAI;AAAA,IACF,MAAM,SAAS,MAAM,yBAAY,SAAS,MAAM,YAAY,MAAM,GAAG,CAAC;AAAA,IACtE,OAAO,QAAQ,IAAI;AAAA,IACnB,OAAO;AAAA,IACP,OAAO,OAAO;AAAA,IACd,MAAM,kBAAkB,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC;AAAA,IAChF,OAAO,QAAQ,MAAM,EAAE,OAAO,gBAAgB,CAAC;AAAA,IAC/C,MAAM;AAAA;AAAA;AAsBV,eAAsB,QAAQ,GAAkB;AAAA,EAC9C,IAAI,YAAY;AAAA,IACd,OAAO,KAAK,6BAA6B;AAAA,IACzC,MAAM,WAAW,SAAS;AAAA,EAC5B;AAAA,EACA,aAAa;AAAA,EACb,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": "534355E26FF7C82564756E2164756E21",
   "names": []
 }

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-export { type Config, type HttpTransportConfig, type Span, SpanStatus, SpanType, } from '@prefactor/core';
-export { getTracer, init, shutdown } from './init.js';
+export { type Config, type CoreRuntime, type HttpTransportConfig, type Span, SpanStatus, SpanType, } from '@prefactor/core';
+export { getTracer, init, shutdown, withSpan } 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

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

package/dist/index.js

@@ -8,17 +8,18 @@ import {
 import {
   configureLogging,
   createConfig,
+  createCore,
   getLogger,
-  HttpTransport,
-  HttpTransportConfigSchema,
-  StdioTransport,
-  Tracer
+  SpanContext as SpanContext2
 } from "@prefactor/core";
-import { extractPartition } from "@prefactor/pfid";
 import { createMiddleware } from "langchain";
 
 // packages/langchain/src/middleware.ts
-import { SpanContext, SpanType } from "@prefactor/core";
+import {
+  SpanContext,
+  SpanType,
+  serializeValue
+} from "@prefactor/core";
 
 // packages/langchain/src/metadata-extractor.ts
 function extractTokenUsage(response) {
@@ -56,22 +57,24 @@ function extractTokenUsage(response) {
 // packages/langchain/src/middleware.ts
 class PrefactorMiddleware {
   tracer;
+  agentManager;
+  agentInfo;
   rootSpan = null;
-  constructor(tracer) {
+  constructor(tracer, agentManager, agentInfo) {
     this.tracer = tracer;
+    this.agentManager = agentManager;
+    this.agentInfo = agentInfo;
   }
   async beforeAgent(state) {
-    const parentSpan = SpanContext.getCurrent();
     const messages = state?.messages ?? [];
-    this.tracer.startAgentInstance();
+    this.agentManager.startInstance(this.agentInfo);
     const span = this.tracer.startSpan({
-      name: "agent",
+      name: "langchain:agent",
       spanType: SpanType.AGENT,
-      inputs: { messages: messages.slice(-3).map((m) => String(m)) },
-      parentSpanId: parentSpan?.spanId,
-      traceId: parentSpan?.traceId
+      inputs: { messages: serializeValue(messages.slice(-3)) }
     });
     this.rootSpan = span;
+    SpanContext.enter(span);
   }
   async afterAgent(state) {
     if (!this.rootSpan) {
@@ -79,20 +82,17 @@ class PrefactorMiddleware {
     }
     const messages = state?.messages ?? [];
     this.tracer.endSpan(this.rootSpan, {
-      outputs: { messages: messages.slice(-3).map((m) => String(m)) }
+      outputs: { messages: serializeValue(messages.slice(-3)) }
     });
-    this.tracer.finishAgentInstance();
-    SpanContext.clear();
+    this.agentManager.finishInstance();
+    SpanContext.exit();
     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
+      inputs: this.extractModelInputs(request)
     });
     try {
       const response = await SpanContext.runAsync(span, async () => {
@@ -108,13 +108,10 @@ class PrefactorMiddleware {
     }
   }
   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
+      inputs: this.extractToolInputs(request)
     });
     try {
       const response = await SpanContext.runAsync(span, async () => {
@@ -130,15 +127,31 @@ class PrefactorMiddleware {
     }
   }
   extractModelName(request) {
-    return request?.model ?? request?.modelName ?? "unknown";
+    const candidate = request?.model ?? request?.modelName;
+    if (typeof candidate === "string") {
+      return candidate;
+    }
+    if (candidate && typeof candidate === "object") {
+      const modelObject = candidate;
+      if (Array.isArray(modelObject.id) && modelObject.id.every((item) => typeof item === "string")) {
+        return modelObject.id.join(".");
+      }
+      if (typeof modelObject.modelName === "string") {
+        return modelObject.modelName;
+      }
+      if (typeof modelObject.name === "string") {
+        return modelObject.name;
+      }
+    }
+    return "unknown";
   }
   extractModelInputs(request) {
     const messages = request?.messages ?? [];
-    return { messages: messages.slice(-3).map((m) => String(m)) };
+    return { messages: serializeValue(messages.slice(-3)) };
   }
   extractModelOutputs(response) {
     const content = response?.content ?? response?.text ?? "";
-    return { content: String(content) };
+    return { content: serializeValue(content) };
   }
   extractToolName(request) {
     return request?.name ?? request?.tool ?? "unknown";
@@ -153,36 +166,67 @@ class PrefactorMiddleware {
 
 // packages/langchain/src/init.ts
 var logger = getLogger("init");
+var DEFAULT_LANGCHAIN_AGENT_SCHEMA = {
+  external_identifier: "prefactor",
+  span_schemas: {
+    agent: { type: "object", additionalProperties: true },
+    llm: { type: "object", additionalProperties: true },
+    tool: { type: "object", additionalProperties: true },
+    chain: { type: "object", additionalProperties: true }
+  }
+};
+var globalCore = null;
 var globalTracer = null;
 var globalMiddleware = null;
 function init(config) {
   configureLogging();
-  const finalConfig = createConfig(config);
-  logger.info("Initializing Prefactor SDK", { transport: finalConfig.transportType });
+  let configWithHttp = config;
+  const transportType = config?.transportType ?? process.env.PREFACTOR_TRANSPORT ?? "http";
+  if (transportType === "http" && !config?.httpConfig) {
+    const apiUrl = process.env.PREFACTOR_API_URL;
+    const apiToken = process.env.PREFACTOR_API_TOKEN;
+    if (!apiUrl || !apiToken) {
+      throw new Error("HTTP transport requires PREFACTOR_API_URL and PREFACTOR_API_TOKEN environment variables, " + "or httpConfig to be provided in configuration");
+    }
+    configWithHttp = {
+      ...config,
+      transportType: "http",
+      httpConfig: {
+        apiUrl,
+        apiToken,
+        agentId: process.env.PREFACTOR_AGENT_ID,
+        agentName: process.env.PREFACTOR_AGENT_NAME,
+        agentIdentifier: process.env.PREFACTOR_AGENT_IDENTIFIER || "1.0.0",
+        agentSchema: DEFAULT_LANGCHAIN_AGENT_SCHEMA
+      }
+    };
+  } else if (transportType === "http" && config?.httpConfig && !config.httpConfig.agentSchema) {
+    configWithHttp = {
+      ...config,
+      httpConfig: {
+        ...config.httpConfig,
+        agentSchema: DEFAULT_LANGCHAIN_AGENT_SCHEMA
+      }
+    };
+  }
+  const finalConfig = createConfig(configWithHttp);
   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 });
-    }
+  const core = createCore(finalConfig);
+  globalCore = core;
+  globalTracer = core.tracer;
+  const httpConfig = finalConfig.httpConfig;
+  if (httpConfig?.agentSchema) {
+    core.agentManager.registerSchema(httpConfig.agentSchema);
   }
-  globalTracer = new Tracer(transport, partition);
-  const prefactorMiddleware = new PrefactorMiddleware(globalTracer);
+  const agentInfo = finalConfig.httpConfig ? {
+    agentId: finalConfig.httpConfig.agentId,
+    agentIdentifier: finalConfig.httpConfig.agentIdentifier,
+    agentName: finalConfig.httpConfig.agentName,
+    agentDescription: finalConfig.httpConfig.agentDescription
+  } : undefined;
+  const prefactorMiddleware = new PrefactorMiddleware(core.tracer, core.agentManager, agentInfo);
   const middleware = createMiddleware({
     name: "prefactor",
     wrapModelCall: async (request, handler) => {
@@ -207,11 +251,25 @@ function getTracer() {
   }
   return globalTracer;
 }
+async function withSpan(options, fn) {
+  const tracer = getTracer();
+  const span = tracer.startSpan(options);
+  try {
+    const result = await SpanContext2.runAsync(span, async () => await fn());
+    tracer.endSpan(span);
+    return result;
+  } catch (error) {
+    const normalizedError = error instanceof Error ? error : new Error(String(error));
+    tracer.endSpan(span, { error: normalizedError });
+    throw error;
+  }
+}
 async function shutdown() {
-  if (globalTracer) {
+  if (globalCore) {
     logger.info("Shutting down Prefactor SDK");
-    await globalTracer.close();
+    await globalCore.shutdown();
   }
+  globalCore = null;
   globalTracer = null;
   globalMiddleware = null;
 }
@@ -221,6 +279,7 @@ process.on("beforeExit", () => {
   });
 });
 export {
+  withSpan,
   shutdown,
   init,
   getTracer,
@@ -230,4 +289,4 @@ export {
   PrefactorMiddleware
 };
 
-//# debugId=B136D336DD9CEBB764756E2164756E21
+//# debugId=EBAE04F79C70EA8764756E2164756E21

package/dist/index.js.map

@@ -2,12 +2,12 @@
   "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",
+    "// Main entry points\n\n// Convenience re-exports from core\nexport {\n  type Config,\n  type CoreRuntime,\n  type HttpTransportConfig,\n  type Span,\n  SpanStatus,\n  SpanType,\n} from '@prefactor/core';\nexport { getTracer, init, shutdown, withSpan } from './init.js';\nexport { extractTokenUsage } from './metadata-extractor.js';\n// Middleware\nexport { PrefactorMiddleware } from './middleware.js';\n",
+    "import {\n  type Config,\n  type CoreRuntime,\n  configureLogging,\n  createConfig,\n  createCore,\n  getLogger,\n  SpanContext,\n  type Tracer,\n} from '@prefactor/core';\nimport { type AgentMiddleware, createMiddleware } from 'langchain';\nimport { PrefactorMiddleware } from './middleware.js';\n\nconst logger = getLogger('init');\n\nconst DEFAULT_LANGCHAIN_AGENT_SCHEMA = {\n  external_identifier: 'prefactor',\n  span_schemas: {\n    agent: { type: 'object', additionalProperties: true },\n    llm: { type: 'object', additionalProperties: true },\n    tool: { type: 'object', additionalProperties: true },\n    chain: { type: 'object', additionalProperties: true },\n  },\n} as const;\n\nlet globalCore: CoreRuntime | null = null;\nlet globalTracer: Tracer | null = null;\nlet globalMiddleware: AgentMiddleware | null = null;\n\nexport type ManualSpanOptions = {\n  name: string;\n  spanType: string;\n  inputs: Record<string, unknown>;\n  metadata?: Record<string, unknown>;\n};\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 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 *     agentIdentifier: 'my-langchain-agent',\n *   },\n * });\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 *     agentIdentifier: 'my-langchain-agent', // Required\n *     agentId: 'legacy-agent-id', // Optional legacy identifier\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  let configWithHttp = config;\n  const transportType = config?.transportType ?? process.env.PREFACTOR_TRANSPORT ?? 'http';\n\n  if (transportType === 'http' && !config?.httpConfig) {\n    const apiUrl = process.env.PREFACTOR_API_URL;\n    const apiToken = process.env.PREFACTOR_API_TOKEN;\n\n    if (!apiUrl || !apiToken) {\n      throw new Error(\n        'HTTP transport requires PREFACTOR_API_URL and PREFACTOR_API_TOKEN environment variables, ' +\n          'or httpConfig to be provided in configuration'\n      );\n    }\n\n    configWithHttp = {\n      ...config,\n      transportType: 'http',\n      httpConfig: {\n        apiUrl,\n        apiToken,\n        agentId: process.env.PREFACTOR_AGENT_ID,\n        agentName: process.env.PREFACTOR_AGENT_NAME,\n        agentIdentifier: process.env.PREFACTOR_AGENT_IDENTIFIER || '1.0.0',\n        agentSchema: DEFAULT_LANGCHAIN_AGENT_SCHEMA,\n      },\n    };\n  } else if (transportType === 'http' && config?.httpConfig && !config.httpConfig.agentSchema) {\n    configWithHttp = {\n      ...config,\n      httpConfig: {\n        ...config.httpConfig,\n        agentSchema: DEFAULT_LANGCHAIN_AGENT_SCHEMA,\n      },\n    };\n  }\n\n  const finalConfig = createConfig(configWithHttp);\n\n  if (globalMiddleware !== null) {\n    return globalMiddleware;\n  }\n\n  const core = createCore(finalConfig);\n  globalCore = core;\n  globalTracer = core.tracer;\n\n  const httpConfig = finalConfig.httpConfig;\n  if (httpConfig?.agentSchema) {\n    core.agentManager.registerSchema(httpConfig.agentSchema);\n  }\n\n  const agentInfo = finalConfig.httpConfig\n    ? {\n        agentId: finalConfig.httpConfig.agentId,\n        agentIdentifier: finalConfig.httpConfig.agentIdentifier,\n        agentName: finalConfig.httpConfig.agentName,\n        agentDescription: finalConfig.httpConfig.agentDescription,\n      }\n    : undefined;\n\n  const prefactorMiddleware = new PrefactorMiddleware(core.tracer, core.agentManager, agentInfo);\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\nexport async function withSpan<T>(\n  options: ManualSpanOptions,\n  fn: () => Promise<T> | T\n): Promise<T> {\n  const tracer = getTracer();\n  const span = tracer.startSpan(options);\n\n  try {\n    const result = await SpanContext.runAsync(span, async () => await fn());\n    tracer.endSpan(span);\n    return result;\n  } catch (error) {\n    const normalizedError = error instanceof Error ? error : new Error(String(error));\n    tracer.endSpan(span, { error: normalizedError });\n    throw error;\n  }\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 (globalCore) {\n    logger.info('Shutting down Prefactor SDK');\n    await globalCore.shutdown();\n  }\n  globalCore = null;\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 {\n  type AgentInstanceManager,\n  SpanContext,\n  SpanType,\n  serializeValue,\n  type Tracer,\n} 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: ReturnType<Tracer['startSpan']> | null = null;\n\n  constructor(\n    private tracer: Tracer,\n    private agentManager: AgentInstanceManager,\n    private agentInfo?: Parameters<AgentInstanceManager['startInstance']>[0]\n  ) {}\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 messages = state?.messages ?? [];\n\n    this.agentManager.startInstance(this.agentInfo);\n\n    const span = this.tracer.startSpan({\n      name: 'langchain:agent',\n      spanType: SpanType.AGENT,\n      inputs: { messages: serializeValue(messages.slice(-3)) },\n    });\n\n    this.rootSpan = span;\n    SpanContext.enter(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: serializeValue(messages.slice(-3)) },\n    });\n\n    this.agentManager.finishInstance();\n    SpanContext.exit();\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 span = this.tracer.startSpan({\n      name: this.extractModelName(request),\n      spanType: SpanType.LLM,\n      inputs: this.extractModelInputs(request),\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 span = this.tracer.startSpan({\n      name: this.extractToolName(request),\n      spanType: SpanType.TOOL,\n      inputs: this.extractToolInputs(request),\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    const candidate = request?.model ?? request?.modelName;\n\n    if (typeof candidate === 'string') {\n      return candidate;\n    }\n\n    if (candidate && typeof candidate === 'object') {\n      const modelObject = candidate as Record<string, unknown>;\n      if (\n        Array.isArray(modelObject.id) &&\n        modelObject.id.every((item) => typeof item === 'string')\n      ) {\n        return (modelObject.id as string[]).join('.');\n      }\n\n      if (typeof modelObject.modelName === 'string') {\n        return modelObject.modelName;\n      }\n\n      if (typeof modelObject.name === 'string') {\n        return modelObject.name;\n      }\n    }\n\n    return '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: serializeValue(messages.slice(-3)) };\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: serializeValue(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",
+  "mappings": ";AAGA;AAAA;AAAA,cAME;AAAA;;;ACTF;AAAA;AAAA;AAAA;AAAA;AAAA,iBAOE;AAAA;AAGF;;;ACVA;AAAA;AAAA;AAAA;AAAA;;;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;;;ADpBJ,MAAM,oBAAoB;AAAA,EAIrB;AAAA,EACA;AAAA,EACA;AAAA,EALF,WAAmD;AAAA,EAE3D,WAAW,CACD,QACA,cACA,WACR;AAAA,IAHQ;AAAA,IACA;AAAA,IACA;AAAA;AAAA,OASJ,YAAW,CAAC,OAA2B;AAAA,IAC3C,MAAM,WAAW,OAAO,YAAY,CAAC;AAAA,IAErC,KAAK,aAAa,cAAc,KAAK,SAAS;AAAA,IAE9C,MAAM,OAAO,KAAK,OAAO,UAAU;AAAA,MACjC,MAAM;AAAA,MACN,UAAU,SAAS;AAAA,MACnB,QAAQ,EAAE,UAAU,eAAe,SAAS,MAAM,EAAE,CAAC,EAAE;AAAA,IACzD,CAAC;AAAA,IAED,KAAK,WAAW;AAAA,IAChB,YAAY,MAAM,IAAI;AAAA;AAAA,OASlB,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,eAAe,SAAS,MAAM,EAAE,CAAC,EAAE;AAAA,IAC1D,CAAC;AAAA,IAED,KAAK,aAAa,eAAe;AAAA,IACjC,YAAY,KAAK;AAAA,IACjB,KAAK,WAAW;AAAA;AAAA,OAWZ,cAAgB,CAAC,SAAc,SAA+C;AAAA,IAClF,MAAM,OAAO,KAAK,OAAO,UAAU;AAAA,MACjC,MAAM,KAAK,iBAAiB,OAAO;AAAA,MACnC,UAAU,SAAS;AAAA,MACnB,QAAQ,KAAK,mBAAmB,OAAO;AAAA,IACzC,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,OAAO,KAAK,OAAO,UAAU;AAAA,MACjC,MAAM,KAAK,gBAAgB,OAAO;AAAA,MAClC,UAAU,SAAS;AAAA,MACnB,QAAQ,KAAK,kBAAkB,OAAO;AAAA,IACxC,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,MAAM,YAAY,SAAS,SAAS,SAAS;AAAA,IAE7C,IAAI,OAAO,cAAc,UAAU;AAAA,MACjC,OAAO;AAAA,IACT;AAAA,IAEA,IAAI,aAAa,OAAO,cAAc,UAAU;AAAA,MAC9C,MAAM,cAAc;AAAA,MACpB,IACE,MAAM,QAAQ,YAAY,EAAE,KAC5B,YAAY,GAAG,MAAM,CAAC,SAAS,OAAO,SAAS,QAAQ,GACvD;AAAA,QACA,OAAQ,YAAY,GAAgB,KAAK,GAAG;AAAA,MAC9C;AAAA,MAEA,IAAI,OAAO,YAAY,cAAc,UAAU;AAAA,QAC7C,OAAO,YAAY;AAAA,MACrB;AAAA,MAEA,IAAI,OAAO,YAAY,SAAS,UAAU;AAAA,QACxC,OAAO,YAAY;AAAA,MACrB;AAAA,IACF;AAAA,IAEA,OAAO;AAAA;AAAA,EAOD,kBAAkB,CAAC,SAAuC;AAAA,IAChE,MAAM,WAAW,SAAS,YAAY,CAAC;AAAA,IACvC,OAAO,EAAE,UAAU,eAAe,SAAS,MAAM,EAAE,CAAC,EAAE;AAAA;AAAA,EAOhD,mBAAmB,CAAC,UAAwC;AAAA,IAClE,MAAM,UAAU,UAAU,WAAW,UAAU,QAAQ;AAAA,IACvD,OAAO,EAAE,SAAS,eAAe,OAAO,EAAE;AAAA;AAAA,EAOpC,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;;;ADhNA,IAAM,SAAS,UAAU,MAAM;AAE/B,IAAM,iCAAiC;AAAA,EACrC,qBAAqB;AAAA,EACrB,cAAc;AAAA,IACZ,OAAO,EAAE,MAAM,UAAU,sBAAsB,KAAK;AAAA,IACpD,KAAK,EAAE,MAAM,UAAU,sBAAsB,KAAK;AAAA,IAClD,MAAM,EAAE,MAAM,UAAU,sBAAsB,KAAK;AAAA,IACnD,OAAO,EAAE,MAAM,UAAU,sBAAsB,KAAK;AAAA,EACtD;AACF;AAEA,IAAI,aAAiC;AACrC,IAAI,eAA8B;AAClC,IAAI,mBAA2C;AAmDxC,SAAS,IAAI,CAAC,QAA2C;AAAA,EAC9D,iBAAiB;AAAA,EAEjB,IAAI,iBAAiB;AAAA,EACrB,MAAM,gBAAgB,QAAQ,iBAAiB,QAAQ,IAAI,uBAAuB;AAAA,EAElF,IAAI,kBAAkB,UAAU,CAAC,QAAQ,YAAY;AAAA,IACnD,MAAM,SAAS,QAAQ,IAAI;AAAA,IAC3B,MAAM,WAAW,QAAQ,IAAI;AAAA,IAE7B,IAAI,CAAC,UAAU,CAAC,UAAU;AAAA,MACxB,MAAM,IAAI,MACR,8FACE,+CACJ;AAAA,IACF;AAAA,IAEA,iBAAiB;AAAA,SACZ;AAAA,MACH,eAAe;AAAA,MACf,YAAY;AAAA,QACV;AAAA,QACA;AAAA,QACA,SAAS,QAAQ,IAAI;AAAA,QACrB,WAAW,QAAQ,IAAI;AAAA,QACvB,iBAAiB,QAAQ,IAAI,8BAA8B;AAAA,QAC3D,aAAa;AAAA,MACf;AAAA,IACF;AAAA,EACF,EAAO,SAAI,kBAAkB,UAAU,QAAQ,cAAc,CAAC,OAAO,WAAW,aAAa;AAAA,IAC3F,iBAAiB;AAAA,SACZ;AAAA,MACH,YAAY;AAAA,WACP,OAAO;AAAA,QACV,aAAa;AAAA,MACf;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,cAAc,aAAa,cAAc;AAAA,EAE/C,IAAI,qBAAqB,MAAM;AAAA,IAC7B,OAAO;AAAA,EACT;AAAA,EAEA,MAAM,OAAO,WAAW,WAAW;AAAA,EACnC,aAAa;AAAA,EACb,eAAe,KAAK;AAAA,EAEpB,MAAM,aAAa,YAAY;AAAA,EAC/B,IAAI,YAAY,aAAa;AAAA,IAC3B,KAAK,aAAa,eAAe,WAAW,WAAW;AAAA,EACzD;AAAA,EAEA,MAAM,YAAY,YAAY,aAC1B;AAAA,IACE,SAAS,YAAY,WAAW;AAAA,IAChC,iBAAiB,YAAY,WAAW;AAAA,IACxC,WAAW,YAAY,WAAW;AAAA,IAClC,kBAAkB,YAAY,WAAW;AAAA,EAC3C,IACA;AAAA,EAEJ,MAAM,sBAAsB,IAAI,oBAAoB,KAAK,QAAQ,KAAK,cAAc,SAAS;AAAA,EAE7F,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;AAGT,eAAsB,QAAW,CAC/B,SACA,IACY;AAAA,EACZ,MAAM,SAAS,UAAU;AAAA,EACzB,MAAM,OAAO,OAAO,UAAU,OAAO;AAAA,EAErC,IAAI;AAAA,IACF,MAAM,SAAS,MAAM,aAAY,SAAS,MAAM,YAAY,MAAM,GAAG,CAAC;AAAA,IACtE,OAAO,QAAQ,IAAI;AAAA,IACnB,OAAO;AAAA,IACP,OAAO,OAAO;AAAA,IACd,MAAM,kBAAkB,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC;AAAA,IAChF,OAAO,QAAQ,MAAM,EAAE,OAAO,gBAAgB,CAAC;AAAA,IAC/C,MAAM;AAAA;AAAA;AAsBV,eAAsB,QAAQ,GAAkB;AAAA,EAC9C,IAAI,YAAY;AAAA,IACd,OAAO,KAAK,6BAA6B;AAAA,IACzC,MAAM,WAAW,SAAS;AAAA,EAC5B;AAAA,EACA,aAAa;AAAA,EACb,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": "EBAE04F79C70EA8764756E2164756E21",
   "names": []
 }

package/dist/init.d.ts

@@ -1,5 +1,11 @@
-import { type Config, Tracer } from '@prefactor/core';
+import { type Config, type Tracer } from '@prefactor/core';
 import { type AgentMiddleware } from 'langchain';
+export type ManualSpanOptions = {
+    name: string;
+    spanType: string;
+    inputs: Record<string, unknown>;
+    metadata?: Record<string, unknown>;
+};
 /**
  * Initialize the Prefactor SDK and return middleware for LangChain.js
  *
@@ -14,8 +20,15 @@ import { type AgentMiddleware } from 'langchain';
  * import { init } from '@prefactor/langchain';
  * import { createAgent } from 'langchain';
  *
- * // Initialize with defaults (stdio transport)
- * const middleware = init();
+ * // Initialize with HTTP transport
+ * const middleware = init({
+ *   transportType: 'http',
+ *   httpConfig: {
+ *     apiUrl: 'https://api.prefactor.ai',
+ *     apiToken: process.env.PREFACTOR_API_TOKEN!,
+ *     agentIdentifier: 'my-langchain-agent',
+ *   },
+ * });
  *
  * // Or configure HTTP transport
  * const middleware = init({
@@ -23,6 +36,8 @@ import { type AgentMiddleware } from 'langchain';
  *   httpConfig: {
  *     apiUrl: 'https://api.prefactor.ai',
  *     apiToken: process.env.PREFACTOR_API_TOKEN!,
+ *     agentIdentifier: 'my-langchain-agent', // Required
+ *     agentId: 'legacy-agent-id', // Optional legacy identifier
  *   }
  * });
  *
@@ -54,6 +69,7 @@ export declare function init(config?: Partial<Config>): AgentMiddleware;
  * ```
  */
 export declare function getTracer(): Tracer;
+export declare function withSpan<T>(options: ManualSpanOptions, fn: () => Promise<T> | T): Promise<T>;
 /**
  * Shutdown the SDK and flush any pending spans.
  *

package/dist/init.d.ts.map

@@ -1 +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"}
+{"version":3,"file":"init.d.ts","sourceRoot":"","sources":["../src/init.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,MAAM,EAOX,KAAK,MAAM,EACZ,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,KAAK,eAAe,EAAoB,MAAM,WAAW,CAAC;AAmBnE,MAAM,MAAM,iBAAiB,GAAG;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAChC,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAyCG;AACH,wBAAgB,IAAI,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,GAAG,eAAe,CAuF9D;AAED;;;;;;;;;;;;;;;;;;GAkBG;AACH,wBAAgB,SAAS,IAAI,MAAM,CAMlC;AAED,wBAAsB,QAAQ,CAAC,CAAC,EAC9B,OAAO,EAAE,iBAAiB,EAC1B,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,GACvB,OAAO,CAAC,CAAC,CAAC,CAaZ;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAsB,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC,CAQ9C"}

package/dist/init.js

@@ -1,8 +1,17 @@
-import { configureLogging, createConfig, getLogger, HttpTransport, HttpTransportConfigSchema, StdioTransport, Tracer, } from '@prefactor/core';
-import { extractPartition } from '@prefactor/pfid';
+import { configureLogging, createConfig, createCore, getLogger, SpanContext, } from '@prefactor/core';
 import { createMiddleware } from 'langchain';
 import { PrefactorMiddleware } from './middleware.js';
 const logger = getLogger('init');
+const DEFAULT_LANGCHAIN_AGENT_SCHEMA = {
+    external_identifier: 'prefactor',
+    span_schemas: {
+        agent: { type: 'object', additionalProperties: true },
+        llm: { type: 'object', additionalProperties: true },
+        tool: { type: 'object', additionalProperties: true },
+        chain: { type: 'object', additionalProperties: true },
+    },
+};
+let globalCore = null;
 let globalTracer = null;
 let globalMiddleware = null;
 /**
@@ -19,8 +28,15 @@ let globalMiddleware = null;
  * import { init } from '@prefactor/langchain';
  * import { createAgent } from 'langchain';
  *
- * // Initialize with defaults (stdio transport)
- * const middleware = init();
+ * // Initialize with HTTP transport
+ * const middleware = init({
+ *   transportType: 'http',
+ *   httpConfig: {
+ *     apiUrl: 'https://api.prefactor.ai',
+ *     apiToken: process.env.PREFACTOR_API_TOKEN!,
+ *     agentIdentifier: 'my-langchain-agent',
+ *   },
+ * });
  *
  * // Or configure HTTP transport
  * const middleware = init({
@@ -28,6 +44,8 @@ let globalMiddleware = null;
  *   httpConfig: {
  *     apiUrl: 'https://api.prefactor.ai',
  *     apiToken: process.env.PREFACTOR_API_TOKEN!,
+ *     agentIdentifier: 'my-langchain-agent', // Required
+ *     agentId: 'legacy-agent-id', // Optional legacy identifier
  *   }
  * });
  *
@@ -40,36 +58,57 @@ let globalMiddleware = null;
  */
 export function init(config) {
     configureLogging();
-    const finalConfig = createConfig(config);
-    logger.info('Initializing Prefactor SDK', { transport: finalConfig.transportType });
+    let configWithHttp = config;
+    const transportType = config?.transportType ?? process.env.PREFACTOR_TRANSPORT ?? 'http';
+    if (transportType === 'http' && !config?.httpConfig) {
+        const apiUrl = process.env.PREFACTOR_API_URL;
+        const apiToken = process.env.PREFACTOR_API_TOKEN;
+        if (!apiUrl || !apiToken) {
+            throw new Error('HTTP transport requires PREFACTOR_API_URL and PREFACTOR_API_TOKEN environment variables, ' +
+                'or httpConfig to be provided in configuration');
+        }
+        configWithHttp = {
+            ...config,
+            transportType: 'http',
+            httpConfig: {
+                apiUrl,
+                apiToken,
+                agentId: process.env.PREFACTOR_AGENT_ID,
+                agentName: process.env.PREFACTOR_AGENT_NAME,
+                agentIdentifier: process.env.PREFACTOR_AGENT_IDENTIFIER || '1.0.0',
+                agentSchema: DEFAULT_LANGCHAIN_AGENT_SCHEMA,
+            },
+        };
+    }
+    else if (transportType === 'http' && config?.httpConfig && !config.httpConfig.agentSchema) {
+        configWithHttp = {
+            ...config,
+            httpConfig: {
+                ...config.httpConfig,
+                agentSchema: DEFAULT_LANGCHAIN_AGENT_SCHEMA,
+            },
+        };
+    }
+    const finalConfig = createConfig(configWithHttp);
     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);
+    const core = createCore(finalConfig);
+    globalCore = core;
+    globalTracer = core.tracer;
+    const httpConfig = finalConfig.httpConfig;
+    if (httpConfig?.agentSchema) {
+        core.agentManager.registerSchema(httpConfig.agentSchema);
     }
-    // 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 });
+    const agentInfo = finalConfig.httpConfig
+        ? {
+            agentId: finalConfig.httpConfig.agentId,
+            agentIdentifier: finalConfig.httpConfig.agentIdentifier,
+            agentName: finalConfig.httpConfig.agentName,
+            agentDescription: finalConfig.httpConfig.agentDescription,
         }
-    }
-    globalTracer = new Tracer(transport, partition);
-    const prefactorMiddleware = new PrefactorMiddleware(globalTracer);
+        : undefined;
+    const prefactorMiddleware = new PrefactorMiddleware(core.tracer, core.agentManager, agentInfo);
     const middleware = createMiddleware({
         name: 'prefactor',
         // biome-ignore lint/suspicious/noExplicitAny: LangChain middleware hooks use dynamic types
@@ -118,6 +157,20 @@ export function getTracer() {
     // Safe because init() always sets globalTracer
     return globalTracer;
 }
+export async function withSpan(options, fn) {
+    const tracer = getTracer();
+    const span = tracer.startSpan(options);
+    try {
+        const result = await SpanContext.runAsync(span, async () => await fn());
+        tracer.endSpan(span);
+        return result;
+    }
+    catch (error) {
+        const normalizedError = error instanceof Error ? error : new Error(String(error));
+        tracer.endSpan(span, { error: normalizedError });
+        throw error;
+    }
+}
 /**
  * Shutdown the SDK and flush any pending spans.
  *
@@ -137,10 +190,11 @@ export function getTracer() {
  * ```
  */
 export async function shutdown() {
-    if (globalTracer) {
+    if (globalCore) {
         logger.info('Shutting down Prefactor SDK');
-        await globalTracer.close();
+        await globalCore.shutdown();
     }
+    globalCore = null;
     globalTracer = null;
     globalMiddleware = null;
 }

package/dist/init.js.map

@@ -1 +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"}
+{"version":3,"file":"init.js","sourceRoot":"","sources":["../src/init.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,gBAAgB,EAChB,YAAY,EACZ,UAAU,EACV,SAAS,EACT,WAAW,GAEZ,MAAM,iBAAiB,CAAC;AACzB,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,MAAM,8BAA8B,GAAG;IACrC,mBAAmB,EAAE,WAAW;IAChC,YAAY,EAAE;QACZ,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,oBAAoB,EAAE,IAAI,EAAE;QACrD,GAAG,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,oBAAoB,EAAE,IAAI,EAAE;QACnD,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,oBAAoB,EAAE,IAAI,EAAE;QACpD,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,oBAAoB,EAAE,IAAI,EAAE;KACtD;CACO,CAAC;AAEX,IAAI,UAAU,GAAuB,IAAI,CAAC;AAC1C,IAAI,YAAY,GAAkB,IAAI,CAAC;AACvC,IAAI,gBAAgB,GAA2B,IAAI,CAAC;AASpD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAyCG;AACH,MAAM,UAAU,IAAI,CAAC,MAAwB;IAC3C,gBAAgB,EAAE,CAAC;IAEnB,IAAI,cAAc,GAAG,MAAM,CAAC;IAC5B,MAAM,aAAa,GAAG,MAAM,EAAE,aAAa,IAAI,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,MAAM,CAAC;IAEzF,IAAI,aAAa,KAAK,MAAM,IAAI,CAAC,MAAM,EAAE,UAAU,EAAE,CAAC;QACpD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;QAC7C,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC;QAEjD,IAAI,CAAC,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACzB,MAAM,IAAI,KAAK,CACb,2FAA2F;gBACzF,+CAA+C,CAClD,CAAC;QACJ,CAAC;QAED,cAAc,GAAG;YACf,GAAG,MAAM;YACT,aAAa,EAAE,MAAM;YACrB,UAAU,EAAE;gBACV,MAAM;gBACN,QAAQ;gBACR,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,kBAAkB;gBACvC,SAAS,EAAE,OAAO,CAAC,GAAG,CAAC,oBAAoB;gBAC3C,eAAe,EAAE,OAAO,CAAC,GAAG,CAAC,0BAA0B,IAAI,OAAO;gBAClE,WAAW,EAAE,8BAA8B;aAC5C;SACF,CAAC;IACJ,CAAC;SAAM,IAAI,aAAa,KAAK,MAAM,IAAI,MAAM,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC;QAC5F,cAAc,GAAG;YACf,GAAG,MAAM;YACT,UAAU,EAAE;gBACV,GAAG,MAAM,CAAC,UAAU;gBACpB,WAAW,EAAE,8BAA8B;aAC5C;SACF,CAAC;IACJ,CAAC;IAED,MAAM,WAAW,GAAG,YAAY,CAAC,cAAc,CAAC,CAAC;IAEjD,IAAI,gBAAgB,KAAK,IAAI,EAAE,CAAC;QAC9B,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAED,MAAM,IAAI,GAAG,UAAU,CAAC,WAAW,CAAC,CAAC;IACrC,UAAU,GAAG,IAAI,CAAC;IAClB,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC;IAE3B,MAAM,UAAU,GAAG,WAAW,CAAC,UAAU,CAAC;IAC1C,IAAI,UAAU,EAAE,WAAW,EAAE,CAAC;QAC5B,IAAI,CAAC,YAAY,CAAC,cAAc,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;IAC3D,CAAC;IAED,MAAM,SAAS,GAAG,WAAW,CAAC,UAAU;QACtC,CAAC,CAAC;YACE,OAAO,EAAE,WAAW,CAAC,UAAU,CAAC,OAAO;YACvC,eAAe,EAAE,WAAW,CAAC,UAAU,CAAC,eAAe;YACvD,SAAS,EAAE,WAAW,CAAC,UAAU,CAAC,SAAS;YAC3C,gBAAgB,EAAE,WAAW,CAAC,UAAU,CAAC,gBAAgB;SAC1D;QACH,CAAC,CAAC,SAAS,CAAC;IAEd,MAAM,mBAAmB,GAAG,IAAI,mBAAmB,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;IAE/F,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,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC5B,OAA0B,EAC1B,EAAwB;IAExB,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAEvC,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,QAAQ,CAAC,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QACxE,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACrB,OAAO,MAAM,CAAC;IAChB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,eAAe,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QAClF,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC,CAAC;QACjD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,CAAC,KAAK,UAAU,QAAQ;IAC5B,IAAI,UAAU,EAAE,CAAC;QACf,MAAM,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;QAC3C,MAAM,UAAU,CAAC,QAAQ,EAAE,CAAC;IAC9B,CAAC;IACD,UAAU,GAAG,IAAI,CAAC;IAClB,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/middleware.d.ts

@@ -1,4 +1,4 @@
-import { type Tracer } from '@prefactor/core';
+import { type AgentInstanceManager, type Tracer } from '@prefactor/core';
 /**
  * Prefactor middleware for LangChain.js agents.
  *
@@ -26,8 +26,10 @@ import { type Tracer } from '@prefactor/core';
  */
 export declare class PrefactorMiddleware {
     private tracer;
+    private agentManager;
+    private agentInfo?;
     private rootSpan;
-    constructor(tracer: Tracer);
+    constructor(tracer: Tracer, agentManager: AgentInstanceManager, agentInfo?: Parameters<AgentInstanceManager['startInstance']>[0]);
     /**
      * Called before agent execution starts
      *

package/dist/middleware.d.ts.map

@@ -1 +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"}
+{"version":3,"file":"middleware.d.ts","sourceRoot":"","sources":["../src/middleware.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,oBAAoB,EAIzB,KAAK,MAAM,EACZ,MAAM,iBAAiB,CAAC;AAGzB;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,qBAAa,mBAAmB;IAI5B,OAAO,CAAC,MAAM;IACd,OAAO,CAAC,YAAY;IACpB,OAAO,CAAC,SAAS,CAAC;IALpB,OAAO,CAAC,QAAQ,CAAgD;gBAGtD,MAAM,EAAE,MAAM,EACd,YAAY,EAAE,oBAAoB,EAClC,SAAS,CAAC,EAAE,UAAU,CAAC,oBAAoB,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;IAG1E;;;;OAIG;IAEG,WAAW,CAAC,KAAK,EAAE,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC;IAe5C;;;;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;IAwBnF;;;;;;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;IAuBlF;;OAEG;IAEH,OAAO,CAAC,gBAAgB;IA4BxB;;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

@@ -1,4 +1,4 @@
-import { SpanContext, SpanType } from '@prefactor/core';
+import { SpanContext, SpanType, serializeValue, } from '@prefactor/core';
 import { extractTokenUsage } from './metadata-extractor.js';
 /**
  * Prefactor middleware for LangChain.js agents.
@@ -27,9 +27,13 @@ import { extractTokenUsage } from './metadata-extractor.js';
  */
 export class PrefactorMiddleware {
     tracer;
+    agentManager;
+    agentInfo;
     rootSpan = null;
-    constructor(tracer) {
+    constructor(tracer, agentManager, agentInfo) {
         this.tracer = tracer;
+        this.agentManager = agentManager;
+        this.agentInfo = agentInfo;
     }
     /**
      * Called before agent execution starts
@@ -38,17 +42,15 @@ export class PrefactorMiddleware {
      */
     // 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();
+        this.agentManager.startInstance(this.agentInfo);
         const span = this.tracer.startSpan({
-            name: 'agent',
+            name: 'langchain:agent',
             spanType: SpanType.AGENT,
-            inputs: { messages: messages.slice(-3).map((m) => String(m)) },
-            parentSpanId: parentSpan?.spanId,
-            traceId: parentSpan?.traceId,
+            inputs: { messages: serializeValue(messages.slice(-3)) },
         });
         this.rootSpan = span;
+        SpanContext.enter(span);
     }
     /**
      * Called after agent execution completes
@@ -62,10 +64,10 @@ export class PrefactorMiddleware {
         }
         const messages = state?.messages ?? [];
         this.tracer.endSpan(this.rootSpan, {
-            outputs: { messages: messages.slice(-3).map((m) => String(m)) },
+            outputs: { messages: serializeValue(messages.slice(-3)) },
         });
-        this.tracer.finishAgentInstance();
-        SpanContext.clear();
+        this.agentManager.finishInstance();
+        SpanContext.exit();
         this.rootSpan = null;
     }
     /**
@@ -77,13 +79,10 @@ export class PrefactorMiddleware {
      */
     // 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
@@ -109,13 +108,10 @@ export class PrefactorMiddleware {
      */
     // 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
@@ -137,7 +133,24 @@ export class PrefactorMiddleware {
      */
     // biome-ignore lint/suspicious/noExplicitAny: LangChain request structure is dynamic
     extractModelName(request) {
-        return request?.model ?? request?.modelName ?? 'unknown';
+        const candidate = request?.model ?? request?.modelName;
+        if (typeof candidate === 'string') {
+            return candidate;
+        }
+        if (candidate && typeof candidate === 'object') {
+            const modelObject = candidate;
+            if (Array.isArray(modelObject.id) &&
+                modelObject.id.every((item) => typeof item === 'string')) {
+                return modelObject.id.join('.');
+            }
+            if (typeof modelObject.modelName === 'string') {
+                return modelObject.modelName;
+            }
+            if (typeof modelObject.name === 'string') {
+                return modelObject.name;
+            }
+        }
+        return 'unknown';
     }
     /**
      * Extract model inputs from request
@@ -145,7 +158,7 @@ export class PrefactorMiddleware {
     // 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)) };
+        return { messages: serializeValue(messages.slice(-3)) };
     }
     /**
      * Extract model outputs from response
@@ -153,7 +166,7 @@ export class PrefactorMiddleware {
     // biome-ignore lint/suspicious/noExplicitAny: LangChain response structure is dynamic
     extractModelOutputs(response) {
         const content = response?.content ?? response?.text ?? '';
-        return { content: String(content) };
+        return { content: serializeValue(content) };
     }
     /**
      * Extract tool name from request

package/dist/middleware.js.map

@@ -1 +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"}
+{"version":3,"file":"middleware.js","sourceRoot":"","sources":["../src/middleware.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,WAAW,EACX,QAAQ,EACR,cAAc,GAEf,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAE5D;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,MAAM,OAAO,mBAAmB;IAIpB;IACA;IACA;IALF,QAAQ,GAA2C,IAAI,CAAC;IAEhE,YACU,MAAc,EACd,YAAkC,EAClC,SAAgE;QAFhE,WAAM,GAAN,MAAM,CAAQ;QACd,iBAAY,GAAZ,YAAY,CAAsB;QAClC,cAAS,GAAT,SAAS,CAAuD;IACvE,CAAC;IAEJ;;;;OAIG;IACH,mFAAmF;IACnF,KAAK,CAAC,WAAW,CAAC,KAAU;QAC1B,MAAM,QAAQ,GAAG,KAAK,EAAE,QAAQ,IAAI,EAAE,CAAC;QAEvC,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAEhD,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;YACjC,IAAI,EAAE,iBAAiB;YACvB,QAAQ,EAAE,QAAQ,CAAC,KAAK;YACxB,MAAM,EAAE,EAAE,QAAQ,EAAE,cAAc,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE;SACzD,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC1B,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,cAAc,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE;SAC1D,CAAC,CAAC;QAEH,IAAI,CAAC,YAAY,CAAC,cAAc,EAAE,CAAC;QACnC,WAAW,CAAC,IAAI,EAAE,CAAC;QACnB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;IACvB,CAAC;IAED;;;;;;OAMG;IACH,0FAA0F;IAC1F,KAAK,CAAC,aAAa,CAAI,OAAY,EAAE,OAAiC;QACpE,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;SACzC,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,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;SACxC,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,MAAM,SAAS,GAAG,OAAO,EAAE,KAAK,IAAI,OAAO,EAAE,SAAS,CAAC;QAEvD,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;YAClC,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,IAAI,SAAS,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;YAC/C,MAAM,WAAW,GAAG,SAAoC,CAAC;YACzD,IACE,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;gBAC7B,WAAW,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC,EACxD,CAAC;gBACD,OAAQ,WAAW,CAAC,EAAe,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAChD,CAAC;YAED,IAAI,OAAO,WAAW,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;gBAC9C,OAAO,WAAW,CAAC,SAAS,CAAC;YAC/B,CAAC;YAED,IAAI,OAAO,WAAW,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACzC,OAAO,WAAW,CAAC,IAAI,CAAC;YAC1B,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,qFAAqF;IAC7E,kBAAkB,CAAC,OAAY;QACrC,MAAM,QAAQ,GAAG,OAAO,EAAE,QAAQ,IAAI,EAAE,CAAC;QACzC,OAAO,EAAE,QAAQ,EAAE,cAAc,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC1D,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,cAAc,CAAC,OAAO,CAAC,EAAE,CAAC;IAC9C,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

@@ -1,6 +1,6 @@
 {
   "name": "@prefactor/langchain",
-  "version": "0.1.1",
+  "version": "0.2.1",
   "description": "LangChain.js integration for Prefactor observability",
   "type": "module",
   "main": "dist/index.cjs",
@@ -28,7 +28,7 @@
   "author": "Prefactor",
   "license": "MIT",
   "dependencies": {
-    "@prefactor/core": "0.1.0",
+    "@prefactor/core": "0.1.1",
     "@prefactor/pfid": "^0.1.0"
   },
   "peerDependencies": {