@turingpulse/sdk-langchain 1.0.1

package/package.json

@@ -0,0 +1,16 @@
+{
+  "name": "@turingpulse/sdk-langchain",
+  "version": "1.0.1",
+  "license": "SEE LICENSE IN LICENSE",
+  "description": "TuringPulse SDK integration for LangChain.js",
+  "type": "module",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "dependencies": {
+    "@turingpulse/sdk": ">=1.0.0",
+    "@langchain/core": ">=1.1.24"
+  },
+  "peerDependencies": {
+    "typescript": ">=5.0.0"
+  }
+}

package/src/index.ts

@@ -0,0 +1,7 @@
+/**
+ * TuringPulse SDK integration for LangChain.js.
+ *
+ * Wraps LangChain Runnables and Chains for observability.
+ */
+
+export { instrumentLangchain } from './wrapper';

package/src/wrapper.ts

@@ -0,0 +1,51 @@
+import { withInstrumentation, GovernanceDirective, currentContext } from '@turingpulse/sdk';
+
+interface LangchainInstrumentOptions {
+  name: string;
+  governance?: GovernanceDirective;
+  model?: string;
+  provider?: string;
+}
+
+/**
+ * Wrap a LangChain Runnable for TuringPulse observability.
+ */
+export function instrumentLangchain(
+  runnable: { invoke: (input: unknown, config?: unknown) => Promise<unknown> },
+  options: LangchainInstrumentOptions,
+): (input: unknown, config?: unknown) => Promise<unknown> {
+  const { name, governance, model, provider = 'openai' } = options;
+
+  const wrappedFn = withInstrumentation(
+    async function langchainInvoke(input: unknown, config?: unknown): Promise<unknown> {
+      const result = await runnable.invoke(input, config);
+
+      const ctx = currentContext();
+      if (ctx) {
+        ctx.framework = 'langchain';
+        ctx.nodeType = 'llm';
+        if (model) ctx.setModel(model, provider);
+
+        const resp = result as Record<string, unknown> | null;
+        const content = typeof resp?.content === 'string' ? resp.content : '';
+
+        const responseMeta = resp?.response_metadata as Record<string, unknown> | undefined;
+        const tokenUsage = responseMeta?.tokenUsage as { promptTokens?: number; completionTokens?: number } | undefined;
+        if (tokenUsage) {
+          ctx.setTokens(tokenUsage.promptTokens ?? 0, tokenUsage.completionTokens ?? 0);
+        }
+
+        const detectedModel = (responseMeta?.model_name as string) ?? model;
+        if (detectedModel) ctx.setModel(detectedModel, provider);
+
+        const inputStr = typeof input === 'string' ? input : JSON.stringify(input);
+        ctx.setIO(inputStr, content || JSON.stringify(result));
+      }
+
+      return result;
+    },
+    { name, governance },
+  );
+
+  return wrappedFn as (input: unknown, config?: unknown) => Promise<unknown>;
+}