refactor: unwrap session/ tier-2 namespaces + self-reexport (#22973)

This commit is contained in:
Kit Langton
2026-04-17 00:49:39 +00:00
committed by GitHub
parent d6af5a686c
commit 51d8219c46
14 changed files with 4909 additions and 4909 deletions
+2 -2
View File
@@ -17,7 +17,6 @@ import { Effect, Layer, Context } from "effect"
import { InstanceState } from "@/effect" import { InstanceState } from "@/effect"
import { isOverflow as overflow } from "./overflow" import { isOverflow as overflow } from "./overflow"
export namespace SessionCompaction {
const log = Log.create({ service: "session.compaction" }) const log = Log.create({ service: "session.compaction" })
export const Event = { export const Event = {
@@ -409,4 +408,5 @@ When constructing the summary, try to stick to this template:
Layer.provide(Config.defaultLayer), Layer.provide(Config.defaultLayer),
), ),
) )
}
export * as SessionCompaction from "./compaction"
+2 -2
View File
@@ -50,7 +50,6 @@ function extract(messages: MessageV2.WithParts[]) {
return paths return paths
} }
export namespace Instruction {
export interface Interface { export interface Interface {
readonly clear: (messageID: MessageID) => Effect.Effect<void> readonly clear: (messageID: MessageID) => Effect.Effect<void>
readonly systemPaths: () => Effect.Effect<Set<string>, AppFileSystem.Error> readonly systemPaths: () => Effect.Effect<Set<string>, AppFileSystem.Error>
@@ -240,4 +239,5 @@ export namespace Instruction {
export function loaded(messages: MessageV2.WithParts[]) { export function loaded(messages: MessageV2.WithParts[]) {
return extract(messages) return extract(messages)
} }
}
export * as Instruction from "./instruction"
+2 -2
View File
@@ -25,7 +25,6 @@ import { EffectBridge } from "@/effect"
import * as Option from "effect/Option" import * as Option from "effect/Option"
import * as OtelTracer from "@effect/opentelemetry/Tracer" import * as OtelTracer from "@effect/opentelemetry/Tracer"
export namespace LLM {
const log = Log.create({ service: "llm" }) const log = Log.create({ service: "llm" })
export const OUTPUT_TOKEN_MAX = ProviderTransform.OUTPUT_TOKEN_MAX export const OUTPUT_TOKEN_MAX = ProviderTransform.OUTPUT_TOKEN_MAX
type Result = Awaited<ReturnType<typeof streamText>> type Result = Awaited<ReturnType<typeof streamText>>
@@ -450,4 +449,5 @@ export namespace LLM {
} }
return false return false
} }
}
export * as LLM from "./llm"
+20 -20
View File
@@ -24,7 +24,6 @@ interface FetchDecompressionError extends Error {
path: string path: string
} }
export namespace MessageV2 {
export const SYNTHETIC_ATTACHMENT_PROMPT = "Attached image(s) from tool result:" export const SYNTHETIC_ATTACHMENT_PROMPT = "Attached image(s) from tool result:"
export function isMedia(mime: string) { export function isMedia(mime: string) {
@@ -534,7 +533,7 @@ export namespace MessageV2 {
...row.data, ...row.data,
id: row.id, id: row.id,
sessionID: row.session_id, sessionID: row.session_id,
}) as MessageV2.Info }) as Info
const part = (row: typeof PartTable.$inferSelect) => const part = (row: typeof PartTable.$inferSelect) =>
({ ({
@@ -542,7 +541,7 @@ export namespace MessageV2 {
id: row.id, id: row.id,
sessionID: row.session_id, sessionID: row.session_id,
messageID: row.message_id, messageID: row.message_id,
}) as MessageV2.Part }) as Part
const older = (row: Cursor) => const older = (row: Cursor) =>
or( or(
@@ -552,7 +551,7 @@ export namespace MessageV2 {
function hydrate(rows: (typeof MessageTable.$inferSelect)[]) { function hydrate(rows: (typeof MessageTable.$inferSelect)[]) {
const ids = rows.map((row) => row.id) const ids = rows.map((row) => row.id)
const partByMessage = new Map<string, MessageV2.Part[]>() const partByMessage = new Map<string, Part[]>()
if (ids.length > 0) { if (ids.length > 0) {
const partRows = Database.use((db) => const partRows = Database.use((db) =>
db db
@@ -699,7 +698,7 @@ export namespace MessageV2 {
if ( if (
msg.info.error && msg.info.error &&
!( !(
MessageV2.AbortedError.isInstance(msg.info.error) && AbortedError.isInstance(msg.info.error) &&
msg.parts.some((part) => part.type !== "step-start" && part.type !== "reasoning") msg.parts.some((part) => part.type !== "step-start" && part.type !== "reasoning")
) )
) { ) {
@@ -865,7 +864,7 @@ export namespace MessageV2 {
) )
if (!row) throw new NotFoundError({ message: `Session not found: ${input.sessionID}` }) if (!row) throw new NotFoundError({ message: `Session not found: ${input.sessionID}` })
return { return {
items: [] as MessageV2.WithParts[], items: [] as WithParts[],
more: false, more: false,
} }
} }
@@ -907,7 +906,7 @@ export namespace MessageV2 {
id: row.id, id: row.id,
sessionID: row.session_id, sessionID: row.session_id,
messageID: row.message_id, messageID: row.message_id,
}) as MessageV2.Part, }) as Part,
) )
} }
@@ -926,8 +925,8 @@ export namespace MessageV2 {
} }
} }
export function filterCompacted(msgs: Iterable<MessageV2.WithParts>) { export function filterCompacted(msgs: Iterable<WithParts>) {
const result = [] as MessageV2.WithParts[] const result = [] as WithParts[]
const completed = new Set<string>() const completed = new Set<string>()
for (const msg of msgs) { for (const msg of msgs) {
result.push(msg) result.push(msg)
@@ -954,16 +953,16 @@ export namespace MessageV2 {
): NonNullable<Assistant["error"]> { ): NonNullable<Assistant["error"]> {
switch (true) { switch (true) {
case e instanceof DOMException && e.name === "AbortError": case e instanceof DOMException && e.name === "AbortError":
return new MessageV2.AbortedError( return new AbortedError(
{ message: e.message }, { message: e.message },
{ {
cause: e, cause: e,
}, },
).toObject() ).toObject()
case MessageV2.OutputLengthError.isInstance(e): case OutputLengthError.isInstance(e):
return e return e
case LoadAPIKeyError.isInstance(e): case LoadAPIKeyError.isInstance(e):
return new MessageV2.AuthError( return new AuthError(
{ {
providerID: ctx.providerID, providerID: ctx.providerID,
message: e.message, message: e.message,
@@ -971,7 +970,7 @@ export namespace MessageV2 {
{ cause: e }, { cause: e },
).toObject() ).toObject()
case (e as SystemError)?.code === "ECONNRESET": case (e as SystemError)?.code === "ECONNRESET":
return new MessageV2.APIError( return new APIError(
{ {
message: "Connection reset by server", message: "Connection reset by server",
isRetryable: true, isRetryable: true,
@@ -985,9 +984,9 @@ export namespace MessageV2 {
).toObject() ).toObject()
case e instanceof Error && (e as FetchDecompressionError).code === "ZlibError": case e instanceof Error && (e as FetchDecompressionError).code === "ZlibError":
if (ctx.aborted) { if (ctx.aborted) {
return new MessageV2.AbortedError({ message: e.message }, { cause: e }).toObject() return new AbortedError({ message: e.message }, { cause: e }).toObject()
} }
return new MessageV2.APIError( return new APIError(
{ {
message: "Response decompression failed", message: "Response decompression failed",
isRetryable: true, isRetryable: true,
@@ -1004,7 +1003,7 @@ export namespace MessageV2 {
error: e, error: e,
}) })
if (parsed.type === "context_overflow") { if (parsed.type === "context_overflow") {
return new MessageV2.ContextOverflowError( return new ContextOverflowError(
{ {
message: parsed.message, message: parsed.message,
responseBody: parsed.responseBody, responseBody: parsed.responseBody,
@@ -1013,7 +1012,7 @@ export namespace MessageV2 {
).toObject() ).toObject()
} }
return new MessageV2.APIError( return new APIError(
{ {
message: parsed.message, message: parsed.message,
statusCode: parsed.statusCode, statusCode: parsed.statusCode,
@@ -1031,7 +1030,7 @@ export namespace MessageV2 {
const parsed = ProviderError.parseStreamError(e) const parsed = ProviderError.parseStreamError(e)
if (parsed) { if (parsed) {
if (parsed.type === "context_overflow") { if (parsed.type === "context_overflow") {
return new MessageV2.ContextOverflowError( return new ContextOverflowError(
{ {
message: parsed.message, message: parsed.message,
responseBody: parsed.responseBody, responseBody: parsed.responseBody,
@@ -1039,7 +1038,7 @@ export namespace MessageV2 {
{ cause: e }, { cause: e },
).toObject() ).toObject()
} }
return new MessageV2.APIError( return new APIError(
{ {
message: parsed.message, message: parsed.message,
isRetryable: parsed.isRetryable, isRetryable: parsed.isRetryable,
@@ -1054,4 +1053,5 @@ export namespace MessageV2 {
return new NamedError.Unknown({ message: JSON.stringify(e) }, { cause: e }).toObject() return new NamedError.Unknown({ message: JSON.stringify(e) }, { cause: e }).toObject()
} }
} }
}
export * as MessageV2 from "./message-v2"
+2 -2
View File
@@ -3,7 +3,6 @@ import { SessionID } from "./schema"
import { ModelID, ProviderID } from "../provider/schema" import { ModelID, ProviderID } from "../provider/schema"
import { NamedError } from "@opencode-ai/shared/util/error" import { NamedError } from "@opencode-ai/shared/util/error"
export namespace Message {
export const OutputLengthError = NamedError.create("MessageOutputLengthError", z.object({})) export const OutputLengthError = NamedError.create("MessageOutputLengthError", z.object({}))
export const AuthError = NamedError.create( export const AuthError = NamedError.create(
"ProviderAuthError", "ProviderAuthError",
@@ -188,4 +187,5 @@ export namespace Message {
ref: "Message", ref: "Message",
}) })
export type Info = z.infer<typeof Info> export type Info = z.infer<typeof Info>
}
export * as Message from "./message"
+2 -2
View File
@@ -21,7 +21,6 @@ import { errorMessage } from "@/util/error"
import { Log } from "@/util" import { Log } from "@/util"
import { isRecord } from "@/util/record" import { isRecord } from "@/util/record"
export namespace SessionProcessor {
const DOOM_LOOP_THRESHOLD = 3 const DOOM_LOOP_THRESHOLD = 3
const log = Log.create({ service: "session.processor" }) const log = Log.create({ service: "session.processor" })
@@ -616,4 +615,5 @@ export namespace SessionProcessor {
Layer.provide(Config.defaultLayer), Layer.provide(Config.defaultLayer),
), ),
) )
}
export * as SessionProcessor from "./processor"
+2 -2
View File
@@ -64,7 +64,6 @@ IMPORTANT:
const STRUCTURED_OUTPUT_SYSTEM_PROMPT = `IMPORTANT: The user has requested structured output. You MUST use the StructuredOutput tool to provide your final response. Do NOT respond with plain text - you MUST call the StructuredOutput tool with your answer formatted according to the schema.` const STRUCTURED_OUTPUT_SYSTEM_PROMPT = `IMPORTANT: The user has requested structured output. You MUST use the StructuredOutput tool to provide your final response. Do NOT respond with plain text - you MUST call the StructuredOutput tool with your answer formatted according to the schema.`
export namespace SessionPrompt {
const log = Log.create({ service: "session.prompt" }) const log = Log.create({ service: "session.prompt" })
const elog = EffectLogger.create({ service: "session.prompt" }) const elog = EffectLogger.create({ service: "session.prompt" })
@@ -1851,4 +1850,5 @@ NOTE: At any point in time through this workflow you should feel free to ask the
const argsRegex = /(?:\[Image\s+\d+\]|"[^"]*"|'[^']*'|[^\s"']+)/gi const argsRegex = /(?:\[Image\s+\d+\]|"[^"]*"|'[^']*'|[^\s"']+)/gi
const placeholderRegex = /\$(\d+)/g const placeholderRegex = /\$(\d+)/g
const quoteTrimRegex = /^["']|["']$/g const quoteTrimRegex = /^["']|["']$/g
}
export * as SessionPrompt from "./prompt"
+2 -2
View File
@@ -3,7 +3,6 @@ import { Cause, Clock, Duration, Effect, Schedule } from "effect"
import { MessageV2 } from "./message-v2" import { MessageV2 } from "./message-v2"
import { iife } from "@/util/iife" import { iife } from "@/util/iife"
export namespace SessionRetry {
export type Err = ReturnType<NamedError["toObject"]> export type Err = ReturnType<NamedError["toObject"]>
// This exported message is shared with the TUI upsell detector. Matching on a // This exported message is shared with the TUI upsell detector. Matching on a
@@ -122,4 +121,5 @@ export namespace SessionRetry {
}), }),
) )
} }
}
export * as SessionRetry from "./retry"
+2 -2
View File
@@ -11,7 +11,6 @@ import { SessionID, MessageID, PartID } from "./schema"
import { SessionRunState } from "./run-state" import { SessionRunState } from "./run-state"
import { SessionSummary } from "./summary" import { SessionSummary } from "./summary"
export namespace SessionRevert {
const log = Log.create({ service: "session.revert" }) const log = Log.create({ service: "session.revert" })
export const RevertInput = z.object({ export const RevertInput = z.object({
@@ -158,4 +157,5 @@ export namespace SessionRevert {
Layer.provide(SessionSummary.defaultLayer), Layer.provide(SessionSummary.defaultLayer),
), ),
) )
}
export * as SessionRevert from "./revert"
+2 -2
View File
@@ -6,7 +6,6 @@ import { MessageV2 } from "./message-v2"
import { SessionID } from "./schema" import { SessionID } from "./schema"
import { SessionStatus } from "./status" import { SessionStatus } from "./status"
export namespace SessionRunState {
export interface Interface { export interface Interface {
readonly assertNotBusy: (sessionID: SessionID) => Effect.Effect<void> readonly assertNotBusy: (sessionID: SessionID) => Effect.Effect<void>
readonly cancel: (sessionID: SessionID) => Effect.Effect<void> readonly cancel: (sessionID: SessionID) => Effect.Effect<void>
@@ -105,4 +104,5 @@ export namespace SessionRunState {
) )
export const defaultLayer = layer.pipe(Layer.provide(SessionStatus.defaultLayer)) export const defaultLayer = layer.pipe(Layer.provide(SessionStatus.defaultLayer))
}
export * as SessionRunState from "./run-state"
+2 -2
View File
@@ -5,7 +5,6 @@ import { SessionID } from "./schema"
import { Effect, Layer, Context } from "effect" import { Effect, Layer, Context } from "effect"
import z from "zod" import z from "zod"
export namespace SessionStatus {
export const Info = z export const Info = z
.union([ .union([
z.object({ z.object({
@@ -85,4 +84,5 @@ export namespace SessionStatus {
) )
export const defaultLayer = layer.pipe(Layer.provide(Bus.layer)) export const defaultLayer = layer.pipe(Layer.provide(Bus.layer))
}
export * as SessionStatus from "./status"
+2 -2
View File
@@ -7,7 +7,6 @@ import * as Session from "./session"
import { MessageV2 } from "./message-v2" import { MessageV2 } from "./message-v2"
import { SessionID, MessageID } from "./schema" import { SessionID, MessageID } from "./schema"
export namespace SessionSummary {
function unquoteGitPath(input: string) { function unquoteGitPath(input: string) {
if (!input.startsWith('"')) return input if (!input.startsWith('"')) return input
if (!input.endsWith('"')) return input if (!input.endsWith('"')) return input
@@ -162,4 +161,5 @@ export namespace SessionSummary {
sessionID: SessionID.zod, sessionID: SessionID.zod,
messageID: MessageID.zod.optional(), messageID: MessageID.zod.optional(),
}) })
}
export * as SessionSummary from "./summary"
+2 -2
View File
@@ -16,7 +16,6 @@ import type { Agent } from "@/agent/agent"
import { Permission } from "@/permission" import { Permission } from "@/permission"
import { Skill } from "@/skill" import { Skill } from "@/skill"
export namespace SystemPrompt {
export function provider(model: Provider.Model) { export function provider(model: Provider.Model) {
if (model.api.id.includes("gpt-4") || model.api.id.includes("o1") || model.api.id.includes("o3")) if (model.api.id.includes("gpt-4") || model.api.id.includes("o1") || model.api.id.includes("o3"))
return [PROMPT_BEAST] return [PROMPT_BEAST]
@@ -81,4 +80,5 @@ export namespace SystemPrompt {
) )
export const defaultLayer = layer.pipe(Layer.provide(Skill.defaultLayer)) export const defaultLayer = layer.pipe(Layer.provide(Skill.defaultLayer))
}
export * as SystemPrompt from "./system"
+2 -2
View File
@@ -6,7 +6,6 @@ import z from "zod"
import { Database, eq, asc } from "../storage" import { Database, eq, asc } from "../storage"
import { TodoTable } from "./session.sql" import { TodoTable } from "./session.sql"
export namespace Todo {
export const Info = z export const Info = z
.object({ .object({
content: z.string().describe("Brief description of the task"), content: z.string().describe("Brief description of the task"),
@@ -82,4 +81,5 @@ export namespace Todo {
) )
export const defaultLayer = layer.pipe(Layer.provide(Bus.layer)) export const defaultLayer = layer.pipe(Layer.provide(Bus.layer))
}
export * as Todo from "./todo"