soraacode 0.2.0 → 0.2.2

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 (353) hide show
  1. package/LICENSE +21 -0
  2. package/bin/soraacode.exe +10 -0
  3. package/package.json +17 -25
  4. package/{script/postinstall.mjs → postinstall.mjs} +5 -5
  5. package/AGENTS.md +0 -131
  6. package/Dockerfile +0 -18
  7. package/README.md +0 -15
  8. package/bin/opencode +0 -199
  9. package/bin/soraacode +0 -199
  10. package/bunfig.toml +0 -7
  11. package/git +0 -0
  12. package/migration/20260511173437_session-metadata/migration.sql +0 -1
  13. package/migration/20260511173437_session-metadata/snapshot.json +0 -1500
  14. package/parsers-config.ts +0 -1
  15. package/script/bench-search.ts +0 -94
  16. package/script/bench-test-suite.ts +0 -52
  17. package/script/build-headless.ts +0 -51
  18. package/script/build.ts +0 -245
  19. package/script/generate.ts +0 -14
  20. package/script/httpapi-exercise.ts +0 -1
  21. package/script/profile-test-files.ts +0 -42
  22. package/script/publish.ts +0 -213
  23. package/script/run-workspace-server +0 -106
  24. package/script/schema.ts +0 -77
  25. package/script/time.ts +0 -6
  26. package/script/trace-imports.ts +0 -153
  27. package/soraacode-0.1.7.tgz +0 -0
  28. package/specs/effect/error-boundaries-plan.md +0 -235
  29. package/specs/effect/errors.md +0 -207
  30. package/specs/effect/facades.md +0 -218
  31. package/specs/effect/guide.md +0 -247
  32. package/specs/effect/instance-context.md +0 -13
  33. package/specs/effect/loose-ends.md +0 -30
  34. package/specs/effect/migration.md +0 -62
  35. package/specs/effect/routes.md +0 -61
  36. package/specs/effect/schema.md +0 -88
  37. package/specs/effect/server-package.md +0 -58
  38. package/specs/effect/todo.md +0 -241
  39. package/specs/effect/tools.md +0 -88
  40. package/specs/openapi-translation-cleanup.md +0 -204
  41. package/specs/tui-plugins.md +0 -544
  42. package/specs/v2/api.ts +0 -67
  43. package/specs/v2/message-shape.md +0 -136
  44. package/specs/v2/notifications.md +0 -13
  45. package/specs/v2/tui-command-shim.md +0 -67
  46. package/src/account/account.ts +0 -481
  47. package/src/account/repo.ts +0 -171
  48. package/src/account/schema.ts +0 -99
  49. package/src/account/url.ts +0 -8
  50. package/src/acp/agent.ts +0 -95
  51. package/src/acp/config-option.ts +0 -206
  52. package/src/acp/content.ts +0 -269
  53. package/src/acp/directory.ts +0 -212
  54. package/src/acp/error.ts +0 -97
  55. package/src/acp/event.ts +0 -421
  56. package/src/acp/permission.ts +0 -254
  57. package/src/acp/profile.ts +0 -42
  58. package/src/acp/service.ts +0 -1226
  59. package/src/acp/session.ts +0 -232
  60. package/src/acp/tool.ts +0 -364
  61. package/src/acp/usage.ts +0 -243
  62. package/src/agent/agent.ts +0 -450
  63. package/src/agent/generate.txt +0 -75
  64. package/src/agent/prompt/compaction.txt +0 -5
  65. package/src/agent/prompt/explore.txt +0 -18
  66. package/src/agent/prompt/summary.txt +0 -11
  67. package/src/agent/prompt/title.txt +0 -44
  68. package/src/agent/subagent-permissions.ts +0 -27
  69. package/src/audio.d.ts +0 -14
  70. package/src/auth/index.ts +0 -97
  71. package/src/background/job.ts +0 -37
  72. package/src/bus/global.ts +0 -22
  73. package/src/cli/bootstrap.ts +0 -11
  74. package/src/cli/cmd/cmd.ts +0 -7
  75. package/src/cli/cmd/serve.ts +0 -24
  76. package/src/cli/effect/prompt.ts +0 -37
  77. package/src/cli/effect-cmd.ts +0 -96
  78. package/src/cli/error.ts +0 -130
  79. package/src/cli/heap.ts +0 -45
  80. package/src/cli/logo.ts +0 -2
  81. package/src/cli/network.ts +0 -80
  82. package/src/cli/ui.ts +0 -132
  83. package/src/command/index.ts +0 -177
  84. package/src/command/template/initialize.txt +0 -66
  85. package/src/command/template/review.txt +0 -101
  86. package/src/config/agent.ts +0 -59
  87. package/src/config/command.ts +0 -39
  88. package/src/config/config.ts +0 -707
  89. package/src/config/entry-name.ts +0 -19
  90. package/src/config/managed.ts +0 -69
  91. package/src/config/markdown.ts +0 -36
  92. package/src/config/parse.ts +0 -61
  93. package/src/config/paths.ts +0 -45
  94. package/src/config/plugin.ts +0 -79
  95. package/src/config/v2-compat.ts +0 -449
  96. package/src/config/variable.ts +0 -91
  97. package/src/control-plane/adapters/index.ts +0 -41
  98. package/src/control-plane/adapters/worktree.ts +0 -96
  99. package/src/control-plane/dev/README.md +0 -19
  100. package/src/control-plane/dev/debug-workspace-plugin.ts +0 -73
  101. package/src/control-plane/types.ts +0 -59
  102. package/src/control-plane/util.ts +0 -39
  103. package/src/control-plane/workspace-adapter-runtime.ts +0 -51
  104. package/src/control-plane/workspace-context.ts +0 -26
  105. package/src/control-plane/workspace.ts +0 -966
  106. package/src/effect/app-node-builder-v1.ts +0 -12
  107. package/src/effect/app-runtime.ts +0 -135
  108. package/src/effect/bootstrap-runtime.ts +0 -19
  109. package/src/effect/bridge.ts +0 -84
  110. package/src/effect/config-service.ts +0 -67
  111. package/src/effect/instance-ref.ts +0 -11
  112. package/src/effect/instance-registry.ts +0 -12
  113. package/src/effect/instance-state.ts +0 -69
  114. package/src/effect/promise.ts +0 -17
  115. package/src/effect/run-service.ts +0 -47
  116. package/src/effect/runner.ts +0 -217
  117. package/src/effect/runtime-flags.ts +0 -78
  118. package/src/env/index.ts +0 -41
  119. package/src/event-manifest.ts +0 -3
  120. package/src/event-v2-bridge.ts +0 -71
  121. package/src/format/formatter.ts +0 -404
  122. package/src/format/index.ts +0 -203
  123. package/src/git/index.ts +0 -348
  124. package/src/id/id.ts +0 -80
  125. package/src/ide/index.ts +0 -54
  126. package/src/image/image.ts +0 -172
  127. package/src/index-headless.ts +0 -89
  128. package/src/index.ts +0 -93
  129. package/src/installation/index.ts +0 -336
  130. package/src/lsp/client.ts +0 -650
  131. package/src/lsp/diagnostic.ts +0 -29
  132. package/src/lsp/language.ts +0 -121
  133. package/src/lsp/launch.ts +0 -21
  134. package/src/lsp/lsp.ts +0 -507
  135. package/src/lsp/server.ts +0 -1983
  136. package/src/markdown.d.ts +0 -4
  137. package/src/mcp/auth.ts +0 -163
  138. package/src/mcp/browser.ts +0 -37
  139. package/src/mcp/catalog.ts +0 -170
  140. package/src/mcp/index.ts +0 -1004
  141. package/src/mcp/oauth-callback.ts +0 -194
  142. package/src/mcp/oauth-provider.ts +0 -259
  143. package/src/node.ts +0 -4
  144. package/src/patch/index.ts +0 -686
  145. package/src/permission/arity.ts +0 -163
  146. package/src/permission/evaluate.ts +0 -1
  147. package/src/permission/index.ts +0 -223
  148. package/src/plugin/github-copilot/copilot.ts +0 -415
  149. package/src/plugin/github-copilot/models.ts +0 -261
  150. package/src/plugin/index.ts +0 -298
  151. package/src/plugin/install.ts +0 -439
  152. package/src/plugin/loader.ts +0 -237
  153. package/src/plugin/meta.ts +0 -188
  154. package/src/plugin/openai/README.md +0 -31
  155. package/src/plugin/openai/codex.ts +0 -575
  156. package/src/plugin/openai/ws-pool.ts +0 -271
  157. package/src/plugin/openai/ws.ts +0 -383
  158. package/src/plugin/pty-environment.ts +0 -24
  159. package/src/plugin/shared.ts +0 -323
  160. package/src/project/bootstrap-service.ts +0 -9
  161. package/src/project/bootstrap.ts +0 -58
  162. package/src/project/instance-context.ts +0 -24
  163. package/src/project/instance-runtime.ts +0 -16
  164. package/src/project/instance-store.ts +0 -213
  165. package/src/project/project.ts +0 -483
  166. package/src/project/vcs.ts +0 -423
  167. package/src/provider/auth.ts +0 -229
  168. package/src/provider/error.ts +0 -195
  169. package/src/provider/model-status.ts +0 -8
  170. package/src/provider/provider.ts +0 -1955
  171. package/src/provider/transform.ts +0 -1923
  172. package/src/question/index.ts +0 -161
  173. package/src/question/schema.ts +0 -4
  174. package/src/server/auth.ts +0 -48
  175. package/src/server/event.ts +0 -10
  176. package/src/server/global-lifecycle.ts +0 -28
  177. package/src/server/mdns.ts +0 -47
  178. package/src/server/proxy-util.ts +0 -48
  179. package/src/server/routes/instance/httpapi/AGENTS.md +0 -39
  180. package/src/server/routes/instance/httpapi/api.ts +0 -97
  181. package/src/server/routes/instance/httpapi/errors.ts +0 -193
  182. package/src/server/routes/instance/httpapi/groups/config.ts +0 -65
  183. package/src/server/routes/instance/httpapi/groups/control-plane.ts +0 -35
  184. package/src/server/routes/instance/httpapi/groups/control.ts +0 -76
  185. package/src/server/routes/instance/httpapi/groups/event.ts +0 -29
  186. package/src/server/routes/instance/httpapi/groups/experimental.ts +0 -275
  187. package/src/server/routes/instance/httpapi/groups/file.ts +0 -185
  188. package/src/server/routes/instance/httpapi/groups/global.ts +0 -139
  189. package/src/server/routes/instance/httpapi/groups/instance.ts +0 -206
  190. package/src/server/routes/instance/httpapi/groups/mcp.ts +0 -156
  191. package/src/server/routes/instance/httpapi/groups/metadata.ts +0 -18
  192. package/src/server/routes/instance/httpapi/groups/permission.ts +0 -61
  193. package/src/server/routes/instance/httpapi/groups/project-copy.ts +0 -32
  194. package/src/server/routes/instance/httpapi/groups/project.ts +0 -93
  195. package/src/server/routes/instance/httpapi/groups/provider.ts +0 -101
  196. package/src/server/routes/instance/httpapi/groups/pty.ts +0 -172
  197. package/src/server/routes/instance/httpapi/groups/query.ts +0 -12
  198. package/src/server/routes/instance/httpapi/groups/question.ts +0 -74
  199. package/src/server/routes/instance/httpapi/groups/session.ts +0 -462
  200. package/src/server/routes/instance/httpapi/groups/sync.ts +0 -113
  201. package/src/server/routes/instance/httpapi/groups/tui.ts +0 -208
  202. package/src/server/routes/instance/httpapi/groups/workspace.ts +0 -141
  203. package/src/server/routes/instance/httpapi/handlers/config.ts +0 -34
  204. package/src/server/routes/instance/httpapi/handlers/control-plane.ts +0 -37
  205. package/src/server/routes/instance/httpapi/handlers/control.ts +0 -43
  206. package/src/server/routes/instance/httpapi/handlers/event.ts +0 -99
  207. package/src/server/routes/instance/httpapi/handlers/experimental.ts +0 -193
  208. package/src/server/routes/instance/httpapi/handlers/file.ts +0 -139
  209. package/src/server/routes/instance/httpapi/handlers/global.ts +0 -126
  210. package/src/server/routes/instance/httpapi/handlers/instance.ts +0 -110
  211. package/src/server/routes/instance/httpapi/handlers/mcp.ts +0 -111
  212. package/src/server/routes/instance/httpapi/handlers/permission.ts +0 -41
  213. package/src/server/routes/instance/httpapi/handlers/project-copy.ts +0 -83
  214. package/src/server/routes/instance/httpapi/handlers/project.ts +0 -63
  215. package/src/server/routes/instance/httpapi/handlers/provider.ts +0 -116
  216. package/src/server/routes/instance/httpapi/handlers/pty.ts +0 -273
  217. package/src/server/routes/instance/httpapi/handlers/question.ts +0 -54
  218. package/src/server/routes/instance/httpapi/handlers/session-errors.ts +0 -21
  219. package/src/server/routes/instance/httpapi/handlers/session.ts +0 -442
  220. package/src/server/routes/instance/httpapi/handlers/sync.ts +0 -89
  221. package/src/server/routes/instance/httpapi/handlers/tui.ts +0 -131
  222. package/src/server/routes/instance/httpapi/handlers/workspace.ts +0 -102
  223. package/src/server/routes/instance/httpapi/lifecycle.ts +0 -54
  224. package/src/server/routes/instance/httpapi/middleware/authorization.ts +0 -150
  225. package/src/server/routes/instance/httpapi/middleware/compression.ts +0 -64
  226. package/src/server/routes/instance/httpapi/middleware/cors-vary.ts +0 -29
  227. package/src/server/routes/instance/httpapi/middleware/error.ts +0 -44
  228. package/src/server/routes/instance/httpapi/middleware/fence.ts +0 -25
  229. package/src/server/routes/instance/httpapi/middleware/instance-context.ts +0 -43
  230. package/src/server/routes/instance/httpapi/middleware/proxy.ts +0 -131
  231. package/src/server/routes/instance/httpapi/middleware/schema-error.ts +0 -41
  232. package/src/server/routes/instance/httpapi/middleware/workspace-routing.ts +0 -250
  233. package/src/server/routes/instance/httpapi/public.ts +0 -537
  234. package/src/server/routes/instance/httpapi/server.ts +0 -325
  235. package/src/server/routes/instance/httpapi/websocket-tracker.ts +0 -60
  236. package/src/server/server.ts +0 -224
  237. package/src/server/shared/fence.ts +0 -60
  238. package/src/server/shared/pty-ticket.ts +0 -15
  239. package/src/server/shared/public-ui.ts +0 -12
  240. package/src/server/shared/tui-control.ts +0 -28
  241. package/src/server/shared/ui.ts +0 -108
  242. package/src/server/shared/workspace-routing.ts +0 -45
  243. package/src/server/tui-event.ts +0 -1
  244. package/src/session/compaction.ts +0 -608
  245. package/src/session/instruction.ts +0 -237
  246. package/src/session/llm/AGENTS.md +0 -90
  247. package/src/session/llm/ai-sdk.ts +0 -291
  248. package/src/session/llm/native-request.ts +0 -196
  249. package/src/session/llm/native-runtime.ts +0 -195
  250. package/src/session/llm/request.ts +0 -226
  251. package/src/session/llm.ts +0 -390
  252. package/src/session/message-error.ts +0 -14
  253. package/src/session/message-v2.ts +0 -742
  254. package/src/session/message.ts +0 -148
  255. package/src/session/overflow.ts +0 -34
  256. package/src/session/processor.ts +0 -732
  257. package/src/session/prompt/build-switch.txt +0 -5
  258. package/src/session/prompt/default.txt +0 -35
  259. package/src/session/prompt/plan-mode.txt +0 -70
  260. package/src/session/prompt/plan.txt +0 -26
  261. package/src/session/prompt.ts +0 -1631
  262. package/src/session/reminders.ts +0 -92
  263. package/src/session/retry.ts +0 -209
  264. package/src/session/revert.ts +0 -136
  265. package/src/session/run-state.ts +0 -151
  266. package/src/session/schema.ts +0 -26
  267. package/src/session/session.ts +0 -1016
  268. package/src/session/status.ts +0 -56
  269. package/src/session/summary.ts +0 -160
  270. package/src/session/system.ts +0 -123
  271. package/src/session/todo.ts +0 -74
  272. package/src/session/tools.ts +0 -590
  273. package/src/share/session.ts +0 -58
  274. package/src/share/share-next.ts +0 -371
  275. package/src/skill/discovery.ts +0 -140
  276. package/src/skill/index.ts +0 -354
  277. package/src/snapshot/index.ts +0 -807
  278. package/src/sql.d.ts +0 -4
  279. package/src/storage/schema.ts +0 -5
  280. package/src/storage/storage.ts +0 -327
  281. package/src/sync/README.md +0 -179
  282. package/src/sync/schema.ts +0 -11
  283. package/src/temporary.ts +0 -31
  284. package/src/tool/apply_patch.ts +0 -313
  285. package/src/tool/apply_patch.txt +0 -14
  286. package/src/tool/code-mode.ts +0 -310
  287. package/src/tool/edit.ts +0 -737
  288. package/src/tool/edit.txt +0 -13
  289. package/src/tool/external-directory.ts +0 -49
  290. package/src/tool/glob.ts +0 -76
  291. package/src/tool/glob.txt +0 -7
  292. package/src/tool/grep.ts +0 -115
  293. package/src/tool/grep.txt +0 -7
  294. package/src/tool/invalid.ts +0 -21
  295. package/src/tool/json-schema.ts +0 -164
  296. package/src/tool/lsp.ts +0 -113
  297. package/src/tool/lsp.txt +0 -1
  298. package/src/tool/mcp-websearch.ts +0 -96
  299. package/src/tool/plan-enter.txt +0 -1
  300. package/src/tool/plan-exit.txt +0 -1
  301. package/src/tool/plan.ts +0 -79
  302. package/src/tool/question.ts +0 -44
  303. package/src/tool/question.txt +0 -1
  304. package/src/tool/read.ts +0 -386
  305. package/src/tool/read.txt +0 -13
  306. package/src/tool/registry.ts +0 -455
  307. package/src/tool/schema.ts +0 -14
  308. package/src/tool/shell/id.ts +0 -19
  309. package/src/tool/shell/prompt.ts +0 -293
  310. package/src/tool/shell/shell.txt +0 -21
  311. package/src/tool/shell.ts +0 -645
  312. package/src/tool/skill.ts +0 -70
  313. package/src/tool/skill.txt +0 -1
  314. package/src/tool/task.ts +0 -371
  315. package/src/tool/task.txt +0 -11
  316. package/src/tool/todo.ts +0 -46
  317. package/src/tool/todowrite.txt +0 -11
  318. package/src/tool/tool.ts +0 -183
  319. package/src/tool/truncate.ts +0 -156
  320. package/src/tool/truncation-dir.ts +0 -4
  321. package/src/tool/webfetch.ts +0 -192
  322. package/src/tool/webfetch.txt +0 -9
  323. package/src/tool/websearch.ts +0 -143
  324. package/src/tool/websearch.txt +0 -9
  325. package/src/tool/write.ts +0 -104
  326. package/src/tool/write.txt +0 -11
  327. package/src/util/archive.ts +0 -17
  328. package/src/util/bom.ts +0 -27
  329. package/src/util/data-url.ts +0 -9
  330. package/src/util/defer.ts +0 -10
  331. package/src/util/effect-http-client.ts +0 -11
  332. package/src/util/error.ts +0 -16
  333. package/src/util/filesystem.ts +0 -251
  334. package/src/util/html.ts +0 -8
  335. package/src/util/iife.ts +0 -3
  336. package/src/util/lazy.ts +0 -20
  337. package/src/util/local-context.ts +0 -25
  338. package/src/util/locale.ts +0 -4
  339. package/src/util/media.ts +0 -26
  340. package/src/util/process.ts +0 -177
  341. package/src/util/proxy-env.ts +0 -72
  342. package/src/util/queue.ts +0 -32
  343. package/src/util/record.ts +0 -3
  344. package/src/util/repository.ts +0 -232
  345. package/src/util/rpc.ts +0 -66
  346. package/src/util/signal.ts +0 -12
  347. package/src/util/timeout.ts +0 -13
  348. package/src/util/token.ts +0 -1
  349. package/src/util/wildcard.ts +0 -59
  350. package/src/worktree/index.ts +0 -623
  351. package/sst-env.d.ts +0 -10
  352. package/tsconfig.headless.json +0 -13
  353. package/tsconfig.json +0 -16
@@ -1,742 +0,0 @@
1
- import { SessionID, MessageID } from "./schema"
2
- import { SessionV1 } from "@soraacode/core/v1/session"
3
- import { ProviderV2 } from "@soraacode/core/provider"
4
- import {
5
- APIError,
6
- AbortedError,
7
- Assistant,
8
- AuthError,
9
- CompactionPart,
10
- ContextOverflowError,
11
- Info,
12
- OutputLengthError,
13
- Part,
14
- SubtaskPart,
15
- User,
16
- WithParts,
17
- } from "@soraacode/core/v1/session"
18
-
19
- import { NamedError } from "@soraacode/core/util/error"
20
- import { APICallError, convertToModelMessages, LoadAPIKeyError, type ModelMessage, type UIMessage } from "ai"
21
- import { Database } from "@soraacode/core/database/database"
22
- import { LayerNode } from "@soraacode/core/effect/layer-node"
23
- import { NotFoundError } from "@/storage/storage"
24
- import { and } from "drizzle-orm"
25
- import { desc } from "drizzle-orm"
26
- import { eq } from "drizzle-orm"
27
- import { inArray } from "drizzle-orm"
28
- import { lt } from "drizzle-orm"
29
- import { or } from "drizzle-orm"
30
- import { MessageTable, PartTable, SessionTable } from "@soraacode/core/session/sql"
31
- import { ProviderError } from "@/provider/error"
32
- import { iife } from "@/util/iife"
33
- import { errorMessage } from "@/util/error"
34
- import { isMedia } from "@/util/media"
35
- import type { SystemError } from "bun"
36
- import type { Provider } from "@/provider/provider"
37
- import { Effect, Schema } from "effect"
38
-
39
- /** Error shape thrown by Bun's fetch() when gzip/br decompression fails mid-stream */
40
- interface FetchDecompressionError extends Error {
41
- code: "ZlibError"
42
- errno: number
43
- path: string
44
- }
45
-
46
- export const SYNTHETIC_ATTACHMENT_PROMPT = "Attached media from tool result:"
47
- export { isMedia }
48
-
49
- function truncateToolOutput(text: string, maxChars?: number) {
50
- const limit = maxChars ?? 4000
51
- if (text.length <= limit) return text
52
- const omitted = text.length - limit
53
- return `${text.slice(0, limit)}\n[Tool output truncated: omitted ${omitted} chars]`
54
- }
55
-
56
- export const Event = {
57
- Updated: SessionV1.Event.MessageUpdated,
58
- Removed: SessionV1.Event.MessageRemoved,
59
- PartUpdated: SessionV1.Event.PartUpdated,
60
- PartDelta: SessionV1.Event.PartDelta,
61
- PartRemoved: SessionV1.Event.PartRemoved,
62
- }
63
-
64
- const Cursor = Schema.Struct({
65
- id: MessageID,
66
- time: Schema.Finite.check(Schema.isGreaterThanOrEqualTo(0)),
67
- })
68
- type Cursor = typeof Cursor.Type
69
-
70
- const decodeCursor = Schema.decodeUnknownSync(Cursor)
71
-
72
- export const cursor = {
73
- encode(input: Cursor) {
74
- return Buffer.from(JSON.stringify(input)).toString("base64url")
75
- },
76
- decode(input: string) {
77
- return decodeCursor(JSON.parse(Buffer.from(input, "base64url").toString("utf8")))
78
- },
79
- }
80
-
81
- const info = (row: typeof MessageTable.$inferSelect) =>
82
- ({
83
- ...row.data,
84
- id: row.id,
85
- sessionID: row.session_id,
86
- }) as Info
87
-
88
- const part = (row: typeof PartTable.$inferSelect) =>
89
- ({
90
- ...row.data,
91
- id: row.id,
92
- sessionID: row.session_id,
93
- messageID: row.message_id,
94
- }) as Part
95
-
96
- const older = (row: Cursor) =>
97
- or(lt(MessageTable.time_created, row.time), and(eq(MessageTable.time_created, row.time), lt(MessageTable.id, row.id)))
98
-
99
- function hydrate(db: Database.Interface["db"], rows: (typeof MessageTable.$inferSelect)[]) {
100
- const ids = rows.map((row) => row.id)
101
- const partByMessage = new Map<string, Part[]>()
102
- return Effect.gen(function* () {
103
- if (ids.length > 0) {
104
- const partRows = yield* db
105
- .select()
106
- .from(PartTable)
107
- .where(inArray(PartTable.message_id, ids))
108
- .orderBy(PartTable.message_id, PartTable.id)
109
- .all()
110
- .pipe(Effect.orDie)
111
- for (const row of partRows) {
112
- const next = part(row)
113
- const list = partByMessage.get(row.message_id)
114
- if (list) list.push(next)
115
- else partByMessage.set(row.message_id, [next])
116
- }
117
- }
118
-
119
- return rows.map((row) => ({
120
- info: info(row),
121
- parts: partByMessage.get(row.id) ?? [],
122
- }))
123
- })
124
- }
125
-
126
- function providerMeta(metadata: Record<string, any> | undefined) {
127
- if (!metadata) return undefined
128
- const { providerExecuted: _, ...rest } = metadata
129
- return Object.keys(rest).length > 0 ? rest : undefined
130
- }
131
-
132
- export const toModelMessagesEffect = Effect.fnUntraced(function* (
133
- input: WithParts[],
134
- model: Provider.Model,
135
- options?: { stripMedia?: boolean; toolOutputMaxChars?: number },
136
- ) {
137
- const result: UIMessage[] = []
138
- const toolNames = new Set<string>()
139
- // Track media from tool results that need to be injected as user messages
140
- // for providers that don't support that media type in tool results.
141
- //
142
- // OpenAI-compatible APIs only support string content in tool results, so we need
143
- // to extract media and inject as user messages. Some SDKs only support a subset
144
- // of media in tool results; e.g. Bedrock supports images but not PDFs there.
145
- //
146
- // Only apply this workaround if the model actually supports that media input -
147
- // otherwise unsupportedParts() will turn it into a user-visible error.
148
- const supportsMediaInToolResult = (attachment: { mime: string }) => {
149
- if (model.api.npm === "@ai-sdk/anthropic") return true
150
- if (model.api.npm === "@ai-sdk/openai") return true
151
- if (model.api.npm === "@ai-sdk/amazon-bedrock/mantle") return true
152
- if (model.api.npm === "@ai-sdk/amazon-bedrock") {
153
- if (!attachment.mime.startsWith("image/")) return false
154
- const id = model.api.id.toLowerCase()
155
- return id.includes("anthropic.") || id.includes("nova") || id.includes("llama4") || id.includes("llama-4")
156
- }
157
- if (model.api.npm === "@ai-sdk/xai") return attachment.mime.startsWith("image/")
158
- if (model.api.npm === "@ai-sdk/google-vertex/anthropic") return true
159
- if (model.api.npm === "@ai-sdk/google") {
160
- const id = model.api.id.toLowerCase()
161
- return id.includes("gemini-3") && !id.includes("gemini-2")
162
- }
163
- return false
164
- }
165
-
166
- const toModelOutput = (options: { toolCallId: string; input: unknown; output: unknown }) => {
167
- const output = options.output
168
- if (typeof output === "string") {
169
- return { type: "text", value: output }
170
- }
171
-
172
- if (typeof output === "object") {
173
- const outputObject = output as {
174
- text: string
175
- attachments?: Array<{ mime: string; url: string }>
176
- }
177
- const attachments = (outputObject.attachments ?? []).filter((attachment) => {
178
- return attachment.url.startsWith("data:") && attachment.url.includes(",")
179
- })
180
-
181
- return {
182
- type: "content",
183
- value: [
184
- ...(outputObject.text ? [{ type: "text", text: outputObject.text }] : []),
185
- ...attachments.map((attachment) => ({
186
- type: "media",
187
- mediaType: attachment.mime,
188
- data: iife(() => {
189
- const commaIndex = attachment.url.indexOf(",")
190
- return commaIndex === -1 ? attachment.url : attachment.url.slice(commaIndex + 1)
191
- }),
192
- })),
193
- ],
194
- }
195
- }
196
-
197
- return { type: "json", value: output as never }
198
- }
199
-
200
- for (const msg of input) {
201
- if (msg.parts.length === 0) continue
202
-
203
- if (msg.info.role === "user") {
204
- const userMessage: UIMessage = {
205
- id: msg.info.id,
206
- role: "user",
207
- parts: [],
208
- }
209
- for (const part of msg.parts) {
210
- // User message parts should never be empty
211
- if (part.type === "text" && !part.ignored && part.text !== "")
212
- userMessage.parts.push({
213
- type: "text",
214
- text: part.text,
215
- })
216
- // text/plain and directory files are converted into text parts, ignore them
217
- if (part.type === "file" && part.mime !== "text/plain" && part.mime !== "application/x-directory") {
218
- if (options?.stripMedia && isMedia(part.mime)) {
219
- userMessage.parts.push({
220
- type: "text",
221
- text: `[Attached ${part.mime}: ${part.filename ?? "file"}]`,
222
- })
223
- } else {
224
- userMessage.parts.push({
225
- type: "file",
226
- url: part.url,
227
- mediaType: part.mime,
228
- filename: part.filename,
229
- })
230
- }
231
- }
232
-
233
- if (part.type === "compaction") {
234
- userMessage.parts.push({
235
- type: "text",
236
- text: "What did we do so far?",
237
- })
238
- }
239
- if (part.type === "subtask") {
240
- userMessage.parts.push({
241
- type: "text",
242
- text: "The following tool was executed by the user",
243
- })
244
- }
245
- }
246
- if (userMessage.parts.length > 0) result.push(userMessage)
247
- }
248
-
249
- if (msg.info.role === "assistant") {
250
- const differentModel = `${model.providerID}/${model.id}` !== `${msg.info.providerID}/${msg.info.modelID}`
251
- const media: Array<{ mime: string; url: string; filename?: string }> = []
252
-
253
- if (
254
- msg.info.error &&
255
- !(
256
- AbortedError.isInstance(msg.info.error) &&
257
- msg.parts.some((part) => part.type !== "step-start" && part.type !== "reasoning")
258
- )
259
- ) {
260
- continue
261
- }
262
- const assistantMessage: UIMessage = {
263
- id: msg.info.id,
264
- role: "assistant",
265
- parts: [],
266
- }
267
- // Anthropic adaptive thinking can persist assistant turns like:
268
- // step-start, reasoning(signature), text(""), step-start,
269
- // reasoning(signature). The empty text part is a structural separator,
270
- // but it does not carry the signature metadata itself. Dropping it shifts
271
- // signed thinking positions after step-start splitting/provider regrouping;
272
- // keeping it as "" is filtered by the AI SDK and rejected by Anthropic.
273
- // It is unclear whether this shape originates in our stream processing,
274
- // a proxy, or a lower-level library, but preserving a non-empty separator
275
- // here is the only safe replay point we have.
276
- // Use a single space so the separator survives replay without changing
277
- // the neighboring signed reasoning blocks.
278
- const hasSignedReasoning = msg.parts.some((part) => {
279
- if (part.type !== "reasoning") return false
280
- return part.metadata?.anthropic?.signature != null
281
- })
282
- for (const part of msg.parts) {
283
- if (part.type === "text") {
284
- const text = part.text === "" && hasSignedReasoning ? " " : part.text
285
- assistantMessage.parts.push({
286
- type: "text",
287
- text,
288
- ...(differentModel ? {} : { providerMetadata: part.metadata }),
289
- })
290
- }
291
- if (part.type === "step-start")
292
- assistantMessage.parts.push({
293
- type: "step-start",
294
- })
295
- if (part.type === "tool") {
296
- toolNames.add(part.tool)
297
- if (part.state.status === "completed") {
298
- const outputText = part.state.time.compacted
299
- ? "[Old tool result content cleared]"
300
- : truncateToolOutput(part.state.output, options?.toolOutputMaxChars)
301
- const attachments = part.state.time.compacted || options?.stripMedia ? [] : (part.state.attachments ?? [])
302
-
303
- // For providers that don't support media in tool results, extract media files
304
- // (images, PDFs) to be sent as a separate user message
305
- const mediaAttachments = attachments.filter((a) => isMedia(a.mime))
306
- const extractedMedia = mediaAttachments.filter((a) => !supportsMediaInToolResult(a))
307
- if (extractedMedia.length > 0) {
308
- media.push(...extractedMedia)
309
- }
310
- const finalAttachments = attachments.filter((a) => !isMedia(a.mime) || supportsMediaInToolResult(a))
311
-
312
- const output =
313
- finalAttachments.length > 0
314
- ? {
315
- text: outputText,
316
- attachments: finalAttachments,
317
- }
318
- : outputText
319
-
320
- assistantMessage.parts.push({
321
- type: ("tool-" + part.tool) as `tool-${string}`,
322
- state: "output-available",
323
- toolCallId: part.callID,
324
- input: part.state.input,
325
- output,
326
- ...(part.metadata?.providerExecuted ? { providerExecuted: true } : {}),
327
- ...(differentModel ? {} : { callProviderMetadata: providerMeta(part.metadata) }),
328
- })
329
- }
330
- if (part.state.status === "error") {
331
- const output = part.state.metadata?.interrupted === true ? part.state.metadata.output : undefined
332
- if (typeof output === "string") {
333
- assistantMessage.parts.push({
334
- type: ("tool-" + part.tool) as `tool-${string}`,
335
- state: "output-available",
336
- toolCallId: part.callID,
337
- input: part.state.input,
338
- output,
339
- ...(part.metadata?.providerExecuted ? { providerExecuted: true } : {}),
340
- ...(differentModel ? {} : { callProviderMetadata: providerMeta(part.metadata) }),
341
- })
342
- } else {
343
- assistantMessage.parts.push({
344
- type: ("tool-" + part.tool) as `tool-${string}`,
345
- state: "output-error",
346
- toolCallId: part.callID,
347
- input: part.state.input,
348
- errorText: part.state.error,
349
- ...(part.metadata?.providerExecuted ? { providerExecuted: true } : {}),
350
- ...(differentModel ? {} : { callProviderMetadata: providerMeta(part.metadata) }),
351
- })
352
- }
353
- }
354
- // Handle pending/running tool calls to prevent dangling tool_use blocks
355
- // Anthropic/Claude APIs require every tool_use to have a corresponding tool_result
356
- if (part.state.status === "pending" || part.state.status === "running")
357
- assistantMessage.parts.push({
358
- type: ("tool-" + part.tool) as `tool-${string}`,
359
- state: "output-error",
360
- toolCallId: part.callID,
361
- input: part.state.input,
362
- errorText: "[Tool execution was interrupted]",
363
- ...(part.metadata?.providerExecuted ? { providerExecuted: true } : {}),
364
- ...(differentModel ? {} : { callProviderMetadata: providerMeta(part.metadata) }),
365
- })
366
- }
367
- if (part.type === "reasoning") {
368
- if (differentModel) {
369
- if (part.text.trim().length > 0)
370
- assistantMessage.parts.push({
371
- type: "text",
372
- text: part.text,
373
- })
374
- continue
375
- }
376
- assistantMessage.parts.push({
377
- type: "reasoning",
378
- text: part.text,
379
- providerMetadata: part.metadata,
380
- })
381
- }
382
- }
383
- if (assistantMessage.parts.length > 0) {
384
- result.push(assistantMessage)
385
- // Inject pending media as a user message for providers that don't support
386
- // media (images, PDFs) in tool results
387
- if (media.length > 0) {
388
- result.push({
389
- id: MessageID.ascending(),
390
- role: "user",
391
- parts: [
392
- {
393
- type: "text" as const,
394
- text: SYNTHETIC_ATTACHMENT_PROMPT,
395
- },
396
- ...media.map((attachment) => ({
397
- type: "file" as const,
398
- url: attachment.url,
399
- mediaType: attachment.mime,
400
- filename: attachment.filename,
401
- })),
402
- ],
403
- })
404
- }
405
- }
406
- }
407
- }
408
-
409
- const tools = Object.fromEntries(Array.from(toolNames).map((toolName) => [toolName, { toModelOutput }]))
410
-
411
- return yield* Effect.promise(() =>
412
- convertToModelMessages(
413
- result.filter((msg) => msg.parts.some((part) => part.type !== "step-start")),
414
- {
415
- //@ts-expect-error (convertToModelMessages expects a ToolSet but only actually needs tools[name]?.toModelOutput)
416
- tools,
417
- },
418
- ),
419
- )
420
- })
421
-
422
- export function toModelMessages(
423
- input: WithParts[],
424
- model: Provider.Model,
425
- options?: { stripMedia?: boolean; toolOutputMaxChars?: number },
426
- ): Promise<ModelMessage[]> {
427
- return Effect.runPromise(toModelMessagesEffect(input, model, options))
428
- }
429
-
430
- export const page = Effect.fn("MessageV2.page")(function* (input: {
431
- sessionID: SessionID
432
- limit: number
433
- before?: string
434
- }) {
435
- const { db } = yield* Database.Service
436
- const before = input.before ? cursor.decode(input.before) : undefined
437
- const where = before
438
- ? and(eq(MessageTable.session_id, input.sessionID), older(before))
439
- : eq(MessageTable.session_id, input.sessionID)
440
- const rows = yield* db
441
- .select()
442
- .from(MessageTable)
443
- .where(where)
444
- .orderBy(desc(MessageTable.time_created), desc(MessageTable.id))
445
- .limit(input.limit + 1)
446
- .all()
447
- .pipe(Effect.orDie)
448
- if (rows.length === 0) {
449
- const row = yield* db
450
- .select({ id: SessionTable.id })
451
- .from(SessionTable)
452
- .where(eq(SessionTable.id, input.sessionID))
453
- .get()
454
- .pipe(Effect.orDie)
455
- if (!row) return yield* new NotFoundError({ message: `Session not found: ${input.sessionID}` })
456
- return {
457
- items: [] as WithParts[],
458
- more: false,
459
- }
460
- }
461
-
462
- const more = rows.length > input.limit
463
- const slice = more ? rows.slice(0, input.limit) : rows
464
- const items = yield* hydrate(db, slice)
465
- items.reverse()
466
- const tail = slice.at(-1)
467
- return {
468
- items,
469
- more,
470
- cursor: more && tail ? cursor.encode({ id: tail.id, time: tail.time_created }) : undefined,
471
- }
472
- })
473
-
474
- export function stream(sessionID: SessionID) {
475
- const size = 50
476
- return Effect.gen(function* () {
477
- const result = [] as WithParts[]
478
- let before: string | undefined
479
- while (true) {
480
- const next = yield* page({ sessionID, limit: size, before }).pipe(
481
- Effect.catchIf(NotFoundError.isInstance, () =>
482
- Effect.succeed({ items: [] as WithParts[], more: false, cursor: undefined }),
483
- ),
484
- )
485
- if (next.items.length === 0) break
486
- for (let i = next.items.length - 1; i >= 0; i--) {
487
- const item = next.items[i]
488
- if (item) result.push(item)
489
- }
490
- if (!next.more || !next.cursor) break
491
- before = next.cursor
492
- }
493
- return result
494
- })
495
- }
496
-
497
- export function parts(messageID: MessageID) {
498
- return Effect.gen(function* () {
499
- const { db } = yield* Database.Service
500
- const rows = yield* db
501
- .select()
502
- .from(PartTable)
503
- .where(eq(PartTable.message_id, messageID))
504
- .orderBy(PartTable.id)
505
- .all()
506
- .pipe(Effect.orDie)
507
- return rows.map(part)
508
- })
509
- }
510
-
511
- export const get = Effect.fn("MessageV2.get")(function* (input: { sessionID: SessionID; messageID: MessageID }) {
512
- const { db } = yield* Database.Service
513
- const row = yield* db
514
- .select()
515
- .from(MessageTable)
516
- .where(and(eq(MessageTable.id, input.messageID), eq(MessageTable.session_id, input.sessionID)))
517
- .get()
518
- .pipe(Effect.orDie)
519
- if (!row) return yield* new NotFoundError({ message: `Message not found: ${input.messageID}` })
520
- return {
521
- info: info(row),
522
- parts: yield* parts(input.messageID),
523
- }
524
- })
525
-
526
- export function filterCompacted(msgs: Iterable<WithParts>) {
527
- const result = [] as WithParts[]
528
- const completed = new Set<string>()
529
- let retain: MessageID | undefined
530
- for (const msg of msgs) {
531
- result.push(msg)
532
- if (retain) {
533
- if (msg.info.id === retain) break
534
- continue
535
- }
536
- if (msg.info.role === "user" && completed.has(msg.info.id)) {
537
- const part = msg.parts.find((item): item is CompactionPart => item.type === "compaction")
538
- if (!part) continue
539
- if (!part.tail_start_id) break
540
- retain = part.tail_start_id
541
- if (msg.info.id === retain) break
542
- continue
543
- }
544
- if (msg.info.role === "user" && completed.has(msg.info.id) && msg.parts.some((part) => part.type === "compaction"))
545
- break
546
- if (msg.info.role === "assistant" && msg.info.summary && msg.info.finish && !msg.info.error)
547
- completed.add(msg.info.parentID)
548
- }
549
- result.reverse()
550
- const compactionIndex = result.findLastIndex(
551
- (msg) =>
552
- msg.info.role === "user" &&
553
- msg.parts.some((item): item is CompactionPart => item.type === "compaction" && item.tail_start_id !== undefined),
554
- )
555
- const compaction = result[compactionIndex]
556
- const part = compaction?.parts.find(
557
- (item): item is CompactionPart => item.type === "compaction" && item.tail_start_id !== undefined,
558
- )
559
- const summaryIndex = compaction
560
- ? result.findIndex(
561
- (msg, index) =>
562
- index > compactionIndex &&
563
- msg.info.role === "assistant" &&
564
- msg.info.summary &&
565
- msg.info.parentID === compaction.info.id,
566
- )
567
- : -1
568
- const tailIndex = part?.tail_start_id ? result.findIndex((msg) => msg.info.id === part.tail_start_id) : -1
569
- if (tailIndex >= 0 && tailIndex < compactionIndex && summaryIndex > compactionIndex) {
570
- return [
571
- ...result.slice(compactionIndex, summaryIndex + 1),
572
- ...result.slice(tailIndex, compactionIndex),
573
- ...result.slice(summaryIndex + 1),
574
- ]
575
- }
576
- return result
577
- }
578
-
579
- export const filterCompactedEffect = Effect.fnUntraced(function* (sessionID: SessionID) {
580
- return filterCompacted(yield* stream(sessionID))
581
- })
582
-
583
- // filterCompacted reorders messages for model consumption
584
- // ([compaction-user, summary, ...retained tail..., continue-user]), so array
585
- // position is not chronological. IDs are only a deterministic tie-breaker
586
- // because imported messages do not necessarily have monotonic IDs.
587
- export function latest(msgs: WithParts[]) {
588
- let user: User | undefined
589
- let assistant: Assistant | undefined
590
- let finished: Assistant | undefined
591
- for (const msg of msgs) {
592
- const info = msg.info
593
- if (info.role === "user" && isAfter(info, user)) user = info
594
- if (info.role === "assistant" && isAfter(info, assistant)) assistant = info
595
- if (info.role === "assistant" && info.finish && isAfter(info, finished)) finished = info
596
- }
597
- const tasks = msgs.flatMap((m) =>
598
- finished && !isAfter(m.info, finished)
599
- ? []
600
- : m.parts.filter((p): p is CompactionPart | SubtaskPart => p.type === "compaction" || p.type === "subtask"),
601
- )
602
- return { user, assistant, finished, tasks }
603
- }
604
-
605
- function isAfter(info: Info, other?: Info) {
606
- if (!other) return true
607
- if (info.time.created !== other.time.created) return info.time.created > other.time.created
608
- return info.id > other.id
609
- }
610
-
611
- export function fromError(
612
- e: unknown,
613
- ctx: { providerID: ProviderV2.ID; aborted?: boolean },
614
- ): NonNullable<Assistant["error"]> {
615
- switch (true) {
616
- case e instanceof DOMException && e.name === "AbortError":
617
- return new AbortedError(
618
- { message: e.message },
619
- {
620
- cause: e,
621
- },
622
- ).toObject()
623
- case OutputLengthError.isInstance(e):
624
- return e
625
- case LoadAPIKeyError.isInstance(e):
626
- return new AuthError(
627
- {
628
- providerID: ctx.providerID,
629
- message: e.message,
630
- },
631
- { cause: e },
632
- ).toObject()
633
- case (e as SystemError)?.code === "ECONNRESET":
634
- return new APIError(
635
- {
636
- message: "Connection reset by server",
637
- isRetryable: true,
638
- metadata: {
639
- code: (e as SystemError).code ?? "",
640
- syscall: (e as SystemError).syscall ?? "",
641
- message: (e as SystemError).message ?? "",
642
- },
643
- },
644
- { cause: e },
645
- ).toObject()
646
- case e instanceof Error && (e as FetchDecompressionError).code === "ZlibError":
647
- if (ctx.aborted) {
648
- return new AbortedError({ message: e.message }, { cause: e }).toObject()
649
- }
650
- return new APIError(
651
- {
652
- message: "Response decompression failed",
653
- isRetryable: true,
654
- metadata: {
655
- code: (e as FetchDecompressionError).code,
656
- message: e.message,
657
- },
658
- },
659
- { cause: e },
660
- ).toObject()
661
- case e instanceof ProviderError.HeaderTimeoutError:
662
- return new APIError(
663
- {
664
- message: e.message,
665
- isRetryable: true,
666
- metadata: {
667
- code: e.name,
668
- timeoutMs: String(e.ms),
669
- },
670
- },
671
- { cause: e },
672
- ).toObject()
673
- case e instanceof ProviderError.ResponseStreamError:
674
- return new APIError(
675
- {
676
- message: e.message,
677
- isRetryable: true,
678
- metadata: {
679
- code: e.name,
680
- },
681
- },
682
- { cause: e },
683
- ).toObject()
684
- case APICallError.isInstance(e):
685
- const parsed = ProviderError.parseAPICallError({
686
- providerID: ctx.providerID,
687
- error: e,
688
- })
689
- if (parsed.type === "context_overflow") {
690
- return new ContextOverflowError(
691
- {
692
- message: parsed.message,
693
- responseBody: parsed.responseBody,
694
- },
695
- { cause: e },
696
- ).toObject()
697
- }
698
-
699
- return new APIError(
700
- {
701
- message: parsed.message,
702
- statusCode: parsed.statusCode,
703
- isRetryable: parsed.isRetryable,
704
- responseHeaders: parsed.responseHeaders,
705
- responseBody: parsed.responseBody,
706
- metadata: parsed.metadata,
707
- },
708
- { cause: e },
709
- ).toObject()
710
- case e instanceof Error:
711
- return new NamedError.Unknown({ message: errorMessage(e) }, { cause: e }).toObject()
712
- default:
713
- try {
714
- const parsed = ProviderError.parseStreamError(e)
715
- if (parsed) {
716
- if (parsed.type === "context_overflow") {
717
- return new ContextOverflowError(
718
- {
719
- message: parsed.message,
720
- responseBody: parsed.responseBody,
721
- },
722
- { cause: e },
723
- ).toObject()
724
- }
725
- return new APIError(
726
- {
727
- message: parsed.message,
728
- isRetryable: parsed.isRetryable,
729
- responseBody: parsed.responseBody,
730
- },
731
- {
732
- cause: e,
733
- },
734
- ).toObject()
735
- }
736
- } catch {}
737
- return new NamedError.Unknown({ message: JSON.stringify(e) }, { cause: e }).toObject()
738
- }
739
- }
740
-
741
- export * as MessageV2 from "./message-v2"
742
- export const node = LayerNode.group([Database.node])