diff --git a/packages/core/test/session-runner.test.ts b/packages/core/test/session-runner.test.ts
index 90ccd9374ef..5f5aa218304 100644
--- a/packages/core/test/session-runner.test.ts
+++ b/packages/core/test/session-runner.test.ts
@@ -1610,6 +1610,7 @@ describe("SessionRunnerLLM", () => {
it.effect("seeds a fork with the parent's newest instruction values", () =>
Effect.gen(function* () {
const session = yield* setup
+ const database = yield* Database.Service
yield* runPrompt(session, "First")
systemBaseline = "Changed context"
const second = yield* runPrompt(session, "Second")
@@ -1618,7 +1619,7 @@ describe("SessionRunnerLLM", () => {
const forked = yield* session.fork({ sessionID, boundary: { type: "before", messageID: second.id } })
expect(
- yield* (yield* Database.Service).db
+ yield* database.db
.select()
.from(InstructionStateTable)
.where(eq(InstructionStateTable.session_id, forked.id))
@@ -1666,6 +1667,7 @@ describe("SessionRunnerLLM", () => {
it.effect("keeps nested forks self-contained", () =>
Effect.gen(function* () {
const session = yield* setup
+ const database = yield* Database.Service
yield* runPrompt(session, "First")
systemBaseline = "Changed context"
const second = yield* runPrompt(session, "Second")
@@ -1681,7 +1683,7 @@ describe("SessionRunnerLLM", () => {
})
expect(
- yield* (yield* Database.Service).db
+ yield* database.db
.select()
.from(InstructionStateTable)
.where(eq(InstructionStateTable.session_id, grandchild.id))
@@ -2496,6 +2498,7 @@ describe("SessionRunnerLLM", () => {
it.effect("automatically compacts into a completed summary and retained recent turn", () =>
Effect.gen(function* () {
const session = yield* setup
+ const store = yield* SessionStore.Service
yield* TestLLM.push(TestLLM.textWithUsage("Earlier answer", "text-first", 3_950))
yield* runPrompt(session, "Earlier question ".repeat(180))
@@ -2513,7 +2516,7 @@ describe("SessionRunnerLLM", () => {
expect(userTexts(requests[1])[0]).toContain("\n## Objective\n- Preserve the task\n")
expect(userTexts(requests[1])[0]).toContain(`[User]: ${"Recent exact request ".repeat(180)}`)
- const context = yield* (yield* SessionStore.Service).context(sessionID)
+ const context = yield* store.context(sessionID)
expect(context.map((message) => message.type)).toEqual(["compaction", "assistant"])
expect(context[0]).toMatchObject({
type: "compaction",
@@ -2533,7 +2536,7 @@ describe("SessionRunnerLLM", () => {
"\n## Objective\n- Preserve the task\n",
)
expect(userTexts(requests[0])[0]).toContain("Recent exact request")
- expect((yield* (yield* SessionStore.Service).context(sessionID))[0]).toMatchObject({
+ expect((yield* store.context(sessionID))[0]).toMatchObject({
type: "compaction",
summary: "## Objective\n- Preserve the updated task",
})
@@ -3683,6 +3686,7 @@ describe("SessionRunnerLLM", () => {
it.effect("coalesces multiple active steering prompts into one continuation step", () =>
Effect.gen(function* () {
const session = yield* setup
+ const execution = yield* SessionExecution.Service
yield* admit(session, "Start working")
yield* TestLLM.push(TestLLM.stop(), TestLLM.stop())
@@ -3698,7 +3702,7 @@ describe("SessionRunnerLLM", () => {
expect(requests).toHaveLength(2)
expect(userTexts(requests[1])).toEqual(["Start working", "First steer", "Second steer"])
- yield* (yield* SessionExecution.Service).wake(sessionID)
+ yield* execution.wake(sessionID)
yield* Effect.yieldNow
expect(requests).toHaveLength(2)
}),
@@ -3933,6 +3937,7 @@ describe("SessionRunnerLLM", () => {
it.effect("promotes the first queued input when woken while idle", () =>
Effect.gen(function* () {
const session = yield* setup
+ const execution = yield* SessionExecution.Service
yield* session.prompt({
sessionID,
text: "Wait in queue",
@@ -3941,7 +3946,7 @@ describe("SessionRunnerLLM", () => {
})
const stream = yield* TestLLM.gate
- yield* (yield* SessionExecution.Service).wake(sessionID)
+ yield* execution.wake(sessionID)
yield* stream.started
yield* stream.release
@@ -3953,6 +3958,7 @@ describe("SessionRunnerLLM", () => {
it.effect("retries inbox input after prompt projection rolls back", () =>
Effect.gen(function* () {
const session = yield* setup
+ const execution = yield* SessionExecution.Service
const bus = yield* Bus.Service
const defect = new Error("fail after prompt promotion")
let fail = true
@@ -3965,7 +3971,7 @@ describe("SessionRunnerLLM", () => {
yield* TestLLM.push(TestLLM.stop())
const stream = yield* TestLLM.gate
- yield* (yield* SessionExecution.Service).wake(sessionID)
+ yield* execution.wake(sessionID)
yield* stream.started
yield* stream.release