Something went wrong. Try again.
opencode v2 mirror
Something went wrong. Try again.
TypeScript
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584import { beforeEach, expect } from "bun:test"import { AIError, LLMClient, LLMEvent, LanguageModel, Message, SystemPart, TransportError, type LLMRequest,} from "@opencode/ai"import { OpenAIChat } from "@opencode/ai/protocols"import { Agent } from "@opencode/core/agent"import { Database } from "@opencode/core/database/database"import { AppNodeBuilder } from "@opencode/core/effect/app-node-builder"import { llmClient } from "@opencode/core/effect/app-node-platform"import { LayerNode } from "@opencode/util/effect/layer-node"import { Bus } from "@opencode/core/bus"import { SessionEvent } from "@opencode/core/session/event"import { SessionMessage } from "@opencode/core/session/message"import { SessionProjector } from "@opencode/core/session/projector"import { SessionRunnerModel } from "@opencode/core/session/runner/model"import { SessionTable } from "@opencode/core/session/sql"import { SessionStore } from "@opencode/core/session/store"import { SessionTitle } from "@opencode/core/session/title"import { PluginHooks } from "@opencode/core/plugin/hooks"import { PluginSupervisor } from "@opencode/core/plugin/supervisor"import { Location } from "@opencode/core/location"import { Session } from "@opencode/core/session"import { Project } from "@opencode/core/project"import { ProjectTable } from "@opencode/core/project/sql"import { App } from "@opencode/core/app"import { Model } from "@opencode/core/model"import { Provider } from "@opencode/core/provider"import { AbsolutePath } from "@opencode/core/schema"import { Money } from "@opencode/schema/money"import { Deferred, Effect, Fiber, Layer, Stream } from "effect"import { testEffect } from "./lib/effect"
let requests: LLMRequest[] = []let selectedSmall: Model.Info | undefinedlet selections: Array<Session.Info["model"]> = []const model = LanguageModel.make({ id: "title-model", provider: "test", route: OpenAIChat.route,})const smallModel = LanguageModel.make({ id: "title-small", provider: "test", route: OpenAIChat.route,})const cost = [ { input: Money.USDPerMillionTokens.make(1), output: Money.USDPerMillionTokens.make(2), cache: { read: Money.USDPerMillionTokens.make(0.1), write: Money.USDPerMillionTokens.make(0.5), }, },]const successfulTitle = () => Stream.make( LLMEvent.textDelta({ id: "title", text: "Generated Title\n" }), LLMEvent.stepFinish({ index: 0, reason: { normalized: "stop" }, usage: { inputTokens: 15, outputTokens: 6, nonCachedInputTokens: 10, cacheReadInputTokens: 3, cacheWriteInputTokens: 2, reasoningTokens: 2, }, }), LLMEvent.finish({ reason: { normalized: "stop" }, }), )let titleStream: () => Stream.Stream<LLMEvent, AIError> = successfulTitleconst client = Layer.mock(LLMClient.Service)({ stream: (request: LLMRequest) => { requests.push(request) return titleStream() }, generate: () => Effect.die("unused"),})const models = Layer.mock(SessionRunnerModel.Service)({ resolve: (session) => { selections.push(session.model) return Effect.succeed( SessionRunnerModel.resolved(session.model?.id === "title-small" ? smallModel : model, { capabilities: { tools: true, input: ["text", "image"], output: ["text"] }, cost, limit: { context: 200_000, output: 32_000 }, variant: session.model?.variant, }), ) },})const smallModels = Layer.mock(Model.Service, { small: () => Effect.succeed(selectedSmall),})const it = testEffect( AppNodeBuilder.build( LayerNode.group([ Database.node, Bus.node, SessionProjector.node, SessionStore.node, Agent.node, PluginHooks.node, SessionTitle.node, ]), [ llmClient.replace(client), Model.node.replace(smallModels), SessionRunnerModel.node.replace(models), Location.node.replace(Location.boundNode({ directory: AbsolutePath.make("/project") })), PluginSupervisor.node.replace(Layer.empty), ], ),)
const insertSession = (id: Session.ID, title?: string, created?: number, model?: Model.Ref) => Effect.gen(function* () { const { db } = yield* Database.Service yield* db .insert(ProjectTable) .values({ id: Project.ID.global, worktree: AbsolutePath.make("/project"), sandboxes: [] }) .onConflictDoNothing() .run() .pipe(Effect.orDie) yield* db .insert(SessionTable) .values({ id, project_id: Project.ID.global, slug: id, directory: "/project", title, model, time_created: created, version: "test", }) .onConflictDoNothing() .run() .pipe(Effect.orDie) })
const prompt = (sessionID: Session.ID, text: string) => Effect.gen(function* () { const bus = yield* Bus.Service const messageID = SessionMessage.ID.create() yield* bus.publish(SessionEvent.InboxEnqueued, { sessionID, inboxID: messageID, item: { type: "user", payload: { text }, delivery: "steer" }, }) yield* bus.publish(SessionEvent.InboxDelivered, { sessionID, inboxID: messageID, }) })
const small = Model.Info.make({ ...Model.Info.default(Provider.ID.make("test"), Model.ID.make("title-small")), family: Model.Family.make("gpt-luna"), capabilities: { tools: false, input: ["text"], output: ["text"] }, variants: [ { id: Model.VariantID.make("low") }, { id: Model.VariantID.make("none") }, { id: Model.VariantID.make("high") }, ],})const lowSmall = Model.Info.make({ ...small, variants: [{ id: Model.VariantID.make("low") }, { id: Model.VariantID.make("high") }],})
beforeEach(() => { requests = [] selectedSmall = undefined selections = [] titleStream = successfulTitle})
const enableTitleAgent = Effect.gen(function* () { const agents = yield* Agent.Service yield* agents.transform((editor) => { editor.update(Agent.ID.make("title"), (agent) => { agent.mode = "primary" agent.hidden = true agent.system = "You are a title generator." }) })})
it.effect("generates a title from the sole user message and renames the session", () => Effect.gen(function* () { yield* enableTitleAgent const sessionID = Session.ID.make("ses_title_generate") yield* insertSession(sessionID) yield* prompt(sessionID, "Help me debug the failing build")
const store = yield* SessionStore.Service const title = yield* SessionTitle.Service yield* title.generate(sessionID)
expect(requests).toHaveLength(1) expect(requests[0]?.http?.headers).toEqual({ "x-session-affinity": sessionID, "X-Session-Id": sessionID, "User-Agent": App.useragent(App.make()), "x-opencode-project": Project.ID.global, "x-opencode-session": sessionID, "x-opencode-client": "opencode", }) expect(requests[0]?.promptCacheKey).toBe(sessionID) expect(requests[0]?.tools).toEqual([]) expect(requests[0]?.system.map((part) => part.text)).toEqual(["You are a title generator."]) expect(JSON.stringify(requests[0]?.messages)).toContain("Help me debug the failing build") const renamed = yield* store.get(sessionID) expect(renamed?.title).toBe("Generated Title") expect(renamed?.tokens).toEqual({ input: 10, output: 4, reasoning: 2, cache: { read: 3, write: 2 } }) expect(renamed?.cost).toBeCloseTo(0.0000233) }),)
it.effect("runs title hooks instead of context hooks", () => Effect.gen(function* () { yield* enableTitleAgent const sessionID = Session.ID.make("ses_title_hook") yield* insertSession(sessionID) yield* prompt(sessionID, "Redact this message")
const hooks = yield* PluginHooks.Service let contexts = 0 yield* hooks.register("session", "context", () => Effect.sync(() => contexts++)) yield* hooks.register("session", "title", (event) => Effect.sync(() => { expect(event.sessionID).toBe(sessionID) expect(event.system.map((part) => part.text)).toEqual(["You are a title generator."]) event.system.push(SystemPart.make("Prefer short titles.")) event.messages = [Message.user("[redacted]")] event.options.maxTokens = 32 event.options.reasoningEffort = "low" }), )
const title = yield* SessionTitle.Service yield* title.generate(sessionID)
expect(contexts).toBe(0) expect(requests).toHaveLength(1) expect(requests[0]?.system.map((part) => part.text)).toEqual(["You are a title generator.", "Prefer short titles."]) expect(JSON.stringify(requests[0]?.messages)).not.toContain("Redact this message") expect(requests[0]?.generation).toEqual(expect.objectContaining({ maxTokens: 32 })) expect(requests[0]?.providerOptions).toEqual({ reasoningEffort: "low" }) }),)
it.effect("uses a hook-provided title without a model request", () => Effect.gen(function* () { yield* enableTitleAgent const sessionID = Session.ID.make("ses_title_result") yield* insertSession(sessionID) yield* prompt(sessionID, "Hello")
const hooks = yield* PluginHooks.Service yield* hooks.register("session", "title", (event) => Effect.sync(() => { event.result = "Plugin Title" }), )
const title = yield* SessionTitle.Service yield* title.generate(sessionID)
expect(requests).toHaveLength(0) const store = yield* SessionStore.Service expect((yield* store.get(sessionID))?.title).toBe("Plugin Title") }),)
it.effect("uses a small model from the primary provider", () => Effect.gen(function* () { selectedSmall = small yield* enableTitleAgent const sessionID = Session.ID.make("ses_title_small_model") yield* insertSession(sessionID) yield* prompt(sessionID, "Use a small model for this title")
const title = yield* SessionTitle.Service yield* title.generate(sessionID)
expect(requests.map((request) => String(request.model.id))).toEqual(["title-small"]) expect(selections[1]?.variant).toBe(Model.VariantID.make("none")) const store = yield* SessionStore.Service expect((yield* store.get(sessionID))?.title).toBe("Generated Title") }),)
it.effect("falls back to the primary model when the small model fails", () => Effect.gen(function* () { titleStream = () => requests.length === 1 ? Stream.make(LLMEvent.providerError({ message: "Small model unavailable" })) : successfulTitle() selectedSmall = lowSmall yield* enableTitleAgent const sessionID = Session.ID.make("ses_title_small_fallback") yield* insertSession( sessionID, undefined, undefined, Model.Ref.make({ providerID: Provider.ID.make("test"), id: Model.ID.make("title-model"), variant: Model.VariantID.make("high"), }), ) yield* prompt(sessionID, "Fall back when title generation fails")
const attempted: Model.Ref[] = [] const hooks = yield* PluginHooks.Service yield* hooks.register("session", "model.request", (event) => Effect.sync(() => { attempted.push(event.model) }), )
const title = yield* SessionTitle.Service yield* title.generate(sessionID)
expect(requests.map((request) => String(request.model.id))).toEqual(["title-small", "title-model"]) expect(attempted.map((model) => String(model.variant))).toEqual(["low", "high"]) const store = yield* SessionStore.Service expect((yield* store.get(sessionID))?.title).toBe("Generated Title") }),)
it.effect("generates from the first user message after later messages exist", () => Effect.gen(function* () { yield* enableTitleAgent const sessionID = Session.ID.make("ses_title_second_message") yield* insertSession(sessionID) yield* prompt(sessionID, "First message") yield* prompt(sessionID, "Second message")
const store = yield* SessionStore.Service const title = yield* SessionTitle.Service yield* title.generate(sessionID)
expect(requests).toHaveLength(1) expect(JSON.stringify(requests[0]?.messages)).toContain("First message") expect(JSON.stringify(requests[0]?.messages)).not.toContain("Second message") expect((yield* store.get(sessionID))?.title).toBe("Generated Title") }),)
it.effect("retries a legacy persisted fallback title", () => Effect.gen(function* () { yield* enableTitleAgent const sessionID = Session.ID.make("ses_title_legacy") const created = Date.parse("2026-07-30T18:45:03.662Z") yield* insertSession(sessionID, "New session - 2026-07-30T18:45:03.662Z", created) yield* prompt(sessionID, "Retry the legacy title")
const title = yield* SessionTitle.Service yield* title.generate(sessionID)
const store = yield* SessionStore.Service expect(requests).toHaveLength(1) expect((yield* store.get(sessionID))?.title).toBe("Generated Title") }),)
it.effect("generates a title for an explicitly requested child session", () => Effect.gen(function* () { yield* enableTitleAgent const sessionID = Session.ID.make("ses_title_child") const { db } = yield* Database.Service yield* db .insert(ProjectTable) .values({ id: Project.ID.global, worktree: AbsolutePath.make("/project"), sandboxes: [] }) .onConflictDoNothing() .run() .pipe(Effect.orDie) yield* db .insert(SessionTable) .values({ id: sessionID, project_id: Project.ID.global, parent_id: Session.ID.make("ses_title_parent"), slug: sessionID, directory: "/project", version: "test", }) .onConflictDoNothing() .run() .pipe(Effect.orDie) yield* prompt(sessionID, "Do this subtask")
const title = yield* SessionTitle.Service yield* title.generate(sessionID) const store = yield* SessionStore.Service expect(requests).toHaveLength(1) expect((yield* store.get(sessionID))?.title).toBe("Generated Title") }),)
it.effect("does not generate when the title agent is removed", () => Effect.gen(function* () { const sessionID = Session.ID.make("ses_title_no_agent") yield* insertSession(sessionID) yield* prompt(sessionID, "Help me debug the failing build")
const store = yield* SessionStore.Service const title = yield* SessionTitle.Service yield* title.generate(sessionID)
expect(requests).toHaveLength(0) const untouched = yield* store.get(sessionID) expect(untouched?.title).toBeUndefined() }),)
it.effect("regenerates an existing title using the title agent", () => Effect.gen(function* () { yield* enableTitleAgent const sessionID = Session.ID.make("ses_title_regenerate") yield* insertSession(sessionID, "Original title") yield* prompt(sessionID, "Investigate the login failure") const events = yield* Bus.Service const assistantMessageID = SessionMessage.ID.create() yield* events.publish(SessionEvent.Step.Started, { sessionID, assistantMessageID, agent: Agent.ID.make("build"), model: Model.Ref.make({ id: Model.ID.make("title-model"), providerID: Provider.ID.make("test") }), started: 0, }) yield* events.publish(SessionEvent.Reasoning.Started, { sessionID, assistantMessageID, ordinal: 0 }) yield* events.publish(SessionEvent.Reasoning.Ended, { sessionID, assistantMessageID, ordinal: 0, text: "Private reasoning that should not appear", }) yield* events.publish(SessionEvent.Text.Started, { sessionID, assistantMessageID, ordinal: 1 }) yield* events.publish(SessionEvent.Text.Ended, { sessionID, assistantMessageID, ordinal: 1, text: "The actual issue is expired OAuth credentials.", }) yield* prompt(sessionID, "Switch to fixing OAuth token refresh")
const title = yield* SessionTitle.Service yield* title.generate(sessionID)
const store = yield* SessionStore.Service expect(requests).toHaveLength(1) expect(requests[0]?.system.map((part) => part.text)).toEqual(["You are a title generator."]) expect(JSON.stringify(requests[0]?.messages)).toContain("Investigate the login failure") expect(JSON.stringify(requests[0]?.messages)).toContain("The actual issue is expired OAuth credentials.") expect(JSON.stringify(requests[0]?.messages)).toContain("Switch to fixing OAuth token refresh") expect(JSON.stringify(requests[0]?.messages)).not.toContain("Private reasoning that should not appear") expect((yield* store.get(sessionID))?.title).toBe("Generated Title") }),)
it.effect("bounds regeneration context while preserving the original request and recent conversation", () => Effect.gen(function* () { yield* enableTitleAgent const sessionID = Session.ID.make("ses_title_regenerate_bounded") yield* insertSession(sessionID, "Original title") yield* prompt(sessionID, `ORIGINAL_GOAL ${"a".repeat(3_000)} OMITTED_ORIGINAL_END`) yield* prompt(sessionID, `OMITTED_OLD_CONTEXT ${"b".repeat(9_000)} RECENT_GOAL`)
const title = yield* SessionTitle.Service yield* title.generate(sessionID)
const content = requests[0]?.messages[0]?.content[0] expect(content?.type).toBe("text") if (content?.type !== "text") return expect(content.text.length).toBeLessThanOrEqual(8_000) expect(content.text).toContain("ORIGINAL_GOAL") expect(content.text).toContain("RECENT_GOAL") expect(content.text).not.toContain("OMITTED_ORIGINAL_END") expect(content.text).not.toContain("OMITTED_OLD_CONTEXT") }),)
it.effect("preserves the existing title when regeneration fails", () => Effect.gen(function* () { yield* enableTitleAgent const sessionID = Session.ID.make("ses_title_regenerate_failure") yield* insertSession(sessionID, "Original title") yield* prompt(sessionID, "Fail to regenerate this title") titleStream = () => Stream.make(LLMEvent.providerError({ message: "Provider unavailable" }))
const title = yield* SessionTitle.Service yield* title.generate(sessionID)
const store = yield* SessionStore.Service expect(requests).toHaveLength(1) expect((yield* store.get(sessionID))?.title).toBe("Original title") }),)
it.effect("retries after a failed title request", () => Effect.gen(function* () { yield* enableTitleAgent const sessionID = Session.ID.make("ses_title_retry") yield* insertSession(sessionID) yield* prompt(sessionID, "Retry this title") const title = yield* SessionTitle.Service titleStream = () => Stream.make(LLMEvent.providerError({ message: "Provider unavailable" }))
yield* title.generate(sessionID) titleStream = successfulTitle yield* title.generate(sessionID)
const store = yield* SessionStore.Service expect(requests).toHaveLength(2) expect((yield* store.get(sessionID))?.title).toBe("Generated Title") }),)
it.effect("does not rename after a failed title stream", () => Effect.gen(function* () { yield* enableTitleAgent const sessionID = Session.ID.make("ses_title_stream_failure") yield* insertSession(sessionID) yield* prompt(sessionID, "Fail this title stream") titleStream = () => Stream.fail( new AIError({ reason: new TransportError({ message: "Disconnected", transport: "http", operation: "request" }), }), )
const title = yield* SessionTitle.Service yield* title.generate(sessionID)
const store = yield* SessionStore.Service expect(requests).toHaveLength(1) expect((yield* store.get(sessionID))?.title).toBeUndefined() }),)
it.effect("preserves a manual rename completed while generation is in flight", () => Effect.gen(function* () { yield* enableTitleAgent const sessionID = Session.ID.make("ses_title_manual_rename") yield* insertSession(sessionID) yield* prompt(sessionID, "Generate this title") const started = yield* Deferred.make<void>() const release = yield* Deferred.make<void>() titleStream = () => Stream.unwrap( Deferred.succeed(started, undefined).pipe( Effect.andThen(Deferred.await(release)), Effect.as(successfulTitle()), ), ) const title = yield* SessionTitle.Service const fiber = yield* title.generate(sessionID).pipe(Effect.forkScoped) yield* Deferred.await(started) const events = yield* Bus.Service yield* events.publish(SessionEvent.Renamed, { sessionID, title: "Manual title" }) yield* Deferred.succeed(release, undefined) yield* Fiber.join(fiber)
const store = yield* SessionStore.Service expect(requests).toHaveLength(1) expect((yield* store.get(sessionID))?.title).toBe("Manual title") }),)