@imagekit/api-mcp 7.3.0 → 7.4.0

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.
Files changed (144) hide show
  1. package/auth.d.mts +6 -0
  2. package/auth.d.mts.map +1 -0
  3. package/auth.d.ts +6 -0
  4. package/auth.d.ts.map +1 -0
  5. package/{headers.mjs → auth.js} +21 -2
  6. package/auth.js.map +1 -0
  7. package/{headers.js → auth.mjs} +16 -6
  8. package/auth.mjs.map +1 -0
  9. package/code-tool-paths.cjs +8 -0
  10. package/code-tool-paths.cjs.map +1 -0
  11. package/code-tool-paths.d.cts +2 -0
  12. package/code-tool-paths.d.cts.map +1 -0
  13. package/code-tool-types.d.mts.map +1 -1
  14. package/code-tool-types.d.ts.map +1 -1
  15. package/code-tool-worker.d.mts +5 -0
  16. package/code-tool-worker.d.mts.map +1 -0
  17. package/code-tool-worker.d.ts +5 -0
  18. package/code-tool-worker.d.ts.map +1 -0
  19. package/code-tool-worker.js +314 -0
  20. package/code-tool-worker.js.map +1 -0
  21. package/code-tool-worker.mjs +276 -0
  22. package/code-tool-worker.mjs.map +1 -0
  23. package/code-tool.d.mts +11 -2
  24. package/code-tool.d.mts.map +1 -1
  25. package/code-tool.d.ts +11 -2
  26. package/code-tool.d.ts.map +1 -1
  27. package/code-tool.js +285 -42
  28. package/code-tool.js.map +1 -1
  29. package/code-tool.mjs +253 -43
  30. package/code-tool.mjs.map +1 -1
  31. package/docs-search-tool.d.mts +11 -3
  32. package/docs-search-tool.d.mts.map +1 -1
  33. package/docs-search-tool.d.ts +11 -3
  34. package/docs-search-tool.d.ts.map +1 -1
  35. package/docs-search-tool.js +58 -5
  36. package/docs-search-tool.js.map +1 -1
  37. package/docs-search-tool.mjs +57 -5
  38. package/docs-search-tool.mjs.map +1 -1
  39. package/http.d.mts +6 -3
  40. package/http.d.mts.map +1 -1
  41. package/http.d.ts +6 -3
  42. package/http.d.ts.map +1 -1
  43. package/http.js +130 -28
  44. package/http.js.map +1 -1
  45. package/http.mjs +130 -28
  46. package/http.mjs.map +1 -1
  47. package/index.js +17 -12
  48. package/index.js.map +1 -1
  49. package/index.mjs +17 -12
  50. package/index.mjs.map +1 -1
  51. package/instructions.d.mts +5 -0
  52. package/instructions.d.mts.map +1 -0
  53. package/instructions.d.ts +5 -0
  54. package/instructions.d.ts.map +1 -0
  55. package/instructions.js +61 -0
  56. package/instructions.js.map +1 -0
  57. package/instructions.mjs +55 -0
  58. package/instructions.mjs.map +1 -0
  59. package/local-docs-search.d.mts +28 -0
  60. package/local-docs-search.d.mts.map +1 -0
  61. package/local-docs-search.d.ts +28 -0
  62. package/local-docs-search.d.ts.map +1 -0
  63. package/local-docs-search.js +2807 -0
  64. package/local-docs-search.js.map +1 -0
  65. package/local-docs-search.mjs +2767 -0
  66. package/local-docs-search.mjs.map +1 -0
  67. package/logger.d.mts +7 -0
  68. package/logger.d.mts.map +1 -0
  69. package/logger.d.ts +7 -0
  70. package/logger.d.ts.map +1 -0
  71. package/logger.js +29 -0
  72. package/logger.js.map +1 -0
  73. package/logger.mjs +22 -0
  74. package/logger.mjs.map +1 -0
  75. package/methods.d.mts +10 -0
  76. package/methods.d.mts.map +1 -0
  77. package/methods.d.ts +10 -0
  78. package/methods.d.ts.map +1 -0
  79. package/methods.js +349 -0
  80. package/methods.js.map +1 -0
  81. package/methods.mjs +345 -0
  82. package/methods.mjs.map +1 -0
  83. package/options.d.mts +12 -0
  84. package/options.d.mts.map +1 -1
  85. package/options.d.ts +12 -0
  86. package/options.d.ts.map +1 -1
  87. package/options.js +81 -12
  88. package/options.js.map +1 -1
  89. package/options.mjs +81 -12
  90. package/options.mjs.map +1 -1
  91. package/package.json +89 -15
  92. package/server.d.mts +18 -11
  93. package/server.d.mts.map +1 -1
  94. package/server.d.ts +18 -11
  95. package/server.d.ts.map +1 -1
  96. package/server.js +89 -49
  97. package/server.js.map +1 -1
  98. package/server.mjs +88 -45
  99. package/server.mjs.map +1 -1
  100. package/src/{headers.ts → auth.ts} +18 -1
  101. package/src/code-tool-paths.cts +5 -0
  102. package/src/code-tool-types.ts +1 -0
  103. package/src/code-tool-worker.ts +327 -0
  104. package/src/code-tool.ts +339 -55
  105. package/src/docs-search-tool.ts +86 -7
  106. package/src/http.ts +154 -33
  107. package/src/index.ts +19 -13
  108. package/src/instructions.ts +83 -0
  109. package/src/local-docs-search.ts +3380 -0
  110. package/src/logger.ts +28 -0
  111. package/src/methods.ts +370 -0
  112. package/src/options.ts +105 -12
  113. package/src/server.ts +111 -57
  114. package/src/stdio.ts +9 -4
  115. package/src/types.ts +15 -4
  116. package/src/util.ts +25 -0
  117. package/stdio.d.mts +2 -1
  118. package/stdio.d.mts.map +1 -1
  119. package/stdio.d.ts +2 -1
  120. package/stdio.d.ts.map +1 -1
  121. package/stdio.js +8 -4
  122. package/stdio.js.map +1 -1
  123. package/stdio.mjs +8 -4
  124. package/stdio.mjs.map +1 -1
  125. package/types.d.mts +14 -1
  126. package/types.d.mts.map +1 -1
  127. package/types.d.ts +14 -1
  128. package/types.d.ts.map +1 -1
  129. package/types.js.map +1 -1
  130. package/types.mjs.map +1 -1
  131. package/util.d.mts +4 -0
  132. package/util.d.mts.map +1 -0
  133. package/util.d.ts +4 -0
  134. package/util.d.ts.map +1 -0
  135. package/util.js +30 -0
  136. package/util.js.map +1 -0
  137. package/util.mjs +24 -0
  138. package/util.mjs.map +1 -0
  139. package/headers.d.mts +0 -4
  140. package/headers.d.mts.map +0 -1
  141. package/headers.d.ts +0 -4
  142. package/headers.d.ts.map +0 -1
  143. package/headers.js.map +0 -1
  144. package/headers.mjs.map +0 -1
package/code-tool.mjs CHANGED
@@ -1,9 +1,10 @@
1
1
  // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
- import { asErrorResult, asTextContentResult } from "./types.mjs";
3
- import { readEnv, requireValue } from "./server.mjs";
2
+ import { asErrorResult, asTextContentResult, } from "./types.mjs";
3
+ import { readEnv, requireValue } from "./util.mjs";
4
+ import { getLogger } from "./logger.mjs";
4
5
  const prompt = `Runs JavaScript code to interact with the Image Kit API.
5
6
 
6
- You are a skilled programmer writing code to interface with the service.
7
+ You are a skilled TypeScript programmer writing code to interface with the service.
7
8
  Define an async function named "run" that takes a single parameter of an initialized SDK client and it will be run.
8
9
  For example:
9
10
 
@@ -19,7 +20,9 @@ You will be returned anything that your function returns, plus the results of an
19
20
  Do not add try-catch blocks for single API calls. The tool will handle errors for you.
20
21
  Do not add comments unless necessary for generating better code.
21
22
  Code will run in a container, and cannot interact with the network outside of the given SDK client.
22
- Variables will not persist between calls, so make sure to return or log any data you might need later.`;
23
+ Variables will not persist between calls, so make sure to return or log any data you might need later.
24
+ Remember that you are writing TypeScript code, so you need to be careful with your types.
25
+ Always type dynamic key-value stores explicitly as Record<string, YourValueType> instead of {}.`;
23
26
  /**
24
27
  * A tool that runs code against a copy of the SDK.
25
28
  *
@@ -27,9 +30,13 @@ Variables will not persist between calls, so make sure to return or log any data
27
30
  * we expose a single tool that can be used to search for endpoints by name, resource, operation, or tag, and then
28
31
  * a generic endpoint that can be used to invoke any endpoint with the provided arguments.
29
32
  *
30
- * @param endpoints - The endpoints to include in the list.
33
+ * @param blockedMethods - The methods to block for code execution. Blocking is done by simple string
34
+ * matching, so it is not secure against obfuscation. For stronger security, block in the downstream API
35
+ * with limited API keys.
36
+ * @param codeExecutionMode - Whether to execute code in a local Deno environment or in a remote
37
+ * sandbox environment hosted by Stainless.
31
38
  */
32
- export function codeTool() {
39
+ export function codeTool({ blockedMethods, codeExecutionMode, }) {
33
40
  const metadata = { resource: 'all', operation: 'write', tags: [] };
34
41
  const tool = {
35
42
  name: 'execute',
@@ -49,47 +56,250 @@ export function codeTool() {
49
56
  required: ['code'],
50
57
  },
51
58
  };
52
- const handler = async (client, args) => {
59
+ const logger = getLogger();
60
+ const handler = async ({ reqContext, args, }) => {
53
61
  const code = args.code;
54
- const intent = args.intent;
55
- // this is not required, but passing a Stainless API key for the matching project_name
56
- // will allow you to run code-mode queries against non-published versions of your SDK.
57
- const stainlessAPIKey = readEnv('STAINLESS_API_KEY');
58
- const codeModeEndpoint = readEnv('CODE_MODE_ENDPOINT_URL') ?? 'https://api.stainless.com/api/ai/code-tool';
59
- const res = await fetch(codeModeEndpoint, {
60
- method: 'POST',
61
- headers: {
62
- ...(stainlessAPIKey && { Authorization: stainlessAPIKey }),
63
- 'Content-Type': 'application/json',
64
- client_envs: JSON.stringify({
65
- IMAGEKIT_PRIVATE_KEY: requireValue(readEnv('IMAGEKIT_PRIVATE_KEY') ?? client.privateKey, 'set IMAGEKIT_PRIVATE_KEY environment variable or provide privateKey client option'),
66
- OPTIONAL_IMAGEKIT_IGNORES_THIS: readEnv('OPTIONAL_IMAGEKIT_IGNORES_THIS') ?? client.password ?? undefined,
67
- IMAGEKIT_WEBHOOK_SECRET: readEnv('IMAGEKIT_WEBHOOK_SECRET') ?? client.webhookSecret ?? undefined,
68
- IMAGE_KIT_BASE_URL: readEnv('IMAGE_KIT_BASE_URL') ?? client.baseURL ?? undefined,
69
- }),
70
- },
71
- body: JSON.stringify({
72
- project_name: 'imagekit',
73
- code,
74
- intent,
75
- client_opts: {},
76
- }),
77
- });
78
- if (!res.ok) {
79
- throw new Error(`${res.status}: ${res.statusText} error when trying to contact Code Tool server. Details: ${await res.text()}`);
62
+ // Do very basic blocking of code that includes forbidden method names.
63
+ //
64
+ // WARNING: This is not secure against obfuscation and other evasion methods. If
65
+ // stronger security blocks are required, then these should be enforced in the downstream
66
+ // API (e.g., by having users call the MCP server with API keys with limited permissions).
67
+ if (blockedMethods) {
68
+ const blockedMatches = blockedMethods.filter((method) => code.includes(method.fullyQualifiedName));
69
+ if (blockedMatches.length > 0) {
70
+ return asErrorResult(`The following methods have been blocked by the MCP server and cannot be used in code execution: ${blockedMatches
71
+ .map((m) => m.fullyQualifiedName)
72
+ .join(', ')}`);
73
+ }
80
74
  }
81
- const { is_error, result, log_lines, err_lines } = (await res.json());
82
- const hasLogs = log_lines.length > 0 || err_lines.length > 0;
83
- const output = {
84
- result,
85
- ...(log_lines.length > 0 && { log_lines }),
86
- ...(err_lines.length > 0 && { err_lines }),
87
- };
88
- if (is_error) {
89
- return asErrorResult(typeof result === 'string' && !hasLogs ? result : JSON.stringify(output, null, 2));
75
+ let result;
76
+ const startTime = Date.now();
77
+ if (codeExecutionMode === 'local') {
78
+ logger.debug('Executing code in local Deno environment');
79
+ result = await localDenoHandler({ reqContext, args });
90
80
  }
91
- return asTextContentResult(output);
81
+ else {
82
+ logger.debug('Executing code in remote Stainless environment');
83
+ result = await remoteStainlessHandler({ reqContext, args });
84
+ }
85
+ logger.info({
86
+ codeExecutionMode,
87
+ durationMs: Date.now() - startTime,
88
+ isError: result.isError,
89
+ contentRows: result.content?.length ?? 0,
90
+ }, 'Got code tool execution result');
91
+ return result;
92
92
  };
93
93
  return { metadata, tool, handler };
94
94
  }
95
+ const remoteStainlessHandler = async ({ reqContext, args, }) => {
96
+ const code = args.code;
97
+ const intent = args.intent;
98
+ const client = reqContext.client;
99
+ const codeModeEndpoint = readEnv('CODE_MODE_ENDPOINT_URL') ?? 'https://api.stainless.com/api/ai/code-tool';
100
+ const localClientEnvs = {
101
+ IMAGEKIT_PRIVATE_KEY: requireValue(readEnv('IMAGEKIT_PRIVATE_KEY') ?? client.privateKey, 'set IMAGEKIT_PRIVATE_KEY environment variable or provide privateKey client option'),
102
+ OPTIONAL_IMAGEKIT_IGNORES_THIS: readEnv('OPTIONAL_IMAGEKIT_IGNORES_THIS') ?? client.password ?? undefined,
103
+ IMAGEKIT_WEBHOOK_SECRET: readEnv('IMAGEKIT_WEBHOOK_SECRET') ?? client.webhookSecret ?? undefined,
104
+ IMAGE_KIT_BASE_URL: readEnv('IMAGE_KIT_BASE_URL') ?? client.baseURL ?? undefined,
105
+ };
106
+ // Merge any upstream client envs from the request header, with upstream values taking precedence.
107
+ const mergedClientEnvs = { ...localClientEnvs, ...reqContext.upstreamClientEnvs };
108
+ // Setting a Stainless API key authenticates requests to the code tool endpoint.
109
+ const res = await fetch(codeModeEndpoint, {
110
+ method: 'POST',
111
+ headers: {
112
+ ...(reqContext.stainlessApiKey && { Authorization: reqContext.stainlessApiKey }),
113
+ 'Content-Type': 'application/json',
114
+ 'x-stainless-mcp-client-envs': JSON.stringify(mergedClientEnvs),
115
+ },
116
+ body: JSON.stringify({
117
+ project_name: 'imagekit',
118
+ code,
119
+ intent,
120
+ client_opts: {},
121
+ }),
122
+ });
123
+ if (!res.ok) {
124
+ if (res.status === 404 && !reqContext.stainlessApiKey) {
125
+ throw new Error('Could not access code tool for this project. You may need to provide a Stainless API key via the STAINLESS_API_KEY environment variable, the --stainless-api-key flag, or the x-stainless-api-key HTTP header.');
126
+ }
127
+ throw new Error(`${res.status}: ${res.statusText} error when trying to contact Code Tool server. Details: ${await res.text()}`);
128
+ }
129
+ const { is_error, result, log_lines, err_lines } = (await res.json());
130
+ const hasLogs = log_lines.length > 0 || err_lines.length > 0;
131
+ const output = {
132
+ result,
133
+ ...(log_lines.length > 0 && { log_lines }),
134
+ ...(err_lines.length > 0 && { err_lines }),
135
+ };
136
+ if (is_error) {
137
+ return asErrorResult(typeof result === 'string' && !hasLogs ? result : JSON.stringify(output, null, 2));
138
+ }
139
+ return asTextContentResult(output);
140
+ };
141
+ const localDenoHandler = async ({ reqContext, args, }) => {
142
+ const fs = await import('node:fs');
143
+ const path = await import('node:path');
144
+ const url = await import('node:url');
145
+ const { newDenoHTTPWorker } = await import('@valtown/deno-http-worker');
146
+ const { getWorkerPath } = await import('./code-tool-paths.cjs');
147
+ const workerPath = getWorkerPath();
148
+ const client = reqContext.client;
149
+ const baseURLHostname = new URL(client.baseURL).hostname;
150
+ const { code } = args;
151
+ let denoPath;
152
+ const packageRoot = path.resolve(path.dirname(workerPath), '..');
153
+ const packageNodeModulesPath = path.resolve(packageRoot, 'node_modules');
154
+ // Check if deno is in PATH
155
+ const { execSync } = await import('node:child_process');
156
+ try {
157
+ execSync('command -v deno', { stdio: 'ignore' });
158
+ denoPath = 'deno';
159
+ }
160
+ catch {
161
+ try {
162
+ // Use deno binary in node_modules if it's found
163
+ const denoNodeModulesPath = path.resolve(packageNodeModulesPath, 'deno', 'bin.cjs');
164
+ await fs.promises.access(denoNodeModulesPath, fs.constants.X_OK);
165
+ denoPath = denoNodeModulesPath;
166
+ }
167
+ catch {
168
+ return asErrorResult('Deno is required for code execution but was not found. ' +
169
+ 'Install it from https://deno.land or run: npm install deno');
170
+ }
171
+ }
172
+ const allowReadPaths = [
173
+ 'code-tool-worker.mjs',
174
+ `${workerPath.replace(/([\/\\]node_modules)[\/\\].+$/, '$1')}/`,
175
+ packageRoot,
176
+ ];
177
+ // Follow symlinks in node_modules to allow read access to workspace-linked packages
178
+ try {
179
+ const sdkPkgName = '@imagekit/nodejs';
180
+ const sdkDir = path.resolve(packageNodeModulesPath, sdkPkgName);
181
+ const realSdkDir = fs.realpathSync(sdkDir);
182
+ if (realSdkDir !== sdkDir) {
183
+ allowReadPaths.push(realSdkDir);
184
+ }
185
+ }
186
+ catch {
187
+ // Ignore if symlink resolution fails
188
+ }
189
+ const allowRead = allowReadPaths.join(',');
190
+ const worker = await newDenoHTTPWorker(url.pathToFileURL(workerPath), {
191
+ denoExecutable: denoPath,
192
+ runFlags: [
193
+ `--node-modules-dir=manual`,
194
+ `--allow-read=${allowRead}`,
195
+ `--allow-net=${baseURLHostname}`,
196
+ // Allow environment variables because instantiating the client will try to read from them,
197
+ // even though they are not set.
198
+ '--allow-env',
199
+ ],
200
+ printOutput: true,
201
+ spawnOptions: {
202
+ cwd: path.dirname(workerPath),
203
+ // Merge any upstream client envs into the Deno subprocess environment,
204
+ // with the upstream env vars taking precedence.
205
+ env: { ...process.env, ...reqContext.upstreamClientEnvs },
206
+ },
207
+ });
208
+ try {
209
+ const resp = await new Promise((resolve, reject) => {
210
+ worker.addEventListener('exit', (exitCode) => {
211
+ reject(new Error(`Worker exited with code ${exitCode}`));
212
+ });
213
+ // Strip null/undefined values so that the worker SDK client can fall back to
214
+ // reading from environment variables (including any upstreamClientEnvs).
215
+ const opts = {
216
+ ...(client.baseURL != null ? { baseURL: client.baseURL } : undefined),
217
+ ...(client.privateKey != null ? { privateKey: client.privateKey } : undefined),
218
+ ...(client.password != null ? { password: client.password } : undefined),
219
+ ...(client.webhookSecret != null ? { webhookSecret: client.webhookSecret } : undefined),
220
+ defaultHeaders: {
221
+ 'X-Stainless-MCP': 'true',
222
+ },
223
+ };
224
+ const req = worker.request('http://localhost', {
225
+ headers: {
226
+ 'content-type': 'application/json',
227
+ },
228
+ method: 'POST',
229
+ }, (resp) => {
230
+ const body = [];
231
+ resp.on('error', (err) => {
232
+ reject(err);
233
+ });
234
+ resp.on('data', (chunk) => {
235
+ body.push(chunk);
236
+ });
237
+ resp.on('end', () => {
238
+ resolve(new Response(Buffer.concat(body).toString(), {
239
+ status: resp.statusCode ?? 200,
240
+ headers: resp.headers,
241
+ }));
242
+ });
243
+ });
244
+ const body = JSON.stringify({
245
+ opts,
246
+ code,
247
+ });
248
+ req.write(body, (err) => {
249
+ if (err != null) {
250
+ reject(err);
251
+ }
252
+ });
253
+ req.end();
254
+ });
255
+ if (resp.status === 200) {
256
+ const { result, log_lines, err_lines } = (await resp.json());
257
+ const returnOutput = result == null ? null : ({
258
+ type: 'text',
259
+ text: typeof result === 'string' ? result : JSON.stringify(result),
260
+ });
261
+ const logOutput = log_lines.length === 0 ?
262
+ null
263
+ : {
264
+ type: 'text',
265
+ text: log_lines.join('\n'),
266
+ };
267
+ const errOutput = err_lines.length === 0 ?
268
+ null
269
+ : {
270
+ type: 'text',
271
+ text: 'Error output:\n' + err_lines.join('\n'),
272
+ };
273
+ return {
274
+ content: [returnOutput, logOutput, errOutput].filter((block) => block !== null),
275
+ };
276
+ }
277
+ else {
278
+ const { result, log_lines, err_lines } = (await resp.json());
279
+ const messageOutput = result == null ? null : ({
280
+ type: 'text',
281
+ text: typeof result === 'string' ? result : JSON.stringify(result),
282
+ });
283
+ const logOutput = log_lines.length === 0 ?
284
+ null
285
+ : {
286
+ type: 'text',
287
+ text: log_lines.join('\n'),
288
+ };
289
+ const errOutput = err_lines.length === 0 ?
290
+ null
291
+ : {
292
+ type: 'text',
293
+ text: 'Error output:\n' + err_lines.join('\n'),
294
+ };
295
+ return {
296
+ content: [messageOutput, logOutput, errOutput].filter((block) => block !== null),
297
+ isError: true,
298
+ };
299
+ }
300
+ }
301
+ finally {
302
+ worker.terminate();
303
+ }
304
+ };
95
305
  //# sourceMappingURL=code-tool.mjs.map
package/code-tool.mjs.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"file":"code-tool.mjs","sourceRoot":"","sources":["src/code-tool.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAqC,aAAa,EAAE,mBAAmB,EAAE;OAEzE,EAAE,OAAO,EAAE,YAAY,EAAE;AAIhC,MAAM,MAAM,GAAG;;;;;;;;;;;;;;;;;;uGAkBwF,CAAC;AAExG;;;;;;;;GAQG;AACH,MAAM,UAAU,QAAQ;IACtB,MAAM,QAAQ,GAAa,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;IAC7E,MAAM,IAAI,GAAS;QACjB,IAAI,EAAE,SAAS;QACf,WAAW,EAAE,MAAM;QACnB,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,IAAI,EAAE;oBACJ,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,kBAAkB;iBAChC;gBACD,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,iEAAiE;iBAC/E;aACF;YACD,QAAQ,EAAE,CAAC,MAAM,CAAC;SACnB;KACF,CAAC;IACF,MAAM,OAAO,GAAG,KAAK,EAAE,MAAgB,EAAE,IAAS,EAA2B,EAAE;QAC7E,MAAM,IAAI,GAAG,IAAI,CAAC,IAAc,CAAC;QACjC,MAAM,MAAM,GAAG,IAAI,CAAC,MAA4B,CAAC;QAEjD,sFAAsF;QACtF,sFAAsF;QACtF,MAAM,eAAe,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAAC;QACrD,MAAM,gBAAgB,GACpB,OAAO,CAAC,wBAAwB,CAAC,IAAI,4CAA4C,CAAC;QAEpF,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,gBAAgB,EAAE;YACxC,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,GAAG,CAAC,eAAe,IAAI,EAAE,aAAa,EAAE,eAAe,EAAE,CAAC;gBAC1D,cAAc,EAAE,kBAAkB;gBAClC,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC;oBAC1B,oBAAoB,EAAE,YAAY,CAChC,OAAO,CAAC,sBAAsB,CAAC,IAAI,MAAM,CAAC,UAAU,EACpD,mFAAmF,CACpF;oBACD,8BAA8B,EAC5B,OAAO,CAAC,gCAAgC,CAAC,IAAI,MAAM,CAAC,QAAQ,IAAI,SAAS;oBAC3E,uBAAuB,EAAE,OAAO,CAAC,yBAAyB,CAAC,IAAI,MAAM,CAAC,aAAa,IAAI,SAAS;oBAChG,kBAAkB,EAAE,OAAO,CAAC,oBAAoB,CAAC,IAAI,MAAM,CAAC,OAAO,IAAI,SAAS;iBACjF,CAAC;aACH;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,YAAY,EAAE,UAAU;gBACxB,IAAI;gBACJ,MAAM;gBACN,WAAW,EAAE,EAAE;aACM,CAAC;SACzB,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CACb,GAAG,GAAG,CAAC,MAAM,KACX,GAAG,CAAC,UACN,4DAA4D,MAAM,GAAG,CAAC,IAAI,EAAE,EAAE,CAC/E,CAAC;QACJ,CAAC;QAED,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAiB,CAAC;QACtF,MAAM,OAAO,GAAG,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC;QAC7D,MAAM,MAAM,GAAG;YACb,MAAM;YACN,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;YAC1C,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;SAC3C,CAAC;QACF,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,aAAa,CAAC,OAAO,MAAM,KAAK,QAAQ,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1G,CAAC;QACD,OAAO,mBAAmB,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC,CAAC;IAEF,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;AACrC,CAAC"}
1
+ {"version":3,"file":"code-tool.mjs","sourceRoot":"","sources":["src/code-tool.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAML,aAAa,EACb,mBAAmB,GACpB;OAEM,EAAE,OAAO,EAAE,YAAY,EAAE;OAEzB,EAAE,SAAS,EAAE;AAKpB,MAAM,MAAM,GAAG;;;;;;;;;;;;;;;;;;;;gGAoBiF,CAAC;AAEjG;;;;;;;;;;;;GAYG;AACH,MAAM,UAAU,QAAQ,CAAC,EACvB,cAAc,EACd,iBAAiB,GAIlB;IACC,MAAM,QAAQ,GAAa,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;IAC7E,MAAM,IAAI,GAAS;QACjB,IAAI,EAAE,SAAS;QACf,WAAW,EAAE,MAAM;QACnB,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,IAAI,EAAE;oBACJ,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,kBAAkB;iBAChC;gBACD,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,iEAAiE;iBAC/E;aACF;YACD,QAAQ,EAAE,CAAC,MAAM,CAAC;SACnB;KACF,CAAC;IAEF,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAE3B,MAAM,OAAO,GAAG,KAAK,EAAE,EACrB,UAAU,EACV,IAAI,GAIL,EAA2B,EAAE;QAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,IAAc,CAAC;QACjC,uEAAuE;QACvE,EAAE;QACF,gFAAgF;QAChF,yFAAyF;QACzF,0FAA0F;QAC1F,IAAI,cAAc,EAAE,CAAC;YACnB,MAAM,cAAc,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,CAAC;YACnG,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC9B,OAAO,aAAa,CAClB,mGAAmG,cAAc;qBAC9G,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,kBAAkB,CAAC;qBAChC,IAAI,CAAC,IAAI,CAAC,EAAE,CAChB,CAAC;YACJ,CAAC;QACH,CAAC;QAED,IAAI,MAAsB,CAAC;QAC3B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,iBAAiB,KAAK,OAAO,EAAE,CAAC;YAClC,MAAM,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;YACzD,MAAM,GAAG,MAAM,gBAAgB,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;QACxD,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,KAAK,CAAC,gDAAgD,CAAC,CAAC;YAC/D,MAAM,GAAG,MAAM,sBAAsB,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9D,CAAC;QAED,MAAM,CAAC,IAAI,CACT;YACE,iBAAiB;YACjB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;YAClC,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,WAAW,EAAE,MAAM,CAAC,OAAO,EAAE,MAAM,IAAI,CAAC;SACzC,EACD,gCAAgC,CACjC,CAAC;QACF,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;IAEF,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;AACrC,CAAC;AAED,MAAM,sBAAsB,GAAG,KAAK,EAAE,EACpC,UAAU,EACV,IAAI,GAIL,EAA2B,EAAE;IAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,IAAc,CAAC;IACjC,MAAM,MAAM,GAAG,IAAI,CAAC,MAA4B,CAAC;IACjD,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;IAEjC,MAAM,gBAAgB,GAAG,OAAO,CAAC,wBAAwB,CAAC,IAAI,4CAA4C,CAAC;IAE3G,MAAM,eAAe,GAAG;QACtB,oBAAoB,EAAE,YAAY,CAChC,OAAO,CAAC,sBAAsB,CAAC,IAAI,MAAM,CAAC,UAAU,EACpD,mFAAmF,CACpF;QACD,8BAA8B,EAAE,OAAO,CAAC,gCAAgC,CAAC,IAAI,MAAM,CAAC,QAAQ,IAAI,SAAS;QACzG,uBAAuB,EAAE,OAAO,CAAC,yBAAyB,CAAC,IAAI,MAAM,CAAC,aAAa,IAAI,SAAS;QAChG,kBAAkB,EAAE,OAAO,CAAC,oBAAoB,CAAC,IAAI,MAAM,CAAC,OAAO,IAAI,SAAS;KACjF,CAAC;IACF,kGAAkG;IAClG,MAAM,gBAAgB,GAAG,EAAE,GAAG,eAAe,EAAE,GAAG,UAAU,CAAC,kBAAkB,EAAE,CAAC;IAElF,gFAAgF;IAChF,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,gBAAgB,EAAE;QACxC,MAAM,EAAE,MAAM;QACd,OAAO,EAAE;YACP,GAAG,CAAC,UAAU,CAAC,eAAe,IAAI,EAAE,aAAa,EAAE,UAAU,CAAC,eAAe,EAAE,CAAC;YAChF,cAAc,EAAE,kBAAkB;YAClC,6BAA6B,EAAE,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC;SAChE;QACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;YACnB,YAAY,EAAE,UAAU;YACxB,IAAI;YACJ,MAAM;YACN,WAAW,EAAE,EAAE;SACM,CAAC;KACzB,CAAC,CAAC;IAEH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;QACZ,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,eAAe,EAAE,CAAC;YACtD,MAAM,IAAI,KAAK,CACb,gNAAgN,CACjN,CAAC;QACJ,CAAC;QACD,MAAM,IAAI,KAAK,CACb,GAAG,GAAG,CAAC,MAAM,KACX,GAAG,CAAC,UACN,4DAA4D,MAAM,GAAG,CAAC,IAAI,EAAE,EAAE,CAC/E,CAAC;IACJ,CAAC;IAED,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAiB,CAAC;IACtF,MAAM,OAAO,GAAG,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC;IAC7D,MAAM,MAAM,GAAG;QACb,MAAM;QACN,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;QAC1C,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;KAC3C,CAAC;IACF,IAAI,QAAQ,EAAE,CAAC;QACb,OAAO,aAAa,CAAC,OAAO,MAAM,KAAK,QAAQ,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IAC1G,CAAC;IACD,OAAO,mBAAmB,CAAC,MAAM,CAAC,CAAC;AACrC,CAAC,CAAC;AAEF,MAAM,gBAAgB,GAAG,KAAK,EAAE,EAC9B,UAAU,EACV,IAAI,GAIL,EAA2B,EAAE;IAC5B,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC;IACnC,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC;IACvC,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,CAAC;IACrC,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAC;IACxE,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,uBAAuB,CAAC,CAAC;IAChE,MAAM,UAAU,GAAG,aAAa,EAAE,CAAC;IAEnC,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;IACjC,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC;IACzD,MAAM,EAAE,IAAI,EAAE,GAAG,IAAwB,CAAC;IAE1C,IAAI,QAAgB,CAAC;IAErB,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,CAAC;IACjE,MAAM,sBAAsB,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC;IAEzE,2BAA2B;IAC3B,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,MAAM,CAAC,oBAAoB,CAAC,CAAC;IACxD,IAAI,CAAC;QACH,QAAQ,CAAC,iBAAiB,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;QACjD,QAAQ,GAAG,MAAM,CAAC;IACpB,CAAC;IAAC,MAAM,CAAC;QACP,IAAI,CAAC;YACH,gDAAgD;YAChD,MAAM,mBAAmB,GAAG,IAAI,CAAC,OAAO,CAAC,sBAAsB,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YACpF,MAAM,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,mBAAmB,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YACjE,QAAQ,GAAG,mBAAmB,CAAC;QACjC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,aAAa,CAClB,yDAAyD;gBACvD,4DAA4D,CAC/D,CAAC;QACJ,CAAC;IACH,CAAC;IAED,MAAM,cAAc,GAAG;QACrB,sBAAsB;QACtB,GAAG,UAAU,CAAC,OAAO,CAAC,+BAA+B,EAAE,IAAI,CAAC,GAAG;QAC/D,WAAW;KACZ,CAAC;IAEF,oFAAoF;IACpF,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,kBAAkB,CAAC;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,sBAAsB,EAAE,UAAU,CAAC,CAAC;QAChE,MAAM,UAAU,GAAG,EAAE,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QAC3C,IAAI,UAAU,KAAK,MAAM,EAAE,CAAC;YAC1B,cAAc,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,qCAAqC;IACvC,CAAC;IAED,MAAM,SAAS,GAAG,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAE3C,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,GAAG,CAAC,aAAa,CAAC,UAAU,CAAC,EAAE;QACpE,cAAc,EAAE,QAAQ;QACxB,QAAQ,EAAE;YACR,2BAA2B;YAC3B,gBAAgB,SAAS,EAAE;YAC3B,eAAe,eAAe,EAAE;YAChC,2FAA2F;YAC3F,gCAAgC;YAChC,aAAa;SACd;QACD,WAAW,EAAE,IAAI;QACjB,YAAY,EAAE;YACZ,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;YAC7B,uEAAuE;YACvE,gDAAgD;YAChD,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,GAAG,UAAU,CAAC,kBAAkB,EAAE;SAC1D;KACF,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,IAAI,OAAO,CAAW,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC3D,MAAM,CAAC,gBAAgB,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,EAAE;gBAC3C,MAAM,CAAC,IAAI,KAAK,CAAC,2BAA2B,QAAQ,EAAE,CAAC,CAAC,CAAC;YAC3D,CAAC,CAAC,CAAC;YAEH,6EAA6E;YAC7E,yEAAyE;YACzE,MAAM,IAAI,GAAG;gBACX,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;gBACrE,GAAG,CAAC,MAAM,CAAC,UAAU,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;gBAC9E,GAAG,CAAC,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;gBACxE,GAAG,CAAC,MAAM,CAAC,aAAa,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;gBACvF,cAAc,EAAE;oBACd,iBAAiB,EAAE,MAAM;iBAC1B;aACgD,CAAC;YAEpD,MAAM,GAAG,GAAG,MAAM,CAAC,OAAO,CACxB,kBAAkB,EAClB;gBACE,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;iBACnC;gBACD,MAAM,EAAE,MAAM;aACf,EACD,CAAC,IAAI,EAAE,EAAE;gBACP,MAAM,IAAI,GAAiB,EAAE,CAAC;gBAC9B,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;oBACvB,MAAM,CAAC,GAAG,CAAC,CAAC;gBACd,CAAC,CAAC,CAAC;gBACH,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;oBACxB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACnB,CAAC,CAAC,CAAC;gBACH,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;oBAClB,OAAO,CACL,IAAI,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE;wBAC3C,MAAM,EAAE,IAAI,CAAC,UAAU,IAAI,GAAG;wBAC9B,OAAO,EAAE,IAAI,CAAC,OAAc;qBAC7B,CAAC,CACH,CAAC;gBACJ,CAAC,CAAC,CAAC;YACL,CAAC,CACF,CAAC;YAEF,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC;gBAC1B,IAAI;gBACJ,IAAI;aACL,CAAC,CAAC;YAEH,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,GAAG,EAAE,EAAE;gBACtB,IAAI,GAAG,IAAI,IAAI,EAAE,CAAC;oBAChB,MAAM,CAAC,GAAG,CAAC,CAAC;gBACd,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,GAAG,CAAC,GAAG,EAAE,CAAC;QACZ,CAAC,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YACxB,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,CAAiB,CAAC;YAC7E,MAAM,YAAY,GAChB,MAAM,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CACtB;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aACnE,CACF,CAAC;YACJ,MAAM,SAAS,GACb,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC;gBACtB,IAAI;gBACN,CAAC,CAAC;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;iBAC3B,CAAC;YACN,MAAM,SAAS,GACb,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC;gBACtB,IAAI;gBACN,CAAC,CAAC;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,iBAAiB,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;iBAC/C,CAAC;YACN,OAAO;gBACL,OAAO,EAAE,CAAC,YAAY,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC;aAChF,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,CAAiB,CAAC;YAC7E,MAAM,aAAa,GACjB,MAAM,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CACtB;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aACnE,CACF,CAAC;YACJ,MAAM,SAAS,GACb,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC;gBACtB,IAAI;gBACN,CAAC,CAAC;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;iBAC3B,CAAC;YACN,MAAM,SAAS,GACb,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC;gBACtB,IAAI;gBACN,CAAC,CAAC;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,iBAAiB,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;iBAC/C,CAAC;YACN,OAAO;gBACL,OAAO,EAAE,CAAC,aAAa,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC;gBAChF,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;IACH,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,SAAS,EAAE,CAAC;IACrB,CAAC;AACH,CAAC,CAAC"}
@@ -1,8 +1,13 @@
1
- import { Metadata } from "./types.mjs";
2
1
  import { Tool } from '@modelcontextprotocol/sdk/types.js';
2
+ import { Metadata, McpRequestContext } from "./types.mjs";
3
+ import type { LocalDocsSearch } from "./local-docs-search.mjs";
3
4
  export declare const metadata: Metadata;
4
5
  export declare const tool: Tool;
5
- export declare const handler: (_: unknown, args: Record<string, unknown> | undefined) => Promise<import("./types").ToolCallResult>;
6
+ export declare function setLocalSearch(search: LocalDocsSearch): void;
7
+ export declare const handler: ({ reqContext, args, }: {
8
+ reqContext: McpRequestContext;
9
+ args: Record<string, unknown> | undefined;
10
+ }) => Promise<import("./types").ToolCallResult>;
6
11
  declare const _default: {
7
12
  metadata: Metadata;
8
13
  tool: {
@@ -45,7 +50,10 @@ declare const _default: {
45
50
  }[] | undefined;
46
51
  title?: string | undefined;
47
52
  };
48
- handler: (_: unknown, args: Record<string, unknown> | undefined) => Promise<import("./types").ToolCallResult>;
53
+ handler: ({ reqContext, args, }: {
54
+ reqContext: McpRequestContext;
55
+ args: Record<string, unknown> | undefined;
56
+ }) => Promise<import("./types").ToolCallResult>;
49
57
  };
50
58
  export default _default;
51
59
  //# sourceMappingURL=docs-search-tool.d.mts.map
@@ -1 +1 @@
1
- {"version":3,"file":"docs-search-tool.d.mts","sourceRoot":"","sources":["src/docs-search-tool.ts"],"names":[],"mappings":"OAEO,EAAE,QAAQ,EAAuB;OAEjC,EAAE,IAAI,EAAE,MAAM,oCAAoC;AAEzD,eAAO,MAAM,QAAQ,EAAE,QAKtB,CAAC;AAEF,eAAO,MAAM,IAAI,EAAE,IA0BlB,CAAC;AAKF,eAAO,MAAM,OAAO,GAAU,GAAG,OAAO,EAAE,MAAM,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,8CAYlF,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAZ+B,OAAO,QAAQ,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS;;AAcnF,wBAA2C"}
1
+ {"version":3,"file":"docs-search-tool.d.mts","sourceRoot":"","sources":["src/docs-search-tool.ts"],"names":[],"mappings":"OAEO,EAAE,IAAI,EAAE,MAAM,oCAAoC;OAClD,EAAE,QAAQ,EAAE,iBAAiB,EAAuB;OAEpD,KAAK,EAAE,eAAe,EAAE;AAE/B,eAAO,MAAM,QAAQ,EAAE,QAKtB,CAAC;AAEF,eAAO,MAAM,IAAI,EAAE,IA2BlB,CAAC;AAOF,wBAAgB,cAAc,CAAC,MAAM,EAAE,eAAe,GAAG,IAAI,CAE5D;AAuED,eAAO,MAAM,OAAO,GAAU,uBAG3B;IACD,UAAU,EAAE,iBAAiB,CAAC;IAC9B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC;CAC3C,8CAQA,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;qCAXC;QACD,UAAU,EAAE,iBAAiB,CAAC;QAC9B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC;KAC3C;;AAUD,wBAA2C"}
@@ -1,8 +1,13 @@
1
- import { Metadata } from "./types.js";
2
1
  import { Tool } from '@modelcontextprotocol/sdk/types.js';
2
+ import { Metadata, McpRequestContext } from "./types.js";
3
+ import type { LocalDocsSearch } from "./local-docs-search.js";
3
4
  export declare const metadata: Metadata;
4
5
  export declare const tool: Tool;
5
- export declare const handler: (_: unknown, args: Record<string, unknown> | undefined) => Promise<import("./types").ToolCallResult>;
6
+ export declare function setLocalSearch(search: LocalDocsSearch): void;
7
+ export declare const handler: ({ reqContext, args, }: {
8
+ reqContext: McpRequestContext;
9
+ args: Record<string, unknown> | undefined;
10
+ }) => Promise<import("./types").ToolCallResult>;
6
11
  declare const _default: {
7
12
  metadata: Metadata;
8
13
  tool: {
@@ -45,7 +50,10 @@ declare const _default: {
45
50
  }[] | undefined;
46
51
  title?: string | undefined;
47
52
  };
48
- handler: (_: unknown, args: Record<string, unknown> | undefined) => Promise<import("./types").ToolCallResult>;
53
+ handler: ({ reqContext, args, }: {
54
+ reqContext: McpRequestContext;
55
+ args: Record<string, unknown> | undefined;
56
+ }) => Promise<import("./types").ToolCallResult>;
49
57
  };
50
58
  export default _default;
51
59
  //# sourceMappingURL=docs-search-tool.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"docs-search-tool.d.ts","sourceRoot":"","sources":["src/docs-search-tool.ts"],"names":[],"mappings":"OAEO,EAAE,QAAQ,EAAuB;OAEjC,EAAE,IAAI,EAAE,MAAM,oCAAoC;AAEzD,eAAO,MAAM,QAAQ,EAAE,QAKtB,CAAC;AAEF,eAAO,MAAM,IAAI,EAAE,IA0BlB,CAAC;AAKF,eAAO,MAAM,OAAO,GAAU,GAAG,OAAO,EAAE,MAAM,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,8CAYlF,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAZ+B,OAAO,QAAQ,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS;;AAcnF,wBAA2C"}
1
+ {"version":3,"file":"docs-search-tool.d.ts","sourceRoot":"","sources":["src/docs-search-tool.ts"],"names":[],"mappings":"OAEO,EAAE,IAAI,EAAE,MAAM,oCAAoC;OAClD,EAAE,QAAQ,EAAE,iBAAiB,EAAuB;OAEpD,KAAK,EAAE,eAAe,EAAE;AAE/B,eAAO,MAAM,QAAQ,EAAE,QAKtB,CAAC;AAEF,eAAO,MAAM,IAAI,EAAE,IA2BlB,CAAC;AAOF,wBAAgB,cAAc,CAAC,MAAM,EAAE,eAAe,GAAG,IAAI,CAE5D;AAuED,eAAO,MAAM,OAAO,GAAU,uBAG3B;IACD,UAAU,EAAE,iBAAiB,CAAC;IAC9B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC;CAC3C,8CAQA,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;qCAXC;QACD,UAAU,EAAE,iBAAiB,CAAC;QAC9B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC;KAC3C;;AAUD,wBAA2C"}
@@ -2,7 +2,9 @@
2
2
  // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
3
3
  Object.defineProperty(exports, "__esModule", { value: true });
4
4
  exports.handler = exports.tool = exports.metadata = void 0;
5
+ exports.setLocalSearch = setLocalSearch;
5
6
  const types_1 = require("./types.js");
7
+ const logger_1 = require("./logger.js");
6
8
  exports.metadata = {
7
9
  resource: 'all',
8
10
  operation: 'read',
@@ -11,7 +13,7 @@ exports.metadata = {
11
13
  };
12
14
  exports.tool = {
13
15
  name: 'search_docs',
14
- description: 'Search for documentation for how to use the client to interact with the API.',
16
+ description: 'Search SDK documentation to find methods, parameters, and usage examples for interacting with the API. Use this before writing code when you need to discover the right approach.',
15
17
  inputSchema: {
16
18
  type: 'object',
17
19
  properties: {
@@ -37,14 +39,65 @@ exports.tool = {
37
39
  },
38
40
  };
39
41
  const docsSearchURL = process.env['DOCS_SEARCH_URL'] || 'https://api.stainless.com/api/projects/imagekit/docs/search';
40
- const handler = async (_, args) => {
42
+ let _localSearch;
43
+ function setLocalSearch(search) {
44
+ _localSearch = search;
45
+ }
46
+ async function searchLocal(args) {
47
+ if (!_localSearch) {
48
+ throw new Error('Local search not initialized');
49
+ }
50
+ const query = args['query'] ?? '';
51
+ const language = args['language'] ?? 'typescript';
52
+ const detail = args['detail'] ?? 'default';
53
+ return _localSearch.search({
54
+ query,
55
+ language,
56
+ detail,
57
+ maxResults: 5,
58
+ }).results;
59
+ }
60
+ async function searchRemote(args, reqContext) {
41
61
  const body = args;
42
62
  const query = new URLSearchParams(body).toString();
43
- const result = await fetch(`${docsSearchURL}?${query}`);
63
+ const startTime = Date.now();
64
+ const result = await fetch(`${docsSearchURL}?${query}`, {
65
+ headers: {
66
+ ...(reqContext.stainlessApiKey && { Authorization: reqContext.stainlessApiKey }),
67
+ ...(reqContext.mcpSessionId && { 'x-stainless-mcp-session-id': reqContext.mcpSessionId }),
68
+ ...(reqContext.mcpClientInfo && {
69
+ 'x-stainless-mcp-client-info': JSON.stringify(reqContext.mcpClientInfo),
70
+ }),
71
+ },
72
+ });
73
+ const logger = (0, logger_1.getLogger)();
44
74
  if (!result.ok) {
45
- throw new Error(`${result.status}: ${result.statusText} when using doc search tool. Details: ${await result.text()}`);
75
+ const errorText = await result.text();
76
+ logger.warn({
77
+ durationMs: Date.now() - startTime,
78
+ query: body.query,
79
+ status: result.status,
80
+ statusText: result.statusText,
81
+ errorText,
82
+ }, 'Got error response from docs search tool');
83
+ if (result.status === 404 && !reqContext.stainlessApiKey) {
84
+ throw new Error('Could not find docs for this project. You may need to provide a Stainless API key via the STAINLESS_API_KEY environment variable, the --stainless-api-key flag, or the x-stainless-api-key HTTP header.');
85
+ }
86
+ throw new Error(`${result.status}: ${result.statusText} when using doc search tool. Details: ${errorText}`);
87
+ }
88
+ const resultBody = await result.json();
89
+ logger.info({
90
+ durationMs: Date.now() - startTime,
91
+ query: body.query,
92
+ }, 'Got docs search result');
93
+ return resultBody;
94
+ }
95
+ const handler = async ({ reqContext, args, }) => {
96
+ const body = args ?? {};
97
+ if (_localSearch) {
98
+ return (0, types_1.asTextContentResult)(await searchLocal(body));
46
99
  }
47
- return (0, types_1.asTextContentResult)(await result.json());
100
+ return (0, types_1.asTextContentResult)(await searchRemote(body, reqContext));
48
101
  };
49
102
  exports.handler = handler;
50
103
  exports.default = { metadata: exports.metadata, tool: exports.tool, handler: exports.handler };
@@ -1 +1 @@
1
- {"version":3,"file":"docs-search-tool.js","sourceRoot":"","sources":["src/docs-search-tool.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,sCAAwD;AAI3C,QAAA,QAAQ,GAAa;IAChC,QAAQ,EAAE,KAAK;IACf,SAAS,EAAE,MAAM;IACjB,IAAI,EAAE,EAAE;IACR,UAAU,EAAE,KAAK;CAClB,CAAC;AAEW,QAAA,IAAI,GAAS;IACxB,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE,8EAA8E;IAC3F,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,0BAA0B;aACxC;YACD,QAAQ,EAAE;gBACR,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,yCAAyC;gBACtD,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC;aAClG;YACD,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,iCAAiC;gBAC9C,IAAI,EAAE,CAAC,SAAS,EAAE,SAAS,CAAC;aAC7B;SACF;QACD,QAAQ,EAAE,CAAC,OAAO,EAAE,UAAU,CAAC;KAChC;IACD,WAAW,EAAE;QACX,YAAY,EAAE,IAAI;KACnB;CACF,CAAC;AAEF,MAAM,aAAa,GACjB,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,6DAA6D,CAAC;AAE3F,MAAM,OAAO,GAAG,KAAK,EAAE,CAAU,EAAE,IAAyC,EAAE,EAAE;IACrF,MAAM,IAAI,GAAG,IAAW,CAAC;IACzB,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;IACnD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,GAAG,aAAa,IAAI,KAAK,EAAE,CAAC,CAAC;IAExD,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CACb,GAAG,MAAM,CAAC,MAAM,KAAK,MAAM,CAAC,UAAU,yCAAyC,MAAM,MAAM,CAAC,IAAI,EAAE,EAAE,CACrG,CAAC;IACJ,CAAC;IAED,OAAO,IAAA,2BAAmB,EAAC,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;AAClD,CAAC,CAAC;AAZW,QAAA,OAAO,WAYlB;AAEF,kBAAe,EAAE,QAAQ,EAAR,gBAAQ,EAAE,IAAI,EAAJ,YAAI,EAAE,OAAO,EAAP,eAAO,EAAE,CAAC"}
1
+ {"version":3,"file":"docs-search-tool.js","sourceRoot":"","sources":["src/docs-search-tool.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAgDtF,wCAEC;AA/CD,sCAA2E;AAC3E,wCAAqC;AAGxB,QAAA,QAAQ,GAAa;IAChC,QAAQ,EAAE,KAAK;IACf,SAAS,EAAE,MAAM;IACjB,IAAI,EAAE,EAAE;IACR,UAAU,EAAE,KAAK;CAClB,CAAC;AAEW,QAAA,IAAI,GAAS;IACxB,IAAI,EAAE,aAAa;IACnB,WAAW,EACT,mLAAmL;IACrL,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,0BAA0B;aACxC;YACD,QAAQ,EAAE;gBACR,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,yCAAyC;gBACtD,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC;aAClG;YACD,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,iCAAiC;gBAC9C,IAAI,EAAE,CAAC,SAAS,EAAE,SAAS,CAAC;aAC7B;SACF;QACD,QAAQ,EAAE,CAAC,OAAO,EAAE,UAAU,CAAC;KAChC;IACD,WAAW,EAAE;QACX,YAAY,EAAE,IAAI;KACnB;CACF,CAAC;AAEF,MAAM,aAAa,GACjB,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,6DAA6D,CAAC;AAElG,IAAI,YAAyC,CAAC;AAE9C,SAAgB,cAAc,CAAC,MAAuB;IACpD,YAAY,GAAG,MAAM,CAAC;AACxB,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,IAA6B;IACtD,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,KAAK,GAAI,IAAI,CAAC,OAAO,CAAY,IAAI,EAAE,CAAC;IAC9C,MAAM,QAAQ,GAAI,IAAI,CAAC,UAAU,CAAY,IAAI,YAAY,CAAC;IAC9D,MAAM,MAAM,GAAI,IAAI,CAAC,QAAQ,CAAY,IAAI,SAAS,CAAC;IAEvD,OAAO,YAAY,CAAC,MAAM,CAAC;QACzB,KAAK;QACL,QAAQ;QACR,MAAM;QACN,UAAU,EAAE,CAAC;KACd,CAAC,CAAC,OAAO,CAAC;AACb,CAAC;AAED,KAAK,UAAU,YAAY,CAAC,IAA6B,EAAE,UAA6B;IACtF,MAAM,IAAI,GAAG,IAAW,CAAC;IACzB,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;IAEnD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,GAAG,aAAa,IAAI,KAAK,EAAE,EAAE;QACtD,OAAO,EAAE;YACP,GAAG,CAAC,UAAU,CAAC,eAAe,IAAI,EAAE,aAAa,EAAE,UAAU,CAAC,eAAe,EAAE,CAAC;YAChF,GAAG,CAAC,UAAU,CAAC,YAAY,IAAI,EAAE,4BAA4B,EAAE,UAAU,CAAC,YAAY,EAAE,CAAC;YACzF,GAAG,CAAC,UAAU,CAAC,aAAa,IAAI;gBAC9B,6BAA6B,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,aAAa,CAAC;aACxE,CAAC;SACH;KACF,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,IAAA,kBAAS,GAAE,CAAC;IAE3B,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;QACf,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QACtC,MAAM,CAAC,IAAI,CACT;YACE,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;YAClC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,SAAS;SACV,EACD,0CAA0C,CAC3C,CAAC;QAEF,IAAI,MAAM,CAAC,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,eAAe,EAAE,CAAC;YACzD,MAAM,IAAI,KAAK,CACb,yMAAyM,CAC1M,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,KAAK,CACb,GAAG,MAAM,CAAC,MAAM,KAAK,MAAM,CAAC,UAAU,yCAAyC,SAAS,EAAE,CAC3F,CAAC;IACJ,CAAC;IAED,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;IACvC,MAAM,CAAC,IAAI,CACT;QACE,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;QAClC,KAAK,EAAE,IAAI,CAAC,KAAK;KAClB,EACD,wBAAwB,CACzB,CAAC;IACF,OAAO,UAAU,CAAC;AACpB,CAAC;AAEM,MAAM,OAAO,GAAG,KAAK,EAAE,EAC5B,UAAU,EACV,IAAI,GAIL,EAAE,EAAE;IACH,MAAM,IAAI,GAAG,IAAI,IAAI,EAAE,CAAC;IAExB,IAAI,YAAY,EAAE,CAAC;QACjB,OAAO,IAAA,2BAAmB,EAAC,MAAM,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC;IACtD,CAAC;IAED,OAAO,IAAA,2BAAmB,EAAC,MAAM,YAAY,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC,CAAC;AACnE,CAAC,CAAC;AAdW,QAAA,OAAO,WAclB;AAEF,kBAAe,EAAE,QAAQ,EAAR,gBAAQ,EAAE,IAAI,EAAJ,YAAI,EAAE,OAAO,EAAP,eAAO,EAAE,CAAC"}
@@ -1,5 +1,6 @@
1
1
  // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
2
  import { asTextContentResult } from "./types.mjs";
3
+ import { getLogger } from "./logger.mjs";
3
4
  export const metadata = {
4
5
  resource: 'all',
5
6
  operation: 'read',
@@ -8,7 +9,7 @@ export const metadata = {
8
9
  };
9
10
  export const tool = {
10
11
  name: 'search_docs',
11
- description: 'Search for documentation for how to use the client to interact with the API.',
12
+ description: 'Search SDK documentation to find methods, parameters, and usage examples for interacting with the API. Use this before writing code when you need to discover the right approach.',
12
13
  inputSchema: {
13
14
  type: 'object',
14
15
  properties: {
@@ -34,14 +35,65 @@ export const tool = {
34
35
  },
35
36
  };
36
37
  const docsSearchURL = process.env['DOCS_SEARCH_URL'] || 'https://api.stainless.com/api/projects/imagekit/docs/search';
37
- export const handler = async (_, args) => {
38
+ let _localSearch;
39
+ export function setLocalSearch(search) {
40
+ _localSearch = search;
41
+ }
42
+ async function searchLocal(args) {
43
+ if (!_localSearch) {
44
+ throw new Error('Local search not initialized');
45
+ }
46
+ const query = args['query'] ?? '';
47
+ const language = args['language'] ?? 'typescript';
48
+ const detail = args['detail'] ?? 'default';
49
+ return _localSearch.search({
50
+ query,
51
+ language,
52
+ detail,
53
+ maxResults: 5,
54
+ }).results;
55
+ }
56
+ async function searchRemote(args, reqContext) {
38
57
  const body = args;
39
58
  const query = new URLSearchParams(body).toString();
40
- const result = await fetch(`${docsSearchURL}?${query}`);
59
+ const startTime = Date.now();
60
+ const result = await fetch(`${docsSearchURL}?${query}`, {
61
+ headers: {
62
+ ...(reqContext.stainlessApiKey && { Authorization: reqContext.stainlessApiKey }),
63
+ ...(reqContext.mcpSessionId && { 'x-stainless-mcp-session-id': reqContext.mcpSessionId }),
64
+ ...(reqContext.mcpClientInfo && {
65
+ 'x-stainless-mcp-client-info': JSON.stringify(reqContext.mcpClientInfo),
66
+ }),
67
+ },
68
+ });
69
+ const logger = getLogger();
41
70
  if (!result.ok) {
42
- throw new Error(`${result.status}: ${result.statusText} when using doc search tool. Details: ${await result.text()}`);
71
+ const errorText = await result.text();
72
+ logger.warn({
73
+ durationMs: Date.now() - startTime,
74
+ query: body.query,
75
+ status: result.status,
76
+ statusText: result.statusText,
77
+ errorText,
78
+ }, 'Got error response from docs search tool');
79
+ if (result.status === 404 && !reqContext.stainlessApiKey) {
80
+ throw new Error('Could not find docs for this project. You may need to provide a Stainless API key via the STAINLESS_API_KEY environment variable, the --stainless-api-key flag, or the x-stainless-api-key HTTP header.');
81
+ }
82
+ throw new Error(`${result.status}: ${result.statusText} when using doc search tool. Details: ${errorText}`);
83
+ }
84
+ const resultBody = await result.json();
85
+ logger.info({
86
+ durationMs: Date.now() - startTime,
87
+ query: body.query,
88
+ }, 'Got docs search result');
89
+ return resultBody;
90
+ }
91
+ export const handler = async ({ reqContext, args, }) => {
92
+ const body = args ?? {};
93
+ if (_localSearch) {
94
+ return asTextContentResult(await searchLocal(body));
43
95
  }
44
- return asTextContentResult(await result.json());
96
+ return asTextContentResult(await searchRemote(body, reqContext));
45
97
  };
46
98
  export default { metadata, tool, handler };
47
99
  //# sourceMappingURL=docs-search-tool.mjs.map
@@ -1 +1 @@
1
- {"version":3,"file":"docs-search-tool.mjs","sourceRoot":"","sources":["src/docs-search-tool.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAY,mBAAmB,EAAE;AAIxC,MAAM,CAAC,MAAM,QAAQ,GAAa;IAChC,QAAQ,EAAE,KAAK;IACf,SAAS,EAAE,MAAM;IACjB,IAAI,EAAE,EAAE;IACR,UAAU,EAAE,KAAK;CAClB,CAAC;AAEF,MAAM,CAAC,MAAM,IAAI,GAAS;IACxB,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE,8EAA8E;IAC3F,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,0BAA0B;aACxC;YACD,QAAQ,EAAE;gBACR,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,yCAAyC;gBACtD,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC;aAClG;YACD,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,iCAAiC;gBAC9C,IAAI,EAAE,CAAC,SAAS,EAAE,SAAS,CAAC;aAC7B;SACF;QACD,QAAQ,EAAE,CAAC,OAAO,EAAE,UAAU,CAAC;KAChC;IACD,WAAW,EAAE;QACX,YAAY,EAAE,IAAI;KACnB;CACF,CAAC;AAEF,MAAM,aAAa,GACjB,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,6DAA6D,CAAC;AAElG,MAAM,CAAC,MAAM,OAAO,GAAG,KAAK,EAAE,CAAU,EAAE,IAAyC,EAAE,EAAE;IACrF,MAAM,IAAI,GAAG,IAAW,CAAC;IACzB,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;IACnD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,GAAG,aAAa,IAAI,KAAK,EAAE,CAAC,CAAC;IAExD,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CACb,GAAG,MAAM,CAAC,MAAM,KAAK,MAAM,CAAC,UAAU,yCAAyC,MAAM,MAAM,CAAC,IAAI,EAAE,EAAE,CACrG,CAAC;IACJ,CAAC;IAED,OAAO,mBAAmB,CAAC,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;AAClD,CAAC,CAAC;AAEF,eAAe,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC"}
1
+ {"version":3,"file":"docs-search-tool.mjs","sourceRoot":"","sources":["src/docs-search-tool.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAG/E,EAA+B,mBAAmB,EAAE;OACpD,EAAE,SAAS,EAAE;AAGpB,MAAM,CAAC,MAAM,QAAQ,GAAa;IAChC,QAAQ,EAAE,KAAK;IACf,SAAS,EAAE,MAAM;IACjB,IAAI,EAAE,EAAE;IACR,UAAU,EAAE,KAAK;CAClB,CAAC;AAEF,MAAM,CAAC,MAAM,IAAI,GAAS;IACxB,IAAI,EAAE,aAAa;IACnB,WAAW,EACT,mLAAmL;IACrL,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE;YACV,KAAK,EAAE;gBACL,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,0BAA0B;aACxC;YACD,QAAQ,EAAE;gBACR,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,yCAAyC;gBACtD,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC;aAClG;YACD,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,iCAAiC;gBAC9C,IAAI,EAAE,CAAC,SAAS,EAAE,SAAS,CAAC;aAC7B;SACF;QACD,QAAQ,EAAE,CAAC,OAAO,EAAE,UAAU,CAAC;KAChC;IACD,WAAW,EAAE;QACX,YAAY,EAAE,IAAI;KACnB;CACF,CAAC;AAEF,MAAM,aAAa,GACjB,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,6DAA6D,CAAC;AAElG,IAAI,YAAyC,CAAC;AAE9C,MAAM,UAAU,cAAc,CAAC,MAAuB;IACpD,YAAY,GAAG,MAAM,CAAC;AACxB,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,IAA6B;IACtD,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,KAAK,GAAI,IAAI,CAAC,OAAO,CAAY,IAAI,EAAE,CAAC;IAC9C,MAAM,QAAQ,GAAI,IAAI,CAAC,UAAU,CAAY,IAAI,YAAY,CAAC;IAC9D,MAAM,MAAM,GAAI,IAAI,CAAC,QAAQ,CAAY,IAAI,SAAS,CAAC;IAEvD,OAAO,YAAY,CAAC,MAAM,CAAC;QACzB,KAAK;QACL,QAAQ;QACR,MAAM;QACN,UAAU,EAAE,CAAC;KACd,CAAC,CAAC,OAAO,CAAC;AACb,CAAC;AAED,KAAK,UAAU,YAAY,CAAC,IAA6B,EAAE,UAA6B;IACtF,MAAM,IAAI,GAAG,IAAW,CAAC;IACzB,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;IAEnD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,GAAG,aAAa,IAAI,KAAK,EAAE,EAAE;QACtD,OAAO,EAAE;YACP,GAAG,CAAC,UAAU,CAAC,eAAe,IAAI,EAAE,aAAa,EAAE,UAAU,CAAC,eAAe,EAAE,CAAC;YAChF,GAAG,CAAC,UAAU,CAAC,YAAY,IAAI,EAAE,4BAA4B,EAAE,UAAU,CAAC,YAAY,EAAE,CAAC;YACzF,GAAG,CAAC,UAAU,CAAC,aAAa,IAAI;gBAC9B,6BAA6B,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,aAAa,CAAC;aACxE,CAAC;SACH;KACF,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAE3B,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;QACf,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QACtC,MAAM,CAAC,IAAI,CACT;YACE,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;YAClC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,SAAS;SACV,EACD,0CAA0C,CAC3C,CAAC;QAEF,IAAI,MAAM,CAAC,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,eAAe,EAAE,CAAC;YACzD,MAAM,IAAI,KAAK,CACb,yMAAyM,CAC1M,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,KAAK,CACb,GAAG,MAAM,CAAC,MAAM,KAAK,MAAM,CAAC,UAAU,yCAAyC,SAAS,EAAE,CAC3F,CAAC;IACJ,CAAC;IAED,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;IACvC,MAAM,CAAC,IAAI,CACT;QACE,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;QAClC,KAAK,EAAE,IAAI,CAAC,KAAK;KAClB,EACD,wBAAwB,CACzB,CAAC;IACF,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,MAAM,CAAC,MAAM,OAAO,GAAG,KAAK,EAAE,EAC5B,UAAU,EACV,IAAI,GAIL,EAAE,EAAE;IACH,MAAM,IAAI,GAAG,IAAI,IAAI,EAAE,CAAC;IAExB,IAAI,YAAY,EAAE,CAAC;QACjB,OAAO,mBAAmB,CAAC,MAAM,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC;IACtD,CAAC;IAED,OAAO,mBAAmB,CAAC,MAAM,YAAY,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC,CAAC;AACnE,CAAC,CAAC;AAEF,eAAe,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC"}