agaltsovav/opencode
1
0
Fork 0
Code Issues Pull Requests Actions 80 Packages Projects Releases Wiki Activity

fix(session): make message reads effectful (#27291)

Browse Source
This commit is contained in:
Shoubhit Dash
2026-05-13 14:40:12 +05:30
committed by GitHub
parent 4b041716fc
commit ccf93f3523
7 changed files with 60 additions and 99 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/opencode/test/session/messages-pagination.test.ts
+33 -61
View File
@@ -12,20 +12,6 @@ void Log.init({ print: false })
const it = testEffect(SessionNs.defaultLayer)
function expectNotFound(fn: () => unknown, message: string) {
let thrown: unknown
try {
fn()
} catch (error) {
thrown = error
}
expect(thrown).toBeInstanceOf(NotFoundError)
if (thrown instanceof NotFoundError) {
expect(thrown._tag).toBe("NotFoundError")
expect(thrown.message).toBe(message)
}
}
const withSession = <A, E, R>(
fn: (input: { session: SessionNs.Interface; sessionID: SessionID }) => Effect.Effect<A, E, R>,
) =>
@@ -140,12 +126,12 @@ const addCompactionPart = Effect.fn("Test.addCompactionPart")(function* (
})
describe("MessageV2.page", () => {
it.instance("returns sync result", () =>
it.instance("returns page result", () =>
withSession(({ sessionID }) =>
Effect.gen(function* () {
yield* fill(sessionID, 2)
const result = MessageV2.page({ sessionID, limit: 10 })
const result = yield* MessageV2.page({ sessionID, limit: 10 })
expect(result).toBeDefined()
expect(result.items).toBeArray()
}),
@@ -157,18 +143,18 @@ describe("MessageV2.page", () => {
Effect.gen(function* () {
const ids = yield* fill(sessionID, 6)
const a = MessageV2.page({ sessionID, limit: 2 })
const a = yield* MessageV2.page({ sessionID, limit: 2 })
expect(a.items.map((item) => item.info.id)).toEqual(ids.slice(-2))
expect(a.items.every((item) => item.parts.length === 1)).toBe(true)
expect(a.more).toBe(true)
expect(a.cursor).toBeTruthy()
const b = MessageV2.page({ sessionID, limit: 2, before: a.cursor! })
const b = yield* MessageV2.page({ sessionID, limit: 2, before: a.cursor! })
expect(b.items.map((item) => item.info.id)).toEqual(ids.slice(-4, -2))
expect(b.more).toBe(true)
expect(b.cursor).toBeTruthy()
const c = MessageV2.page({ sessionID, limit: 2, before: b.cursor! })
const c = yield* MessageV2.page({ sessionID, limit: 2, before: b.cursor! })
expect(c.items.map((item) => item.info.id)).toEqual(ids.slice(0, 2))
expect(c.more).toBe(false)
expect(c.cursor).toBeUndefined()
@@ -181,7 +167,7 @@ describe("MessageV2.page", () => {
Effect.gen(function* () {
const ids = yield* fill(sessionID, 4)
const result = MessageV2.page({ sessionID, limit: 4 })
const result = yield* MessageV2.page({ sessionID, limit: 4 })
expect(result.items.map((item) => item.info.id)).toEqual(ids)
}),
),
@@ -190,7 +176,7 @@ describe("MessageV2.page", () => {
it.instance("returns empty items for session with no messages", () =>
withSession(({ sessionID }) =>
Effect.gen(function* () {
const result = MessageV2.page({ sessionID, limit: 10 })
const result = yield* MessageV2.page({ sessionID, limit: 10 })
expect(result.items).toEqual([])
expect(result.more).toBe(false)
expect(result.cursor).toBeUndefined()
@@ -198,17 +184,10 @@ describe("MessageV2.page", () => {
),
)
it.instance("throws NotFoundError for non-existent session", () =>
it.instance("fails with NotFoundError for non-existent session", () =>
Effect.gen(function* () {
const fake = "non-existent-session" as SessionID
expectNotFound(() => MessageV2.page({ sessionID: fake, limit: 10 }), `Session not found: ${fake}`)
}),
)
it.instance("fails pageEffect with NotFoundError for non-existent session", () =>
Effect.gen(function* () {
const fake = "non-existent-session" as SessionID
const error = yield* Effect.flip(MessageV2.pageEffect({ sessionID: fake, limit: 10 }))
const error = yield* Effect.flip(MessageV2.page({ sessionID: fake, limit: 10 }))
expect(error).toBeInstanceOf(NotFoundError)
expect(error.message).toBe(`Session not found: ${fake}`)
}),
@@ -219,7 +198,7 @@ describe("MessageV2.page", () => {
Effect.gen(function* () {
const ids = yield* fill(sessionID, 3)
const result = MessageV2.page({ sessionID, limit: 3 })
const result = yield* MessageV2.page({ sessionID, limit: 3 })
expect(result.items.map((item) => item.info.id)).toEqual(ids)
expect(result.more).toBe(false)
expect(result.cursor).toBeUndefined()
@@ -232,7 +211,7 @@ describe("MessageV2.page", () => {
Effect.gen(function* () {
const ids = yield* fill(sessionID, 5)
const result = MessageV2.page({ sessionID, limit: 1 })
const result = yield* MessageV2.page({ sessionID, limit: 1 })
expect(result.items).toHaveLength(1)
expect(result.items[0].info.id).toBe(ids[ids.length - 1])
expect(result.more).toBe(true)
@@ -253,7 +232,7 @@ describe("MessageV2.page", () => {
text: "extra",
})
const result = MessageV2.page({ sessionID, limit: 10 })
const result = yield* MessageV2.page({ sessionID, limit: 10 })
expect(result.items).toHaveLength(1)
expect(result.items[0].parts).toHaveLength(2)
}),
@@ -265,8 +244,8 @@ describe("MessageV2.page", () => {
Effect.gen(function* () {
const ids = yield* fill(sessionID, 4, (i: number) => 1000.5 + i)
const a = MessageV2.page({ sessionID, limit: 2 })
const b = MessageV2.page({ sessionID, limit: 2, before: a.cursor! })
const a = yield* MessageV2.page({ sessionID, limit: 2 })
const b = yield* MessageV2.page({ sessionID, limit: 2, before: a.cursor! })
expect(a.items.map((item) => item.info.id)).toEqual(ids.slice(-2))
expect(b.items.map((item) => item.info.id)).toEqual(ids.slice(0, 2))
@@ -279,11 +258,11 @@ describe("MessageV2.page", () => {
Effect.gen(function* () {
const ids = yield* fill(sessionID, 4, () => 1000)
const a = MessageV2.page({ sessionID, limit: 2 })
const a = yield* MessageV2.page({ sessionID, limit: 2 })
expect(a.items.map((item) => item.info.id)).toEqual(ids.slice(-2))
expect(a.more).toBe(true)
const b = MessageV2.page({ sessionID, limit: 2, before: a.cursor! })
const b = yield* MessageV2.page({ sessionID, limit: 2, before: a.cursor! })
expect(b.items.map((item) => item.info.id)).toEqual(ids.slice(0, 2))
expect(b.more).toBe(false)
}),
@@ -298,8 +277,8 @@ describe("MessageV2.page", () => {
yield* fill(a.id, 3)
yield* fill(b.id, 2)
const resultA = MessageV2.page({ sessionID: a.id, limit: 10 })
const resultB = MessageV2.page({ sessionID: b.id, limit: 10 })
const resultA = yield* MessageV2.page({ sessionID: a.id, limit: 10 })
const resultB = yield* MessageV2.page({ sessionID: b.id, limit: 10 })
expect(resultA.items).toHaveLength(3)
expect(resultB.items).toHaveLength(2)
expect(resultA.items.every((item) => item.info.sessionID === a.id)).toBe(true)
@@ -315,7 +294,7 @@ describe("MessageV2.page", () => {
Effect.gen(function* () {
const ids = yield* fill(sessionID, 10)
const result = MessageV2.page({ sessionID, limit: 100 })
const result = yield* MessageV2.page({ sessionID, limit: 100 })
expect(result.items).toHaveLength(10)
expect(result.items.map((item) => item.info.id)).toEqual(ids)
expect(result.more).toBe(false)
@@ -482,7 +461,7 @@ describe("MessageV2.get", () => {
Effect.gen(function* () {
const [id] = yield* fill(sessionID, 1)
const result = MessageV2.get({ sessionID, messageID: id })
const result = yield* MessageV2.get({ sessionID, messageID: id })
expect(result.info.id).toBe(id)
expect(result.info.sessionID).toBe(sessionID)
expect(result.info.role).toBe("user")
@@ -492,20 +471,11 @@ describe("MessageV2.get", () => {
),
)
it.instance("throws NotFoundError for non-existent message", () =>
it.instance("fails with NotFoundError for non-existent message", () =>
withSession(({ sessionID }) =>
Effect.gen(function* () {
const messageID = MessageID.ascending()
expectNotFound(() => MessageV2.get({ sessionID, messageID }), `Message not found: ${messageID}`)
}),
),
)
it.instance("fails getEffect with NotFoundError for non-existent message", () =>
withSession(({ sessionID }) =>
Effect.gen(function* () {
const messageID = MessageID.ascending()
const error = yield* Effect.flip(MessageV2.getEffect({ sessionID, messageID }))
const error = yield* Effect.flip(MessageV2.get({ sessionID, messageID }))
expect(error).toBeInstanceOf(NotFoundError)
expect(error.message).toBe(`Message not found: ${messageID}`)
}),
@@ -519,8 +489,10 @@ describe("MessageV2.get", () => {
const b = yield* session.create({})
const [id] = yield* fill(a.id, 1)
expectNotFound(() => MessageV2.get({ sessionID: b.id, messageID: id }), `Message not found: ${id}`)
const result = MessageV2.get({ sessionID: a.id, messageID: id })
const error = yield* Effect.flip(MessageV2.get({ sessionID: b.id, messageID: id }))
expect(error).toBeInstanceOf(NotFoundError)
expect(error.message).toBe(`Message not found: ${id}`)
const result = yield* MessageV2.get({ sessionID: a.id, messageID: id })
expect(result.info.id).toBe(id)
yield* session.remove(a.id)
@@ -541,7 +513,7 @@ describe("MessageV2.get", () => {
text: "extra",
})
const result = MessageV2.get({ sessionID, messageID: id })
const result = yield* MessageV2.get({ sessionID, messageID: id })
expect(result.parts).toHaveLength(2)
}),
),
@@ -561,7 +533,7 @@ describe("MessageV2.get", () => {
text: "response",
})
const result = MessageV2.get({ sessionID, messageID: aid })
const result = yield* MessageV2.get({ sessionID, messageID: aid })
expect(result.info.role).toBe("assistant")
expect(result.parts).toHaveLength(1)
expect((result.parts[0] as MessageV2.TextPart).text).toBe("response")
@@ -574,7 +546,7 @@ describe("MessageV2.get", () => {
Effect.gen(function* () {
const id = yield* addUser(sessionID)
const result = MessageV2.get({ sessionID, messageID: id })
const result = yield* MessageV2.get({ sessionID, messageID: id })
expect(result.info.id).toBe(id)
expect(result.parts).toEqual([])
}),
@@ -1026,9 +998,9 @@ describe("MessageV2 consistency", () => {
Effect.gen(function* () {
yield* fill(sessionID, 3)
const paged = MessageV2.page({ sessionID, limit: 10 })
const paged = yield* MessageV2.page({ sessionID, limit: 10 })
for (const item of paged.items) {
const got = MessageV2.get({ sessionID, messageID: item.info.id as MessageID })
const got = yield* MessageV2.get({ sessionID, messageID: item.info.id as MessageID })
expect(got.info).toEqual(item.info)
expect(got.parts).toEqual(item.parts)
}
@@ -1041,7 +1013,7 @@ describe("MessageV2 consistency", () => {
Effect.gen(function* () {
const [id] = yield* fill(sessionID, 1)
const got = MessageV2.get({ sessionID, messageID: id })
const got = yield* MessageV2.get({ sessionID, messageID: id })
const standalone = MessageV2.parts(id)
expect(got.parts).toEqual(standalone)
}),
@@ -1058,7 +1030,7 @@ describe("MessageV2 consistency", () => {
const paged = [] as MessageV2.WithParts[]
let cursor: string | undefined
while (true) {
const result = MessageV2.page({ sessionID, limit: 3, before: cursor })
const result = yield* MessageV2.page({ sessionID, limit: 3, before: cursor })
for (let i = result.items.length - 1; i >= 0; i--) {
paged.push(result.items[i])
}
packages/opencode/test/session/processor-effect.test.ts
+2 -2
View File
@@ -767,7 +767,7 @@ it.live("session.processor effect tests record aborted errors and idle state", (
const exit = yield* Fiber.await(run)
yield* Effect.promise(() => seen.promise)
const stored = MessageV2.get({ sessionID: chat.id, messageID: msg.id })
const stored = yield* MessageV2.get({ sessionID: chat.id, messageID: msg.id })
const state = yield* sts.get(chat.id)
off()
@@ -829,7 +829,7 @@ it.live("session.processor effect tests mark interruptions aborted without manua
yield* Fiber.interrupt(run)
const exit = yield* Fiber.await(run)
const stored = MessageV2.get({ sessionID: chat.id, messageID: msg.id })
const stored = yield* MessageV2.get({ sessionID: chat.id, messageID: msg.id })
const state = yield* sts.get(chat.id)
expect(Exit.isFailure(exit)).toBe(true)
packages/opencode/test/session/prompt.test.ts
+4 -4
View File
@@ -1787,7 +1787,7 @@ it.instance(
if (msg.info.role !== "user") throw new Error("expected user message")
const stored = MessageV2.get({
const stored = yield* MessageV2.get({
sessionID: session.id,
messageID: msg.info.id,
})
@@ -1875,7 +1875,7 @@ it.instance(
parts: yield* prompt.resolvePromptParts("Use @docs for context"),
})
const stored = MessageV2.get({ sessionID: session.id, messageID: message.info.id })
const stored = yield* MessageV2.get({ sessionID: session.id, messageID: message.info.id })
const synthetic = stored.parts.filter(
(part): part is MessageV2.TextPart => part.type === "text" && part.synthetic === true,
)
@@ -1931,7 +1931,7 @@ it.instance(
],
})
const stored = MessageV2.get({ sessionID: session.id, messageID: message.info.id })
const stored = yield* MessageV2.get({ sessionID: session.id, messageID: message.info.id })
const synthetic = stored.parts.filter(
(part): part is MessageV2.TextPart => part.type === "text" && part.synthetic === true,
)
@@ -1991,7 +1991,7 @@ it.instance(
parts,
noReply: true,
})
const stored = MessageV2.get({ sessionID: session.id, messageID: message.info.id })
const stored = yield* MessageV2.get({ sessionID: session.id, messageID: message.info.id })
const textParts = stored.parts.filter((part) => part.type === "text")
const hasContent = textParts.some((part) => part.text.includes("special content"))
expect(hasContent).toBe(true)