clanka 0.0.1
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 +3 -0
- package/dist/Agent.d.ts +119 -0
- package/dist/Agent.d.ts.map +1 -0
- package/dist/Agent.js +240 -0
- package/dist/Agent.js.map +1 -0
- package/dist/AgentTools.d.ts +246 -0
- package/dist/AgentTools.d.ts.map +1 -0
- package/dist/AgentTools.js +374 -0
- package/dist/AgentTools.js.map +1 -0
- package/dist/AgentTools.test.d.ts +2 -0
- package/dist/AgentTools.test.d.ts.map +1 -0
- package/dist/AgentTools.test.js +147 -0
- package/dist/AgentTools.test.js.map +1 -0
- package/dist/ApplyPatch.d.ts +27 -0
- package/dist/ApplyPatch.d.ts.map +1 -0
- package/dist/ApplyPatch.js +343 -0
- package/dist/ApplyPatch.js.map +1 -0
- package/dist/ApplyPatch.test.d.ts +2 -0
- package/dist/ApplyPatch.test.d.ts.map +1 -0
- package/dist/ApplyPatch.test.js +99 -0
- package/dist/ApplyPatch.test.js.map +1 -0
- package/dist/Codex.d.ts +11 -0
- package/dist/Codex.d.ts.map +1 -0
- package/dist/Codex.js +14 -0
- package/dist/Codex.js.map +1 -0
- package/dist/CodexAuth.d.ts +68 -0
- package/dist/CodexAuth.d.ts.map +1 -0
- package/dist/CodexAuth.js +270 -0
- package/dist/CodexAuth.js.map +1 -0
- package/dist/CodexAuth.test.d.ts +2 -0
- package/dist/CodexAuth.test.d.ts.map +1 -0
- package/dist/CodexAuth.test.js +425 -0
- package/dist/CodexAuth.test.js.map +1 -0
- package/dist/Executor.d.ts +20 -0
- package/dist/Executor.d.ts.map +1 -0
- package/dist/Executor.js +76 -0
- package/dist/Executor.js.map +1 -0
- package/dist/OutputFormatter.d.ts +11 -0
- package/dist/OutputFormatter.d.ts.map +1 -0
- package/dist/OutputFormatter.js +5 -0
- package/dist/OutputFormatter.js.map +1 -0
- package/dist/ToolkitRenderer.d.ts +17 -0
- package/dist/ToolkitRenderer.d.ts.map +1 -0
- package/dist/ToolkitRenderer.js +25 -0
- package/dist/ToolkitRenderer.js.map +1 -0
- package/dist/TypeBuilder.d.ts +11 -0
- package/dist/TypeBuilder.d.ts.map +1 -0
- package/dist/TypeBuilder.js +383 -0
- package/dist/TypeBuilder.js.map +1 -0
- package/dist/TypeBuilder.test.d.ts +2 -0
- package/dist/TypeBuilder.test.d.ts.map +1 -0
- package/dist/TypeBuilder.test.js +243 -0
- package/dist/TypeBuilder.test.js.map +1 -0
- package/dist/index.d.ts +25 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +25 -0
- package/dist/index.js.map +1 -0
- package/package.json +72 -0
- package/src/Agent.ts +398 -0
- package/src/AgentTools.test.ts +215 -0
- package/src/AgentTools.ts +507 -0
- package/src/ApplyPatch.test.ts +154 -0
- package/src/ApplyPatch.ts +473 -0
- package/src/Codex.ts +14 -0
- package/src/CodexAuth.test.ts +729 -0
- package/src/CodexAuth.ts +571 -0
- package/src/Executor.ts +129 -0
- package/src/OutputFormatter.ts +17 -0
- package/src/ToolkitRenderer.ts +39 -0
- package/src/TypeBuilder.test.ts +508 -0
- package/src/TypeBuilder.ts +670 -0
- package/src/index.ts +29 -0
|
@@ -0,0 +1,425 @@
|
|
|
1
|
+
import { assert, describe, it } from "@effect/vitest";
|
|
2
|
+
import { Deferred, Effect, Encoding, Fiber, Option, Ref } from "effect";
|
|
3
|
+
import { HttpClient, HttpClientRequest, HttpClientResponse, } from "effect/unstable/http";
|
|
4
|
+
import { KeyValueStore } from "effect/unstable/persistence";
|
|
5
|
+
import { CodexAuth, CodexAuthError, ISSUER, STORE_PREFIX, STORE_TOKEN_KEY, TOKEN_EXPIRY_BUFFER_MS, TokenData, extractAccountIdFromClaims, extractAccountIdFromToken, parseJwtClaims, toCodexAuthKeyValueStore, toTokenStore, } from "./CodexAuth.js";
|
|
6
|
+
const createJwt = (payload) => `${Encoding.encodeBase64Url(JSON.stringify({ alg: "none" }))}.${Encoding.encodeBase64Url(payload)}.sig`;
|
|
7
|
+
const createTestJwt = (payload) => createJwt(JSON.stringify(payload));
|
|
8
|
+
const jsonResponse = (body, status = 200) => new Response(JSON.stringify(body), {
|
|
9
|
+
status,
|
|
10
|
+
headers: {
|
|
11
|
+
"content-type": "application/json",
|
|
12
|
+
},
|
|
13
|
+
});
|
|
14
|
+
const getBody = (request) => {
|
|
15
|
+
if (request.body._tag !== "Uint8Array") {
|
|
16
|
+
throw new Error("Expected request body to be a Uint8Array payload");
|
|
17
|
+
}
|
|
18
|
+
return new TextDecoder().decode(request.body.body);
|
|
19
|
+
};
|
|
20
|
+
const makeClient = Effect.fn("makeClient")(function* (handler) {
|
|
21
|
+
const attempts = yield* Ref.make(0);
|
|
22
|
+
const requests = yield* Ref.make([]);
|
|
23
|
+
const client = HttpClient.make((request) => Effect.gen(function* () {
|
|
24
|
+
const attempt = yield* Ref.updateAndGet(attempts, (count) => count + 1);
|
|
25
|
+
yield* Ref.update(requests, (current) => [...current, request]);
|
|
26
|
+
return HttpClientResponse.fromWeb(request, handler(request, attempt));
|
|
27
|
+
}));
|
|
28
|
+
return {
|
|
29
|
+
attempts,
|
|
30
|
+
client,
|
|
31
|
+
requests,
|
|
32
|
+
};
|
|
33
|
+
});
|
|
34
|
+
const makeEffectClient = Effect.fn("makeEffectClient")(function* (handler) {
|
|
35
|
+
const attempts = yield* Ref.make(0);
|
|
36
|
+
const requests = yield* Ref.make([]);
|
|
37
|
+
const client = HttpClient.make((request) => Effect.gen(function* () {
|
|
38
|
+
const attempt = yield* Ref.updateAndGet(attempts, (count) => count + 1);
|
|
39
|
+
yield* Ref.update(requests, (current) => [...current, request]);
|
|
40
|
+
return HttpClientResponse.fromWeb(request, yield* handler(request, attempt));
|
|
41
|
+
}));
|
|
42
|
+
return {
|
|
43
|
+
attempts,
|
|
44
|
+
client,
|
|
45
|
+
requests,
|
|
46
|
+
};
|
|
47
|
+
});
|
|
48
|
+
describe("CodexAuth", () => {
|
|
49
|
+
it.effect("persists token data through the prefixed schema store", Effect.fn(function* () {
|
|
50
|
+
const kvs = yield* KeyValueStore.KeyValueStore;
|
|
51
|
+
const tokenStore = toTokenStore(kvs);
|
|
52
|
+
const token = new TokenData({
|
|
53
|
+
access: "access-token",
|
|
54
|
+
refresh: "refresh-token",
|
|
55
|
+
expires: 1_700_000_000_000,
|
|
56
|
+
accountId: Option.some("account_123"),
|
|
57
|
+
});
|
|
58
|
+
yield* Effect.orDie(tokenStore.set(STORE_TOKEN_KEY, token));
|
|
59
|
+
const stored = yield* Effect.orDie(tokenStore.get(STORE_TOKEN_KEY));
|
|
60
|
+
assert.strictEqual(Option.isSome(stored), true);
|
|
61
|
+
if (Option.isNone(stored)) {
|
|
62
|
+
return;
|
|
63
|
+
}
|
|
64
|
+
assert.strictEqual(stored.value.access, token.access);
|
|
65
|
+
assert.strictEqual(stored.value.refresh, token.refresh);
|
|
66
|
+
assert.strictEqual(stored.value.expires, token.expires);
|
|
67
|
+
assert.strictEqual(Option.isSome(stored.value.accountId), true);
|
|
68
|
+
if (Option.isSome(stored.value.accountId)) {
|
|
69
|
+
assert.strictEqual(stored.value.accountId.value, "account_123");
|
|
70
|
+
}
|
|
71
|
+
const rawValue = yield* Effect.orDie(kvs.get(`${STORE_PREFIX}${STORE_TOKEN_KEY}`));
|
|
72
|
+
const unprefixedValue = yield* Effect.orDie(kvs.get(STORE_TOKEN_KEY));
|
|
73
|
+
assert.strictEqual(typeof rawValue, "string");
|
|
74
|
+
assert.strictEqual(unprefixedValue, undefined);
|
|
75
|
+
}, Effect.provide(KeyValueStore.layerMemory)));
|
|
76
|
+
it.effect("round-trips missing account ids as Option.none", Effect.fn(function* () {
|
|
77
|
+
const kvs = yield* KeyValueStore.KeyValueStore;
|
|
78
|
+
const prefixedStore = toCodexAuthKeyValueStore(kvs);
|
|
79
|
+
const tokenStore = toTokenStore(kvs);
|
|
80
|
+
const token = new TokenData({
|
|
81
|
+
access: "access-token",
|
|
82
|
+
refresh: "refresh-token",
|
|
83
|
+
expires: 1_700_000_000_000,
|
|
84
|
+
accountId: Option.none(),
|
|
85
|
+
});
|
|
86
|
+
yield* Effect.orDie(tokenStore.set(STORE_TOKEN_KEY, token));
|
|
87
|
+
const stored = yield* Effect.orDie(tokenStore.get(STORE_TOKEN_KEY));
|
|
88
|
+
assert.strictEqual(Option.isSome(stored), true);
|
|
89
|
+
if (Option.isNone(stored)) {
|
|
90
|
+
return;
|
|
91
|
+
}
|
|
92
|
+
assert.strictEqual(Option.isNone(stored.value.accountId), true);
|
|
93
|
+
assert.strictEqual(yield* Effect.orDie(prefixedStore.has(STORE_TOKEN_KEY)), true);
|
|
94
|
+
}, Effect.provide(KeyValueStore.layerMemory)));
|
|
95
|
+
it("marks tokens expired using the refresh buffer", () => {
|
|
96
|
+
const now = Date.now();
|
|
97
|
+
const expiredSoon = new TokenData({
|
|
98
|
+
access: "access-token",
|
|
99
|
+
refresh: "refresh-token",
|
|
100
|
+
expires: now + TOKEN_EXPIRY_BUFFER_MS - 1_000,
|
|
101
|
+
accountId: Option.none(),
|
|
102
|
+
});
|
|
103
|
+
const stillValid = new TokenData({
|
|
104
|
+
access: "access-token",
|
|
105
|
+
refresh: "refresh-token",
|
|
106
|
+
expires: now + TOKEN_EXPIRY_BUFFER_MS + 60_000,
|
|
107
|
+
accountId: Option.none(),
|
|
108
|
+
});
|
|
109
|
+
assert.strictEqual(expiredSoon.isExpired(), true);
|
|
110
|
+
assert.strictEqual(stillValid.isExpired(), false);
|
|
111
|
+
});
|
|
112
|
+
it("constructs CodexAuthError with the expected tagged shape", () => {
|
|
113
|
+
const error = new CodexAuthError({
|
|
114
|
+
reason: "RefreshFailed",
|
|
115
|
+
message: "Could not refresh the token",
|
|
116
|
+
});
|
|
117
|
+
assert.strictEqual(error._tag, "CodexAuthError");
|
|
118
|
+
assert.strictEqual(error.reason, "RefreshFailed");
|
|
119
|
+
assert.strictEqual(error.message, "Could not refresh the token");
|
|
120
|
+
});
|
|
121
|
+
it("parses valid JWT claims from a base64url payload", () => {
|
|
122
|
+
assert.deepStrictEqual(Option.getOrUndefined(parseJwtClaims(createTestJwt({
|
|
123
|
+
email: "test@example.com",
|
|
124
|
+
chatgpt_account_id: "acc-123",
|
|
125
|
+
}))), {
|
|
126
|
+
chatgpt_account_id: "acc-123",
|
|
127
|
+
});
|
|
128
|
+
});
|
|
129
|
+
it("returns none for JWTs without three parts", () => {
|
|
130
|
+
assert.strictEqual(Option.isNone(parseJwtClaims("invalid")), true);
|
|
131
|
+
assert.strictEqual(Option.isNone(parseJwtClaims("only.two")), true);
|
|
132
|
+
});
|
|
133
|
+
it("returns none for invalid base64url payloads", () => {
|
|
134
|
+
assert.strictEqual(Option.isNone(parseJwtClaims("a.!!!invalid!!!.b")), true);
|
|
135
|
+
});
|
|
136
|
+
it("returns none for invalid JSON payloads", () => {
|
|
137
|
+
assert.strictEqual(Option.isNone(parseJwtClaims(createJwt("not json"))), true);
|
|
138
|
+
});
|
|
139
|
+
it("ignores malformed claim types instead of failing the parse", () => {
|
|
140
|
+
assert.deepStrictEqual(Option.getOrUndefined(parseJwtClaims(createTestJwt({ chatgpt_account_id: 123 }))), {});
|
|
141
|
+
});
|
|
142
|
+
it("keeps valid account ids when other claim locations are malformed", () => {
|
|
143
|
+
assert.strictEqual(Option.getOrUndefined(extractAccountIdFromToken(createTestJwt({
|
|
144
|
+
chatgpt_account_id: "acc-root",
|
|
145
|
+
"https://api.openai.com/auth": "invalid",
|
|
146
|
+
}))), "acc-root");
|
|
147
|
+
assert.strictEqual(Option.getOrUndefined(extractAccountIdFromToken(createTestJwt({
|
|
148
|
+
"https://api.openai.com/auth": {
|
|
149
|
+
chatgpt_account_id: "acc-nested",
|
|
150
|
+
},
|
|
151
|
+
organizations: [123],
|
|
152
|
+
}))), "acc-nested");
|
|
153
|
+
});
|
|
154
|
+
it("extracts account ids from the documented claim locations", () => {
|
|
155
|
+
assert.strictEqual(Option.getOrUndefined(extractAccountIdFromClaims({ chatgpt_account_id: "acc-root" })), "acc-root");
|
|
156
|
+
assert.strictEqual(Option.getOrUndefined(extractAccountIdFromClaims({
|
|
157
|
+
"https://api.openai.com/auth": {
|
|
158
|
+
chatgpt_account_id: "acc-nested",
|
|
159
|
+
},
|
|
160
|
+
})), "acc-nested");
|
|
161
|
+
assert.strictEqual(Option.getOrUndefined(extractAccountIdFromClaims({
|
|
162
|
+
organizations: [{ id: "org-123" }, { id: "org-456" }],
|
|
163
|
+
})), "org-123");
|
|
164
|
+
});
|
|
165
|
+
it("prefers root claims over nested and organization fallbacks", () => {
|
|
166
|
+
assert.strictEqual(Option.getOrUndefined(extractAccountIdFromClaims({
|
|
167
|
+
chatgpt_account_id: "acc-root",
|
|
168
|
+
"https://api.openai.com/auth": {
|
|
169
|
+
chatgpt_account_id: "acc-nested",
|
|
170
|
+
},
|
|
171
|
+
organizations: [{ id: "org-123" }],
|
|
172
|
+
})), "acc-root");
|
|
173
|
+
});
|
|
174
|
+
it("treats empty root and nested account ids as missing", () => {
|
|
175
|
+
assert.strictEqual(Option.getOrUndefined(extractAccountIdFromClaims({
|
|
176
|
+
chatgpt_account_id: "",
|
|
177
|
+
"https://api.openai.com/auth": {
|
|
178
|
+
chatgpt_account_id: "acc-nested",
|
|
179
|
+
},
|
|
180
|
+
})), "acc-nested");
|
|
181
|
+
assert.strictEqual(Option.getOrUndefined(extractAccountIdFromClaims({
|
|
182
|
+
chatgpt_account_id: "",
|
|
183
|
+
"https://api.openai.com/auth": {
|
|
184
|
+
chatgpt_account_id: "",
|
|
185
|
+
},
|
|
186
|
+
organizations: [{ id: "org-123" }],
|
|
187
|
+
})), "org-123");
|
|
188
|
+
});
|
|
189
|
+
it("treats empty organization ids as missing", () => {
|
|
190
|
+
assert.strictEqual(Option.isNone(extractAccountIdFromClaims({
|
|
191
|
+
organizations: [{ id: "" }],
|
|
192
|
+
})), true);
|
|
193
|
+
});
|
|
194
|
+
it("returns none when no account id claim is present", () => {
|
|
195
|
+
assert.strictEqual(Option.isNone(extractAccountIdFromClaims({
|
|
196
|
+
organizations: [],
|
|
197
|
+
})), true);
|
|
198
|
+
});
|
|
199
|
+
it("extracts account ids directly from JWTs", () => {
|
|
200
|
+
assert.strictEqual(Option.getOrUndefined(extractAccountIdFromToken(createTestJwt({
|
|
201
|
+
"https://api.openai.com/auth": {
|
|
202
|
+
chatgpt_account_id: "acc-token",
|
|
203
|
+
},
|
|
204
|
+
}))), "acc-token");
|
|
205
|
+
assert.strictEqual(Option.isNone(extractAccountIdFromToken("invalid")), true);
|
|
206
|
+
});
|
|
207
|
+
it.effect("preserves the stored account id when refreshed tokens omit parseable claims", () => Effect.gen(function* () {
|
|
208
|
+
const kvs = yield* KeyValueStore.KeyValueStore;
|
|
209
|
+
const tokenStore = toTokenStore(kvs);
|
|
210
|
+
yield* Effect.orDie(tokenStore.set(STORE_TOKEN_KEY, new TokenData({
|
|
211
|
+
access: "stale-access-token",
|
|
212
|
+
refresh: "stale-refresh-token",
|
|
213
|
+
expires: Date.now() - 60_000,
|
|
214
|
+
accountId: Option.some("persisted-account"),
|
|
215
|
+
})));
|
|
216
|
+
const { attempts, client } = yield* makeClient(() => jsonResponse({
|
|
217
|
+
id_token: "invalid",
|
|
218
|
+
access_token: "also-invalid",
|
|
219
|
+
refresh_token: "next-refresh-token",
|
|
220
|
+
expires_in: 120,
|
|
221
|
+
}));
|
|
222
|
+
const auth = yield* CodexAuth.make.pipe(Effect.provideService(HttpClient.HttpClient, client));
|
|
223
|
+
const token = yield* auth.get;
|
|
224
|
+
assert.strictEqual(token.refresh, "next-refresh-token");
|
|
225
|
+
assert.strictEqual(Option.getOrUndefined(token.accountId), "persisted-account");
|
|
226
|
+
assert.strictEqual(yield* Ref.get(attempts), 1);
|
|
227
|
+
const stored = yield* Effect.orDie(tokenStore.get(STORE_TOKEN_KEY));
|
|
228
|
+
assert.strictEqual(Option.isSome(stored), true);
|
|
229
|
+
if (Option.isNone(stored)) {
|
|
230
|
+
return;
|
|
231
|
+
}
|
|
232
|
+
assert.strictEqual(Option.getOrUndefined(stored.value.accountId), "persisted-account");
|
|
233
|
+
}).pipe(Effect.provide(KeyValueStore.layerMemory)));
|
|
234
|
+
it.effect("falls back to device auth when refreshing an expired token fails", () => Effect.gen(function* () {
|
|
235
|
+
const kvs = yield* KeyValueStore.KeyValueStore;
|
|
236
|
+
const tokenStore = toTokenStore(kvs);
|
|
237
|
+
yield* Effect.orDie(tokenStore.set(STORE_TOKEN_KEY, new TokenData({
|
|
238
|
+
access: "stale-access-token",
|
|
239
|
+
refresh: "stale-refresh-token",
|
|
240
|
+
expires: Date.now() - 60_000,
|
|
241
|
+
accountId: Option.some("stale-account"),
|
|
242
|
+
})));
|
|
243
|
+
const { client, requests } = yield* makeClient((request) => {
|
|
244
|
+
if (request.url === `${ISSUER}/api/accounts/deviceauth/usercode`) {
|
|
245
|
+
return jsonResponse({
|
|
246
|
+
device_auth_id: "device-auth-id",
|
|
247
|
+
user_code: "WXYZ-9876",
|
|
248
|
+
interval: "1",
|
|
249
|
+
});
|
|
250
|
+
}
|
|
251
|
+
if (request.url === `${ISSUER}/api/accounts/deviceauth/token`) {
|
|
252
|
+
return jsonResponse({
|
|
253
|
+
authorization_code: "authorization-code",
|
|
254
|
+
code_verifier: "code-verifier",
|
|
255
|
+
});
|
|
256
|
+
}
|
|
257
|
+
if (request.url === `${ISSUER}/oauth/token`) {
|
|
258
|
+
const body = new URLSearchParams(getBody(request));
|
|
259
|
+
if (body.get("grant_type") === "refresh_token") {
|
|
260
|
+
return new Response(null, { status: 401 });
|
|
261
|
+
}
|
|
262
|
+
return jsonResponse({
|
|
263
|
+
id_token: createTestJwt({
|
|
264
|
+
chatgpt_account_id: "account-from-id",
|
|
265
|
+
}),
|
|
266
|
+
access_token: createTestJwt({
|
|
267
|
+
chatgpt_account_id: "account-from-access",
|
|
268
|
+
}),
|
|
269
|
+
refresh_token: "next-refresh-token",
|
|
270
|
+
expires_in: 120,
|
|
271
|
+
});
|
|
272
|
+
}
|
|
273
|
+
return new Response(null, { status: 500 });
|
|
274
|
+
});
|
|
275
|
+
const auth = yield* CodexAuth.make.pipe(Effect.provideService(HttpClient.HttpClient, client));
|
|
276
|
+
const token = yield* auth.get;
|
|
277
|
+
assert.strictEqual(token.refresh, "next-refresh-token");
|
|
278
|
+
assert.strictEqual(Option.getOrUndefined(token.accountId), "account-from-id");
|
|
279
|
+
const stored = yield* Effect.orDie(tokenStore.get(STORE_TOKEN_KEY));
|
|
280
|
+
assert.strictEqual(Option.isSome(stored), true);
|
|
281
|
+
if (Option.isSome(stored)) {
|
|
282
|
+
assert.strictEqual(stored.value.refresh, "next-refresh-token");
|
|
283
|
+
}
|
|
284
|
+
const seenRequests = yield* Ref.get(requests);
|
|
285
|
+
assert.strictEqual(seenRequests.length, 4);
|
|
286
|
+
assert.strictEqual(seenRequests[0]?.url, `${ISSUER}/oauth/token`);
|
|
287
|
+
assert.strictEqual(new URLSearchParams(getBody(seenRequests[0])).get("grant_type"), "refresh_token");
|
|
288
|
+
assert.strictEqual(new URLSearchParams(getBody(seenRequests[3])).get("grant_type"), "authorization_code");
|
|
289
|
+
}).pipe(Effect.provide(KeyValueStore.layerMemory)));
|
|
290
|
+
it.effect("clears corrupted persisted tokens before re-authenticating", () => Effect.gen(function* () {
|
|
291
|
+
const kvs = yield* KeyValueStore.KeyValueStore;
|
|
292
|
+
yield* Effect.orDie(kvs.set(`${STORE_PREFIX}${STORE_TOKEN_KEY}`, "not-json"));
|
|
293
|
+
const { attempts, client } = yield* makeClient((request) => {
|
|
294
|
+
if (request.url === `${ISSUER}/api/accounts/deviceauth/usercode`) {
|
|
295
|
+
return jsonResponse({
|
|
296
|
+
device_auth_id: "device-auth-id",
|
|
297
|
+
user_code: "ABCD-EFGH",
|
|
298
|
+
interval: "1",
|
|
299
|
+
});
|
|
300
|
+
}
|
|
301
|
+
if (request.url === `${ISSUER}/api/accounts/deviceauth/token`) {
|
|
302
|
+
return jsonResponse({
|
|
303
|
+
authorization_code: "authorization-code",
|
|
304
|
+
code_verifier: "code-verifier",
|
|
305
|
+
});
|
|
306
|
+
}
|
|
307
|
+
if (request.url === `${ISSUER}/oauth/token`) {
|
|
308
|
+
return jsonResponse({
|
|
309
|
+
access_token: createTestJwt({
|
|
310
|
+
chatgpt_account_id: "fresh-account",
|
|
311
|
+
}),
|
|
312
|
+
refresh_token: "fresh-refresh-token",
|
|
313
|
+
expires_in: 120,
|
|
314
|
+
});
|
|
315
|
+
}
|
|
316
|
+
return new Response(null, { status: 500 });
|
|
317
|
+
});
|
|
318
|
+
const auth = yield* CodexAuth.make.pipe(Effect.provideService(HttpClient.HttpClient, client));
|
|
319
|
+
assert.strictEqual(yield* Effect.orDie(kvs.get(`${STORE_PREFIX}${STORE_TOKEN_KEY}`)), undefined);
|
|
320
|
+
assert.strictEqual(yield* Ref.get(attempts), 0);
|
|
321
|
+
const token = yield* auth.get;
|
|
322
|
+
assert.strictEqual(token.refresh, "fresh-refresh-token");
|
|
323
|
+
assert.strictEqual(Option.getOrUndefined(token.accountId), "fresh-account");
|
|
324
|
+
assert.strictEqual(yield* Ref.get(attempts), 3);
|
|
325
|
+
}).pipe(Effect.provide(KeyValueStore.layerMemory)));
|
|
326
|
+
it.effect("serializes concurrent get calls behind one refresh", () => Effect.gen(function* () {
|
|
327
|
+
const kvs = yield* KeyValueStore.KeyValueStore;
|
|
328
|
+
const tokenStore = toTokenStore(kvs);
|
|
329
|
+
yield* Effect.orDie(tokenStore.set(STORE_TOKEN_KEY, new TokenData({
|
|
330
|
+
access: "expired-access-token",
|
|
331
|
+
refresh: "refresh-token",
|
|
332
|
+
expires: Date.now() - 60_000,
|
|
333
|
+
accountId: Option.none(),
|
|
334
|
+
})));
|
|
335
|
+
const refreshStarted = yield* Deferred.make();
|
|
336
|
+
const releaseRefresh = yield* Deferred.make();
|
|
337
|
+
const { attempts, client } = yield* makeEffectClient((request) => {
|
|
338
|
+
if (request.url !== `${ISSUER}/oauth/token`) {
|
|
339
|
+
return Effect.succeed(new Response(null, { status: 500 }));
|
|
340
|
+
}
|
|
341
|
+
const body = new URLSearchParams(getBody(request));
|
|
342
|
+
if (body.get("grant_type") !== "refresh_token") {
|
|
343
|
+
return Effect.succeed(new Response(null, { status: 500 }));
|
|
344
|
+
}
|
|
345
|
+
return Effect.gen(function* () {
|
|
346
|
+
yield* Deferred.succeed(refreshStarted, void 0);
|
|
347
|
+
yield* Deferred.await(releaseRefresh);
|
|
348
|
+
return jsonResponse({
|
|
349
|
+
access_token: createTestJwt({
|
|
350
|
+
chatgpt_account_id: "fresh-account",
|
|
351
|
+
}),
|
|
352
|
+
refresh_token: "fresh-refresh-token",
|
|
353
|
+
expires_in: 120,
|
|
354
|
+
});
|
|
355
|
+
});
|
|
356
|
+
});
|
|
357
|
+
const auth = yield* CodexAuth.make.pipe(Effect.provideService(HttpClient.HttpClient, client));
|
|
358
|
+
const firstFiber = yield* auth.get.pipe(Effect.forkChild({ startImmediately: true }));
|
|
359
|
+
yield* Deferred.await(refreshStarted);
|
|
360
|
+
const secondFiber = yield* auth.get.pipe(Effect.forkChild({ startImmediately: true }));
|
|
361
|
+
yield* Effect.yieldNow;
|
|
362
|
+
assert.strictEqual(yield* Ref.get(attempts), 1);
|
|
363
|
+
yield* Deferred.succeed(releaseRefresh, void 0);
|
|
364
|
+
const first = yield* Fiber.join(firstFiber);
|
|
365
|
+
const second = yield* Fiber.join(secondFiber);
|
|
366
|
+
assert.strictEqual(yield* Ref.get(attempts), 1);
|
|
367
|
+
assert.strictEqual(first.refresh, "fresh-refresh-token");
|
|
368
|
+
assert.strictEqual(second.refresh, "fresh-refresh-token");
|
|
369
|
+
assert.strictEqual(Option.getOrUndefined(first.accountId), "fresh-account");
|
|
370
|
+
assert.strictEqual(Option.getOrUndefined(second.accountId), "fresh-account");
|
|
371
|
+
}).pipe(Effect.provide(KeyValueStore.layerMemory)));
|
|
372
|
+
it.effect("forces device auth on authenticate and clears cache plus storage on logout", () => Effect.gen(function* () {
|
|
373
|
+
const kvs = yield* KeyValueStore.KeyValueStore;
|
|
374
|
+
const tokenStore = toTokenStore(kvs);
|
|
375
|
+
yield* Effect.orDie(tokenStore.set(STORE_TOKEN_KEY, new TokenData({
|
|
376
|
+
access: "cached-access-token",
|
|
377
|
+
refresh: "cached-refresh-token",
|
|
378
|
+
expires: Date.now() + 600_000,
|
|
379
|
+
accountId: Option.some("cached-account"),
|
|
380
|
+
})));
|
|
381
|
+
let deviceFlowCount = 0;
|
|
382
|
+
const { attempts, client } = yield* makeClient((request) => {
|
|
383
|
+
if (request.url === `${ISSUER}/api/accounts/deviceauth/usercode`) {
|
|
384
|
+
deviceFlowCount += 1;
|
|
385
|
+
return jsonResponse({
|
|
386
|
+
device_auth_id: `device-auth-${deviceFlowCount}`,
|
|
387
|
+
user_code: `CODE-${deviceFlowCount}`,
|
|
388
|
+
interval: "1",
|
|
389
|
+
});
|
|
390
|
+
}
|
|
391
|
+
if (request.url === `${ISSUER}/api/accounts/deviceauth/token`) {
|
|
392
|
+
return jsonResponse({
|
|
393
|
+
authorization_code: `authorization-code-${deviceFlowCount}`,
|
|
394
|
+
code_verifier: `code-verifier-${deviceFlowCount}`,
|
|
395
|
+
});
|
|
396
|
+
}
|
|
397
|
+
if (request.url === `${ISSUER}/oauth/token`) {
|
|
398
|
+
return jsonResponse({
|
|
399
|
+
access_token: createTestJwt({
|
|
400
|
+
chatgpt_account_id: `device-account-${deviceFlowCount}`,
|
|
401
|
+
}),
|
|
402
|
+
refresh_token: `device-refresh-${deviceFlowCount}`,
|
|
403
|
+
expires_in: 120,
|
|
404
|
+
});
|
|
405
|
+
}
|
|
406
|
+
return new Response(null, { status: 500 });
|
|
407
|
+
});
|
|
408
|
+
const auth = yield* CodexAuth.make.pipe(Effect.provideService(HttpClient.HttpClient, client));
|
|
409
|
+
const cachedToken = yield* auth.get;
|
|
410
|
+
assert.strictEqual(cachedToken.refresh, "cached-refresh-token");
|
|
411
|
+
assert.strictEqual(yield* Ref.get(attempts), 0);
|
|
412
|
+
const authenticated = yield* auth.authenticate;
|
|
413
|
+
assert.strictEqual(authenticated.refresh, "device-refresh-1");
|
|
414
|
+
assert.strictEqual(Option.getOrUndefined(authenticated.accountId), "device-account-1");
|
|
415
|
+
assert.strictEqual(yield* Ref.get(attempts), 3);
|
|
416
|
+
yield* auth.logout;
|
|
417
|
+
const storedAfterLogout = yield* Effect.orDie(tokenStore.get(STORE_TOKEN_KEY));
|
|
418
|
+
assert.strictEqual(Option.isNone(storedAfterLogout), true);
|
|
419
|
+
const tokenAfterLogout = yield* auth.get;
|
|
420
|
+
assert.strictEqual(tokenAfterLogout.refresh, "device-refresh-2");
|
|
421
|
+
assert.strictEqual(Option.getOrUndefined(tokenAfterLogout.accountId), "device-account-2");
|
|
422
|
+
assert.strictEqual(yield* Ref.get(attempts), 6);
|
|
423
|
+
}).pipe(Effect.provide(KeyValueStore.layerMemory)));
|
|
424
|
+
});
|
|
425
|
+
//# sourceMappingURL=CodexAuth.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"CodexAuth.test.js","sourceRoot":"","sources":["../src/CodexAuth.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,gBAAgB,CAAA;AACrD,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,QAAQ,CAAA;AACvE,OAAO,EACL,UAAU,EACV,iBAAiB,EACjB,kBAAkB,GACnB,MAAM,sBAAsB,CAAA;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAA;AAC3D,OAAO,EACL,SAAS,EACT,cAAc,EACd,MAAM,EACN,YAAY,EACZ,eAAe,EACf,sBAAsB,EACtB,SAAS,EACT,0BAA0B,EAC1B,yBAAyB,EACzB,cAAc,EACd,wBAAwB,EACxB,YAAY,GACb,MAAM,gBAAgB,CAAA;AAEvB,MAAM,SAAS,GAAG,CAAC,OAAe,EAAU,EAAE,CAC5C,GAAG,QAAQ,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,QAAQ,CAAC,eAAe,CAAC,OAAO,CAAC,MAAM,CAAA;AAEzG,MAAM,aAAa,GAAG,CAAC,OAAgC,EAAU,EAAE,CACjE,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAA;AAEpC,MAAM,YAAY,GAAG,CAAC,IAAa,EAAE,MAAM,GAAG,GAAG,EAAY,EAAE,CAC7D,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;IACjC,MAAM;IACN,OAAO,EAAE;QACP,cAAc,EAAE,kBAAkB;KACnC;CACF,CAAC,CAAA;AAEJ,MAAM,OAAO,GAAG,CAAC,OAA4C,EAAU,EAAE;IACvE,IAAI,OAAO,CAAC,IAAI,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;QACvC,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAA;IACrE,CAAC;IAED,OAAO,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;AACpD,CAAC,CAAA;AAED,MAAM,UAAU,GAAG,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,QAAQ,CAAC,EAClD,OAGa;IAEb,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IACnC,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,IAAI,CAC9B,EAAE,CACH,CAAA;IACD,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CACzC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;QAClB,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,GAAG,CAAC,CAAC,CAAA;QACvE,KAAK,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,GAAG,OAAO,EAAE,OAAO,CAAC,CAAC,CAAA;QAC/D,OAAO,kBAAkB,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAA;IACvE,CAAC,CAAC,CACH,CAAA;IAED,OAAO;QACL,QAAQ;QACR,MAAM;QACN,QAAQ;KACA,CAAA;AACZ,CAAC,CAAC,CAAA;AAEF,MAAM,gBAAgB,GAAG,MAAM,CAAC,EAAE,CAAC,kBAAkB,CAAC,CAAC,QAAQ,CAAC,EAC9D,OAG4B;IAE5B,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IACnC,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,IAAI,CAC9B,EAAE,CACH,CAAA;IACD,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CACzC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;QAClB,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,GAAG,CAAC,CAAC,CAAA;QACvE,KAAK,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,GAAG,OAAO,EAAE,OAAO,CAAC,CAAC,CAAA;QAC/D,OAAO,kBAAkB,CAAC,OAAO,CAC/B,OAAO,EACP,KAAK,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CACjC,CAAA;IACH,CAAC,CAAC,CACH,CAAA;IAED,OAAO;QACL,QAAQ;QACR,MAAM;QACN,QAAQ;KACA,CAAA;AACZ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,EAAE,CAAC,MAAM,CACP,uDAAuD,EACvD,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC;QACjB,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,aAAa,CAAC,aAAa,CAAA;QAC9C,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,CAAA;QACpC,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC;YAC1B,MAAM,EAAE,cAAc;YACtB,OAAO,EAAE,eAAe;YACxB,OAAO,EAAE,iBAAiB;YAC1B,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC;SACtC,CAAC,CAAA;QAEF,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC,CAAA;QAE3D,MAAM,MAAM,GAAG,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAA;QAEnE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,CAAA;QAC/C,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1B,OAAM;QACR,CAAC;QAED,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,CAAA;QACrD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAA;QACvD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAA;QACvD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,CAAA;QAC/D,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;YAC1C,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,EAAE,aAAa,CAAC,CAAA;QACjE,CAAC;QAED,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAClC,GAAG,CAAC,GAAG,CAAC,GAAG,YAAY,GAAG,eAAe,EAAE,CAAC,CAC7C,CAAA;QACD,MAAM,eAAe,GAAG,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAA;QAErE,MAAM,CAAC,WAAW,CAAC,OAAO,QAAQ,EAAE,QAAQ,CAAC,CAAA;QAC7C,MAAM,CAAC,WAAW,CAAC,eAAe,EAAE,SAAS,CAAC,CAAA;IAChD,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,CAC9C,CAAA;IAED,EAAE,CAAC,MAAM,CACP,gDAAgD,EAChD,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC;QACjB,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,aAAa,CAAC,aAAa,CAAA;QAC9C,MAAM,aAAa,GAAG,wBAAwB,CAAC,GAAG,CAAC,CAAA;QACnD,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,CAAA;QACpC,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC;YAC1B,MAAM,EAAE,cAAc;YACtB,OAAO,EAAE,eAAe;YACxB,OAAO,EAAE,iBAAiB;YAC1B,SAAS,EAAE,MAAM,CAAC,IAAI,EAAE;SACzB,CAAC,CAAA;QAEF,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC,CAAA;QAE3D,MAAM,MAAM,GAAG,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAA;QAEnE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,CAAA;QAC/C,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1B,OAAM;QACR,CAAC;QAED,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,CAAA;QAC/D,MAAM,CAAC,WAAW,CAChB,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,EACvD,IAAI,CACL,CAAA;IACH,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,CAC9C,CAAA;IAED,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;QACtB,MAAM,WAAW,GAAG,IAAI,SAAS,CAAC;YAChC,MAAM,EAAE,cAAc;YACtB,OAAO,EAAE,eAAe;YACxB,OAAO,EAAE,GAAG,GAAG,sBAAsB,GAAG,KAAK;YAC7C,SAAS,EAAE,MAAM,CAAC,IAAI,EAAE;SACzB,CAAC,CAAA;QACF,MAAM,UAAU,GAAG,IAAI,SAAS,CAAC;YAC/B,MAAM,EAAE,cAAc;YACtB,OAAO,EAAE,eAAe;YACxB,OAAO,EAAE,GAAG,GAAG,sBAAsB,GAAG,MAAM;YAC9C,SAAS,EAAE,MAAM,CAAC,IAAI,EAAE;SACzB,CAAC,CAAA;QAEF,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,SAAS,EAAE,EAAE,IAAI,CAAC,CAAA;QACjD,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,SAAS,EAAE,EAAE,KAAK,CAAC,CAAA;IACnD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,MAAM,KAAK,GAAG,IAAI,cAAc,CAAC;YAC/B,MAAM,EAAE,eAAe;YACvB,OAAO,EAAE,6BAA6B;SACvC,CAAC,CAAA;QAEF,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAA;QAChD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,EAAE,eAAe,CAAC,CAAA;QACjD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE,6BAA6B,CAAC,CAAA;IAClE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,CAAC,eAAe,CACpB,MAAM,CAAC,cAAc,CACnB,cAAc,CACZ,aAAa,CAAC;YACZ,KAAK,EAAE,kBAAkB;YACzB,kBAAkB,EAAE,SAAS;SAC9B,CAAC,CACH,CACF,EACD;YACE,kBAAkB,EAAE,SAAS;SAC9B,CACF,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;QAClE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;IACrE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,mBAAmB,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;IAC9E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,EACpD,IAAI,CACL,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,CAAC,eAAe,CACpB,MAAM,CAAC,cAAc,CACnB,cAAc,CAAC,aAAa,CAAC,EAAE,kBAAkB,EAAE,GAAG,EAAE,CAAC,CAAC,CAC3D,EACD,EAAE,CACH,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,cAAc,CACnB,yBAAyB,CACvB,aAAa,CAAC;YACZ,kBAAkB,EAAE,UAAU;YAC9B,6BAA6B,EAAE,SAAS;SACzC,CAAC,CACH,CACF,EACD,UAAU,CACX,CAAA;QACD,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,cAAc,CACnB,yBAAyB,CACvB,aAAa,CAAC;YACZ,6BAA6B,EAAE;gBAC7B,kBAAkB,EAAE,YAAY;aACjC;YACD,aAAa,EAAE,CAAC,GAAG,CAAC;SACrB,CAAC,CACH,CACF,EACD,YAAY,CACb,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,cAAc,CACnB,0BAA0B,CAAC,EAAE,kBAAkB,EAAE,UAAU,EAAE,CAAC,CAC/D,EACD,UAAU,CACX,CAAA;QACD,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,cAAc,CACnB,0BAA0B,CAAC;YACzB,6BAA6B,EAAE;gBAC7B,kBAAkB,EAAE,YAAY;aACjC;SACF,CAAC,CACH,EACD,YAAY,CACb,CAAA;QACD,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,cAAc,CACnB,0BAA0B,CAAC;YACzB,aAAa,EAAE,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC;SACtD,CAAC,CACH,EACD,SAAS,CACV,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,cAAc,CACnB,0BAA0B,CAAC;YACzB,kBAAkB,EAAE,UAAU;YAC9B,6BAA6B,EAAE;gBAC7B,kBAAkB,EAAE,YAAY;aACjC;YACD,aAAa,EAAE,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC;SACnC,CAAC,CACH,EACD,UAAU,CACX,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,cAAc,CACnB,0BAA0B,CAAC;YACzB,kBAAkB,EAAE,EAAE;YACtB,6BAA6B,EAAE;gBAC7B,kBAAkB,EAAE,YAAY;aACjC;SACF,CAAC,CACH,EACD,YAAY,CACb,CAAA;QACD,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,cAAc,CACnB,0BAA0B,CAAC;YACzB,kBAAkB,EAAE,EAAE;YACtB,6BAA6B,EAAE;gBAC7B,kBAAkB,EAAE,EAAE;aACvB;YACD,aAAa,EAAE,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC;SACnC,CAAC,CACH,EACD,SAAS,CACV,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,MAAM,CACX,0BAA0B,CAAC;YACzB,aAAa,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;SAC5B,CAAC,CACH,EACD,IAAI,CACL,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,MAAM,CACX,0BAA0B,CAAC;YACzB,aAAa,EAAE,EAAE;SAClB,CAAC,CACH,EACD,IAAI,CACL,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,cAAc,CACnB,yBAAyB,CACvB,aAAa,CAAC;YACZ,6BAA6B,EAAE;gBAC7B,kBAAkB,EAAE,WAAW;aAChC;SACF,CAAC,CACH,CACF,EACD,WAAW,CACZ,CAAA;QACD,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,MAAM,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC,EACnD,IAAI,CACL,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,MAAM,CACP,6EAA6E,EAC7E,GAAG,EAAE,CACH,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;QAClB,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,aAAa,CAAC,aAAa,CAAA;QAC9C,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,CAAA;QACpC,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CACjB,UAAU,CAAC,GAAG,CACZ,eAAe,EACf,IAAI,SAAS,CAAC;YACZ,MAAM,EAAE,oBAAoB;YAC5B,OAAO,EAAE,qBAAqB;YAC9B,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM;YAC5B,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,mBAAmB,CAAC;SAC5C,CAAC,CACH,CACF,CAAA;QAED,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC,CAAC,UAAU,CAAC,GAAG,EAAE,CAClD,YAAY,CAAC;YACX,QAAQ,EAAE,SAAS;YACnB,YAAY,EAAE,cAAc;YAC5B,aAAa,EAAE,oBAAoB;YACnC,UAAU,EAAE,GAAG;SAChB,CAAC,CACH,CAAA;QAED,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CACrC,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,UAAU,EAAE,MAAM,CAAC,CACrD,CAAA;QAED,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAA;QAE7B,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE,oBAAoB,CAAC,CAAA;QACvD,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,SAAS,CAAC,EACtC,mBAAmB,CACpB,CAAA;QACD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAA;QAE/C,MAAM,MAAM,GAAG,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAA;QACnE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,CAAA;QAC/C,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1B,OAAM;QACR,CAAC;QAED,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,EAC7C,mBAAmB,CACpB,CAAA;IACH,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,CACrD,CAAA;IAED,EAAE,CAAC,MAAM,CACP,kEAAkE,EAClE,GAAG,EAAE,CACH,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;QAClB,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,aAAa,CAAC,aAAa,CAAA;QAC9C,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,CAAA;QACpC,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CACjB,UAAU,CAAC,GAAG,CACZ,eAAe,EACf,IAAI,SAAS,CAAC;YACZ,MAAM,EAAE,oBAAoB;YAC5B,OAAO,EAAE,qBAAqB;YAC9B,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM;YAC5B,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC;SACxC,CAAC,CACH,CACF,CAAA;QAED,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,EAAE;YACzD,IAAI,OAAO,CAAC,GAAG,KAAK,GAAG,MAAM,mCAAmC,EAAE,CAAC;gBACjE,OAAO,YAAY,CAAC;oBAClB,cAAc,EAAE,gBAAgB;oBAChC,SAAS,EAAE,WAAW;oBACtB,QAAQ,EAAE,GAAG;iBACd,CAAC,CAAA;YACJ,CAAC;YAED,IAAI,OAAO,CAAC,GAAG,KAAK,GAAG,MAAM,gCAAgC,EAAE,CAAC;gBAC9D,OAAO,YAAY,CAAC;oBAClB,kBAAkB,EAAE,oBAAoB;oBACxC,aAAa,EAAE,eAAe;iBAC/B,CAAC,CAAA;YACJ,CAAC;YAED,IAAI,OAAO,CAAC,GAAG,KAAK,GAAG,MAAM,cAAc,EAAE,CAAC;gBAC5C,MAAM,IAAI,GAAG,IAAI,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAA;gBAClD,IAAI,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,KAAK,eAAe,EAAE,CAAC;oBAC/C,OAAO,IAAI,QAAQ,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAA;gBAC5C,CAAC;gBAED,OAAO,YAAY,CAAC;oBAClB,QAAQ,EAAE,aAAa,CAAC;wBACtB,kBAAkB,EAAE,iBAAiB;qBACtC,CAAC;oBACF,YAAY,EAAE,aAAa,CAAC;wBAC1B,kBAAkB,EAAE,qBAAqB;qBAC1C,CAAC;oBACF,aAAa,EAAE,oBAAoB;oBACnC,UAAU,EAAE,GAAG;iBAChB,CAAC,CAAA;YACJ,CAAC;YAED,OAAO,IAAI,QAAQ,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAA;QAC5C,CAAC,CAAC,CAAA;QAEF,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CACrC,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,UAAU,EAAE,MAAM,CAAC,CACrD,CAAA;QAED,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAA;QAE7B,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE,oBAAoB,CAAC,CAAA;QACvD,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,SAAS,CAAC,EACtC,iBAAiB,CAClB,CAAA;QAED,MAAM,MAAM,GAAG,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAA;QACnE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,CAAA;QAC/C,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1B,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,oBAAoB,CAAC,CAAA;QAChE,CAAC;QAED,MAAM,YAAY,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;QAC7C,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QAC1C,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,GAAG,MAAM,cAAc,CAAC,CAAA;QACjE,MAAM,CAAC,WAAW,CAChB,IAAI,eAAe,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC,EAChE,eAAe,CAChB,CAAA;QACD,MAAM,CAAC,WAAW,CAChB,IAAI,eAAe,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC,EAChE,oBAAoB,CACrB,CAAA;IACH,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,CACrD,CAAA;IAED,EAAE,CAAC,MAAM,CAAC,4DAA4D,EAAE,GAAG,EAAE,CAC3E,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;QAClB,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,aAAa,CAAC,aAAa,CAAA;QAC9C,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CACjB,GAAG,CAAC,GAAG,CAAC,GAAG,YAAY,GAAG,eAAe,EAAE,EAAE,UAAU,CAAC,CACzD,CAAA;QAED,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,EAAE;YACzD,IAAI,OAAO,CAAC,GAAG,KAAK,GAAG,MAAM,mCAAmC,EAAE,CAAC;gBACjE,OAAO,YAAY,CAAC;oBAClB,cAAc,EAAE,gBAAgB;oBAChC,SAAS,EAAE,WAAW;oBACtB,QAAQ,EAAE,GAAG;iBACd,CAAC,CAAA;YACJ,CAAC;YAED,IAAI,OAAO,CAAC,GAAG,KAAK,GAAG,MAAM,gCAAgC,EAAE,CAAC;gBAC9D,OAAO,YAAY,CAAC;oBAClB,kBAAkB,EAAE,oBAAoB;oBACxC,aAAa,EAAE,eAAe;iBAC/B,CAAC,CAAA;YACJ,CAAC;YAED,IAAI,OAAO,CAAC,GAAG,KAAK,GAAG,MAAM,cAAc,EAAE,CAAC;gBAC5C,OAAO,YAAY,CAAC;oBAClB,YAAY,EAAE,aAAa,CAAC;wBAC1B,kBAAkB,EAAE,eAAe;qBACpC,CAAC;oBACF,aAAa,EAAE,qBAAqB;oBACpC,UAAU,EAAE,GAAG;iBAChB,CAAC,CAAA;YACJ,CAAC;YAED,OAAO,IAAI,QAAQ,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAA;QAC5C,CAAC,CAAC,CAAA;QAEF,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CACrC,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,UAAU,EAAE,MAAM,CAAC,CACrD,CAAA;QAED,MAAM,CAAC,WAAW,CAChB,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,YAAY,GAAG,eAAe,EAAE,CAAC,CAAC,EACjE,SAAS,CACV,CAAA;QACD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAA;QAE/C,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAA;QAE7B,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE,qBAAqB,CAAC,CAAA;QACxD,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,SAAS,CAAC,EACtC,eAAe,CAChB,CAAA;QACD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAA;IACjD,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,CACnD,CAAA;IAED,EAAE,CAAC,MAAM,CAAC,oDAAoD,EAAE,GAAG,EAAE,CACnE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;QAClB,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,aAAa,CAAC,aAAa,CAAA;QAC9C,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,CAAA;QACpC,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CACjB,UAAU,CAAC,GAAG,CACZ,eAAe,EACf,IAAI,SAAS,CAAC;YACZ,MAAM,EAAE,sBAAsB;YAC9B,OAAO,EAAE,eAAe;YACxB,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM;YAC5B,SAAS,EAAE,MAAM,CAAC,IAAI,EAAE;SACzB,CAAC,CACH,CACF,CAAA;QAED,MAAM,cAAc,GAAG,KAAK,CAAC,CAAC,QAAQ,CAAC,IAAI,EAAQ,CAAA;QACnD,MAAM,cAAc,GAAG,KAAK,CAAC,CAAC,QAAQ,CAAC,IAAI,EAAQ,CAAA;QACnD,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC,CAAC,gBAAgB,CAAC,CAAC,OAAO,EAAE,EAAE;YAC/D,IAAI,OAAO,CAAC,GAAG,KAAK,GAAG,MAAM,cAAc,EAAE,CAAC;gBAC5C,OAAO,MAAM,CAAC,OAAO,CAAC,IAAI,QAAQ,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,CAAA;YAC5D,CAAC;YAED,MAAM,IAAI,GAAG,IAAI,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAA;YAClD,IAAI,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,KAAK,eAAe,EAAE,CAAC;gBAC/C,OAAO,MAAM,CAAC,OAAO,CAAC,IAAI,QAAQ,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,CAAA;YAC5D,CAAC;YAED,OAAO,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;gBACzB,KAAK,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC,CAAA;gBAC/C,KAAK,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,cAAc,CAAC,CAAA;gBACrC,OAAO,YAAY,CAAC;oBAClB,YAAY,EAAE,aAAa,CAAC;wBAC1B,kBAAkB,EAAE,eAAe;qBACpC,CAAC;oBACF,aAAa,EAAE,qBAAqB;oBACpC,UAAU,EAAE,GAAG;iBAChB,CAAC,CAAA;YACJ,CAAC,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;QAEF,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CACrC,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,UAAU,EAAE,MAAM,CAAC,CACrD,CAAA;QAED,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CACrC,MAAM,CAAC,SAAS,CAAC,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC,CAC7C,CAAA;QACD,KAAK,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,cAAc,CAAC,CAAA;QAErC,MAAM,WAAW,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CACtC,MAAM,CAAC,SAAS,CAAC,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC,CAC7C,CAAA;QACD,KAAK,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAA;QAEtB,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAA;QAE/C,KAAK,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC,CAAA;QAE/C,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QAC3C,MAAM,MAAM,GAAG,KAAK,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QAE7C,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAA;QAC/C,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE,qBAAqB,CAAC,CAAA;QACxD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,EAAE,qBAAqB,CAAC,CAAA;QACzD,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,SAAS,CAAC,EACtC,eAAe,CAChB,CAAA;QACD,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,SAAS,CAAC,EACvC,eAAe,CAChB,CAAA;IACH,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,CACnD,CAAA;IAED,EAAE,CAAC,MAAM,CACP,4EAA4E,EAC5E,GAAG,EAAE,CACH,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;QAClB,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,aAAa,CAAC,aAAa,CAAA;QAC9C,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,CAAA;QACpC,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CACjB,UAAU,CAAC,GAAG,CACZ,eAAe,EACf,IAAI,SAAS,CAAC;YACZ,MAAM,EAAE,qBAAqB;YAC7B,OAAO,EAAE,sBAAsB;YAC/B,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO;YAC7B,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC;SACzC,CAAC,CACH,CACF,CAAA;QAED,IAAI,eAAe,GAAG,CAAC,CAAA;QACvB,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,EAAE;YACzD,IAAI,OAAO,CAAC,GAAG,KAAK,GAAG,MAAM,mCAAmC,EAAE,CAAC;gBACjE,eAAe,IAAI,CAAC,CAAA;gBACpB,OAAO,YAAY,CAAC;oBAClB,cAAc,EAAE,eAAe,eAAe,EAAE;oBAChD,SAAS,EAAE,QAAQ,eAAe,EAAE;oBACpC,QAAQ,EAAE,GAAG;iBACd,CAAC,CAAA;YACJ,CAAC;YAED,IAAI,OAAO,CAAC,GAAG,KAAK,GAAG,MAAM,gCAAgC,EAAE,CAAC;gBAC9D,OAAO,YAAY,CAAC;oBAClB,kBAAkB,EAAE,sBAAsB,eAAe,EAAE;oBAC3D,aAAa,EAAE,iBAAiB,eAAe,EAAE;iBAClD,CAAC,CAAA;YACJ,CAAC;YAED,IAAI,OAAO,CAAC,GAAG,KAAK,GAAG,MAAM,cAAc,EAAE,CAAC;gBAC5C,OAAO,YAAY,CAAC;oBAClB,YAAY,EAAE,aAAa,CAAC;wBAC1B,kBAAkB,EAAE,kBAAkB,eAAe,EAAE;qBACxD,CAAC;oBACF,aAAa,EAAE,kBAAkB,eAAe,EAAE;oBAClD,UAAU,EAAE,GAAG;iBAChB,CAAC,CAAA;YACJ,CAAC;YAED,OAAO,IAAI,QAAQ,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAA;QAC5C,CAAC,CAAC,CAAA;QAEF,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CACrC,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,UAAU,EAAE,MAAM,CAAC,CACrD,CAAA;QAED,MAAM,WAAW,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAA;QACnC,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,OAAO,EAAE,sBAAsB,CAAC,CAAA;QAC/D,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAA;QAE/C,MAAM,aAAa,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAA;QAC9C,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,OAAO,EAAE,kBAAkB,CAAC,CAAA;QAC7D,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,cAAc,CAAC,aAAa,CAAC,SAAS,CAAC,EAC9C,kBAAkB,CACnB,CAAA;QACD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAA;QAE/C,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAA;QAClB,MAAM,iBAAiB,GAAG,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAC3C,UAAU,CAAC,GAAG,CAAC,eAAe,CAAC,CAChC,CAAA;QACD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,CAAA;QAE1D,MAAM,gBAAgB,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAA;QACxC,MAAM,CAAC,WAAW,CAAC,gBAAgB,CAAC,OAAO,EAAE,kBAAkB,CAAC,CAAA;QAChE,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,cAAc,CAAC,gBAAgB,CAAC,SAAS,CAAC,EACjD,kBAAkB,CACnB,CAAA;QACD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAA;IACjD,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,CACrD,CAAA;AACH,CAAC,CAAC,CAAA"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @since 1.0.0
|
|
3
|
+
*/
|
|
4
|
+
import { Layer, ServiceMap, Stream } from "effect";
|
|
5
|
+
import { Tool, Toolkit } from "effect/unstable/ai";
|
|
6
|
+
declare const Executor_base: ServiceMap.ServiceClass<Executor, "clanka/Executor", {
|
|
7
|
+
execute<Tools extends Record<string, Tool.Any>>(options: {
|
|
8
|
+
readonly tools: Toolkit.WithHandler<Tools>;
|
|
9
|
+
readonly script: string;
|
|
10
|
+
}): Stream.Stream<string, never, Tool.HandlerServices<Tools[keyof Tools]>>;
|
|
11
|
+
}>;
|
|
12
|
+
/**
|
|
13
|
+
* @since 1.0.0
|
|
14
|
+
* @category Services
|
|
15
|
+
*/
|
|
16
|
+
export declare class Executor extends Executor_base {
|
|
17
|
+
static readonly layer: Layer.Layer<Executor, never, never>;
|
|
18
|
+
}
|
|
19
|
+
export {};
|
|
20
|
+
//# sourceMappingURL=Executor.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"Executor.d.ts","sourceRoot":"","sources":["../src/Executor.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EAKL,KAAK,EAEL,UAAU,EACV,MAAM,EACP,MAAM,QAAQ,CAAA;AACf,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAA;;YAYtC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW;QACvD,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAA;QAC1C,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;KACxB,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC;;AAV9E;;;GAGG;AACH,qBAAa,QAAS,SAAQ,aAQR;IACpB,MAAM,CAAC,QAAQ,CAAC,KAAK,sCA2DpB;CACF"}
|
package/dist/Executor.js
ADDED
|
@@ -0,0 +1,76 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @since 1.0.0
|
|
3
|
+
*/
|
|
4
|
+
import { Cause, Console, Effect, FiberSet, Layer, Queue, ServiceMap, Stream, } from "effect";
|
|
5
|
+
import { Tool, Toolkit } from "effect/unstable/ai";
|
|
6
|
+
import * as NodeConsole from "node:console";
|
|
7
|
+
import * as NodeVm from "node:vm";
|
|
8
|
+
import { Writable } from "node:stream";
|
|
9
|
+
/**
|
|
10
|
+
* @since 1.0.0
|
|
11
|
+
* @category Services
|
|
12
|
+
*/
|
|
13
|
+
export class Executor extends ServiceMap.Service()("clanka/Executor") {
|
|
14
|
+
static layer = Layer.effect(Executor,
|
|
15
|
+
// oxlint-disable-next-line require-yield
|
|
16
|
+
Effect.gen(function* () {
|
|
17
|
+
const execute = Effect.fnUntraced(function* (options) {
|
|
18
|
+
const output = yield* Queue.unbounded();
|
|
19
|
+
yield* Effect.gen(function* () {
|
|
20
|
+
const console = yield* Console.Console;
|
|
21
|
+
const services = yield* Effect.services();
|
|
22
|
+
const runPromise = yield* FiberSet.makeRuntimePromise();
|
|
23
|
+
const script = new NodeVm.Script(`async function main() {
|
|
24
|
+
${options.script}
|
|
25
|
+
}`);
|
|
26
|
+
const sandbox = {
|
|
27
|
+
main: defaultMain,
|
|
28
|
+
console,
|
|
29
|
+
fetch,
|
|
30
|
+
process: undefined,
|
|
31
|
+
};
|
|
32
|
+
for (const [name, tool] of Object.entries(options.tools.tools)) {
|
|
33
|
+
const handler = services.mapUnsafe.get(tool.id);
|
|
34
|
+
// oxlint-disable-next-line typescript/no-explicit-any
|
|
35
|
+
sandbox[name] = function (params) {
|
|
36
|
+
return handler
|
|
37
|
+
.handler(params, {})
|
|
38
|
+
.pipe(Effect.provideServices(handler.services), runPromise);
|
|
39
|
+
};
|
|
40
|
+
}
|
|
41
|
+
script.runInNewContext(sandbox, {
|
|
42
|
+
timeout: 1000,
|
|
43
|
+
});
|
|
44
|
+
yield* Effect.promise(sandbox.main);
|
|
45
|
+
}).pipe(Effect.timeout("3 minutes"), Effect.catchCause(Effect.logFatal), Effect.provideServiceEffect(Console.Console, makeConsole(output)), Effect.scoped, Effect.ensuring(Queue.end(output)), Effect.forkScoped);
|
|
46
|
+
return Stream.fromQueue(output);
|
|
47
|
+
}, Stream.unwrap);
|
|
48
|
+
return Executor.of({
|
|
49
|
+
execute,
|
|
50
|
+
});
|
|
51
|
+
}));
|
|
52
|
+
}
|
|
53
|
+
const defaultMain = () => Promise.resolve();
|
|
54
|
+
const makeConsole = Effect.fn(function* (queue) {
|
|
55
|
+
const writable = new QueueWriteStream(queue);
|
|
56
|
+
const newConsole = new NodeConsole.Console(writable);
|
|
57
|
+
yield* Effect.addFinalizer(() => {
|
|
58
|
+
writable.end();
|
|
59
|
+
return Effect.void;
|
|
60
|
+
});
|
|
61
|
+
return newConsole;
|
|
62
|
+
});
|
|
63
|
+
class QueueWriteStream extends Writable {
|
|
64
|
+
queue;
|
|
65
|
+
constructor(queue) {
|
|
66
|
+
super();
|
|
67
|
+
this.queue = queue;
|
|
68
|
+
}
|
|
69
|
+
_write(
|
|
70
|
+
// oxlint-disable-next-line typescript/no-explicit-any
|
|
71
|
+
chunk, _encoding, callback) {
|
|
72
|
+
Queue.offerUnsafe(this.queue, chunk.toString());
|
|
73
|
+
callback();
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
//# sourceMappingURL=Executor.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"Executor.js","sourceRoot":"","sources":["../src/Executor.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EACL,KAAK,EACL,OAAO,EACP,MAAM,EACN,QAAQ,EACR,KAAK,EACL,KAAK,EACL,UAAU,EACV,MAAM,GACP,MAAM,QAAQ,CAAA;AACf,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAA;AAClD,OAAO,KAAK,WAAW,MAAM,cAAc,CAAA;AAC3C,OAAO,KAAK,MAAM,MAAM,SAAS,CAAA;AACjC,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAA;AAEtC;;;GAGG;AACH,MAAM,OAAO,QAAS,SAAQ,UAAU,CAAC,OAAO,EAQ7C,CAAC,iBAAiB,CAAC;IACpB,MAAM,CAAU,KAAK,GAAG,KAAK,CAAC,MAAM,CAClC,QAAQ;IACR,yCAAyC;IACzC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;QAClB,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,EAEzC,OAGD;YACC,MAAM,MAAM,GAAG,KAAK,CAAC,CAAC,KAAK,CAAC,SAAS,EAAsB,CAAA;YAE3D,KAAK,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;gBACzB,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,OAAO,CAAC,OAAO,CAAA;gBACtC,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAA;gBACzC,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,QAAQ,CAAC,kBAAkB,EAAE,CAAA;gBAEvD,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,MAAM,CAAC;EACzC,OAAO,CAAC,MAAM;EACd,CAAC,CAAA;gBACO,MAAM,OAAO,GAAkB;oBAC7B,IAAI,EAAE,WAAW;oBACjB,OAAO;oBACP,KAAK;oBACL,OAAO,EAAE,SAAS;iBACnB,CAAA;gBAED,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;oBAC/D,MAAM,OAAO,GAAG,QAAQ,CAAC,SAAS,CAAC,GAAG,CACpC,IAAI,CAAC,EAAE,CACgB,CAAA;oBACzB,sDAAsD;oBACtD,OAAO,CAAC,IAAI,CAAC,GAAG,UAAU,MAAW;wBACnC,OAAO,OAAO;6BACX,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;6BACnB,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,UAAU,CAAC,CAAA;oBAC/D,CAAC,CAAA;gBACH,CAAC;gBAED,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE;oBAC9B,OAAO,EAAE,IAAI;iBACd,CAAC,CAAA;gBACF,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;YACrC,CAAC,CAAC,CAAC,IAAI,CACL,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,EAC3B,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,EAClC,MAAM,CAAC,oBAAoB,CAAC,OAAO,CAAC,OAAO,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,EACjE,MAAM,CAAC,MAAM,EACb,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,EAClC,MAAM,CAAC,UAAU,CAClB,CAAA;YAED,OAAO,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAA;QACjC,CAAC,EAAE,MAAM,CAAC,MAAM,CAAC,CAAA;QAEjB,OAAO,QAAQ,CAAC,EAAE,CAAC;YACjB,OAAO;SACR,CAAC,CAAA;IACJ,CAAC,CAAC,CACH,CAAA;;AASH,MAAM,WAAW,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAA;AAE3C,MAAM,WAAW,GAAG,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,EACrC,KAAsC;IAEtC,MAAM,QAAQ,GAAG,IAAI,gBAAgB,CAAC,KAAK,CAAC,CAAA;IAC5C,MAAM,UAAU,GAAG,IAAI,WAAW,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAA;IACpD,KAAK,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,GAAG,EAAE;QAC9B,QAAQ,CAAC,GAAG,EAAE,CAAA;QACd,OAAO,MAAM,CAAC,IAAI,CAAA;IACpB,CAAC,CAAC,CAAA;IACF,OAAO,UAAU,CAAA;AACnB,CAAC,CAAC,CAAA;AAEF,MAAM,gBAAiB,SAAQ,QAAQ;IAC5B,KAAK,CAAmC;IACjD,YAAY,KAAwC;QAClD,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,KAAK,GAAG,KAAK,CAAA;IACpB,CAAC;IACD,MAAM;IACJ,sDAAsD;IACtD,KAAU,EACV,SAAyB,EACzB,QAAwC;QAExC,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAA;QAC/C,QAAQ,EAAE,CAAA;IACZ,CAAC;CACF"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @since 1.0.0
|
|
3
|
+
*/
|
|
4
|
+
import { Sink } from "effect";
|
|
5
|
+
import type { Output } from "./Agent.ts";
|
|
6
|
+
/**
|
|
7
|
+
* @since 1.0.0
|
|
8
|
+
* @category Models
|
|
9
|
+
*/
|
|
10
|
+
export type OutputFormatter<E = never, R = never> = Sink.Sink<void, Output, never, E, R>;
|
|
11
|
+
//# sourceMappingURL=OutputFormatter.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"OutputFormatter.d.ts","sourceRoot":"","sources":["../src/OutputFormatter.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAA;AAC7B,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,YAAY,CAAA;AAExC;;;GAGG;AACH,MAAM,MAAM,eAAe,CAAC,CAAC,GAAG,KAAK,EAAE,CAAC,GAAG,KAAK,IAAI,IAAI,CAAC,IAAI,CAC3D,IAAI,EACJ,MAAM,EACN,KAAK,EACL,CAAC,EACD,CAAC,CACF,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"OutputFormatter.js","sourceRoot":"","sources":["../src/OutputFormatter.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAA"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @since 1.0.0
|
|
3
|
+
*/
|
|
4
|
+
import { Layer, ServiceMap } from "effect";
|
|
5
|
+
import { Tool, Toolkit } from "effect/unstable/ai";
|
|
6
|
+
declare const ToolkitRenderer_base: ServiceMap.ServiceClass<ToolkitRenderer, "clanka/ToolkitRenderer", {
|
|
7
|
+
render<Tools extends Record<string, Tool.Any>>(tools: Toolkit.Toolkit<Tools>): string;
|
|
8
|
+
}>;
|
|
9
|
+
/**
|
|
10
|
+
* @since 1.0.0
|
|
11
|
+
* @category Services
|
|
12
|
+
*/
|
|
13
|
+
export declare class ToolkitRenderer extends ToolkitRenderer_base {
|
|
14
|
+
static readonly layer: Layer.Layer<ToolkitRenderer, never, never>;
|
|
15
|
+
}
|
|
16
|
+
export {};
|
|
17
|
+
//# sourceMappingURL=ToolkitRenderer.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ToolkitRenderer.d.ts","sourceRoot":"","sources":["../src/ToolkitRenderer.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EAAS,KAAK,EAAa,UAAU,EAAE,MAAM,QAAQ,CAAA;AAC5D,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAA;;WAWvC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,SACpC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,GAC5B,MAAM;;AATb;;;GAGG;AACH,qBAAa,eAAgB,SAAQ,oBAOR;IAC3B,MAAM,CAAC,QAAQ,CAAC,KAAK,6CAiBnB;CACH"}
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @since 1.0.0
|
|
3
|
+
*/
|
|
4
|
+
import { Array, Layer, SchemaAST, ServiceMap } from "effect";
|
|
5
|
+
import { Tool, Toolkit } from "effect/unstable/ai";
|
|
6
|
+
import * as TypeBuilder from "./TypeBuilder.js";
|
|
7
|
+
import { memoize } from "effect/Function";
|
|
8
|
+
/**
|
|
9
|
+
* @since 1.0.0
|
|
10
|
+
* @category Services
|
|
11
|
+
*/
|
|
12
|
+
export class ToolkitRenderer extends ServiceMap.Service()("clanka/ToolkitRenderer") {
|
|
13
|
+
static layer = Layer.succeed(ToolkitRenderer, {
|
|
14
|
+
render: memoize((tools) => {
|
|
15
|
+
const output = Array.empty();
|
|
16
|
+
for (const [name, tool] of Object.entries(tools.tools)) {
|
|
17
|
+
const paramName = SchemaAST.resolveIdentifier(tool.parametersSchema.ast) ?? "options";
|
|
18
|
+
output.push(`/** ${tool.description} */
|
|
19
|
+
declare function ${name}(${paramName}: ${TypeBuilder.render(tool.parametersSchema)}): Promise<${TypeBuilder.render(tool.successSchema)}>`);
|
|
20
|
+
}
|
|
21
|
+
return output.join("\n\n");
|
|
22
|
+
}),
|
|
23
|
+
});
|
|
24
|
+
}
|
|
25
|
+
//# sourceMappingURL=ToolkitRenderer.js.map
|