managed-deepagents 0.6.2-dev.2 → 0.6.2-dev.20
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +13 -3
- package/dist/agent-auth.d.ts +104 -0
- package/dist/agent-auth.d.ts.map +1 -0
- package/dist/agent-auth.js +224 -0
- package/dist/agent-auth.js.map +1 -0
- package/dist/channels/errors.d.ts +0 -1
- package/dist/channels/errors.d.ts.map +1 -1
- package/dist/channels/errors.js +0 -12
- package/dist/channels/errors.js.map +1 -1
- package/dist/channels/manifest.d.ts +2 -0
- package/dist/channels/manifest.d.ts.map +1 -1
- package/dist/channels/messaging-registry.d.ts +0 -2
- package/dist/channels/messaging-registry.d.ts.map +1 -1
- package/dist/channels/messaging-registry.js +0 -6
- package/dist/channels/messaging-registry.js.map +1 -1
- package/dist/channels/registry-types.d.ts +1 -1
- package/dist/channels/schedule-delivery.d.ts +3 -1
- package/dist/channels/schedule-delivery.d.ts.map +1 -1
- package/dist/channels/schedule-delivery.js +41 -5
- package/dist/channels/schedule-delivery.js.map +1 -1
- package/dist/channels/slack/http.d.ts +1 -1
- package/dist/channels/trigger/client.d.ts +1 -0
- package/dist/channels/trigger/client.d.ts.map +1 -1
- package/dist/channels/trigger/client.js +19 -8
- package/dist/channels/trigger/client.js.map +1 -1
- package/dist/channels/trigger/types.d.ts +1 -1
- package/dist/channels/trigger/types.d.ts.map +1 -1
- package/dist/connectors/mcp/index.d.ts +29 -89
- package/dist/connectors/mcp/index.d.ts.map +1 -1
- package/dist/connectors/mcp/index.js +240 -81
- package/dist/connectors/mcp/index.js.map +1 -1
- package/dist/connectors/mcp/schema.d.ts +12 -0
- package/dist/connectors/mcp/schema.d.ts.map +1 -0
- package/dist/connectors/mcp/schema.js +55 -0
- package/dist/connectors/mcp/schema.js.map +1 -0
- package/dist/connectors/mcp/types.d.ts +62 -0
- package/dist/connectors/mcp/types.d.ts.map +1 -0
- package/dist/connectors/mcp/types.js +2 -0
- package/dist/connectors/mcp/types.js.map +1 -0
- package/dist/connectors/tool-selection.d.ts +12 -14
- package/dist/connectors/tool-selection.d.ts.map +1 -1
- package/dist/connectors/tool-selection.js +25 -39
- package/dist/connectors/tool-selection.js.map +1 -1
- package/dist/credential-gate.d.ts +39 -0
- package/dist/credential-gate.d.ts.map +1 -0
- package/dist/credential-gate.js +209 -0
- package/dist/credential-gate.js.map +1 -0
- package/dist/credentials.d.ts +95 -0
- package/dist/credentials.d.ts.map +1 -0
- package/dist/credentials.js +288 -0
- package/dist/credentials.js.map +1 -0
- package/dist/identity/index.d.ts +1 -1
- package/dist/identity/index.d.ts.map +1 -1
- package/dist/identity/index.js.map +1 -1
- package/dist/identity/types.d.ts +79 -5
- package/dist/identity/types.d.ts.map +1 -1
- package/dist/index.d.ts +11 -6
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +1 -0
- package/dist/index.js.map +1 -1
- package/dist/runtime/auth.d.ts +31 -4
- package/dist/runtime/auth.d.ts.map +1 -1
- package/dist/runtime/auth.js +58 -3
- package/dist/runtime/auth.js.map +1 -1
- package/dist/runtime/channel.d.ts +1 -3
- package/dist/runtime/channel.d.ts.map +1 -1
- package/dist/runtime/channel.js +0 -4
- package/dist/runtime/channel.js.map +1 -1
- package/dist/runtime/connector.d.ts +1 -3
- package/dist/runtime/connector.d.ts.map +1 -1
- package/dist/runtime/connector.js +0 -4
- package/dist/runtime/connector.js.map +1 -1
- package/dist/runtime/cors.d.ts +8 -6
- package/dist/runtime/cors.d.ts.map +1 -1
- package/dist/runtime/cors.js +18 -19
- package/dist/runtime/cors.js.map +1 -1
- package/dist/runtime/http.d.ts +1 -1
- package/dist/runtime/http.d.ts.map +1 -1
- package/dist/runtime/http.js.map +1 -1
- package/dist/runtime/identity-http.d.ts.map +1 -1
- package/dist/runtime/identity-http.js +22 -13
- package/dist/runtime/identity-http.js.map +1 -1
- package/dist/runtime/identity-runtime.d.ts +66 -6
- package/dist/runtime/identity-runtime.d.ts.map +1 -1
- package/dist/runtime/identity-runtime.js +149 -1
- package/dist/runtime/identity-runtime.js.map +1 -1
- package/dist/runtime/index.d.ts +3 -8
- package/dist/runtime/index.d.ts.map +1 -1
- package/dist/runtime/index.js +15 -13
- package/dist/runtime/index.js.map +1 -1
- package/dist/runtime/managed-middleware.d.ts +2 -2
- package/dist/runtime/managed-middleware.js +6 -6
- package/dist/runtime/managed-tools.d.ts +19 -15
- package/dist/runtime/managed-tools.d.ts.map +1 -1
- package/dist/runtime/managed-tools.js +14 -13
- package/dist/runtime/managed-tools.js.map +1 -1
- package/dist/runtime/sandbox-manager.d.ts +1 -1
- package/dist/runtime/sandbox-manager.d.ts.map +1 -1
- package/dist/runtime/sandbox-manager.js +1 -1
- package/dist/runtime/sandbox-manager.js.map +1 -1
- package/dist/runtime/types.d.ts +2 -2
- package/package.json +8 -8
- package/dist/connectors/mcp/load.d.ts +0 -12
- package/dist/connectors/mcp/load.d.ts.map +0 -1
- package/dist/connectors/mcp/load.js +0 -122
- package/dist/connectors/mcp/load.js.map +0 -1
|
@@ -8,13 +8,11 @@
|
|
|
8
8
|
* silently dropping a tool the agent was meant to have.
|
|
9
9
|
*/
|
|
10
10
|
import { z } from "zod";
|
|
11
|
-
function nonEmpty(field) {
|
|
12
|
-
const message = `${field} must be a non-empty string`;
|
|
13
|
-
return z.string(message).refine((value) => value.trim().length > 0, message);
|
|
14
|
-
}
|
|
15
11
|
/** Tool names: non-empty, no duplicates within one list. */
|
|
16
|
-
const
|
|
17
|
-
.array(
|
|
12
|
+
export const optionalToolNameListSchema = z
|
|
13
|
+
.array(z
|
|
14
|
+
.string("tool name must be a non-empty string")
|
|
15
|
+
.refine((name) => name.trim().length > 0, "tool name must be a non-empty string"))
|
|
18
16
|
.check((ctx) => {
|
|
19
17
|
const seen = new Set();
|
|
20
18
|
for (const name of ctx.value) {
|
|
@@ -29,39 +27,27 @@ const toolNames = z
|
|
|
29
27
|
}
|
|
30
28
|
})
|
|
31
29
|
.optional();
|
|
32
|
-
|
|
33
|
-
includeTools
|
|
34
|
-
excludeTools
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
ctx.issues.push({
|
|
54
|
-
code: "custom",
|
|
55
|
-
input: ctx.value,
|
|
56
|
-
message: `selects tool(s) in both \`includeTools\` and \`excludeTools\`: ${[
|
|
57
|
-
...new Set(overlap),
|
|
58
|
-
]
|
|
59
|
-
.sort()
|
|
60
|
-
.join(", ")}`,
|
|
61
|
-
});
|
|
62
|
-
}
|
|
63
|
-
})
|
|
64
|
-
.prefault({});
|
|
30
|
+
function toolSelectionOverlapMessage(selection) {
|
|
31
|
+
const { includeTools, excludeTools } = selection;
|
|
32
|
+
const overlap = (includeTools ?? []).filter((name) => (excludeTools ?? []).includes(name));
|
|
33
|
+
if (overlap.length === 0) {
|
|
34
|
+
return undefined;
|
|
35
|
+
}
|
|
36
|
+
return `selects tool(s) in both \`includeTools\` and \`excludeTools\`: ${[
|
|
37
|
+
...new Set(overlap),
|
|
38
|
+
]
|
|
39
|
+
.sort()
|
|
40
|
+
.join(", ")}`;
|
|
41
|
+
}
|
|
42
|
+
export function appendToolSelectionOverlapIssue(ctx) {
|
|
43
|
+
const message = toolSelectionOverlapMessage(ctx.value);
|
|
44
|
+
if (message) {
|
|
45
|
+
ctx.issues.push({
|
|
46
|
+
code: "custom",
|
|
47
|
+
input: ctx.value,
|
|
48
|
+
message,
|
|
49
|
+
});
|
|
50
|
+
}
|
|
65
51
|
}
|
|
66
52
|
/** Read a loaded tool's own `name`. The default for connectors that don't rename. */
|
|
67
53
|
export const toolName = (tool) => {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"tool-selection.js","sourceRoot":"","sources":["../../src/connectors/tool-selection.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;
|
|
1
|
+
{"version":3,"file":"tool-selection.js","sourceRoot":"","sources":["../../src/connectors/tool-selection.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAexB,4DAA4D;AAC5D,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC;KACxC,KAAK,CACJ,CAAC;KACE,MAAM,CAAC,sCAAsC,CAAC;KAC9C,MAAM,CACL,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAChC,sCAAsC,CACvC,CACJ;KACA,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACb,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;QAC7B,IAAI,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YACnB,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC;gBACd,IAAI,EAAE,QAAQ;gBACd,KAAK,EAAE,GAAG,CAAC,KAAK;gBAChB,OAAO,EAAE,iCAAiC,IAAI,GAAG;aAClD,CAAC,CAAC;QACL,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACjB,CAAC;AACH,CAAC,CAAC;KACD,QAAQ,EAAE,CAAC;AAEd,SAAS,2BAA2B,CAClC,SAA+D;IAE/D,MAAM,EAAE,YAAY,EAAE,YAAY,EAAE,GAAG,SAAS,CAAC;IACjD,MAAM,OAAO,GAAG,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CACnD,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CACpC,CAAC;IACF,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,kEAAkE;QACvE,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC;KACpB;SACE,IAAI,EAAE;SACN,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,+BAA+B,CAAC,GAK/C;IACC,MAAM,OAAO,GAAG,2BAA2B,CACzC,GAAG,CAAC,KAA6D,CAClE,CAAC;IACF,IAAI,OAAO,EAAE,CAAC;QACZ,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC;YACd,IAAI,EAAE,QAAQ;YACd,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,OAAO;SACR,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAKD,qFAAqF;AACrF,MAAM,CAAC,MAAM,QAAQ,GAAmB,CAAC,IAAI,EAAE,EAAE;IAC/C,MAAM,IAAI,GAAI,IAA2B,CAAC,IAAI,CAAC;IAC/C,OAAO,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;AACrD,CAAC,CAAC;AAEF;;;;;;;GAOG;AACH,MAAM,UAAU,WAAW,CACzB,OAAe,EACf,SAAwB,EACxB,KAAyB,EACzB,QAAQ,GAAmB,QAAQ;IAEnC,MAAM,EAAE,YAAY,EAAE,YAAY,EAAE,GAAG,SAAS,CAAC;IACjD,IAAI,CAAC,YAAY,IAAI,CAAC,YAAY,EAAE,CAAC;QACnC,OAAO,CAAC,GAAG,KAAK,CAAC,CAAC;IACpB,CAAC;IAED,MAAM,SAAS,GAAG,IAAI,GAAG,CACvB,KAAK;SACF,GAAG,CAAC,QAAQ,CAAC;SACb,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC,CAC9D,CAAC;IACF,wBAAwB,CAAC,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,YAAY,CAAC,CAAC;IAEzE,MAAM,QAAQ,GAAG,YAAY,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAClE,MAAM,QAAQ,GAAG,YAAY,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAClE,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE;QAC3B,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC5B,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,OAAO,QAAQ,KAAK,SAAS,CAAC;QAChC,CAAC;QACD,OAAO,CACL,CAAC,QAAQ,KAAK,SAAS,IAAI,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,CAAC,IAAI,CAAC,CACvE,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,wBAAwB,CAC/B,OAAe,EACf,SAAsB,EACtB,YAAuB,EACvB,YAAuB;IAEvB,MAAM,QAAQ,GAAG,CAAC,GAAG,CAAC,YAAY,IAAI,EAAE,CAAC,EAAE,GAAG,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC,CAAC;IACpE,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;IAChE,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO;IACT,CAAC;IAED,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACtD,MAAM,cAAc,GAAG,CAAC,GAAG,SAAS,CAAC,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,QAAQ,CAAC;IACpE,MAAM,IAAI,KAAK,CACb,GAAG,OAAO,8BAA8B,KAAK,IAAI;QAC/C,oBAAoB,cAAc,GAAG,CACxC,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,39 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Pre-run gate for declarations that need a user-owned credential.
|
|
3
|
+
*
|
|
4
|
+
* Before the first model turn the gate reads the caller's OAuth grants from
|
|
5
|
+
* Agent Auth. Missing grants raise one LangGraph interrupt listing connect
|
|
6
|
+
* URLs; the tools a grant unlocks are loaded on first use and cached per
|
|
7
|
+
* caller.
|
|
8
|
+
*/
|
|
9
|
+
import { createMiddleware } from "langchain";
|
|
10
|
+
import { type CredentialReference } from "./credentials.js";
|
|
11
|
+
export declare const CREDENTIAL_GATE_MIDDLEWARE_NAME = "mda-credential-gate";
|
|
12
|
+
export interface PendingCredential {
|
|
13
|
+
slug: string;
|
|
14
|
+
connect_url: string;
|
|
15
|
+
auth_id: string;
|
|
16
|
+
}
|
|
17
|
+
/** Interrupt payload when the caller has not granted one or more credentials. */
|
|
18
|
+
export interface CredentialAuthorizationRequiredInterrupt {
|
|
19
|
+
type: "credential_authorization_required";
|
|
20
|
+
message: string;
|
|
21
|
+
credentials: PendingCredential[];
|
|
22
|
+
}
|
|
23
|
+
/** Tools a declaration contributes for one caller, plus their transport. */
|
|
24
|
+
export interface LoadedCredentialTools {
|
|
25
|
+
tools: unknown[];
|
|
26
|
+
close: () => Promise<void>;
|
|
27
|
+
}
|
|
28
|
+
/** A declaration whose tools are unlocked by user-owned credentials. */
|
|
29
|
+
export interface CredentialConsumer {
|
|
30
|
+
/** The declaration itself, used to key the per-caller tool cache. */
|
|
31
|
+
declaration: object;
|
|
32
|
+
/** User-owned credential references the declaration needs. */
|
|
33
|
+
credentials: readonly CredentialReference[];
|
|
34
|
+
load: (accessTokens: ReadonlyMap<string, string>) => Promise<LoadedCredentialTools>;
|
|
35
|
+
}
|
|
36
|
+
export declare function createCredentialGateMiddleware(consumers: readonly CredentialConsumer[]): ReturnType<typeof createMiddleware> | undefined;
|
|
37
|
+
/** Drop every cached caller entry and close its transports. Test-only. */
|
|
38
|
+
export declare function clearCredentialToolCacheForTests(): Promise<void>;
|
|
39
|
+
//# sourceMappingURL=credential-gate.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"credential-gate.d.ts","sourceRoot":"","sources":["../src/credential-gate.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAQ7C,OAAO,EAGL,KAAK,mBAAmB,EACzB,MAAM,kBAAkB,CAAC;AAY1B,eAAO,MAAM,+BAA+B,wBAAwB,CAAC;AAUrE,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,iFAAiF;AACjF,MAAM,WAAW,wCAAwC;IACvD,IAAI,EAAE,mCAAmC,CAAC;IAC1C,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,iBAAiB,EAAE,CAAC;CAClC;AAED,4EAA4E;AAC5E,MAAM,WAAW,qBAAqB;IACpC,KAAK,EAAE,OAAO,EAAE,CAAC;IACjB,KAAK,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CAC5B;AAED,wEAAwE;AACxE,MAAM,WAAW,kBAAkB;IACjC,qEAAqE;IACrE,WAAW,EAAE,MAAM,CAAC;IACpB,8DAA8D;IAC9D,WAAW,EAAE,SAAS,mBAAmB,EAAE,CAAC;IAC5C,IAAI,EAAE,CACJ,YAAY,EAAE,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,KACtC,OAAO,CAAC,qBAAqB,CAAC,CAAC;CACrC;AAuCD,wBAAgB,8BAA8B,CAC5C,SAAS,EAAE,SAAS,kBAAkB,EAAE,GACvC,UAAU,CAAC,OAAO,gBAAgB,CAAC,GAAG,SAAS,CAuCjD;AA+KD,0EAA0E;AAC1E,wBAAsB,gCAAgC,IAAI,OAAO,CAAC,IAAI,CAAC,CAStE"}
|
|
@@ -0,0 +1,209 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Pre-run gate for declarations that need a user-owned credential.
|
|
3
|
+
*
|
|
4
|
+
* Before the first model turn the gate reads the caller's OAuth grants from
|
|
5
|
+
* Agent Auth. Missing grants raise one LangGraph interrupt listing connect
|
|
6
|
+
* URLs; the tools a grant unlocks are loaded on first use and cached per
|
|
7
|
+
* caller.
|
|
8
|
+
*/
|
|
9
|
+
import { createMiddleware } from "langchain";
|
|
10
|
+
import { interrupt } from "@langchain/langgraph";
|
|
11
|
+
import { AgentAuth, CredentialAuthorizationRequiredError, } from "./agent-auth.js";
|
|
12
|
+
import { resolveAgentAuthContext, resolveOwnerId, } from "./credentials.js";
|
|
13
|
+
import { buildRuntimeIdentity, resolveRuntimeConfigurable, } from "./runtime/identity-runtime.js";
|
|
14
|
+
import { wrapToolsWithManagedRuntime, } from "./runtime/managed-tools.js";
|
|
15
|
+
export const CREDENTIAL_GATE_MIDDLEWARE_NAME = "mda-credential-gate";
|
|
16
|
+
/**
|
|
17
|
+
* How long a caller's loaded tools stay cached.
|
|
18
|
+
*
|
|
19
|
+
* The access token is bound into the transport when the tools are built, so the
|
|
20
|
+
* entry has to lapse well before the token does.
|
|
21
|
+
*/
|
|
22
|
+
const TOOL_CACHE_TTL_MS = 5 * 60_000;
|
|
23
|
+
/**
|
|
24
|
+
* Loaded tools by declaration and caller.
|
|
25
|
+
*
|
|
26
|
+
* Module scope, not middleware scope: a resumed run re-enters through a fresh
|
|
27
|
+
* graph factory, so a cache owned by the middleware would be empty for every
|
|
28
|
+
* turn after an interrupt. It is only ever a latency optimization — a resume
|
|
29
|
+
* can also land on another replica, so every read path rebuilds on a miss.
|
|
30
|
+
*/
|
|
31
|
+
const toolCache = new Map();
|
|
32
|
+
const declarationIds = new WeakMap();
|
|
33
|
+
let nextDeclarationId = 0;
|
|
34
|
+
class UnauthenticatedCallerError extends Error {
|
|
35
|
+
constructor() {
|
|
36
|
+
super("credentials with a user owner require an authenticated caller. " +
|
|
37
|
+
"Anonymous or agent-only runs cannot complete the credential gate.");
|
|
38
|
+
this.name = "UnauthenticatedCallerError";
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
export function createCredentialGateMiddleware(consumers) {
|
|
42
|
+
if (consumers.length === 0) {
|
|
43
|
+
return undefined;
|
|
44
|
+
}
|
|
45
|
+
return createMiddleware({
|
|
46
|
+
name: CREDENTIAL_GATE_MIDDLEWARE_NAME,
|
|
47
|
+
async beforeAgent(_state, runtime) {
|
|
48
|
+
const caller = authContext(runtime);
|
|
49
|
+
if (!caller) {
|
|
50
|
+
throw new UnauthenticatedCallerError();
|
|
51
|
+
}
|
|
52
|
+
await ensureGrants(consumers, caller);
|
|
53
|
+
await toolsForContext(consumers, caller.context);
|
|
54
|
+
},
|
|
55
|
+
async wrapModelCall(request, handler) {
|
|
56
|
+
const extra = await toolsForRuntime(consumers, request.runtime);
|
|
57
|
+
if (extra.length === 0) {
|
|
58
|
+
return handler(request);
|
|
59
|
+
}
|
|
60
|
+
const existing = Array.isArray(request.tools) ? request.tools : [];
|
|
61
|
+
return handler({ ...request, tools: [...existing, ...extra] });
|
|
62
|
+
},
|
|
63
|
+
async wrapToolCall(request, handler) {
|
|
64
|
+
const name = request.toolCall?.name;
|
|
65
|
+
if (request.tool != null || typeof name !== "string") {
|
|
66
|
+
return handler(request);
|
|
67
|
+
}
|
|
68
|
+
const tools = await toolsForRuntime(consumers, request.runtime);
|
|
69
|
+
const tool = tools.find((candidate) => candidate.name === name);
|
|
70
|
+
return tool ? handler({ ...request, tool }) : handler(request);
|
|
71
|
+
},
|
|
72
|
+
});
|
|
73
|
+
}
|
|
74
|
+
/**
|
|
75
|
+
* Interrupt once when the caller is missing a grant.
|
|
76
|
+
*
|
|
77
|
+
* Reads never mint, so the probe is safe to repeat on every replay of this
|
|
78
|
+
* node. Sessions are minted once, immediately before the interrupt that shows
|
|
79
|
+
* their URLs.
|
|
80
|
+
*/
|
|
81
|
+
async function ensureGrants(consumers, caller) {
|
|
82
|
+
const agentAuth = new AgentAuth(caller.context);
|
|
83
|
+
const granted = await Promise.all(consumerCredentials(consumers, caller).map(async ({ slug }) => ({
|
|
84
|
+
slug,
|
|
85
|
+
token: await agentAuth.readUserAccessToken(slug),
|
|
86
|
+
})));
|
|
87
|
+
const missing = granted.filter((row) => !row.token).map((row) => row.slug);
|
|
88
|
+
if (missing.length === 0) {
|
|
89
|
+
return;
|
|
90
|
+
}
|
|
91
|
+
const pending = await Promise.all(missing.map((slug) => agentAuth.startAuthorizationSession(slug)));
|
|
92
|
+
const payload = {
|
|
93
|
+
type: "credential_authorization_required",
|
|
94
|
+
message: "Connect the following integrations to continue.",
|
|
95
|
+
credentials: pending.map((session) => ({
|
|
96
|
+
slug: session.slug,
|
|
97
|
+
connect_url: session.connectUrl,
|
|
98
|
+
auth_id: session.authId,
|
|
99
|
+
})),
|
|
100
|
+
};
|
|
101
|
+
// `interrupt` throws while no resume value is pending and returns once one
|
|
102
|
+
// is, so reaching the line below means the run resumed with a grant still
|
|
103
|
+
// missing.
|
|
104
|
+
interrupt(payload);
|
|
105
|
+
throw new CredentialAuthorizationRequiredError(missing);
|
|
106
|
+
}
|
|
107
|
+
/**
|
|
108
|
+
* Declared credentials, one per slug.
|
|
109
|
+
*
|
|
110
|
+
* Every reference is resolved through {@link resolveOwnerId}, the same check
|
|
111
|
+
* `credentials.get` applies when a tool awaits a handle, so a declaration that
|
|
112
|
+
* names another user is rejected here rather than quietly run as the caller.
|
|
113
|
+
*/
|
|
114
|
+
function consumerCredentials(consumers, caller) {
|
|
115
|
+
const bySlug = new Map();
|
|
116
|
+
for (const consumer of consumers) {
|
|
117
|
+
for (const credential of consumer.credentials) {
|
|
118
|
+
resolveOwnerId(credential.owner, caller.identity, caller.context.agentId);
|
|
119
|
+
bySlug.set(credential.slug, credential);
|
|
120
|
+
}
|
|
121
|
+
}
|
|
122
|
+
return [...bySlug.values()];
|
|
123
|
+
}
|
|
124
|
+
async function toolsForRuntime(consumers, runtime) {
|
|
125
|
+
const caller = authContext(runtime);
|
|
126
|
+
return caller ? await toolsForContext(consumers, caller.context) : [];
|
|
127
|
+
}
|
|
128
|
+
async function toolsForContext(consumers, context) {
|
|
129
|
+
const loaded = await Promise.all(consumers.map((consumer) => cachedTools(consumer, context)));
|
|
130
|
+
return loaded.flat();
|
|
131
|
+
}
|
|
132
|
+
async function cachedTools(consumer, context) {
|
|
133
|
+
const key = `${declarationId(consumer.declaration)} ${context.principalId}`;
|
|
134
|
+
for (;;) {
|
|
135
|
+
const cached = toolCache.get(key);
|
|
136
|
+
if (!cached) {
|
|
137
|
+
return refresh(key, consumer, context, undefined);
|
|
138
|
+
}
|
|
139
|
+
const entry = await cached.catch(() => undefined);
|
|
140
|
+
if (entry && entry.expiresAt > Date.now()) {
|
|
141
|
+
return entry.tools;
|
|
142
|
+
}
|
|
143
|
+
// Nothing suspends between this read and the write inside `refresh`, so a
|
|
144
|
+
// lapsed entry is retired by exactly one caller. The rest see the
|
|
145
|
+
// replacement on the next pass and join it.
|
|
146
|
+
if (toolCache.get(key) === cached) {
|
|
147
|
+
return refresh(key, consumer, context, entry);
|
|
148
|
+
}
|
|
149
|
+
}
|
|
150
|
+
}
|
|
151
|
+
/** Publish the load before awaiting it, so concurrent callers share one transport. */
|
|
152
|
+
async function refresh(key, consumer, context, lapsed) {
|
|
153
|
+
const loading = loadTools(consumer, context);
|
|
154
|
+
toolCache.set(key, loading);
|
|
155
|
+
lapsed?.close().catch(() => undefined);
|
|
156
|
+
try {
|
|
157
|
+
return (await loading).tools;
|
|
158
|
+
}
|
|
159
|
+
catch (error) {
|
|
160
|
+
if (toolCache.get(key) === loading) {
|
|
161
|
+
toolCache.delete(key);
|
|
162
|
+
}
|
|
163
|
+
throw error;
|
|
164
|
+
}
|
|
165
|
+
}
|
|
166
|
+
async function loadTools(consumer, context) {
|
|
167
|
+
const agentAuth = new AgentAuth(context);
|
|
168
|
+
const resolved = await Promise.all(consumer.credentials.map(async ({ slug }) => ({
|
|
169
|
+
slug,
|
|
170
|
+
token: await agentAuth.readUserAccessToken(slug),
|
|
171
|
+
})));
|
|
172
|
+
const missing = resolved.filter((row) => !row.token).map((row) => row.slug);
|
|
173
|
+
if (missing.length > 0) {
|
|
174
|
+
throw new CredentialAuthorizationRequiredError(missing);
|
|
175
|
+
}
|
|
176
|
+
const accessTokens = new Map(resolved.map((row) => [row.slug, row.token]));
|
|
177
|
+
const loaded = await consumer.load(accessTokens);
|
|
178
|
+
return {
|
|
179
|
+
tools: wrapToolsWithManagedRuntime(loaded.tools),
|
|
180
|
+
close: loaded.close,
|
|
181
|
+
expiresAt: Date.now() + TOOL_CACHE_TTL_MS,
|
|
182
|
+
};
|
|
183
|
+
}
|
|
184
|
+
function declarationId(declaration) {
|
|
185
|
+
const existing = declarationIds.get(declaration);
|
|
186
|
+
if (existing)
|
|
187
|
+
return existing;
|
|
188
|
+
nextDeclarationId += 1;
|
|
189
|
+
const id = String(nextDeclarationId);
|
|
190
|
+
declarationIds.set(declaration, id);
|
|
191
|
+
return id;
|
|
192
|
+
}
|
|
193
|
+
function authContext(runtime) {
|
|
194
|
+
const identity = buildRuntimeIdentity(resolveRuntimeConfigurable(runtime));
|
|
195
|
+
if (!identity?.user.id) {
|
|
196
|
+
return undefined;
|
|
197
|
+
}
|
|
198
|
+
return { context: resolveAgentAuthContext(identity), identity };
|
|
199
|
+
}
|
|
200
|
+
/** Drop every cached caller entry and close its transports. Test-only. */
|
|
201
|
+
export async function clearCredentialToolCacheForTests() {
|
|
202
|
+
const entries = [...toolCache.values()];
|
|
203
|
+
toolCache.clear();
|
|
204
|
+
await Promise.all(entries.map(async (entry) => {
|
|
205
|
+
const resolved = await entry.catch(() => undefined);
|
|
206
|
+
await resolved?.close().catch(() => undefined);
|
|
207
|
+
}));
|
|
208
|
+
}
|
|
209
|
+
//# sourceMappingURL=credential-gate.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"credential-gate.js","sourceRoot":"","sources":["../src/credential-gate.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAC7C,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AAEjD,OAAO,EACL,SAAS,EACT,oCAAoC,GAErC,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACL,uBAAuB,EACvB,cAAc,GAEf,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EACL,oBAAoB,EACpB,0BAA0B,GAE3B,MAAM,+BAA+B,CAAC;AAEvC,OAAO,EACL,2BAA2B,GAE5B,MAAM,4BAA4B,CAAC;AAEpC,MAAM,CAAC,MAAM,+BAA+B,GAAG,qBAAqB,CAAC;AAErE;;;;;GAKG;AACH,MAAM,iBAAiB,GAAG,CAAC,GAAG,MAAM,CAAC;AA8CrC;;;;;;;GAOG;AACH,MAAM,SAAS,GAAG,IAAI,GAAG,EAA+B,CAAC;AAEzD,MAAM,cAAc,GAAG,IAAI,OAAO,EAAkB,CAAC;AACrD,IAAI,iBAAiB,GAAG,CAAC,CAAC;AAE1B,MAAM,0BAA2B,SAAQ,KAAK;IAC5C;QACE,KAAK,CACH,iEAAiE;YAC/D,mEAAmE,CACtE,CAAC;QACF,IAAI,CAAC,IAAI,GAAG,4BAA4B,CAAC;IAC3C,CAAC;CACF;AAED,MAAM,UAAU,8BAA8B,CAC5C,SAAwC;IAExC,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC3B,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,gBAAgB,CAAC;QACtB,IAAI,EAAE,+BAA+B;QACrC,KAAK,CAAC,WAAW,CAAC,MAAM,EAAE,OAAO;YAC/B,MAAM,MAAM,GAAG,WAAW,CAAC,OAAwC,CAAC,CAAC;YACrE,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,0BAA0B,EAAE,CAAC;YACzC,CAAC;YACD,MAAM,YAAY,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;YACtC,MAAM,eAAe,CAAC,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;QACnD,CAAC;QACD,KAAK,CAAC,aAAa,CAAC,OAAO,EAAE,OAAO;YAClC,MAAM,KAAK,GAAG,MAAM,eAAe,CACjC,SAAS,EACT,OAAO,CAAC,OAAwC,CACjD,CAAC;YACF,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACvB,OAAO,OAAO,CAAC,OAAO,CAAC,CAAC;YAC1B,CAAC;YACD,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;YACnE,OAAO,OAAO,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,CAAC,GAAG,QAAQ,EAAE,GAAG,KAAK,CAAC,EAAE,CAAC,CAAC;QACjE,CAAC;QACD,KAAK,CAAC,YAAY,CAAC,OAAO,EAAE,OAAO;YACjC,MAAM,IAAI,GAAG,OAAO,CAAC,QAAQ,EAAE,IAAI,CAAC;YACpC,IAAI,OAAO,CAAC,IAAI,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACrD,OAAO,OAAO,CAAC,OAAO,CAAC,CAAC;YAC1B,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,eAAe,CACjC,SAAS,EACT,OAAO,CAAC,OAAwC,CACjD,CAAC;YACF,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;YAChE,OAAO,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,GAAG,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACjE,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED;;;;;;GAMG;AACH,KAAK,UAAU,YAAY,CACzB,SAAwC,EACxC,MAAqB;IAErB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAChD,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,mBAAmB,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QAC9D,IAAI;QACJ,KAAK,EAAE,MAAM,SAAS,CAAC,mBAAmB,CAAC,IAAI,CAAC;KACjD,CAAC,CAAC,CACJ,CAAC;IACF,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAC3E,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO;IACT,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,SAAS,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC,CACjE,CAAC;IACF,MAAM,OAAO,GAA6C;QACxD,IAAI,EAAE,mCAAmC;QACzC,OAAO,EAAE,iDAAiD;QAC1D,WAAW,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YACrC,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,WAAW,EAAE,OAAO,CAAC,UAAU;YAC/B,OAAO,EAAE,OAAO,CAAC,MAAM;SACxB,CAAC,CAAC;KACJ,CAAC;IACF,2EAA2E;IAC3E,0EAA0E;IAC1E,WAAW;IACX,SAAS,CAAC,OAAO,CAAC,CAAC;IACnB,MAAM,IAAI,oCAAoC,CAAC,OAAO,CAAC,CAAC;AAC1D,CAAC;AAED;;;;;;GAMG;AACH,SAAS,mBAAmB,CAC1B,SAAwC,EACxC,MAAqB;IAErB,MAAM,MAAM,GAAG,IAAI,GAAG,EAA+B,CAAC;IACtD,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,KAAK,MAAM,UAAU,IAAI,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC9C,cAAc,CAAC,UAAU,CAAC,KAAK,EAAE,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC1E,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IACD,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;AAC9B,CAAC;AAED,KAAK,UAAU,eAAe,CAC5B,SAAwC,EACxC,OAAsC;IAEtC,MAAM,MAAM,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;IACpC,OAAO,MAAM,CAAC,CAAC,CAAC,MAAM,eAAe,CAAC,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AACxE,CAAC;AAED,KAAK,UAAU,eAAe,CAC5B,SAAwC,EACxC,OAAyB;IAEzB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,GAAG,CAC9B,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,WAAW,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,CAC5D,CAAC;IACF,OAAO,MAAM,CAAC,IAAI,EAAE,CAAC;AACvB,CAAC;AAED,KAAK,UAAU,WAAW,CACxB,QAA4B,EAC5B,OAAyB;IAEzB,MAAM,GAAG,GAAG,GAAG,aAAa,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;IAE5E,SAAS,CAAC;QACR,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAClC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,OAAO,CAAC,GAAG,EAAE,QAAQ,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;QACpD,CAAC;QACD,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAClD,IAAI,KAAK,IAAI,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;YAC1C,OAAO,KAAK,CAAC,KAAK,CAAC;QACrB,CAAC;QACD,0EAA0E;QAC1E,kEAAkE;QAClE,4CAA4C;QAC5C,IAAI,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,MAAM,EAAE,CAAC;YAClC,OAAO,OAAO,CAAC,GAAG,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;AACH,CAAC;AAED,sFAAsF;AACtF,KAAK,UAAU,OAAO,CACpB,GAAW,EACX,QAA4B,EAC5B,OAAyB,EACzB,MAA8B;IAE9B,MAAM,OAAO,GAAG,SAAS,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC7C,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IAC5B,MAAM,EAAE,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IACvC,IAAI,CAAC;QACH,OAAO,CAAC,MAAM,OAAO,CAAC,CAAC,KAAK,CAAC;IAC/B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,OAAO,EAAE,CAAC;YACnC,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACxB,CAAC;QACD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,KAAK,UAAU,SAAS,CACtB,QAA4B,EAC5B,OAAyB;IAEzB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,OAAO,CAAC,CAAC;IACzC,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,GAAG,CAChC,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QAC5C,IAAI;QACJ,KAAK,EAAE,MAAM,SAAS,CAAC,mBAAmB,CAAC,IAAI,CAAC;KACjD,CAAC,CAAC,CACJ,CAAC;IACF,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAC5E,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvB,MAAM,IAAI,oCAAoC,CAAC,OAAO,CAAC,CAAC;IAC1D,CAAC;IAED,MAAM,YAAY,GAAG,IAAI,GAAG,CAC1B,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,KAAe,CAAC,CAAC,CACvD,CAAC;IACF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACjD,OAAO;QACL,KAAK,EAAE,2BAA2B,CAChC,MAAM,CAAC,KAAwB,CACjB;QAChB,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,iBAAiB;KAC1C,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,WAAmB;IACxC,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IACjD,IAAI,QAAQ;QAAE,OAAO,QAAQ,CAAC;IAC9B,iBAAiB,IAAI,CAAC,CAAC;IACvB,MAAM,EAAE,GAAG,MAAM,CAAC,iBAAiB,CAAC,CAAC;IACrC,cAAc,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;IACpC,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,WAAW,CAClB,OAAsC;IAEtC,MAAM,QAAQ,GAAG,oBAAoB,CAAC,0BAA0B,CAAC,OAAO,CAAC,CAAC,CAAC;IAC3E,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;QACvB,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,uBAAuB,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,CAAC;AAClE,CAAC;AAED,0EAA0E;AAC1E,MAAM,CAAC,KAAK,UAAU,gCAAgC;IACpD,MAAM,OAAO,GAAG,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC;IACxC,SAAS,CAAC,KAAK,EAAE,CAAC;IAClB,MAAM,OAAO,CAAC,GAAG,CACf,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;QAC1B,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACpD,MAAM,QAAQ,EAAE,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IACjD,CAAC,CAAC,CACH,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,95 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* `credentials.get(...)` — read a managed credential from a tool.
|
|
3
|
+
*
|
|
4
|
+
* `owner` is the scope the credential resolves under, and names the same axis as
|
|
5
|
+
* Agent Auth's `owner_type` / `owner_id`. `{ type: "agent" }` is the deployment,
|
|
6
|
+
* so every caller resolves the same static secret. `{ type: "user" }` is the
|
|
7
|
+
* current caller, always backed by an OAuth grant. Resolution reads Agent Auth
|
|
8
|
+
* on every await and never caches.
|
|
9
|
+
*
|
|
10
|
+
* `mda dev` runs without a deployment, so it substitutes local values instead:
|
|
11
|
+
* `{ type: "agent" }` reads a normalized `MDA_DEV_*` environment variable, and
|
|
12
|
+
* `{ type: "user" }` has no local stand-in for an OAuth grant, so it fails with
|
|
13
|
+
* an actionable error. Both scopes work normally against a deployment.
|
|
14
|
+
*/
|
|
15
|
+
import type { RuntimeIdentity } from "./identity/index.js";
|
|
16
|
+
import { type AgentAuthContext } from "./agent-auth.js";
|
|
17
|
+
export { AGENT_AUTH_TIMEOUT_MS, AgentAuthUnknownProviderError, CredentialAuthorizationRequiredError, CredentialStoreError, } from "./agent-auth.js";
|
|
18
|
+
export type { AgentAuthContext, PendingAuthorization } from "./agent-auth.js";
|
|
19
|
+
export type AgentCredentialOwner = {
|
|
20
|
+
type: "agent";
|
|
21
|
+
};
|
|
22
|
+
export type UserCredentialOwner = {
|
|
23
|
+
type: "user";
|
|
24
|
+
id?: string;
|
|
25
|
+
};
|
|
26
|
+
export type CredentialOwner = AgentCredentialOwner | UserCredentialOwner;
|
|
27
|
+
export interface CredentialRequest {
|
|
28
|
+
slug: string;
|
|
29
|
+
owner: CredentialOwner;
|
|
30
|
+
identity?: RuntimeIdentity;
|
|
31
|
+
}
|
|
32
|
+
declare const CREDENTIAL_REFERENCE: unique symbol;
|
|
33
|
+
export declare class CredentialNotFoundError extends Error {
|
|
34
|
+
readonly slug: string;
|
|
35
|
+
readonly ownerType: CredentialOwner["type"];
|
|
36
|
+
constructor(slug: string, ownerType: CredentialOwner["type"]);
|
|
37
|
+
}
|
|
38
|
+
/**
|
|
39
|
+
* A declarative credential handle. Connector declarations inspect its owner;
|
|
40
|
+
* awaiting the same handle inside a tool resolves fresh material every time.
|
|
41
|
+
*/
|
|
42
|
+
export declare class CredentialReference implements PromiseLike<string> {
|
|
43
|
+
readonly slug: string;
|
|
44
|
+
readonly owner: CredentialOwner;
|
|
45
|
+
private readonly accessor;
|
|
46
|
+
readonly [CREDENTIAL_REFERENCE] = true;
|
|
47
|
+
constructor(slug: string, owner: CredentialOwner, accessor: Credentials);
|
|
48
|
+
then<TResult1 = string, TResult2 = never>(onfulfilled?: ((value: string) => TResult1 | PromiseLike<TResult1>) | null, onrejected?: ((reason: unknown) => TResult2 | PromiseLike<TResult2>) | null): PromiseLike<TResult1 | TResult2>;
|
|
49
|
+
}
|
|
50
|
+
export declare function isCredentialReference(value: unknown): value is CredentialReference;
|
|
51
|
+
export declare class AgentAuthCredentialStore {
|
|
52
|
+
private readonly env;
|
|
53
|
+
private readonly fetchImpl;
|
|
54
|
+
private readonly timeoutMs;
|
|
55
|
+
constructor(env?: NodeJS.ProcessEnv, fetchImpl?: typeof fetch, timeoutMs?: number);
|
|
56
|
+
fetch(request: CredentialRequest): Promise<string | undefined>;
|
|
57
|
+
}
|
|
58
|
+
export declare class Credentials {
|
|
59
|
+
private readonly store;
|
|
60
|
+
constructor(store?: AgentAuthCredentialStore);
|
|
61
|
+
get(slug: string, owner: CredentialOwner): CredentialReference;
|
|
62
|
+
resolve(slug: string, owner: CredentialOwner): Promise<string>;
|
|
63
|
+
/**
|
|
64
|
+
* Read an agent-owned secret outside a run.
|
|
65
|
+
*
|
|
66
|
+
* Connector declarations are loaded once per deployment, before any caller
|
|
67
|
+
* exists. An agent-owned secret is deployment-scoped, so it is the one lookup
|
|
68
|
+
* that carries no caller and can be resolved there.
|
|
69
|
+
*/
|
|
70
|
+
resolveDeploymentSecret(slug: string): Promise<string>;
|
|
71
|
+
}
|
|
72
|
+
/** Environment variable used for one agent-owned development credential. */
|
|
73
|
+
export declare function agentCredentialEnvKey(slug: string): string;
|
|
74
|
+
/**
|
|
75
|
+
* The `owner_id` a lookup is keyed by: the caller for a user, else the
|
|
76
|
+
* deployment. Shared with the credential gate so a declared owner is checked
|
|
77
|
+
* the same way wherever it is resolved.
|
|
78
|
+
*/
|
|
79
|
+
export declare function resolveOwnerId(owner: CredentialOwner, identity: RuntimeIdentity | undefined, deploymentId: string | undefined): string;
|
|
80
|
+
/**
|
|
81
|
+
* The Agent Auth binding for one caller.
|
|
82
|
+
*
|
|
83
|
+
* `credentials.get` and the pre-run credential gate both resolve their owner id
|
|
84
|
+
* here, so a slug addresses the same Agent Auth row on either path.
|
|
85
|
+
*/
|
|
86
|
+
export declare function resolveAgentAuthContext(identity: RuntimeIdentity | undefined, env?: NodeJS.ProcessEnv): AgentAuthContext;
|
|
87
|
+
/**
|
|
88
|
+
* Agent Auth is served by the LangSmith API.
|
|
89
|
+
*
|
|
90
|
+
* Rejects a URL that carries a path, query, or userinfo, so a misconfigured
|
|
91
|
+
* endpoint cannot send the API key at a different service.
|
|
92
|
+
*/
|
|
93
|
+
export declare function resolveAgentAuthBaseUrl(env?: NodeJS.ProcessEnv): string;
|
|
94
|
+
export declare const credentials: Credentials;
|
|
95
|
+
//# sourceMappingURL=credentials.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"credentials.d.ts","sourceRoot":"","sources":["../src/credentials.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAEH,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAC3D,OAAO,EAKL,KAAK,gBAAgB,EACtB,MAAM,iBAAiB,CAAC;AAMzB,OAAO,EACL,qBAAqB,EACrB,6BAA6B,EAC7B,oCAAoC,EACpC,oBAAoB,GACrB,MAAM,iBAAiB,CAAC;AACzB,YAAY,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAE9E,MAAM,MAAM,oBAAoB,GAAG;IAAE,IAAI,EAAE,OAAO,CAAA;CAAE,CAAC;AACrD,MAAM,MAAM,mBAAmB,GAAG;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,EAAE,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC;AAChE,MAAM,MAAM,eAAe,GAAG,oBAAoB,GAAG,mBAAmB,CAAC;AAEzE,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,eAAe,CAAC;IACvB,QAAQ,CAAC,EAAE,eAAe,CAAC;CAC5B;AAED,QAAA,MAAM,oBAAoB,eAA0C,CAAC;AAuBrE,qBAAa,uBAAwB,SAAQ,KAAK;IAChD,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,SAAS,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC;IAE5C,YAAY,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,eAAe,CAAC,MAAM,CAAC,EAK3D;CACF;AAED;;;GAGG;AACH,qBAAa,mBAAoB,YAAW,WAAW,CAAC,MAAM,CAAC;IAI3D,QAAQ,CAAC,IAAI,EAAE,MAAM;IACrB,QAAQ,CAAC,KAAK,EAAE,eAAe;IAC/B,OAAO,CAAC,QAAQ,CAAC,QAAQ;IAL3B,QAAQ,CAAC,CAAC,oBAAoB,CAAC,QAAQ;IAEvC,YACW,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,eAAe,EACd,QAAQ,EAAE,WAAW,EACpC;IAEJ,IAAI,CAAC,QAAQ,GAAG,MAAM,EAAE,QAAQ,GAAG,KAAK,EACtC,WAAW,CAAC,EAAE,CAAC,CAAC,KAAK,EAAE,MAAM,KAAK,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC,GAAG,IAAI,EAC1E,UAAU,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,OAAO,KAAK,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC,GAAG,IAAI,GAC1E,WAAW,CAAC,QAAQ,GAAG,QAAQ,CAAC,CAIlC;CACF;AAED,wBAAgB,qBAAqB,CACnC,KAAK,EAAE,OAAO,GACb,KAAK,IAAI,mBAAmB,CAO9B;AASD,qBAAa,wBAAwB;IAEjC,OAAO,CAAC,QAAQ,CAAC,GAAG;IACpB,OAAO,CAAC,QAAQ,CAAC,SAAS;IAC1B,OAAO,CAAC,QAAQ,CAAC,SAAS;IAH5B,YACmB,GAAG,GAAE,MAAM,CAAC,UAAwB,EACpC,SAAS,GAAE,OAAO,KAAa,EAC/B,SAAS,SAAwB,EAChD;IAEE,KAAK,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CA8CnE;CACF;AAED,qBAAa,WAAW;IAEpB,OAAO,CAAC,QAAQ,CAAC,KAAK;IADxB,YACmB,KAAK,GAAE,wBAAyD,EAC/E;IAEJ,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,eAAe,GAAG,mBAAmB,CAM7D;IAEK,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,eAAe,GAAG,OAAO,CAAC,MAAM,CAAC,CAkBnE;IAED;;;;;;OAMG;IACG,uBAAuB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAS3D;CACF;AAED,4EAA4E;AAC5E,wBAAgB,qBAAqB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAI1D;AA+CD;;;;GAIG;AACH,wBAAgB,cAAc,CAC5B,KAAK,EAAE,eAAe,EACtB,QAAQ,EAAE,eAAe,GAAG,SAAS,EACrC,YAAY,EAAE,MAAM,GAAG,SAAS,GAC/B,MAAM,CAqBR;AAED;;;;;GAKG;AACH,wBAAgB,uBAAuB,CACrC,QAAQ,EAAE,eAAe,GAAG,SAAS,EACrC,GAAG,GAAE,MAAM,CAAC,UAAwB,GACnC,gBAAgB,CAuBlB;AAiBD;;;;;GAKG;AACH,wBAAgB,uBAAuB,CACrC,GAAG,GAAE,MAAM,CAAC,UAAwB,GACnC,MAAM,CAoBR;AAED,eAAO,MAAM,WAAW,aAAoB,CAAC"}
|