@stonerzju/opencode 1.2.17 → 1.2.18

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 (261) hide show
  1. package/bin/opencode +29 -157
  2. package/package.json +29 -29
  3. package/src/acp/agent.ts +4 -4
  4. package/src/acp/session.ts +1 -1
  5. package/src/agent/agent.ts +3 -3
  6. package/src/bun/index.ts +2 -2
  7. package/src/cli/cmd/acp.ts +3 -3
  8. package/src/cli/cmd/debug/file.ts +1 -1
  9. package/src/cli/cmd/github.ts +2 -2
  10. package/src/cli/cmd/pr.ts +1 -1
  11. package/src/cli/cmd/tui/app.tsx +24 -24
  12. package/src/cli/cmd/tui/attach.ts +3 -3
  13. package/src/cli/cmd/tui/component/dialog-agent.tsx +3 -3
  14. package/src/cli/cmd/tui/component/dialog-command.tsx +3 -3
  15. package/src/cli/cmd/tui/component/dialog-mcp.tsx +5 -5
  16. package/src/cli/cmd/tui/component/dialog-model.tsx +4 -4
  17. package/src/cli/cmd/tui/component/dialog-provider.tsx +4 -4
  18. package/src/cli/cmd/tui/component/dialog-session-list.tsx +5 -5
  19. package/src/cli/cmd/tui/component/dialog-session-rename.tsx +3 -3
  20. package/src/cli/cmd/tui/component/dialog-skill.tsx +3 -3
  21. package/src/cli/cmd/tui/component/dialog-stash.tsx +3 -3
  22. package/src/cli/cmd/tui/component/dialog-status.tsx +2 -2
  23. package/src/cli/cmd/tui/component/dialog-tag.tsx +3 -3
  24. package/src/cli/cmd/tui/component/logo.tsx +2 -2
  25. package/src/cli/cmd/tui/component/prompt/autocomplete.tsx +6 -6
  26. package/src/cli/cmd/tui/component/prompt/frecency.tsx +2 -2
  27. package/src/cli/cmd/tui/component/prompt/history.tsx +2 -2
  28. package/src/cli/cmd/tui/component/prompt/index.tsx +14 -14
  29. package/src/cli/cmd/tui/component/prompt/stash.tsx +2 -2
  30. package/src/cli/cmd/tui/component/textarea-keybindings.ts +1 -1
  31. package/src/cli/cmd/tui/component/tips.tsx +1 -1
  32. package/src/cli/cmd/tui/context/directory.ts +1 -1
  33. package/src/cli/cmd/tui/context/exit.tsx +1 -1
  34. package/src/cli/cmd/tui/context/keybind.tsx +2 -2
  35. package/src/cli/cmd/tui/context/kv.tsx +2 -2
  36. package/src/cli/cmd/tui/context/local.tsx +6 -6
  37. package/src/cli/cmd/tui/context/sync.tsx +4 -4
  38. package/src/cli/cmd/tui/context/theme/opencode.json +245 -0
  39. package/src/cli/cmd/tui/context/theme.tsx +2 -2
  40. package/src/cli/cmd/tui/context/tui-config.tsx +1 -1
  41. package/src/cli/cmd/tui/event.ts +2 -2
  42. package/src/cli/cmd/tui/routes/home.tsx +6 -6
  43. package/src/cli/cmd/tui/routes/session/dialog-fork-from-timeline.tsx +6 -6
  44. package/src/cli/cmd/tui/routes/session/dialog-message.tsx +6 -6
  45. package/src/cli/cmd/tui/routes/session/dialog-subagent.tsx +2 -2
  46. package/src/cli/cmd/tui/routes/session/dialog-timeline.tsx +3 -3
  47. package/src/cli/cmd/tui/routes/session/header.tsx +5 -5
  48. package/src/cli/cmd/tui/routes/session/index.tsx +32 -32
  49. package/src/cli/cmd/tui/routes/session/permission.tsx +4 -4
  50. package/src/cli/cmd/tui/routes/session/sidebar.tsx +4 -4
  51. package/src/cli/cmd/tui/thread.ts +9 -9
  52. package/src/cli/cmd/tui/ui/dialog-confirm.tsx +1 -1
  53. package/src/cli/cmd/tui/ui/dialog-help.tsx +2 -2
  54. package/src/cli/cmd/tui/ui/dialog-select.tsx +5 -5
  55. package/src/cli/cmd/tui/ui/dialog.tsx +3 -3
  56. package/src/cli/cmd/tui/ui/toast.tsx +1 -1
  57. package/src/cli/cmd/tui/util/editor.ts +3 -3
  58. package/src/cli/cmd/tui/util/transcript.ts +1 -1
  59. package/src/cli/cmd/tui/worker.ts +10 -10
  60. package/src/cli/error.ts +1 -1
  61. package/src/cli/ui.ts +1 -1
  62. package/src/cli/upgrade.ts +4 -4
  63. package/src/command/index.ts +1 -1
  64. package/src/config/config.ts +10 -10
  65. package/src/config/markdown.ts +1 -1
  66. package/src/config/migrate-tui-config.ts +5 -5
  67. package/src/config/paths.ts +4 -4
  68. package/src/config/tui.ts +4 -4
  69. package/src/control/control.sql.ts +1 -1
  70. package/src/control/index.ts +1 -1
  71. package/src/control-plane/adaptors/worktree.ts +1 -1
  72. package/src/control-plane/session-proxy-middleware.ts +1 -1
  73. package/src/control-plane/workspace.sql.ts +1 -1
  74. package/src/control-plane/workspace.ts +7 -7
  75. package/src/file/index.ts +1 -1
  76. package/src/file/ripgrep.ts +2 -2
  77. package/src/file/watcher.ts +5 -5
  78. package/src/format/formatter.ts +1 -1
  79. package/src/ide/index.ts +3 -3
  80. package/src/index.ts +1 -1
  81. package/src/installation/index.ts +3 -3
  82. package/src/lsp/client.ts +3 -3
  83. package/src/lsp/index.ts +3 -3
  84. package/src/mcp/index.ts +4 -4
  85. package/src/permission/index.ts +2 -2
  86. package/src/permission/next.ts +10 -10
  87. package/src/plugin/codex.ts +1 -1
  88. package/src/plugin/copilot.ts +2 -2
  89. package/src/plugin/index.ts +1 -1
  90. package/src/project/bootstrap.ts +2 -2
  91. package/src/project/instance.ts +4 -4
  92. package/src/project/project.sql.ts +1 -1
  93. package/src/project/project.ts +5 -5
  94. package/src/project/state.ts +1 -1
  95. package/src/project/vcs.ts +4 -4
  96. package/src/provider/auth.ts +4 -4
  97. package/src/provider/error.ts +1 -1
  98. package/src/provider/models-snapshot.ts +2 -0
  99. package/src/provider/models.ts +1 -1
  100. package/src/provider/provider.ts +2 -2
  101. package/src/provider/transform.ts +2 -2
  102. package/src/pty/index.ts +5 -5
  103. package/src/question/index.ts +5 -5
  104. package/src/server/event.ts +1 -1
  105. package/src/server/mdns.ts +1 -1
  106. package/src/server/routes/global.ts +3 -3
  107. package/src/server/routes/permission.ts +1 -1
  108. package/src/server/routes/pty.ts +1 -1
  109. package/src/server/routes/session.ts +4 -4
  110. package/src/server/routes/tui.ts +1 -1
  111. package/src/server/server.ts +3 -3
  112. package/src/session/compaction.ts +7 -7
  113. package/src/session/index.ts +10 -10
  114. package/src/session/instruction.ts +1 -1
  115. package/src/session/llm.ts +11 -11
  116. package/src/session/message-v2.ts +10 -10
  117. package/src/session/message.ts +1 -1
  118. package/src/session/processor.ts +10 -10
  119. package/src/session/prompt.ts +8 -8
  120. package/src/session/retry.ts +2 -2
  121. package/src/session/revert.ts +1 -1
  122. package/src/session/session.sql.ts +3 -3
  123. package/src/session/status.ts +3 -3
  124. package/src/session/summary.ts +5 -5
  125. package/src/session/system.ts +1 -1
  126. package/src/session/todo.ts +2 -2
  127. package/src/share/share-next.ts +7 -7
  128. package/src/share/share.sql.ts +1 -1
  129. package/src/shell/shell.ts +3 -3
  130. package/src/skill/skill.ts +6 -6
  131. package/src/storage/db.ts +1 -1
  132. package/src/storage/storage.ts +1 -1
  133. package/src/tool/bash.ts +6 -6
  134. package/src/tool/edit.ts +1 -1
  135. package/src/tool/registry.ts +2 -2
  136. package/src/tool/skill.ts +1 -1
  137. package/src/tool/task.ts +3 -3
  138. package/src/util/array.ts +10 -0
  139. package/src/util/binary.ts +41 -0
  140. package/src/util/encode.ts +51 -0
  141. package/src/util/error.ts +54 -0
  142. package/src/util/identifier.ts +48 -0
  143. package/src/util/lazy.ts +4 -16
  144. package/src/util/path.ts +37 -0
  145. package/src/util/retry.ts +41 -0
  146. package/src/util/slug.ts +74 -0
  147. package/src/worktree/index.ts +3 -3
  148. package/AGENTS.md +0 -10
  149. package/BUN_SHELL_MIGRATION_PLAN.md +0 -136
  150. package/Dockerfile +0 -18
  151. package/README.md +0 -15
  152. package/bunfig.toml +0 -7
  153. package/drizzle.config.ts +0 -10
  154. package/script/build.ts +0 -224
  155. package/script/check-migrations.ts +0 -16
  156. package/script/postinstall.mjs +0 -131
  157. package/script/publish.ts +0 -181
  158. package/script/schema.ts +0 -63
  159. package/script/seed-e2e.ts +0 -50
  160. package/sst-env.d.ts +0 -10
  161. package/test/AGENTS.md +0 -81
  162. package/test/acp/agent-interface.test.ts +0 -51
  163. package/test/acp/event-subscription.test.ts +0 -683
  164. package/test/agent/agent.test.ts +0 -689
  165. package/test/bun.test.ts +0 -53
  166. package/test/cli/github-action.test.ts +0 -197
  167. package/test/cli/github-remote.test.ts +0 -80
  168. package/test/cli/import.test.ts +0 -38
  169. package/test/cli/plugin-auth-picker.test.ts +0 -120
  170. package/test/cli/tui/transcript.test.ts +0 -322
  171. package/test/config/agent-color.test.ts +0 -71
  172. package/test/config/config.test.ts +0 -1886
  173. package/test/config/fixtures/empty-frontmatter.md +0 -4
  174. package/test/config/fixtures/frontmatter.md +0 -28
  175. package/test/config/fixtures/markdown-header.md +0 -11
  176. package/test/config/fixtures/no-frontmatter.md +0 -1
  177. package/test/config/fixtures/weird-model-id.md +0 -13
  178. package/test/config/markdown.test.ts +0 -228
  179. package/test/config/tui.test.ts +0 -510
  180. package/test/control-plane/session-proxy-middleware.test.ts +0 -147
  181. package/test/control-plane/sse.test.ts +0 -56
  182. package/test/control-plane/workspace-server-sse.test.ts +0 -65
  183. package/test/control-plane/workspace-sync.test.ts +0 -97
  184. package/test/file/ignore.test.ts +0 -10
  185. package/test/file/index.test.ts +0 -394
  186. package/test/file/path-traversal.test.ts +0 -198
  187. package/test/file/ripgrep.test.ts +0 -39
  188. package/test/file/time.test.ts +0 -361
  189. package/test/fixture/db.ts +0 -11
  190. package/test/fixture/fixture.ts +0 -45
  191. package/test/fixture/lsp/fake-lsp-server.js +0 -77
  192. package/test/fixture/skills/agents-sdk/SKILL.md +0 -152
  193. package/test/fixture/skills/agents-sdk/references/callable.md +0 -92
  194. package/test/fixture/skills/cloudflare/SKILL.md +0 -211
  195. package/test/fixture/skills/index.json +0 -6
  196. package/test/ide/ide.test.ts +0 -82
  197. package/test/keybind.test.ts +0 -421
  198. package/test/lsp/client.test.ts +0 -95
  199. package/test/mcp/headers.test.ts +0 -153
  200. package/test/mcp/oauth-browser.test.ts +0 -249
  201. package/test/memory/abort-leak.test.ts +0 -136
  202. package/test/patch/patch.test.ts +0 -348
  203. package/test/permission/arity.test.ts +0 -33
  204. package/test/permission/next.test.ts +0 -689
  205. package/test/permission-task.test.ts +0 -319
  206. package/test/plugin/auth-override.test.ts +0 -44
  207. package/test/plugin/codex.test.ts +0 -123
  208. package/test/preload.ts +0 -80
  209. package/test/project/project.test.ts +0 -348
  210. package/test/project/worktree-remove.test.ts +0 -65
  211. package/test/provider/amazon-bedrock.test.ts +0 -446
  212. package/test/provider/copilot/convert-to-copilot-messages.test.ts +0 -523
  213. package/test/provider/copilot/copilot-chat-model.test.ts +0 -592
  214. package/test/provider/gitlab-duo.test.ts +0 -262
  215. package/test/provider/provider.test.ts +0 -2220
  216. package/test/provider/transform.test.ts +0 -2353
  217. package/test/pty/pty-output-isolation.test.ts +0 -140
  218. package/test/question/question.test.ts +0 -300
  219. package/test/scheduler.test.ts +0 -73
  220. package/test/server/global-session-list.test.ts +0 -89
  221. package/test/server/session-list.test.ts +0 -90
  222. package/test/server/session-select.test.ts +0 -78
  223. package/test/session/compaction.test.ts +0 -423
  224. package/test/session/instruction.test.ts +0 -170
  225. package/test/session/llm.test.ts +0 -667
  226. package/test/session/message-v2.test.ts +0 -924
  227. package/test/session/prompt.test.ts +0 -211
  228. package/test/session/retry.test.ts +0 -188
  229. package/test/session/revert-compact.test.ts +0 -285
  230. package/test/session/session.test.ts +0 -71
  231. package/test/session/structured-output-integration.test.ts +0 -233
  232. package/test/session/structured-output.test.ts +0 -385
  233. package/test/skill/discovery.test.ts +0 -110
  234. package/test/skill/skill.test.ts +0 -388
  235. package/test/snapshot/snapshot.test.ts +0 -1180
  236. package/test/storage/json-migration.test.ts +0 -846
  237. package/test/tool/__snapshots__/tool.test.ts.snap +0 -9
  238. package/test/tool/apply_patch.test.ts +0 -566
  239. package/test/tool/bash.test.ts +0 -402
  240. package/test/tool/edit.test.ts +0 -496
  241. package/test/tool/external-directory.test.ts +0 -127
  242. package/test/tool/fixtures/large-image.png +0 -0
  243. package/test/tool/fixtures/models-api.json +0 -38413
  244. package/test/tool/grep.test.ts +0 -110
  245. package/test/tool/question.test.ts +0 -107
  246. package/test/tool/read.test.ts +0 -504
  247. package/test/tool/registry.test.ts +0 -122
  248. package/test/tool/skill.test.ts +0 -112
  249. package/test/tool/truncation.test.ts +0 -160
  250. package/test/tool/webfetch.test.ts +0 -100
  251. package/test/tool/write.test.ts +0 -348
  252. package/test/util/filesystem.test.ts +0 -443
  253. package/test/util/format.test.ts +0 -59
  254. package/test/util/glob.test.ts +0 -164
  255. package/test/util/iife.test.ts +0 -36
  256. package/test/util/lazy.test.ts +0 -50
  257. package/test/util/lock.test.ts +0 -72
  258. package/test/util/process.test.ts +0 -59
  259. package/test/util/timeout.test.ts +0 -21
  260. package/test/util/wildcard.test.ts +0 -90
  261. package/tsconfig.json +0 -16
@@ -1,211 +0,0 @@
1
- import path from "path"
2
- import { describe, expect, test } from "bun:test"
3
- import { fileURLToPath } from "url"
4
- import { Instance } from "../../src/project/instance"
5
- import { Session } from "../../src/session"
6
- import { MessageV2 } from "../../src/session/message-v2"
7
- import { SessionPrompt } from "../../src/session/prompt"
8
- import { Log } from "../../src/util/log"
9
- import { tmpdir } from "../fixture/fixture"
10
-
11
- Log.init({ print: false })
12
-
13
- describe("session.prompt missing file", () => {
14
- test("does not fail the prompt when a file part is missing", async () => {
15
- await using tmp = await tmpdir({
16
- git: true,
17
- config: {
18
- agent: {
19
- build: {
20
- model: "openai/gpt-5.2",
21
- },
22
- },
23
- },
24
- })
25
-
26
- await Instance.provide({
27
- directory: tmp.path,
28
- fn: async () => {
29
- const session = await Session.create({})
30
-
31
- const missing = path.join(tmp.path, "does-not-exist.ts")
32
- const msg = await SessionPrompt.prompt({
33
- sessionID: session.id,
34
- agent: "build",
35
- noReply: true,
36
- parts: [
37
- { type: "text", text: "please review @does-not-exist.ts" },
38
- {
39
- type: "file",
40
- mime: "text/plain",
41
- url: `file://${missing}`,
42
- filename: "does-not-exist.ts",
43
- },
44
- ],
45
- })
46
-
47
- if (msg.info.role !== "user") throw new Error("expected user message")
48
-
49
- const hasFailure = msg.parts.some(
50
- (part) => part.type === "text" && part.synthetic && part.text.includes("Read tool failed to read"),
51
- )
52
- expect(hasFailure).toBe(true)
53
-
54
- await Session.remove(session.id)
55
- },
56
- })
57
- })
58
-
59
- test("keeps stored part order stable when file resolution is async", async () => {
60
- await using tmp = await tmpdir({
61
- git: true,
62
- config: {
63
- agent: {
64
- build: {
65
- model: "openai/gpt-5.2",
66
- },
67
- },
68
- },
69
- })
70
-
71
- await Instance.provide({
72
- directory: tmp.path,
73
- fn: async () => {
74
- const session = await Session.create({})
75
-
76
- const missing = path.join(tmp.path, "still-missing.ts")
77
- const msg = await SessionPrompt.prompt({
78
- sessionID: session.id,
79
- agent: "build",
80
- noReply: true,
81
- parts: [
82
- {
83
- type: "file",
84
- mime: "text/plain",
85
- url: `file://${missing}`,
86
- filename: "still-missing.ts",
87
- },
88
- { type: "text", text: "after-file" },
89
- ],
90
- })
91
-
92
- if (msg.info.role !== "user") throw new Error("expected user message")
93
-
94
- const stored = await MessageV2.get({
95
- sessionID: session.id,
96
- messageID: msg.info.id,
97
- })
98
- const text = stored.parts.filter((part) => part.type === "text").map((part) => part.text)
99
-
100
- expect(text[0]?.startsWith("Called the Read tool with the following input:")).toBe(true)
101
- expect(text[1]?.includes("Read tool failed to read")).toBe(true)
102
- expect(text[2]).toBe("after-file")
103
-
104
- await Session.remove(session.id)
105
- },
106
- })
107
- })
108
- })
109
-
110
- describe("session.prompt special characters", () => {
111
- test("handles filenames with # character", async () => {
112
- await using tmp = await tmpdir({
113
- git: true,
114
- init: async (dir) => {
115
- await Bun.write(path.join(dir, "file#name.txt"), "special content\n")
116
- },
117
- })
118
-
119
- await Instance.provide({
120
- directory: tmp.path,
121
- fn: async () => {
122
- const session = await Session.create({})
123
- const template = "Read @file#name.txt"
124
- const parts = await SessionPrompt.resolvePromptParts(template)
125
- const fileParts = parts.filter((part) => part.type === "file")
126
-
127
- expect(fileParts.length).toBe(1)
128
- expect(fileParts[0].filename).toBe("file#name.txt")
129
- expect(fileParts[0].url).toContain("%23")
130
-
131
- const decodedPath = fileURLToPath(fileParts[0].url)
132
- expect(decodedPath).toBe(path.join(tmp.path, "file#name.txt"))
133
-
134
- const message = await SessionPrompt.prompt({
135
- sessionID: session.id,
136
- parts,
137
- noReply: true,
138
- })
139
- const stored = await MessageV2.get({ sessionID: session.id, messageID: message.info.id })
140
- const textParts = stored.parts.filter((part) => part.type === "text")
141
- const hasContent = textParts.some((part) => part.text.includes("special content"))
142
- expect(hasContent).toBe(true)
143
-
144
- await Session.remove(session.id)
145
- },
146
- })
147
- })
148
- })
149
-
150
- describe("session.prompt agent variant", () => {
151
- test("applies agent variant only when using agent model", async () => {
152
- const prev = process.env.OPENAI_API_KEY
153
- process.env.OPENAI_API_KEY = "test-openai-key"
154
-
155
- try {
156
- await using tmp = await tmpdir({
157
- git: true,
158
- config: {
159
- agent: {
160
- build: {
161
- model: "openai/gpt-5.2",
162
- variant: "xhigh",
163
- },
164
- },
165
- },
166
- })
167
-
168
- await Instance.provide({
169
- directory: tmp.path,
170
- fn: async () => {
171
- const session = await Session.create({})
172
-
173
- const other = await SessionPrompt.prompt({
174
- sessionID: session.id,
175
- agent: "build",
176
- model: { providerID: "opencode", modelID: "kimi-k2.5-free" },
177
- noReply: true,
178
- parts: [{ type: "text", text: "hello" }],
179
- })
180
- if (other.info.role !== "user") throw new Error("expected user message")
181
- expect(other.info.variant).toBeUndefined()
182
-
183
- const match = await SessionPrompt.prompt({
184
- sessionID: session.id,
185
- agent: "build",
186
- noReply: true,
187
- parts: [{ type: "text", text: "hello again" }],
188
- })
189
- if (match.info.role !== "user") throw new Error("expected user message")
190
- expect(match.info.model).toEqual({ providerID: "openai", modelID: "gpt-5.2" })
191
- expect(match.info.variant).toBe("xhigh")
192
-
193
- const override = await SessionPrompt.prompt({
194
- sessionID: session.id,
195
- agent: "build",
196
- noReply: true,
197
- variant: "high",
198
- parts: [{ type: "text", text: "hello third" }],
199
- })
200
- if (override.info.role !== "user") throw new Error("expected user message")
201
- expect(override.info.variant).toBe("high")
202
-
203
- await Session.remove(session.id)
204
- },
205
- })
206
- } finally {
207
- if (prev === undefined) delete process.env.OPENAI_API_KEY
208
- else process.env.OPENAI_API_KEY = prev
209
- }
210
- })
211
- })
@@ -1,188 +0,0 @@
1
- import { describe, expect, test } from "bun:test"
2
- import type { NamedError } from "@opencode-ai/util/error"
3
- import { APICallError } from "ai"
4
- import { SessionRetry } from "../../src/session/retry"
5
- import { MessageV2 } from "../../src/session/message-v2"
6
-
7
- function apiError(headers?: Record<string, string>): MessageV2.APIError {
8
- return new MessageV2.APIError({
9
- message: "boom",
10
- isRetryable: true,
11
- responseHeaders: headers,
12
- }).toObject() as MessageV2.APIError
13
- }
14
-
15
- function wrap(message: unknown): ReturnType<NamedError["toObject"]> {
16
- return { data: { message } } as ReturnType<NamedError["toObject"]>
17
- }
18
-
19
- describe("session.retry.delay", () => {
20
- test("caps delay at 30 seconds when headers missing", () => {
21
- const error = apiError()
22
- const delays = Array.from({ length: 10 }, (_, index) => SessionRetry.delay(index + 1, error))
23
- expect(delays).toStrictEqual([2000, 4000, 8000, 16000, 30000, 30000, 30000, 30000, 30000, 30000])
24
- })
25
-
26
- test("prefers retry-after-ms when shorter than exponential", () => {
27
- const error = apiError({ "retry-after-ms": "1500" })
28
- expect(SessionRetry.delay(4, error)).toBe(1500)
29
- })
30
-
31
- test("uses retry-after seconds when reasonable", () => {
32
- const error = apiError({ "retry-after": "30" })
33
- expect(SessionRetry.delay(3, error)).toBe(30000)
34
- })
35
-
36
- test("accepts http-date retry-after values", () => {
37
- const date = new Date(Date.now() + 20000).toUTCString()
38
- const error = apiError({ "retry-after": date })
39
- const d = SessionRetry.delay(1, error)
40
- expect(d).toBeGreaterThanOrEqual(19000)
41
- expect(d).toBeLessThanOrEqual(20000)
42
- })
43
-
44
- test("ignores invalid retry hints", () => {
45
- const error = apiError({ "retry-after": "not-a-number" })
46
- expect(SessionRetry.delay(1, error)).toBe(2000)
47
- })
48
-
49
- test("ignores malformed date retry hints", () => {
50
- const error = apiError({ "retry-after": "Invalid Date String" })
51
- expect(SessionRetry.delay(1, error)).toBe(2000)
52
- })
53
-
54
- test("ignores past date retry hints", () => {
55
- const pastDate = new Date(Date.now() - 5000).toUTCString()
56
- const error = apiError({ "retry-after": pastDate })
57
- expect(SessionRetry.delay(1, error)).toBe(2000)
58
- })
59
-
60
- test("uses retry-after values even when exceeding 10 minutes with headers", () => {
61
- const error = apiError({ "retry-after": "50" })
62
- expect(SessionRetry.delay(1, error)).toBe(50000)
63
-
64
- const longError = apiError({ "retry-after-ms": "700000" })
65
- expect(SessionRetry.delay(1, longError)).toBe(700000)
66
- })
67
-
68
- test("sleep caps delay to max 32-bit signed integer to avoid TimeoutOverflowWarning", async () => {
69
- const controller = new AbortController()
70
-
71
- const warnings: string[] = []
72
- const originalWarn = process.emitWarning
73
- process.emitWarning = (warning: string | Error) => {
74
- warnings.push(typeof warning === "string" ? warning : warning.message)
75
- }
76
-
77
- const promise = SessionRetry.sleep(2_560_914_000, controller.signal)
78
- controller.abort()
79
-
80
- try {
81
- await promise
82
- } catch {}
83
-
84
- process.emitWarning = originalWarn
85
- expect(warnings.some((w) => w.includes("TimeoutOverflowWarning"))).toBe(false)
86
- })
87
- })
88
-
89
- describe("session.retry.retryable", () => {
90
- test("maps too_many_requests json messages", () => {
91
- const error = wrap(JSON.stringify({ type: "error", error: { type: "too_many_requests" } }))
92
- expect(SessionRetry.retryable(error)).toBe("Too Many Requests")
93
- })
94
-
95
- test("maps overloaded provider codes", () => {
96
- const error = wrap(JSON.stringify({ code: "resource_exhausted" }))
97
- expect(SessionRetry.retryable(error)).toBe("Provider is overloaded")
98
- })
99
-
100
- test("handles json messages without code", () => {
101
- const error = wrap(JSON.stringify({ error: { message: "no_kv_space" } }))
102
- expect(SessionRetry.retryable(error)).toBe(`{"error":{"message":"no_kv_space"}}`)
103
- })
104
-
105
- test("does not throw on numeric error codes", () => {
106
- const error = wrap(JSON.stringify({ type: "error", error: { code: 123 } }))
107
- const result = SessionRetry.retryable(error)
108
- expect(result).toBeUndefined()
109
- })
110
-
111
- test("returns undefined for non-json message", () => {
112
- const error = wrap("not-json")
113
- expect(SessionRetry.retryable(error)).toBeUndefined()
114
- })
115
-
116
- test("does not retry context overflow errors", () => {
117
- const error = new MessageV2.ContextOverflowError({
118
- message: "Input exceeds context window of this model",
119
- responseBody: '{"error":{"code":"context_length_exceeded"}}',
120
- }).toObject() as ReturnType<NamedError["toObject"]>
121
-
122
- expect(SessionRetry.retryable(error)).toBeUndefined()
123
- })
124
- })
125
-
126
- describe("session.message-v2.fromError", () => {
127
- test.concurrent(
128
- "converts ECONNRESET socket errors to retryable APIError",
129
- async () => {
130
- using server = Bun.serve({
131
- port: 0,
132
- idleTimeout: 8,
133
- async fetch(req) {
134
- return new Response(
135
- new ReadableStream({
136
- async pull(controller) {
137
- controller.enqueue("Hello,")
138
- await Bun.sleep(10000)
139
- controller.enqueue(" World!")
140
- controller.close()
141
- },
142
- }),
143
- { headers: { "Content-Type": "text/plain" } },
144
- )
145
- },
146
- })
147
-
148
- const error = await fetch(new URL("/", server.url.origin))
149
- .then((res) => res.text())
150
- .catch((e) => e)
151
-
152
- const result = MessageV2.fromError(error, { providerID: "test" })
153
-
154
- expect(MessageV2.APIError.isInstance(result)).toBe(true)
155
- expect((result as MessageV2.APIError).data.isRetryable).toBe(true)
156
- expect((result as MessageV2.APIError).data.message).toBe("Connection reset by server")
157
- expect((result as MessageV2.APIError).data.metadata?.code).toBe("ECONNRESET")
158
- expect((result as MessageV2.APIError).data.metadata?.message).toInclude("socket connection")
159
- },
160
- 15_000,
161
- )
162
-
163
- test("ECONNRESET socket error is retryable", () => {
164
- const error = new MessageV2.APIError({
165
- message: "Connection reset by server",
166
- isRetryable: true,
167
- metadata: { code: "ECONNRESET", message: "The socket connection was closed unexpectedly" },
168
- }).toObject() as MessageV2.APIError
169
-
170
- const retryable = SessionRetry.retryable(error)
171
- expect(retryable).toBeDefined()
172
- expect(retryable).toBe("Connection reset by server")
173
- })
174
-
175
- test("marks OpenAI 404 status codes as retryable", () => {
176
- const error = new APICallError({
177
- message: "boom",
178
- url: "https://api.openai.com/v1/chat/completions",
179
- requestBodyValues: {},
180
- statusCode: 404,
181
- responseHeaders: { "content-type": "application/json" },
182
- responseBody: '{"error":"boom"}',
183
- isRetryable: false,
184
- })
185
- const result = MessageV2.fromError(error, { providerID: "openai" }) as MessageV2.APIError
186
- expect(result.data.isRetryable).toBe(true)
187
- })
188
- })
@@ -1,285 +0,0 @@
1
- import { describe, expect, test, beforeEach, afterEach } from "bun:test"
2
- import path from "path"
3
- import { Session } from "../../src/session"
4
- import { SessionRevert } from "../../src/session/revert"
5
- import { SessionCompaction } from "../../src/session/compaction"
6
- import { MessageV2 } from "../../src/session/message-v2"
7
- import { Log } from "../../src/util/log"
8
- import { Instance } from "../../src/project/instance"
9
- import { Identifier } from "../../src/id/id"
10
- import { tmpdir } from "../fixture/fixture"
11
-
12
- const projectRoot = path.join(__dirname, "../..")
13
- Log.init({ print: false })
14
-
15
- describe("revert + compact workflow", () => {
16
- test("should properly handle compact command after revert", async () => {
17
- await using tmp = await tmpdir({ git: true })
18
- await Instance.provide({
19
- directory: tmp.path,
20
- fn: async () => {
21
- // Create a session
22
- const session = await Session.create({})
23
- const sessionID = session.id
24
-
25
- // Create a user message
26
- const userMsg1 = await Session.updateMessage({
27
- id: Identifier.ascending("message"),
28
- role: "user",
29
- sessionID,
30
- agent: "default",
31
- model: {
32
- providerID: "openai",
33
- modelID: "gpt-4",
34
- },
35
- time: {
36
- created: Date.now(),
37
- },
38
- })
39
-
40
- // Add a text part to the user message
41
- await Session.updatePart({
42
- id: Identifier.ascending("part"),
43
- messageID: userMsg1.id,
44
- sessionID,
45
- type: "text",
46
- text: "Hello, please help me",
47
- })
48
-
49
- // Create an assistant response message
50
- const assistantMsg1: MessageV2.Assistant = {
51
- id: Identifier.ascending("message"),
52
- role: "assistant",
53
- sessionID,
54
- mode: "default",
55
- agent: "default",
56
- path: {
57
- cwd: tmp.path,
58
- root: tmp.path,
59
- },
60
- cost: 0,
61
- tokens: {
62
- output: 0,
63
- input: 0,
64
- reasoning: 0,
65
- cache: { read: 0, write: 0 },
66
- },
67
- modelID: "gpt-4",
68
- providerID: "openai",
69
- parentID: userMsg1.id,
70
- time: {
71
- created: Date.now(),
72
- },
73
- finish: "end_turn",
74
- }
75
- await Session.updateMessage(assistantMsg1)
76
-
77
- // Add a text part to the assistant message
78
- await Session.updatePart({
79
- id: Identifier.ascending("part"),
80
- messageID: assistantMsg1.id,
81
- sessionID,
82
- type: "text",
83
- text: "Sure, I'll help you!",
84
- })
85
-
86
- // Create another user message
87
- const userMsg2 = await Session.updateMessage({
88
- id: Identifier.ascending("message"),
89
- role: "user",
90
- sessionID,
91
- agent: "default",
92
- model: {
93
- providerID: "openai",
94
- modelID: "gpt-4",
95
- },
96
- time: {
97
- created: Date.now(),
98
- },
99
- })
100
-
101
- await Session.updatePart({
102
- id: Identifier.ascending("part"),
103
- messageID: userMsg2.id,
104
- sessionID,
105
- type: "text",
106
- text: "What's the capital of France?",
107
- })
108
-
109
- // Create another assistant response
110
- const assistantMsg2: MessageV2.Assistant = {
111
- id: Identifier.ascending("message"),
112
- role: "assistant",
113
- sessionID,
114
- mode: "default",
115
- agent: "default",
116
- path: {
117
- cwd: tmp.path,
118
- root: tmp.path,
119
- },
120
- cost: 0,
121
- tokens: {
122
- output: 0,
123
- input: 0,
124
- reasoning: 0,
125
- cache: { read: 0, write: 0 },
126
- },
127
- modelID: "gpt-4",
128
- providerID: "openai",
129
- parentID: userMsg2.id,
130
- time: {
131
- created: Date.now(),
132
- },
133
- finish: "end_turn",
134
- }
135
- await Session.updateMessage(assistantMsg2)
136
-
137
- await Session.updatePart({
138
- id: Identifier.ascending("part"),
139
- messageID: assistantMsg2.id,
140
- sessionID,
141
- type: "text",
142
- text: "The capital of France is Paris.",
143
- })
144
-
145
- // Verify messages before revert
146
- let messages = await Session.messages({ sessionID })
147
- expect(messages.length).toBe(4) // 2 user + 2 assistant messages
148
- const messageIds = messages.map((m) => m.info.id)
149
- expect(messageIds).toContain(userMsg1.id)
150
- expect(messageIds).toContain(userMsg2.id)
151
- expect(messageIds).toContain(assistantMsg1.id)
152
- expect(messageIds).toContain(assistantMsg2.id)
153
-
154
- // Revert the last user message (userMsg2)
155
- await SessionRevert.revert({
156
- sessionID,
157
- messageID: userMsg2.id,
158
- })
159
-
160
- // Check that revert state is set
161
- let sessionInfo = await Session.get(sessionID)
162
- expect(sessionInfo.revert).toBeDefined()
163
- const revertMessageID = sessionInfo.revert?.messageID
164
- expect(revertMessageID).toBeDefined()
165
-
166
- // Messages should still be in the list (not removed yet, just marked for revert)
167
- messages = await Session.messages({ sessionID })
168
- expect(messages.length).toBe(4)
169
-
170
- // Now clean up the revert state (this is what the compact endpoint should do)
171
- await SessionRevert.cleanup(sessionInfo)
172
-
173
- // After cleanup, the reverted messages (those after the revert point) should be removed
174
- messages = await Session.messages({ sessionID })
175
- const remainingIds = messages.map((m) => m.info.id)
176
- // The revert point is somewhere in the message chain, so we should have fewer messages
177
- expect(messages.length).toBeLessThan(4)
178
- // userMsg2 and assistantMsg2 should be removed (they come after the revert point)
179
- expect(remainingIds).not.toContain(userMsg2.id)
180
- expect(remainingIds).not.toContain(assistantMsg2.id)
181
-
182
- // Revert state should be cleared
183
- sessionInfo = await Session.get(sessionID)
184
- expect(sessionInfo.revert).toBeUndefined()
185
-
186
- // Clean up
187
- await Session.remove(sessionID)
188
- },
189
- })
190
- })
191
-
192
- test("should properly clean up revert state before creating compaction message", async () => {
193
- await using tmp = await tmpdir({ git: true })
194
- await Instance.provide({
195
- directory: tmp.path,
196
- fn: async () => {
197
- // Create a session
198
- const session = await Session.create({})
199
- const sessionID = session.id
200
-
201
- // Create initial messages
202
- const userMsg = await Session.updateMessage({
203
- id: Identifier.ascending("message"),
204
- role: "user",
205
- sessionID,
206
- agent: "default",
207
- model: {
208
- providerID: "openai",
209
- modelID: "gpt-4",
210
- },
211
- time: {
212
- created: Date.now(),
213
- },
214
- })
215
-
216
- await Session.updatePart({
217
- id: Identifier.ascending("part"),
218
- messageID: userMsg.id,
219
- sessionID,
220
- type: "text",
221
- text: "Hello",
222
- })
223
-
224
- const assistantMsg: MessageV2.Assistant = {
225
- id: Identifier.ascending("message"),
226
- role: "assistant",
227
- sessionID,
228
- mode: "default",
229
- agent: "default",
230
- path: {
231
- cwd: tmp.path,
232
- root: tmp.path,
233
- },
234
- cost: 0,
235
- tokens: {
236
- output: 0,
237
- input: 0,
238
- reasoning: 0,
239
- cache: { read: 0, write: 0 },
240
- },
241
- modelID: "gpt-4",
242
- providerID: "openai",
243
- parentID: userMsg.id,
244
- time: {
245
- created: Date.now(),
246
- },
247
- finish: "end_turn",
248
- }
249
- await Session.updateMessage(assistantMsg)
250
-
251
- await Session.updatePart({
252
- id: Identifier.ascending("part"),
253
- messageID: assistantMsg.id,
254
- sessionID,
255
- type: "text",
256
- text: "Hi there!",
257
- })
258
-
259
- // Revert the user message
260
- await SessionRevert.revert({
261
- sessionID,
262
- messageID: userMsg.id,
263
- })
264
-
265
- // Check that revert state is set
266
- let sessionInfo = await Session.get(sessionID)
267
- expect(sessionInfo.revert).toBeDefined()
268
-
269
- // Simulate what the compact endpoint does: cleanup revert before creating compaction
270
- await SessionRevert.cleanup(sessionInfo)
271
-
272
- // Verify revert state is cleared
273
- sessionInfo = await Session.get(sessionID)
274
- expect(sessionInfo.revert).toBeUndefined()
275
-
276
- // Verify messages are properly cleaned up
277
- const messages = await Session.messages({ sessionID })
278
- expect(messages.length).toBe(0) // All messages should be reverted
279
-
280
- // Clean up
281
- await Session.remove(sessionID)
282
- },
283
- })
284
- })
285
- })