agents 0.0.0-25aeaf2 → 0.0.0-25b261e
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 +131 -25
- package/dist/ai-chat-agent.d.ts +35 -6
- package/dist/ai-chat-agent.js +139 -117
- package/dist/ai-chat-agent.js.map +1 -1
- package/dist/ai-react.d.ts +13 -9
- package/dist/ai-react.js +27 -29
- package/dist/ai-react.js.map +1 -1
- package/dist/chunk-DQJFYHG3.js +1290 -0
- package/dist/chunk-DQJFYHG3.js.map +1 -0
- package/dist/{chunk-25YDMV4H.js → chunk-EM3J4KV7.js} +179 -45
- package/dist/chunk-EM3J4KV7.js.map +1 -0
- package/dist/{chunk-RN4SNE73.js → chunk-KUH345EY.js} +17 -34
- package/dist/chunk-KUH345EY.js.map +1 -0
- package/dist/{chunk-D6UOOELW.js → chunk-PVQZBKN7.js} +15 -15
- package/dist/chunk-PVQZBKN7.js.map +1 -0
- package/dist/client-DgyzBU_8.d.ts +4601 -0
- package/dist/client.d.ts +10 -3
- package/dist/client.js +1 -2
- package/dist/index.d.ts +197 -17
- package/dist/index.js +10 -5
- package/dist/mcp/client.d.ts +9 -781
- package/dist/mcp/client.js +1 -2
- package/dist/mcp/do-oauth-client-provider.d.ts +3 -3
- package/dist/mcp/do-oauth-client-provider.js +1 -2
- package/dist/mcp/index.d.ts +48 -11
- package/dist/mcp/index.js +317 -180
- package/dist/mcp/index.js.map +1 -1
- package/dist/observability/index.d.ts +46 -0
- package/dist/observability/index.js +10 -0
- package/dist/react.d.ts +80 -11
- package/dist/react.js +16 -8
- package/dist/react.js.map +1 -1
- package/dist/schedule.d.ts +6 -6
- package/dist/schedule.js +4 -6
- package/dist/schedule.js.map +1 -1
- package/dist/serializable.d.ts +32 -0
- package/dist/serializable.js +1 -0
- package/dist/serializable.js.map +1 -0
- package/package.json +79 -71
- package/src/index.ts +898 -222
- package/dist/chunk-25YDMV4H.js.map +0 -1
- package/dist/chunk-D6UOOELW.js.map +0 -1
- package/dist/chunk-HMLY7DHA.js +0 -16
- package/dist/chunk-RN4SNE73.js.map +0 -1
- package/dist/chunk-YFPCCSZO.js +0 -787
- package/dist/chunk-YFPCCSZO.js.map +0 -1
- /package/dist/{chunk-HMLY7DHA.js.map → observability/index.js.map} +0 -0
|
@@ -1,9 +1,3 @@
|
|
|
1
|
-
import {
|
|
2
|
-
__privateAdd,
|
|
3
|
-
__privateGet,
|
|
4
|
-
__privateSet
|
|
5
|
-
} from "./chunk-HMLY7DHA.js";
|
|
6
|
-
|
|
7
1
|
// src/client.ts
|
|
8
2
|
import {
|
|
9
3
|
PartySocket
|
|
@@ -19,54 +13,52 @@ function camelCaseToKebabCase(str) {
|
|
|
19
13
|
kebabified = kebabified.startsWith("-") ? kebabified.slice(1) : kebabified;
|
|
20
14
|
return kebabified.replace(/_/g, "-").replace(/-$/, "");
|
|
21
15
|
}
|
|
22
|
-
var _options, _pendingCalls;
|
|
23
16
|
var AgentClient = class extends PartySocket {
|
|
24
17
|
constructor(options) {
|
|
25
18
|
const agentNamespace = camelCaseToKebabCase(options.agent);
|
|
26
19
|
super({
|
|
27
|
-
prefix: "agents",
|
|
28
20
|
party: agentNamespace,
|
|
21
|
+
prefix: "agents",
|
|
29
22
|
room: options.name || "default",
|
|
30
23
|
...options
|
|
31
24
|
});
|
|
32
|
-
|
|
33
|
-
__privateAdd(this, _pendingCalls, /* @__PURE__ */ new Map());
|
|
25
|
+
this._pendingCalls = /* @__PURE__ */ new Map();
|
|
34
26
|
this.agent = agentNamespace;
|
|
35
27
|
this.name = options.name || "default";
|
|
36
|
-
|
|
28
|
+
this.options = options;
|
|
37
29
|
this.addEventListener("message", (event) => {
|
|
38
30
|
if (typeof event.data === "string") {
|
|
39
31
|
let parsedMessage;
|
|
40
32
|
try {
|
|
41
33
|
parsedMessage = JSON.parse(event.data);
|
|
42
|
-
} catch (
|
|
34
|
+
} catch (_error) {
|
|
43
35
|
return;
|
|
44
36
|
}
|
|
45
37
|
if (parsedMessage.type === "cf_agent_state") {
|
|
46
|
-
|
|
38
|
+
this.options.onStateUpdate?.(parsedMessage.state, "server");
|
|
47
39
|
return;
|
|
48
40
|
}
|
|
49
41
|
if (parsedMessage.type === "rpc") {
|
|
50
42
|
const response = parsedMessage;
|
|
51
|
-
const pending =
|
|
43
|
+
const pending = this._pendingCalls.get(response.id);
|
|
52
44
|
if (!pending) return;
|
|
53
45
|
if (!response.success) {
|
|
54
46
|
pending.reject(new Error(response.error));
|
|
55
|
-
|
|
47
|
+
this._pendingCalls.delete(response.id);
|
|
56
48
|
pending.stream?.onError?.(response.error);
|
|
57
49
|
return;
|
|
58
50
|
}
|
|
59
51
|
if ("done" in response) {
|
|
60
52
|
if (response.done) {
|
|
61
53
|
pending.resolve(response.result);
|
|
62
|
-
|
|
54
|
+
this._pendingCalls.delete(response.id);
|
|
63
55
|
pending.stream?.onDone?.(response.result);
|
|
64
56
|
} else {
|
|
65
57
|
pending.stream?.onChunk?.(response.result);
|
|
66
58
|
}
|
|
67
59
|
} else {
|
|
68
60
|
pending.resolve(response.result);
|
|
69
|
-
|
|
61
|
+
this._pendingCalls.delete(response.id);
|
|
70
62
|
}
|
|
71
63
|
}
|
|
72
64
|
}
|
|
@@ -81,43 +73,34 @@ var AgentClient = class extends PartySocket {
|
|
|
81
73
|
);
|
|
82
74
|
}
|
|
83
75
|
setState(state) {
|
|
84
|
-
this.send(JSON.stringify({ type: "cf_agent_state"
|
|
85
|
-
|
|
76
|
+
this.send(JSON.stringify({ state, type: "cf_agent_state" }));
|
|
77
|
+
this.options.onStateUpdate?.(state, "client");
|
|
86
78
|
}
|
|
87
|
-
/**
|
|
88
|
-
* Call a method on the Agent
|
|
89
|
-
* @param method Name of the method to call
|
|
90
|
-
* @param args Arguments to pass to the method
|
|
91
|
-
* @param streamOptions Options for handling streaming responses
|
|
92
|
-
* @returns Promise that resolves with the method's return value
|
|
93
|
-
*/
|
|
94
79
|
async call(method, args = [], streamOptions) {
|
|
95
80
|
return new Promise((resolve, reject) => {
|
|
96
81
|
const id = Math.random().toString(36).slice(2);
|
|
97
|
-
|
|
98
|
-
resolve: (value) => resolve(value),
|
|
82
|
+
this._pendingCalls.set(id, {
|
|
99
83
|
reject,
|
|
84
|
+
resolve: (value) => resolve(value),
|
|
100
85
|
stream: streamOptions,
|
|
101
86
|
type: null
|
|
102
87
|
});
|
|
103
88
|
const request = {
|
|
104
|
-
|
|
89
|
+
args,
|
|
105
90
|
id,
|
|
106
91
|
method,
|
|
107
|
-
|
|
92
|
+
type: "rpc"
|
|
108
93
|
};
|
|
109
94
|
this.send(JSON.stringify(request));
|
|
110
95
|
});
|
|
111
96
|
}
|
|
112
97
|
};
|
|
113
|
-
_options = new WeakMap();
|
|
114
|
-
_pendingCalls = new WeakMap();
|
|
115
98
|
function agentFetch(opts, init) {
|
|
116
99
|
const agentNamespace = camelCaseToKebabCase(opts.agent);
|
|
117
100
|
return PartySocket.fetch(
|
|
118
101
|
{
|
|
119
|
-
prefix: "agents",
|
|
120
102
|
party: agentNamespace,
|
|
103
|
+
prefix: "agents",
|
|
121
104
|
room: opts.name || "default",
|
|
122
105
|
...opts
|
|
123
106
|
},
|
|
@@ -130,4 +113,4 @@ export {
|
|
|
130
113
|
AgentClient,
|
|
131
114
|
agentFetch
|
|
132
115
|
};
|
|
133
|
-
//# sourceMappingURL=chunk-
|
|
116
|
+
//# sourceMappingURL=chunk-KUH345EY.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["../src/client.ts"],"sourcesContent":["import {\n type PartyFetchOptions,\n PartySocket,\n type PartySocketOptions\n} from \"partysocket\";\nimport type { RPCRequest, RPCResponse } from \"./\";\nimport type {\n SerializableReturnValue,\n SerializableValue\n} from \"./serializable\";\n\n/**\n * Options for creating an AgentClient\n */\nexport type AgentClientOptions<State = unknown> = Omit<\n PartySocketOptions,\n \"party\" | \"room\"\n> & {\n /** Name of the agent to connect to */\n agent: string;\n /** Name of the specific Agent instance */\n name?: string;\n /** Called when the Agent's state is updated */\n onStateUpdate?: (state: State, source: \"server\" | \"client\") => void;\n};\n\n/**\n * Options for streaming RPC calls\n */\nexport type StreamOptions = {\n /** Called when a chunk of data is received */\n onChunk?: (chunk: unknown) => void;\n /** Called when the stream ends */\n onDone?: (finalChunk: unknown) => void;\n /** Called when an error occurs */\n onError?: (error: string) => void;\n};\n\n/**\n * Options for the agentFetch function\n */\nexport type AgentClientFetchOptions = Omit<\n PartyFetchOptions,\n \"party\" | \"room\"\n> & {\n /** Name of the agent to connect to */\n agent: string;\n /** Name of the specific Agent instance */\n name?: string;\n};\n\n/**\n * Convert a camelCase string to a kebab-case string\n * @param str The string to convert\n * @returns The kebab-case string\n */\nexport function camelCaseToKebabCase(str: string): string {\n // If string is all uppercase, convert to lowercase\n if (str === str.toUpperCase() && str !== str.toLowerCase()) {\n return str.toLowerCase().replace(/_/g, \"-\");\n }\n\n // Otherwise handle camelCase to kebab-case\n let kebabified = str.replace(\n /[A-Z]/g,\n (letter) => `-${letter.toLowerCase()}`\n );\n kebabified = kebabified.startsWith(\"-\") ? kebabified.slice(1) : kebabified;\n // Convert any remaining underscores to hyphens and remove trailing -'s\n return kebabified.replace(/_/g, \"-\").replace(/-$/, \"\");\n}\n\n/**\n * WebSocket client for connecting to an Agent\n */\nexport class AgentClient<State = unknown> extends PartySocket {\n /**\n * @deprecated Use agentFetch instead\n */\n static fetch(_opts: PartyFetchOptions): Promise<Response> {\n throw new Error(\n \"AgentClient.fetch is not implemented, use agentFetch instead\"\n );\n }\n agent: string;\n name: string;\n private options: AgentClientOptions<State>;\n private _pendingCalls = new Map<\n string,\n {\n resolve: (value: unknown) => void;\n reject: (error: Error) => void;\n stream?: StreamOptions;\n type?: unknown;\n }\n >();\n\n constructor(options: AgentClientOptions<State>) {\n const agentNamespace = camelCaseToKebabCase(options.agent);\n super({\n party: agentNamespace,\n prefix: \"agents\",\n room: options.name || \"default\",\n ...options\n });\n this.agent = agentNamespace;\n this.name = options.name || \"default\";\n this.options = options;\n\n this.addEventListener(\"message\", (event) => {\n if (typeof event.data === \"string\") {\n let parsedMessage: Record<string, unknown>;\n try {\n parsedMessage = JSON.parse(event.data);\n } catch (_error) {\n // silently ignore invalid messages for now\n // TODO: log errors with log levels\n return;\n }\n if (parsedMessage.type === \"cf_agent_state\") {\n this.options.onStateUpdate?.(parsedMessage.state as State, \"server\");\n return;\n }\n if (parsedMessage.type === \"rpc\") {\n const response = parsedMessage as RPCResponse;\n const pending = this._pendingCalls.get(response.id);\n if (!pending) return;\n\n if (!response.success) {\n pending.reject(new Error(response.error));\n this._pendingCalls.delete(response.id);\n pending.stream?.onError?.(response.error);\n return;\n }\n\n // Handle streaming responses\n if (\"done\" in response) {\n if (response.done) {\n pending.resolve(response.result);\n this._pendingCalls.delete(response.id);\n pending.stream?.onDone?.(response.result);\n } else {\n pending.stream?.onChunk?.(response.result);\n }\n } else {\n // Non-streaming response\n pending.resolve(response.result);\n this._pendingCalls.delete(response.id);\n }\n }\n }\n });\n }\n\n setState(state: State) {\n this.send(JSON.stringify({ state, type: \"cf_agent_state\" }));\n this.options.onStateUpdate?.(state, \"client\");\n }\n\n /**\n * Call a method on the Agent\n * @param method Name of the method to call\n * @param args Arguments to pass to the method\n * @param streamOptions Options for handling streaming responses\n * @returns Promise that resolves with the method's return value\n */\n call<T extends SerializableReturnValue>(\n method: string,\n args?: SerializableValue[],\n streamOptions?: StreamOptions\n ): Promise<T>;\n call<T = unknown>(\n method: string,\n args?: unknown[],\n streamOptions?: StreamOptions\n ): Promise<T>;\n async call<T>(\n method: string,\n args: unknown[] = [],\n streamOptions?: StreamOptions\n ): Promise<T> {\n return new Promise<T>((resolve, reject) => {\n const id = Math.random().toString(36).slice(2);\n this._pendingCalls.set(id, {\n reject,\n resolve: (value: unknown) => resolve(value as T),\n stream: streamOptions,\n type: null as T\n });\n\n const request: RPCRequest = {\n args,\n id,\n method,\n type: \"rpc\"\n };\n\n this.send(JSON.stringify(request));\n });\n }\n}\n\n/**\n * Make an HTTP request to an Agent\n * @param opts Connection options\n * @param init Request initialization options\n * @returns Promise resolving to a Response\n */\nexport function agentFetch(opts: AgentClientFetchOptions, init?: RequestInit) {\n const agentNamespace = camelCaseToKebabCase(opts.agent);\n\n return PartySocket.fetch(\n {\n party: agentNamespace,\n prefix: \"agents\",\n room: opts.name || \"default\",\n ...opts\n },\n init\n );\n}\n"],"mappings":";AAAA;AAAA,EAEE;AAAA,OAEK;AAoDA,SAAS,qBAAqB,KAAqB;AAExD,MAAI,QAAQ,IAAI,YAAY,KAAK,QAAQ,IAAI,YAAY,GAAG;AAC1D,WAAO,IAAI,YAAY,EAAE,QAAQ,MAAM,GAAG;AAAA,EAC5C;AAGA,MAAI,aAAa,IAAI;AAAA,IACnB;AAAA,IACA,CAAC,WAAW,IAAI,OAAO,YAAY,CAAC;AAAA,EACtC;AACA,eAAa,WAAW,WAAW,GAAG,IAAI,WAAW,MAAM,CAAC,IAAI;AAEhE,SAAO,WAAW,QAAQ,MAAM,GAAG,EAAE,QAAQ,MAAM,EAAE;AACvD;AAKO,IAAM,cAAN,cAA2C,YAAY;AAAA,EAsB5D,YAAY,SAAoC;AAC9C,UAAM,iBAAiB,qBAAqB,QAAQ,KAAK;AACzD,UAAM;AAAA,MACJ,OAAO;AAAA,MACP,QAAQ;AAAA,MACR,MAAM,QAAQ,QAAQ;AAAA,MACtB,GAAG;AAAA,IACL,CAAC;AAjBH,SAAQ,gBAAgB,oBAAI,IAQ1B;AAUA,SAAK,QAAQ;AACb,SAAK,OAAO,QAAQ,QAAQ;AAC5B,SAAK,UAAU;AAEf,SAAK,iBAAiB,WAAW,CAAC,UAAU;AAC1C,UAAI,OAAO,MAAM,SAAS,UAAU;AAClC,YAAI;AACJ,YAAI;AACF,0BAAgB,KAAK,MAAM,MAAM,IAAI;AAAA,QACvC,SAAS,QAAQ;AAGf;AAAA,QACF;AACA,YAAI,cAAc,SAAS,kBAAkB;AAC3C,eAAK,QAAQ,gBAAgB,cAAc,OAAgB,QAAQ;AACnE;AAAA,QACF;AACA,YAAI,cAAc,SAAS,OAAO;AAChC,gBAAM,WAAW;AACjB,gBAAM,UAAU,KAAK,cAAc,IAAI,SAAS,EAAE;AAClD,cAAI,CAAC,QAAS;AAEd,cAAI,CAAC,SAAS,SAAS;AACrB,oBAAQ,OAAO,IAAI,MAAM,SAAS,KAAK,CAAC;AACxC,iBAAK,cAAc,OAAO,SAAS,EAAE;AACrC,oBAAQ,QAAQ,UAAU,SAAS,KAAK;AACxC;AAAA,UACF;AAGA,cAAI,UAAU,UAAU;AACtB,gBAAI,SAAS,MAAM;AACjB,sBAAQ,QAAQ,SAAS,MAAM;AAC/B,mBAAK,cAAc,OAAO,SAAS,EAAE;AACrC,sBAAQ,QAAQ,SAAS,SAAS,MAAM;AAAA,YAC1C,OAAO;AACL,sBAAQ,QAAQ,UAAU,SAAS,MAAM;AAAA,YAC3C;AAAA,UACF,OAAO;AAEL,oBAAQ,QAAQ,SAAS,MAAM;AAC/B,iBAAK,cAAc,OAAO,SAAS,EAAE;AAAA,UACvC;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA,EAzEA,OAAO,MAAM,OAA6C;AACxD,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAAA,EAuEA,SAAS,OAAc;AACrB,SAAK,KAAK,KAAK,UAAU,EAAE,OAAO,MAAM,iBAAiB,CAAC,CAAC;AAC3D,SAAK,QAAQ,gBAAgB,OAAO,QAAQ;AAAA,EAC9C;AAAA,EAmBA,MAAM,KACJ,QACA,OAAkB,CAAC,GACnB,eACY;AACZ,WAAO,IAAI,QAAW,CAAC,SAAS,WAAW;AACzC,YAAM,KAAK,KAAK,OAAO,EAAE,SAAS,EAAE,EAAE,MAAM,CAAC;AAC7C,WAAK,cAAc,IAAI,IAAI;AAAA,QACzB;AAAA,QACA,SAAS,CAAC,UAAmB,QAAQ,KAAU;AAAA,QAC/C,QAAQ;AAAA,QACR,MAAM;AAAA,MACR,CAAC;AAED,YAAM,UAAsB;AAAA,QAC1B;AAAA,QACA;AAAA,QACA;AAAA,QACA,MAAM;AAAA,MACR;AAEA,WAAK,KAAK,KAAK,UAAU,OAAO,CAAC;AAAA,IACnC,CAAC;AAAA,EACH;AACF;AAQO,SAAS,WAAW,MAA+B,MAAoB;AAC5E,QAAM,iBAAiB,qBAAqB,KAAK,KAAK;AAEtD,SAAO,YAAY;AAAA,IACjB;AAAA,MACE,OAAO;AAAA,MACP,QAAQ;AAAA,MACR,MAAM,KAAK,QAAQ;AAAA,MACnB,GAAG;AAAA,IACL;AAAA,IACA;AAAA,EACF;AACF;","names":[]}
|
|
@@ -7,34 +7,34 @@ var DurableObjectOAuthClientProvider = class {
|
|
|
7
7
|
}
|
|
8
8
|
get clientMetadata() {
|
|
9
9
|
return {
|
|
10
|
-
|
|
11
|
-
|
|
10
|
+
client_name: this.clientName,
|
|
11
|
+
client_uri: "example.com",
|
|
12
12
|
grant_types: ["authorization_code", "refresh_token"],
|
|
13
|
+
redirect_uris: [this.redirectUrl],
|
|
13
14
|
response_types: ["code"],
|
|
14
|
-
|
|
15
|
-
client_uri: "example.com"
|
|
15
|
+
token_endpoint_auth_method: "none"
|
|
16
16
|
};
|
|
17
17
|
}
|
|
18
18
|
get redirectUrl() {
|
|
19
19
|
return `${this.baseRedirectUrl}/${this.serverId}`;
|
|
20
20
|
}
|
|
21
21
|
get clientId() {
|
|
22
|
-
if (!this.
|
|
22
|
+
if (!this._clientId_) {
|
|
23
23
|
throw new Error("Trying to access clientId before it was set");
|
|
24
24
|
}
|
|
25
|
-
return this.
|
|
25
|
+
return this._clientId_;
|
|
26
26
|
}
|
|
27
27
|
set clientId(clientId_) {
|
|
28
|
-
this.
|
|
28
|
+
this._clientId_ = clientId_;
|
|
29
29
|
}
|
|
30
30
|
get serverId() {
|
|
31
|
-
if (!this.
|
|
31
|
+
if (!this._serverId_) {
|
|
32
32
|
throw new Error("Trying to access serverId before it was set");
|
|
33
33
|
}
|
|
34
|
-
return this.
|
|
34
|
+
return this._serverId_;
|
|
35
35
|
}
|
|
36
36
|
set serverId(serverId_) {
|
|
37
|
-
this.
|
|
37
|
+
this._serverId_ = serverId_;
|
|
38
38
|
}
|
|
39
39
|
keyPrefix(clientId) {
|
|
40
40
|
return `/${this.clientName}/${this.serverId}/${clientId}`;
|
|
@@ -43,7 +43,7 @@ var DurableObjectOAuthClientProvider = class {
|
|
|
43
43
|
return `${this.keyPrefix(clientId)}/client_info/`;
|
|
44
44
|
}
|
|
45
45
|
async clientInformation() {
|
|
46
|
-
if (!this.
|
|
46
|
+
if (!this._clientId_) {
|
|
47
47
|
return void 0;
|
|
48
48
|
}
|
|
49
49
|
return await this.storage.get(
|
|
@@ -61,7 +61,7 @@ var DurableObjectOAuthClientProvider = class {
|
|
|
61
61
|
return `${this.keyPrefix(clientId)}/token`;
|
|
62
62
|
}
|
|
63
63
|
async tokens() {
|
|
64
|
-
if (!this.
|
|
64
|
+
if (!this._clientId_) {
|
|
65
65
|
return void 0;
|
|
66
66
|
}
|
|
67
67
|
return await this.storage.get(this.tokenKey(this.clientId)) ?? void 0;
|
|
@@ -70,7 +70,7 @@ var DurableObjectOAuthClientProvider = class {
|
|
|
70
70
|
await this.storage.put(this.tokenKey(this.clientId), tokens);
|
|
71
71
|
}
|
|
72
72
|
get authUrl() {
|
|
73
|
-
return this.
|
|
73
|
+
return this._authUrl_;
|
|
74
74
|
}
|
|
75
75
|
/**
|
|
76
76
|
* Because this operates on the server side (but we need browser auth), we send this url back to the user
|
|
@@ -81,7 +81,7 @@ var DurableObjectOAuthClientProvider = class {
|
|
|
81
81
|
if (client_id) {
|
|
82
82
|
authUrl.searchParams.append("state", client_id);
|
|
83
83
|
}
|
|
84
|
-
this.
|
|
84
|
+
this._authUrl_ = authUrl.toString();
|
|
85
85
|
}
|
|
86
86
|
codeVerifierKey(clientId) {
|
|
87
87
|
return `${this.keyPrefix(clientId)}/code_verifier`;
|
|
@@ -103,4 +103,4 @@ var DurableObjectOAuthClientProvider = class {
|
|
|
103
103
|
export {
|
|
104
104
|
DurableObjectOAuthClientProvider
|
|
105
105
|
};
|
|
106
|
-
//# sourceMappingURL=chunk-
|
|
106
|
+
//# sourceMappingURL=chunk-PVQZBKN7.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["../src/mcp/do-oauth-client-provider.ts"],"sourcesContent":["import type { OAuthClientProvider } from \"@modelcontextprotocol/sdk/client/auth.js\";\nimport type {\n OAuthClientInformation,\n OAuthClientInformationFull,\n OAuthClientMetadata,\n OAuthTokens\n} from \"@modelcontextprotocol/sdk/shared/auth.js\";\n\n// A slight extension to the standard OAuthClientProvider interface because `redirectToAuthorization` doesn't give us the interface we need\n// This allows us to track authentication for a specific server and associated dynamic client registration\nexport interface AgentsOAuthProvider extends OAuthClientProvider {\n authUrl: string | undefined;\n clientId: string | undefined;\n serverId: string | undefined;\n}\n\nexport class DurableObjectOAuthClientProvider implements AgentsOAuthProvider {\n private _authUrl_: string | undefined;\n private _serverId_: string | undefined;\n private _clientId_: string | undefined;\n\n constructor(\n public storage: DurableObjectStorage,\n public clientName: string,\n public baseRedirectUrl: string\n ) {}\n\n get clientMetadata(): OAuthClientMetadata {\n return {\n client_name: this.clientName,\n client_uri: \"example.com\",\n grant_types: [\"authorization_code\", \"refresh_token\"],\n redirect_uris: [this.redirectUrl],\n response_types: [\"code\"],\n token_endpoint_auth_method: \"none\"\n };\n }\n\n get redirectUrl() {\n return `${this.baseRedirectUrl}/${this.serverId}`;\n }\n\n get clientId() {\n if (!this._clientId_) {\n throw new Error(\"Trying to access clientId before it was set\");\n }\n return this._clientId_;\n }\n\n set clientId(clientId_: string) {\n this._clientId_ = clientId_;\n }\n\n get serverId() {\n if (!this._serverId_) {\n throw new Error(\"Trying to access serverId before it was set\");\n }\n return this._serverId_;\n }\n\n set serverId(serverId_: string) {\n this._serverId_ = serverId_;\n }\n\n keyPrefix(clientId: string) {\n return `/${this.clientName}/${this.serverId}/${clientId}`;\n }\n\n clientInfoKey(clientId: string) {\n return `${this.keyPrefix(clientId)}/client_info/`;\n }\n\n async clientInformation(): Promise<OAuthClientInformation | undefined> {\n if (!this._clientId_) {\n return undefined;\n }\n return (\n (await this.storage.get<OAuthClientInformation>(\n this.clientInfoKey(this.clientId)\n )) ?? undefined\n );\n }\n\n async saveClientInformation(\n clientInformation: OAuthClientInformationFull\n ): Promise<void> {\n await this.storage.put(\n this.clientInfoKey(clientInformation.client_id),\n clientInformation\n );\n this.clientId = clientInformation.client_id;\n }\n\n tokenKey(clientId: string) {\n return `${this.keyPrefix(clientId)}/token`;\n }\n\n async tokens(): Promise<OAuthTokens | undefined> {\n if (!this._clientId_) {\n return undefined;\n }\n return (\n (await this.storage.get<OAuthTokens>(this.tokenKey(this.clientId))) ??\n undefined\n );\n }\n\n async saveTokens(tokens: OAuthTokens): Promise<void> {\n await this.storage.put(this.tokenKey(this.clientId), tokens);\n }\n\n get authUrl() {\n return this._authUrl_;\n }\n\n /**\n * Because this operates on the server side (but we need browser auth), we send this url back to the user\n * and require user interact to initiate the redirect flow\n */\n async redirectToAuthorization(authUrl: URL): Promise<void> {\n // We want to track the client ID in state here because the typescript SSE client sometimes does\n // a dynamic client registration AFTER generating this redirect URL.\n const client_id = authUrl.searchParams.get(\"client_id\");\n if (client_id) {\n authUrl.searchParams.append(\"state\", client_id);\n }\n this._authUrl_ = authUrl.toString();\n }\n\n codeVerifierKey(clientId: string) {\n return `${this.keyPrefix(clientId)}/code_verifier`;\n }\n\n async saveCodeVerifier(verifier: string): Promise<void> {\n await this.storage.put(this.codeVerifierKey(this.clientId), verifier);\n }\n\n async codeVerifier(): Promise<string> {\n const codeVerifier = await this.storage.get<string>(\n this.codeVerifierKey(this.clientId)\n );\n if (!codeVerifier) {\n throw new Error(\"No code verifier found\");\n }\n return codeVerifier;\n }\n}\n"],"mappings":";AAgBO,IAAM,mCAAN,MAAsE;AAAA,EAK3E,YACS,SACA,YACA,iBACP;AAHO;AACA;AACA;AAAA,EACN;AAAA,EAEH,IAAI,iBAAsC;AACxC,WAAO;AAAA,MACL,aAAa,KAAK;AAAA,MAClB,YAAY;AAAA,MACZ,aAAa,CAAC,sBAAsB,eAAe;AAAA,MACnD,eAAe,CAAC,KAAK,WAAW;AAAA,MAChC,gBAAgB,CAAC,MAAM;AAAA,MACvB,4BAA4B;AAAA,IAC9B;AAAA,EACF;AAAA,EAEA,IAAI,cAAc;AAChB,WAAO,GAAG,KAAK,eAAe,IAAI,KAAK,QAAQ;AAAA,EACjD;AAAA,EAEA,IAAI,WAAW;AACb,QAAI,CAAC,KAAK,YAAY;AACpB,YAAM,IAAI,MAAM,6CAA6C;AAAA,IAC/D;AACA,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAI,SAAS,WAAmB;AAC9B,SAAK,aAAa;AAAA,EACpB;AAAA,EAEA,IAAI,WAAW;AACb,QAAI,CAAC,KAAK,YAAY;AACpB,YAAM,IAAI,MAAM,6CAA6C;AAAA,IAC/D;AACA,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAI,SAAS,WAAmB;AAC9B,SAAK,aAAa;AAAA,EACpB;AAAA,EAEA,UAAU,UAAkB;AAC1B,WAAO,IAAI,KAAK,UAAU,IAAI,KAAK,QAAQ,IAAI,QAAQ;AAAA,EACzD;AAAA,EAEA,cAAc,UAAkB;AAC9B,WAAO,GAAG,KAAK,UAAU,QAAQ,CAAC;AAAA,EACpC;AAAA,EAEA,MAAM,oBAAiE;AACrE,QAAI,CAAC,KAAK,YAAY;AACpB,aAAO;AAAA,IACT;AACA,WACG,MAAM,KAAK,QAAQ;AAAA,MAClB,KAAK,cAAc,KAAK,QAAQ;AAAA,IAClC,KAAM;AAAA,EAEV;AAAA,EAEA,MAAM,sBACJ,mBACe;AACf,UAAM,KAAK,QAAQ;AAAA,MACjB,KAAK,cAAc,kBAAkB,SAAS;AAAA,MAC9C;AAAA,IACF;AACA,SAAK,WAAW,kBAAkB;AAAA,EACpC;AAAA,EAEA,SAAS,UAAkB;AACzB,WAAO,GAAG,KAAK,UAAU,QAAQ,CAAC;AAAA,EACpC;AAAA,EAEA,MAAM,SAA2C;AAC/C,QAAI,CAAC,KAAK,YAAY;AACpB,aAAO;AAAA,IACT;AACA,WACG,MAAM,KAAK,QAAQ,IAAiB,KAAK,SAAS,KAAK,QAAQ,CAAC,KACjE;AAAA,EAEJ;AAAA,EAEA,MAAM,WAAW,QAAoC;AACnD,UAAM,KAAK,QAAQ,IAAI,KAAK,SAAS,KAAK,QAAQ,GAAG,MAAM;AAAA,EAC7D;AAAA,EAEA,IAAI,UAAU;AACZ,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,wBAAwB,SAA6B;AAGzD,UAAM,YAAY,QAAQ,aAAa,IAAI,WAAW;AACtD,QAAI,WAAW;AACb,cAAQ,aAAa,OAAO,SAAS,SAAS;AAAA,IAChD;AACA,SAAK,YAAY,QAAQ,SAAS;AAAA,EACpC;AAAA,EAEA,gBAAgB,UAAkB;AAChC,WAAO,GAAG,KAAK,UAAU,QAAQ,CAAC;AAAA,EACpC;AAAA,EAEA,MAAM,iBAAiB,UAAiC;AACtD,UAAM,KAAK,QAAQ,IAAI,KAAK,gBAAgB,KAAK,QAAQ,GAAG,QAAQ;AAAA,EACtE;AAAA,EAEA,MAAM,eAAgC;AACpC,UAAM,eAAe,MAAM,KAAK,QAAQ;AAAA,MACtC,KAAK,gBAAgB,KAAK,QAAQ;AAAA,IACpC;AACA,QAAI,CAAC,cAAc;AACjB,YAAM,IAAI,MAAM,wBAAwB;AAAA,IAC1C;AACA,WAAO;AAAA,EACT;AACF;","names":[]}
|