@auvira.ai/sdk 0.2.9

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (303) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +414 -0
  3. package/dist/agent/Agent.d.ts +27 -0
  4. package/dist/agent/Agent.d.ts.map +1 -0
  5. package/dist/agent/Agent.js +144 -0
  6. package/dist/agent/Agent.js.map +1 -0
  7. package/dist/agent/AgentRegistry.d.ts +13 -0
  8. package/dist/agent/AgentRegistry.d.ts.map +1 -0
  9. package/dist/agent/AgentRegistry.js +28 -0
  10. package/dist/agent/AgentRegistry.js.map +1 -0
  11. package/dist/agent/AgentRun.d.ts +38 -0
  12. package/dist/agent/AgentRun.d.ts.map +1 -0
  13. package/dist/agent/AgentRun.js +101 -0
  14. package/dist/agent/AgentRun.js.map +1 -0
  15. package/dist/agent/ConversationStore.d.ts +18 -0
  16. package/dist/agent/ConversationStore.d.ts.map +1 -0
  17. package/dist/agent/ConversationStore.js +38 -0
  18. package/dist/agent/ConversationStore.js.map +1 -0
  19. package/dist/agent/RunEventBus.d.ts +43 -0
  20. package/dist/agent/RunEventBus.d.ts.map +1 -0
  21. package/dist/agent/RunEventBus.js +289 -0
  22. package/dist/agent/RunEventBus.js.map +1 -0
  23. package/dist/agent/assistantEvent.d.ts +11 -0
  24. package/dist/agent/assistantEvent.d.ts.map +1 -0
  25. package/dist/agent/assistantEvent.js +20 -0
  26. package/dist/agent/assistantEvent.js.map +1 -0
  27. package/dist/agent/buildAgentPrompt.d.ts +10 -0
  28. package/dist/agent/buildAgentPrompt.d.ts.map +1 -0
  29. package/dist/agent/buildAgentPrompt.js +39 -0
  30. package/dist/agent/buildAgentPrompt.js.map +1 -0
  31. package/dist/agent/cancellation.d.ts +9 -0
  32. package/dist/agent/cancellation.d.ts.map +1 -0
  33. package/dist/agent/cancellation.js +18 -0
  34. package/dist/agent/cancellation.js.map +1 -0
  35. package/dist/agent/editCompletion.d.ts +35 -0
  36. package/dist/agent/editCompletion.d.ts.map +1 -0
  37. package/dist/agent/editCompletion.js +179 -0
  38. package/dist/agent/editCompletion.js.map +1 -0
  39. package/dist/agent/editEvents.d.ts +11 -0
  40. package/dist/agent/editEvents.d.ts.map +1 -0
  41. package/dist/agent/editEvents.js +36 -0
  42. package/dist/agent/editEvents.js.map +1 -0
  43. package/dist/agent/evaluateTaskCompletion.d.ts +49 -0
  44. package/dist/agent/evaluateTaskCompletion.d.ts.map +1 -0
  45. package/dist/agent/evaluateTaskCompletion.js +257 -0
  46. package/dist/agent/evaluateTaskCompletion.js.map +1 -0
  47. package/dist/agent/events.d.ts +182 -0
  48. package/dist/agent/events.d.ts.map +1 -0
  49. package/dist/agent/events.js +25 -0
  50. package/dist/agent/events.js.map +1 -0
  51. package/dist/agent/parseSendPayload.d.ts +24 -0
  52. package/dist/agent/parseSendPayload.d.ts.map +1 -0
  53. package/dist/agent/parseSendPayload.js +71 -0
  54. package/dist/agent/parseSendPayload.js.map +1 -0
  55. package/dist/agent/patchUtils.d.ts +17 -0
  56. package/dist/agent/patchUtils.d.ts.map +1 -0
  57. package/dist/agent/patchUtils.js +90 -0
  58. package/dist/agent/patchUtils.js.map +1 -0
  59. package/dist/agent/resolveModelConfig.d.ts +4 -0
  60. package/dist/agent/resolveModelConfig.d.ts.map +1 -0
  61. package/dist/agent/resolveModelConfig.js +27 -0
  62. package/dist/agent/resolveModelConfig.js.map +1 -0
  63. package/dist/agent/runAgentTask.d.ts +14 -0
  64. package/dist/agent/runAgentTask.d.ts.map +1 -0
  65. package/dist/agent/runAgentTask.js +376 -0
  66. package/dist/agent/runAgentTask.js.map +1 -0
  67. package/dist/agent/runResult.d.ts +10 -0
  68. package/dist/agent/runResult.d.ts.map +1 -0
  69. package/dist/agent/runResult.js +34 -0
  70. package/dist/agent/runResult.js.map +1 -0
  71. package/dist/agent/runValidationWithRepair.d.ts +32 -0
  72. package/dist/agent/runValidationWithRepair.d.ts.map +1 -0
  73. package/dist/agent/runValidationWithRepair.js +114 -0
  74. package/dist/agent/runValidationWithRepair.js.map +1 -0
  75. package/dist/agent/tools/edit.d.ts +4 -0
  76. package/dist/agent/tools/edit.d.ts.map +1 -0
  77. package/dist/agent/tools/edit.js +98 -0
  78. package/dist/agent/tools/edit.js.map +1 -0
  79. package/dist/agent/tools/executeTool.d.ts +8 -0
  80. package/dist/agent/tools/executeTool.d.ts.map +1 -0
  81. package/dist/agent/tools/executeTool.js +104 -0
  82. package/dist/agent/tools/executeTool.js.map +1 -0
  83. package/dist/agent/tools/glob.d.ts +6 -0
  84. package/dist/agent/tools/glob.d.ts.map +1 -0
  85. package/dist/agent/tools/glob.js +77 -0
  86. package/dist/agent/tools/glob.js.map +1 -0
  87. package/dist/agent/tools/grep.d.ts +4 -0
  88. package/dist/agent/tools/grep.d.ts.map +1 -0
  89. package/dist/agent/tools/grep.js +61 -0
  90. package/dist/agent/tools/grep.js.map +1 -0
  91. package/dist/agent/tools/pathGuard.d.ts +11 -0
  92. package/dist/agent/tools/pathGuard.d.ts.map +1 -0
  93. package/dist/agent/tools/pathGuard.js +55 -0
  94. package/dist/agent/tools/pathGuard.js.map +1 -0
  95. package/dist/agent/tools/read.d.ts +4 -0
  96. package/dist/agent/tools/read.d.ts.map +1 -0
  97. package/dist/agent/tools/read.js +41 -0
  98. package/dist/agent/tools/read.js.map +1 -0
  99. package/dist/agent/tools/types.d.ts +46 -0
  100. package/dist/agent/tools/types.d.ts.map +1 -0
  101. package/dist/agent/tools/types.js +2 -0
  102. package/dist/agent/tools/types.js.map +1 -0
  103. package/dist/agent/tools/write.d.ts +4 -0
  104. package/dist/agent/tools/write.d.ts.map +1 -0
  105. package/dist/agent/tools/write.js +63 -0
  106. package/dist/agent/tools/write.js.map +1 -0
  107. package/dist/agent/types.d.ts +193 -0
  108. package/dist/agent/types.d.ts.map +1 -0
  109. package/dist/agent/types.js +18 -0
  110. package/dist/agent/types.js.map +1 -0
  111. package/dist/agent/validateOptions.d.ts +12 -0
  112. package/dist/agent/validateOptions.d.ts.map +1 -0
  113. package/dist/agent/validateOptions.js +91 -0
  114. package/dist/agent/validateOptions.js.map +1 -0
  115. package/dist/git/cloneGitLabProject.d.ts +10 -0
  116. package/dist/git/cloneGitLabProject.d.ts.map +1 -0
  117. package/dist/git/cloneGitLabProject.js +24 -0
  118. package/dist/git/cloneGitLabProject.js.map +1 -0
  119. package/dist/git/createAgentBranch.d.ts +3 -0
  120. package/dist/git/createAgentBranch.d.ts.map +1 -0
  121. package/dist/git/createAgentBranch.js +14 -0
  122. package/dist/git/createAgentBranch.js.map +1 -0
  123. package/dist/git/gitDiff.d.ts +9 -0
  124. package/dist/git/gitDiff.d.ts.map +1 -0
  125. package/dist/git/gitDiff.js +34 -0
  126. package/dist/git/gitDiff.js.map +1 -0
  127. package/dist/git/maskSecrets.d.ts +3 -0
  128. package/dist/git/maskSecrets.d.ts.map +1 -0
  129. package/dist/git/maskSecrets.js +15 -0
  130. package/dist/git/maskSecrets.js.map +1 -0
  131. package/dist/git/resolveGitLabProject.d.ts +26 -0
  132. package/dist/git/resolveGitLabProject.d.ts.map +1 -0
  133. package/dist/git/resolveGitLabProject.js +68 -0
  134. package/dist/git/resolveGitLabProject.js.map +1 -0
  135. package/dist/git/sanitizeGitLabUrl.d.ts +8 -0
  136. package/dist/git/sanitizeGitLabUrl.d.ts.map +1 -0
  137. package/dist/git/sanitizeGitLabUrl.js +38 -0
  138. package/dist/git/sanitizeGitLabUrl.js.map +1 -0
  139. package/dist/index.d.ts +13 -0
  140. package/dist/index.d.ts.map +1 -0
  141. package/dist/index.js +7 -0
  142. package/dist/index.js.map +1 -0
  143. package/dist/providers/agentic/AgenticCustomProvider.d.ts +9 -0
  144. package/dist/providers/agentic/AgenticCustomProvider.d.ts.map +1 -0
  145. package/dist/providers/agentic/AgenticCustomProvider.js +366 -0
  146. package/dist/providers/agentic/AgenticCustomProvider.js.map +1 -0
  147. package/dist/providers/agentic/agentLoop.d.ts +79 -0
  148. package/dist/providers/agentic/agentLoop.d.ts.map +1 -0
  149. package/dist/providers/agentic/agentLoop.js +646 -0
  150. package/dist/providers/agentic/agentLoop.js.map +1 -0
  151. package/dist/providers/agentic/priomptContext.d.ts +29 -0
  152. package/dist/providers/agentic/priomptContext.d.ts.map +1 -0
  153. package/dist/providers/agentic/priomptContext.js +90 -0
  154. package/dist/providers/agentic/priomptContext.js.map +1 -0
  155. package/dist/providers/agentic/seedRetrieval.d.ts +12 -0
  156. package/dist/providers/agentic/seedRetrieval.d.ts.map +1 -0
  157. package/dist/providers/agentic/seedRetrieval.js +58 -0
  158. package/dist/providers/agentic/seedRetrieval.js.map +1 -0
  159. package/dist/providers/agentic/toolPolicy.d.ts +15 -0
  160. package/dist/providers/agentic/toolPolicy.d.ts.map +1 -0
  161. package/dist/providers/agentic/toolPolicy.js +53 -0
  162. package/dist/providers/agentic/toolPolicy.js.map +1 -0
  163. package/dist/providers/agentic/toolSchemas.d.ts +177 -0
  164. package/dist/providers/agentic/toolSchemas.d.ts.map +1 -0
  165. package/dist/providers/agentic/toolSchemas.js +104 -0
  166. package/dist/providers/agentic/toolSchemas.js.map +1 -0
  167. package/dist/providers/agentic/trimAgentHistory.d.ts +18 -0
  168. package/dist/providers/agentic/trimAgentHistory.d.ts.map +1 -0
  169. package/dist/providers/agentic/trimAgentHistory.js +68 -0
  170. package/dist/providers/agentic/trimAgentHistory.js.map +1 -0
  171. package/dist/providers/custom/CustomModelProvider.d.ts +4 -0
  172. package/dist/providers/custom/CustomModelProvider.d.ts.map +1 -0
  173. package/dist/providers/custom/CustomModelProvider.js +538 -0
  174. package/dist/providers/custom/CustomModelProvider.js.map +1 -0
  175. package/dist/providers/custom/discoverTargetFiles.d.ts +16 -0
  176. package/dist/providers/custom/discoverTargetFiles.d.ts.map +1 -0
  177. package/dist/providers/custom/discoverTargetFiles.js +231 -0
  178. package/dist/providers/custom/discoverTargetFiles.js.map +1 -0
  179. package/dist/providers/custom/harnessPinTypes.d.ts +15 -0
  180. package/dist/providers/custom/harnessPinTypes.d.ts.map +1 -0
  181. package/dist/providers/custom/harnessPinTypes.js +46 -0
  182. package/dist/providers/custom/harnessPinTypes.js.map +1 -0
  183. package/dist/providers/custom/harnessPrompt.d.ts +29 -0
  184. package/dist/providers/custom/harnessPrompt.d.ts.map +1 -0
  185. package/dist/providers/custom/harnessPrompt.js +356 -0
  186. package/dist/providers/custom/harnessPrompt.js.map +1 -0
  187. package/dist/providers/custom/harnessSectionContext.d.ts +7 -0
  188. package/dist/providers/custom/harnessSectionContext.d.ts.map +1 -0
  189. package/dist/providers/custom/harnessSectionContext.js +82 -0
  190. package/dist/providers/custom/harnessSectionContext.js.map +1 -0
  191. package/dist/providers/custom/modelRequestDefaults.d.ts +16 -0
  192. package/dist/providers/custom/modelRequestDefaults.d.ts.map +1 -0
  193. package/dist/providers/custom/modelRequestDefaults.js +71 -0
  194. package/dist/providers/custom/modelRequestDefaults.js.map +1 -0
  195. package/dist/providers/custom/parseModelEditPlan.d.ts +21 -0
  196. package/dist/providers/custom/parseModelEditPlan.d.ts.map +1 -0
  197. package/dist/providers/custom/parseModelEditPlan.js +164 -0
  198. package/dist/providers/custom/parseModelEditPlan.js.map +1 -0
  199. package/dist/providers/custom/selectPromptFiles.d.ts +18 -0
  200. package/dist/providers/custom/selectPromptFiles.d.ts.map +1 -0
  201. package/dist/providers/custom/selectPromptFiles.js +180 -0
  202. package/dist/providers/custom/selectPromptFiles.js.map +1 -0
  203. package/dist/providers/custom/visionImages.d.ts +4 -0
  204. package/dist/providers/custom/visionImages.d.ts.map +1 -0
  205. package/dist/providers/custom/visionImages.js +10 -0
  206. package/dist/providers/custom/visionImages.js.map +1 -0
  207. package/dist/providers/openhands/OpenHandsMockProvider.d.ts +6 -0
  208. package/dist/providers/openhands/OpenHandsMockProvider.d.ts.map +1 -0
  209. package/dist/providers/openhands/OpenHandsMockProvider.js +73 -0
  210. package/dist/providers/openhands/OpenHandsMockProvider.js.map +1 -0
  211. package/dist/providers/openhands/OpenHandsProvider.d.ts +4 -0
  212. package/dist/providers/openhands/OpenHandsProvider.d.ts.map +1 -0
  213. package/dist/providers/openhands/OpenHandsProvider.js +163 -0
  214. package/dist/providers/openhands/OpenHandsProvider.js.map +1 -0
  215. package/dist/providers/openhands/openHandsEvents.d.ts +6 -0
  216. package/dist/providers/openhands/openHandsEvents.d.ts.map +1 -0
  217. package/dist/providers/openhands/openHandsEvents.js +54 -0
  218. package/dist/providers/openhands/openHandsEvents.js.map +1 -0
  219. package/dist/providers/openhands/types.d.ts +16 -0
  220. package/dist/providers/openhands/types.d.ts.map +1 -0
  221. package/dist/providers/openhands/types.js +2 -0
  222. package/dist/providers/openhands/types.js.map +1 -0
  223. package/dist/providers/runProvider.d.ts +4 -0
  224. package/dist/providers/runProvider.d.ts.map +1 -0
  225. package/dist/providers/runProvider.js +16 -0
  226. package/dist/providers/runProvider.js.map +1 -0
  227. package/dist/providers/types.d.ts +35 -0
  228. package/dist/providers/types.d.ts.map +1 -0
  229. package/dist/providers/types.js +5 -0
  230. package/dist/providers/types.js.map +1 -0
  231. package/dist/server/routes.d.ts +9 -0
  232. package/dist/server/routes.d.ts.map +1 -0
  233. package/dist/server/routes.js +58 -0
  234. package/dist/server/routes.js.map +1 -0
  235. package/dist/server/server.d.ts +10 -0
  236. package/dist/server/server.d.ts.map +1 -0
  237. package/dist/server/server.js +39 -0
  238. package/dist/server/server.js.map +1 -0
  239. package/dist/tools/fileExists.d.ts +3 -0
  240. package/dist/tools/fileExists.d.ts.map +1 -0
  241. package/dist/tools/fileExists.js +12 -0
  242. package/dist/tools/fileExists.js.map +1 -0
  243. package/dist/tools/resolveBuildValidation.d.ts +8 -0
  244. package/dist/tools/resolveBuildValidation.d.ts.map +1 -0
  245. package/dist/tools/resolveBuildValidation.js +35 -0
  246. package/dist/tools/resolveBuildValidation.js.map +1 -0
  247. package/dist/tools/runCommand.d.ts +17 -0
  248. package/dist/tools/runCommand.d.ts.map +1 -0
  249. package/dist/tools/runCommand.js +92 -0
  250. package/dist/tools/runCommand.js.map +1 -0
  251. package/dist/tools/safeValidateCommand.d.ts +9 -0
  252. package/dist/tools/safeValidateCommand.d.ts.map +1 -0
  253. package/dist/tools/safeValidateCommand.js +59 -0
  254. package/dist/tools/safeValidateCommand.js.map +1 -0
  255. package/dist/util/apiThrottle.d.ts +14 -0
  256. package/dist/util/apiThrottle.d.ts.map +1 -0
  257. package/dist/util/apiThrottle.js +50 -0
  258. package/dist/util/apiThrottle.js.map +1 -0
  259. package/dist/util/globMatch.d.ts +7 -0
  260. package/dist/util/globMatch.d.ts.map +1 -0
  261. package/dist/util/globMatch.js +25 -0
  262. package/dist/util/globMatch.js.map +1 -0
  263. package/dist/util/modelApiClient.d.ts +47 -0
  264. package/dist/util/modelApiClient.d.ts.map +1 -0
  265. package/dist/util/modelApiClient.js +245 -0
  266. package/dist/util/modelApiClient.js.map +1 -0
  267. package/dist/util/screenshotBase64.d.ts +3 -0
  268. package/dist/util/screenshotBase64.d.ts.map +1 -0
  269. package/dist/util/screenshotBase64.js +7 -0
  270. package/dist/util/screenshotBase64.js.map +1 -0
  271. package/dist/workspace/cleanupWorkspace.d.ts +3 -0
  272. package/dist/workspace/cleanupWorkspace.d.ts.map +1 -0
  273. package/dist/workspace/cleanupWorkspace.js +6 -0
  274. package/dist/workspace/cleanupWorkspace.js.map +1 -0
  275. package/dist/workspace/copyLocalRepo.d.ts +5 -0
  276. package/dist/workspace/copyLocalRepo.d.ts.map +1 -0
  277. package/dist/workspace/copyLocalRepo.js +34 -0
  278. package/dist/workspace/copyLocalRepo.js.map +1 -0
  279. package/dist/workspace/createWorkspace.d.ts +7 -0
  280. package/dist/workspace/createWorkspace.d.ts.map +1 -0
  281. package/dist/workspace/createWorkspace.js +10 -0
  282. package/dist/workspace/createWorkspace.js.map +1 -0
  283. package/dist/workspace/prepareWorkspace.d.ts +4 -0
  284. package/dist/workspace/prepareWorkspace.d.ts.map +1 -0
  285. package/dist/workspace/prepareWorkspace.js +46 -0
  286. package/dist/workspace/prepareWorkspace.js.map +1 -0
  287. package/dist/workspace/resolveWorkspaceMode.d.ts +7 -0
  288. package/dist/workspace/resolveWorkspaceMode.d.ts.map +1 -0
  289. package/dist/workspace/resolveWorkspaceMode.js +15 -0
  290. package/dist/workspace/resolveWorkspaceMode.js.map +1 -0
  291. package/dist/workspace/syncRepoToOwner.d.ts +3 -0
  292. package/dist/workspace/syncRepoToOwner.d.ts.map +1 -0
  293. package/dist/workspace/syncRepoToOwner.js +14 -0
  294. package/dist/workspace/syncRepoToOwner.js.map +1 -0
  295. package/dist/workspace/workspaceCheckpoint.d.ts +20 -0
  296. package/dist/workspace/workspaceCheckpoint.d.ts.map +1 -0
  297. package/dist/workspace/workspaceCheckpoint.js +243 -0
  298. package/dist/workspace/workspaceCheckpoint.js.map +1 -0
  299. package/dist/workspace/writeScreenshot.d.ts +15 -0
  300. package/dist/workspace/writeScreenshot.d.ts.map +1 -0
  301. package/dist/workspace/writeScreenshot.js +56 -0
  302. package/dist/workspace/writeScreenshot.js.map +1 -0
  303. package/package.json +53 -0
@@ -0,0 +1,68 @@
1
+ const DEFAULT_TOOL_ROUNDS = 3;
2
+ /** Max assistant+tool round pairs to retain (`AUVIRA_AGENT_HISTORY_TOOL_ROUNDS`). */
3
+ export function getHistoryToolRounds() {
4
+ const raw = process.env.AUVIRA_AGENT_HISTORY_TOOL_ROUNDS?.trim();
5
+ if (!raw) {
6
+ return DEFAULT_TOOL_ROUNDS;
7
+ }
8
+ const parsed = Number.parseInt(raw, 10);
9
+ if (!Number.isFinite(parsed) || parsed < 1) {
10
+ return DEFAULT_TOOL_ROUNDS;
11
+ }
12
+ return parsed;
13
+ }
14
+ function isAssistantWithTools(message) {
15
+ return (message.role === "assistant" &&
16
+ Array.isArray(message.tool_calls) &&
17
+ message.tool_calls.length > 0);
18
+ }
19
+ function countToolRounds(messages) {
20
+ let rounds = 0;
21
+ for (let index = 0; index < messages.length; index += 1) {
22
+ if (isAssistantWithTools(messages[index])) {
23
+ rounds += 1;
24
+ }
25
+ }
26
+ return rounds;
27
+ }
28
+ function findFirstUserIndex(messages) {
29
+ return messages.findIndex((message) => message.role === "user");
30
+ }
31
+ /**
32
+ * Trims agent message history before each model call.
33
+ * Keeps system, first user bootstrap, and the last N tool rounds intact.
34
+ */
35
+ export function trimAgentHistory(messages, maxToolRounds = getHistoryToolRounds()) {
36
+ if (messages.length <= 2) {
37
+ return messages;
38
+ }
39
+ const systemMessages = messages.filter((message) => message.role === "system");
40
+ const firstUserIndex = findFirstUserIndex(messages);
41
+ const firstUser = firstUserIndex >= 0 ? messages[firstUserIndex] : undefined;
42
+ const roundStartIndices = [];
43
+ for (let index = 0; index < messages.length; index += 1) {
44
+ if (isAssistantWithTools(messages[index])) {
45
+ roundStartIndices.push(index);
46
+ }
47
+ }
48
+ if (roundStartIndices.length <= maxToolRounds) {
49
+ return messages;
50
+ }
51
+ const keepFromIndex = roundStartIndices[roundStartIndices.length - maxToolRounds] ?? 0;
52
+ const trimmed = [...systemMessages];
53
+ if (firstUser && !trimmed.includes(firstUser)) {
54
+ trimmed.push(firstUser);
55
+ }
56
+ for (let index = keepFromIndex; index < messages.length; index += 1) {
57
+ const message = messages[index];
58
+ if (message.role === "system") {
59
+ continue;
60
+ }
61
+ if (firstUser && index === firstUserIndex) {
62
+ continue;
63
+ }
64
+ trimmed.push(message);
65
+ }
66
+ return trimmed;
67
+ }
68
+ //# sourceMappingURL=trimAgentHistory.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"trimAgentHistory.js","sourceRoot":"","sources":["../../../src/providers/agentic/trimAgentHistory.ts"],"names":[],"mappings":"AAWA,MAAM,mBAAmB,GAAG,CAAC,CAAC;AAE9B,qFAAqF;AACrF,MAAM,UAAU,oBAAoB;IAClC,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,gCAAgC,EAAE,IAAI,EAAE,CAAC;IACjE,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IACD,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;IACxC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3C,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,oBAAoB,CAAC,OAA4B;IACxD,OAAO,CACL,OAAO,CAAC,IAAI,KAAK,WAAW;QAC5B,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC;QACjC,OAAO,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAC9B,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,QAA+B;IACtD,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;QACxD,IAAI,oBAAoB,CAAC,QAAQ,CAAC,KAAK,CAAE,CAAC,EAAE,CAAC;YAC3C,MAAM,IAAI,CAAC,CAAC;QACd,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,kBAAkB,CAAC,QAA+B;IACzD,OAAO,QAAQ,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;AAClE,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAC9B,QAA+B,EAC/B,aAAa,GAAG,oBAAoB,EAAE;IAEtC,IAAI,QAAQ,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;QACzB,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;IAC/E,MAAM,cAAc,GAAG,kBAAkB,CAAC,QAAQ,CAAC,CAAC;IACpD,MAAM,SAAS,GACb,cAAc,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAE7D,MAAM,iBAAiB,GAAa,EAAE,CAAC;IACvC,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;QACxD,IAAI,oBAAoB,CAAC,QAAQ,CAAC,KAAK,CAAE,CAAC,EAAE,CAAC;YAC3C,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAED,IAAI,iBAAiB,CAAC,MAAM,IAAI,aAAa,EAAE,CAAC;QAC9C,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,MAAM,aAAa,GACjB,iBAAiB,CAAC,iBAAiB,CAAC,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;IAEnE,MAAM,OAAO,GAA0B,CAAC,GAAG,cAAc,CAAC,CAAC;IAC3D,IAAI,SAAS,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QAC9C,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK,IAAI,KAAK,GAAG,aAAa,EAAE,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;QACpE,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAE,CAAC;QACjC,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC9B,SAAS;QACX,CAAC;QACD,IAAI,SAAS,IAAI,KAAK,KAAK,cAAc,EAAE,CAAC;YAC1C,SAAS;QACX,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACxB,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC"}
@@ -0,0 +1,4 @@
1
+ import type { ProviderRunInput, ProviderRunOutput } from "../types.js";
2
+ /** Calls a custom OpenAI-compatible model API and applies returned file edits. */
3
+ export declare function runCustomModelProvider(input: ProviderRunInput): Promise<ProviderRunOutput>;
4
+ //# sourceMappingURL=CustomModelProvider.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"CustomModelProvider.d.ts","sourceRoot":"","sources":["../../../src/providers/custom/CustomModelProvider.ts"],"names":[],"mappings":"AAcA,OAAO,KAAK,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAoUvE,kFAAkF;AAClF,wBAAsB,sBAAsB,CAC1C,KAAK,EAAE,gBAAgB,GACtB,OAAO,CAAC,iBAAiB,CAAC,CAuZ5B"}
@@ -0,0 +1,538 @@
1
+ import { promises as fs } from "node:fs";
2
+ import { join } from "node:path";
3
+ import { fileExists } from "../../tools/fileExists.js";
4
+ import { buildAssistantEvent } from "../../agent/assistantEvent.js";
5
+ import { splitHarnessPrompt } from "../../agent/parseSendPayload.js";
6
+ import { throwIfCancelled } from "../../agent/cancellation.js";
7
+ import { withApiConcurrencyLimit } from "../../util/apiThrottle.js";
8
+ import { formatRateLimitReport, formatTimingReport, postJsonWithRetry, } from "../../util/modelApiClient.js";
9
+ import { parseModelEditPlan, truncateModelOutput, } from "./parseModelEditPlan.js";
10
+ import { selectPromptFiles } from "./selectPromptFiles.js";
11
+ import { describePinnedTarget, discoverTargetFiles, pickExampleEditPath, } from "./discoverTargetFiles.js";
12
+ import { buildHarnessUserPrompt, buildPinnedSectionSnapshots, discoverMergedHarnessPaths, extractPinnedTargetFromHarness, formatPinnedTargetLabel, harnessTargetFromDom, mergePinnedTarget, } from "./harnessPrompt.js";
13
+ import { imagesToDataUrls } from "./visionImages.js";
14
+ import { applyCustomModelRequestDefaults } from "./modelRequestDefaults.js";
15
+ import { emitEditEventsFromPatch } from "../../agent/editEvents.js";
16
+ function getMaxEditAttempts() {
17
+ const parsed = Number(process.env.CUSTOM_MODEL_EDIT_MAX_ATTEMPTS ?? "3");
18
+ return Number.isFinite(parsed) && parsed > 0 ? parsed : 3;
19
+ }
20
+ function isJsonModeEnabled(harnessPrompt) {
21
+ if (harnessPrompt) {
22
+ return true;
23
+ }
24
+ return process.env.CUSTOM_MODEL_JSON_MODE === "1";
25
+ }
26
+ async function loadFileSnapshots(repoPath, relativePaths) {
27
+ const snapshots = [];
28
+ for (const relativePath of relativePaths) {
29
+ const absolutePath = join(repoPath, relativePath);
30
+ if (!(await fileExists(absolutePath))) {
31
+ continue;
32
+ }
33
+ snapshots.push({
34
+ path: relativePath,
35
+ content: await fs.readFile(absolutePath, "utf8"),
36
+ });
37
+ }
38
+ return snapshots;
39
+ }
40
+ async function applyEdits(repoPath, edits, bus) {
41
+ let applied = 0;
42
+ for (const edit of edits) {
43
+ const absolutePath = join(repoPath, edit.path);
44
+ if (!(await fileExists(absolutePath))) {
45
+ if (bus) {
46
+ emitEditEventsFromPatch(bus, {
47
+ path: edit.path,
48
+ before: "",
49
+ after: "",
50
+ applied: false,
51
+ errorMessage: `File not found: ${edit.path}`,
52
+ });
53
+ }
54
+ continue;
55
+ }
56
+ const original = await fs.readFile(absolutePath, "utf8");
57
+ let didApply = false;
58
+ let after = original;
59
+ if (edit.content !== undefined) {
60
+ if (edit.content !== original) {
61
+ await fs.writeFile(absolutePath, edit.content, "utf8");
62
+ applied += 1;
63
+ didApply = true;
64
+ after = edit.content;
65
+ }
66
+ if (bus) {
67
+ if (didApply) {
68
+ emitEditEventsFromPatch(bus, {
69
+ path: edit.path,
70
+ before: original,
71
+ after,
72
+ applied: true,
73
+ });
74
+ }
75
+ else {
76
+ emitEditEventsFromPatch(bus, {
77
+ path: edit.path,
78
+ before: original,
79
+ after,
80
+ applied: false,
81
+ errorMessage: "No change: content already matches",
82
+ });
83
+ }
84
+ }
85
+ continue;
86
+ }
87
+ if (edit.search &&
88
+ edit.replace !== undefined &&
89
+ original.includes(edit.search)) {
90
+ const updated = original.replace(edit.search, edit.replace);
91
+ if (updated !== original) {
92
+ await fs.writeFile(absolutePath, updated, "utf8");
93
+ applied += 1;
94
+ didApply = true;
95
+ after = updated;
96
+ }
97
+ }
98
+ if (bus) {
99
+ if (didApply) {
100
+ emitEditEventsFromPatch(bus, {
101
+ path: edit.path,
102
+ before: original,
103
+ after,
104
+ applied: true,
105
+ });
106
+ }
107
+ else {
108
+ emitEditEventsFromPatch(bus, {
109
+ path: edit.path,
110
+ before: original,
111
+ after: original,
112
+ applied: false,
113
+ errorMessage: edit.search
114
+ ? "search string not found or produced no change"
115
+ : "edit not applied",
116
+ });
117
+ }
118
+ }
119
+ }
120
+ return applied;
121
+ }
122
+ async function loadScreenshotDataUrl(screenshotPath) {
123
+ if (!screenshotPath || !(await fileExists(screenshotPath))) {
124
+ return undefined;
125
+ }
126
+ const buffer = await fs.readFile(screenshotPath);
127
+ const mime = screenshotPath.endsWith(".png") ? "image/png" : "image/jpeg";
128
+ return `data:${mime};base64,${buffer.toString("base64")}`;
129
+ }
130
+ function buildUserMessage(userPrompt, imageDataUrls) {
131
+ if (imageDataUrls.length === 0) {
132
+ return userPrompt;
133
+ }
134
+ return [
135
+ {
136
+ type: "text",
137
+ text: `${userPrompt}\n\nUse the attached reference image(s) for visual vibe, color palette, and mood.`,
138
+ },
139
+ ...imageDataUrls.map((url) => ({
140
+ type: "image_url",
141
+ image_url: { url },
142
+ })),
143
+ ];
144
+ }
145
+ function buildHarnessJsonRules(examplePath, hasScreenshot, pinnedTarget, pinned) {
146
+ const visionRule = hasScreenshot
147
+ ? pinnedTarget
148
+ ? `\n- Apply the attached reference image vibe to the pinned target (${pinnedTarget}).`
149
+ : "\n- Use the attached reference image for visual vibe when editing."
150
+ : pinnedTarget
151
+ ? `\n- Prioritize edits for the pinned target (${pinnedTarget}).`
152
+ : "";
153
+ const pinScopeRule = pinned?.sectionIndex != null && pinned.sectionTitle
154
+ ? `\n- Apply vibe/style changes ONLY to pinned section index ${pinned.sectionIndex} ("${pinned.sectionTitle}"). Do not edit other sections' custom components unless they are the pinned target.`
155
+ : "";
156
+ return `Return ONLY valid JSON with this shape:
157
+ {
158
+ "summary": "short summary",
159
+ "response": "user-facing response",
160
+ "edits": [
161
+ { "path": "${examplePath}", "search": "exact text to replace", "replace": "new text" }
162
+ ]
163
+ }
164
+
165
+ Rules:
166
+ - The "edits" array MUST contain at least one edit.
167
+ - Make the smallest safe edit that satisfies the user request.
168
+ - Use exact search strings copied from the provided files.${visionRule}${pinScopeRule}
169
+ - After two failed search/replace attempts you may use { "path": "${examplePath}", "content": "full file contents" }.
170
+ - Do not include markdown fences or commentary outside JSON.`;
171
+ }
172
+ function buildSystemPrompt(examplePath, hasScreenshot, pinnedTarget, pinned) {
173
+ return `You are Auvira Agent. ${buildHarnessJsonRules(examplePath, hasScreenshot, pinnedTarget, pinned)}
174
+ - For vibe/style requests, update CSS variables, Tailwind classes, inline styles, or section component markup as appropriate for the pinned target.`;
175
+ }
176
+ function formatRepositoryFiles(files) {
177
+ return files
178
+ .map((file) => `--- ${file.path} ---\n${file.content}`)
179
+ .join("\n\n");
180
+ }
181
+ function buildRepairMessage(input) {
182
+ const snippet = truncateModelOutput(input.priorOutput);
183
+ const pinnedRule = input.pinnedTarget
184
+ ? `\nEdits must target the pinned section: ${input.pinnedTarget}.`
185
+ : "";
186
+ const excerptBlock = input.pinnedSnippet
187
+ ? `\nPinned section excerpt (copy search strings exactly from here):\n${input.pinnedSnippet.slice(0, 4_000)}\n`
188
+ : "";
189
+ const fullFileRule = input.attempt >= input.maxAttempts
190
+ ? `\nYou may return a full-file replace: { "path": "${input.examplePath}", "content": "..." }`
191
+ : "";
192
+ return `Your previous response was invalid (${input.failure}).
193
+
194
+ Previous output snippet:
195
+ ${snippet}
196
+ ${excerptBlock}
197
+ Return ONLY valid JSON with a non-empty "edits" array. Use exact search strings from the repository files provided earlier.${pinnedRule}${fullFileRule}
198
+
199
+ Required shape example:
200
+ {
201
+ "summary": "Applied style update",
202
+ "response": "Updated colors",
203
+ "edits": [
204
+ { "path": "${input.examplePath}", "search": "exact existing text", "replace": "new text" }
205
+ ]
206
+ }`;
207
+ }
208
+ function accumulateTiming(current, next) {
209
+ return {
210
+ queueWaitMs: current.queueWaitMs + next.queueWaitMs,
211
+ requestMs: current.requestMs + next.requestMs,
212
+ totalMs: current.totalMs + next.totalMs,
213
+ retries: current.retries + next.retries,
214
+ };
215
+ }
216
+ /** Calls a custom OpenAI-compatible model API and applies returned file edits. */
217
+ export async function runCustomModelProvider(input) {
218
+ const model = input.model;
219
+ const bus = input.events;
220
+ const baseURL = model.baseURL?.replace(/\/+$/, "");
221
+ const apiKey = model.apiKey;
222
+ if (!baseURL || !apiKey) {
223
+ return {
224
+ ok: false,
225
+ response: "",
226
+ summary: "Custom model provider requires baseURL and apiKey",
227
+ error: {
228
+ message: "model.baseURL and model.apiKey are required for custom provider",
229
+ code: "PROVIDER_ERROR",
230
+ },
231
+ };
232
+ }
233
+ const harnessPrompt = input.harnessPrompt === true;
234
+ const pinnedTargetStruct = harnessPrompt
235
+ ? mergePinnedTarget(harnessTargetFromDom(input.selectedDom), extractPinnedTargetFromHarness(input.prompt))
236
+ : harnessTargetFromDom(input.selectedDom);
237
+ const discoveredPaths = harnessPrompt
238
+ ? await discoverMergedHarnessPaths(input.repoPath, input.prompt, input.selectedDom, pinnedTargetStruct)
239
+ : await discoverTargetFiles(input.repoPath, input.selectedDom);
240
+ const fileSnapshots = await loadFileSnapshots(input.repoPath, discoveredPaths);
241
+ const pinnedSnippets = harnessPrompt
242
+ ? await buildPinnedSectionSnapshots(input.repoPath, pinnedTargetStruct)
243
+ : [];
244
+ const allSnapshots = [...pinnedSnippets, ...fileSnapshots];
245
+ const screenshotDataUrl = await loadScreenshotDataUrl(input.screenshotPath);
246
+ const imageDataUrls = input.images && input.images.length > 0
247
+ ? imagesToDataUrls(input.images)
248
+ : screenshotDataUrl
249
+ ? [screenshotDataUrl]
250
+ : [];
251
+ const hasVision = imageDataUrls.length > 0;
252
+ const pinnedLabel = harnessPrompt
253
+ ? formatPinnedTargetLabel(pinnedTargetStruct) ??
254
+ describePinnedTarget(input.selectedDom)
255
+ : describePinnedTarget(input.selectedDom);
256
+ const selectedDomForFiles = input.selectedDom
257
+ ? {
258
+ ...input.selectedDom,
259
+ componentHint: input.selectedDom.componentHint ?? pinnedTargetStruct?.componentHint,
260
+ text: input.selectedDom.text ?? pinnedTargetStruct?.sectionTitle,
261
+ sectionIndex: input.selectedDom.sectionIndex ?? pinnedTargetStruct?.sectionIndex,
262
+ sectionType: input.selectedDom.sectionType ?? pinnedTargetStruct?.sectionType,
263
+ sectionTitle: input.selectedDom.sectionTitle ?? pinnedTargetStruct?.sectionTitle,
264
+ kind: input.selectedDom.kind ?? pinnedTargetStruct?.kind,
265
+ }
266
+ : pinnedTargetStruct
267
+ ? {
268
+ componentHint: pinnedTargetStruct.componentHint,
269
+ text: pinnedTargetStruct.sectionTitle,
270
+ sectionIndex: pinnedTargetStruct.sectionIndex,
271
+ sectionType: pinnedTargetStruct.sectionType,
272
+ sectionTitle: pinnedTargetStruct.sectionTitle,
273
+ kind: pinnedTargetStruct.kind,
274
+ }
275
+ : undefined;
276
+ const stylePrompt = hasVision || /vibe|style|background|mood/i.test(input.prompt);
277
+ const promptFiles = selectPromptFiles({
278
+ prompt: input.prompt,
279
+ files: allSnapshots,
280
+ hasScreenshot: hasVision,
281
+ selectedDom: selectedDomForFiles,
282
+ pinnedTarget: pinnedTargetStruct,
283
+ harnessMode: harnessPrompt,
284
+ });
285
+ bus?.emit({
286
+ type: "context.files_selected",
287
+ files: promptFiles.map((file) => file.path),
288
+ });
289
+ const examplePath = pickExampleEditPath(promptFiles, selectedDomForFiles, stylePrompt);
290
+ const pinnedSnippet = promptFiles.find((file) => file.path.startsWith("__harness__/"))?.content;
291
+ let systemPrompt;
292
+ let userPrompt;
293
+ if (harnessPrompt) {
294
+ const { system, user } = splitHarnessPrompt(input.prompt);
295
+ const jsonRules = buildHarnessJsonRules(examplePath, hasVision, pinnedLabel, pinnedTargetStruct);
296
+ systemPrompt = system.trim()
297
+ ? `${system}\n\n${jsonRules}`
298
+ : buildSystemPrompt(examplePath, hasVision, pinnedLabel, pinnedTargetStruct);
299
+ userPrompt = buildHarnessUserPrompt(user, promptFiles);
300
+ }
301
+ else {
302
+ systemPrompt = buildSystemPrompt(examplePath, hasVision, pinnedLabel, pinnedTargetStruct);
303
+ userPrompt = `${input.prompt}
304
+
305
+ Repository files:
306
+ ${formatRepositoryFiles(promptFiles)}`;
307
+ }
308
+ const messages = [{ role: "system", content: systemPrompt }];
309
+ for (const turn of input.conversationHistory ?? []) {
310
+ if (turn.role === "user" || turn.role === "assistant") {
311
+ messages.push({ role: turn.role, content: turn.content });
312
+ }
313
+ }
314
+ messages.push({
315
+ role: "user",
316
+ content: buildUserMessage(userPrompt, imageDataUrls),
317
+ });
318
+ const maxAttempts = getMaxEditAttempts();
319
+ let timing = {
320
+ queueWaitMs: 0,
321
+ requestMs: 0,
322
+ totalMs: 0,
323
+ retries: 0,
324
+ };
325
+ let lastContent = "";
326
+ let lastFailure = "MISSING_EDITS";
327
+ let lastPlan;
328
+ let lastUsage;
329
+ for (let attempt = 1; attempt <= maxAttempts; attempt += 1) {
330
+ throwIfCancelled(input.signal);
331
+ bus?.emit({
332
+ type: "model.attempt",
333
+ attempt,
334
+ maxAttempts,
335
+ });
336
+ const requestBody = {
337
+ model: model.id,
338
+ messages,
339
+ temperature: 0.2,
340
+ };
341
+ applyCustomModelRequestDefaults(requestBody, {
342
+ profile: "json",
343
+ modelId: model.id,
344
+ });
345
+ if (isJsonModeEnabled(harnessPrompt)) {
346
+ requestBody.response_format = { type: "json_object" };
347
+ }
348
+ const callModel = async () => {
349
+ bus?.emit({
350
+ type: "model.request",
351
+ provider: "custom",
352
+ model: model.id,
353
+ hasVision,
354
+ });
355
+ const { value: result } = await withApiConcurrencyLimit((queueWaitMs) => postJsonWithRetry({
356
+ url: `${baseURL}/chat/completions`,
357
+ headers: {
358
+ Authorization: `Bearer ${apiKey}`,
359
+ "Content-Type": "application/json",
360
+ },
361
+ body: requestBody,
362
+ timeoutMs: input.timeoutMs ?? 300_000,
363
+ signal: input.signal,
364
+ }, queueWaitMs));
365
+ if (result.ok && result.data) {
366
+ const content = result.data.choices?.[0]?.message?.content?.trim() ?? "";
367
+ bus?.emit({
368
+ type: "model.response",
369
+ contentPreview: truncateModelOutput(content),
370
+ timingMs: result.totalMs,
371
+ httpRetries: result.retries,
372
+ });
373
+ }
374
+ return result;
375
+ };
376
+ const apiResult = bus
377
+ ? (await bus.withTool("model_completion", { attempt, model: model.id }, callModel))
378
+ : await callModel();
379
+ throwIfCancelled(input.signal);
380
+ timing = accumulateTiming(timing, {
381
+ queueWaitMs: apiResult.queueWaitMs,
382
+ requestMs: apiResult.requestMs,
383
+ totalMs: apiResult.totalMs,
384
+ retries: apiResult.retries,
385
+ });
386
+ if (timing.totalMs > 60_000) {
387
+ console.warn(`[auvira-agent] Slow model API call: ${formatTimingReport(timing)}`);
388
+ }
389
+ if (!apiResult.ok || !apiResult.data) {
390
+ const rateLimit = apiResult.rateLimit;
391
+ const message = rateLimit
392
+ ? `${formatRateLimitReport(rateLimit)} | ${formatTimingReport(timing)} | attempts=${attempt}`
393
+ : `${apiResult.errorMessage ?? "Model API request failed"} | ${formatTimingReport(timing)} | attempts=${attempt}`;
394
+ return {
395
+ ok: false,
396
+ response: "",
397
+ summary: rateLimit ? "MiniMax rate limit exceeded" : "Custom model API request failed",
398
+ error: {
399
+ message,
400
+ code: rateLimit ? "RATE_LIMIT" : "PROVIDER_ERROR",
401
+ rateLimit: rateLimit
402
+ ? {
403
+ httpStatus: rateLimit.httpStatus,
404
+ providerStatusCode: rateLimit.providerStatusCode,
405
+ retryAfterMs: rateLimit.retryAfterMs,
406
+ headers: rateLimit.headers,
407
+ }
408
+ : undefined,
409
+ timing,
410
+ attempts: attempt,
411
+ lastFailure: rateLimit ? "RATE_LIMIT" : "PROVIDER_ERROR",
412
+ },
413
+ };
414
+ }
415
+ const payload = apiResult.data;
416
+ lastUsage = payload.usage;
417
+ if (!payload.choices?.length) {
418
+ lastFailure = "MISSING_EDITS";
419
+ if (attempt < maxAttempts) {
420
+ bus?.emit({
421
+ type: "model.repair",
422
+ reason: lastFailure,
423
+ priorOutputSnippet: "",
424
+ });
425
+ messages.push({
426
+ role: "user",
427
+ content: buildRepairMessage({
428
+ failure: lastFailure,
429
+ priorOutput: "",
430
+ examplePath,
431
+ pinnedTarget: pinnedLabel,
432
+ attempt,
433
+ maxAttempts,
434
+ pinnedSnippet,
435
+ }),
436
+ });
437
+ continue;
438
+ }
439
+ break;
440
+ }
441
+ lastContent = payload.choices[0]?.message?.content?.trim() ?? "";
442
+ messages.push({ role: "assistant", content: lastContent });
443
+ const parsed = parseModelEditPlan(lastContent);
444
+ if (!parsed.plan) {
445
+ lastFailure = parsed.failure ?? "PARSE_FAILED";
446
+ if (!lastContent.trim().startsWith("{")) {
447
+ bus?.emit({
448
+ type: "thinking",
449
+ text: truncateModelOutput(lastContent),
450
+ });
451
+ }
452
+ if (attempt < maxAttempts) {
453
+ bus?.emit({
454
+ type: "model.repair",
455
+ reason: lastFailure,
456
+ priorOutputSnippet: truncateModelOutput(lastContent),
457
+ });
458
+ messages.push({
459
+ role: "user",
460
+ content: buildRepairMessage({
461
+ failure: lastFailure,
462
+ priorOutput: lastContent,
463
+ examplePath,
464
+ pinnedTarget: pinnedLabel,
465
+ attempt,
466
+ maxAttempts,
467
+ pinnedSnippet,
468
+ }),
469
+ });
470
+ continue;
471
+ }
472
+ break;
473
+ }
474
+ lastPlan = parsed.plan;
475
+ bus?.emit({
476
+ type: "edit.parsed",
477
+ editCount: parsed.plan.edits.length,
478
+ paths: parsed.plan.edits.map((edit) => edit.path),
479
+ });
480
+ const assistantText = parsed.plan.response ?? parsed.plan.summary ?? "Applied model edits";
481
+ bus?.emit(buildAssistantEvent(assistantText));
482
+ const applied = bus
483
+ ? await bus.withTool("apply_file_edit", { editCount: parsed.plan.edits.length }, () => applyEdits(input.repoPath, parsed.plan.edits, bus))
484
+ : await applyEdits(input.repoPath, parsed.plan.edits, bus);
485
+ if (applied > 0) {
486
+ return {
487
+ ok: true,
488
+ response: parsed.plan.response ?? parsed.plan.summary ?? "Applied model edits",
489
+ summary: parsed.plan.summary ?? "Applied model edits",
490
+ usage: {
491
+ model: model.id,
492
+ provider: "custom",
493
+ inputTokens: lastUsage?.prompt_tokens,
494
+ outputTokens: lastUsage?.completion_tokens,
495
+ totalTokens: lastUsage?.total_tokens,
496
+ },
497
+ };
498
+ }
499
+ lastFailure = "EDIT_NOT_APPLIED";
500
+ if (attempt < maxAttempts) {
501
+ bus?.emit({
502
+ type: "model.repair",
503
+ reason: lastFailure,
504
+ priorOutputSnippet: truncateModelOutput(lastContent),
505
+ });
506
+ messages.push({
507
+ role: "user",
508
+ content: buildRepairMessage({
509
+ failure: lastFailure,
510
+ priorOutput: lastContent,
511
+ examplePath,
512
+ pinnedTarget: pinnedLabel,
513
+ attempt,
514
+ maxAttempts,
515
+ pinnedSnippet,
516
+ }),
517
+ });
518
+ }
519
+ }
520
+ const isApplyFailure = lastFailure === "EDIT_NOT_APPLIED";
521
+ const code = isApplyFailure ? "EDIT_APPLY_FAILED" : "INVALID_MODEL_OUTPUT";
522
+ return {
523
+ ok: false,
524
+ response: lastContent,
525
+ summary: isApplyFailure
526
+ ? "Model returned edits but none could be applied"
527
+ : "Model did not return valid edit JSON",
528
+ error: {
529
+ message: `${isApplyFailure ? "No edits applied to repository files" : "Model response missing valid edits"} | attempts=${maxAttempts} | lastFailure=${lastFailure} | ${formatTimingReport(timing)}`,
530
+ code,
531
+ timing,
532
+ attempts: maxAttempts,
533
+ lastFailure,
534
+ lastOutputSnippet: truncateModelOutput(lastContent),
535
+ },
536
+ };
537
+ }
538
+ //# sourceMappingURL=CustomModelProvider.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"CustomModelProvider.js","sourceRoot":"","sources":["../../../src/providers/custom/CustomModelProvider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,SAAS,CAAC;AACzC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AACvD,OAAO,EAAE,mBAAmB,EAAE,MAAM,+BAA+B,CAAC;AACpE,OAAO,EAAE,kBAAkB,EAAE,MAAM,iCAAiC,CAAC;AAGrE,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,EAAE,uBAAuB,EAAE,MAAM,2BAA2B,CAAC;AACpE,OAAO,EACL,qBAAqB,EACrB,kBAAkB,EAClB,iBAAiB,GAClB,MAAM,8BAA8B,CAAC;AAEtC,OAAO,EAIL,kBAAkB,EAClB,mBAAmB,GACpB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,iBAAiB,EAAqB,MAAM,wBAAwB,CAAC;AAC9E,OAAO,EACL,oBAAoB,EACpB,mBAAmB,EACnB,mBAAmB,GACpB,MAAM,0BAA0B,CAAC;AAElC,OAAO,EACL,sBAAsB,EACtB,2BAA2B,EAC3B,0BAA0B,EAC1B,8BAA8B,EAC9B,uBAAuB,EACvB,oBAAoB,EACpB,iBAAiB,GAClB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,EAAE,+BAA+B,EAAE,MAAM,2BAA2B,CAAC;AAC5E,OAAO,EAAE,uBAAuB,EAAE,MAAM,2BAA2B,CAAC;AAoCpE,SAAS,kBAAkB;IACzB,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,8BAA8B,IAAI,GAAG,CAAC,CAAC;IACzE,OAAO,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;AAC5D,CAAC;AAED,SAAS,iBAAiB,CAAC,aAAsB;IAC/C,IAAI,aAAa,EAAE,CAAC;QAClB,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,OAAO,CAAC,GAAG,CAAC,sBAAsB,KAAK,GAAG,CAAC;AACpD,CAAC;AAED,KAAK,UAAU,iBAAiB,CAC9B,QAAgB,EAChB,aAAuB;IAEvB,MAAM,SAAS,GAAmB,EAAE,CAAC;IAErC,KAAK,MAAM,YAAY,IAAI,aAAa,EAAE,CAAC;QACzC,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;QAClD,IAAI,CAAC,CAAC,MAAM,UAAU,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC;YACtC,SAAS;QACX,CAAC;QAED,SAAS,CAAC,IAAI,CAAC;YACb,IAAI,EAAE,YAAY;YAClB,OAAO,EAAE,MAAM,EAAE,CAAC,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;SACjD,CAAC,CAAC;IACL,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,KAAK,UAAU,UAAU,CACvB,QAAgB,EAChB,KAAkB,EAClB,GAAiB;IAEjB,IAAI,OAAO,GAAG,CAAC,CAAC;IAEhB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/C,IAAI,CAAC,CAAC,MAAM,UAAU,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC;YACtC,IAAI,GAAG,EAAE,CAAC;gBACR,uBAAuB,CAAC,GAAG,EAAE;oBAC3B,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,MAAM,EAAE,EAAE;oBACV,KAAK,EAAE,EAAE;oBACT,OAAO,EAAE,KAAK;oBACd,YAAY,EAAE,mBAAmB,IAAI,CAAC,IAAI,EAAE;iBAC7C,CAAC,CAAC;YACL,CAAC;YACD,SAAS;QACX,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QACzD,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,IAAI,KAAK,GAAG,QAAQ,CAAC;QAErB,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YAC/B,IAAI,IAAI,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;gBAC9B,MAAM,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;gBACvD,OAAO,IAAI,CAAC,CAAC;gBACb,QAAQ,GAAG,IAAI,CAAC;gBAChB,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC;YACvB,CAAC;YACD,IAAI,GAAG,EAAE,CAAC;gBACR,IAAI,QAAQ,EAAE,CAAC;oBACb,uBAAuB,CAAC,GAAG,EAAE;wBAC3B,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,MAAM,EAAE,QAAQ;wBAChB,KAAK;wBACL,OAAO,EAAE,IAAI;qBACd,CAAC,CAAC;gBACL,CAAC;qBAAM,CAAC;oBACN,uBAAuB,CAAC,GAAG,EAAE;wBAC3B,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,MAAM,EAAE,QAAQ;wBAChB,KAAK;wBACL,OAAO,EAAE,KAAK;wBACd,YAAY,EAAE,oCAAoC;qBACnD,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YACD,SAAS;QACX,CAAC;QAED,IACE,IAAI,CAAC,MAAM;YACX,IAAI,CAAC,OAAO,KAAK,SAAS;YAC1B,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAC9B,CAAC;YACD,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;YAC5D,IAAI,OAAO,KAAK,QAAQ,EAAE,CAAC;gBACzB,MAAM,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;gBAClD,OAAO,IAAI,CAAC,CAAC;gBACb,QAAQ,GAAG,IAAI,CAAC;gBAChB,KAAK,GAAG,OAAO,CAAC;YAClB,CAAC;QACH,CAAC;QAED,IAAI,GAAG,EAAE,CAAC;YACR,IAAI,QAAQ,EAAE,CAAC;gBACb,uBAAuB,CAAC,GAAG,EAAE;oBAC3B,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,MAAM,EAAE,QAAQ;oBAChB,KAAK;oBACL,OAAO,EAAE,IAAI;iBACd,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,uBAAuB,CAAC,GAAG,EAAE;oBAC3B,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,MAAM,EAAE,QAAQ;oBAChB,KAAK,EAAE,QAAQ;oBACf,OAAO,EAAE,KAAK;oBACd,YAAY,EAAE,IAAI,CAAC,MAAM;wBACvB,CAAC,CAAC,+CAA+C;wBACjD,CAAC,CAAC,kBAAkB;iBACvB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,KAAK,UAAU,qBAAqB,CAClC,cAAuB;IAEvB,IAAI,CAAC,cAAc,IAAI,CAAC,CAAC,MAAM,UAAU,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC;QAC3D,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;IACjD,MAAM,IAAI,GAAG,cAAc,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,YAAY,CAAC;IAC1E,OAAO,QAAQ,IAAI,WAAW,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;AAC5D,CAAC;AAED,SAAS,gBAAgB,CACvB,UAAkB,EAClB,aAAuB;IAEvB,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/B,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,OAAO;QACL;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,GAAG,UAAU,mFAAmF;SACvG;QACD,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YAC7B,IAAI,EAAE,WAAoB;YAC1B,SAAS,EAAE,EAAE,GAAG,EAAE;SACnB,CAAC,CAAC;KACJ,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAC5B,WAAmB,EACnB,aAAsB,EACtB,YAAqB,EACrB,MAA4B;IAE5B,MAAM,UAAU,GAAG,aAAa;QAC9B,CAAC,CAAC,YAAY;YACZ,CAAC,CAAC,qEAAqE,YAAY,IAAI;YACvF,CAAC,CAAC,oEAAoE;QACxE,CAAC,CAAC,YAAY;YACZ,CAAC,CAAC,+CAA+C,YAAY,IAAI;YACjE,CAAC,CAAC,EAAE,CAAC;IAET,MAAM,YAAY,GAChB,MAAM,EAAE,YAAY,IAAI,IAAI,IAAI,MAAM,CAAC,YAAY;QACjD,CAAC,CAAC,6DAA6D,MAAM,CAAC,YAAY,MAAM,MAAM,CAAC,YAAY,sFAAsF;QACjM,CAAC,CAAC,EAAE,CAAC;IAET,OAAO;;;;;iBAKQ,WAAW;;;;;;;4DAOgC,UAAU,GAAG,YAAY;oEACjB,WAAW;6DAClB,CAAC;AAC9D,CAAC;AAED,SAAS,iBAAiB,CACxB,WAAmB,EACnB,aAAsB,EACtB,YAAqB,EACrB,MAA4B;IAE5B,OAAO,yBAAyB,qBAAqB,CAAC,WAAW,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,CAAC;oJAC2C,CAAC;AACrJ,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAqB;IAClD,OAAO,KAAK;SACT,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,IAAI,CAAC,IAAI,SAAS,IAAI,CAAC,OAAO,EAAE,CAAC;SACtD,IAAI,CAAC,MAAM,CAAC,CAAC;AAClB,CAAC;AAED,SAAS,kBAAkB,CAAC,KAQ3B;IACC,MAAM,OAAO,GAAG,mBAAmB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;IACvD,MAAM,UAAU,GAAG,KAAK,CAAC,YAAY;QACnC,CAAC,CAAC,2CAA2C,KAAK,CAAC,YAAY,GAAG;QAClE,CAAC,CAAC,EAAE,CAAC;IAEP,MAAM,YAAY,GAAG,KAAK,CAAC,aAAa;QACtC,CAAC,CAAC,sEAAsE,KAAK,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,IAAI;QAC/G,CAAC,CAAC,EAAE,CAAC;IAEP,MAAM,YAAY,GAChB,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,WAAW;QAChC,CAAC,CAAC,oDAAoD,KAAK,CAAC,WAAW,uBAAuB;QAC9F,CAAC,CAAC,EAAE,CAAC;IAET,OAAO,uCAAuC,KAAK,CAAC,OAAO;;;EAG3D,OAAO;EACP,YAAY;6HAC+G,UAAU,GAAG,YAAY;;;;;;;iBAOrI,KAAK,CAAC,WAAW;;EAEhC,CAAC;AACH,CAAC;AAED,SAAS,gBAAgB,CACvB,OAAsB,EACtB,IAAmB;IAEnB,OAAO;QACL,WAAW,EAAE,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW;QACnD,SAAS,EAAE,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS;QAC7C,OAAO,EAAE,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO;QACvC,OAAO,EAAE,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO;KACxC,CAAC;AACJ,CAAC;AAED,kFAAkF;AAClF,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,KAAuB;IAEvB,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;IAC1B,MAAM,GAAG,GAAG,KAAK,CAAC,MAAM,CAAC;IACzB,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACnD,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;IAE5B,IAAI,CAAC,OAAO,IAAI,CAAC,MAAM,EAAE,CAAC;QACxB,OAAO;YACL,EAAE,EAAE,KAAK;YACT,QAAQ,EAAE,EAAE;YACZ,OAAO,EAAE,mDAAmD;YAC5D,KAAK,EAAE;gBACL,OAAO,EAAE,iEAAiE;gBAC1E,IAAI,EAAE,gBAAgB;aACvB;SACF,CAAC;IACJ,CAAC;IAED,MAAM,aAAa,GAAG,KAAK,CAAC,aAAa,KAAK,IAAI,CAAC;IACnD,MAAM,kBAAkB,GAAG,aAAa;QACtC,CAAC,CAAC,iBAAiB,CACf,oBAAoB,CAAC,KAAK,CAAC,WAAW,CAAC,EACvC,8BAA8B,CAAC,KAAK,CAAC,MAAM,CAAC,CAC7C;QACH,CAAC,CAAC,oBAAoB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;IAE5C,MAAM,eAAe,GAAG,aAAa;QACnC,CAAC,CAAC,MAAM,0BAA0B,CAC9B,KAAK,CAAC,QAAQ,EACd,KAAK,CAAC,MAAM,EACZ,KAAK,CAAC,WAAW,EACjB,kBAAkB,CACnB;QACH,CAAC,CAAC,MAAM,mBAAmB,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,WAAW,CAAC,CAAC;IAEjE,MAAM,aAAa,GAAG,MAAM,iBAAiB,CAAC,KAAK,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;IAC/E,MAAM,cAAc,GAAG,aAAa;QAClC,CAAC,CAAC,MAAM,2BAA2B,CAAC,KAAK,CAAC,QAAQ,EAAE,kBAAkB,CAAC;QACvE,CAAC,CAAC,EAAE,CAAC;IACP,MAAM,YAAY,GAAG,CAAC,GAAG,cAAc,EAAE,GAAG,aAAa,CAAC,CAAC;IAE3D,MAAM,iBAAiB,GAAG,MAAM,qBAAqB,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IAC5E,MAAM,aAAa,GACjB,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC;QACrC,CAAC,CAAC,gBAAgB,CAAC,KAAK,CAAC,MAAM,CAAC;QAChC,CAAC,CAAC,iBAAiB;YACjB,CAAC,CAAC,CAAC,iBAAiB,CAAC;YACrB,CAAC,CAAC,EAAE,CAAC;IACX,MAAM,SAAS,GAAG,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC;IAC3C,MAAM,WAAW,GAAG,aAAa;QAC/B,CAAC,CAAC,uBAAuB,CAAC,kBAAkB,CAAC;YAC3C,oBAAoB,CAAC,KAAK,CAAC,WAAW,CAAC;QACzC,CAAC,CAAC,oBAAoB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;IAE5C,MAAM,mBAAmB,GAAmC,KAAK,CAAC,WAAW;QAC3E,CAAC,CAAC;YACE,GAAG,KAAK,CAAC,WAAW;YACpB,aAAa,EACX,KAAK,CAAC,WAAW,CAAC,aAAa,IAAI,kBAAkB,EAAE,aAAa;YACtE,IAAI,EAAE,KAAK,CAAC,WAAW,CAAC,IAAI,IAAI,kBAAkB,EAAE,YAAY;YAChE,YAAY,EACV,KAAK,CAAC,WAAW,CAAC,YAAY,IAAI,kBAAkB,EAAE,YAAY;YACpE,WAAW,EACT,KAAK,CAAC,WAAW,CAAC,WAAW,IAAI,kBAAkB,EAAE,WAAW;YAClE,YAAY,EACV,KAAK,CAAC,WAAW,CAAC,YAAY,IAAI,kBAAkB,EAAE,YAAY;YACpE,IAAI,EAAE,KAAK,CAAC,WAAW,CAAC,IAAI,IAAI,kBAAkB,EAAE,IAAI;SACzD;QACH,CAAC,CAAC,kBAAkB;YAClB,CAAC,CAAC;gBACE,aAAa,EAAE,kBAAkB,CAAC,aAAa;gBAC/C,IAAI,EAAE,kBAAkB,CAAC,YAAY;gBACrC,YAAY,EAAE,kBAAkB,CAAC,YAAY;gBAC7C,WAAW,EAAE,kBAAkB,CAAC,WAAW;gBAC3C,YAAY,EAAE,kBAAkB,CAAC,YAAY;gBAC7C,IAAI,EAAE,kBAAkB,CAAC,IAAI;aAC9B;YACH,CAAC,CAAC,SAAS,CAAC;IAEhB,MAAM,WAAW,GACf,SAAS,IAAI,6BAA6B,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAEhE,MAAM,WAAW,GAAG,iBAAiB,CAAC;QACpC,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,KAAK,EAAE,YAAY;QACnB,aAAa,EAAE,SAAS;QACxB,WAAW,EAAE,mBAAmB;QAChC,YAAY,EAAE,kBAAkB;QAChC,WAAW,EAAE,aAAa;KAC3B,CAAC,CAAC;IAEH,GAAG,EAAE,IAAI,CAAC;QACR,IAAI,EAAE,wBAAwB;QAC9B,KAAK,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;KAC5C,CAAC,CAAC;IAEH,MAAM,WAAW,GAAG,mBAAmB,CACrC,WAAW,EACX,mBAAmB,EACnB,WAAW,CACZ,CAAC;IAEF,MAAM,aAAa,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAC9C,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,CACrC,EAAE,OAAO,CAAC;IAEX,IAAI,YAAoB,CAAC;IACzB,IAAI,UAAkB,CAAC;IAEvB,IAAI,aAAa,EAAE,CAAC;QAClB,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,kBAAkB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAC1D,MAAM,SAAS,GAAG,qBAAqB,CACrC,WAAW,EACX,SAAS,EACT,WAAW,EACX,kBAAkB,CACnB,CAAC;QACF,YAAY,GAAG,MAAM,CAAC,IAAI,EAAE;YAC1B,CAAC,CAAC,GAAG,MAAM,OAAO,SAAS,EAAE;YAC7B,CAAC,CAAC,iBAAiB,CAAC,WAAW,EAAE,SAAS,EAAE,WAAW,EAAE,kBAAkB,CAAC,CAAC;QAC/E,UAAU,GAAG,sBAAsB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;IACzD,CAAC;SAAM,CAAC;QACN,YAAY,GAAG,iBAAiB,CAC9B,WAAW,EACX,SAAS,EACT,WAAW,EACX,kBAAkB,CACnB,CAAC;QACF,UAAU,GAAG,GAAG,KAAK,CAAC,MAAM;;;EAG9B,qBAAqB,CAAC,WAAW,CAAC,EAAE,CAAC;IACrC,CAAC;IAED,MAAM,QAAQ,GAAkB,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;IAE5E,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,mBAAmB,IAAI,EAAE,EAAE,CAAC;QACnD,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACtD,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,IAAI,CAAC;QACZ,IAAI,EAAE,MAAM;QACZ,OAAO,EAAE,gBAAgB,CAAC,UAAU,EAAE,aAAa,CAAC;KACrD,CAAC,CAAC;IAEH,MAAM,WAAW,GAAG,kBAAkB,EAAE,CAAC;IACzC,IAAI,MAAM,GAAkB;QAC1B,WAAW,EAAE,CAAC;QACd,SAAS,EAAE,CAAC;QACZ,OAAO,EAAE,CAAC;QACV,OAAO,EAAE,CAAC;KACX,CAAC;IAEF,IAAI,WAAW,GAAG,EAAE,CAAC;IACrB,IAAI,WAAW,GAAoC,eAAe,CAAC;IACnE,IAAI,QAAmC,CAAC;IACxC,IAAI,SAA0C,CAAC;IAE/C,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,WAAW,EAAE,OAAO,IAAI,CAAC,EAAE,CAAC;QAC3D,gBAAgB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAE/B,GAAG,EAAE,IAAI,CAAC;YACR,IAAI,EAAE,eAAe;YACrB,OAAO;YACP,WAAW;SACZ,CAAC,CAAC;QAEH,MAAM,WAAW,GAA4B;YAC3C,KAAK,EAAE,KAAK,CAAC,EAAE;YACf,QAAQ;YACR,WAAW,EAAE,GAAG;SACjB,CAAC;QACF,+BAA+B,CAAC,WAAW,EAAE;YAC3C,OAAO,EAAE,MAAM;YACf,OAAO,EAAE,KAAK,CAAC,EAAE;SAClB,CAAC,CAAC;QAEH,IAAI,iBAAiB,CAAC,aAAa,CAAC,EAAE,CAAC;YACrC,WAAW,CAAC,eAAe,GAAG,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;QACxD,CAAC;QAED,MAAM,SAAS,GAAG,KAAK,IAAI,EAAE;YAC3B,GAAG,EAAE,IAAI,CAAC;gBACR,IAAI,EAAE,eAAe;gBACrB,QAAQ,EAAE,QAAQ;gBAClB,KAAK,EAAE,KAAK,CAAC,EAAE;gBACf,SAAS;aACV,CAAC,CAAC;YAEH,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,uBAAuB,CAAC,CAAC,WAAW,EAAE,EAAE,CACtE,iBAAiB,CACf;gBACE,GAAG,EAAE,GAAG,OAAO,mBAAmB;gBAClC,OAAO,EAAE;oBACP,aAAa,EAAE,UAAU,MAAM,EAAE;oBACjC,cAAc,EAAE,kBAAkB;iBACnC;gBACH,IAAI,EAAE,WAAW;gBACjB,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,OAAO;gBACrC,MAAM,EAAE,KAAK,CAAC,MAAM;aACrB,EACC,WAAW,CACZ,CACF,CAAC;YAEF,IAAI,MAAM,CAAC,EAAE,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC7B,MAAM,OAAO,GACX,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;gBAC3D,GAAG,EAAE,IAAI,CAAC;oBACR,IAAI,EAAE,gBAAgB;oBACtB,cAAc,EAAE,mBAAmB,CAAC,OAAO,CAAC;oBAC5C,QAAQ,EAAE,MAAM,CAAC,OAAO;oBACxB,WAAW,EAAE,MAAM,CAAC,OAAO;iBAC5B,CAAC,CAAC;YACL,CAAC;YAED,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC;QAEF,MAAM,SAAS,GAAG,GAAG;YACnB,CAAC,CAAC,CACE,MAAM,GAAG,CAAC,QAAQ,CAAC,kBAAkB,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,EAAE,EAAE,EAAE,SAAS,CAAC,CAChF;YACH,CAAC,CAAC,MAAM,SAAS,EAAE,CAAC;QAEtB,gBAAgB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAE/B,MAAM,GAAG,gBAAgB,CAAC,MAAM,EAAE;YAChC,WAAW,EAAE,SAAS,CAAC,WAAW;YAClC,SAAS,EAAE,SAAS,CAAC,SAAS;YAC9B,OAAO,EAAE,SAAS,CAAC,OAAO;YAC1B,OAAO,EAAE,SAAS,CAAC,OAAO;SAC3B,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,OAAO,GAAG,MAAM,EAAE,CAAC;YAC5B,OAAO,CAAC,IAAI,CACV,uCAAuC,kBAAkB,CAAC,MAAM,CAAC,EAAE,CACpE,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;YACrC,MAAM,SAAS,GAAG,SAAS,CAAC,SAAS,CAAC;YACtC,MAAM,OAAO,GAAG,SAAS;gBACvB,CAAC,CAAC,GAAG,qBAAqB,CAAC,SAAS,CAAC,MAAM,kBAAkB,CAAC,MAAM,CAAC,eAAe,OAAO,EAAE;gBAC7F,CAAC,CAAC,GAAG,SAAS,CAAC,YAAY,IAAI,0BAA0B,MAAM,kBAAkB,CAAC,MAAM,CAAC,eAAe,OAAO,EAAE,CAAC;YAEpH,OAAO;gBACL,EAAE,EAAE,KAAK;gBACT,QAAQ,EAAE,EAAE;gBACZ,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC,6BAA6B,CAAC,CAAC,CAAC,iCAAiC;gBACtF,KAAK,EAAE;oBACL,OAAO;oBACP,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,gBAAgB;oBACjD,SAAS,EAAE,SAAS;wBAClB,CAAC,CAAC;4BACE,UAAU,EAAE,SAAS,CAAC,UAAU;4BAChC,kBAAkB,EAAE,SAAS,CAAC,kBAAkB;4BAChD,YAAY,EAAE,SAAS,CAAC,YAAY;4BACpC,OAAO,EAAE,SAAS,CAAC,OAAO;yBAC3B;wBACH,CAAC,CAAC,SAAS;oBACb,MAAM;oBACN,QAAQ,EAAE,OAAO;oBACjB,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,gBAAgB;iBACzD;aACF,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,SAAS,CAAC,IAAI,CAAC;QAC/B,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC;QAE1B,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;YAC7B,WAAW,GAAG,eAAe,CAAC;YAC9B,IAAI,OAAO,GAAG,WAAW,EAAE,CAAC;gBAC1B,GAAG,EAAE,IAAI,CAAC;oBACR,IAAI,EAAE,cAAc;oBACpB,MAAM,EAAE,WAAW;oBACnB,kBAAkB,EAAE,EAAE;iBACvB,CAAC,CAAC;gBACH,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,kBAAkB,CAAC;wBAC1B,OAAO,EAAE,WAAW;wBACpB,WAAW,EAAE,EAAE;wBACf,WAAW;wBACX,YAAY,EAAE,WAAW;wBACzB,OAAO;wBACP,WAAW;wBACX,aAAa;qBACd,CAAC;iBACH,CAAC,CAAC;gBACH,SAAS;YACX,CAAC;YACD,MAAM;QACR,CAAC;QAED,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QACjE,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;QAE3D,MAAM,MAAM,GAAG,kBAAkB,CAAC,WAAW,CAAC,CAAC;QAC/C,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACjB,WAAW,GAAG,MAAM,CAAC,OAAO,IAAI,cAAc,CAAC;YAC/C,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gBACxC,GAAG,EAAE,IAAI,CAAC;oBACR,IAAI,EAAE,UAAU;oBAChB,IAAI,EAAE,mBAAmB,CAAC,WAAW,CAAC;iBACvC,CAAC,CAAC;YACL,CAAC;YACD,IAAI,OAAO,GAAG,WAAW,EAAE,CAAC;gBAC1B,GAAG,EAAE,IAAI,CAAC;oBACR,IAAI,EAAE,cAAc;oBACpB,MAAM,EAAE,WAAW;oBACnB,kBAAkB,EAAE,mBAAmB,CAAC,WAAW,CAAC;iBACrD,CAAC,CAAC;gBACH,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,kBAAkB,CAAC;wBAC1B,OAAO,EAAE,WAAW;wBACpB,WAAW,EAAE,WAAW;wBACxB,WAAW;wBACX,YAAY,EAAE,WAAW;wBACzB,OAAO;wBACP,WAAW;wBACX,aAAa;qBACd,CAAC;iBACH,CAAC,CAAC;gBACH,SAAS;YACX,CAAC;YACD,MAAM;QACR,CAAC;QAED,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC;QAEvB,GAAG,EAAE,IAAI,CAAC;YACR,IAAI,EAAE,aAAa;YACnB,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM;YACnC,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;SAClD,CAAC,CAAC;QAEH,MAAM,aAAa,GACjB,MAAM,CAAC,IAAI,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,qBAAqB,CAAC;QACvE,GAAG,EAAE,IAAI,CAAC,mBAAmB,CAAC,aAAa,CAAC,CAAC,CAAC;QAE9C,MAAM,OAAO,GAAG,GAAG;YACjB,CAAC,CAAC,MAAM,GAAG,CAAC,QAAQ,CAChB,iBAAiB,EACjB,EAAE,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,EACvC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,IAAK,CAAC,KAAK,EAAE,GAAG,CAAC,CAC1D;YACH,CAAC,CAAC,MAAM,UAAU,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAC7D,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;YAChB,OAAO;gBACL,EAAE,EAAE,IAAI;gBACR,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,qBAAqB;gBAC9E,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,qBAAqB;gBACrD,KAAK,EAAE;oBACL,KAAK,EAAE,KAAK,CAAC,EAAE;oBACf,QAAQ,EAAE,QAAQ;oBAClB,WAAW,EAAE,SAAS,EAAE,aAAa;oBACrC,YAAY,EAAE,SAAS,EAAE,iBAAiB;oBAC1C,WAAW,EAAE,SAAS,EAAE,YAAY;iBACrC;aACF,CAAC;QACJ,CAAC;QAED,WAAW,GAAG,kBAAkB,CAAC;QACjC,IAAI,OAAO,GAAG,WAAW,EAAE,CAAC;YAC1B,GAAG,EAAE,IAAI,CAAC;gBACR,IAAI,EAAE,cAAc;gBACpB,MAAM,EAAE,WAAW;gBACnB,kBAAkB,EAAE,mBAAmB,CAAC,WAAW,CAAC;aACrD,CAAC,CAAC;YACH,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,kBAAkB,CAAC;oBAC1B,OAAO,EAAE,WAAW;oBACpB,WAAW,EAAE,WAAW;oBACxB,WAAW;oBACX,YAAY,EAAE,WAAW;oBACzB,OAAO;oBACP,WAAW;oBACX,aAAa;iBACd,CAAC;aACH,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,MAAM,cAAc,GAAG,WAAW,KAAK,kBAAkB,CAAC;IAC1D,MAAM,IAAI,GAAG,cAAc,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,sBAAsB,CAAC;IAE3E,OAAO;QACL,EAAE,EAAE,KAAK;QACT,QAAQ,EAAE,WAAW;QACrB,OAAO,EAAE,cAAc;YACrB,CAAC,CAAC,gDAAgD;YAClD,CAAC,CAAC,sCAAsC;QAC1C,KAAK,EAAE;YACL,OAAO,EAAE,GAAG,cAAc,CAAC,CAAC,CAAC,sCAAsC,CAAC,CAAC,CAAC,oCAAoC,eAAe,WAAW,kBAAkB,WAAW,MAAM,kBAAkB,CAAC,MAAM,CAAC,EAAE;YACnM,IAAI;YACJ,MAAM;YACN,QAAQ,EAAE,WAAW;YACrB,WAAW;YACX,iBAAiB,EAAE,mBAAmB,CAAC,WAAW,CAAC;SACpD;KACF,CAAC;AACJ,CAAC"}
@@ -0,0 +1,16 @@
1
+ import type { SelectedDomContext } from "../../agent/types.js";
2
+ import type { HarnessPinnedTarget } from "./harnessPinTypes.js";
3
+ /** Common edit targets for Next.js / migration-site repos. */
4
+ export declare const TARGET_FILE_CANDIDATES: string[];
5
+ /** Extracts searchable tokens from a pinned component hint string. */
6
+ export declare function extractHintTokens(componentHint: string): string[];
7
+ /** Discovers repository files likely relevant to the user request and pinned DOM. */
8
+ export declare function discoverTargetFiles(repoPath: string, selectedDom?: SelectedDomContext): Promise<string[]>;
9
+ /** Picks the best example edit path for system/repair prompts. */
10
+ export declare function pickExampleEditPath(files: Array<{
11
+ path: string;
12
+ content?: string;
13
+ }>, selectedDom?: SelectedDomContext, stylePrompt?: boolean): string;
14
+ /** Describes the pinned edit target for vision/style prompts. */
15
+ export declare function describePinnedTarget(selectedDom?: SelectedDomContext | HarnessPinnedTarget): string | undefined;
16
+ //# sourceMappingURL=discoverTargetFiles.d.ts.map