@agent-assembly/sdk 0.0.1-alpha.3

package/dist/esm/hooks/ai-sdk.js

@@ -0,0 +1,123 @@
+import { PolicyViolationError } from "../errors/policy-violation-error.js";
+import { runWithAgentId } from "../lineage/agent-context-store.js";
+export const vercelAiSdkPatchState = {
+    isPatched: false,
+    originalToolFactory: undefined,
+    patchedModule: undefined
+};
+export function captureOriginalToolFactory(module) {
+    const candidate = module.tool;
+    if (typeof candidate !== "function") {
+        return undefined;
+    }
+    if (!vercelAiSdkPatchState.originalToolFactory) {
+        vercelAiSdkPatchState.originalToolFactory = candidate;
+    }
+    return vercelAiSdkPatchState.originalToolFactory;
+}
+export function recordToolResultNonBlocking(gatewayClient, runId, output) {
+    void gatewayClient.recordResult({ runId, output }).catch(() => undefined);
+}
+export function createWrappedExecute(originalExecute, description, gatewayClient, options) {
+    return async function wrappedExecute(args, executionOptions) {
+        const runId = executionOptions.toolCallId ?? options.fallbackRunId;
+        const executeOriginal = async () => {
+            const run = async () => {
+                const result = await originalExecute(args, executionOptions);
+                recordToolResultNonBlocking(gatewayClient, runId, result);
+                return result;
+            };
+            return options.agentId ? runWithAgentId(options.agentId, run) : run();
+        };
+        let decision;
+        try {
+            decision = await gatewayClient.check({
+                action: "tool_call",
+                toolName: description,
+                args: args,
+                runId
+            });
+        }
+        catch {
+            return executeOriginal();
+        }
+        if (decision.denied) {
+            throw new PolicyViolationError(`Tool blocked by governance policy: ${decision.reason ?? "Denied"}`);
+        }
+        if (decision.pending) {
+            let approval;
+            try {
+                approval = await gatewayClient.waitForApproval(description, runId, options.approvalTimeoutMs);
+            }
+            catch {
+                return executeOriginal();
+            }
+            if (approval.denied) {
+                throw new PolicyViolationError(`Approval rejected: ${approval.reason ?? "Rejected"}`);
+            }
+        }
+        return executeOriginal();
+    };
+}
+export function createPatchedToolFactory(originalToolFactory, gatewayClient, options) {
+    return function patchedTool(definition) {
+        const toolResult = originalToolFactory(definition);
+        if (!toolResult.execute) {
+            return toolResult;
+        }
+        const description = toolResult.description ?? "unknown_tool";
+        return {
+            ...toolResult,
+            execute: createWrappedExecute(toolResult.execute, description, gatewayClient, options)
+        };
+    };
+}
+async function loadVercelAiSdkModule() {
+    try {
+        const moduleName = "ai";
+        const module = (await import(moduleName));
+        return module;
+    }
+    catch {
+        return undefined;
+    }
+}
+export async function patchVercelAiSdk(options) {
+    if (vercelAiSdkPatchState.isPatched) {
+        return true;
+    }
+    const loadModule = options.loadModule ?? loadVercelAiSdkModule;
+    const module = await loadModule();
+    if (!module) {
+        return false;
+    }
+    const originalToolFactory = captureOriginalToolFactory(module);
+    if (!originalToolFactory) {
+        return false;
+    }
+    module.tool = createPatchedToolFactory(originalToolFactory, options.gatewayClient, {
+        approvalTimeoutMs: options.approvalTimeoutMs ?? 30_000,
+        fallbackRunId: options.fallbackRunId ?? "vercel-ai-sdk",
+        ...(options.agentId !== undefined ? { agentId: options.agentId } : {})
+    });
+    vercelAiSdkPatchState.isPatched = true;
+    vercelAiSdkPatchState.patchedModule = module;
+    return true;
+}
+export function unpatchVercelAiSdk() {
+    if (!vercelAiSdkPatchState.isPatched) {
+        return false;
+    }
+    if (!vercelAiSdkPatchState.patchedModule) {
+        return false;
+    }
+    if (!vercelAiSdkPatchState.originalToolFactory) {
+        return false;
+    }
+    vercelAiSdkPatchState.patchedModule.tool =
+        vercelAiSdkPatchState.originalToolFactory;
+    vercelAiSdkPatchState.isPatched = false;
+    vercelAiSdkPatchState.patchedModule = undefined;
+    return true;
+}
+//# sourceMappingURL=ai-sdk.js.map

package/dist/esm/hooks/ai-sdk.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-sdk.js","sourceRoot":"","sources":["../../../src/hooks/ai-sdk.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,oBAAoB,EAAE,MAAM,qCAAqC,CAAC;AAC3E,OAAO,EAAE,cAAc,EAAE,MAAM,mCAAmC,CAAC;AAYnE,MAAM,CAAC,MAAM,qBAAqB,GAA0B;IAC1D,SAAS,EAAE,KAAK;IAChB,mBAAmB,EAAE,SAAS;IAC9B,aAAa,EAAE,SAAS;CACzB,CAAC;AAEF,MAAM,UAAU,0BAA0B,CACxC,MAAyB;IAEzB,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC;IAC9B,IAAI,OAAO,SAAS,KAAK,UAAU,EAAE,CAAC;QACpC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,CAAC,qBAAqB,CAAC,mBAAmB,EAAE,CAAC;QAC/C,qBAAqB,CAAC,mBAAmB,GAAG,SAAS,CAAC;IACxD,CAAC;IAED,OAAO,qBAAqB,CAAC,mBAAmB,CAAC;AACnD,CAAC;AASD,MAAM,UAAU,2BAA2B,CACzC,aAA4B,EAC5B,KAAa,EACb,MAAe;IAEf,KAAK,aAAa,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;AAC5E,CAAC;AAED,MAAM,UAAU,oBAAoB,CAClC,eAAyF,EACzF,WAAmB,EACnB,aAA4B,EAC5B,OAAoC;IAEpC,OAAO,KAAK,UAAU,cAAc,CAClC,IAAW,EACX,gBAA8C;QAE9C,MAAM,KAAK,GAAG,gBAAgB,CAAC,UAAU,IAAI,OAAO,CAAC,aAAa,CAAC;QAEnE,MAAM,eAAe,GAAG,KAAK,IAAsB,EAAE;YACnD,MAAM,GAAG,GAAG,KAAK,IAAsB,EAAE;gBACvC,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;gBAC7D,2BAA2B,CAAC,aAAa,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;gBAC1D,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC;YACF,OAAO,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;QACxE,CAAC,CAAC;QAEF,IAAI,QAAQ,CAAC;QACb,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,aAAa,CAAC,KAAK,CAAC;gBACnC,MAAM,EAAE,WAAW;gBACnB,QAAQ,EAAE,WAAW;gBACrB,IAAI,EAAE,IAA+B;gBACrC,KAAK;aACN,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,eAAe,EAAE,CAAC;QAC3B,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;YACpB,MAAM,IAAI,oBAAoB,CAC5B,sCAAsC,QAAQ,CAAC,MAAM,IAAI,QAAQ,EAAE,CACpE,CAAC;QACJ,CAAC;QAED,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACrB,IAAI,QAAQ,CAAC;YACb,IAAI,CAAC;gBACH,QAAQ,GAAG,MAAM,aAAa,CAAC,eAAe,CAC5C,WAAW,EACX,KAAK,EACL,OAAO,CAAC,iBAAiB,CAC1B,CAAC;YACJ,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,eAAe,EAAE,CAAC;YAC3B,CAAC;YACD,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;gBACpB,MAAM,IAAI,oBAAoB,CAC5B,sBAAsB,QAAQ,CAAC,MAAM,IAAI,UAAU,EAAE,CACtD,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO,eAAe,EAAE,CAAC;IAC3B,CAAC,CAAC;AACJ,CAAC;AAQD,MAAM,UAAU,wBAAwB,CACtC,mBAAwC,EACxC,aAA4B,EAC5B,OAAwC;IAExC,OAAO,SAAS,WAAW,CACzB,UAAkD;QAElD,MAAM,UAAU,GAAG,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAEnD,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YACxB,OAAO,UAAU,CAAC;QACpB,CAAC;QAED,MAAM,WAAW,GAAG,UAAU,CAAC,WAAW,IAAI,cAAc,CAAC;QAE7D,OAAO;YACL,GAAG,UAAU;YACb,OAAO,EAAE,oBAAoB,CAC3B,UAAU,CAAC,OAAO,EAClB,WAAW,EACX,aAAa,EACb,OAAO,CACR;SACF,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAUD,KAAK,UAAU,qBAAqB;IAClC,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,IAAI,CAAC;QACxB,MAAM,MAAM,GAAG,CAAC,MAAM,MAAM,CAAC,UAAU,CAAC,CAAsB,CAAC;QAC/D,OAAO,MAAM,CAAC;IAChB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,OAAgC;IAEhC,IAAI,qBAAqB,CAAC,SAAS,EAAE,CAAC;QACpC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,qBAAqB,CAAC;IAC/D,MAAM,MAAM,GAAG,MAAM,UAAU,EAAE,CAAC;IAClC,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,mBAAmB,GAAG,0BAA0B,CAAC,MAAM,CAAC,CAAC;IAC/D,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,CAAC,IAAI,GAAG,wBAAwB,CACpC,mBAAmB,EACnB,OAAO,CAAC,aAAa,EACrB;QACE,iBAAiB,EAAE,OAAO,CAAC,iBAAiB,IAAI,MAAM;QACtD,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,eAAe;QACvD,GAAG,CAAC,OAAO,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACvE,CACF,CAAC;IACF,qBAAqB,CAAC,SAAS,GAAG,IAAI,CAAC;IACvC,qBAAqB,CAAC,aAAa,GAAG,MAAM,CAAC;IAC7C,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,kBAAkB;IAChC,IAAI,CAAC,qBAAqB,CAAC,SAAS,EAAE,CAAC;QACrC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,CAAC,qBAAqB,CAAC,aAAa,EAAE,CAAC;QACzC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,CAAC,qBAAqB,CAAC,mBAAmB,EAAE,CAAC;QAC/C,OAAO,KAAK,CAAC;IACf,CAAC;IAED,qBAAqB,CAAC,aAAa,CAAC,IAAI;QACtC,qBAAqB,CAAC,mBAAmB,CAAC;IAC5C,qBAAqB,CAAC,SAAS,GAAG,KAAK,CAAC;IACxC,qBAAqB,CAAC,aAAa,GAAG,SAAS,CAAC;IAChD,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/esm/hooks/langchain.js

@@ -0,0 +1,7 @@
+export async function patchLangChain(client) {
+    if (client.mode === "grpc-sidecar" || client.mode === "napi-inprocess") {
+        return false;
+    }
+    return false;
+}
+//# sourceMappingURL=langchain.js.map

package/dist/esm/hooks/langchain.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"langchain.js","sourceRoot":"","sources":["../../../src/hooks/langchain.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,MAAoB;IACvD,IAAI,MAAM,CAAC,IAAI,KAAK,cAAc,IAAI,MAAM,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;QACvE,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC"}

package/dist/esm/hooks/langgraph-detection.js

@@ -0,0 +1,12 @@
+import { createRequire } from "node:module";
+const requireFromCwd = createRequire(`${process.cwd()}/`);
+export function hasLangGraph() {
+    try {
+        requireFromCwd.resolve("@langchain/langgraph");
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+//# sourceMappingURL=langgraph-detection.js.map

package/dist/esm/hooks/langgraph-detection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"langgraph-detection.js","sourceRoot":"","sources":["../../../src/hooks/langgraph-detection.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAE5C,MAAM,cAAc,GAAG,aAAa,CAAC,GAAG,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;AAE1D,MAAM,UAAU,YAAY;IAC1B,IAAI,CAAC;QACH,cAAc,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC;QAC/C,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC"}

package/dist/esm/hooks/langgraph.js

@@ -0,0 +1,69 @@
+import { runWithAgentId } from "../lineage/agent-context-store.js";
+export const langGraphPatchState = {
+    isPatched: false,
+    originalCompile: undefined,
+    patchedClass: undefined
+};
+export function wrapCompiledGraph(compiled, agentId) {
+    const originalInvoke = compiled.invoke.bind(compiled);
+    const originalStream = compiled.stream.bind(compiled);
+    compiled.invoke = (...args) => runWithAgentId(agentId, () => originalInvoke(...args));
+    compiled.stream = (...args) => runWithAgentId(agentId, () => originalStream(...args));
+    return compiled;
+}
+async function loadLangGraphModule() {
+    try {
+        const moduleName = "@langchain/langgraph";
+        const module = (await import(moduleName));
+        return module;
+    }
+    catch {
+        return undefined;
+    }
+}
+export async function patchLangGraph(options) {
+    if (langGraphPatchState.isPatched) {
+        return true;
+    }
+    const loadModule = options.loadModule ?? loadLangGraphModule;
+    let module;
+    try {
+        module = await loadModule();
+    }
+    catch {
+        return false;
+    }
+    if (!module?.StateGraph?.prototype?.compile) {
+        return false;
+    }
+    const originalCompile = module.StateGraph.prototype.compile;
+    langGraphPatchState.originalCompile = originalCompile;
+    langGraphPatchState.patchedClass = module.StateGraph;
+    const { agentId } = options;
+    module.StateGraph.prototype.compile = function patchedCompile(...args) {
+        const compiled = originalCompile.apply(this, args);
+        try {
+            return wrapCompiledGraph(compiled, agentId);
+        }
+        catch (e) {
+            console.warn("[assembly] LangGraph lineage patch error on compile; falling back:", e);
+            return compiled;
+        }
+    };
+    langGraphPatchState.isPatched = true;
+    return true;
+}
+export function unpatchLangGraph() {
+    if (!langGraphPatchState.isPatched) {
+        return false;
+    }
+    if (!langGraphPatchState.patchedClass || !langGraphPatchState.originalCompile) {
+        return false;
+    }
+    langGraphPatchState.patchedClass.prototype.compile = langGraphPatchState.originalCompile;
+    langGraphPatchState.isPatched = false;
+    langGraphPatchState.originalCompile = undefined;
+    langGraphPatchState.patchedClass = undefined;
+    return true;
+}
+//# sourceMappingURL=langgraph.js.map

package/dist/esm/hooks/langgraph.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"langgraph.js","sourceRoot":"","sources":["../../../src/hooks/langgraph.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,mCAAmC,CAAC;AAuBnE,MAAM,CAAC,MAAM,mBAAmB,GAAwB;IACtD,SAAS,EAAE,KAAK;IAChB,eAAe,EAAE,SAAS;IAC1B,YAAY,EAAE,SAAS;CACxB,CAAC;AAEF,MAAM,UAAU,iBAAiB,CAAC,QAAuB,EAAE,OAAe;IACxE,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACtD,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAEtD,QAAQ,CAAC,MAAM,GAAG,CAAC,GAAG,IAAe,EAAE,EAAE,CACvC,cAAc,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;IAEzD,QAAQ,CAAC,MAAM,GAAG,CAAC,GAAG,IAAe,EAAE,EAAE,CACvC,cAAc,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;IAEzD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAOD,KAAK,UAAU,mBAAmB;IAChC,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,sBAAsB,CAAC;QAC1C,MAAM,MAAM,GAAG,CAAC,MAAM,MAAM,CAAC,UAAU,CAAC,CAAoB,CAAC;QAC7D,OAAO,MAAM,CAAC;IAChB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,OAA8B;IACjE,IAAI,mBAAmB,CAAC,SAAS,EAAE,CAAC;QAClC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,mBAAmB,CAAC;IAC7D,IAAI,MAAmC,CAAC;IACxC,IAAI,CAAC;QACH,MAAM,GAAG,MAAM,UAAU,EAAE,CAAC;IAC9B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,CAAC,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC;QAC5C,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,eAAe,GAAG,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,OAAO,CAAC;IAC5D,mBAAmB,CAAC,eAAe,GAAG,eAAe,CAAC;IACtD,mBAAmB,CAAC,YAAY,GAAG,MAAM,CAAC,UAAU,CAAC;IAErD,MAAM,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;IAC5B,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,OAAO,GAAG,SAAS,cAAc,CAC3D,GAAG,IAAe;QAElB,MAAM,QAAQ,GAAG,eAAe,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAEnD,IAAI,CAAC;YACH,OAAO,iBAAiB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAC9C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,IAAI,CAAC,oEAAoE,EAAE,CAAC,CAAC,CAAC;YACtF,OAAO,QAAQ,CAAC;QAClB,CAAC;IACH,CAAC,CAAC;IAEF,mBAAmB,CAAC,SAAS,GAAG,IAAI,CAAC;IACrC,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,gBAAgB;IAC9B,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,CAAC;QACnC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,CAAC,mBAAmB,CAAC,YAAY,IAAI,CAAC,mBAAmB,CAAC,eAAe,EAAE,CAAC;QAC9E,OAAO,KAAK,CAAC;IACf,CAAC;IAED,mBAAmB,CAAC,YAAY,CAAC,SAAS,CAAC,OAAO,GAAG,mBAAmB,CAAC,eAAe,CAAC;IACzF,mBAAmB,CAAC,SAAS,GAAG,KAAK,CAAC;IACtC,mBAAmB,CAAC,eAAe,GAAG,SAAS,CAAC;IAChD,mBAAmB,CAAC,YAAY,GAAG,SAAS,CAAC;IAC7C,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/esm/hooks/mastra-detection.js

@@ -0,0 +1,12 @@
+import { createRequire } from "node:module";
+const requireFromCwd = createRequire(`${process.cwd()}/`);
+export function hasMastra() {
+    try {
+        requireFromCwd.resolve("@mastra/core");
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+//# sourceMappingURL=mastra-detection.js.map

package/dist/esm/hooks/mastra-detection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mastra-detection.js","sourceRoot":"","sources":["../../../src/hooks/mastra-detection.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAE5C,MAAM,cAAc,GAAG,aAAa,CAAC,GAAG,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;AAE1D,MAAM,UAAU,SAAS;IACvB,IAAI,CAAC;QACH,cAAc,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACvC,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC"}

package/dist/esm/hooks/mastra.js

@@ -0,0 +1,83 @@
+import { runWithAgentId } from "../lineage/agent-context-store.js";
+export const mastraPatchState = {
+    isPatched: false,
+    originalGenerate: undefined,
+    originalExecute: undefined,
+    patchedAgentClass: undefined,
+    patchedWorkflowClass: undefined
+};
+async function loadMastraModule() {
+    try {
+        const moduleName = "@mastra/core";
+        const module = (await import(moduleName));
+        return module;
+    }
+    catch {
+        return undefined;
+    }
+}
+export async function patchMastra(options) {
+    if (mastraPatchState.isPatched) {
+        return true;
+    }
+    const loadModule = options.loadModule ?? loadMastraModule;
+    let module;
+    try {
+        module = await loadModule();
+    }
+    catch {
+        return false;
+    }
+    if (!module?.Agent?.prototype?.generate) {
+        return false;
+    }
+    const { agentId } = options;
+    // Wrap Agent.prototype.generate
+    const originalGenerate = module.Agent.prototype.generate;
+    mastraPatchState.originalGenerate = originalGenerate;
+    mastraPatchState.patchedAgentClass = module.Agent;
+    module.Agent.prototype.generate = function patchedGenerate(...args) {
+        try {
+            return runWithAgentId(agentId, () => originalGenerate.apply(this, args));
+        }
+        catch (e) {
+            console.warn("[assembly] Mastra lineage patch error on generate; falling back:", e);
+            return originalGenerate.apply(this, args);
+        }
+    };
+    // Wrap Workflow.prototype.execute if present
+    if (module.Workflow?.prototype?.execute) {
+        const originalExecute = module.Workflow.prototype.execute;
+        mastraPatchState.originalExecute = originalExecute;
+        mastraPatchState.patchedWorkflowClass = module.Workflow;
+        module.Workflow.prototype.execute = function patchedExecute(...args) {
+            try {
+                return runWithAgentId(agentId, () => originalExecute.apply(this, args));
+            }
+            catch (e) {
+                console.warn("[assembly] Mastra lineage patch error on execute; falling back:", e);
+                return originalExecute.apply(this, args);
+            }
+        };
+    }
+    mastraPatchState.isPatched = true;
+    return true;
+}
+export function unpatchMastra() {
+    if (!mastraPatchState.isPatched) {
+        return false;
+    }
+    if (mastraPatchState.patchedAgentClass && mastraPatchState.originalGenerate) {
+        mastraPatchState.patchedAgentClass.prototype.generate = mastraPatchState.originalGenerate;
+    }
+    if (mastraPatchState.patchedWorkflowClass && mastraPatchState.originalExecute) {
+        mastraPatchState.patchedWorkflowClass.prototype.execute = mastraPatchState.originalExecute;
+    }
+    mastraPatchState.isPatched = false;
+    mastraPatchState.originalGenerate = undefined;
+    mastraPatchState.originalExecute = undefined;
+    mastraPatchState.patchedAgentClass = undefined;
+    mastraPatchState.patchedWorkflowClass = undefined;
+    return true;
+}
+//# sourceMappingURL=mastra.js.map

package/dist/esm/hooks/mastra.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mastra.js","sourceRoot":"","sources":["../../../src/hooks/mastra.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,mCAAmC,CAAC;AA2BnE,MAAM,CAAC,MAAM,gBAAgB,GAAqB;IAChD,SAAS,EAAE,KAAK;IAChB,gBAAgB,EAAE,SAAS;IAC3B,eAAe,EAAE,SAAS;IAC1B,iBAAiB,EAAE,SAAS;IAC5B,oBAAoB,EAAE,SAAS;CAChC,CAAC;AAOF,KAAK,UAAU,gBAAgB;IAC7B,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,cAAc,CAAC;QAClC,MAAM,MAAM,GAAG,CAAC,MAAM,MAAM,CAAC,UAAU,CAAC,CAAiB,CAAC;QAC1D,OAAO,MAAM,CAAC;IAChB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,OAA2B;IAC3D,IAAI,gBAAgB,CAAC,SAAS,EAAE,CAAC;QAC/B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,gBAAgB,CAAC;IAC1D,IAAI,MAAgC,CAAC;IACrC,IAAI,CAAC;QACH,MAAM,GAAG,MAAM,UAAU,EAAE,CAAC;IAC9B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC;QACxC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;IAE5B,gCAAgC;IAChC,MAAM,gBAAgB,GAAG,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,QAAQ,CAAC;IACzD,gBAAgB,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;IACrD,gBAAgB,CAAC,iBAAiB,GAAG,MAAM,CAAC,KAAK,CAAC;IAElD,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,QAAQ,GAAG,SAAS,eAAe,CACxD,GAAG,IAAe;QAElB,IAAI,CAAC;YACH,OAAO,cAAc,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,gBAAgB,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;QAC3E,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,IAAI,CAAC,kEAAkE,EAAE,CAAC,CAAC,CAAC;YACpF,OAAO,gBAAgB,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC,CAAC;IAEF,6CAA6C;IAC7C,IAAI,MAAM,CAAC,QAAQ,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC;QACxC,MAAM,eAAe,GAAG,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC;QAC1D,gBAAgB,CAAC,eAAe,GAAG,eAAe,CAAC;QACnD,gBAAgB,CAAC,oBAAoB,GAAG,MAAM,CAAC,QAAQ,CAAC;QAExD,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,GAAG,SAAS,cAAc,CACzD,GAAG,IAAe;YAElB,IAAI,CAAC;gBACH,OAAO,cAAc,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;YAC1E,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,OAAO,CAAC,IAAI,CAAC,iEAAiE,EAAE,CAAC,CAAC,CAAC;gBACnF,OAAO,eAAe,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC,CAAC;IACJ,CAAC;IAED,gBAAgB,CAAC,SAAS,GAAG,IAAI,CAAC;IAClC,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,aAAa;IAC3B,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,CAAC;QAChC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,gBAAgB,CAAC,iBAAiB,IAAI,gBAAgB,CAAC,gBAAgB,EAAE,CAAC;QAC5E,gBAAgB,CAAC,iBAAiB,CAAC,SAAS,CAAC,QAAQ,GAAG,gBAAgB,CAAC,gBAAgB,CAAC;IAC5F,CAAC;IACD,IAAI,gBAAgB,CAAC,oBAAoB,IAAI,gBAAgB,CAAC,eAAe,EAAE,CAAC;QAC9E,gBAAgB,CAAC,oBAAoB,CAAC,SAAS,CAAC,OAAO,GAAG,gBAAgB,CAAC,eAAe,CAAC;IAC7F,CAAC;IAED,gBAAgB,CAAC,SAAS,GAAG,KAAK,CAAC;IACnC,gBAAgB,CAAC,gBAAgB,GAAG,SAAS,CAAC;IAC9C,gBAAgB,CAAC,eAAe,GAAG,SAAS,CAAC;IAC7C,gBAAgB,CAAC,iBAAiB,GAAG,SAAS,CAAC;IAC/C,gBAAgB,CAAC,oBAAoB,GAAG,SAAS,CAAC;IAClD,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/esm/hooks/openai-agents-detection.js

@@ -0,0 +1,12 @@
+import { createRequire } from "node:module";
+const requireFromCwd = createRequire(`${process.cwd()}/`);
+export function hasOpenAIAgentsSDK() {
+    try {
+        requireFromCwd.resolve("@openai/agents");
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+//# sourceMappingURL=openai-agents-detection.js.map

package/dist/esm/hooks/openai-agents-detection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-agents-detection.js","sourceRoot":"","sources":["../../../src/hooks/openai-agents-detection.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAE5C,MAAM,cAAc,GAAG,aAAa,CAAC,GAAG,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;AAE1D,MAAM,UAAU,kBAAkB;IAChC,IAAI,CAAC;QACH,cAAc,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QACzC,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC"}

package/dist/esm/hooks/openai-agents.js

@@ -0,0 +1,145 @@
+import { hasOpenAIAgentsSDK } from "./openai-agents-detection.js";
+export const openAIAgentsPatchState = {
+    isPatched: false,
+    originalRunTool: undefined,
+    patchedAgentClass: undefined
+};
+export function captureOriginalRunTool(agentClass) {
+    const candidate = agentClass.prototype._runTool;
+    if (!candidate) {
+        return undefined;
+    }
+    if (!openAIAgentsPatchState.originalRunTool) {
+        openAIAgentsPatchState.originalRunTool = candidate;
+    }
+    return openAIAgentsPatchState.originalRunTool;
+}
+export function parseToolCallArguments(toolCall) {
+    try {
+        return JSON.parse(toolCall.function.arguments);
+    }
+    catch {
+        return toolCall.function.arguments;
+    }
+}
+export function extractToolCallContextMetadata(context) {
+    return {
+        agentId: context?.agentId ?? undefined,
+        runId: context?.runId ?? undefined
+    };
+}
+export function formatDeniedToolCallOutput(reason, prefix) {
+    const detail = reason?.trim() ? reason : "denied";
+    return { error: `${prefix}: ${detail}` };
+}
+export async function handlePendingApproval(gatewayClient, options) {
+    const approval = await gatewayClient.waitForApproval(options.toolName, options.runId, options.timeoutMs);
+    if (!approval.denied) {
+        return undefined;
+    }
+    return formatDeniedToolCallOutput(approval.reason, "Approval denied");
+}
+export function recordToolResultNonBlocking(gatewayClient, runId, output) {
+    void gatewayClient.recordResult({ runId, output }).catch(() => undefined);
+}
+export function createPatchedRunTool(originalRunTool, gatewayClient, options) {
+    return async function patchedRunTool(toolCall, context) {
+        const toolName = toolCall.function.name;
+        const args = parseToolCallArguments(toolCall);
+        const metadata = extractToolCallContextMetadata(context);
+        const runId = metadata.runId ?? options.fallbackRunId;
+        const executeOriginal = async () => {
+            const result = await originalRunTool.call(this, toolCall, context);
+            recordToolResultNonBlocking(gatewayClient, runId, {
+                toolName,
+                args,
+                result,
+                agentId: metadata.agentId
+            });
+            return result;
+        };
+        let decision;
+        try {
+            decision = await gatewayClient.check({
+                action: "tool_call",
+                toolName,
+                args,
+                runId
+            });
+        }
+        catch {
+            return executeOriginal();
+        }
+        if (decision.denied) {
+            return formatDeniedToolCallOutput(decision.reason, "Blocked by governance policy");
+        }
+        if (decision.pending) {
+            let pendingOutput;
+            try {
+                pendingOutput = await handlePendingApproval(gatewayClient, {
+                    toolName,
+                    runId,
+                    timeoutMs: options.approvalTimeoutMs
+                });
+            }
+            catch {
+                return executeOriginal();
+            }
+            if (pendingOutput) {
+                return pendingOutput;
+            }
+        }
+        return executeOriginal();
+    };
+}
+async function loadAgentClassFromSdk() {
+    if (!hasOpenAIAgentsSDK()) {
+        return undefined;
+    }
+    try {
+        const moduleName = "@openai/agents";
+        const module = (await import(moduleName));
+        return module.Agent;
+    }
+    catch {
+        return undefined;
+    }
+}
+export async function patchOpenAIAgents(options) {
+    if (openAIAgentsPatchState.isPatched) {
+        return true;
+    }
+    const loadAgentClass = options.loadAgentClass ?? loadAgentClassFromSdk;
+    const agentClass = await loadAgentClass();
+    if (!agentClass) {
+        return false;
+    }
+    const originalRunTool = captureOriginalRunTool(agentClass);
+    if (!originalRunTool) {
+        return false;
+    }
+    agentClass.prototype._runTool = createPatchedRunTool(originalRunTool, options.gatewayClient, {
+        approvalTimeoutMs: options.approvalTimeoutMs ?? 30_000,
+        fallbackRunId: options.fallbackRunId ?? "openai-agents"
+    });
+    openAIAgentsPatchState.isPatched = true;
+    openAIAgentsPatchState.patchedAgentClass = agentClass;
+    return true;
+}
+export function unpatchOpenAIAgents() {
+    if (!openAIAgentsPatchState.isPatched) {
+        return false;
+    }
+    if (!openAIAgentsPatchState.patchedAgentClass) {
+        return false;
+    }
+    if (!openAIAgentsPatchState.originalRunTool) {
+        return false;
+    }
+    openAIAgentsPatchState.patchedAgentClass.prototype._runTool =
+        openAIAgentsPatchState.originalRunTool;
+    openAIAgentsPatchState.isPatched = false;
+    openAIAgentsPatchState.patchedAgentClass = undefined;
+    return true;
+}
+//# sourceMappingURL=openai-agents.js.map

package/dist/esm/hooks/openai-agents.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-agents.js","sourceRoot":"","sources":["../../../src/hooks/openai-agents.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AAclE,MAAM,CAAC,MAAM,sBAAsB,GAA2B;IAC5D,SAAS,EAAE,KAAK;IAChB,eAAe,EAAE,SAAS;IAC1B,iBAAiB,EAAE,SAAS;CAC7B,CAAC;AAEF,MAAM,UAAU,sBAAsB,CACpC,UAAkC;IAElC,MAAM,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC,QAAQ,CAAC;IAChD,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,CAAC,sBAAsB,CAAC,eAAe,EAAE,CAAC;QAC5C,sBAAsB,CAAC,eAAe,GAAG,SAAS,CAAC;IACrD,CAAC;IAED,OAAO,sBAAsB,CAAC,eAAe,CAAC;AAChD,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,QAA8B;IACnE,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IACjD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;IACrC,CAAC;AACH,CAAC;AAOD,MAAM,UAAU,8BAA8B,CAC5C,OAA2C;IAE3C,OAAO;QACL,OAAO,EAAE,OAAO,EAAE,OAAO,IAAI,SAAS;QACtC,KAAK,EAAE,OAAO,EAAE,KAAK,IAAI,SAAS;KACnC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,MAA0B,EAC1B,MAAc;IAEd,MAAM,MAAM,GAAG,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC;IAClD,OAAO,EAAE,KAAK,EAAE,GAAG,MAAM,KAAK,MAAM,EAAE,EAAE,CAAC;AAC3C,CAAC;AAQD,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,aAA4B,EAC5B,OAA6B;IAE7B,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,eAAe,CAClD,OAAO,CAAC,QAAQ,EAChB,OAAO,CAAC,KAAK,EACb,OAAO,CAAC,SAAS,CAClB,CAAC;IAEF,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;QACrB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,0BAA0B,CAAC,QAAQ,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC;AACxE,CAAC;AAED,MAAM,UAAU,2BAA2B,CACzC,aAA4B,EAC5B,KAAa,EACb,MAAe;IAEf,KAAK,aAAa,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;AAC5E,CAAC;AAOD,MAAM,UAAU,oBAAoB,CAClC,eAAoC,EACpC,aAA4B,EAC5B,OAAoC;IAEpC,OAAO,KAAK,UAAU,cAAc,CAElC,QAA8B,EAC9B,OAA+B;QAE/B,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC;QACxC,MAAM,IAAI,GAAG,sBAAsB,CAAC,QAAQ,CAAC,CAAC;QAC9C,MAAM,QAAQ,GAAG,8BAA8B,CAAC,OAAO,CAAC,CAAC;QACzD,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,IAAI,OAAO,CAAC,aAAa,CAAC;QAEtD,MAAM,eAAe,GAAG,KAAK,IAAyC,EAAE;YACtE,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,IAAI,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;YACnE,2BAA2B,CAAC,aAAa,EAAE,KAAK,EAAE;gBAChD,QAAQ;gBACR,IAAI;gBACJ,MAAM;gBACN,OAAO,EAAE,QAAQ,CAAC,OAAO;aAC1B,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC;QAEF,IAAI,QAAQ,CAAC;QACb,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,aAAa,CAAC,KAAK,CAAC;gBACnC,MAAM,EAAE,WAAW;gBACnB,QAAQ;gBACR,IAAI;gBACJ,KAAK;aACN,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,eAAe,EAAE,CAAC;QAC3B,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;YACpB,OAAO,0BAA0B,CAC/B,QAAQ,CAAC,MAAM,EACf,8BAA8B,CAC/B,CAAC;QACJ,CAAC;QAED,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACrB,IAAI,aAAa,CAAC;YAClB,IAAI,CAAC;gBACH,aAAa,GAAG,MAAM,qBAAqB,CAAC,aAAa,EAAE;oBACzD,QAAQ;oBACR,KAAK;oBACL,SAAS,EAAE,OAAO,CAAC,iBAAiB;iBACrC,CAAC,CAAC;YACL,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,eAAe,EAAE,CAAC;YAC3B,CAAC;YACD,IAAI,aAAa,EAAE,CAAC;gBAClB,OAAO,aAAa,CAAC;YACvB,CAAC;QACH,CAAC;QAED,OAAO,eAAe,EAAE,CAAC;IAC3B,CAAC,CAAC;AACJ,CAAC;AASD,KAAK,UAAU,qBAAqB;IAClC,IAAI,CAAC,kBAAkB,EAAE,EAAE,CAAC;QAC1B,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,gBAAgB,CAAC;QACpC,MAAM,MAAM,GAAG,CAAC,MAAM,MAAM,CAAC,UAAU,CAAC,CAAuC,CAAC;QAChF,OAAO,MAAM,CAAC,KAAK,CAAC;IACtB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,OAAiC;IAEjC,IAAI,sBAAsB,CAAC,SAAS,EAAE,CAAC;QACrC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,cAAc,GAAG,OAAO,CAAC,cAAc,IAAI,qBAAqB,CAAC;IACvE,MAAM,UAAU,GAAG,MAAM,cAAc,EAAE,CAAC;IAC1C,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,eAAe,GAAG,sBAAsB,CAAC,UAAU,CAAC,CAAC;IAC3D,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,UAAU,CAAC,SAAS,CAAC,QAAQ,GAAG,oBAAoB,CAClD,eAAe,EACf,OAAO,CAAC,aAAa,EACrB;QACE,iBAAiB,EAAE,OAAO,CAAC,iBAAiB,IAAI,MAAM;QACtD,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,eAAe;KACxD,CACF,CAAC;IACF,sBAAsB,CAAC,SAAS,GAAG,IAAI,CAAC;IACxC,sBAAsB,CAAC,iBAAiB,GAAG,UAAU,CAAC;IACtD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,mBAAmB;IACjC,IAAI,CAAC,sBAAsB,CAAC,SAAS,EAAE,CAAC;QACtC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,CAAC,sBAAsB,CAAC,iBAAiB,EAAE,CAAC;QAC9C,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,CAAC,sBAAsB,CAAC,eAAe,EAAE,CAAC;QAC5C,OAAO,KAAK,CAAC;IACf,CAAC;IAED,sBAAsB,CAAC,iBAAiB,CAAC,SAAS,CAAC,QAAQ;QACzD,sBAAsB,CAAC,eAAe,CAAC;IACzC,sBAAsB,CAAC,SAAS,GAAG,KAAK,CAAC;IACzC,sBAAsB,CAAC,iBAAiB,GAAG,SAAS,CAAC;IACrD,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/esm/index.js

@@ -0,0 +1,12 @@
+export { initAssembly } from "./core/init-assembly.js";
+export { withAssembly } from "./wrappers/index.js";
+export { ENFORCEMENT_MODES } from "./types/index.js";
+export { decodeAuditEvent, decodeCallStackNode, encodeAuditEvent, encodeCallStackNode } from "./audit/index.js";
+export { currentAgentId, runWithAgentId } from "./lineage/index.js";
+// Runtime binary resolution helpers (F113 / AAASM-1221). Re-export the
+// discovery + error-message surfaces from src/runtime.ts; the lifecycle
+// `initAssembly` from runtime.ts intentionally stays scoped to the
+// `@agent-assembly/sdk/runtime` subpath export to avoid colliding with
+// the gateway-based `initAssembly` re-exported above.
+export { INSTALL_HINT, findAasmBinary } from "./runtime.js";
+//# sourceMappingURL=index.js.map

package/dist/esm/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AACvD,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAYnD,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EACL,gBAAgB,EAChB,mBAAmB,EACnB,gBAAgB,EAChB,mBAAmB,EACpB,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpE,uEAAuE;AACvE,wEAAwE;AACxE,mEAAmE;AACnE,uEAAuE;AACvE,sDAAsD;AACtD,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC"}

package/dist/esm/lineage/agent-context-store.js

@@ -0,0 +1,16 @@
+import { AsyncLocalStorage } from "node:async_hooks";
+/**
+ * Holds the current agent's ID for automatic lineage propagation.
+ * When a framework spawns a child agent, the child's initAssembly reads
+ * this store to auto-populate parentAgentId without manual threading.
+ */
+export const agentContextStore = new AsyncLocalStorage();
+/** Run fn within a context where the current agent ID is set. */
+export function runWithAgentId(agentId, fn) {
+    return agentContextStore.run(agentId, fn);
+}
+/** Return the current agent ID from the async context, or undefined if absent. */
+export function currentAgentId() {
+    return agentContextStore.getStore();
+}
+//# sourceMappingURL=agent-context-store.js.map

package/dist/esm/lineage/agent-context-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-context-store.js","sourceRoot":"","sources":["../../../src/lineage/agent-context-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAErD;;;;GAIG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,IAAI,iBAAiB,EAAU,CAAC;AAEjE,iEAAiE;AACjE,MAAM,UAAU,cAAc,CAAI,OAAe,EAAE,EAAW;IAC5D,OAAO,iBAAiB,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;AAC5C,CAAC;AAED,kFAAkF;AAClF,MAAM,UAAU,cAAc;IAC5B,OAAO,iBAAiB,CAAC,QAAQ,EAAE,CAAC;AACtC,CAAC"}

package/dist/esm/lineage/index.js

@@ -0,0 +1,2 @@
+export { agentContextStore, currentAgentId, runWithAgentId } from "./agent-context-store.js";
+//# sourceMappingURL=index.js.map

package/dist/esm/lineage/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/lineage/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC"}