@polka-codes/core 0.9.46 → 0.9.47

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/dist/index.js CHANGED
@@ -1,29 +1,232 @@
1
- // src/tools/askFollowupQuestion.ts
1
+ // src/Agent/parseJsonFromMarkdown.ts
2
+ var parseJsonFromMarkdown = (markdown) => {
3
+ const jsonRegex = /```(?:json)?\n([\s\S]*?)\n```/;
4
+ const match = markdown.match(jsonRegex);
5
+ const tryParse = (str) => {
6
+ try {
7
+ let parsed = JSON.parse(str);
8
+ if (typeof parsed === "string") {
9
+ try {
10
+ parsed = JSON.parse(parsed);
11
+ } catch {
12
+ }
13
+ }
14
+ return { success: true, data: parsed };
15
+ } catch (e) {
16
+ const error = e instanceof Error ? e.message : String(e);
17
+ return { success: false, error: `Failed to parse JSON: ${error}` };
18
+ }
19
+ };
20
+ if (match?.[1]) {
21
+ const content = match[1].trim();
22
+ return tryParse(content);
23
+ }
24
+ const parseResult = tryParse(markdown);
25
+ if (parseResult.success) {
26
+ return parseResult;
27
+ }
28
+ return { success: false, error: "No JSON object found in the string." };
29
+ };
30
+
31
+ // src/Agent/prompts.ts
32
+ var responsePrompts = {
33
+ errorInvokeTool: (tool, error) => `An error occurred while invoking the tool "${tool}": ${error}`,
34
+ requireUseTool: `Error: No tool use detected. You MUST use a tool before proceeding.
35
+ e.g. <tool_tool_name>tool_name</tool_tool_name>
36
+
37
+ Ensure the opening and closing tags are correctly nested and closed, and that you are using the correct tool name.
38
+ Avoid unnecessary text or symbols before or after the tool use.
39
+ Avoid unnecessary escape characters or special characters.
40
+ `,
41
+ requireUseToolNative: `Error: No tool use detected. You MUST use a tool before proceeding.
42
+ `,
43
+ toolResults: (tool, result) => {
44
+ switch (result.type) {
45
+ case "text":
46
+ return [
47
+ {
48
+ type: "text",
49
+ text: `<tool_response name=${tool}>${result.value}</tool_response>`
50
+ }
51
+ ];
52
+ case "error-text":
53
+ return [
54
+ {
55
+ type: "text",
56
+ text: `<tool_response_error name=${tool}>${result.value}</tool_response_error>`
57
+ }
58
+ ];
59
+ case "json":
60
+ return [
61
+ {
62
+ type: "text",
63
+ text: `<tool_response_json name=${tool}>${JSON.stringify(result.value)}</tool_response_json>`
64
+ }
65
+ ];
66
+ case "error-json":
67
+ return [
68
+ {
69
+ type: "text",
70
+ text: `<tool_response_error_json name=${tool}>${JSON.stringify(result.value)}</tool_response_error_json>`
71
+ }
72
+ ];
73
+ case "content":
74
+ return [
75
+ {
76
+ type: "text",
77
+ text: `<tool_response name=${tool}>`
78
+ },
79
+ ...result.value.map((part) => {
80
+ if (part.type === "text") {
81
+ return part;
82
+ }
83
+ if (part.mediaType.startsWith("image/")) {
84
+ return {
85
+ type: "image",
86
+ mediaType: part.mediaType,
87
+ image: part.data
88
+ };
89
+ }
90
+ return {
91
+ type: "file",
92
+ mediaType: part.mediaType,
93
+ data: part.data
94
+ };
95
+ }),
96
+ {
97
+ type: "text",
98
+ text: "</tool_response>"
99
+ }
100
+ ];
101
+ }
102
+ },
103
+ commandResult: (command, exitCode, stdout, stderr) => `<command>${command}</command>
104
+ <command_exit_code>${exitCode}</command_exit_code>
105
+ <command_stdout>
106
+ ${stdout}
107
+ </command_stdout>
108
+ <command_stderr>
109
+ ${stderr}
110
+ </command_stderr>`
111
+ };
112
+
113
+ // src/Agent/backoff.ts
114
+ function computeRateLimitBackoffSeconds(count, baseSeconds = 2, capSeconds = 60) {
115
+ if (!Number.isFinite(count) || count <= 0) {
116
+ count = 1;
117
+ }
118
+ if (!Number.isFinite(baseSeconds) || baseSeconds <= 0) {
119
+ baseSeconds = 2;
120
+ }
121
+ if (!Number.isFinite(capSeconds) || capSeconds <= 0) {
122
+ capSeconds = 60;
123
+ }
124
+ const delay = baseSeconds * 2 ** (count - 1);
125
+ return Math.min(delay, capSeconds);
126
+ }
127
+
128
+ // src/config.ts
2
129
  import { z } from "zod";
130
+ var providerModelSchema = z.object({
131
+ provider: z.string().optional(),
132
+ model: z.string().optional(),
133
+ parameters: z.record(z.string(), z.any()).optional()
134
+ });
135
+ var configSchema = z.object({
136
+ prices: z.record(
137
+ z.string(),
138
+ // provider
139
+ z.record(
140
+ z.string(),
141
+ // model
142
+ z.object({
143
+ inputPrice: z.number().optional(),
144
+ outputPrice: z.number().optional(),
145
+ cacheWritesPrice: z.number().optional(),
146
+ cacheReadsPrice: z.number().optional()
147
+ })
148
+ )
149
+ ).optional(),
150
+ providers: z.record(
151
+ z.string(),
152
+ z.object({
153
+ apiKey: z.string().optional(),
154
+ defaultModel: z.string().optional(),
155
+ defaultParameters: z.record(z.string(), z.any()).optional(),
156
+ location: z.string().optional(),
157
+ project: z.string().optional(),
158
+ keyFile: z.string().optional()
159
+ })
160
+ ).optional(),
161
+ defaultProvider: z.string().optional(),
162
+ defaultModel: z.string().optional(),
163
+ defaultParameters: z.record(z.string(), z.any()).optional(),
164
+ maxMessageCount: z.number().int().positive().optional(),
165
+ budget: z.number().positive().optional(),
166
+ retryCount: z.number().int().min(0).optional(),
167
+ requestTimeoutSeconds: z.number().int().positive().optional(),
168
+ summaryThreshold: z.number().int().positive().optional(),
169
+ scripts: z.record(
170
+ z.string(),
171
+ z.string().or(
172
+ z.object({
173
+ command: z.string(),
174
+ description: z.string()
175
+ })
176
+ )
177
+ ).optional(),
178
+ commands: z.object({
179
+ default: providerModelSchema.optional()
180
+ }).catchall(providerModelSchema).optional(),
181
+ rules: z.array(z.string()).optional().or(z.string()).optional(),
182
+ excludeFiles: z.array(z.string()).optional()
183
+ }).strict();
3
184
 
4
185
  // src/tool.ts
5
186
  var ToolResponseType = /* @__PURE__ */ ((ToolResponseType2) => {
6
187
  ToolResponseType2["Reply"] = "Reply";
7
188
  ToolResponseType2["Exit"] = "Exit";
8
- ToolResponseType2["Invalid"] = "Invalid";
9
189
  ToolResponseType2["Error"] = "Error";
10
- ToolResponseType2["Interrupted"] = "Interrupted";
11
- ToolResponseType2["HandOver"] = "HandOver";
12
- ToolResponseType2["Delegate"] = "Delegate";
13
- ToolResponseType2["Pause"] = "Pause";
14
190
  return ToolResponseType2;
15
191
  })(ToolResponseType || {});
16
192
 
193
+ // src/tools/appendMemory.ts
194
+ import { z as z2 } from "zod";
195
+ var toolInfo = {
196
+ name: "appendMemory",
197
+ description: "Appends content to a memory topic.",
198
+ parameters: z2.object({
199
+ topic: z2.string().nullish().describe('The topic to append to in memory. Defaults to ":default:".'),
200
+ content: z2.string().describe("The content to append.")
201
+ })
202
+ };
203
+ var handler = async (provider, args) => {
204
+ const { topic, content } = toolInfo.parameters.parse(args);
205
+ await provider.appendMemory(topic ?? void 0, content);
206
+ return {
207
+ type: "Reply" /* Reply */,
208
+ message: {
209
+ type: "text",
210
+ value: `Content appended to memory topic '${topic || ""}'.`
211
+ }
212
+ };
213
+ };
214
+ var appendMemory_default = {
215
+ ...toolInfo,
216
+ handler
217
+ };
218
+
17
219
  // src/tools/askFollowupQuestion.ts
18
- var questionObject = z.object({
19
- prompt: z.string().describe("The text of the question.").meta({ usageValue: "question text here" }),
20
- options: z.array(z.string()).default([]).describe("Ordered list of suggested answers (omit if none).").meta({ usageValue: "suggested answer here" })
220
+ import { z as z3 } from "zod";
221
+ var questionObject = z3.object({
222
+ prompt: z3.string().describe("The text of the question.").meta({ usageValue: "question text here" }),
223
+ options: z3.array(z3.string()).default([]).describe("Ordered list of suggested answers (omit if none).").meta({ usageValue: "suggested answer here" })
21
224
  });
22
- var toolInfo = {
23
- name: "ask_followup_question",
225
+ var toolInfo2 = {
226
+ name: "askFollowupQuestion",
24
227
  description: "Call this when vital details are missing. Pose each follow-up as one direct, unambiguous question. If it speeds the reply, add up to five short, mutually-exclusive answer options. Group any related questions in the same call to avoid a back-and-forth chain.",
25
- parameters: z.object({
26
- questions: z.array(questionObject).describe("One or more follow-up questions you need answered before you can continue.").meta({ usageValue: "questions here" })
228
+ parameters: z3.object({
229
+ questions: z3.array(questionObject).describe("One or more follow-up questions you need answered before you can continue.").meta({ usageValue: "questions here" })
27
230
  }).meta({
28
231
  examples: [
29
232
  {
@@ -69,7 +272,7 @@ var toolInfo = {
69
272
  ]
70
273
  })
71
274
  };
72
- var handler = async (provider, args) => {
275
+ var handler2 = async (provider, args) => {
73
276
  if (!provider.askFollowupQuestion) {
74
277
  return {
75
278
  type: "Error" /* Error */,
@@ -79,10 +282,10 @@ var handler = async (provider, args) => {
79
282
  }
80
283
  };
81
284
  }
82
- const { questions } = toolInfo.parameters.parse(args);
285
+ const { questions } = toolInfo2.parameters.parse(args);
83
286
  if (questions.length === 0) {
84
287
  return {
85
- type: "Invalid" /* Invalid */,
288
+ type: "Error" /* Error */,
86
289
  message: {
87
290
  type: "error-text",
88
291
  value: "No questions provided"
@@ -105,132 +308,15 @@ ${answer}
105
308
  }
106
309
  };
107
310
  };
108
- var isAvailable = (provider) => {
109
- return !!provider.askFollowupQuestion;
110
- };
111
311
  var askFollowupQuestion_default = {
112
- ...toolInfo,
113
- handler,
114
- isAvailable
115
- };
116
-
117
- // src/tools/attemptCompletion.ts
118
- import { z as z2 } from "zod";
119
- var toolInfo2 = {
120
- name: "attempt_completion",
121
- description: "Use this tool when you believe the user\u2019s requested task is complete. Indicate that your work is finished, but acknowledge the user may still provide additional instructions or questions if they want to continue. This tool MUST NOT to be used with any other tool.",
122
- parameters: z2.object({
123
- result: z2.string().describe(
124
- "The result of the task. Formulate this result in a way that is final and does not require further input from the user. Don't end your result with questions or offers for further assistance."
125
- ).meta({ usageValue: "Your final result description here" })
126
- }).meta({
127
- examples: [
128
- {
129
- description: "Request to present the result of the task",
130
- input: {
131
- result: "Your final result description here"
132
- }
133
- }
134
- ]
135
- })
136
- };
137
- var handler2 = async (provider, args) => {
138
- const parsed = toolInfo2.parameters.safeParse(args);
139
- if (!parsed.success) {
140
- return {
141
- type: "Invalid" /* Invalid */,
142
- message: {
143
- type: "error-text",
144
- value: `Invalid arguments for attempt_completion: ${parsed.error.message}`
145
- }
146
- };
147
- }
148
- const { result } = parsed.data;
149
- const moreMessage = await provider.attemptCompletion?.(result);
150
- if (!moreMessage) {
151
- return {
152
- type: "Exit" /* Exit */,
153
- message: result
154
- };
155
- }
156
- return {
157
- type: "Reply" /* Reply */,
158
- message: {
159
- type: "text",
160
- value: `<user_message>${moreMessage}</user_message>`
161
- }
162
- };
163
- };
164
- var isAvailable2 = (_provider) => {
165
- return true;
166
- };
167
- var attemptCompletion_default = {
168
312
  ...toolInfo2,
169
- handler: handler2,
170
- isAvailable: isAvailable2
171
- };
172
-
173
- // src/tools/delegate.ts
174
- import { z as z3 } from "zod";
175
- var toolInfo3 = {
176
- name: "delegate",
177
- description: "Temporarily delegate a task to another agent and receive the result back. This tool MUST NOT to be used with any other tool.",
178
- parameters: z3.object({
179
- agentName: z3.string().describe("The name of the agent to delegate the task to").meta({ usageValue: "Name of the target agent" }),
180
- task: z3.string().describe("The task to be completed by the target agent").meta({ usageValue: "Task description" }),
181
- context: z3.string().describe("The context information for the task").meta({ usageValue: "Context information" }),
182
- files: z3.preprocess((val) => {
183
- if (!val) return [];
184
- const values = Array.isArray(val) ? val : [val];
185
- return values.flatMap((i) => typeof i === "string" ? i.split(",") : []).filter((s) => s.length > 0);
186
- }, z3.array(z3.string())).optional().describe("The files relevant to the task. Comma separated paths").meta({ usageValue: "Relevant files" })
187
- }).meta({
188
- examples: [
189
- {
190
- description: "Delegate a code analysis task to the analyzer agent",
191
- input: {
192
- agentName: "analyzer",
193
- task: "Analyze the authentication implementation",
194
- context: "Need to understand the security implications of the current auth system",
195
- files: "src/auth/login.ts,src/auth/types.ts"
196
- }
197
- }
198
- ]
199
- })
200
- };
201
- var handler3 = async (_provider, args) => {
202
- const parsed = toolInfo3.parameters.safeParse(args);
203
- if (!parsed.success) {
204
- return {
205
- type: "Invalid" /* Invalid */,
206
- message: {
207
- type: "error-text",
208
- value: `Invalid arguments for delegate: ${parsed.error.message}`
209
- }
210
- };
211
- }
212
- const { agentName, task, context, files } = parsed.data;
213
- return {
214
- type: "Delegate" /* Delegate */,
215
- agentName,
216
- task,
217
- context,
218
- files: files ?? []
219
- };
220
- };
221
- var isAvailable3 = (_provider) => {
222
- return true;
223
- };
224
- var delegate_default = {
225
- ...toolInfo3,
226
- handler: handler3,
227
- isAvailable: isAvailable3
313
+ handler: handler2
228
314
  };
229
315
 
230
316
  // src/tools/executeCommand.ts
231
317
  import { z as z4 } from "zod";
232
- var toolInfo4 = {
233
- name: "execute_command",
318
+ var toolInfo3 = {
319
+ name: "executeCommand",
234
320
  description: "Run a single CLI command. The command is always executed in the project-root working directory (regardless of earlier commands). Prefer one-off shell commands over wrapper scripts for flexibility. **IMPORTANT**: After an `execute_command` call, you MUST stop and NOT allowed to make further tool calls in the same message.",
235
321
  parameters: z4.object({
236
322
  command: z4.string().describe("The exact command to run (valid for the current OS). It must be correctly formatted and free of harmful instructions.").meta({ usageValue: "your-command-here" }),
@@ -256,7 +342,7 @@ var toolInfo4 = {
256
342
  ]
257
343
  })
258
344
  };
259
- var handler4 = async (provider, args) => {
345
+ var handler3 = async (provider, args) => {
260
346
  if (!provider.executeCommand) {
261
347
  return {
262
348
  type: "Error" /* Error */,
@@ -266,7 +352,7 @@ var handler4 = async (provider, args) => {
266
352
  }
267
353
  };
268
354
  }
269
- const { command, requiresApproval } = toolInfo4.parameters.parse(args);
355
+ const { command, requiresApproval } = toolInfo3.parameters.parse(args);
270
356
  try {
271
357
  const result = await provider.executeCommand(command, requiresApproval);
272
358
  let message = `<command>${command}</command>
@@ -312,19 +398,15 @@ ${result.stderr}
312
398
  };
313
399
  }
314
400
  };
315
- var isAvailable4 = (provider) => {
316
- return !!provider.executeCommand;
317
- };
318
401
  var executeCommand_default = {
319
- ...toolInfo4,
320
- handler: handler4,
321
- isAvailable: isAvailable4
402
+ ...toolInfo3,
403
+ handler: handler3
322
404
  };
323
405
 
324
406
  // src/tools/fetchUrl.ts
325
407
  import { z as z5 } from "zod";
326
- var toolInfo5 = {
327
- name: "fetch_url",
408
+ var toolInfo4 = {
409
+ name: "fetchUrl",
328
410
  description: "Fetch the content located at one or more HTTP(S) URLs and return it in Markdown format. This works for standard web pages as well as raw files (e.g. README.md, source code) hosted on platforms like GitHub.",
329
411
  parameters: z5.object({
330
412
  url: z5.preprocess((val) => {
@@ -355,7 +437,7 @@ var toolInfo5 = {
355
437
  ]
356
438
  })
357
439
  };
358
- var handler5 = async (provider, args) => {
440
+ var handler4 = async (provider, args) => {
359
441
  if (!provider.fetchUrl) {
360
442
  return {
361
443
  type: "Error" /* Error */,
@@ -365,7 +447,7 @@ var handler5 = async (provider, args) => {
365
447
  }
366
448
  };
367
449
  }
368
- const { url: urls } = toolInfo5.parameters.parse(args);
450
+ const { url: urls } = toolInfo4.parameters.parse(args);
369
451
  if (urls.length === 0) {
370
452
  return {
371
453
  type: "Error" /* Error */,
@@ -394,96 +476,35 @@ var handler5 = async (provider, args) => {
394
476
  }
395
477
  };
396
478
  };
397
- var isAvailable5 = (provider) => {
398
- return typeof provider.fetchUrl === "function";
399
- };
400
479
  var fetchUrl_default = {
401
- ...toolInfo5,
402
- handler: handler5,
403
- isAvailable: isAvailable5
480
+ ...toolInfo4,
481
+ handler: handler4
404
482
  };
405
483
 
406
- // src/tools/handOver.ts
484
+ // src/tools/listFiles.ts
407
485
  import { z as z6 } from "zod";
408
- var toolInfo6 = {
409
- name: "hand_over",
410
- description: "Hand over the current task to another agent to complete. This tool MUST NOT to be used with any other tool.",
486
+ var toolInfo5 = {
487
+ name: "listFiles",
488
+ description: "Request to list files and directories within the specified directory. If recursive is true, it will list all files and directories recursively. If recursive is false or not provided, it will only list the top-level contents. Do not use this tool to confirm the existence of files you may have created, as the user will let you know if the files were created successfully or not.",
411
489
  parameters: z6.object({
412
- agentName: z6.string().describe("The name of the agent to hand over the task to").meta({ usageValue: "Name of the target agent" }),
413
- task: z6.string().describe("The task to be completed by the target agent").meta({ usageValue: "Task description" }),
414
- context: z6.string().describe("The context information for the task").meta({ usageValue: "Context information" }),
415
- files: z6.preprocess((val) => {
416
- if (!val) return [];
417
- const values = Array.isArray(val) ? val : [val];
418
- return values.flatMap((i) => typeof i === "string" ? i.split(",") : []).filter((s) => s.length > 0);
419
- }, z6.array(z6.string())).optional().describe("The files relevant to the task. Comma separated paths").meta({ usageValue: "Relevant files" })
420
- }).meta({
421
- examples: [
422
- {
423
- description: "Hand over a coding task to the coder agent",
424
- input: {
425
- agentName: "coder",
426
- task: "Implement the login feature",
427
- context: "We need a secure login system with email and password",
428
- files: "src/auth/login.ts,src/auth/types.ts"
429
- }
430
- }
431
- ]
432
- })
433
- };
434
- var handler6 = async (_provider, args) => {
435
- const parsed = toolInfo6.parameters.safeParse(args);
436
- if (!parsed.success) {
437
- return {
438
- type: "Invalid" /* Invalid */,
439
- message: {
440
- type: "error-text",
441
- value: `Invalid arguments for hand_over: ${parsed.error.message}`
442
- }
443
- };
444
- }
445
- const { agentName, task, context, files } = parsed.data;
446
- return {
447
- type: "HandOver" /* HandOver */,
448
- agentName,
449
- task,
450
- context,
451
- files: files ?? []
452
- };
453
- };
454
- var isAvailable6 = (_provider) => {
455
- return true;
456
- };
457
- var handOver_default = {
458
- ...toolInfo6,
459
- handler: handler6,
460
- isAvailable: isAvailable6
461
- };
462
-
463
- // src/tools/listFiles.ts
464
- import { z as z7 } from "zod";
465
- var toolInfo7 = {
466
- name: "list_files",
467
- description: "Request to list files and directories within the specified directory. If recursive is true, it will list all files and directories recursively. If recursive is false or not provided, it will only list the top-level contents. Do not use this tool to confirm the existence of files you may have created, as the user will let you know if the files were created successfully or not.",
468
- parameters: z7.object({
469
- path: z7.string().describe("The path of the directory to list contents for (relative to the current working directory)").meta({ usageValue: "Directory path here" }),
470
- maxCount: z7.coerce.number().optional().default(2e3).describe("The maximum number of files to list. Default to 2000").meta({ usageValue: "Maximum number of files to list (optional)" }),
471
- recursive: z7.preprocess((val) => {
490
+ path: z6.string().describe("The path of the directory to list contents for (relative to the current working directory)").meta({ usageValue: "Directory path here" }),
491
+ maxCount: z6.coerce.number().optional().default(2e3).describe("The maximum number of files to list. Default to 2000").meta({ usageValue: "Maximum number of files to list (optional)" }),
492
+ recursive: z6.preprocess((val) => {
472
493
  if (typeof val === "string") {
473
494
  const lower = val.toLowerCase();
474
495
  if (lower === "false") return false;
475
496
  if (lower === "true") return true;
476
497
  }
477
498
  return val;
478
- }, z7.boolean().optional().default(true)).describe("Whether to list files recursively. Use true for recursive listing, false or omit for top-level only.").meta({ usageValue: "true or false (optional)" }),
479
- includeIgnored: z7.preprocess((val) => {
499
+ }, z6.boolean().optional().default(true)).describe("Whether to list files recursively. Use true for recursive listing, false or omit for top-level only.").meta({ usageValue: "true or false (optional)" }),
500
+ includeIgnored: z6.preprocess((val) => {
480
501
  if (typeof val === "string") {
481
502
  const lower = val.toLowerCase();
482
503
  if (lower === "false") return false;
483
504
  if (lower === "true") return true;
484
505
  }
485
506
  return val;
486
- }, z7.boolean().optional().default(false)).describe("Whether to include ignored files. Use true to include files ignored by .gitignore.").meta({ usageValue: "true or false (optional)" })
507
+ }, z6.boolean().optional().default(false)).describe("Whether to include ignored files. Use true to include files ignored by .gitignore.").meta({ usageValue: "true or false (optional)" })
487
508
  }).meta({
488
509
  examples: [
489
510
  {
@@ -496,7 +517,7 @@ var toolInfo7 = {
496
517
  ]
497
518
  })
498
519
  };
499
- var handler7 = async (provider, args) => {
520
+ var handler5 = async (provider, args) => {
500
521
  if (!provider.listFiles) {
501
522
  return {
502
523
  type: "Error" /* Error */,
@@ -506,7 +527,7 @@ var handler7 = async (provider, args) => {
506
527
  }
507
528
  };
508
529
  }
509
- const { path, maxCount, recursive, includeIgnored } = toolInfo7.parameters.parse(args);
530
+ const { path, maxCount, recursive, includeIgnored } = toolInfo5.parameters.parse(args);
510
531
  const [files, limitReached] = await provider.listFiles(path, recursive, maxCount, includeIgnored);
511
532
  return {
512
533
  type: "Reply" /* Reply */,
@@ -520,13 +541,35 @@ ${files.join("\n")}
520
541
  }
521
542
  };
522
543
  };
523
- var isAvailable7 = (provider) => {
524
- return !!provider.listFiles;
525
- };
526
544
  var listFiles_default = {
527
- ...toolInfo7,
528
- handler: handler7,
529
- isAvailable: isAvailable7
545
+ ...toolInfo5,
546
+ handler: handler5
547
+ };
548
+
549
+ // src/tools/listMemoryTopics.ts
550
+ import { z as z7 } from "zod";
551
+ var toolInfo6 = {
552
+ name: "listMemoryTopics",
553
+ description: "Lists all topics in memory.",
554
+ parameters: z7.object({})
555
+ };
556
+ var handler6 = async (provider, _args) => {
557
+ const topics = await provider.listMemoryTopics();
558
+ if (!topics.length) {
559
+ return { type: "Reply" /* Reply */, message: { type: "text", value: "No topics found." } };
560
+ }
561
+ return {
562
+ type: "Reply" /* Reply */,
563
+ message: {
564
+ type: "text",
565
+ value: `Memory topics:
566
+ ${topics.join("\n")}`
567
+ }
568
+ };
569
+ };
570
+ var listMemoryTopics_default = {
571
+ ...toolInfo6,
572
+ handler: handler6
530
573
  };
531
574
 
532
575
  // src/tools/provider.ts
@@ -555,21 +598,18 @@ var MockProvider = class {
555
598
  async askFollowupQuestion(_question, _options) {
556
599
  return "mock answer";
557
600
  }
558
- async attemptCompletion(_result) {
559
- return "mock completion";
560
- }
561
601
  };
562
602
 
563
603
  // src/tools/readBinaryFile.ts
564
604
  import { z as z8 } from "zod";
565
- var toolInfo8 = {
566
- name: "read_binary_file",
605
+ var toolInfo7 = {
606
+ name: "readBinaryFile",
567
607
  description: "Read a binary file from a URL or local path. Use file:// prefix to access local files. This can be used to access non-text files such as PDFs or images.",
568
608
  parameters: z8.object({
569
609
  url: z8.string().describe("The URL or local path of the file to read.")
570
610
  })
571
611
  };
572
- var handler8 = async (provider, args) => {
612
+ var handler7 = async (provider, args) => {
573
613
  if (!provider.readBinaryFile) {
574
614
  return {
575
615
  type: "Error" /* Error */,
@@ -579,7 +619,7 @@ var handler8 = async (provider, args) => {
579
619
  }
580
620
  };
581
621
  }
582
- const { url } = toolInfo8.parameters.parse(args);
622
+ const { url } = toolInfo7.parameters.parse(args);
583
623
  try {
584
624
  const filePart = await provider.readBinaryFile(url);
585
625
  return {
@@ -607,19 +647,15 @@ var handler8 = async (provider, args) => {
607
647
  };
608
648
  }
609
649
  };
610
- var isAvailable8 = (provider) => {
611
- return typeof provider.readBinaryFile === "function";
612
- };
613
650
  var readBinaryFile_default = {
614
- ...toolInfo8,
615
- handler: handler8,
616
- isAvailable: isAvailable8
651
+ ...toolInfo7,
652
+ handler: handler7
617
653
  };
618
654
 
619
655
  // src/tools/readFile.ts
620
656
  import { z as z9 } from "zod";
621
- var toolInfo9 = {
622
- name: "read_file",
657
+ var toolInfo8 = {
658
+ name: "readFile",
623
659
  description: "Request to read the contents of one or multiple files at the specified paths. Use comma separated paths to read multiple files. Use this when you need to examine the contents of an existing file you do not know the contents of, for example to analyze code, review text files, or extract information from configuration files. May not be suitable for other types of binary files, as it returns the raw content as a string. Try to list all the potential files are relevent to the task, and then use this tool to read all the relevant files.",
624
660
  parameters: z9.object({
625
661
  path: z9.preprocess((val) => {
@@ -652,7 +688,7 @@ var toolInfo9 = {
652
688
  ]
653
689
  })
654
690
  };
655
- var handler9 = async (provider, args) => {
691
+ var handler8 = async (provider, args) => {
656
692
  if (!provider.readFile) {
657
693
  return {
658
694
  type: "Error" /* Error */,
@@ -662,7 +698,7 @@ var handler9 = async (provider, args) => {
662
698
  }
663
699
  };
664
700
  }
665
- const { path: paths, includeIgnored } = toolInfo9.parameters.parse(args);
701
+ const { path: paths, includeIgnored } = toolInfo8.parameters.parse(args);
666
702
  const resp = [];
667
703
  for (const path of paths) {
668
704
  const fileContent = await provider.readFile(path, includeIgnored);
@@ -685,22 +721,54 @@ var handler9 = async (provider, args) => {
685
721
  }
686
722
  };
687
723
  };
688
- var isAvailable9 = (provider) => {
689
- return !!provider.readFile;
690
- };
691
724
  var readFile_default = {
725
+ ...toolInfo8,
726
+ handler: handler8
727
+ };
728
+
729
+ // src/tools/readMemory.ts
730
+ import { z as z10 } from "zod";
731
+ var toolInfo9 = {
732
+ name: "readMemory",
733
+ description: "Reads content from a memory topic.",
734
+ parameters: z10.object({
735
+ topic: z10.string().optional().describe('The topic to read from memory. Defaults to ":default:".')
736
+ })
737
+ };
738
+ var handler9 = async (provider, args) => {
739
+ const { topic } = toolInfo9.parameters.parse(args);
740
+ const content = await provider.readMemory(topic);
741
+ if (content) {
742
+ return {
743
+ type: "Reply" /* Reply */,
744
+ message: {
745
+ type: "text",
746
+ value: `<memory${topic ? ` topic="${topic}"` : ""}>
747
+ ${content}
748
+ </memory>`
749
+ }
750
+ };
751
+ }
752
+ return {
753
+ type: "Reply" /* Reply */,
754
+ message: {
755
+ type: "text",
756
+ value: `<memory ${topic ? `topic="${topic}"` : ""} isEmpty="true" />`
757
+ }
758
+ };
759
+ };
760
+ var readMemory_default = {
692
761
  ...toolInfo9,
693
- handler: handler9,
694
- isAvailable: isAvailable9
762
+ handler: handler9
695
763
  };
696
764
 
697
765
  // src/tools/removeFile.ts
698
- import { z as z10 } from "zod";
766
+ import { z as z11 } from "zod";
699
767
  var toolInfo10 = {
700
- name: "remove_file",
768
+ name: "removeFile",
701
769
  description: "Request to remove a file at the specified path.",
702
- parameters: z10.object({
703
- path: z10.string().describe("The path of the file to remove").meta({ usageValue: "File path here" })
770
+ parameters: z11.object({
771
+ path: z11.string().describe("The path of the file to remove").meta({ usageValue: "File path here" })
704
772
  }).meta({
705
773
  examples: [
706
774
  {
@@ -725,10 +793,10 @@ var handler10 = async (provider, args) => {
725
793
  const parsed = toolInfo10.parameters.safeParse(args);
726
794
  if (!parsed.success) {
727
795
  return {
728
- type: "Invalid" /* Invalid */,
796
+ type: "Error" /* Error */,
729
797
  message: {
730
798
  type: "error-text",
731
- value: `Invalid arguments for remove_file: ${parsed.error.message}`
799
+ value: `Invalid arguments for removeFile: ${parsed.error.message}`
732
800
  }
733
801
  };
734
802
  }
@@ -742,23 +810,44 @@ var handler10 = async (provider, args) => {
742
810
  }
743
811
  };
744
812
  };
745
- var isAvailable10 = (provider) => {
746
- return !!provider.removeFile;
747
- };
748
813
  var removeFile_default = {
749
814
  ...toolInfo10,
750
- handler: handler10,
751
- isAvailable: isAvailable10
815
+ handler: handler10
752
816
  };
753
817
 
754
- // src/tools/renameFile.ts
755
- import { z as z11 } from "zod";
818
+ // src/tools/removeMemory.ts
819
+ import { z as z12 } from "zod";
756
820
  var toolInfo11 = {
757
- name: "rename_file",
821
+ name: "removeMemory",
822
+ description: "Removes a topic from memory.",
823
+ parameters: z12.object({
824
+ topic: z12.string().optional().describe('The topic to remove from memory. Defaults to ":default:".')
825
+ })
826
+ };
827
+ var handler11 = async (provider, args) => {
828
+ const { topic } = toolInfo11.parameters.parse(args);
829
+ await provider.removeMemory(topic);
830
+ return {
831
+ type: "Reply" /* Reply */,
832
+ message: {
833
+ type: "text",
834
+ value: `Memory topic '${topic || ""}' removed.`
835
+ }
836
+ };
837
+ };
838
+ var removeMemory_default = {
839
+ ...toolInfo11,
840
+ handler: handler11
841
+ };
842
+
843
+ // src/tools/renameFile.ts
844
+ import { z as z13 } from "zod";
845
+ var toolInfo12 = {
846
+ name: "renameFile",
758
847
  description: "Request to rename a file from source path to target path.",
759
- parameters: z11.object({
760
- source_path: z11.string().describe("The current path of the file").meta({ usageValue: "Source file path here" }),
761
- target_path: z11.string().describe("The new path for the file").meta({ usageValue: "Target file path here" })
848
+ parameters: z13.object({
849
+ source_path: z13.string().describe("The current path of the file").meta({ usageValue: "Source file path here" }),
850
+ target_path: z13.string().describe("The new path for the file").meta({ usageValue: "Target file path here" })
762
851
  }).meta({
763
852
  examples: [
764
853
  {
@@ -771,7 +860,7 @@ var toolInfo11 = {
771
860
  ]
772
861
  })
773
862
  };
774
- var handler11 = async (provider, args) => {
863
+ var handler12 = async (provider, args) => {
775
864
  if (!provider.renameFile) {
776
865
  return {
777
866
  type: "Error" /* Error */,
@@ -781,7 +870,7 @@ var handler11 = async (provider, args) => {
781
870
  }
782
871
  };
783
872
  }
784
- const { source_path, target_path } = toolInfo11.parameters.parse(args);
873
+ const { source_path, target_path } = toolInfo12.parameters.parse(args);
785
874
  await provider.renameFile(source_path, target_path);
786
875
  return {
787
876
  type: "Reply" /* Reply */,
@@ -791,17 +880,13 @@ var handler11 = async (provider, args) => {
791
880
  }
792
881
  };
793
882
  };
794
- var isAvailable11 = (provider) => {
795
- return !!provider.renameFile;
796
- };
797
883
  var renameFile_default = {
798
- ...toolInfo11,
799
- handler: handler11,
800
- isAvailable: isAvailable11
884
+ ...toolInfo12,
885
+ handler: handler12
801
886
  };
802
887
 
803
888
  // src/tools/replaceInFile.ts
804
- import { z as z12 } from "zod";
889
+ import { z as z14 } from "zod";
805
890
 
806
891
  // src/tools/utils/replaceInFile.ts
807
892
  var replaceInFile = (fileContent, diff) => {
@@ -878,12 +963,12 @@ var replaceInFile = (fileContent, diff) => {
878
963
  };
879
964
 
880
965
  // src/tools/replaceInFile.ts
881
- var toolInfo12 = {
882
- name: "replace_in_file",
966
+ var toolInfo13 = {
967
+ name: "replaceInFile",
883
968
  description: "Request to replace sections of content in an existing file using SEARCH/REPLACE blocks that define exact changes to specific parts of the file. This tool should be used when you need to make targeted changes to specific parts of a file.",
884
- parameters: z12.object({
885
- path: z12.string().describe("The path of the file to modify").meta({ usageValue: "File path here" }),
886
- diff: z12.string().describe(
969
+ parameters: z14.object({
970
+ path: z14.string().describe("The path of the file to modify").meta({ usageValue: "File path here" }),
971
+ diff: z14.string().describe(
887
972
  `One or more SEARCH/REPLACE blocks following this exact format:
888
973
  \`\`\`
889
974
  <<<<<<< SEARCH
@@ -990,7 +1075,7 @@ function oldFeature() {
990
1075
  ]
991
1076
  })
992
1077
  };
993
- var handler12 = async (provider, args) => {
1078
+ var handler13 = async (provider, args) => {
994
1079
  if (!provider.readFile || !provider.writeFile) {
995
1080
  return {
996
1081
  type: "Error" /* Error */,
@@ -1000,13 +1085,13 @@ var handler12 = async (provider, args) => {
1000
1085
  }
1001
1086
  };
1002
1087
  }
1003
- const parsed = toolInfo12.parameters.safeParse(args);
1088
+ const parsed = toolInfo13.parameters.safeParse(args);
1004
1089
  if (!parsed.success) {
1005
1090
  return {
1006
- type: "Invalid" /* Invalid */,
1091
+ type: "Error" /* Error */,
1007
1092
  message: {
1008
1093
  type: "error-text",
1009
- value: `Invalid arguments for replace_in_file: ${parsed.error.message}`
1094
+ value: `Invalid arguments for replaceInFile: ${parsed.error.message}`
1010
1095
  }
1011
1096
  };
1012
1097
  }
@@ -1055,36 +1140,58 @@ var handler12 = async (provider, args) => {
1055
1140
  };
1056
1141
  } catch (error) {
1057
1142
  return {
1058
- type: "Invalid" /* Invalid */,
1143
+ type: "Error" /* Error */,
1059
1144
  message: {
1060
1145
  type: "error-text",
1061
- value: `Invalid arguments for replace_in_file: ${error}`
1146
+ value: `Invalid arguments for replaceInFile: ${error}`
1062
1147
  }
1063
1148
  };
1064
1149
  }
1065
1150
  };
1066
- var isAvailable12 = (provider) => {
1067
- return !!provider.readFile && !!provider.writeFile;
1068
- };
1069
1151
  var replaceInFile_default = {
1070
- ...toolInfo12,
1071
- handler: handler12,
1072
- isAvailable: isAvailable12
1152
+ ...toolInfo13,
1153
+ handler: handler13
1154
+ };
1155
+
1156
+ // src/tools/replaceMemory.ts
1157
+ import { z as z15 } from "zod";
1158
+ var toolInfo14 = {
1159
+ name: "replaceMemory",
1160
+ description: "Replaces content of a memory topic.",
1161
+ parameters: z15.object({
1162
+ topic: z15.string().optional().describe('The topic to replace in memory. Defaults to ":default:".'),
1163
+ content: z15.string().describe("The new content.")
1164
+ })
1165
+ };
1166
+ var handler14 = async (provider, args) => {
1167
+ const { topic, content } = toolInfo14.parameters.parse(args);
1168
+ await provider.replaceMemory(topic, content);
1169
+ return {
1170
+ type: "Reply" /* Reply */,
1171
+ message: {
1172
+ type: "text",
1173
+ value: `Memory topic '${topic || ""}' replaced.`
1174
+ }
1175
+ };
1176
+ };
1177
+ var replaceMemory_default = {
1178
+ ...toolInfo14,
1179
+ handler: handler14
1073
1180
  };
1074
1181
 
1075
1182
  // src/tools/searchFiles.ts
1076
- import { z as z13 } from "zod";
1077
- var toolInfo13 = {
1078
- name: "search_files",
1183
+ import { z as z16 } from "zod";
1184
+ var toolInfo15 = {
1185
+ name: "searchFiles",
1079
1186
  description: "Request to perform a regex search across files in a specified directory, outputting context-rich results that include surrounding lines. This tool searches for patterns or specific content across multiple files, displaying each match with encapsulating context.",
1080
- parameters: z13.object({
1081
- path: z13.string().describe(
1187
+ parameters: z16.object({
1188
+ path: z16.string().describe(
1082
1189
  "The path of the directory to search in (relative to the current working directory). This directory will be recursively searched."
1083
1190
  ).meta({ usageValue: "Directory path here" }),
1084
- regex: z13.string().describe("The regular expression pattern to search for. Uses Rust regex syntax.").meta({
1191
+ regex: z16.string().describe("The regular expression pattern to search for. Uses Rust regex syntax.").meta({
1085
1192
  usageValue: "Your regex pattern here"
1086
1193
  }),
1087
- filePattern: z13.string().optional().describe(
1194
+ filePattern: z16.string().optional().describe(
1088
1195
  'Comma-separated glob pattern to filter files (e.g., "*.ts" for TypeScript files or "*.ts,*.js" for both TypeScript and JavaScript files). If not provided, it will search all files (*).'
1089
1196
  ).meta({
1090
1197
  usageValue: "file pattern here (optional)"
@@ -1102,7 +1209,7 @@ var toolInfo13 = {
1102
1209
  ]
1103
1210
  })
1104
1211
  };
1105
- var handler13 = async (provider, args) => {
1212
+ var handler15 = async (provider, args) => {
1106
1213
  if (!provider.searchFiles) {
1107
1214
  return {
1108
1215
  type: "Error" /* Error */,
@@ -1112,13 +1219,13 @@ var handler13 = async (provider, args) => {
1112
1219
  }
1113
1220
  };
1114
1221
  }
1115
- const parsed = toolInfo13.parameters.safeParse(args);
1222
+ const parsed = toolInfo15.parameters.safeParse(args);
1116
1223
  if (!parsed.success) {
1117
1224
  return {
1118
- type: "Invalid" /* Invalid */,
1225
+ type: "Error" /* Error */,
1119
1226
  message: {
1120
1227
  type: "error-text",
1121
- value: `Invalid arguments for search_files: ${parsed.error.message}`
1228
+ value: `Invalid arguments for searchFiles: ${parsed.error.message}`
1122
1229
  }
1123
1230
  };
1124
1231
  }
@@ -1148,23 +1255,19 @@ ${files.join("\n")}
1148
1255
  };
1149
1256
  }
1150
1257
  };
1151
- var isAvailable13 = (provider) => {
1152
- return !!provider.searchFiles;
1153
- };
1154
1258
  var searchFiles_default = {
1155
- ...toolInfo13,
1156
- handler: handler13,
1157
- isAvailable: isAvailable13
1259
+ ...toolInfo15,
1260
+ handler: handler15
1158
1261
  };
1159
1262
 
1160
1263
  // src/tools/writeToFile.ts
1161
- import { z as z14 } from "zod";
1162
- var toolInfo14 = {
1163
- name: "write_to_file",
1264
+ import { z as z17 } from "zod";
1265
+ var toolInfo16 = {
1266
+ name: "writeToFile",
1164
1267
  description: "Request to write content to a file at the specified path. If the file exists, it will be overwritten with the provided content. If the file doesn't exist, it will be created. This tool will automatically create any directories needed to write the file. Ensure that the output content does not include incorrect escaped character patterns such as `&lt;`, `&gt;`, or `&amp;`. Also ensure there is no unwanted CDATA tags in the content.",
1165
- parameters: z14.object({
1166
- path: z14.string().describe("The path of the file to write to").meta({ usageValue: "File path here" }),
1167
- content: z14.string().describe(
1268
+ parameters: z17.object({
1269
+ path: z17.string().describe("The path of the file to write to").meta({ usageValue: "File path here" }),
1270
+ content: z17.string().describe(
1168
1271
  "The content to write to the file. ALWAYS provide the COMPLETE intended content of the file, without any truncation or omissions. You MUST include ALL parts of the file, even if they haven't been modified."
1169
1272
  ).meta({ usageValue: "Your file content here" })
1170
1273
  }).meta({
@@ -1190,7 +1293,7 @@ export default App;
1190
1293
  ]
1191
1294
  })
1192
1295
  };
1193
- var handler14 = async (provider, args) => {
1296
+ var handler16 = async (provider, args) => {
1194
1297
  if (!provider.writeFile) {
1195
1298
  return {
1196
1299
  type: "Error" /* Error */,
@@ -1200,13 +1303,13 @@ var handler14 = async (provider, args) => {
1200
1303
  }
1201
1304
  };
1202
1305
  }
1203
- const parsed = toolInfo14.parameters.safeParse(args);
1306
+ const parsed = toolInfo16.parameters.safeParse(args);
1204
1307
  if (!parsed.success) {
1205
1308
  return {
1206
- type: "Invalid" /* Invalid */,
1309
+ type: "Error" /* Error */,
1207
1310
  message: {
1208
1311
  type: "error-text",
1209
- value: `Invalid arguments for write_to_file: ${parsed.error.message}`
1312
+ value: `Invalid arguments for writeToFile: ${parsed.error.message}`
1210
1313
  }
1211
1314
  };
1212
1315
  }
@@ -1222,35 +1325,9 @@ var handler14 = async (provider, args) => {
1222
1325
  }
1223
1326
  };
1224
1327
  };
1225
- var isAvailable14 = (provider) => {
1226
- return !!provider.writeFile;
1227
- };
1228
1328
  var writeToFile_default = {
1229
- ...toolInfo14,
1230
- handler: handler14,
1231
- isAvailable: isAvailable14
1232
- };
1233
-
1234
- // src/getAvailableTools.ts
1235
- var getAvailableTools = ({
1236
- provider,
1237
- allTools,
1238
- hasAgent
1239
- }) => {
1240
- const tools = [];
1241
- for (const tool of allTools) {
1242
- if (!hasAgent) {
1243
- switch (tool.name) {
1244
- case handOver_default.name:
1245
- case delegate_default.name:
1246
- continue;
1247
- }
1248
- }
1249
- if (tool.isAvailable(provider)) {
1250
- tools.push(tool);
1251
- }
1252
- }
1253
- return tools;
1329
+ ...toolInfo16,
1330
+ handler: handler16
1254
1331
  };
1255
1332
 
1256
1333
  // src/UsageMeter.ts
@@ -1400,2036 +1477,29 @@ var UsageMeter = class {
1400
1477
  };
1401
1478
  }
1402
1479
  };
1403
-
1404
- // src/Agent/AgentBase.ts
1405
- import { jsonSchema, streamText } from "ai";
1406
- import { camelCase } from "lodash-es";
1407
- import { toJSONSchema } from "zod";
1408
-
1409
- // src/tool-v1-compat.ts
1410
- import { z as z15 } from "zod";
1411
- function zodSchemaToParameters(schema) {
1412
- const parameters = [];
1413
- const { shape } = schema;
1414
- for (const name in shape) {
1415
- const def = shape[name];
1416
- const isOptional = def.safeParse(void 0).success;
1417
- const description = def.description || "";
1418
- const param = {
1419
- name,
1420
- description,
1421
- required: !isOptional
1422
- };
1423
- const usageValue = def.meta()?.usageValue;
1424
- if (usageValue) {
1425
- param.usageValue = usageValue;
1426
- }
1427
- if (def instanceof z15.ZodObject) {
1428
- param.children = zodSchemaToParameters(def);
1429
- } else if (def instanceof z15.ZodArray) {
1430
- param.allowMultiple = true;
1431
- const element = def.element;
1432
- if (element instanceof z15.ZodObject) {
1433
- param.children = zodSchemaToParameters(element);
1434
- }
1435
- }
1436
- parameters.push(param);
1437
- }
1438
- return parameters;
1439
- }
1440
- function toToolInfoV1(tool) {
1441
- const { parameters: zodSchema, ...rest } = tool;
1442
- const v1Parameters = zodSchemaToParameters(zodSchema);
1443
- const examples = zodSchema.meta()?.examples;
1444
- const v1Tool = {
1445
- ...rest,
1446
- parameters: v1Parameters
1447
- };
1448
- if (examples) {
1449
- v1Tool.examples = examples;
1450
- }
1451
- return v1Tool;
1452
- }
1453
-
1454
- // src/Agent/backoff.ts
1455
- function computeRateLimitBackoffSeconds(count, baseSeconds = 2, capSeconds = 60) {
1456
- if (!Number.isFinite(count) || count <= 0) {
1457
- count = 1;
1458
- }
1459
- if (!Number.isFinite(baseSeconds) || baseSeconds <= 0) {
1460
- baseSeconds = 2;
1461
- }
1462
- if (!Number.isFinite(capSeconds) || capSeconds <= 0) {
1463
- capSeconds = 60;
1464
- }
1465
- const delay = baseSeconds * 2 ** (count - 1);
1466
- return Math.min(delay, capSeconds);
1467
- }
1468
-
1469
- // src/Agent/parseAssistantMessage.ts
1470
- function parseNestedParameters(content, parameterPrefix, childrenParams) {
1471
- const result = {};
1472
- const nestedParamRegex = new RegExp(`<${parameterPrefix}([^>]+)>([\\s\\S]*?)<\\/${parameterPrefix}\\1>`, "gs");
1473
- const arrayParams = /* @__PURE__ */ new Set();
1474
- if (childrenParams) {
1475
- for (const param of childrenParams) {
1476
- if (param.allowMultiple) {
1477
- arrayParams.add(param.name);
1478
- }
1479
- }
1480
- }
1481
- while (true) {
1482
- const match = nestedParamRegex.exec(content);
1483
- if (match === null) break;
1484
- const paramName = match[1];
1485
- const paramContent = match[2].trim();
1486
- const childParam = childrenParams?.find((p) => p.name === paramName);
1487
- if (paramContent.includes(`<${parameterPrefix}`) && childParam?.children) {
1488
- const nestedResult = parseNestedParameters(paramContent, parameterPrefix, childParam.children);
1489
- if (arrayParams.has(paramName)) {
1490
- if (!result[paramName]) {
1491
- result[paramName] = [];
1492
- }
1493
- if (Array.isArray(result[paramName])) {
1494
- result[paramName].push(nestedResult);
1495
- } else {
1496
- result[paramName] = [nestedResult];
1497
- }
1498
- } else {
1499
- result[paramName] = nestedResult;
1500
- }
1501
- } else if (arrayParams.has(paramName)) {
1502
- if (!result[paramName]) {
1503
- result[paramName] = [];
1504
- }
1505
- if (Array.isArray(result[paramName])) {
1506
- result[paramName].push(paramContent);
1507
- } else {
1508
- result[paramName] = [paramContent];
1509
- }
1510
- } else {
1511
- result[paramName] = paramContent;
1512
- }
1513
- }
1514
- return result;
1515
- }
1516
- function parseAssistantMessage(assistantMessage, tools, toolNamePrefix) {
1517
- const parameterPrefix = `${toolNamePrefix}parameter_`;
1518
- const results = [];
1519
- const toolTags = tools.map((tool) => `${toolNamePrefix}${tool.name}`);
1520
- const toolPattern = toolTags.join("|");
1521
- let remainingMessage = assistantMessage;
1522
- let match;
1523
- const tagRegex = new RegExp(`<(${toolPattern})>([\\s\\S]*?)<\\/\\1>`, "s");
1524
- while (true) {
1525
- match = tagRegex.exec(remainingMessage);
1526
- if (match === null) break;
1527
- const beforeTag = remainingMessage.slice(0, match.index).trim();
1528
- if (beforeTag) {
1529
- results.push({
1530
- type: "text",
1531
- content: beforeTag
1532
- });
1533
- }
1534
- const tagName = match[1];
1535
- const toolName = tagName.replace(toolNamePrefix, "");
1536
- const tool = tools.find((t) => t.name === toolName);
1537
- const fullTagContent = match[0];
1538
- if (tool) {
1539
- const params = {};
1540
- for (const param of tool.parameters) {
1541
- const paramName = `${parameterPrefix}${param.name}`;
1542
- const escapedParamName = paramName.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
1543
- const paramRegex = new RegExp(`<${escapedParamName}>([\\s\\S]*?)<\\/${escapedParamName}>`, "gs");
1544
- const paramMatches = [];
1545
- while (true) {
1546
- const paramMatch = paramRegex.exec(fullTagContent);
1547
- if (paramMatch === null) break;
1548
- paramMatches.push(paramMatch[1].trim());
1549
- }
1550
- if (paramMatches.length > 0) {
1551
- if (paramMatches.length === 1 && !param.allowMultiple) {
1552
- const paramContent = paramMatches[0];
1553
- if (paramContent.includes(`<${parameterPrefix}`) && param.children) {
1554
- params[param.name] = parseNestedParameters(paramContent, parameterPrefix, param.children);
1555
- } else {
1556
- params[param.name] = paramContent;
1557
- }
1558
- } else {
1559
- if (param.allowMultiple) {
1560
- params[param.name] = paramMatches.map((paramContent) => {
1561
- if (paramContent.includes(`<${parameterPrefix}`) && param.children) {
1562
- return parseNestedParameters(paramContent, parameterPrefix, param.children);
1563
- }
1564
- return paramContent;
1565
- });
1566
- } else {
1567
- const paramContent = paramMatches[0];
1568
- if (paramContent.includes(`<${parameterPrefix}`) && param.children) {
1569
- params[param.name] = parseNestedParameters(paramContent, parameterPrefix, param.children);
1570
- } else {
1571
- params[param.name] = paramContent;
1572
- }
1573
- }
1574
- }
1575
- }
1576
- }
1577
- results.push({
1578
- type: "tool_use",
1579
- name: toolName,
1580
- params
1581
- });
1582
- } else {
1583
- results.push({
1584
- type: "text",
1585
- content: fullTagContent
1586
- });
1587
- }
1588
- remainingMessage = remainingMessage.slice(match.index + fullTagContent.length);
1589
- }
1590
- if (remainingMessage.trim()) {
1591
- results.push({
1592
- type: "text",
1593
- content: remainingMessage.trim()
1594
- });
1595
- }
1596
- if (results.length === 0) {
1597
- results.push({
1598
- type: "text",
1599
- content: assistantMessage
1600
- });
1601
- }
1602
- return results;
1603
- }
1604
-
1605
- // src/Agent/prompts.ts
1606
- var renderParameterValue = (key, value, parameterPrefix) => {
1607
- if (typeof value === "string") {
1608
- return `<${parameterPrefix}${key}>${value}</${parameterPrefix}${key}>`;
1609
- }
1610
- if (Array.isArray(value)) {
1611
- return value.map((v) => renderParameterValue(key, v, parameterPrefix)).join("\n");
1612
- }
1613
- const inner = Object.entries(value).map(([key2, v]) => renderParameterValue(key2, v, parameterPrefix)).join("\n");
1614
- return `<${parameterPrefix}${key}>
1615
- ${inner}
1616
- </${parameterPrefix}${key}>`;
1617
- };
1618
- var toolInfoPrompt = (tool, toolNamePrefix, parameterPrefix) => `
1619
- ## ${toolNamePrefix}${tool.name}
1620
-
1621
- Description: ${tool.description}
1622
-
1623
- Parameters:
1624
- ${tool.parameters.map(
1625
- (param) => `- ${parameterPrefix}${param.name}: (${param.required ? "required" : "optional"})${param.allowMultiple ? " (multiple allowed)" : ""} ${param.description}`
1626
- ).join("\n")}
1627
-
1628
- Usage:
1629
- <${toolNamePrefix}${tool.name}>
1630
- ${tool.parameters.map(
1631
- (param) => param.allowMultiple ? `<${parameterPrefix}${param.name}>${param.usageValue}</${parameterPrefix}${param.name}>
1632
- <${parameterPrefix}${param.name}>${param.usageValue}</${parameterPrefix}${param.name}>` : `<${parameterPrefix}${param.name}>${param.usageValue}</${parameterPrefix}${param.name}>`
1633
- ).join("\n")}
1634
- </${toolNamePrefix}${tool.name}>`;
1635
- var toolInfoExamplesPrompt = (tool, example, toolNamePrefix, parameterPrefix) => `
1636
- ## Example: ${example.description}
1637
-
1638
- <${toolNamePrefix}${tool.name}>
1639
- ${Object.entries(example.input).map(([name, value]) => renderParameterValue(name, value, parameterPrefix)).join("\n")}
1640
- </${toolNamePrefix}${tool.name}>
1641
- `;
1642
- var toolUsePrompt = (useNativeTool, tools, toolNamePrefix) => {
1643
- if (useNativeTool) {
1644
- return `
1645
- ====
1646
-
1647
- TOOL USE
1648
-
1649
- - You MUST use a tool.
1650
- - Batch tool use when feasible; avoid redundant calls.
1651
- `;
1652
- }
1653
- if (tools.length === 0) {
1654
- return "";
1655
- }
1656
- const parameterPrefix = `${toolNamePrefix}parameter_`;
1657
- const v1Tools = tools.map(toToolInfoV1);
1658
- return `
1659
- ====
1660
-
1661
- TOOL USE
1662
-
1663
- You have access to a set of tools that are executed upon the user's approval. You can use up to 5 tool calls per message, and will receive the results of those tool uses in the user's response. You use tools step-by-step to accomplish a given task, with each tool use informed by the result of the previous tool use.
1664
-
1665
- # Tool Use Formatting
1666
-
1667
- Tool use is formatted using XML-style tags. The tool name is enclosed in opening and closing tags, and each parameter is similarly enclosed within its own set of tags. Here's the structure:
1668
-
1669
- <${toolNamePrefix}tool_name>
1670
- <${parameterPrefix}name1>value1</${parameterPrefix}name1>
1671
- <${parameterPrefix}name2>value2</${parameterPrefix}name2>
1672
- ...
1673
- </${toolNamePrefix}tool_name>
1674
-
1675
- **It is crucial that all tags are correctly nested and closed.**
1676
-
1677
- ## Array Parameters
1678
-
1679
- To create an array of values for a parameter, repeat the parameter tag multiple times:
1680
-
1681
- <${toolNamePrefix}process_file>
1682
- <${parameterPrefix}path>test.ts</${parameterPrefix}path>
1683
- <${parameterPrefix}path>main.ts</${parameterPrefix}path>
1684
- </${toolNamePrefix}process_file>
1685
-
1686
- ## Nested Object Parameters
1687
-
1688
- To create nested objects, nest parameter tags within other parameter tags:
1689
-
1690
- <${toolNamePrefix}example_tool>
1691
- <${parameterPrefix}key>
1692
- <${parameterPrefix}key2>value</${parameterPrefix}key2>
1693
- <${parameterPrefix}key3>value2</${parameterPrefix}key3>
1694
- </${parameterPrefix}key>
1695
- </${toolNamePrefix}example_tool>
1696
-
1697
- You can also combine array parameters with nested objects:
1698
-
1699
- <${toolNamePrefix}example_tool>
1700
- <${parameterPrefix}key>
1701
- <${parameterPrefix}key2>value</${parameterPrefix}key2>
1702
- <${parameterPrefix}key3>value2</${parameterPrefix}key3>
1703
- </${parameterPrefix}key>
1704
- <${parameterPrefix}key>
1705
- <${parameterPrefix}key2>value3</${parameterPrefix}key2>
1706
- <${parameterPrefix}key3>value4</${parameterPrefix}key3>
1707
- <${parameterPrefix}key3>value5</${parameterPrefix}key3>
1708
- </${parameterPrefix}key>
1709
- </${toolNamePrefix}example_tool>
1710
-
1711
- Always adhere to this format, ensuring every opening tag has a matching closing tag, to ensure proper parsing and execution.
1712
-
1713
- NEVER surround tool use with triple backticks (\`\`\`).
1714
-
1715
- # Tools
1716
- ${v1Tools.map((tool) => toolInfoPrompt(tool, toolNamePrefix, parameterPrefix)).join("\n")}
1717
-
1718
- # Tool Use Examples
1719
- ${v1Tools.map((tool) => {
1720
- let promp = "";
1721
- for (const example of tool.examples ?? []) {
1722
- promp += toolInfoExamplesPrompt(tool, example, toolNamePrefix, parameterPrefix);
1723
- }
1724
- return promp;
1725
- }).join("")}
1726
- # Tool Use Guidelines
1727
-
1728
- 1. **Wait for Feedback**
1729
- - After using a tool, wait for the user's response indicating success/failure or any output logs. Do not assume the result of a tool without explicit confirmation.
1730
- 2. **Error Handling**
1731
- - If a tool fails or produces an unexpected result, analyze the error, decide on an alternative approach or tool, and proceed carefully.
1732
- 3. **Avoid Repetition**
1733
- - Do not quote or repeat previous commands or prompts verbatim. Move the conversation forward by focusing on the latest required action.
1734
- 4. **Tool Call Limit**
1735
- - It is **STRIGHTLY FORBIDDEN** to make more than 5 tool calls in a single message.`;
1736
- };
1737
- var agentsPrompt = (agents, name) => `
1738
- ====
1739
-
1740
- AVAILABLE AGENTS
1741
-
1742
- The following agents are available for task handover/delegate:
1743
- ${agents.map(
1744
- (agent) => `
1745
- - **${agent.name}**
1746
- - Responsibilities:
1747
- ${agent.responsibilities.map((resp) => ` - ${resp}`).join("\n")}`
1748
- ).join("\n")}
1749
-
1750
- - **Current Agent Role**
1751
- You are currently acting as **${name}**. If you identify the task is beyond your current scope, use the handover or delegate tool to transition to the other agent. Include sufficient context so the new agent can seamlessly continue the work.
1752
- `;
1753
- var capabilities = (_toolNamePrefix) => `
1754
- ====
1755
-
1756
- CAPABILITIES
1757
-
1758
- - You have access to a range of tools to aid you in your work. These tools help you effectively accomplish a wide range of tasks.
1759
- - When the user initially gives you a task, a recursive list of all filepaths in the current working directory will be included in context. This provides an overview of the project's file structure, offering key insights into the project from directory/file names (how developers conceptualize and organize their code) and file extensions (the language used). This can also guide decision-making on which files to explore further.`;
1760
- var systemInformation = (info) => `
1761
- ====
1762
-
1763
- SYSTEM INFORMATION
1764
-
1765
- Operating System: ${info.os}`;
1766
- var customInstructions = (customInstructions2) => {
1767
- const joined = customInstructions2.join("\n");
1768
- if (joined.trim() === "") {
1769
- return "";
1770
- }
1771
- return `
1772
- ====
1773
-
1774
- USER'S CUSTOM INSTRUCTIONS
1775
-
1776
- The following additional instructions are provided by the user, and should be followed to the best of your ability without interfering with the TOOL USE guidelines.
1777
-
1778
- ${joined}`;
1779
- };
1780
- var customScripts = (commands) => {
1781
- const joined = Object.entries(commands).map(([name, command]) => {
1782
- if (typeof command === "string") {
1783
- return `- ${name}
1784
- - Command: \`${command}\``;
1785
- }
1786
- return `- ${name}
1787
- - Command: \`${command.command}\`
1788
- - Description: ${command.description}`;
1789
- }).join("\n");
1790
- if (joined.trim() === "") {
1791
- return "";
1792
- }
1793
- return `
1794
- ====
1795
-
1796
- USER'S CUSTOM COMMANDS
1797
-
1798
- The following additional commands are provided by the user, and should be followed to the best of your ability without interfering with the TOOL USE guidelines.
1799
-
1800
- ${joined}`;
1801
- };
1802
- var responsePrompts = {
1803
- errorInvokeTool: (tool, error) => `An error occurred while invoking the tool "${tool}": ${error}`,
1804
- requireUseTool: `Error: No tool use detected. You MUST use a tool before proceeding.
1805
- e.g. <tool_tool_name>tool_name</tool_tool_name>
1806
-
1807
- Ensure the opening and closing tags are correctly nested and closed, and that you are using the correct tool name.
1808
- Avoid unnecessary text or symbols before or after the tool use.
1809
- Avoid unnecessary escape characters or special characters.
1810
- `,
1811
- requireUseToolNative: `Error: No tool use detected. You MUST use a tool before proceeding.
1812
- `,
1813
- toolResults: (tool, result) => {
1814
- switch (result.type) {
1815
- case "text":
1816
- return [
1817
- {
1818
- type: "text",
1819
- text: `<tool_response name=${tool}>${result.value}</tool_response>`
1820
- }
1821
- ];
1822
- case "error-text":
1823
- return [
1824
- {
1825
- type: "text",
1826
- text: `<tool_response_error name=${tool}>${result.value}</tool_response_error>`
1827
- }
1828
- ];
1829
- case "json":
1830
- return [
1831
- {
1832
- type: "text",
1833
- text: `<tool_response_json name=${tool}>${JSON.stringify(result.value)}</tool_response_json>`
1834
- }
1835
- ];
1836
- case "error-json":
1837
- return [
1838
- {
1839
- type: "text",
1840
- text: `<tool_response_error_json name=${tool}>${JSON.stringify(result.value)}</tool_response_error_json>`
1841
- }
1842
- ];
1843
- case "content":
1844
- return [
1845
- {
1846
- type: "text",
1847
- text: `<tool_response name=${tool}>`
1848
- },
1849
- ...result.value.map((part) => {
1850
- if (part.type === "text") {
1851
- return part;
1852
- }
1853
- if (part.mediaType.startsWith("image/")) {
1854
- return {
1855
- type: "image",
1856
- mediaType: part.mediaType,
1857
- image: part.data
1858
- };
1859
- }
1860
- return {
1861
- type: "file",
1862
- mediaType: part.mediaType,
1863
- data: part.data
1864
- };
1865
- }),
1866
- {
1867
- type: "text",
1868
- text: "</tool_response>"
1869
- }
1870
- ];
1871
- }
1872
- },
1873
- commandResult: (command, exitCode, stdout, stderr) => `<command>${command}</command>
1874
- <command_exit_code>${exitCode}</command_exit_code>
1875
- <command_stdout>
1876
- ${stdout}
1877
- </command_stdout>
1878
- <command_stderr>
1879
- ${stderr}
1880
- </command_stderr>`
1881
- };
1882
-
1883
- // src/Agent/AgentBase.ts
1884
- var TaskEventKind = /* @__PURE__ */ ((TaskEventKind2) => {
1885
- TaskEventKind2["StartTask"] = "StartTask";
1886
- TaskEventKind2["StartRequest"] = "StartRequest";
1887
- TaskEventKind2["EndRequest"] = "EndRequest";
1888
- TaskEventKind2["Usage"] = "Usage";
1889
- TaskEventKind2["Text"] = "Text";
1890
- TaskEventKind2["Reasoning"] = "Reasoning";
1891
- TaskEventKind2["ToolUse"] = "ToolUse";
1892
- TaskEventKind2["ToolReply"] = "ToolReply";
1893
- TaskEventKind2["ToolInvalid"] = "ToolInvalid";
1894
- TaskEventKind2["ToolError"] = "ToolError";
1895
- TaskEventKind2["ToolInterrupted"] = "ToolInterrupted";
1896
- TaskEventKind2["ToolHandOver"] = "ToolHandOver";
1897
- TaskEventKind2["ToolDelegate"] = "ToolDelegate";
1898
- TaskEventKind2["ToolPause"] = "ToolPause";
1899
- TaskEventKind2["UsageExceeded"] = "UsageExceeded";
1900
- TaskEventKind2["EndTask"] = "EndTask";
1901
- return TaskEventKind2;
1902
- })(TaskEventKind || {});
1903
- var AgentBase = class {
1904
- ai;
1905
- config;
1906
- handlers;
1907
- #policies;
1908
- #toolSet;
1909
- #messages = [];
1910
- #aborted = false;
1911
- #abortController;
1912
- constructor(name, ai, config) {
1913
- this.ai = ai;
1914
- if (config.agents && config.agents.length > 0) {
1915
- const agents = agentsPrompt(config.agents, name);
1916
- config.systemPrompt += `
1917
- ${agents}`;
1918
- }
1919
- const handlers = {};
1920
- for (const tool of config.tools) {
1921
- handlers[tool.name] = tool;
1922
- }
1923
- const policies = [];
1924
- for (const policy of config.policies) {
1925
- const instance = policy(handlers, config.parameters);
1926
- if (instance) {
1927
- policies.push(instance);
1928
- if (instance.prompt) {
1929
- config.systemPrompt += `
1930
- ${instance.prompt}`;
1931
- }
1932
- if (instance.tools) {
1933
- for (const tool of instance.tools) {
1934
- handlers[tool.name] = tool;
1935
- }
1936
- }
1937
- }
1938
- }
1939
- this.handlers = handlers;
1940
- this.config = config;
1941
- this.#policies = policies;
1942
- this.#messages.push({
1943
- role: "system",
1944
- content: this.config.systemPrompt
1945
- });
1946
- if (this.config.toolFormat === "native") {
1947
- const tools = {};
1948
- for (const tool of Object.values(this.handlers)) {
1949
- const toolName = camelCase(tool.name);
1950
- tools[toolName] = {
1951
- description: tool.description,
1952
- inputSchema: jsonSchema(toJSONSchema(tool.parameters))
1953
- };
1954
- this.handlers[toolName] = tool;
1955
- }
1956
- this.#toolSet = tools;
1957
- } else {
1958
- this.#toolSet = {};
1959
- }
1960
- }
1961
- abort() {
1962
- this.#aborted = true;
1963
- if (this.#abortController) {
1964
- this.#abortController.abort();
1965
- }
1966
- }
1967
- get parameters() {
1968
- return this.config.parameters;
1969
- }
1970
- get messages() {
1971
- return this.#messages;
1972
- }
1973
- setMessages(messages) {
1974
- this.#messages = [...messages];
1975
- }
1976
- async #callback(event) {
1977
- await this.config.callback?.(event);
1978
- }
1979
- async start(prompt) {
1980
- this.#callback({ kind: "StartTask" /* StartTask */, systemPrompt: this.config.systemPrompt });
1981
- return await this.#processLoop(prompt);
1982
- }
1983
- async step(message) {
1984
- if (this.#messages.length === 0) {
1985
- this.#callback({ kind: "StartTask" /* StartTask */, systemPrompt: this.config.systemPrompt });
1986
- }
1987
- return await this.#request(message);
1988
- }
1989
- async handleStepResponse(response) {
1990
- return this.#handleResponse(response);
1991
- }
1992
- async #processLoop(userMessage) {
1993
- let nextRequest = [
1994
- {
1995
- role: "user",
1996
- content: userMessage
1997
- }
1998
- ];
1999
- while (true) {
2000
- if (this.#aborted) {
2001
- return { type: "Aborted" };
2002
- }
2003
- if (this.config.usageMeter.isLimitExceeded().result) {
2004
- this.#callback({ kind: "UsageExceeded" /* UsageExceeded */ });
2005
- return { type: "UsageExceeded" };
2006
- }
2007
- const response = await this.#request(nextRequest);
2008
- if (this.#aborted) {
2009
- return { type: "Aborted" };
2010
- }
2011
- const resp = await this.#handleResponse(response);
2012
- if (resp.type === "exit") {
2013
- this.#callback({ kind: "EndTask" /* EndTask */, exitReason: resp.reason });
2014
- return resp.reason;
2015
- }
2016
- nextRequest = resp.message;
2017
- }
2018
- }
2019
- async continueTask(userMessage) {
2020
- return await this.#processLoop(userMessage);
2021
- }
2022
- async #request(userMessage) {
2023
- if (!userMessage) {
2024
- throw new Error("userMessage is missing");
2025
- }
2026
- await this.#callback({ kind: "StartRequest" /* StartRequest */, userMessage });
2027
- this.#messages.push(...userMessage);
2028
- for (const instance of this.#policies) {
2029
- if (instance.onBeforeRequest) {
2030
- await instance.onBeforeRequest(this);
2031
- }
2032
- }
2033
- let messages = this.#messages;
2034
- for (const instance of this.#policies) {
2035
- if (instance.prepareMessages) {
2036
- messages = await instance.prepareMessages(this, messages);
2037
- }
2038
- }
2039
- const retryCount = this.config.retryCount ?? 5;
2040
- const requestTimeoutSeconds = this.config.requestTimeoutSeconds ?? 90;
2041
- let respMessages = [];
2042
- let rateLimitErrorCount = 0;
2043
- for (let i = 0; i < retryCount; i++) {
2044
- if (this.#aborted) {
2045
- break;
2046
- }
2047
- respMessages = [];
2048
- let timeout;
2049
- let requestAbortController;
2050
- requestAbortController = new AbortController();
2051
- this.#abortController = requestAbortController;
2052
- const resetTimeout = () => {
2053
- if (timeout) {
2054
- clearTimeout(timeout);
2055
- }
2056
- if (requestTimeoutSeconds > 0 && requestAbortController) {
2057
- timeout = setTimeout(() => {
2058
- console.error(
2059
- `
2060
- Request timeout after ${requestTimeoutSeconds} seconds. Canceling current request attempt ${i + 1}/${retryCount}.`
2061
- );
2062
- requestAbortController?.abort();
2063
- }, requestTimeoutSeconds * 1e3);
2064
- }
2065
- };
2066
- try {
2067
- resetTimeout();
2068
- const usageMeterOnFinishHandler = this.config.usageMeter.onFinishHandler(this.ai);
2069
- const streamTextOptions = {
2070
- model: this.ai,
2071
- temperature: 0,
2072
- messages,
2073
- providerOptions: this.config.parameters?.providerOptions,
2074
- onChunk: async ({ chunk }) => {
2075
- resetTimeout();
2076
- switch (chunk.type) {
2077
- case "text-delta":
2078
- await this.#callback({ kind: "Text" /* Text */, newText: chunk.text });
2079
- break;
2080
- case "reasoning-delta":
2081
- await this.#callback({ kind: "Reasoning" /* Reasoning */, newText: chunk.text });
2082
- break;
2083
- case "tool-call":
2084
- break;
2085
- }
2086
- },
2087
- onFinish: (evt) => {
2088
- usageMeterOnFinishHandler(evt);
2089
- this.#callback({ kind: "Usage" /* Usage */, usage: evt.totalUsage });
2090
- },
2091
- onError: async (error) => {
2092
- console.error("Error in stream:", error);
2093
- },
2094
- abortSignal: requestAbortController.signal
2095
- };
2096
- if (this.config.toolFormat === "native") {
2097
- streamTextOptions.tools = this.#toolSet;
2098
- }
2099
- const stream = streamText(streamTextOptions);
2100
- await stream.consumeStream({
2101
- onError: (error) => {
2102
- console.error("Error in stream:", error);
2103
- }
2104
- });
2105
- const resp = await stream.response;
2106
- respMessages = resp.messages;
2107
- rateLimitErrorCount = 0;
2108
- if (timeout) {
2109
- clearTimeout(timeout);
2110
- timeout = void 0;
2111
- }
2112
- } catch (error) {
2113
- if (error instanceof Error && error.name === "AbortError") {
2114
- if (this.#aborted) {
2115
- break;
2116
- }
2117
- console.error(`Request attempt ${i + 1} timed out, will retry`);
2118
- } else if (error?.error?.error?.code === "rate_limit_exceeded" || error?.error?.code === "rate_limit_exceeded" || error?.code === "rate_limit_exceeded" || error?.status === 429 || error?.error?.status === 429) {
2119
- rateLimitErrorCount++;
2120
- const waitSeconds = computeRateLimitBackoffSeconds(rateLimitErrorCount);
2121
- console.error(`Rate limit exceeded. Waiting ${waitSeconds}s before retrying...`);
2122
- if (timeout) {
2123
- clearTimeout(timeout);
2124
- timeout = void 0;
2125
- }
2126
- await new Promise((resolve) => setTimeout(resolve, waitSeconds * 1e3));
2127
- console.error("Retrying request...");
2128
- } else {
2129
- rateLimitErrorCount = 0;
2130
- console.error("Error in stream:", error);
2131
- }
2132
- } finally {
2133
- if (timeout) {
2134
- clearTimeout(timeout);
2135
- }
2136
- }
2137
- if (this.config.toolFormat === "native") {
2138
- if (respMessages.some((m) => {
2139
- if (m.role === "tool") {
2140
- return true;
2141
- }
2142
- if (typeof m.content === "string") {
2143
- return true;
2144
- }
2145
- return m.content.some((part) => part.type === "tool-call" || part.type === "tool-result" || part.type === "text");
2146
- })) {
2147
- break;
2148
- }
2149
- } else {
2150
- if (respMessages.length > 0) {
2151
- break;
2152
- }
2153
- }
2154
- if (this.#aborted) {
2155
- break;
2156
- }
2157
- if (i < retryCount - 1) {
2158
- console.error(`
2159
- Retrying request ${i + 2} of ${retryCount}`);
2160
- }
2161
- }
2162
- if (respMessages.length === 0) {
2163
- if (this.#aborted) {
2164
- return [];
2165
- }
2166
- throw new Error("No assistant message received after all retry attempts");
2167
- }
2168
- this.#messages.push(...respMessages);
2169
- if (this.config.toolFormat === "native") {
2170
- const assistantText = respMessages.map((msg) => {
2171
- if (typeof msg.content === "string") {
2172
- return msg.content;
2173
- }
2174
- return msg.content.map((part) => part.type === "text" || part.type === "reasoning" ? part.text : "").join("");
2175
- }).join("\n");
2176
- await this.#callback({ kind: "EndRequest" /* EndRequest */, message: assistantText });
2177
- return respMessages.flatMap((msg) => {
2178
- if (msg.role === "assistant") {
2179
- const content = msg.content;
2180
- if (typeof content === "string") {
2181
- return [{ type: "text", content }];
2182
- }
2183
- return content.flatMap((part) => {
2184
- if (part.type === "text" || part.type === "reasoning") {
2185
- return [{ type: "text", content: part.text }];
2186
- }
2187
- if (part.type === "tool-call") {
2188
- return [{ type: "tool_use", id: part.toolCallId, name: part.toolName, params: part.input }];
2189
- }
2190
- return [];
2191
- });
2192
- }
2193
- return [];
2194
- });
2195
- }
2196
- const currentAssistantMessage = respMessages.map((msg) => {
2197
- if (typeof msg.content === "string") {
2198
- return msg.content;
2199
- }
2200
- return msg.content.map((part) => {
2201
- if (part.type === "text") {
2202
- return part.text;
2203
- }
2204
- return "";
2205
- });
2206
- }).join("\n");
2207
- const ret = parseAssistantMessage(currentAssistantMessage, this.config.tools.map(toToolInfoV1), this.config.toolNamePrefix);
2208
- await this.#callback({ kind: "EndRequest" /* EndRequest */, message: currentAssistantMessage });
2209
- return ret;
2210
- }
2211
- async #handleResponse(response) {
2212
- const toolResponses = [];
2213
- const medias = [];
2214
- const processResponse = (resp) => {
2215
- if (resp.type === "content") {
2216
- return {
2217
- type: "content",
2218
- value: resp.value.map((part) => {
2219
- if (part.type === "media") {
2220
- medias.push(part);
2221
- return {
2222
- type: "text",
2223
- text: `<media url="${part.url}" media-type="${part.mediaType}" />`
2224
- };
2225
- }
2226
- return part;
2227
- })
2228
- };
2229
- }
2230
- return resp;
2231
- };
2232
- let hasPause = false;
2233
- const toolUseContents = response.filter((c) => c.type === "tool_use");
2234
- outer: for (const content of response) {
2235
- switch (content.type) {
2236
- case "text":
2237
- break;
2238
- case "tool_use": {
2239
- await this.#callback({ kind: "ToolUse" /* ToolUse */, tool: content.name, params: content.params });
2240
- const toolResp = await this.#invokeTool(content.name, content.params);
2241
- switch (toolResp.type) {
2242
- case "Reply" /* Reply */: {
2243
- await this.#callback({ kind: "ToolReply" /* ToolReply */, tool: content.name, content: toolResp.message });
2244
- toolResponses.push({ type: "response", tool: content.name, response: processResponse(toolResp.message), id: content.id });
2245
- break;
2246
- }
2247
- case "Exit" /* Exit */:
2248
- case "HandOver" /* HandOver */:
2249
- case "Delegate" /* Delegate */: {
2250
- if (this.config.toolFormat === "native" && toolUseContents.length > 1) {
2251
- const message = {
2252
- type: "Error" /* Error */,
2253
- message: {
2254
- type: "error-text",
2255
- value: `Error: The tool '${content.name}' must be called alone, but it was called with other tools.`
2256
- },
2257
- canRetry: false
2258
- };
2259
- await this.#callback({ kind: "ToolError" /* ToolError */, tool: content.name, error: message });
2260
- toolResponses.push({
2261
- type: "response",
2262
- tool: content.name,
2263
- response: processResponse(message.message),
2264
- id: content.id
2265
- });
2266
- break;
2267
- }
2268
- if (toolResponses.length > 0) {
2269
- break outer;
2270
- }
2271
- if (toolResp.type === "Exit" /* Exit */) {
2272
- return { type: "exit", reason: toolResp };
2273
- }
2274
- if (toolResp.type === "HandOver" /* HandOver */) {
2275
- await this.#callback({
2276
- kind: "ToolHandOver" /* ToolHandOver */,
2277
- tool: content.name,
2278
- agentName: toolResp.agentName,
2279
- task: toolResp.task,
2280
- context: toolResp.context,
2281
- files: toolResp.files
2282
- });
2283
- return { type: "exit", reason: toolResp };
2284
- }
2285
- if (toolResp.type === "Delegate" /* Delegate */) {
2286
- await this.#callback({
2287
- kind: "ToolDelegate" /* ToolDelegate */,
2288
- tool: content.name,
2289
- agentName: toolResp.agentName,
2290
- task: toolResp.task,
2291
- context: toolResp.context,
2292
- files: toolResp.files
2293
- });
2294
- return { type: "exit", reason: toolResp };
2295
- }
2296
- break;
2297
- }
2298
- case "Invalid" /* Invalid */: {
2299
- await this.#callback({ kind: "ToolInvalid" /* ToolInvalid */, tool: content.name, content: toolResp.message });
2300
- toolResponses.push({ type: "response", tool: content.name, response: processResponse(toolResp.message), id: content.id });
2301
- if (this.config.toolFormat !== "native") {
2302
- break outer;
2303
- }
2304
- break;
2305
- }
2306
- case "Error" /* Error */: {
2307
- await this.#callback({ kind: "ToolError" /* ToolError */, tool: content.name, error: toolResp.message });
2308
- toolResponses.push({ type: "response", tool: content.name, response: processResponse(toolResp.message), id: content.id });
2309
- if (this.config.toolFormat !== "native") {
2310
- break outer;
2311
- }
2312
- break;
2313
- }
2314
- case "Pause" /* Pause */: {
2315
- await this.#callback({ kind: "ToolPause" /* ToolPause */, tool: content.name, object: toolResp.object });
2316
- toolResponses.push({ type: "pause", tool: content.name, object: toolResp.object, id: content.id });
2317
- hasPause = true;
2318
- }
2319
- }
2320
- break;
2321
- }
2322
- }
2323
- }
2324
- if (hasPause) {
2325
- return { type: "exit", reason: { type: "Pause", responses: toolResponses } };
2326
- }
2327
- if (toolResponses.length === 0) {
2328
- if (this.config.requireToolUse) {
2329
- return {
2330
- type: "reply",
2331
- message: [
2332
- {
2333
- role: "user",
2334
- content: responsePrompts.requireUseToolNative
2335
- }
2336
- ]
2337
- };
2338
- }
2339
- return {
2340
- type: "exit",
2341
- reason: {
2342
- type: "Exit" /* Exit */,
2343
- message: response.filter((c) => c.type === "text").map((c) => c.content).join("")
2344
- }
2345
- };
2346
- }
2347
- const mediaUserMessage = medias.length > 0 ? [
2348
- {
2349
- role: "user",
2350
- content: medias.map((m) => {
2351
- if (m.mediaType.startsWith("image/")) {
2352
- return {
2353
- type: "image",
2354
- image: m.base64Data,
2355
- mediaType: m.mediaType
2356
- };
2357
- }
2358
- return {
2359
- type: "file",
2360
- data: m.base64Data,
2361
- mediaType: m.mediaType,
2362
- filename: m.url.split("/").pop()
2363
- };
2364
- })
2365
- }
2366
- ] : [];
2367
- if (this.config.toolFormat === "native") {
2368
- const toolResults = toolResponses.filter((resp) => resp.type === "response").map(
2369
- (resp) => ({
2370
- type: "tool-result",
2371
- toolCallId: resp.id,
2372
- toolName: resp.tool,
2373
- output: resp.response
2374
- })
2375
- );
2376
- return {
2377
- type: "reply",
2378
- message: [
2379
- {
2380
- role: "tool",
2381
- content: toolResults
2382
- },
2383
- ...mediaUserMessage
2384
- ]
2385
- };
2386
- }
2387
- if (toolResponses.length === 0) {
2388
- if (this.config.requireToolUse) {
2389
- return {
2390
- type: "reply",
2391
- message: [
2392
- {
2393
- role: "user",
2394
- content: responsePrompts.requireUseTool
2395
- }
2396
- ]
2397
- };
2398
- }
2399
- return {
2400
- type: "exit",
2401
- reason: {
2402
- type: "Exit" /* Exit */,
2403
- message: response.filter((c) => c.type === "text").map((c) => c.content).join("")
2404
- }
2405
- };
2406
- }
2407
- const finalResp = toolResponses.filter((resp) => resp.type === "response").flatMap(({ tool, response: response2 }) => responsePrompts.toolResults(tool, response2));
2408
- return {
2409
- type: "reply",
2410
- message: [
2411
- {
2412
- role: "user",
2413
- content: finalResp
2414
- },
2415
- ...mediaUserMessage
2416
- ]
2417
- };
2418
- }
2419
- async #invokeTool(name, args) {
2420
- try {
2421
- const handler15 = this.handlers[name]?.handler;
2422
- if (!handler15) {
2423
- return {
2424
- type: "Error" /* Error */,
2425
- message: {
2426
- type: "error-text",
2427
- value: responsePrompts.errorInvokeTool(name, "Tool not found")
2428
- },
2429
- canRetry: false
2430
- };
2431
- }
2432
- for (const instance of this.#policies) {
2433
- if (instance.onBeforeInvokeTool) {
2434
- const resp2 = await instance.onBeforeInvokeTool(name, args);
2435
- if (resp2) {
2436
- return resp2;
2437
- }
2438
- }
2439
- }
2440
- const resp = await this.onBeforeInvokeTool(this.handlers[name].name, args);
2441
- if (resp) {
2442
- return resp;
2443
- }
2444
- return await handler15(this.config.provider, args);
2445
- } catch (error) {
2446
- return {
2447
- type: "Error" /* Error */,
2448
- message: {
2449
- type: "error-text",
2450
- value: responsePrompts.errorInvokeTool(name, error)
2451
- },
2452
- canRetry: false
2453
- };
2454
- }
2455
- }
2456
- get model() {
2457
- return `${this.ai.provider}:${this.ai.modelId}`;
2458
- }
2459
- get usage() {
2460
- return this.config.usageMeter.usage;
2461
- }
2462
- };
2463
-
2464
- // src/Agent/AnalyzerAgent/prompts.ts
2465
- var fullSystemPrompt = (info, tools, toolNamePrefix, instructions, scripts, useNativeTool) => `
2466
- # Analyzer Agent
2467
-
2468
- ## Role
2469
- You are the **Analyzer** agent, responsible for:
2470
- 1. **Project Structure Analysis** \u2013 Understand codebase organization and architecture.
2471
- 2. **Code Pattern Analysis** \u2013 Identify common patterns, conventions, and best practices.
2472
- 3. **Dependency Analysis** \u2013 Examine project dependencies and their usage.
2473
- 4. **Workflow Analysis** \u2013 Understand development tools, scripts, and processes.
2474
- 5. **Documentation Review** \u2013 Analyze documentation and code comments.
2475
-
2476
- > **Note**: The **Analyzer** agent focuses on understanding and analyzing the codebase without making modifications. Your role is to provide insights and understanding that can inform development decisions.
2477
-
2478
- ## Rules
2479
- 1. **Thoroughness**: Conduct comprehensive analysis of relevant project aspects.
2480
- 2. **Pattern Recognition**: Identify recurring patterns, conventions, and architectural decisions.
2481
- 3. **Dependency Mapping**: Track and understand relationships between components.
2482
- 4. **Workflow Understanding**: Analyze build processes, testing approaches, and development tools.
2483
- 5. **Documentation Assessment**: Review documentation quality and completeness.
2484
- 6. **Non-Modification**: Never modify code or files - focus solely on analysis.
2485
-
2486
- ${toolUsePrompt(useNativeTool, tools, toolNamePrefix)}
2487
- ${capabilities(toolNamePrefix)}
2488
- ${systemInformation(info)}
2489
- ${customInstructions(instructions)}
2490
- ${customScripts(scripts)}
2491
- `;
2492
-
2493
- // src/Agent/AnalyzerAgent/index.ts
2494
- var agentTools = [attemptCompletion_default, delegate_default, fetchUrl_default, handOver_default, listFiles_default, readBinaryFile_default, readFile_default, searchFiles_default];
2495
- var AnalyzerAgent = class extends AgentBase {
2496
- constructor(options) {
2497
- const combinedTools = [...options.additionalTools ?? [], ...agentTools];
2498
- const tools = getAvailableTools({
2499
- provider: options.provider,
2500
- allTools: combinedTools,
2501
- hasAgent: (options.agents?.length ?? 0) > 0
2502
- });
2503
- const toolNamePrefix = options.toolFormat === "native" ? "" : "tool_";
2504
- const systemPrompt = fullSystemPrompt(
2505
- {
2506
- os: options.os
2507
- },
2508
- tools,
2509
- toolNamePrefix,
2510
- options.customInstructions ?? [],
2511
- options.scripts ?? {},
2512
- options.toolFormat === "native"
2513
- );
2514
- super(analyzerAgentInfo.name, options.ai, {
2515
- systemPrompt,
2516
- tools,
2517
- toolNamePrefix,
2518
- provider: options.provider,
2519
- agents: options.agents,
2520
- scripts: options.scripts,
2521
- callback: options.callback,
2522
- policies: options.policies,
2523
- toolFormat: options.toolFormat,
2524
- parameters: options.parameters ?? {},
2525
- usageMeter: options.usageMeter ?? new UsageMeter(),
2526
- requestTimeoutSeconds: options.requestTimeoutSeconds,
2527
- requireToolUse: options.requireToolUse ?? true
2528
- });
2529
- }
2530
- onBeforeInvokeTool() {
2531
- return Promise.resolve(void 0);
2532
- }
2533
- };
2534
- var analyzerAgentInfo = {
2535
- name: "analyzer",
2536
- responsibilities: [
2537
- "Analyzing project structure and organization",
2538
- "Identifying key source code files and their relationships",
2539
- "Understanding common coding patterns and conventions",
2540
- "Examining development workflow and tooling",
2541
- "Analyzing dependencies and their usage patterns"
2542
- ]
2543
- };
2544
-
2545
- // src/Agent/ArchitectAgent/prompts.ts
2546
- var fullSystemPrompt2 = (info, tools, toolNamePrefix, instructions, scripts, useNativeTool) => `
2547
- # Architect Agent
2548
-
2549
- ## Role
2550
- You are the **Architect** agent, responsible for:
2551
- 1. **Task Analysis** - Understand requirements.
2552
- 2. **File Identification** - Select relevant files.
2553
- 3. **File Reading** - Use provided tools to gather information.
2554
- 4. **Implementation Plan** - Draft a *code-free* plan (pseudocode/interface stubs permitted).
2555
- 5. **Review & Improve** - Refine the plan.
2556
- 6. **Handover/Delegate** - Provide the final plan, context, and files to the **Coder** agent.
2557
-
2558
- > **Never** modify project files directly. Only produce the implementation plan; the **Coder** agent performs all code changes.
2559
-
2560
- ## Rules
2561
- 1. **Consistency** - Align with user objectives.
2562
- 2. **Relevance** - Consult only essential files.
2563
- 3. **Conciseness** - Be succinct; avoid repetition.
2564
- 4. **Accuracy** - Ensure conclusions are verifiable.
2565
- 5. **Clarity** - Present information in a structured format.
2566
- 6. **Minimal Queries** - Ask questions only when truly needed.
2567
- 7. **Completion** - Only use the \`attemptCompletion\` tool if the user's request has been fully satisfied and no coding tasks need to be delegated to the Coder agent.
2568
-
2569
- ## Steps
2570
- 1. **Analyze Task** - Capture goals, constraints, and success criteria.
2571
- 2. **Identify Relevant Files** - List files and justify each choice.
2572
- 3. **Read Files via Tools** - Summarize key insights.
2573
- 4. **Create Implementation Plan** - Provide a detailed, step-by-step breakdown:
2574
- * Tasks, resources, and dependencies.
2575
- * Pseudocode or interface declarations only\u2014no concrete code.
2576
- * Sufficient detail for the **Coder** to implement without follow-ups.
2577
- 5. **Review & Improve** - Confirm the plan is unambiguous and complete.
2578
- 6. **Handover/Delegate**
2579
- - If the plan consists of a single self-contained step, hand it over as one task.
2580
- - If multiple steps are required, break them into numbered tasks to delegate to the **Coder** agent.
2581
- - When handing over or delegating, you MUST provide the full implementation plan. Include all necessary context, file references, and clarifications for successful execution.
2582
-
2583
- ${toolUsePrompt(useNativeTool, tools, toolNamePrefix)}
2584
- ${capabilities(toolNamePrefix)}
2585
- ${systemInformation(info)}
2586
- ${customInstructions(instructions)}
2587
- ${customScripts(scripts)}
2588
- `;
2589
-
2590
- // src/Agent/ArchitectAgent/index.ts
2591
- var agentTools2 = [attemptCompletion_default, delegate_default, fetchUrl_default, handOver_default, listFiles_default, readBinaryFile_default, readFile_default, searchFiles_default];
2592
- var ArchitectAgent = class extends AgentBase {
2593
- constructor(options) {
2594
- const combinedTools = [...options.additionalTools ?? [], ...agentTools2];
2595
- const tools = getAvailableTools({
2596
- provider: options.provider,
2597
- allTools: combinedTools,
2598
- hasAgent: (options.agents?.length ?? 0) > 0
2599
- });
2600
- const toolNamePrefix = options.toolFormat === "native" ? "" : "tool_";
2601
- const systemPrompt = fullSystemPrompt2(
2602
- {
2603
- os: options.os
2604
- },
2605
- tools,
2606
- toolNamePrefix,
2607
- options.customInstructions ?? [],
2608
- options.scripts ?? {},
2609
- options.toolFormat === "native"
2610
- );
2611
- super(architectAgentInfo.name, options.ai, {
2612
- systemPrompt,
2613
- tools,
2614
- toolNamePrefix,
2615
- provider: options.provider,
2616
- agents: options.agents,
2617
- scripts: options.scripts,
2618
- callback: options.callback,
2619
- policies: options.policies,
2620
- toolFormat: options.toolFormat,
2621
- parameters: options.parameters ?? {},
2622
- usageMeter: options.usageMeter ?? new UsageMeter(),
2623
- requestTimeoutSeconds: options.requestTimeoutSeconds,
2624
- requireToolUse: options.requireToolUse ?? true
2625
- });
2626
- }
2627
- onBeforeInvokeTool() {
2628
- return Promise.resolve(void 0);
2629
- }
2630
- };
2631
- var architectAgentInfo = {
2632
- name: "architect",
2633
- responsibilities: [
2634
- "Analyzing the user\u2019s overall task and requirements.",
2635
- "Creating plans and making higher-level decisions about system structure and design.",
2636
- "Reviewing and analyzing existing code or components for maintainability and scalability.",
2637
- "Laying out the roadmap for implementation."
2638
- ]
2639
- };
2640
-
2641
- // src/Agent/CodeFixerAgent/prompts.ts
2642
- var basePrompt = `You are a highly skilled software engineer specializing in debugging and fixing code issues. You have extensive experience with:
2643
- - Type systems and type checking
2644
- - Test frameworks and debugging test failures
2645
- - Code quality tools and best practices
2646
- - Systematic debugging approaches`;
2647
- var codeFixingStrategies = `
2648
- ====
2649
-
2650
- CODE FIXING STRATEGIES
2651
-
2652
- 1. Type Errors
2653
- - Analyze type error messages carefully
2654
- - Check type definitions and imports
2655
- - Consider type assertions only as a last resort
2656
- - Verify type compatibility across function boundaries
2657
- - Look for null/undefined handling issues
2658
-
2659
- 2. Test Failures
2660
- - Examine test output and error messages
2661
- - Check test setup and fixtures
2662
- - Verify assertions and expectations
2663
- - Look for async/timing issues
2664
- - Consider edge cases and input validation
2665
-
2666
- 3. Code Quality Issues
2667
- - Follow project's coding standards
2668
- - Address linter warnings systematically
2669
- - Improve code readability
2670
- - Fix potential runtime issues
2671
- - Consider performance implications
2672
-
2673
- 4. General Approach
2674
- - Start with the most critical issues
2675
- - Make minimal necessary changes
2676
- - Verify fixes don't introduce new issues
2677
- - Document complex fixes with comments
2678
- - Track attempted solutions for each issue`;
2679
- var retryGuidelines = `
2680
- ====
2681
-
2682
- RETRY GUIDELINES
2683
-
2684
- 1. Before Retrying
2685
- - Analyze previous attempt's failure
2686
- - Consider alternative approaches
2687
- - Check if similar issues were fixed
2688
- - Verify no new issues were introduced
2689
-
2690
- 2. When to Retry
2691
- - Error message changed but issue persists
2692
- - New information available about the root cause
2693
- - Different fixing strategy available
2694
- - Previous attempt partially successful
2695
-
2696
- 3. When to Stop
2697
- - Maximum retry limit reached
2698
- - Same error occurs repeatedly
2699
- - Fix would require major refactoring
2700
- - Issue requires human intervention
2701
-
2702
- 4. After Maximum Retries
2703
- - Document attempted solutions
2704
- - Explain why the issue remains
2705
- - Suggest manual intervention steps
2706
- - Report any partial improvements`;
2707
- var fullSystemPrompt3 = (info, tools, toolNamePrefix, instructions, scripts, useNativeTool) => `
2708
- ${basePrompt}
2709
- ${toolUsePrompt(useNativeTool, tools, toolNamePrefix)}
2710
- ${codeFixingStrategies}
2711
- ${retryGuidelines}
2712
- ${capabilities(toolNamePrefix)}
2713
- ${systemInformation(info)}
2714
- ${customInstructions(instructions)}
2715
- ${customScripts(scripts)}`;
2716
-
2717
- // src/Agent/CodeFixerAgent/index.ts
2718
- var agentTools3 = [
2719
- attemptCompletion_default,
2720
- delegate_default,
2721
- executeCommand_default,
2722
- fetchUrl_default,
2723
- handOver_default,
2724
- listFiles_default,
2725
- readBinaryFile_default,
2726
- readFile_default,
2727
- removeFile_default,
2728
- renameFile_default,
2729
- replaceInFile_default,
2730
- searchFiles_default,
2731
- writeToFile_default
2732
- ];
2733
- var CodeFixerAgent = class extends AgentBase {
2734
- #maxRetries;
2735
- #retryCount = 0;
2736
- constructor(options) {
2737
- const combinedTools = [...options.additionalTools ?? [], ...agentTools3];
2738
- const tools = getAvailableTools({
2739
- provider: options.provider,
2740
- allTools: combinedTools,
2741
- hasAgent: (options.agents?.length ?? 0) > 0
2742
- });
2743
- const toolNamePrefix = options.toolFormat === "native" ? "" : "tool_";
2744
- const systemPrompt = fullSystemPrompt3(
2745
- {
2746
- os: options.os
2747
- },
2748
- tools,
2749
- toolNamePrefix,
2750
- options.customInstructions ?? [],
2751
- options.scripts ?? {},
2752
- options.toolFormat === "native"
2753
- );
2754
- super(codeFixerAgentInfo.name, options.ai, {
2755
- systemPrompt,
2756
- tools,
2757
- toolNamePrefix,
2758
- provider: options.provider,
2759
- agents: options.agents,
2760
- scripts: options.scripts,
2761
- callback: options.callback,
2762
- policies: options.policies,
2763
- toolFormat: options.toolFormat,
2764
- parameters: options.parameters ?? {},
2765
- usageMeter: options.usageMeter ?? new UsageMeter(),
2766
- requestTimeoutSeconds: options.requestTimeoutSeconds,
2767
- requireToolUse: options.requireToolUse ?? true
2768
- });
2769
- this.#maxRetries = options.maxRetries ?? 5;
2770
- }
2771
- async onBeforeInvokeTool(name, _args) {
2772
- if (name === attemptCompletion_default.name) {
2773
- if (this.#retryCount > this.#maxRetries) {
2774
- return;
2775
- }
2776
- this.#retryCount++;
2777
- const executeCommand = this.config.provider.executeCommand;
2778
- if (!executeCommand) {
2779
- return;
2780
- }
2781
- const check = this.config.scripts?.check;
2782
- const checkCommand = typeof check === "string" ? check : check?.command;
2783
- if (checkCommand) {
2784
- try {
2785
- const { exitCode, stdout, stderr } = await executeCommand(checkCommand, false);
2786
- if (exitCode !== 0) {
2787
- return {
2788
- type: "Reply" /* Reply */,
2789
- message: {
2790
- type: "error-text",
2791
- value: responsePrompts.commandResult(checkCommand, exitCode, stdout, stderr)
2792
- }
2793
- };
2794
- }
2795
- } catch (error) {
2796
- console.warn(`Failed to check code using command: ${checkCommand}`, error);
2797
- }
2798
- }
2799
- const test = this.config.scripts?.test;
2800
- const testCommand = typeof test === "string" ? test : test?.command;
2801
- if (testCommand) {
2802
- try {
2803
- const { exitCode, stdout, stderr } = await executeCommand(testCommand, false);
2804
- if (exitCode !== 0) {
2805
- return {
2806
- type: "Reply" /* Reply */,
2807
- message: {
2808
- type: "error-text",
2809
- value: responsePrompts.commandResult(testCommand, exitCode, stdout, stderr)
2810
- }
2811
- };
2812
- }
2813
- } catch (error) {
2814
- console.warn(`Failed to test code using command: ${testCommand}`, error);
2815
- }
2816
- }
2817
- }
2818
- }
2819
- };
2820
- var codeFixerAgentInfo = {
2821
- name: "codefixer",
2822
- responsibilities: [
2823
- "Fixing type errors and type-related issues",
2824
- "Resolving failing tests",
2825
- "Addressing code quality issues",
2826
- "Tracking and reporting unfixed issues"
2827
- ]
2828
- };
2829
-
2830
- // src/Agent/CoderAgent/prompts.ts
2831
- var basePrompt2 = "You are a highly skilled software engineer with extensive knowledge in many programming languages, frameworks, design patterns, and best practices.";
2832
- var editingFilesPrompt = (toolNamePrefix) => `
2833
- ====
2834
-
2835
- EDITING FILES
2836
-
2837
- You have two file-manipulation tools: **${toolNamePrefix}write_to_file** (full overwrite) and **${toolNamePrefix}replace_in_file** (targeted anchor-based edits). Choose the smallest safe operation for every change.
2838
-
2839
- # ${toolNamePrefix}write_to_file
2840
-
2841
- ## Purpose
2842
-
2843
- - Create a new file, or overwrite the entire contents of an existing file.
2844
-
2845
- ## When to Use
2846
-
2847
- - Initial file creation, such as when scaffolding a new project.
2848
- - Overwriting large boilerplate files where you want to replace the entire content at once.
2849
- - When the complexity or number of changes would make ${toolNamePrefix}replace_in_file unwieldy or error-prone.
2850
- - When you need to completely restructure a file's content or change its fundamental organization.
2851
-
2852
- ## Important Considerations
2853
-
2854
- - Using ${toolNamePrefix}write_to_file requires providing the file's complete final content.
2855
- - If you only need to make small changes to an existing file, consider using ${toolNamePrefix}replace_in_file instead to avoid unnecessarily rewriting the entire file.
2856
- - While ${toolNamePrefix}write_to_file should not be your default choice, don't hesitate to use it when the situation truly calls for it.
2857
-
2858
- # ${toolNamePrefix}replace_in_file
2859
-
2860
- ## Purpose
2861
-
2862
- - Make targeted edits to specific parts of an existing file without overwriting the entire file.
2863
-
2864
- ## When to Use
2865
-
2866
- - Small, localized changes like updating a few lines, function implementations, changing variable names, modifying a section of text, etc.
2867
- - Targeted improvements where only specific portions of the file's content needs to be altered.
2868
- - Especially useful for long files where much of the file will remain unchanged.
2869
-
2870
- ## Advantages
2871
-
2872
- - More efficient for minor edits, since you don't need to supply the entire file content.
2873
- - Reduces the chance of errors that can occur when overwriting large files.
2874
-
2875
- # Choosing the Appropriate Tool
2876
-
2877
- - **Default to ${toolNamePrefix}replace_in_file** for most changes. It keeps diffs small and reduces risk.
2878
- - **Use ${toolNamePrefix}write_to_file** when:
2879
- - Creating new files
2880
- - The changes are so extensive that using ${toolNamePrefix}replace_in_file would be more complex or risky
2881
- - You need to completely reorganize or restructure a file
2882
- - The file is relatively small and the changes affect most of its content
2883
- - You're generating boilerplate or template files
2884
-
2885
- # Workflow Tips
2886
-
2887
- 1. Before editing, assess the scope of your changes and decide which tool to use.
2888
- 2. For targeted edits, apply ${toolNamePrefix}replace_in_file with carefully crafted before/after text anchors. If you need multiple changes, you can stack multiple operations within a single ${toolNamePrefix}replace_in_file call.
2889
- 3. For major overhauls or initial file creation, rely on ${toolNamePrefix}write_to_file.
2890
- 4. Once the file has been edited with either ${toolNamePrefix}write_to_file or ${toolNamePrefix}replace_in_file, the system will provide you with the final state of the modified file. Use this updated content as the reference point for any subsequent operations, since it reflects any auto-formatting or user-applied changes.
2891
-
2892
- Picking the right tool keeps edits minimal, safe, and easy to review.
2893
- `;
2894
- var rules = (toolNamePrefix) => `
2895
- ====
2896
-
2897
- RULES
2898
-
2899
- - Work only with relative paths; you may \`cd\` into child directories but never use \`cd ..\`, root, or absolute paths.
2900
- - When generating code, tests, or other comment-capable files, prepend a comment describing the file's purpose plus \u201Cgenerated by polka.codes\u201D.
2901
- For text files (e.g. README.md), append a footer with the same notice.
2902
- - Never describe what changed inside code comments; comments must focus on purpose or usage only.
2903
- - Before using ${toolNamePrefix}execute_command, consider SYSTEM INFORMATION to ensure commands suit the user's OS. If a command must run in a subdirectory, prepend a single \`cd childDir &&\` segment.
2904
- - Use ${toolNamePrefix}search_files for broad analysis, then ${toolNamePrefix}read_file to inspect context, and finally ${toolNamePrefix}replace_in_file or ${toolNamePrefix}write_to_file to modify.
2905
- - Prefer ${toolNamePrefix}replace_in_file for focused edits; choose ${toolNamePrefix}write_to_file for new files or complete rewrites.
2906
- - When creating a new file, look for existing files with similar content or patterns; if found, read them and use their structure or conventions as a reference.
2907
- - Use before/after text anchors in ${toolNamePrefix}replace_in_file to target changes. If multiple operations are needed, list them in file order.
2908
- - Do not guess unseen content. Read existing files first unless creating new ones.
2909
- - Follow existing style, lint, and naming conventions. Ensure all changes compile and pass tests where applicable.
2910
- - ALWAYS wait for the user's confirmation after each tool call before starting the next step.
2911
- - The agent must never invoke more than 5 tools in a single response.
2912
- - Do not end ${toolNamePrefix}attempt_completion output with questions or conversational prompts.
2913
- - Avoid filler words like \u201CGreat\u201D, \u201CCertainly\u201D, \u201COkay\u201D, \u201CSure\u201D at the start of responses; be direct and technical.
2914
- - Keep inline documentation current as you edit.
2915
- `;
2916
- var objectives = (toolNamePrefix) => `
2917
- ====
2918
-
2919
- OBJECTIVE
2920
-
2921
- You solve the user's task by working in small, verifiable steps.
2922
-
2923
- 1. **Plan** - Parse the task, list clear goals, and order them logically.
2924
- 2. **Think** - Wrap private reasoning in <thinking></thinking>.
2925
- \u2022 Review project context.
2926
- \u2022 Select the single best tool for the next goal.
2927
- \u2022 Ensure every required parameter is available or can be inferred.
2928
- 3. **Act** - Invoke one tool per step. Wait for the system's response (and user confirmation where required) before continuing.
2929
- 4. **Iterate** - Repeat Plan \u2192 Think \u2192 Act until all goals are complete.
2930
- 5. **Complete** - Use ${toolNamePrefix}attempt_completion to deliver the final result. Do not invite further discussion unless the user explicitly requests changes.
2931
- `;
2932
- var fullSystemPrompt4 = (info, tools, toolNamePrefix, instructions, scripts, useNativeTool) => `
2933
- ${basePrompt2}
2934
- ${toolUsePrompt(useNativeTool, tools, toolNamePrefix)}
2935
- ${editingFilesPrompt(toolNamePrefix)}
2936
- ${capabilities(toolNamePrefix)}
2937
- ${rules(toolNamePrefix)}
2938
- ${objectives(toolNamePrefix)}
2939
- ${systemInformation(info)}
2940
- ${customInstructions(instructions)}
2941
- ${customScripts(scripts)}
2942
- `;
2943
-
2944
- // src/Agent/CoderAgent/index.ts
2945
- var agentTools4 = [
2946
- attemptCompletion_default,
2947
- delegate_default,
2948
- executeCommand_default,
2949
- fetchUrl_default,
2950
- handOver_default,
2951
- listFiles_default,
2952
- readBinaryFile_default,
2953
- readFile_default,
2954
- removeFile_default,
2955
- renameFile_default,
2956
- replaceInFile_default,
2957
- searchFiles_default,
2958
- writeToFile_default
2959
- ];
2960
- var CoderAgent = class extends AgentBase {
2961
- constructor(options) {
2962
- const combinedTools = [...options.additionalTools ?? [], ...agentTools4];
2963
- const tools = getAvailableTools({
2964
- provider: options.provider,
2965
- allTools: combinedTools,
2966
- hasAgent: (options.agents?.length ?? 0) > 0
2967
- });
2968
- const toolNamePrefix = options.toolFormat === "native" ? "" : "tool_";
2969
- const systemPrompt = fullSystemPrompt4(
2970
- {
2971
- os: options.os
2972
- },
2973
- tools,
2974
- toolNamePrefix,
2975
- options.customInstructions ?? [],
2976
- options.scripts ?? {},
2977
- options.toolFormat === "native"
2978
- );
2979
- super(coderAgentInfo.name, options.ai, {
2980
- systemPrompt,
2981
- tools,
2982
- toolNamePrefix,
2983
- provider: options.provider,
2984
- agents: options.agents,
2985
- scripts: options.scripts,
2986
- callback: options.callback,
2987
- policies: options.policies,
2988
- toolFormat: options.toolFormat,
2989
- parameters: options.parameters ?? {},
2990
- usageMeter: options.usageMeter ?? new UsageMeter(),
2991
- requestTimeoutSeconds: options.requestTimeoutSeconds,
2992
- requireToolUse: options.requireToolUse ?? true
2993
- });
2994
- }
2995
- async #runScript(scriptName, shouldReplyWithError) {
2996
- const executeCommand = this.config.provider.executeCommand;
2997
- if (!executeCommand) {
2998
- return;
2999
- }
3000
- const script = this.config.scripts?.[scriptName];
3001
- const command = typeof script === "string" ? script : script?.command;
3002
- if (command) {
3003
- try {
3004
- const { exitCode, stdout, stderr } = await executeCommand(command, false);
3005
- if (exitCode !== 0 && shouldReplyWithError) {
3006
- return {
3007
- type: "Reply" /* Reply */,
3008
- message: {
3009
- type: "error-text",
3010
- value: responsePrompts.commandResult(command, exitCode, stdout, stderr)
3011
- }
3012
- };
3013
- }
3014
- } catch (error) {
3015
- console.warn(`Failed to run ${scriptName} using command: ${command}`, error);
3016
- }
3017
- }
3018
- }
3019
- async onBeforeInvokeTool(name, _args) {
3020
- if (name !== attemptCompletion_default.name) {
3021
- return;
3022
- }
3023
- await this.#runScript("format", false);
3024
- const checkResult = await this.#runScript("check", true);
3025
- if (checkResult) {
3026
- return checkResult;
3027
- }
3028
- const testResult = await this.#runScript("test", true);
3029
- if (testResult) {
3030
- return testResult;
3031
- }
3032
- }
3033
- };
3034
- var coderAgentInfo = {
3035
- name: "coder",
3036
- responsibilities: [
3037
- "Editing and refactoring existing code.",
3038
- "Creating new features or modules.",
3039
- "Running tests and analyzing test results.",
3040
- "Maintaining coding standards, lint rules, and general code quality."
3041
- ]
3042
- };
3043
-
3044
- // src/Agent/MultiAgent.ts
3045
- var MultiAgent = class {
3046
- #config;
3047
- #agents = [];
3048
- #originalTask;
3049
- constructor(config) {
3050
- this.#config = config;
3051
- }
3052
- async #handleTaskResult(exitReason) {
3053
- switch (exitReason.type) {
3054
- case "HandOver" /* HandOver */: {
3055
- this.#agents.pop();
3056
- const prompt = await this.#config.getPrompt?.(
3057
- exitReason.agentName,
3058
- exitReason.task,
3059
- exitReason.context,
3060
- exitReason.files,
3061
- this.#originalTask
3062
- ) ?? exitReason.task;
3063
- return await this.#startTask(exitReason.agentName, prompt);
3064
- }
3065
- case "Delegate" /* Delegate */: {
3066
- const prompt = await this.#config.getPrompt?.(
3067
- exitReason.agentName,
3068
- exitReason.task,
3069
- exitReason.context,
3070
- exitReason.files,
3071
- this.#originalTask
3072
- ) ?? exitReason.task;
3073
- const delegateResult = await this.#startTask(exitReason.agentName, prompt);
3074
- switch (delegateResult.type) {
3075
- case "HandOver" /* HandOver */:
3076
- case "Delegate" /* Delegate */:
3077
- console.warn("Unexpected exit reason", delegateResult);
3078
- break;
3079
- case "Aborted":
3080
- case "Interrupted" /* Interrupted */:
3081
- return delegateResult;
3082
- case "Exit" /* Exit */:
3083
- return this.continueTask(delegateResult.message);
3084
- }
3085
- return delegateResult;
3086
- }
3087
- case "Aborted":
3088
- case "Interrupted" /* Interrupted */:
3089
- case "Exit" /* Exit */:
3090
- this.#agents.pop();
3091
- return exitReason;
3092
- default:
3093
- return exitReason;
3094
- }
3095
- }
3096
- async #startTask(agentName, task) {
3097
- const newAgent = await this.#config.createAgent(agentName);
3098
- this.#agents.push(newAgent);
3099
- const exitReason = await newAgent.start(task);
3100
- return await this.#handleTaskResult(exitReason);
3101
- }
3102
- async startTask(options) {
3103
- if (this.#agents.length > 0) {
3104
- throw new Error("An active agent already exists");
3105
- }
3106
- this.#originalTask = options.task;
3107
- const userContent = options.files ?? [];
3108
- userContent.push({
3109
- type: "text",
3110
- text: `<task>${options.task}</task>
3111
- <context>${options.context}</context>`
3112
- });
3113
- return this.#startTask(options.agentName, userContent);
3114
- }
3115
- async continueTask(userMessage) {
3116
- if (!this.#agents.length) {
3117
- throw new Error("No active agent");
3118
- }
3119
- const exitReason = await this.#agents[this.#agents.length - 1].continueTask(userMessage);
3120
- return await this.#handleTaskResult(exitReason);
3121
- }
3122
- get hasActiveAgent() {
3123
- return this.#agents.length > 0;
3124
- }
3125
- abort() {
3126
- if (this.hasActiveAgent) {
3127
- this.#agents[this.#agents.length - 1].abort();
3128
- }
3129
- }
3130
- };
3131
-
3132
- // src/Agent/parseJsonFromMarkdown.ts
3133
- var parseJsonFromMarkdown = (markdown) => {
3134
- const jsonRegex = /```(?:json)?\n([\s\S]*?)\n```/;
3135
- const match = markdown.match(jsonRegex);
3136
- const tryParse = (str) => {
3137
- try {
3138
- let parsed = JSON.parse(str);
3139
- if (typeof parsed === "string") {
3140
- try {
3141
- parsed = JSON.parse(parsed);
3142
- } catch {
3143
- }
3144
- }
3145
- return { success: true, data: parsed };
3146
- } catch (e) {
3147
- const error = e instanceof Error ? e.message : String(e);
3148
- return { success: false, error: `Failed to parse JSON: ${error}` };
3149
- }
3150
- };
3151
- if (match?.[1]) {
3152
- const content = match[1].trim();
3153
- return tryParse(content);
3154
- }
3155
- const parseResult = tryParse(markdown);
3156
- if (parseResult.success) {
3157
- return parseResult;
3158
- }
3159
- return { success: false, error: "No JSON object found in the string." };
3160
- };
3161
-
3162
- // src/config.ts
3163
- import { z as z16 } from "zod";
3164
- var toolFormatSchema = z16.enum(["native", "polka-codes"]).optional();
3165
- var providerModelSchema = z16.object({
3166
- provider: z16.string().optional(),
3167
- model: z16.string().optional(),
3168
- parameters: z16.record(z16.string(), z16.any()).optional(),
3169
- toolFormat: toolFormatSchema
3170
- });
3171
- var agentSchema = providerModelSchema.extend({
3172
- initialContext: z16.object({
3173
- maxFileCount: z16.number().int().positive().optional(),
3174
- excludes: z16.array(z16.string()).optional()
3175
- }).optional(),
3176
- retryCount: z16.number().int().min(0).optional(),
3177
- requestTimeoutSeconds: z16.number().int().positive().optional()
3178
- });
3179
- var configSchema = z16.object({
3180
- agent: z16.string().optional(),
3181
- prices: z16.record(
3182
- z16.string(),
3183
- // provider
3184
- z16.record(
3185
- z16.string(),
3186
- // model
3187
- z16.object({
3188
- inputPrice: z16.number().optional(),
3189
- outputPrice: z16.number().optional(),
3190
- cacheWritesPrice: z16.number().optional(),
3191
- cacheReadsPrice: z16.number().optional()
3192
- })
3193
- )
3194
- ).optional(),
3195
- providers: z16.record(
3196
- z16.string(),
3197
- z16.object({
3198
- apiKey: z16.string().optional(),
3199
- defaultModel: z16.string().optional(),
3200
- defaultParameters: z16.record(z16.string(), z16.any()).optional(),
3201
- location: z16.string().optional(),
3202
- project: z16.string().optional(),
3203
- keyFile: z16.string().optional()
3204
- })
3205
- ).optional(),
3206
- defaultProvider: z16.string().optional(),
3207
- defaultModel: z16.string().optional(),
3208
- defaultParameters: z16.record(z16.string(), z16.any()).optional(),
3209
- toolFormat: toolFormatSchema,
3210
- maxMessageCount: z16.number().int().positive().optional(),
3211
- budget: z16.number().positive().optional(),
3212
- retryCount: z16.number().int().min(0).optional(),
3213
- requestTimeoutSeconds: z16.number().int().positive().optional(),
3214
- summaryThreshold: z16.number().int().positive().optional(),
3215
- scripts: z16.record(
3216
- z16.string(),
3217
- z16.string().or(
3218
- z16.object({
3219
- command: z16.string(),
3220
- description: z16.string()
3221
- })
3222
- )
3223
- ).optional(),
3224
- agents: z16.record(z16.string(), agentSchema).optional(),
3225
- commands: z16.object({
3226
- default: providerModelSchema.optional()
3227
- }).catchall(providerModelSchema).optional(),
3228
- rules: z16.array(z16.string()).optional().or(z16.string()).optional(),
3229
- excludeFiles: z16.array(z16.string()).optional(),
3230
- policies: z16.array(z16.string()).optional()
3231
- }).strict();
3232
- var Policies = /* @__PURE__ */ ((Policies2) => {
3233
- Policies2["TruncateContext"] = "truncatecontext";
3234
- Policies2["EnableCache"] = "enablecache";
3235
- return Policies2;
3236
- })(Policies || {});
3237
-
3238
- // src/Agent/policies/EnableCachePolicy.ts
3239
- var CACHEABLE_MODELS = ["sonnet", "opus", "haiku", "gemini"];
3240
- function isCacheableModel(modelId) {
3241
- return CACHEABLE_MODELS.some((model) => modelId.includes(model));
3242
- }
3243
- function getProviderKey(provider) {
3244
- if (provider === "openrouter") {
3245
- return "openrouter";
3246
- }
3247
- if (provider.includes("anthropic")) {
3248
- return "anthropic";
3249
- }
3250
- return void 0;
3251
- }
3252
- var EnableCachePolicy = () => {
3253
- return {
3254
- name: "enablecache" /* EnableCache */,
3255
- async prepareMessages(agent, messages) {
3256
- const providerKey = getProviderKey(agent.ai.provider);
3257
- if (!providerKey || !isCacheableModel(agent.ai.modelId)) {
3258
- return messages;
3259
- }
3260
- const providerOptions = { [providerKey]: { cacheControl: { type: "ephemeral" } } };
3261
- const newMessages = messages.slice();
3262
- let userMessagesToUpdate = 2;
3263
- for (let i = newMessages.length - 1; i >= 0; i--) {
3264
- const message = newMessages[i];
3265
- if (message.role === "user" && userMessagesToUpdate > 0) {
3266
- newMessages[i] = {
3267
- ...message,
3268
- providerOptions: {
3269
- ...providerOptions,
3270
- ...message.providerOptions ?? {}
3271
- }
3272
- };
3273
- userMessagesToUpdate--;
3274
- } else if (message.role === "system") {
3275
- newMessages[i] = {
3276
- ...message,
3277
- providerOptions: {
3278
- ...providerOptions,
3279
- ...message.providerOptions ?? {}
3280
- }
3281
- };
3282
- break;
3283
- }
3284
- }
3285
- return newMessages;
3286
- }
3287
- };
3288
- };
3289
-
3290
- // src/Agent/policies/TruncateContextPolicy.ts
3291
- var DEFAULT_MAX_TOKENS_ESTIMATE = 32e3;
3292
- function getMaxTokens(agent) {
3293
- const params = agent.parameters || {};
3294
- if (params.maxTokens && typeof params.maxTokens === "number" && params.maxTokens > 0) {
3295
- return params.maxTokens;
3296
- }
3297
- return DEFAULT_MAX_TOKENS_ESTIMATE;
3298
- }
3299
- function estimateTokens(text) {
3300
- return Math.ceil(text.length / 4);
3301
- }
3302
- function estimateMessageTokens(msg) {
3303
- if (typeof msg.content === "string") {
3304
- return estimateTokens(msg.content);
3305
- }
3306
- if (Array.isArray(msg.content)) {
3307
- let messageTokens = 0;
3308
- for (const block of msg.content) {
3309
- if (typeof block === "object" && "text" in block && typeof block.text === "string") {
3310
- messageTokens += estimateTokens(block.text);
3311
- }
3312
- }
3313
- return messageTokens;
3314
- }
3315
- return 0;
3316
- }
3317
- var TruncateContextPolicy = (() => {
3318
- return {
3319
- name: "truncatecontext" /* TruncateContext */,
3320
- async onBeforeRequest(agent) {
3321
- const messages = agent.messages;
3322
- if (messages.length < 3) {
3323
- return;
3324
- }
3325
- let totalTokens = messages.reduce((acc, msg) => acc + estimateMessageTokens(msg), 0);
3326
- const maxTokens = getMaxTokens(agent);
3327
- if (totalTokens <= maxTokens) {
3328
- return;
3329
- }
3330
- const protectedIndices = /* @__PURE__ */ new Set();
3331
- let userMessagesFound = 0;
3332
- messages.forEach((msg, index) => {
3333
- if (msg.role === "system") {
3334
- protectedIndices.add(index);
3335
- }
3336
- if (msg.role === "user") {
3337
- if (userMessagesFound < 2) {
3338
- protectedIndices.add(index);
3339
- }
3340
- userMessagesFound++;
3341
- }
3342
- });
3343
- if (messages.length > 0) {
3344
- protectedIndices.add(messages.length - 1);
3345
- }
3346
- const truncatableIndices = [];
3347
- for (let i = 0; i < messages.length; i++) {
3348
- if (!protectedIndices.has(i)) {
3349
- truncatableIndices.push(i);
3350
- }
3351
- }
3352
- const removedIndices = /* @__PURE__ */ new Set();
3353
- const initialTotalTokens = totalTokens;
3354
- while (totalTokens > maxTokens && truncatableIndices.length > 0) {
3355
- const middleIndex = Math.floor(truncatableIndices.length / 2);
3356
- const indexToRemove = truncatableIndices.splice(middleIndex, 1)[0];
3357
- removedIndices.add(indexToRemove);
3358
- totalTokens -= estimateMessageTokens(messages[indexToRemove]);
3359
- }
3360
- if (removedIndices.size === 0) {
3361
- if (initialTotalTokens > maxTokens) {
3362
- console.warn("Warning: Protected messages exceed token limit. Cannot truncate further.");
3363
- }
3364
- return;
3365
- }
3366
- const truncatedMessages = [];
3367
- let truncationNoticeAdded = false;
3368
- for (let i = 0; i < messages.length; i++) {
3369
- if (removedIndices.has(i)) {
3370
- if (!truncationNoticeAdded) {
3371
- truncatedMessages.push({
3372
- role: "user",
3373
- content: `Note: ${removedIndices.size} messages were truncated from the conversation to prevent context overflow.`
3374
- });
3375
- truncationNoticeAdded = true;
3376
- }
3377
- } else {
3378
- truncatedMessages.push(messages[i]);
3379
- }
3380
- }
3381
- console.log(`Truncated context: removed ${removedIndices.size} messages.`);
3382
- agent.setMessages(truncatedMessages);
3383
- }
3384
- };
3385
- });
3386
-
3387
- // src/Agent/index.ts
3388
- var allAgents = [architectAgentInfo, coderAgentInfo, analyzerAgentInfo, codeFixerAgentInfo];
3389
1480
  export {
3390
- AgentBase,
3391
- AnalyzerAgent,
3392
- ArchitectAgent,
3393
- CodeFixerAgent,
3394
- CoderAgent,
3395
- EnableCachePolicy,
3396
1481
  MockProvider,
3397
- MultiAgent,
3398
- Policies,
3399
- TaskEventKind,
3400
1482
  ToolResponseType,
3401
- TruncateContextPolicy,
3402
1483
  UsageMeter,
3403
- agentsPrompt,
3404
- allAgents,
3405
- analyzerAgentInfo,
3406
- architectAgentInfo,
1484
+ appendMemory_default as appendMemory,
3407
1485
  askFollowupQuestion_default as askFollowupQuestion,
3408
- attemptCompletion_default as attemptCompletion,
3409
- capabilities,
3410
- codeFixerAgentInfo,
3411
- coderAgentInfo,
3412
1486
  computeRateLimitBackoffSeconds,
3413
1487
  configSchema,
3414
- customInstructions,
3415
- customScripts,
3416
- delegate_default as delegate,
3417
1488
  executeCommand_default as executeCommand,
3418
1489
  fetchUrl_default as fetchUrl,
3419
- getAvailableTools,
3420
- handOver_default as handOver,
3421
1490
  listFiles_default as listFiles,
3422
- parseAssistantMessage,
1491
+ listMemoryTopics_default as listMemoryTopics,
3423
1492
  parseJsonFromMarkdown,
3424
1493
  readBinaryFile_default as readBinaryFile,
3425
1494
  readFile_default as readFile,
1495
+ readMemory_default as readMemory,
3426
1496
  removeFile_default as removeFile,
1497
+ removeMemory_default as removeMemory,
3427
1498
  renameFile_default as renameFile,
3428
1499
  replaceInFile_default as replaceInFile,
3429
1500
  replaceInFile as replaceInFileHelper,
1501
+ replaceMemory_default as replaceMemory,
3430
1502
  responsePrompts,
3431
1503
  searchFiles_default as searchFiles,
3432
- systemInformation,
3433
- toolUsePrompt,
3434
1504
  writeToFile_default as writeToFile
3435
1505
  };