@tenet-ai/sdk 0.1.0

package/dist/integrations/openai.d.ts

@@ -0,0 +1,166 @@
+/**
+ * OpenAI Node SDK integration for Tenet AI.
+ *
+ * Provides a wrapper around `openai.chat.completions.create()` that
+ * automatically records prompts, responses, model versions, and tool calls
+ * in the Tenet audit trail, with replay support.
+ *
+ * This module has **no hard dependency** on the `openai` package. It defines
+ * the relevant OpenAI types inline and works with any object that matches
+ * the expected shapes.
+ *
+ * @example
+ * ```ts
+ * import OpenAI from 'openai';
+ * import { Tenet } from '@tenet-ai/sdk';
+ * import { TenetOpenAITracker } from '@tenet-ai/sdk/integrations/openai';
+ *
+ * const openai = new OpenAI();
+ * const tenet = new Tenet({ apiKey: 'tenet_xxx' });
+ * const tracker = new TenetOpenAITracker(tenet, {
+ *   agentId: 'openai-chatbot',
+ *   tags: ['production'],
+ * });
+ *
+ * // Tracked call -- automatically logs to Tenet
+ * const response = await tracker.chatCompletion(openai, {
+ *   model: 'gpt-4o',
+ *   messages: [{ role: 'user', content: 'Hello!' }],
+ * });
+ *
+ * // Or wrap an existing client for automatic tracking
+ * const wrappedOpenAI = tracker.wrap(openai);
+ * const res = await wrappedOpenAI.chat.completions.create({
+ *   model: 'gpt-4o',
+ *   messages: [{ role: 'user', content: 'Hello!' }],
+ * });
+ * ```
+ *
+ * @module
+ */
+import { Tenet } from "../client";
+/** A chat message in OpenAI format. */
+export interface ChatMessage {
+    role: "system" | "user" | "assistant" | "tool" | "function" | string;
+    content?: string | null;
+    name?: string;
+    tool_calls?: ToolCall[];
+    tool_call_id?: string;
+    [key: string]: unknown;
+}
+/** A tool call returned in an assistant message. */
+export interface ToolCall {
+    id: string;
+    type: "function";
+    function: {
+        name: string;
+        arguments: string;
+    };
+}
+/** Parameters for `chat.completions.create()`. */
+export interface ChatCompletionCreateParams {
+    model: string;
+    messages: ChatMessage[];
+    temperature?: number;
+    max_tokens?: number;
+    tools?: unknown[];
+    tool_choice?: unknown;
+    [key: string]: unknown;
+}
+/** A single choice in a chat completion response. */
+export interface ChatCompletionChoice {
+    index: number;
+    message: ChatMessage;
+    finish_reason: string | null;
+}
+/** The response from `chat.completions.create()`. */
+export interface ChatCompletionResponse {
+    id: string;
+    object: string;
+    model: string;
+    choices: ChatCompletionChoice[];
+    usage?: {
+        prompt_tokens: number;
+        completion_tokens: number;
+        total_tokens: number;
+    };
+    [key: string]: unknown;
+}
+/**
+ * Minimal interface for an OpenAI client instance.
+ * Matches the shape of `new OpenAI()` from the `openai` package.
+ */
+export interface OpenAIClientLike {
+    chat: {
+        completions: {
+            create(params: ChatCompletionCreateParams): Promise<ChatCompletionResponse>;
+        };
+    };
+}
+/** Configuration options for {@link TenetOpenAITracker}. */
+export interface TenetOpenAITrackerConfig {
+    /** Identifier for the agent. Defaults to `"openai-chat"`. */
+    agentId?: string;
+    /** Explicit session ID. A random UUID is generated per tracker if omitted. */
+    sessionId?: string;
+    /** Tags attached to every decision. */
+    tags?: string[];
+    /** Arbitrary metadata attached to every decision. */
+    metadata?: Record<string, unknown>;
+}
+/**
+ * Wrapper for the OpenAI Node SDK that records `chat.completions.create()`
+ * calls in the Tenet audit trail.
+ *
+ * Each call records the full intent -> context -> decision -> execution
+ * chain including the prompt messages, model version, chosen response,
+ * tool calls, and token usage. The recorded decision includes a replay
+ * prompt so it can be replayed later via `tenet.replay()`.
+ */
+export declare class TenetOpenAITracker {
+    private client;
+    private agentId;
+    private sessionId;
+    private tags;
+    private metadata;
+    /**
+     * @param client  - An initialised {@link Tenet} client instance.
+     * @param config  - Optional configuration overrides.
+     */
+    constructor(client: Tenet, config?: TenetOpenAITrackerConfig);
+    /**
+     * Execute a `chat.completions.create()` call and record it in Tenet.
+     *
+     * @param openai  - An OpenAI client instance (or anything with the same shape).
+     * @param params  - The parameters to pass to `chat.completions.create()`.
+     * @returns       The original OpenAI `ChatCompletionResponse`.
+     *
+     * @example
+     * ```ts
+     * const response = await tracker.chatCompletion(openai, {
+     *   model: 'gpt-4o',
+     *   messages: [{ role: 'user', content: 'Summarise this doc.' }],
+     * });
+     * ```
+     */
+    chatCompletion(openai: OpenAIClientLike, params: ChatCompletionCreateParams): Promise<ChatCompletionResponse>;
+    /**
+     * Return a proxied OpenAI client whose `chat.completions.create()` calls
+     * are automatically tracked. All other methods pass through untouched.
+     *
+     * @param openai - The original OpenAI client instance.
+     * @returns A proxied client with transparent tracking.
+     *
+     * @example
+     * ```ts
+     * const tracked = tracker.wrap(openai);
+     * // This call is automatically recorded in Tenet:
+     * const res = await tracked.chat.completions.create({ ... });
+     * ```
+     */
+    wrap<T extends OpenAIClientLike>(openai: T): T;
+    /**
+     * Record a single chat completion call in Tenet.
+     */
+    private record;
+}

package/dist/integrations/openai.js

@@ -0,0 +1,250 @@
+"use strict";
+/**
+ * OpenAI Node SDK integration for Tenet AI.
+ *
+ * Provides a wrapper around `openai.chat.completions.create()` that
+ * automatically records prompts, responses, model versions, and tool calls
+ * in the Tenet audit trail, with replay support.
+ *
+ * This module has **no hard dependency** on the `openai` package. It defines
+ * the relevant OpenAI types inline and works with any object that matches
+ * the expected shapes.
+ *
+ * @example
+ * ```ts
+ * import OpenAI from 'openai';
+ * import { Tenet } from '@tenet-ai/sdk';
+ * import { TenetOpenAITracker } from '@tenet-ai/sdk/integrations/openai';
+ *
+ * const openai = new OpenAI();
+ * const tenet = new Tenet({ apiKey: 'tenet_xxx' });
+ * const tracker = new TenetOpenAITracker(tenet, {
+ *   agentId: 'openai-chatbot',
+ *   tags: ['production'],
+ * });
+ *
+ * // Tracked call -- automatically logs to Tenet
+ * const response = await tracker.chatCompletion(openai, {
+ *   model: 'gpt-4o',
+ *   messages: [{ role: 'user', content: 'Hello!' }],
+ * });
+ *
+ * // Or wrap an existing client for automatic tracking
+ * const wrappedOpenAI = tracker.wrap(openai);
+ * const res = await wrappedOpenAI.chat.completions.create({
+ *   model: 'gpt-4o',
+ *   messages: [{ role: 'user', content: 'Hello!' }],
+ * });
+ * ```
+ *
+ * @module
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TenetOpenAITracker = void 0;
+const crypto_1 = require("crypto");
+const types_1 = require("../types");
+// ---------------------------------------------------------------------------
+// Tracker
+// ---------------------------------------------------------------------------
+/**
+ * Wrapper for the OpenAI Node SDK that records `chat.completions.create()`
+ * calls in the Tenet audit trail.
+ *
+ * Each call records the full intent -> context -> decision -> execution
+ * chain including the prompt messages, model version, chosen response,
+ * tool calls, and token usage. The recorded decision includes a replay
+ * prompt so it can be replayed later via `tenet.replay()`.
+ */
+class TenetOpenAITracker {
+    client;
+    agentId;
+    sessionId;
+    tags;
+    metadata;
+    /**
+     * @param client  - An initialised {@link Tenet} client instance.
+     * @param config  - Optional configuration overrides.
+     */
+    constructor(client, config = {}) {
+        this.client = client;
+        this.agentId = config.agentId ?? "openai-chat";
+        this.sessionId = config.sessionId ?? (0, crypto_1.randomUUID)();
+        this.tags = config.tags ?? [];
+        this.metadata = config.metadata ?? {};
+    }
+    /**
+     * Execute a `chat.completions.create()` call and record it in Tenet.
+     *
+     * @param openai  - An OpenAI client instance (or anything with the same shape).
+     * @param params  - The parameters to pass to `chat.completions.create()`.
+     * @returns       The original OpenAI `ChatCompletionResponse`.
+     *
+     * @example
+     * ```ts
+     * const response = await tracker.chatCompletion(openai, {
+     *   model: 'gpt-4o',
+     *   messages: [{ role: 'user', content: 'Summarise this doc.' }],
+     * });
+     * ```
+     */
+    async chatCompletion(openai, params) {
+        let response;
+        let error;
+        try {
+            response = await openai.chat.completions.create(params);
+        }
+        catch (e) {
+            error = e instanceof Error ? e : new Error(String(e));
+        }
+        await this.record(params, response, error);
+        if (error) {
+            throw error;
+        }
+        return response;
+    }
+    /**
+     * Return a proxied OpenAI client whose `chat.completions.create()` calls
+     * are automatically tracked. All other methods pass through untouched.
+     *
+     * @param openai - The original OpenAI client instance.
+     * @returns A proxied client with transparent tracking.
+     *
+     * @example
+     * ```ts
+     * const tracked = tracker.wrap(openai);
+     * // This call is automatically recorded in Tenet:
+     * const res = await tracked.chat.completions.create({ ... });
+     * ```
+     */
+    wrap(openai) {
+        const tracker = this;
+        const completionsProxy = new Proxy(openai.chat.completions, {
+            get(target, prop, receiver) {
+                if (prop === "create") {
+                    return async (params) => {
+                        return tracker.chatCompletion(openai, params);
+                    };
+                }
+                return Reflect.get(target, prop, receiver);
+            },
+        });
+        const chatProxy = new Proxy(openai.chat, {
+            get(target, prop, receiver) {
+                if (prop === "completions") {
+                    return completionsProxy;
+                }
+                return Reflect.get(target, prop, receiver);
+            },
+        });
+        return new Proxy(openai, {
+            get(target, prop, receiver) {
+                if (prop === "chat") {
+                    return chatProxy;
+                }
+                return Reflect.get(target, prop, receiver);
+            },
+        });
+    }
+    // ----- Internal -----
+    /**
+     * Record a single chat completion call in Tenet.
+     */
+    async record(params, response, error) {
+        const success = !error && !!response;
+        // Build a concise description of the prompt
+        const lastUserMessage = [...params.messages]
+            .reverse()
+            .find((m) => m.role === "user");
+        const goal = lastUserMessage?.content
+            ? String(lastUserMessage.content).slice(0, 200)
+            : `chat:${params.model}`;
+        // Extract the assistant reply and tool calls
+        const choice = response?.choices?.[0];
+        const assistantContent = choice?.message?.content ?? "";
+        const toolCalls = choice?.message?.tool_calls ?? [];
+        // Build options
+        const options = [];
+        if (toolCalls.length > 0) {
+            for (let i = 0; i < toolCalls.length; i++) {
+                options.push({
+                    action: toolCalls[i].function.name,
+                    score: 1.0 / (i + 1),
+                    reason: toolCalls[i].function.arguments.slice(0, 100),
+                });
+            }
+        }
+        else {
+            options.push({
+                action: "generate_response",
+                score: success ? 1.0 : 0.0,
+                reason: success ? "Direct text response" : `error: ${error?.message}`,
+            });
+        }
+        const chosenAction = toolCalls.length > 0
+            ? toolCalls[0].function.name
+            : "generate_response";
+        // Build a replay prompt from the messages
+        const replayPrompt = params.messages
+            .map((m) => `${m.role}: ${m.content ?? ""}`)
+            .join("\n")
+            .slice(0, 4000);
+        try {
+            // 1. Intent
+            const intent = await this.client.createIntent({
+                goal,
+                agentId: this.agentId,
+                sessionId: this.sessionId,
+                origin: types_1.OriginType.AGENT,
+            });
+            // 2. Context
+            await intent.snapshotContext({
+                messages: params.messages,
+                model: params.model,
+                temperature: params.temperature,
+                max_tokens: params.max_tokens,
+                tools: params.tools ?? [],
+            });
+            // 3. Decision
+            await intent.decide({
+                options,
+                chosenAction,
+                confidence: success ? 0.95 : 0.0,
+                modelVersion: response?.model ?? params.model,
+                reasoning: success
+                    ? `Model responded with ${choice?.finish_reason ?? "unknown"} finish reason`
+                    : `Error: ${error?.message}`,
+                tags: this.tags,
+                metadata: {
+                    ...this.metadata,
+                    usage: response?.usage ?? null,
+                    finish_reason: choice?.finish_reason ?? null,
+                },
+                replayPrompt,
+                replayConfig: {
+                    model: params.model,
+                    temperature: params.temperature,
+                    maxTokens: params.max_tokens,
+                },
+            });
+            // 4. Execution
+            await intent.execute({
+                action: chosenAction,
+                target: {
+                    response: assistantContent.slice(0, 2000),
+                    tool_calls: toolCalls.map((tc) => ({
+                        name: tc.function.name,
+                        arguments: tc.function.arguments,
+                    })),
+                },
+                result: success ? types_1.ResultType.SUCCESS : types_1.ResultType.FAILURE,
+                sideEffects: toolCalls.map((tc) => `tool:${tc.function.name}`),
+                actor: types_1.ActorType.AGENT,
+            });
+        }
+        catch {
+            // Swallow tracking errors so the caller is not impacted.
+        }
+    }
+}
+exports.TenetOpenAITracker = TenetOpenAITracker;
+//# sourceMappingURL=openai.js.map

package/dist/integrations/openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/integrations/openai.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAuCG;;;AAEH,mCAAoC;AAEpC,oCAA2E;AAsF3E,8EAA8E;AAC9E,UAAU;AACV,8EAA8E;AAE9E;;;;;;;;GAQG;AACH,MAAa,kBAAkB;IACrB,MAAM,CAAQ;IACd,OAAO,CAAS;IAChB,SAAS,CAAS;IAClB,IAAI,CAAW;IACf,QAAQ,CAA0B;IAE1C;;;OAGG;IACH,YAAY,MAAa,EAAE,SAAmC,EAAE;QAC9D,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,aAAa,CAAC;QAC/C,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,IAAA,mBAAU,GAAE,CAAC;QAClD,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;QAC9B,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC;IACxC,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,KAAK,CAAC,cAAc,CAClB,MAAwB,EACxB,MAAkC;QAElC,IAAI,QAA4C,CAAC;QACjD,IAAI,KAAwB,CAAC;QAE7B,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC1D,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,KAAK,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QACxD,CAAC;QAED,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;QAE3C,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,KAAK,CAAC;QACd,CAAC;QAED,OAAO,QAAS,CAAC;IACnB,CAAC;IAED;;;;;;;;;;;;;OAaG;IACH,IAAI,CAA6B,MAAS;QACxC,MAAM,OAAO,GAAG,IAAI,CAAC;QAErB,MAAM,gBAAgB,GAAG,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE;YAC1D,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ;gBACxB,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;oBACtB,OAAO,KAAK,EAAE,MAAkC,EAAE,EAAE;wBAClD,OAAO,OAAO,CAAC,cAAc,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;oBAChD,CAAC,CAAC;gBACJ,CAAC;gBACD,OAAO,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;YAC7C,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE;YACvC,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ;gBACxB,IAAI,IAAI,KAAK,aAAa,EAAE,CAAC;oBAC3B,OAAO,gBAAgB,CAAC;gBAC1B,CAAC;gBACD,OAAO,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;YAC7C,CAAC;SACF,CAAC,CAAC;QAEH,OAAO,IAAI,KAAK,CAAC,MAAM,EAAE;YACvB,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ;gBACxB,IAAI,IAAI,KAAK,MAAM,EAAE,CAAC;oBACpB,OAAO,SAAS,CAAC;gBACnB,CAAC;gBACD,OAAO,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;YAC7C,CAAC;SACF,CAAC,CAAC;IACL,CAAC;IAED,uBAAuB;IAEvB;;OAEG;IACK,KAAK,CAAC,MAAM,CAClB,MAAkC,EAClC,QAAiC,EACjC,KAAa;QAEb,MAAM,OAAO,GAAG,CAAC,KAAK,IAAI,CAAC,CAAC,QAAQ,CAAC;QAErC,4CAA4C;QAC5C,MAAM,eAAe,GAAG,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC;aACzC,OAAO,EAAE;aACT,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;QAClC,MAAM,IAAI,GAAG,eAAe,EAAE,OAAO;YACnC,CAAC,CAAC,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;YAC/C,CAAC,CAAC,QAAQ,MAAM,CAAC,KAAK,EAAE,CAAC;QAE3B,6CAA6C;QAC7C,MAAM,MAAM,GAAG,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,gBAAgB,GAAG,MAAM,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QACxD,MAAM,SAAS,GAAG,MAAM,EAAE,OAAO,EAAE,UAAU,IAAI,EAAE,CAAC;QAEpD,gBAAgB;QAChB,MAAM,OAAO,GAAmB,EAAE,CAAC;QACnC,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC1C,OAAO,CAAC,IAAI,CAAC;oBACX,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI;oBAClC,KAAK,EAAE,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;oBACpB,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;iBACtD,CAAC,CAAC;YACL,CAAC;QACH,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,IAAI,CAAC;gBACX,MAAM,EAAE,mBAAmB;gBAC3B,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG;gBAC1B,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,UAAU,KAAK,EAAE,OAAO,EAAE;aACtE,CAAC,CAAC;QACL,CAAC;QAED,MAAM,YAAY,GAAG,SAAS,CAAC,MAAM,GAAG,CAAC;YACvC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI;YAC5B,CAAC,CAAC,mBAAmB,CAAC;QAExB,0CAA0C;QAC1C,MAAM,YAAY,GAAG,MAAM,CAAC,QAAQ;aACjC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;aAC3C,IAAI,CAAC,IAAI,CAAC;aACV,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;QAElB,IAAI,CAAC;YACH,YAAY;YACZ,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;gBAC5C,IAAI;gBACJ,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,MAAM,EAAE,kBAAU,CAAC,KAAK;aACzB,CAAC,CAAC;YAEH,aAAa;YACb,MAAM,MAAM,CAAC,eAAe,CAAC;gBAC3B,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,WAAW,EAAE,MAAM,CAAC,WAAW;gBAC/B,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,EAAE;aAC1B,CAAC,CAAC;YAEH,cAAc;YACd,MAAM,MAAM,CAAC,MAAM,CAAC;gBAClB,OAAO;gBACP,YAAY;gBACZ,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG;gBAChC,YAAY,EAAE,QAAQ,EAAE,KAAK,IAAI,MAAM,CAAC,KAAK;gBAC7C,SAAS,EAAE,OAAO;oBAChB,CAAC,CAAC,wBAAwB,MAAM,EAAE,aAAa,IAAI,SAAS,gBAAgB;oBAC5E,CAAC,CAAC,UAAU,KAAK,EAAE,OAAO,EAAE;gBAC9B,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,QAAQ,EAAE;oBACR,GAAG,IAAI,CAAC,QAAQ;oBAChB,KAAK,EAAE,QAAQ,EAAE,KAAK,IAAI,IAAI;oBAC9B,aAAa,EAAE,MAAM,EAAE,aAAa,IAAI,IAAI;iBAC7C;gBACD,YAAY;gBACZ,YAAY,EAAE;oBACZ,KAAK,EAAE,MAAM,CAAC,KAAK;oBACnB,WAAW,EAAE,MAAM,CAAC,WAAW;oBAC/B,SAAS,EAAE,MAAM,CAAC,UAAU;iBAC7B;aACF,CAAC,CAAC;YAEH,eAAe;YACf,MAAM,MAAM,CAAC,OAAO,CAAC;gBACnB,MAAM,EAAE,YAAY;gBACpB,MAAM,EAAE;oBACN,QAAQ,EAAE,gBAAgB,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC;oBACzC,UAAU,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;wBACjC,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;wBACtB,SAAS,EAAE,EAAE,CAAC,QAAQ,CAAC,SAAS;qBACjC,CAAC,CAAC;iBACJ;gBACD,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,kBAAU,CAAC,OAAO,CAAC,CAAC,CAAC,kBAAU,CAAC,OAAO;gBACzD,WAAW,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;gBAC9D,KAAK,EAAE,iBAAS,CAAC,KAAK;aACvB,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,yDAAyD;QAC3D,CAAC;IACH,CAAC;CACF;AAvND,gDAuNC"}

package/dist/integrations/vercel-ai.d.ts

@@ -0,0 +1,185 @@
+/**
+ * Vercel AI SDK integration for Tenet AI.
+ *
+ * Provides wrappers for the Vercel AI SDK's `generateText` and `streamText`
+ * functions that automatically record prompts, responses, tool calls, and
+ * model metadata in the Tenet audit trail.
+ *
+ * This module has **no hard dependency** on the `ai` (Vercel AI SDK) package.
+ * All relevant types are defined inline and the wrappers accept generic
+ * function references.
+ *
+ * @example
+ * ```ts
+ * import { generateText, streamText } from 'ai';
+ * import { openai } from '@ai-sdk/openai';
+ * import { Tenet } from '@tenet-ai/sdk';
+ * import { TenetVercelAITracker } from '@tenet-ai/sdk/integrations/vercel-ai';
+ *
+ * const tenet = new Tenet({ apiKey: 'tenet_xxx' });
+ * const tracker = new TenetVercelAITracker(tenet, {
+ *   agentId: 'vercel-ai-agent',
+ * });
+ *
+ * // Wrap generateText for automatic tracking
+ * const trackedGenerate = tracker.wrapGenerateText(generateText);
+ * const result = await trackedGenerate({
+ *   model: openai('gpt-4o'),
+ *   prompt: 'What is the meaning of life?',
+ * });
+ *
+ * // Wrap streamText for automatic tracking
+ * const trackedStream = tracker.wrapStreamText(streamText);
+ * const stream = await trackedStream({
+ *   model: openai('gpt-4o'),
+ *   prompt: 'Tell me a story.',
+ * });
+ * ```
+ *
+ * @module
+ */
+import { Tenet } from "../client";
+/** A tool call result shape from the Vercel AI SDK. */
+export interface VercelToolCall {
+    toolCallId: string;
+    toolName: string;
+    args: Record<string, unknown>;
+}
+/** A tool result shape from the Vercel AI SDK. */
+export interface VercelToolResult {
+    toolCallId: string;
+    toolName: string;
+    args: Record<string, unknown>;
+    result: unknown;
+}
+/** A message in Vercel AI SDK format. */
+export interface VercelMessage {
+    role: "system" | "user" | "assistant" | "tool" | string;
+    content: string;
+    [key: string]: unknown;
+}
+/** Token usage information. */
+export interface VercelTokenUsage {
+    promptTokens: number;
+    completionTokens: number;
+    totalTokens: number;
+}
+/**
+ * Parameters for `generateText` / `streamText`.
+ * This is a minimal subset -- the real type has many more fields.
+ */
+export interface VercelGenerateParams {
+    model: unknown;
+    prompt?: string;
+    messages?: VercelMessage[];
+    system?: string;
+    tools?: Record<string, unknown>;
+    maxTokens?: number;
+    temperature?: number;
+    [key: string]: unknown;
+}
+/** The result returned by `generateText`. */
+export interface VercelGenerateTextResult {
+    text: string;
+    toolCalls?: VercelToolCall[];
+    toolResults?: VercelToolResult[];
+    finishReason?: string;
+    usage?: VercelTokenUsage;
+    [key: string]: unknown;
+}
+/** The result returned by `streamText` (before consuming the stream). */
+export interface VercelStreamTextResult {
+    textStream: AsyncIterable<string>;
+    fullStream: AsyncIterable<unknown>;
+    text: Promise<string>;
+    toolCalls?: Promise<VercelToolCall[]>;
+    toolResults?: Promise<VercelToolResult[]>;
+    finishReason?: Promise<string>;
+    usage?: Promise<VercelTokenUsage>;
+    [key: string]: unknown;
+}
+/** Signature of the `generateText` function from `ai`. */
+export type GenerateTextFn = (params: VercelGenerateParams) => Promise<VercelGenerateTextResult>;
+/** Signature of the `streamText` function from `ai`. */
+export type StreamTextFn = (params: VercelGenerateParams) => Promise<VercelStreamTextResult>;
+/** Configuration options for {@link TenetVercelAITracker}. */
+export interface TenetVercelAITrackerConfig {
+    /** Identifier for the agent. Defaults to `"vercel-ai-agent"`. */
+    agentId?: string;
+    /** Explicit session ID. A random UUID is generated if omitted. */
+    sessionId?: string;
+    /** Tags attached to every decision. */
+    tags?: string[];
+    /** Arbitrary metadata attached to every decision. */
+    metadata?: Record<string, unknown>;
+}
+/**
+ * Wrapper for the Vercel AI SDK that records `generateText` and `streamText`
+ * calls in the Tenet audit trail.
+ *
+ * Each call records the full intent -> context -> decision -> execution
+ * chain including the prompt, model, tool calls, and token usage.
+ */
+export declare class TenetVercelAITracker {
+    private client;
+    private agentId;
+    private sessionId;
+    private tags;
+    private metadata;
+    /**
+     * @param client  - An initialised {@link Tenet} client instance.
+     * @param config  - Optional configuration overrides.
+     */
+    constructor(client: Tenet, config?: TenetVercelAITrackerConfig);
+    /**
+     * Wrap the Vercel AI SDK `generateText` function with automatic Tenet
+     * tracking. Returns a new function with the same signature.
+     *
+     * @param generateText - The `generateText` function from the `ai` package.
+     * @returns A wrapped function that records calls in Tenet.
+     *
+     * @example
+     * ```ts
+     * import { generateText } from 'ai';
+     * const trackedGenerate = tracker.wrapGenerateText(generateText);
+     * const result = await trackedGenerate({ model, prompt: 'Hello' });
+     * ```
+     */
+    wrapGenerateText(generateText: GenerateTextFn): GenerateTextFn;
+    /**
+     * Wrap the Vercel AI SDK `streamText` function with automatic Tenet
+     * tracking. The tracking data is recorded after the stream completes
+     * (resolves its promises).
+     *
+     * @param streamText - The `streamText` function from the `ai` package.
+     * @returns A wrapped function that records calls in Tenet.
+     *
+     * @example
+     * ```ts
+     * import { streamText } from 'ai';
+     * const trackedStream = tracker.wrapStreamText(streamText);
+     * const result = await trackedStream({ model, prompt: 'Hello' });
+     * for await (const chunk of result.textStream) {
+     *   process.stdout.write(chunk);
+     * }
+     * ```
+     */
+    wrapStreamText(streamText: StreamTextFn): StreamTextFn;
+    /**
+     * Manually record a `generateText` call. Useful when you want to call
+     * `generateText` yourself and record after the fact.
+     *
+     * @param params   - The parameters that were passed to `generateText`.
+     * @param result   - The result, if the call succeeded.
+     * @param error    - The error, if the call failed.
+     */
+    recordGenerateText(params: VercelGenerateParams, result?: VercelGenerateTextResult, error?: Error): Promise<void>;
+    /**
+     * Record a streaming result after its promises resolve.
+     */
+    private recordStreamResult;
+    /**
+     * Core recording logic shared by generate and stream paths.
+     */
+    private record;
+}