@turingpulse/sdk-vercel-ai 1.0.1

package/package.json

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

package/src/index.ts

@@ -0,0 +1,5 @@
+/**
+ * TuringPulse SDK integration for Vercel AI SDK.
+ */
+
+export { instrumentVercelAI } from './wrapper';

package/src/wrapper.ts

@@ -0,0 +1,49 @@
+import { withInstrumentation, GovernanceDirective, currentContext } from '@turingpulse/sdk';
+
+interface VercelAIInstrumentOptions {
+  name: string;
+  governance?: GovernanceDirective;
+  model?: string;
+  provider?: string;
+}
+
+/**
+ * Wrap Vercel AI SDK's generateText/streamText for TuringPulse observability.
+ */
+export function instrumentVercelAI(
+  generateFn: (...args: unknown[]) => Promise<unknown>,
+  options: VercelAIInstrumentOptions,
+): (...args: unknown[]) => Promise<unknown> {
+  const { name, governance, model, provider = 'openai' } = options;
+
+  const wrappedFn = withInstrumentation(
+    async function vercelAIGenerate(...args: unknown[]): Promise<unknown> {
+      const result = await generateFn(...args);
+
+      const ctx = currentContext();
+      if (ctx) {
+        ctx.framework = 'vercel-ai';
+        ctx.nodeType = 'llm';
+
+        const resp = result as Record<string, unknown>;
+        const usage = resp?.usage as { promptTokens?: number; completionTokens?: number } | undefined;
+        if (usage) {
+          ctx.setTokens(usage.promptTokens ?? 0, usage.completionTokens ?? 0);
+        }
+        if (model) {
+          ctx.setModel(model, provider);
+        }
+
+        const firstArg = args[0] as Record<string, unknown> | undefined;
+        const inputStr = (firstArg?.prompt as string) ?? JSON.stringify(firstArg);
+        const outputStr = (resp?.text as string) ?? JSON.stringify(result);
+        ctx.setIO(inputStr, outputStr);
+      }
+
+      return result;
+    },
+    { name, governance },
+  );
+
+  return wrappedFn as (...args: unknown[]) => Promise<unknown>;
+}