arkna-sdk 0.1.0

package/dist/instrumentations/langchain.js

@@ -0,0 +1,398 @@
+"use strict";
+/**
+ * ARKNA LangChain Auto-Instrumentation
+ *
+ * Callback handler that automatically captures LangChain chain/agent
+ * executions as ARKNA observability runs — no manual SDK calls needed.
+ *
+ * Usage:
+ *   import { ArknaClient, ArknaLangChainHandler } from '@arkna/sdk';
+ *   const client = new ArknaClient();
+ *   const handler = new ArknaLangChainHandler(client);
+ *   const result = await chain.invoke(input, { callbacks: [handler] });
+ *
+ * Requires: langchain >= 0.2.0 (peer dependency, not bundled)
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ArknaLangChainHandler = void 0;
+/**
+ * LangChain callback handler that auto-captures runs into ARKNA observability.
+ *
+ * Event mapping:
+ *   chain start  → startRun() (top-level) or recordStep(reasoning) (nested)
+ *   LLM start    → recordStep(reasoning)
+ *   LLM end      → update step with output + token usage
+ *   tool start   → recordToolCall()
+ *   tool end     → (tool result captured)
+ *   retriever    → recordStep(retrieval)
+ *   chain end    → completeRun(completed)
+ *   chain error  → completeRun(failed)
+ *
+ * This class dynamically extends BaseCallbackHandler at runtime so that
+ * `@langchain/core` is only required at runtime (peer dependency).
+ */
+class ArknaLangChainHandler {
+    client;
+    sessionId;
+    trigger;
+    runMetadata;
+    // Track active runs: LangChain runId → ARKNA state
+    runs = new Map();
+    // Map LangChain parent runId → top-level ARKNA runId
+    parentToArknaRun = new Map();
+    // Track the root run ID
+    rootRunId = null;
+    // The actual LangChain BaseCallbackHandler instance (created lazily)
+    _handler = null;
+    constructor(client, options) {
+        this.client = client;
+        this.sessionId = options?.sessionId;
+        this.trigger = options?.trigger ?? 'langchain';
+        this.runMetadata = options?.metadata ?? {};
+    }
+    /**
+     * Get the LangChain-compatible callback handler instance.
+     * Lazily creates a BaseCallbackHandler subclass on first access.
+     * Pass this to LangChain's `callbacks` option.
+     */
+    get handler() {
+        if (!this._handler) {
+            this._handler = this.createHandler();
+        }
+        return this._handler;
+    }
+    /**
+     * Also allow using the ArknaLangChainHandler directly as a callback
+     * by implementing the key callback methods on the instance itself.
+     * LangChain checks for method existence, so this works as a duck-typed handler.
+     */
+    createHandler() {
+        // eslint-disable-next-line @typescript-eslint/no-this-alias
+        const self = this;
+        try {
+            // Try to dynamically import and extend BaseCallbackHandler
+            const { BaseCallbackHandler } = require('@langchain/core/callbacks/base');
+            class ArknaHandler extends BaseCallbackHandler {
+                name = 'ArknaLangChainHandler';
+                async handleChainStart(chain, inputs, runId, parentRunId) {
+                    return self.handleChainStart(chain, inputs, runId, parentRunId);
+                }
+                async handleChainEnd(outputs, runId) {
+                    return self.handleChainEnd(outputs, runId);
+                }
+                async handleChainError(err, runId) {
+                    return self.handleChainError(err, runId);
+                }
+                async handleLLMStart(llm, prompts, runId, parentRunId) {
+                    return self.handleLLMStart(llm, prompts, runId, parentRunId);
+                }
+                async handleLLMEnd(output, runId) {
+                    return self.handleLLMEnd(output, runId);
+                }
+                async handleLLMError(err, runId) {
+                    return self.handleLLMError(err, runId);
+                }
+                async handleToolStart(tool, input, runId, parentRunId) {
+                    return self.handleToolStart(tool, input, runId, parentRunId);
+                }
+                async handleToolEnd(_output, _runId) { }
+                async handleToolError(err, runId) {
+                    return self.handleToolError(err, runId);
+                }
+                async handleRetrieverStart(retriever, query, runId, parentRunId) {
+                    return self.handleRetrieverStart(retriever, query, runId, parentRunId);
+                }
+                async handleRetrieverEnd(documents, runId) {
+                    return self.handleRetrieverEnd(documents, runId);
+                }
+                async handleRetrieverError(err, runId) {
+                    return self.handleRetrieverError(err, runId);
+                }
+            }
+            return new ArknaHandler();
+        }
+        catch {
+            throw new Error('@langchain/core is required for ArknaLangChainHandler. Install it: npm install @langchain/core');
+        }
+    }
+    async post(path, body) {
+        return this.client.ingestionRequest('POST', path, body);
+    }
+    async patch(path, body) {
+        return this.client.ingestionRequest('PATCH', path, body);
+    }
+    // ────────────────────────────────────────────────────────────
+    // Chain Events
+    // ────────────────────────────────────────────────────────────
+    async handleChainStart(chain, inputs, runId, parentRunId) {
+        try {
+            const chainName = chain.id?.[chain.id.length - 1] ?? 'chain';
+            if (!parentRunId || !this.rootRunId) {
+                const inputStr = typeof inputs === 'string' ? inputs : JSON.stringify(inputs);
+                const resp = await this.post('/runs', {
+                    input: inputStr,
+                    trigger_type: this.trigger,
+                    session_id: this.sessionId,
+                    metadata: {
+                        ...this.runMetadata,
+                        chain_name: chainName,
+                        framework: 'langchain',
+                    },
+                });
+                const arknaRunId = resp.run_id;
+                this.runs.set(runId, { runId: arknaRunId, currentStepId: null, depth: 0 });
+                this.parentToArknaRun.set(runId, arknaRunId);
+                this.rootRunId = runId;
+            }
+            else {
+                const arknaRunId = this.getArknaRunId(parentRunId);
+                if (!arknaRunId)
+                    return;
+                const resp = await this.post(`/runs/${arknaRunId}/steps`, {
+                    step_type: 'reasoning',
+                    input: JSON.stringify(inputs),
+                    metadata: { chain_name: chainName, nested: true },
+                });
+                const depth = (this.runs.get(parentRunId)?.depth ?? 0) + 1;
+                this.runs.set(runId, { runId: arknaRunId, currentStepId: resp.step_id, depth });
+                this.parentToArknaRun.set(runId, arknaRunId);
+            }
+        }
+        catch {
+            // Non-critical — don't break the chain execution
+        }
+    }
+    async handleChainEnd(outputs, runId) {
+        try {
+            if (runId === this.rootRunId) {
+                const state = this.runs.get(runId);
+                if (!state)
+                    return;
+                const outputStr = typeof outputs === 'string' ? outputs : JSON.stringify(outputs);
+                await this.patch(`/runs/${state.runId}`, { status: 'completed', output: outputStr });
+                this.cleanup();
+            }
+        }
+        catch {
+            // Non-critical
+        }
+    }
+    async handleChainError(err, runId) {
+        try {
+            if (runId === this.rootRunId) {
+                const state = this.runs.get(runId);
+                if (!state)
+                    return;
+                await this.patch(`/runs/${state.runId}`, {
+                    status: 'failed',
+                    error_type: err.name || 'Error',
+                    error_message: err.message,
+                });
+                this.cleanup();
+            }
+        }
+        catch {
+            // Non-critical
+        }
+    }
+    // ────────────────────────────────────────────────────────────
+    // LLM Events
+    // ────────────────────────────────────────────────────────────
+    async handleLLMStart(llm, prompts, runId, parentRunId) {
+        try {
+            const arknaRunId = this.getArknaRunId(parentRunId || runId);
+            if (!arknaRunId)
+                return;
+            const modelName = llm.id?.[llm.id.length - 1] ?? 'llm';
+            const resp = await this.post(`/runs/${arknaRunId}/steps`, {
+                step_type: 'reasoning',
+                input: prompts.length === 1 ? prompts[0] : JSON.stringify(prompts),
+                metadata: { model: modelName },
+            });
+            const state = this.runs.get(parentRunId || runId);
+            if (state) {
+                this.runs.set(runId, { ...state, currentStepId: resp.step_id });
+            }
+            else {
+                this.runs.set(runId, { runId: arknaRunId, currentStepId: resp.step_id, depth: 0 });
+            }
+            this.parentToArknaRun.set(runId, arknaRunId);
+        }
+        catch {
+            // Non-critical
+        }
+    }
+    async handleLLMEnd(output, runId) {
+        try {
+            const state = this.runs.get(runId);
+            if (!state?.currentStepId)
+                return;
+            const tokenUsage = output.llmOutput?.tokenUsage || output.llmOutput?.usage;
+            const totalTokens = tokenUsage?.totalTokens ?? tokenUsage?.total_tokens;
+            const generationText = output.generations?.[0]?.[0]?.text ?? '';
+            await this.post(`/runs/${state.runId}/steps`, {
+                step_type: 'reasoning',
+                output: generationText.substring(0, 10000),
+                reasoning: generationText.substring(0, 5000),
+                tokens_used: totalTokens ?? undefined,
+                metadata: {
+                    llm_output: true,
+                    ...(tokenUsage ? {
+                        prompt_tokens: tokenUsage.promptTokens ?? tokenUsage.prompt_tokens,
+                        completion_tokens: tokenUsage.completionTokens ?? tokenUsage.completion_tokens,
+                    } : {}),
+                },
+            });
+        }
+        catch {
+            // Non-critical
+        }
+    }
+    async handleLLMError(err, runId) {
+        try {
+            const state = this.runs.get(runId);
+            if (!state)
+                return;
+            await this.post(`/runs/${state.runId}/steps`, {
+                step_type: 'reasoning',
+                output: `LLM Error: ${err.message}`,
+                metadata: { error: true, error_type: err.name },
+            });
+        }
+        catch {
+            // Non-critical
+        }
+    }
+    // ────────────────────────────────────────────────────────────
+    // Tool Events
+    // ────────────────────────────────────────────────────────────
+    async handleToolStart(tool, input, runId, parentRunId) {
+        try {
+            const arknaRunId = this.getArknaRunId(parentRunId || runId);
+            if (!arknaRunId)
+                return;
+            const toolName = tool.id?.[tool.id.length - 1] ?? 'tool';
+            const parentState = this.runs.get(parentRunId || runId);
+            let stepId = parentState?.currentStepId ?? null;
+            if (!stepId) {
+                const stepResp = await this.post(`/runs/${arknaRunId}/steps`, {
+                    step_type: 'action',
+                    input: `Tool call: ${toolName}`,
+                });
+                stepId = stepResp.step_id;
+            }
+            let parsedArgs = {};
+            try {
+                parsedArgs = JSON.parse(input);
+            }
+            catch {
+                parsedArgs = { input };
+            }
+            await this.post(`/runs/${arknaRunId}/tools`, {
+                step_id: stepId,
+                tool_name: toolName,
+                arguments: parsedArgs,
+            });
+            this.runs.set(runId, {
+                runId: arknaRunId,
+                currentStepId: stepId,
+                depth: parentState?.depth ?? 0,
+            });
+            this.parentToArknaRun.set(runId, arknaRunId);
+        }
+        catch {
+            // Non-critical
+        }
+    }
+    async handleToolError(err, runId) {
+        try {
+            const state = this.runs.get(runId);
+            if (!state)
+                return;
+            await this.post(`/runs/${state.runId}/steps`, {
+                step_type: 'action',
+                output: `Tool Error: ${err.message}`,
+                metadata: { error: true, error_type: err.name },
+            });
+        }
+        catch {
+            // Non-critical
+        }
+    }
+    // ────────────────────────────────────────────────────────────
+    // Retriever Events
+    // ────────────────────────────────────────────────────────────
+    async handleRetrieverStart(retriever, query, runId, parentRunId) {
+        try {
+            const arknaRunId = this.getArknaRunId(parentRunId || runId);
+            if (!arknaRunId)
+                return;
+            const retrieverName = retriever.id?.[retriever.id.length - 1] ?? 'retriever';
+            const resp = await this.post(`/runs/${arknaRunId}/steps`, {
+                step_type: 'retrieval',
+                input: query,
+                metadata: { retriever: retrieverName },
+            });
+            this.runs.set(runId, {
+                runId: arknaRunId,
+                currentStepId: resp.step_id,
+                depth: this.runs.get(parentRunId || runId)?.depth ?? 0,
+            });
+            this.parentToArknaRun.set(runId, arknaRunId);
+        }
+        catch {
+            // Non-critical
+        }
+    }
+    async handleRetrieverEnd(documents, runId) {
+        try {
+            const state = this.runs.get(runId);
+            if (!state)
+                return;
+            const docSummary = documents.map((d, i) => ({
+                index: i,
+                content_length: d.pageContent?.length ?? 0,
+                metadata: d.metadata,
+            }));
+            await this.post(`/runs/${state.runId}/steps`, {
+                step_type: 'retrieval',
+                output: JSON.stringify(docSummary).substring(0, 10000),
+                metadata: { documents_retrieved: documents.length },
+            });
+        }
+        catch {
+            // Non-critical
+        }
+    }
+    async handleRetrieverError(err, runId) {
+        try {
+            const state = this.runs.get(runId);
+            if (!state)
+                return;
+            await this.post(`/runs/${state.runId}/steps`, {
+                step_type: 'retrieval',
+                output: `Retriever Error: ${err.message}`,
+                metadata: { error: true },
+            });
+        }
+        catch {
+            // Non-critical
+        }
+    }
+    // ────────────────────────────────────────────────────────────
+    // Helpers
+    // ────────────────────────────────────────────────────────────
+    getArknaRunId(lcRunId) {
+        const direct = this.runs.get(lcRunId);
+        if (direct)
+            return direct.runId;
+        return this.parentToArknaRun.get(lcRunId) ?? null;
+    }
+    cleanup() {
+        this.runs.clear();
+        this.parentToArknaRun.clear();
+        this.rootRunId = null;
+    }
+}
+exports.ArknaLangChainHandler = ArknaLangChainHandler;
+//# sourceMappingURL=langchain.js.map

package/dist/instrumentations/langchain.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"langchain.js","sourceRoot":"","sources":["../../src/instrumentations/langchain.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;GAaG;;;AAoBH;;;;;;;;;;;;;;;GAeG;AACH,MAAa,qBAAqB;IACxB,MAAM,CAAc;IACpB,SAAS,CAAU;IACnB,OAAO,CAAS;IAChB,WAAW,CAA0B;IAE7C,mDAAmD;IAC3C,IAAI,GAAG,IAAI,GAAG,EAAoB,CAAC;IAC3C,qDAAqD;IAC7C,gBAAgB,GAAG,IAAI,GAAG,EAAkB,CAAC;IACrD,wBAAwB;IAChB,SAAS,GAAkB,IAAI,CAAC;IAExC,qEAAqE;IAC7D,QAAQ,GAAQ,IAAI,CAAC;IAE7B,YAAY,MAAmB,EAAE,OAAsC;QACrE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,SAAS,GAAG,OAAO,EAAE,SAAS,CAAC;QACpC,IAAI,CAAC,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,WAAW,CAAC;QAC/C,IAAI,CAAC,WAAW,GAAG,OAAO,EAAE,QAAQ,IAAI,EAAE,CAAC;IAC7C,CAAC;IAED;;;;OAIG;IACH,IAAI,OAAO;QACT,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;QACvC,CAAC;QACD,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED;;;;OAIG;IACK,aAAa;QACnB,4DAA4D;QAC5D,MAAM,IAAI,GAAG,IAAI,CAAC;QAElB,IAAI,CAAC;YACH,2DAA2D;YAC3D,MAAM,EAAE,mBAAmB,EAAE,GAAG,OAAO,CAAC,gCAAgC,CAAC,CAAC;YAE1E,MAAM,YAAa,SAAQ,mBAAmB;gBAC5C,IAAI,GAAG,uBAAuB,CAAC;gBAE/B,KAAK,CAAC,gBAAgB,CAAC,KAAU,EAAE,MAAW,EAAE,KAAa,EAAE,WAAoB;oBACjF,OAAO,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC;gBAClE,CAAC;gBACD,KAAK,CAAC,cAAc,CAAC,OAAY,EAAE,KAAa;oBAC9C,OAAO,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;gBAC7C,CAAC;gBACD,KAAK,CAAC,gBAAgB,CAAC,GAAQ,EAAE,KAAa;oBAC5C,OAAO,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;gBAC3C,CAAC;gBACD,KAAK,CAAC,cAAc,CAAC,GAAQ,EAAE,OAAiB,EAAE,KAAa,EAAE,WAAoB;oBACnF,OAAO,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC;gBAC/D,CAAC;gBACD,KAAK,CAAC,YAAY,CAAC,MAAW,EAAE,KAAa;oBAC3C,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;gBAC1C,CAAC;gBACD,KAAK,CAAC,cAAc,CAAC,GAAQ,EAAE,KAAa;oBAC1C,OAAO,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;gBACzC,CAAC;gBACD,KAAK,CAAC,eAAe,CAAC,IAAS,EAAE,KAAa,EAAE,KAAa,EAAE,WAAoB;oBACjF,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC;gBAC/D,CAAC;gBACD,KAAK,CAAC,aAAa,CAAC,OAAe,EAAE,MAAc,IAA4B,CAAC;gBAChF,KAAK,CAAC,eAAe,CAAC,GAAQ,EAAE,KAAa;oBAC3C,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;gBAC1C,CAAC;gBACD,KAAK,CAAC,oBAAoB,CAAC,SAAc,EAAE,KAAa,EAAE,KAAa,EAAE,WAAoB;oBAC3F,OAAO,IAAI,CAAC,oBAAoB,CAAC,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC;gBACzE,CAAC;gBACD,KAAK,CAAC,kBAAkB,CAAC,SAAgB,EAAE,KAAa;oBACtD,OAAO,IAAI,CAAC,kBAAkB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;gBACnD,CAAC;gBACD,KAAK,CAAC,oBAAoB,CAAC,GAAQ,EAAE,KAAa;oBAChD,OAAO,IAAI,CAAC,oBAAoB,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;gBAC/C,CAAC;aACF;YAED,OAAO,IAAI,YAAY,EAAE,CAAC;QAC5B,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,IAAI,KAAK,CACb,gGAAgG,CACjG,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,IAAI,CAAC,IAAY,EAAE,IAAa;QAC5C,OAAQ,IAAI,CAAC,MAAc,CAAC,gBAAgB,CAAC,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IACnE,CAAC;IAEO,KAAK,CAAC,KAAK,CAAC,IAAY,EAAE,IAAa;QAC7C,OAAQ,IAAI,CAAC,MAAc,CAAC,gBAAgB,CAAC,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IACpE,CAAC;IAED,+DAA+D;IAC/D,eAAe;IACf,+DAA+D;IAEvD,KAAK,CAAC,gBAAgB,CAC5B,KAAU,EACV,MAA+B,EAC/B,KAAa,EACb,WAAoB;QAEpB,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,OAAO,CAAC;YAE7D,IAAI,CAAC,WAAW,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACpC,MAAM,QAAQ,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;gBAC9E,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE;oBACpC,KAAK,EAAE,QAAQ;oBACf,YAAY,EAAE,IAAI,CAAC,OAAO;oBAC1B,UAAU,EAAE,IAAI,CAAC,SAAS;oBAC1B,QAAQ,EAAE;wBACR,GAAG,IAAI,CAAC,WAAW;wBACnB,UAAU,EAAE,SAAS;wBACrB,SAAS,EAAE,WAAW;qBACvB;iBACF,CAAC,CAAC;gBAEH,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC;gBAC/B,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,aAAa,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;gBAC3E,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;gBAC7C,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;YACzB,CAAC;iBAAM,CAAC;gBACN,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC;gBACnD,IAAI,CAAC,UAAU;oBAAE,OAAO;gBAExB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,UAAU,QAAQ,EAAE;oBACxD,SAAS,EAAE,WAAW;oBACtB,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;oBAC7B,QAAQ,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE;iBAClD,CAAC,CAAC;gBAEH,MAAM,KAAK,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;gBAC3D,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,aAAa,EAAE,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;gBAChF,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;YAC/C,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,iDAAiD;QACnD,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,OAAgC,EAChC,KAAa;QAEb,IAAI,CAAC;YACH,IAAI,KAAK,KAAK,IAAI,CAAC,SAAS,EAAE,CAAC;gBAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBACnC,IAAI,CAAC,KAAK;oBAAE,OAAO;gBAEnB,MAAM,SAAS,GAAG,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;gBAClF,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,KAAK,CAAC,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;gBACrF,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,eAAe;QACjB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC5B,GAAU,EACV,KAAa;QAEb,IAAI,CAAC;YACH,IAAI,KAAK,KAAK,IAAI,CAAC,SAAS,EAAE,CAAC;gBAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBACnC,IAAI,CAAC,KAAK;oBAAE,OAAO;gBAEnB,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,KAAK,CAAC,KAAK,EAAE,EAAE;oBACvC,MAAM,EAAE,QAAQ;oBAChB,UAAU,EAAE,GAAG,CAAC,IAAI,IAAI,OAAO;oBAC/B,aAAa,EAAE,GAAG,CAAC,OAAO;iBAC3B,CAAC,CAAC;gBACH,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,eAAe;QACjB,CAAC;IACH,CAAC;IAED,+DAA+D;IAC/D,aAAa;IACb,+DAA+D;IAEvD,KAAK,CAAC,cAAc,CAC1B,GAAQ,EACR,OAAiB,EACjB,KAAa,EACb,WAAoB;QAEpB,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,WAAW,IAAI,KAAK,CAAC,CAAC;YAC5D,IAAI,CAAC,UAAU;gBAAE,OAAO;YAExB,MAAM,SAAS,GAAG,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC;YAEvD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,UAAU,QAAQ,EAAE;gBACxD,SAAS,EAAE,WAAW;gBACtB,KAAK,EAAE,OAAO,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;gBAClE,QAAQ,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE;aAC/B,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,IAAI,KAAK,CAAC,CAAC;YAClD,IAAI,KAAK,EAAE,CAAC;gBACV,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,GAAG,KAAK,EAAE,aAAa,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;YAClE,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,aAAa,EAAE,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YACrF,CAAC;YACD,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QAC/C,CAAC;QAAC,MAAM,CAAC;YACP,eAAe;QACjB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,MAAW,EACX,KAAa;QAEb,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACnC,IAAI,CAAC,KAAK,EAAE,aAAa;gBAAE,OAAO;YAElC,MAAM,UAAU,GAAG,MAAM,CAAC,SAAS,EAAE,UAAU,IAAI,MAAM,CAAC,SAAS,EAAE,KAAK,CAAC;YAC3E,MAAM,WAAW,GAAG,UAAU,EAAE,WAAW,IAAI,UAAU,EAAE,YAAY,CAAC;YACxE,MAAM,cAAc,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;YAEhE,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,KAAK,CAAC,KAAK,QAAQ,EAAE;gBAC5C,SAAS,EAAE,WAAW;gBACtB,MAAM,EAAE,cAAc,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,CAAC;gBAC1C,SAAS,EAAE,cAAc,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC;gBAC5C,WAAW,EAAE,WAAW,IAAI,SAAS;gBACrC,QAAQ,EAAE;oBACR,UAAU,EAAE,IAAI;oBAChB,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;wBACf,aAAa,EAAE,UAAU,CAAC,YAAY,IAAI,UAAU,CAAC,aAAa;wBAClE,iBAAiB,EAAE,UAAU,CAAC,gBAAgB,IAAI,UAAU,CAAC,iBAAiB;qBAC/E,CAAC,CAAC,CAAC,EAAE,CAAC;iBACR;aACF,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,eAAe;QACjB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,GAAU,EACV,KAAa;QAEb,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACnC,IAAI,CAAC,KAAK;gBAAE,OAAO;YAEnB,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,KAAK,CAAC,KAAK,QAAQ,EAAE;gBAC5C,SAAS,EAAE,WAAW;gBACtB,MAAM,EAAE,cAAc,GAAG,CAAC,OAAO,EAAE;gBACnC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,CAAC,IAAI,EAAE;aAChD,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,eAAe;QACjB,CAAC;IACH,CAAC;IAED,+DAA+D;IAC/D,cAAc;IACd,+DAA+D;IAEvD,KAAK,CAAC,eAAe,CAC3B,IAAS,EACT,KAAa,EACb,KAAa,EACb,WAAoB;QAEpB,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,WAAW,IAAI,KAAK,CAAC,CAAC;YAC5D,IAAI,CAAC,UAAU;gBAAE,OAAO;YAExB,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,MAAM,CAAC;YAEzD,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,IAAI,KAAK,CAAC,CAAC;YACxD,IAAI,MAAM,GAAkB,WAAW,EAAE,aAAa,IAAI,IAAI,CAAC;YAE/D,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,UAAU,QAAQ,EAAE;oBAC5D,SAAS,EAAE,QAAQ;oBACnB,KAAK,EAAE,cAAc,QAAQ,EAAE;iBAChC,CAAC,CAAC;gBACH,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC;YAC5B,CAAC;YAED,IAAI,UAAU,GAA4B,EAAE,CAAC;YAC7C,IAAI,CAAC;gBACH,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACjC,CAAC;YAAC,MAAM,CAAC;gBACP,UAAU,GAAG,EAAE,KAAK,EAAE,CAAC;YACzB,CAAC;YAED,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,UAAU,QAAQ,EAAE;gBAC3C,OAAO,EAAE,MAAM;gBACf,SAAS,EAAE,QAAQ;gBACnB,SAAS,EAAE,UAAU;aACtB,CAAC,CAAC;YAEH,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE;gBACnB,KAAK,EAAE,UAAU;gBACjB,aAAa,EAAE,MAAM;gBACrB,KAAK,EAAE,WAAW,EAAE,KAAK,IAAI,CAAC;aAC/B,CAAC,CAAC;YACH,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QAC/C,CAAC;QAAC,MAAM,CAAC;YACP,eAAe;QACjB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,eAAe,CAC3B,GAAU,EACV,KAAa;QAEb,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACnC,IAAI,CAAC,KAAK;gBAAE,OAAO;YAEnB,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,KAAK,CAAC,KAAK,QAAQ,EAAE;gBAC5C,SAAS,EAAE,QAAQ;gBACnB,MAAM,EAAE,eAAe,GAAG,CAAC,OAAO,EAAE;gBACpC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,CAAC,IAAI,EAAE;aAChD,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,eAAe;QACjB,CAAC;IACH,CAAC;IAED,+DAA+D;IAC/D,mBAAmB;IACnB,+DAA+D;IAEvD,KAAK,CAAC,oBAAoB,CAChC,SAAc,EACd,KAAa,EACb,KAAa,EACb,WAAoB;QAEpB,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,WAAW,IAAI,KAAK,CAAC,CAAC;YAC5D,IAAI,CAAC,UAAU;gBAAE,OAAO;YAExB,MAAM,aAAa,GAAG,SAAS,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,WAAW,CAAC;YAE7E,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,UAAU,QAAQ,EAAE;gBACxD,SAAS,EAAE,WAAW;gBACtB,KAAK,EAAE,KAAK;gBACZ,QAAQ,EAAE,EAAE,SAAS,EAAE,aAAa,EAAE;aACvC,CAAC,CAAC;YAEH,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE;gBACnB,KAAK,EAAE,UAAU;gBACjB,aAAa,EAAE,IAAI,CAAC,OAAO;gBAC3B,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,IAAI,KAAK,CAAC,EAAE,KAAK,IAAI,CAAC;aACvD,CAAC,CAAC;YACH,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QAC/C,CAAC;QAAC,MAAM,CAAC;YACP,eAAe;QACjB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAC9B,SAAgB,EAChB,KAAa;QAEb,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACnC,IAAI,CAAC,KAAK;gBAAE,OAAO;YAEnB,MAAM,UAAU,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,CAAS,EAAE,EAAE,CAAC,CAAC;gBACvD,KAAK,EAAE,CAAC;gBACR,cAAc,EAAE,CAAC,CAAC,WAAW,EAAE,MAAM,IAAI,CAAC;gBAC1C,QAAQ,EAAE,CAAC,CAAC,QAAQ;aACrB,CAAC,CAAC,CAAC;YAEJ,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,KAAK,CAAC,KAAK,QAAQ,EAAE;gBAC5C,SAAS,EAAE,WAAW;gBACtB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,CAAC;gBACtD,QAAQ,EAAE,EAAE,mBAAmB,EAAE,SAAS,CAAC,MAAM,EAAE;aACpD,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,eAAe;QACjB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAChC,GAAU,EACV,KAAa;QAEb,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACnC,IAAI,CAAC,KAAK;gBAAE,OAAO;YAEnB,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,KAAK,CAAC,KAAK,QAAQ,EAAE;gBAC5C,SAAS,EAAE,WAAW;gBACtB,MAAM,EAAE,oBAAoB,GAAG,CAAC,OAAO,EAAE;gBACzC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE;aAC1B,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,eAAe;QACjB,CAAC;IACH,CAAC;IAED,+DAA+D;IAC/D,UAAU;IACV,+DAA+D;IAEvD,aAAa,CAAC,OAAe;QACnC,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACtC,IAAI,MAAM;YAAE,OAAO,MAAM,CAAC,KAAK,CAAC;QAChC,OAAO,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC;IACpD,CAAC;IAEO,OAAO;QACb,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QAClB,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;QAC9B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;IACxB,CAAC;CACF;AAhbD,sDAgbC"}

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

@@ -0,0 +1,40 @@
+/**
+ * ARKNA Vercel AI SDK Auto-Instrumentation
+ *
+ * Wraps Vercel AI SDK's `generateText()` and `streamText()` to automatically
+ * capture runs in ARKNA observability — no manual SDK calls needed.
+ *
+ * Usage:
+ *   import { ArknaClient, wrapVercelAI } from '@arkna/sdk';
+ *   const client = new ArknaClient();
+ *   const ai = wrapVercelAI(client);
+ *   const result = await ai.generateText({ model, prompt });
+ *
+ * Requires: ai (Vercel AI SDK) as peer dependency
+ */
+import type { ArknaClient } from '../client';
+/** Options for wrapping Vercel AI SDK */
+export interface WrapVercelAIOptions {
+    /** Session ID to group related runs */
+    sessionId?: string;
+    /** How the run was triggered */
+    trigger?: string;
+    /** Additional metadata */
+    metadata?: Record<string, unknown>;
+}
+interface WrappedAI {
+    generateText: (params: any) => Promise<any>;
+    streamText: (params: any) => Promise<any>;
+}
+/**
+ * Wrap Vercel AI SDK functions with ARKNA auto-instrumentation.
+ *
+ * Returns wrapped `generateText` and `streamText` that automatically capture:
+ * - Run start/end
+ * - LLM reasoning steps
+ * - Tool calls and results
+ * - Token usage (final response only, not streaming tokens)
+ */
+export declare function wrapVercelAI(client: ArknaClient, options?: WrapVercelAIOptions): WrappedAI;
+export {};
+//# sourceMappingURL=vercel-ai.d.ts.map

package/dist/instrumentations/vercel-ai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vercel-ai.d.ts","sourceRoot":"","sources":["../../src/instrumentations/vercel-ai.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,WAAW,CAAC;AAE7C,yCAAyC;AACzC,MAAM,WAAW,mBAAmB;IAClC,uCAAuC;IACvC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,gCAAgC;IAChC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,0BAA0B;IAC1B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED,UAAU,SAAS;IACjB,YAAY,EAAE,CAAC,MAAM,EAAE,GAAG,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;IAC5C,UAAU,EAAE,CAAC,MAAM,EAAE,GAAG,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;CAC3C;AAED;;;;;;;;GAQG;AACH,wBAAgB,YAAY,CAC1B,MAAM,EAAE,WAAW,EACnB,OAAO,CAAC,EAAE,mBAAmB,GAC5B,SAAS,CAyMX"}

package/dist/instrumentations/vercel-ai.js

@@ -0,0 +1,212 @@
+"use strict";
+/**
+ * ARKNA Vercel AI SDK Auto-Instrumentation
+ *
+ * Wraps Vercel AI SDK's `generateText()` and `streamText()` to automatically
+ * capture runs in ARKNA observability — no manual SDK calls needed.
+ *
+ * Usage:
+ *   import { ArknaClient, wrapVercelAI } from '@arkna/sdk';
+ *   const client = new ArknaClient();
+ *   const ai = wrapVercelAI(client);
+ *   const result = await ai.generateText({ model, prompt });
+ *
+ * Requires: ai (Vercel AI SDK) as peer dependency
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.wrapVercelAI = wrapVercelAI;
+/**
+ * Wrap Vercel AI SDK functions with ARKNA auto-instrumentation.
+ *
+ * Returns wrapped `generateText` and `streamText` that automatically capture:
+ * - Run start/end
+ * - LLM reasoning steps
+ * - Tool calls and results
+ * - Token usage (final response only, not streaming tokens)
+ */
+function wrapVercelAI(client, options) {
+    const sessionId = options?.sessionId;
+    const trigger = options?.trigger ?? 'vercel-ai';
+    const meta = options?.metadata ?? {};
+    async function post(path, body) {
+        return client.ingestionRequest('POST', path, body);
+    }
+    async function patch(path, body) {
+        return client.ingestionRequest('PATCH', path, body);
+    }
+    async function wrappedGenerateText(params) {
+        // Dynamically require — ai is a peer dependency
+        const { generateText: originalGenerateText } = require('ai');
+        const modelId = params.model?.modelId ?? params.model?.constructor?.name ?? 'unknown';
+        let runId = null;
+        try {
+            // Start run
+            const runResp = await post('/runs', {
+                input: typeof params.prompt === 'string' ? params.prompt : JSON.stringify(params.messages ?? params.prompt),
+                trigger_type: trigger,
+                session_id: sessionId,
+                metadata: { ...meta, model: modelId, framework: 'vercel-ai', mode: 'generate' },
+            });
+            runId = runResp.run_id;
+        }
+        catch {
+            // If we can't start the run, just execute without instrumentation
+            return originalGenerateText(params);
+        }
+        try {
+            const result = await originalGenerateText(params);
+            // Record LLM step
+            try {
+                const stepResp = await post(`/runs/${runId}/steps`, {
+                    step_type: 'reasoning',
+                    input: typeof params.prompt === 'string' ? params.prompt : JSON.stringify(params.messages ?? params.prompt),
+                    output: result.text,
+                    reasoning: result.reasoning ?? undefined,
+                    tokens_used: result.usage?.totalTokens ?? undefined,
+                    metadata: {
+                        model: modelId,
+                        prompt_tokens: result.usage?.promptTokens,
+                        completion_tokens: result.usage?.completionTokens,
+                        finish_reason: result.finishReason,
+                    },
+                });
+                // Record tool calls if any
+                if (result.toolCalls && result.toolCalls.length > 0) {
+                    for (const tc of result.toolCalls) {
+                        await post(`/runs/${runId}/tools`, {
+                            step_id: stepResp.step_id,
+                            tool_name: tc.toolName,
+                            arguments: tc.args ?? {},
+                        }).catch(() => { });
+                    }
+                }
+                // Record tool results if multi-step
+                if (result.steps) {
+                    for (const step of result.steps) {
+                        if (step.toolCalls?.length) {
+                            const sResp = await post(`/runs/${runId}/steps`, {
+                                step_type: 'action',
+                                input: JSON.stringify(step.toolCalls.map((t) => ({ name: t.toolName, args: t.args }))),
+                                output: JSON.stringify(step.toolResults ?? []),
+                            }).catch(() => null);
+                            if (sResp) {
+                                for (const tc of step.toolCalls) {
+                                    await post(`/runs/${runId}/tools`, {
+                                        step_id: sResp.step_id,
+                                        tool_name: tc.toolName,
+                                        arguments: tc.args ?? {},
+                                    }).catch(() => { });
+                                }
+                            }
+                        }
+                    }
+                }
+            }
+            catch {
+                // Non-critical
+            }
+            // Complete run
+            await patch(`/runs/${runId}`, { status: 'completed', output: result.text }).catch(() => { });
+            return result;
+        }
+        catch (err) {
+            // Fail run
+            if (runId) {
+                const error = err instanceof Error ? err : new Error(String(err));
+                await patch(`/runs/${runId}`, {
+                    status: 'failed',
+                    error_type: error.name || 'Error',
+                    error_message: error.message,
+                }).catch(() => { });
+            }
+            throw err;
+        }
+    }
+    async function wrappedStreamText(params) {
+        const { streamText: originalStreamText } = require('ai');
+        const modelId = params.model?.modelId ?? params.model?.constructor?.name ?? 'unknown';
+        let runId = null;
+        try {
+            const runResp = await post('/runs', {
+                input: typeof params.prompt === 'string' ? params.prompt : JSON.stringify(params.messages ?? params.prompt),
+                trigger_type: trigger,
+                session_id: sessionId,
+                metadata: { ...meta, model: modelId, framework: 'vercel-ai', mode: 'stream' },
+            });
+            runId = runResp.run_id;
+        }
+        catch {
+            return originalStreamText(params);
+        }
+        try {
+            const result = await originalStreamText(params);
+            // We don't capture streaming tokens — only the final response.
+            // Wrap the result to capture completion.
+            const capturedRunId = runId;
+            // Use the response promise to capture the final result
+            if (!result.response || typeof result.response.then !== 'function') {
+                // Vercel AI SDK version doesn't support .response promise — skip capture
+                return result;
+            }
+            result.response.then(async (_response) => {
+                try {
+                    const usage = await result.usage;
+                    const text = await result.text;
+                    const toolCalls = result.toolCalls;
+                    const stepResp = await post(`/runs/${capturedRunId}/steps`, {
+                        step_type: 'reasoning',
+                        output: text,
+                        tokens_used: usage?.totalTokens ?? undefined,
+                        metadata: {
+                            model: modelId,
+                            prompt_tokens: usage?.promptTokens,
+                            completion_tokens: usage?.completionTokens,
+                            streaming: true,
+                        },
+                    });
+                    // Capture tool calls from the stream result
+                    if (toolCalls) {
+                        const resolvedTools = await toolCalls;
+                        if (Array.isArray(resolvedTools)) {
+                            for (const tc of resolvedTools) {
+                                await post(`/runs/${capturedRunId}/tools`, {
+                                    step_id: stepResp.step_id,
+                                    tool_name: tc.toolName,
+                                    arguments: tc.args ?? {},
+                                }).catch(() => { });
+                            }
+                        }
+                    }
+                    await patch(`/runs/${capturedRunId}`, { status: 'completed', output: text }).catch(() => { });
+                }
+                catch {
+                    // Non-critical
+                }
+            }).catch(async (err) => {
+                const error = err instanceof Error ? err : new Error(String(err));
+                await patch(`/runs/${capturedRunId}`, {
+                    status: 'failed',
+                    error_type: error.name || 'Error',
+                    error_message: error.message,
+                }).catch(() => { });
+            });
+            return result;
+        }
+        catch (err) {
+            if (runId) {
+                const error = err instanceof Error ? err : new Error(String(err));
+                await patch(`/runs/${runId}`, {
+                    status: 'failed',
+                    error_type: error.name || 'Error',
+                    error_message: error.message,
+                }).catch(() => { });
+            }
+            throw err;
+        }
+    }
+    return {
+        generateText: wrappedGenerateText,
+        streamText: wrappedStreamText,
+    };
+}
+//# sourceMappingURL=vercel-ai.js.map