lmin/zmo-cli
1
0
Fork 0
Code Issues Pull Requests Actions 83 Packages Projects Releases Wiki Activity

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

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
packages/opencode/src/session/compaction.ts
+16 -16
View File
@@ -17,23 +17,22 @@ import { Effect, Layer, Context } from "effect"
import { InstanceState } from "@/effect"
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 = {
Compacted: BusEvent.define(
"session.compacted",
z.object({
sessionID: SessionID.zod,
}),
),
}
}
export const PRUNE_MINIMUM = 20_000
export const PRUNE_PROTECT = 40_000
const PRUNE_PROTECTED_TOOLS = ["skill"]
export const PRUNE_MINIMUM = 20_000
export const PRUNE_PROTECT = 40_000
const PRUNE_PROTECTED_TOOLS = ["skill"]
export interface Interface {
export interface Interface {
readonly isOverflow: (input: {
tokens: MessageV2.Assistant["tokens"]
model: Provider.Model
@@ -53,11 +52,11 @@ export namespace SessionCompaction {
auto: boolean
overflow?: boolean
}) => Effect.Effect<void>
}
}
export class Service extends Context.Service<Service, Interface>()("@opencode/SessionCompaction") {}
export class Service extends Context.Service<Service, Interface>()("@opencode/SessionCompaction") {}
export const layer: Layer.Layer<
export const layer: Layer.Layer<
Service,
never,
| Bus.Service
@@ -67,7 +66,7 @@ export namespace SessionCompaction {
| Plugin.Service
| SessionProcessor.Service
| Provider.Service
> = Layer.effect(
> = Layer.effect(
Service,
Effect.gen(function* () {
const bus = yield* Bus.Service
@@ -396,9 +395,9 @@ When constructing the summary, try to stick to this template:
create,
})
}),
)
)
export const defaultLayer = Layer.suspend(() =>
export const defaultLayer = Layer.suspend(() =>
layer.pipe(
Layer.provide(Provider.defaultLayer),
Layer.provide(Session.defaultLayer),
@@ -408,5 +407,6 @@ When constructing the summary, try to stick to this template:
Layer.provide(Bus.layer),
Layer.provide(Config.defaultLayer),
),
)
}
)
export * as SessionCompaction from "./compaction"
packages/opencode/src/session/instruction.ts
+9 -9
View File
@@ -50,8 +50,7 @@ function extract(messages: MessageV2.WithParts[]) {
return paths
}
export namespace Instruction {
export interface Interface {
export interface Interface {
readonly clear: (messageID: MessageID) => Effect.Effect<void>
readonly systemPaths: () => Effect.Effect<Set<string>, AppFileSystem.Error>
readonly system: () => Effect.Effect<string[], AppFileSystem.Error>
@@ -61,11 +60,11 @@ export namespace Instruction {
filepath: string,
messageID: MessageID,
) => Effect.Effect<{ filepath: string; content: string }[], AppFileSystem.Error>
}
}
export class Service extends Context.Service<Service, Interface>()("@opencode/Instruction") {}
export class Service extends Context.Service<Service, Interface>()("@opencode/Instruction") {}
export const layer: Layer.Layer<Service, never, AppFileSystem.Service | Config.Service | HttpClient.HttpClient> =
export const layer: Layer.Layer<Service, never, AppFileSystem.Service | Config.Service | HttpClient.HttpClient> =
Layer.effect(
Service,
Effect.gen(function* () {
@@ -231,13 +230,14 @@ export namespace Instruction {
}),
)
export const defaultLayer = layer.pipe(
export const defaultLayer = layer.pipe(
Layer.provide(Config.defaultLayer),
Layer.provide(AppFileSystem.defaultLayer),
Layer.provide(FetchHttpClient.layer),
)
)
export function loaded(messages: MessageV2.WithParts[]) {
export function loaded(messages: MessageV2.WithParts[]) {
return extract(messages)
}
}
export * as Instruction from "./instruction"
packages/opencode/src/session/llm.ts
+24 -24
View File
@@ -25,12 +25,11 @@ import { EffectBridge } from "@/effect"
import * as Option from "effect/Option"
import * as OtelTracer from "@effect/opentelemetry/Tracer"
export namespace LLM {
const log = Log.create({ service: "llm" })
export const OUTPUT_TOKEN_MAX = ProviderTransform.OUTPUT_TOKEN_MAX
type Result = Awaited<ReturnType<typeof streamText>>
const log = Log.create({ service: "llm" })
export const OUTPUT_TOKEN_MAX = ProviderTransform.OUTPUT_TOKEN_MAX
type Result = Awaited<ReturnType<typeof streamText>>
export type StreamInput = {
export type StreamInput = {
user: MessageV2.User
sessionID: string
parentSessionID?: string
@@ -43,25 +42,25 @@ export namespace LLM {
tools: Record<string, Tool>
retries?: number
toolChoice?: "auto" | "required" | "none"
}
}
export type StreamRequest = StreamInput & {
export type StreamRequest = StreamInput & {
abort: AbortSignal
}
}
export type Event = Result["fullStream"] extends AsyncIterable<infer T> ? T : never
export type Event = Result["fullStream"] extends AsyncIterable<infer T> ? T : never
export interface Interface {
export interface Interface {
readonly stream: (input: StreamInput) => Stream.Stream<Event, unknown>
}
}
export class Service extends Context.Service<Service, Interface>()("@opencode/LLM") {}
export class Service extends Context.Service<Service, Interface>()("@opencode/LLM") {}
const live: Layer.Layer<
const live: Layer.Layer<
Service,
never,
Auth.Service | Config.Service | Provider.Service | Plugin.Service | Permission.Service
> = Layer.effect(
> = Layer.effect(
Service,
Effect.gen(function* () {
const auth = yield* Auth.Service
@@ -418,30 +417,30 @@ export namespace LLM {
return Service.of({ stream })
}),
)
)
export const layer = live.pipe(Layer.provide(Permission.defaultLayer))
export const layer = live.pipe(Layer.provide(Permission.defaultLayer))
export const defaultLayer = Layer.suspend(() =>
export const defaultLayer = Layer.suspend(() =>
layer.pipe(
Layer.provide(Auth.defaultLayer),
Layer.provide(Config.defaultLayer),
Layer.provide(Provider.defaultLayer),
Layer.provide(Plugin.defaultLayer),
),
)
)
function resolveTools(input: Pick<StreamInput, "tools" | "agent" | "permission" | "user">) {
function resolveTools(input: Pick<StreamInput, "tools" | "agent" | "permission" | "user">) {
const disabled = Permission.disabled(
Object.keys(input.tools),
Permission.merge(input.agent.permission, input.permission ?? []),
)
return Record.filter(input.tools, (_, k) => input.user.tools?.[k] !== false && !disabled.has(k))
}
}
// Check if messages contain any tool-call content
// Used to determine if a dummy tool should be added for LiteLLM proxy compatibility
export function hasToolCalls(messages: ModelMessage[]): boolean {
// Check if messages contain any tool-call content
// Used to determine if a dummy tool should be added for LiteLLM proxy compatibility
export function hasToolCalls(messages: ModelMessage[]): boolean {
for (const msg of messages) {
if (!Array.isArray(msg.content)) continue
for (const part of msg.content) {
@@ -449,5 +448,6 @@ export namespace LLM {
}
}
return false
}
}
export * as LLM from "./llm"
packages/opencode/src/session/message-v2.ts
+162 -162
View File
@@ -24,30 +24,29 @@ interface FetchDecompressionError extends Error {
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) {
return mime.startsWith("image/") || mime === "application/pdf"
}
}
export const OutputLengthError = NamedError.create("MessageOutputLengthError", z.object({}))
export const AbortedError = NamedError.create("MessageAbortedError", z.object({ message: z.string() }))
export const StructuredOutputError = NamedError.create(
export const OutputLengthError = NamedError.create("MessageOutputLengthError", z.object({}))
export const AbortedError = NamedError.create("MessageAbortedError", z.object({ message: z.string() }))
export const StructuredOutputError = NamedError.create(
"StructuredOutputError",
z.object({
message: z.string(),
retries: z.number(),
}),
)
export const AuthError = NamedError.create(
)
export const AuthError = NamedError.create(
"ProviderAuthError",
z.object({
providerID: z.string(),
message: z.string(),
}),
)
export const APIError = NamedError.create(
)
export const APIError = NamedError.create(
"APIError",
z.object({
message: z.string(),
@@ -57,14 +56,14 @@ export namespace MessageV2 {
responseBody: z.string().optional(),
metadata: z.record(z.string(), z.string()).optional(),
}),
)
export type APIError = z.infer<typeof APIError.Schema>
export const ContextOverflowError = NamedError.create(
)
export type APIError = z.infer<typeof APIError.Schema>
export const ContextOverflowError = NamedError.create(
"ContextOverflowError",
z.object({ message: z.string(), responseBody: z.string().optional() }),
)
)
export const OutputFormatText = z
export const OutputFormatText = z
.object({
type: z.literal("text"),
})
@@ -72,7 +71,7 @@ export namespace MessageV2 {
ref: "OutputFormatText",
})
export const OutputFormatJsonSchema = z
export const OutputFormatJsonSchema = z
.object({
type: z.literal("json_schema"),
schema: z.record(z.string(), z.any()).meta({ ref: "JSONSchema" }),
@@ -82,35 +81,35 @@ export namespace MessageV2 {
ref: "OutputFormatJsonSchema",
})
export const Format = z.discriminatedUnion("type", [OutputFormatText, OutputFormatJsonSchema]).meta({
export const Format = z.discriminatedUnion("type", [OutputFormatText, OutputFormatJsonSchema]).meta({
ref: "OutputFormat",
})
export type OutputFormat = z.infer<typeof Format>
})
export type OutputFormat = z.infer<typeof Format>
const PartBase = z.object({
const PartBase = z.object({
id: PartID.zod,
sessionID: SessionID.zod,
messageID: MessageID.zod,
})
})
export const SnapshotPart = PartBase.extend({
export const SnapshotPart = PartBase.extend({
type: z.literal("snapshot"),
snapshot: z.string(),
}).meta({
}).meta({
ref: "SnapshotPart",
})
export type SnapshotPart = z.infer<typeof SnapshotPart>
})
export type SnapshotPart = z.infer<typeof SnapshotPart>
export const PatchPart = PartBase.extend({
export const PatchPart = PartBase.extend({
type: z.literal("patch"),
hash: z.string(),
files: z.string().array(),
}).meta({
}).meta({
ref: "PatchPart",
})
export type PatchPart = z.infer<typeof PatchPart>
})
export type PatchPart = z.infer<typeof PatchPart>
export const TextPart = PartBase.extend({
export const TextPart = PartBase.extend({
type: z.literal("text"),
text: z.string(),
synthetic: z.boolean().optional(),
@@ -122,12 +121,12 @@ export namespace MessageV2 {
})
.optional(),
metadata: z.record(z.string(), z.any()).optional(),
}).meta({
}).meta({
ref: "TextPart",
})
export type TextPart = z.infer<typeof TextPart>
})
export type TextPart = z.infer<typeof TextPart>
export const ReasoningPart = PartBase.extend({
export const ReasoningPart = PartBase.extend({
type: z.literal("reasoning"),
text: z.string(),
metadata: z.record(z.string(), z.any()).optional(),
@@ -135,12 +134,12 @@ export namespace MessageV2 {
start: z.number(),
end: z.number().optional(),
}),
}).meta({
}).meta({
ref: "ReasoningPart",
})
export type ReasoningPart = z.infer<typeof ReasoningPart>
})
export type ReasoningPart = z.infer<typeof ReasoningPart>
const FilePartSourceBase = z.object({
const FilePartSourceBase = z.object({
text: z
.object({
value: z.string(),
@@ -150,49 +149,49 @@ export namespace MessageV2 {
.meta({
ref: "FilePartSourceText",
}),
})
})
export const FileSource = FilePartSourceBase.extend({
export const FileSource = FilePartSourceBase.extend({
type: z.literal("file"),
path: z.string(),
}).meta({
}).meta({
ref: "FileSource",
})
})
export const SymbolSource = FilePartSourceBase.extend({
export const SymbolSource = FilePartSourceBase.extend({
type: z.literal("symbol"),
path: z.string(),
range: LSP.Range,
name: z.string(),
kind: z.number().int(),
}).meta({
}).meta({
ref: "SymbolSource",
})
})
export const ResourceSource = FilePartSourceBase.extend({
export const ResourceSource = FilePartSourceBase.extend({
type: z.literal("resource"),
clientName: z.string(),
uri: z.string(),
}).meta({
}).meta({
ref: "ResourceSource",
})
})
export const FilePartSource = z.discriminatedUnion("type", [FileSource, SymbolSource, ResourceSource]).meta({
export const FilePartSource = z.discriminatedUnion("type", [FileSource, SymbolSource, ResourceSource]).meta({
ref: "FilePartSource",
})
})
export const FilePart = PartBase.extend({
export const FilePart = PartBase.extend({
type: z.literal("file"),
mime: z.string(),
filename: z.string().optional(),
url: z.string(),
source: FilePartSource.optional(),
}).meta({
}).meta({
ref: "FilePart",
})
export type FilePart = z.infer<typeof FilePart>
})
export type FilePart = z.infer<typeof FilePart>
export const AgentPart = PartBase.extend({
export const AgentPart = PartBase.extend({
type: z.literal("agent"),
name: z.string(),
source: z
@@ -202,21 +201,21 @@ export namespace MessageV2 {
end: z.number().int(),
})
.optional(),
}).meta({
}).meta({
ref: "AgentPart",
})
export type AgentPart = z.infer<typeof AgentPart>
})
export type AgentPart = z.infer<typeof AgentPart>
export const CompactionPart = PartBase.extend({
export const CompactionPart = PartBase.extend({
type: z.literal("compaction"),
auto: z.boolean(),
overflow: z.boolean().optional(),
}).meta({
}).meta({
ref: "CompactionPart",
})
export type CompactionPart = z.infer<typeof CompactionPart>
})
export type CompactionPart = z.infer<typeof CompactionPart>
export const SubtaskPart = PartBase.extend({
export const SubtaskPart = PartBase.extend({
type: z.literal("subtask"),
prompt: z.string(),
description: z.string(),
@@ -228,32 +227,32 @@ export namespace MessageV2 {
})
.optional(),
command: z.string().optional(),
}).meta({
}).meta({
ref: "SubtaskPart",
})
export type SubtaskPart = z.infer<typeof SubtaskPart>
})
export type SubtaskPart = z.infer<typeof SubtaskPart>
export const RetryPart = PartBase.extend({
export const RetryPart = PartBase.extend({
type: z.literal("retry"),
attempt: z.number(),
error: APIError.Schema,
time: z.object({
created: z.number(),
}),
}).meta({
}).meta({
ref: "RetryPart",
})
export type RetryPart = z.infer<typeof RetryPart>
})
export type RetryPart = z.infer<typeof RetryPart>
export const StepStartPart = PartBase.extend({
export const StepStartPart = PartBase.extend({
type: z.literal("step-start"),
snapshot: z.string().optional(),
}).meta({
}).meta({
ref: "StepStartPart",
})
export type StepStartPart = z.infer<typeof StepStartPart>
})
export type StepStartPart = z.infer<typeof StepStartPart>
export const StepFinishPart = PartBase.extend({
export const StepFinishPart = PartBase.extend({
type: z.literal("step-finish"),
reason: z.string(),
snapshot: z.string().optional(),
@@ -268,12 +267,12 @@ export namespace MessageV2 {
write: z.number(),
}),
}),
}).meta({
}).meta({
ref: "StepFinishPart",
})
export type StepFinishPart = z.infer<typeof StepFinishPart>
})
export type StepFinishPart = z.infer<typeof StepFinishPart>
export const ToolStatePending = z
export const ToolStatePending = z
.object({
status: z.literal("pending"),
input: z.record(z.string(), z.any()),
@@ -283,9 +282,9 @@ export namespace MessageV2 {
ref: "ToolStatePending",
})
export type ToolStatePending = z.infer<typeof ToolStatePending>
export type ToolStatePending = z.infer<typeof ToolStatePending>
export const ToolStateRunning = z
export const ToolStateRunning = z
.object({
status: z.literal("running"),
input: z.record(z.string(), z.any()),
@@ -298,9 +297,9 @@ export namespace MessageV2 {
.meta({
ref: "ToolStateRunning",
})
export type ToolStateRunning = z.infer<typeof ToolStateRunning>
export type ToolStateRunning = z.infer<typeof ToolStateRunning>
export const ToolStateCompleted = z
export const ToolStateCompleted = z
.object({
status: z.literal("completed"),
input: z.record(z.string(), z.any()),
@@ -317,9 +316,9 @@ export namespace MessageV2 {
.meta({
ref: "ToolStateCompleted",
})
export type ToolStateCompleted = z.infer<typeof ToolStateCompleted>
export type ToolStateCompleted = z.infer<typeof ToolStateCompleted>
export const ToolStateError = z
export const ToolStateError = z
.object({
status: z.literal("error"),
input: z.record(z.string(), z.any()),
@@ -333,31 +332,31 @@ export namespace MessageV2 {
.meta({
ref: "ToolStateError",
})
export type ToolStateError = z.infer<typeof ToolStateError>
export type ToolStateError = z.infer<typeof ToolStateError>
export const ToolState = z
export const ToolState = z
.discriminatedUnion("status", [ToolStatePending, ToolStateRunning, ToolStateCompleted, ToolStateError])
.meta({
ref: "ToolState",
})
export const ToolPart = PartBase.extend({
export const ToolPart = PartBase.extend({
type: z.literal("tool"),
callID: z.string(),
tool: z.string(),
state: ToolState,
metadata: z.record(z.string(), z.any()).optional(),
}).meta({
}).meta({
ref: "ToolPart",
})
export type ToolPart = z.infer<typeof ToolPart>
})
export type ToolPart = z.infer<typeof ToolPart>
const Base = z.object({
const Base = z.object({
id: MessageID.zod,
sessionID: SessionID.zod,
})
})
export const User = Base.extend({
export const User = Base.extend({
role: z.literal("user"),
time: z.object({
created: z.number(),
@@ -378,12 +377,12 @@ export namespace MessageV2 {
}),
system: z.string().optional(),
tools: z.record(z.string(), z.boolean()).optional(),
}).meta({
}).meta({
ref: "UserMessage",
})
export type User = z.infer<typeof User>
})
export type User = z.infer<typeof User>
export const Part = z
export const Part = z
.discriminatedUnion("type", [
TextPart,
SubtaskPart,
@@ -401,9 +400,9 @@ export namespace MessageV2 {
.meta({
ref: "Part",
})
export type Part = z.infer<typeof Part>
export type Part = z.infer<typeof Part>
export const Assistant = Base.extend({
export const Assistant = Base.extend({
role: z.literal("assistant"),
time: z.object({
created: z.number(),
@@ -447,17 +446,17 @@ export namespace MessageV2 {
structured: z.any().optional(),
variant: z.string().optional(),
finish: z.string().optional(),
}).meta({
}).meta({
ref: "AssistantMessage",
})
export type Assistant = z.infer<typeof Assistant>
})
export type Assistant = z.infer<typeof Assistant>
export const Info = z.discriminatedUnion("role", [User, Assistant]).meta({
export const Info = z.discriminatedUnion("role", [User, Assistant]).meta({
ref: "Message",
})
export type Info = z.infer<typeof Info>
})
export type Info = z.infer<typeof Info>
export const Event = {
export const Event = {
Updated: SyncEvent.define({
type: "message.updated",
version: 1,
@@ -506,53 +505,53 @@ export namespace MessageV2 {
partID: PartID.zod,
}),
}),
}
}
export const WithParts = z.object({
export const WithParts = z.object({
info: Info,
parts: z.array(Part),
})
export type WithParts = z.infer<typeof WithParts>
})
export type WithParts = z.infer<typeof WithParts>
const Cursor = z.object({
const Cursor = z.object({
id: MessageID.zod,
time: z.number(),
})
type Cursor = z.infer<typeof Cursor>
})
type Cursor = z.infer<typeof Cursor>
export const cursor = {
export const cursor = {
encode(input: Cursor) {
return Buffer.from(JSON.stringify(input)).toString("base64url")
},
decode(input: string) {
return Cursor.parse(JSON.parse(Buffer.from(input, "base64url").toString("utf8")))
},
}
}
const info = (row: typeof MessageTable.$inferSelect) =>
const info = (row: typeof MessageTable.$inferSelect) =>
({
...row.data,
id: row.id,
sessionID: row.session_id,
}) as MessageV2.Info
}) as Info
const part = (row: typeof PartTable.$inferSelect) =>
const part = (row: typeof PartTable.$inferSelect) =>
({
...row.data,
id: row.id,
sessionID: row.session_id,
messageID: row.message_id,
}) as MessageV2.Part
}) as Part
const older = (row: Cursor) =>
const older = (row: Cursor) =>
or(
lt(MessageTable.time_created, row.time),
and(eq(MessageTable.time_created, row.time), lt(MessageTable.id, row.id)),
)
function hydrate(rows: (typeof MessageTable.$inferSelect)[]) {
function hydrate(rows: (typeof MessageTable.$inferSelect)[]) {
const ids = rows.map((row) => row.id)
const partByMessage = new Map<string, MessageV2.Part[]>()
const partByMessage = new Map<string, Part[]>()
if (ids.length > 0) {
const partRows = Database.use((db) =>
db
@@ -574,19 +573,19 @@ export namespace MessageV2 {
info: info(row),
parts: partByMessage.get(row.id) ?? [],
}))
}
}
function providerMeta(metadata: Record<string, any> | undefined) {
function providerMeta(metadata: Record<string, any> | undefined) {
if (!metadata) return undefined
const { providerExecuted: _, ...rest } = metadata
return Object.keys(rest).length > 0 ? rest : undefined
}
}
export const toModelMessagesEffect = Effect.fnUntraced(function* (
export const toModelMessagesEffect = Effect.fnUntraced(function* (
input: WithParts[],
model: Provider.Model,
options?: { stripMedia?: boolean },
) {
) {
const result: UIMessage[] = []
const toolNames = new Set<string>()
// Track media from tool results that need to be injected as user messages
@@ -699,7 +698,7 @@ export namespace MessageV2 {
if (
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")
)
) {
@@ -835,17 +834,17 @@ export namespace MessageV2 {
},
),
)
})
})
export function toModelMessages(
export function toModelMessages(
input: WithParts[],
model: Provider.Model,
options?: { stripMedia?: boolean },
): Promise<ModelMessage[]> {
): Promise<ModelMessage[]> {
return Effect.runPromise(toModelMessagesEffect(input, model, options).pipe(Effect.provide(EffectLogger.layer)))
}
}
export function page(input: { sessionID: SessionID; limit: number; before?: string }) {
export function page(input: { sessionID: SessionID; limit: number; before?: string }) {
const before = input.before ? cursor.decode(input.before) : undefined
const where = before
? and(eq(MessageTable.session_id, input.sessionID), older(before))
@@ -865,7 +864,7 @@ export namespace MessageV2 {
)
if (!row) throw new NotFoundError({ message: `Session not found: ${input.sessionID}` })
return {
items: [] as MessageV2.WithParts[],
items: [] as WithParts[],
more: false,
}
}
@@ -880,9 +879,9 @@ export namespace MessageV2 {
more,
cursor: more && tail ? cursor.encode({ id: tail.id, time: tail.time_created }) : undefined,
}
}
}
export function* stream(sessionID: SessionID) {
export function* stream(sessionID: SessionID) {
const size = 50
let before: string | undefined
while (true) {
@@ -894,9 +893,9 @@ export namespace MessageV2 {
if (!next.more || !next.cursor) break
before = next.cursor
}
}
}
export function parts(message_id: MessageID) {
export function parts(message_id: MessageID) {
const rows = Database.use((db) =>
db.select().from(PartTable).where(eq(PartTable.message_id, message_id)).orderBy(PartTable.id).all(),
)
@@ -907,11 +906,11 @@ export namespace MessageV2 {
id: row.id,
sessionID: row.session_id,
messageID: row.message_id,
}) as MessageV2.Part,
}) as Part,
)
}
}
export function get(input: { sessionID: SessionID; messageID: MessageID }): WithParts {
export function get(input: { sessionID: SessionID; messageID: MessageID }): WithParts {
const row = Database.use((db) =>
db
.select()
@@ -924,10 +923,10 @@ export namespace MessageV2 {
info: info(row),
parts: parts(input.messageID),
}
}
}
export function filterCompacted(msgs: Iterable<MessageV2.WithParts>) {
const result = [] as MessageV2.WithParts[]
export function filterCompacted(msgs: Iterable<WithParts>) {
const result = [] as WithParts[]
const completed = new Set<string>()
for (const msg of msgs) {
result.push(msg)
@@ -942,28 +941,28 @@ export namespace MessageV2 {
}
result.reverse()
return result
}
}
export const filterCompactedEffect = Effect.fnUntraced(function* (sessionID: SessionID) {
export const filterCompactedEffect = Effect.fnUntraced(function* (sessionID: SessionID) {
return filterCompacted(stream(sessionID))
})
})
export function fromError(
export function fromError(
e: unknown,
ctx: { providerID: ProviderID; aborted?: boolean },
): NonNullable<Assistant["error"]> {
): NonNullable<Assistant["error"]> {
switch (true) {
case e instanceof DOMException && e.name === "AbortError":
return new MessageV2.AbortedError(
return new AbortedError(
{ message: e.message },
{
cause: e,
},
).toObject()
case MessageV2.OutputLengthError.isInstance(e):
case OutputLengthError.isInstance(e):
return e
case LoadAPIKeyError.isInstance(e):
return new MessageV2.AuthError(
return new AuthError(
{
providerID: ctx.providerID,
message: e.message,
@@ -971,7 +970,7 @@ export namespace MessageV2 {
{ cause: e },
).toObject()
case (e as SystemError)?.code === "ECONNRESET":
return new MessageV2.APIError(
return new APIError(
{
message: "Connection reset by server",
isRetryable: true,
@@ -985,9 +984,9 @@ export namespace MessageV2 {
).toObject()
case e instanceof Error && (e as FetchDecompressionError).code === "ZlibError":
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",
isRetryable: true,
@@ -1004,7 +1003,7 @@ export namespace MessageV2 {
error: e,
})
if (parsed.type === "context_overflow") {
return new MessageV2.ContextOverflowError(
return new ContextOverflowError(
{
message: parsed.message,
responseBody: parsed.responseBody,
@@ -1013,7 +1012,7 @@ export namespace MessageV2 {
).toObject()
}
return new MessageV2.APIError(
return new APIError(
{
message: parsed.message,
statusCode: parsed.statusCode,
@@ -1031,7 +1030,7 @@ export namespace MessageV2 {
const parsed = ProviderError.parseStreamError(e)
if (parsed) {
if (parsed.type === "context_overflow") {
return new MessageV2.ContextOverflowError(
return new ContextOverflowError(
{
message: parsed.message,
responseBody: parsed.responseBody,
@@ -1039,7 +1038,7 @@ export namespace MessageV2 {
{ cause: e },
).toObject()
}
return new MessageV2.APIError(
return new APIError(
{
message: parsed.message,
isRetryable: parsed.isRetryable,
@@ -1053,5 +1052,6 @@ export namespace MessageV2 {
} catch {}
return new NamedError.Unknown({ message: JSON.stringify(e) }, { cause: e }).toObject()
}
}
}
export * as MessageV2 from "./message-v2"
packages/opencode/src/session/message.ts
+30 -30
View File
@@ -3,17 +3,16 @@ import { SessionID } from "./schema"
import { ModelID, ProviderID } from "../provider/schema"
import { NamedError } from "@opencode-ai/shared/util/error"
export namespace Message {
export const OutputLengthError = NamedError.create("MessageOutputLengthError", z.object({}))
export const AuthError = NamedError.create(
export const OutputLengthError = NamedError.create("MessageOutputLengthError", z.object({}))
export const AuthError = NamedError.create(
"ProviderAuthError",
z.object({
providerID: z.string(),
message: z.string(),
}),
)
)
export const ToolCall = z
export const ToolCall = z
.object({
state: z.literal("call"),
step: z.number().optional(),
@@ -24,9 +23,9 @@ export namespace Message {
.meta({
ref: "ToolCall",
})
export type ToolCall = z.infer<typeof ToolCall>
export type ToolCall = z.infer<typeof ToolCall>
export const ToolPartialCall = z
export const ToolPartialCall = z
.object({
state: z.literal("partial-call"),
step: z.number().optional(),
@@ -37,9 +36,9 @@ export namespace Message {
.meta({
ref: "ToolPartialCall",
})
export type ToolPartialCall = z.infer<typeof ToolPartialCall>
export type ToolPartialCall = z.infer<typeof ToolPartialCall>
export const ToolResult = z
export const ToolResult = z
.object({
state: z.literal("result"),
step: z.number().optional(),
@@ -51,14 +50,14 @@ export namespace Message {
.meta({
ref: "ToolResult",
})
export type ToolResult = z.infer<typeof ToolResult>
export type ToolResult = z.infer<typeof ToolResult>
export const ToolInvocation = z.discriminatedUnion("state", [ToolCall, ToolPartialCall, ToolResult]).meta({
export const ToolInvocation = z.discriminatedUnion("state", [ToolCall, ToolPartialCall, ToolResult]).meta({
ref: "ToolInvocation",
})
export type ToolInvocation = z.infer<typeof ToolInvocation>
})
export type ToolInvocation = z.infer<typeof ToolInvocation>
export const TextPart = z
export const TextPart = z
.object({
type: z.literal("text"),
text: z.string(),
@@ -66,9 +65,9 @@ export namespace Message {
.meta({
ref: "TextPart",
})
export type TextPart = z.infer<typeof TextPart>
export type TextPart = z.infer<typeof TextPart>
export const ReasoningPart = z
export const ReasoningPart = z
.object({
type: z.literal("reasoning"),
text: z.string(),
@@ -77,9 +76,9 @@ export namespace Message {
.meta({
ref: "ReasoningPart",
})
export type ReasoningPart = z.infer<typeof ReasoningPart>
export type ReasoningPart = z.infer<typeof ReasoningPart>
export const ToolInvocationPart = z
export const ToolInvocationPart = z
.object({
type: z.literal("tool-invocation"),
toolInvocation: ToolInvocation,
@@ -87,9 +86,9 @@ export namespace Message {
.meta({
ref: "ToolInvocationPart",
})
export type ToolInvocationPart = z.infer<typeof ToolInvocationPart>
export type ToolInvocationPart = z.infer<typeof ToolInvocationPart>
export const SourceUrlPart = z
export const SourceUrlPart = z
.object({
type: z.literal("source-url"),
sourceId: z.string(),
@@ -100,9 +99,9 @@ export namespace Message {
.meta({
ref: "SourceUrlPart",
})
export type SourceUrlPart = z.infer<typeof SourceUrlPart>
export type SourceUrlPart = z.infer<typeof SourceUrlPart>
export const FilePart = z
export const FilePart = z
.object({
type: z.literal("file"),
mediaType: z.string(),
@@ -112,25 +111,25 @@ export namespace Message {
.meta({
ref: "FilePart",
})
export type FilePart = z.infer<typeof FilePart>
export type FilePart = z.infer<typeof FilePart>
export const StepStartPart = z
export const StepStartPart = z
.object({
type: z.literal("step-start"),
})
.meta({
ref: "StepStartPart",
})
export type StepStartPart = z.infer<typeof StepStartPart>
export type StepStartPart = z.infer<typeof StepStartPart>
export const MessagePart = z
export const MessagePart = z
.discriminatedUnion("type", [TextPart, ReasoningPart, ToolInvocationPart, SourceUrlPart, FilePart, StepStartPart])
.meta({
ref: "MessagePart",
})
export type MessagePart = z.infer<typeof MessagePart>
export type MessagePart = z.infer<typeof MessagePart>
export const Info = z
export const Info = z
.object({
id: z.string(),
role: z.enum(["user", "assistant"]),
@@ -187,5 +186,6 @@ export namespace Message {
.meta({
ref: "Message",
})
export type Info = z.infer<typeof Info>
}
export type Info = z.infer<typeof Info>
export * as Message from "./message"
packages/opencode/src/session/processor.ts
+23 -23
View File
@@ -21,15 +21,14 @@ import { errorMessage } from "@/util/error"
import { Log } from "@/util"
import { isRecord } from "@/util/record"
export namespace SessionProcessor {
const DOOM_LOOP_THRESHOLD = 3
const log = Log.create({ service: "session.processor" })
const DOOM_LOOP_THRESHOLD = 3
const log = Log.create({ service: "session.processor" })
export type Result = "compact" | "stop" | "continue"
export type Result = "compact" | "stop" | "continue"
export type Event = LLM.Event
export type Event = LLM.Event
export interface Handle {
export interface Handle {
readonly message: MessageV2.Assistant
readonly updateToolCall: (
toolCallID: string,
@@ -45,26 +44,26 @@ export namespace SessionProcessor {
},
) => Effect.Effect<void>
readonly process: (streamInput: LLM.StreamInput) => Effect.Effect<Result>
}
}
type Input = {
type Input = {
assistantMessage: MessageV2.Assistant
sessionID: SessionID
model: Provider.Model
}
}
export interface Interface {
export interface Interface {
readonly create: (input: Input) => Effect.Effect<Handle>
}
}
type ToolCall = {
type ToolCall = {
partID: MessageV2.ToolPart["id"]
messageID: MessageV2.ToolPart["messageID"]
sessionID: MessageV2.ToolPart["sessionID"]
done: Deferred.Deferred<void>
}
}
interface ProcessorContext extends Input {
interface ProcessorContext extends Input {
toolcalls: Record<string, ToolCall>
shouldBreak: boolean
snapshot: string | undefined
@@ -72,13 +71,13 @@ export namespace SessionProcessor {
needsCompaction: boolean
currentText: MessageV2.TextPart | undefined
reasoningMap: Record<string, MessageV2.ReasoningPart>
}
}
type StreamEvent = Event
type StreamEvent = Event
export class Service extends Context.Service<Service, Interface>()("@opencode/SessionProcessor") {}
export class Service extends Context.Service<Service, Interface>()("@opencode/SessionProcessor") {}
export const layer: Layer.Layer<
export const layer: Layer.Layer<
Service,
never,
| Session.Service
@@ -91,7 +90,7 @@ export namespace SessionProcessor {
| Plugin.Service
| SessionSummary.Service
| SessionStatus.Service
> = Layer.effect(
> = Layer.effect(
Service,
Effect.gen(function* () {
const session = yield* Session.Service
@@ -600,9 +599,9 @@ export namespace SessionProcessor {
return Service.of({ create })
}),
)
)
export const defaultLayer = Layer.suspend(() =>
export const defaultLayer = Layer.suspend(() =>
layer.pipe(
Layer.provide(Session.defaultLayer),
Layer.provide(Snapshot.defaultLayer),
@@ -615,5 +614,6 @@ export namespace SessionProcessor {
Layer.provide(Bus.layer),
Layer.provide(Config.defaultLayer),
),
)
}
)
export * as SessionProcessor from "./processor"
packages/opencode/src/session/prompt.ts
+30 -30
View File
@@ -64,22 +64,21 @@ 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.`
export namespace SessionPrompt {
const log = Log.create({ service: "session.prompt" })
const elog = EffectLogger.create({ service: "session.prompt" })
const log = Log.create({ service: "session.prompt" })
const elog = EffectLogger.create({ service: "session.prompt" })
export interface Interface {
export interface Interface {
readonly cancel: (sessionID: SessionID) => Effect.Effect<void>
readonly prompt: (input: PromptInput) => Effect.Effect<MessageV2.WithParts>
readonly loop: (input: z.infer<typeof LoopInput>) => Effect.Effect<MessageV2.WithParts>
readonly shell: (input: ShellInput) => Effect.Effect<MessageV2.WithParts>
readonly command: (input: CommandInput) => Effect.Effect<MessageV2.WithParts>
readonly resolvePromptParts: (template: string) => Effect.Effect<PromptInput["parts"]>
}
}
export class Service extends Context.Service<Service, Interface>()("@opencode/SessionPrompt") {}
export class Service extends Context.Service<Service, Interface>()("@opencode/SessionPrompt") {}
export const layer = Layer.effect(
export const layer = Layer.effect(
Service,
Effect.gen(function* () {
const bus = yield* Bus.Service
@@ -1677,9 +1676,9 @@ NOTE: At any point in time through this workflow you should feel free to ask the
resolvePromptParts,
})
}),
)
)
export const defaultLayer = Layer.suspend(() =>
export const defaultLayer = Layer.suspend(() =>
layer.pipe(
Layer.provide(SessionRunState.defaultLayer),
Layer.provide(SessionStatus.defaultLayer),
@@ -1709,8 +1708,8 @@ NOTE: At any point in time through this workflow you should feel free to ask the
),
),
),
)
export const PromptInput = z.object({
)
export const PromptInput = z.object({
sessionID: SessionID.zod,
messageID: MessageID.zod.optional(),
model: z
@@ -1774,14 +1773,14 @@ NOTE: At any point in time through this workflow you should feel free to ask the
}),
]),
),
})
export type PromptInput = z.infer<typeof PromptInput>
})
export type PromptInput = z.infer<typeof PromptInput>
export const LoopInput = z.object({
export const LoopInput = z.object({
sessionID: SessionID.zod,
})
})
export const ShellInput = z.object({
export const ShellInput = z.object({
sessionID: SessionID.zod,
messageID: MessageID.zod.optional(),
agent: z.string(),
@@ -1792,10 +1791,10 @@ NOTE: At any point in time through this workflow you should feel free to ask the
})
.optional(),
command: z.string(),
})
export type ShellInput = z.infer<typeof ShellInput>
})
export type ShellInput = z.infer<typeof ShellInput>
export const CommandInput = z.object({
export const CommandInput = z.object({
messageID: MessageID.zod.optional(),
sessionID: SessionID.zod,
agent: z.string().optional(),
@@ -1815,14 +1814,14 @@ NOTE: At any point in time through this workflow you should feel free to ask the
]),
)
.optional(),
})
export type CommandInput = z.infer<typeof CommandInput>
})
export type CommandInput = z.infer<typeof CommandInput>
/** @internal Exported for testing */
export function createStructuredOutputTool(input: {
/** @internal Exported for testing */
export function createStructuredOutputTool(input: {
schema: Record<string, any>
onSuccess: (output: unknown) => void
}): AITool {
}): AITool {
// Remove $schema property if present (not needed for tool input)
const { $schema: _, ...toolSchema } = input.schema
@@ -1845,10 +1844,11 @@ NOTE: At any point in time through this workflow you should feel free to ask the
}
},
})
}
const bashRegex = /!`([^`]+)`/g
// Match [Image N] as single token, quoted strings, or non-space sequences
const argsRegex = /(?:\[Image\s+\d+\]|"[^"]*"|'[^']*'|[^\s"']+)/gi
const placeholderRegex = /\$(\d+)/g
const quoteTrimRegex = /^["']|["']$/g
}
const bashRegex = /!`([^`]+)`/g
// Match [Image N] as single token, quoted strings, or non-space sequences
const argsRegex = /(?:\[Image\s+\d+\]|"[^"]*"|'[^']*'|[^\s"']+)/gi
const placeholderRegex = /\$(\d+)/g
const quoteTrimRegex = /^["']|["']$/g
export * as SessionPrompt from "./prompt"
packages/opencode/src/session/retry.ts
+18 -18
View File
@@ -3,23 +3,22 @@ import { Cause, Clock, Duration, Effect, Schedule } from "effect"
import { MessageV2 } from "./message-v2"
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
// literal error string kind of sucks, but it is the simplest for now.
export const GO_UPSELL_MESSAGE = "Free usage exceeded, subscribe to Go https://opencode.ai/go"
// This exported message is shared with the TUI upsell detector. Matching on a
// literal error string kind of sucks, but it is the simplest for now.
export const GO_UPSELL_MESSAGE = "Free usage exceeded, subscribe to Go https://opencode.ai/go"
export const RETRY_INITIAL_DELAY = 2000
export const RETRY_BACKOFF_FACTOR = 2
export const RETRY_MAX_DELAY_NO_HEADERS = 30_000 // 30 seconds
export const RETRY_MAX_DELAY = 2_147_483_647 // max 32-bit signed integer for setTimeout
export const RETRY_INITIAL_DELAY = 2000
export const RETRY_BACKOFF_FACTOR = 2
export const RETRY_MAX_DELAY_NO_HEADERS = 30_000 // 30 seconds
export const RETRY_MAX_DELAY = 2_147_483_647 // max 32-bit signed integer for setTimeout
function cap(ms: number) {
function cap(ms: number) {
return Math.min(ms, RETRY_MAX_DELAY)
}
}
export function delay(attempt: number, error?: MessageV2.APIError) {
export function delay(attempt: number, error?: MessageV2.APIError) {
if (error) {
const headers = error.data.responseHeaders
if (headers) {
@@ -50,9 +49,9 @@ export namespace SessionRetry {
}
return cap(Math.min(RETRY_INITIAL_DELAY * Math.pow(RETRY_BACKOFF_FACTOR, attempt - 1), RETRY_MAX_DELAY_NO_HEADERS))
}
}
export function retryable(error: Err) {
export function retryable(error: Err) {
// context overflow errors should not be retried
if (MessageV2.ContextOverflowError.isInstance(error)) return undefined
if (MessageV2.APIError.isInstance(error)) {
@@ -102,12 +101,12 @@ export namespace SessionRetry {
return "Rate Limited"
}
return undefined
}
}
export function policy(opts: {
export function policy(opts: {
parse: (error: unknown) => Err
set: (input: { attempt: number; message: string; next: number }) => Effect.Effect<void>
}) {
}) {
return Schedule.fromStepWithMetadata(
Effect.succeed((meta: Schedule.InputMetadata<unknown>) => {
const error = opts.parse(meta.input)
@@ -121,5 +120,6 @@ export namespace SessionRetry {
})
}),
)
}
}
export * as SessionRetry from "./retry"
packages/opencode/src/session/revert.ts
+13 -13
View File
@@ -11,25 +11,24 @@ import { SessionID, MessageID, PartID } from "./schema"
import { SessionRunState } from "./run-state"
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({
sessionID: SessionID.zod,
messageID: MessageID.zod,
partID: PartID.zod.optional(),
})
export type RevertInput = z.infer<typeof RevertInput>
})
export type RevertInput = z.infer<typeof RevertInput>
export interface Interface {
export interface Interface {
readonly revert: (input: RevertInput) => Effect.Effect<Session.Info>
readonly unrevert: (input: { sessionID: SessionID }) => Effect.Effect<Session.Info>
readonly cleanup: (session: Session.Info) => Effect.Effect<void>
}
}
export class Service extends Context.Service<Service, Interface>()("@opencode/SessionRevert") {}
export class Service extends Context.Service<Service, Interface>()("@opencode/SessionRevert") {}
export const layer = Layer.effect(
export const layer = Layer.effect(
Service,
Effect.gen(function* () {
const sessions = yield* Session.Service
@@ -146,9 +145,9 @@ export namespace SessionRevert {
return Service.of({ revert, unrevert, cleanup })
}),
)
)
export const defaultLayer = Layer.suspend(() =>
export const defaultLayer = Layer.suspend(() =>
layer.pipe(
Layer.provide(SessionRunState.defaultLayer),
Layer.provide(Session.defaultLayer),
@@ -157,5 +156,6 @@ export namespace SessionRevert {
Layer.provide(Bus.layer),
Layer.provide(SessionSummary.defaultLayer),
),
)
}
)
export * as SessionRevert from "./revert"
packages/opencode/src/session/run-state.ts
+8 -8
View File
@@ -6,8 +6,7 @@ import { MessageV2 } from "./message-v2"
import { SessionID } from "./schema"
import { SessionStatus } from "./status"
export namespace SessionRunState {
export interface Interface {
export interface Interface {
readonly assertNotBusy: (sessionID: SessionID) => Effect.Effect<void>
readonly cancel: (sessionID: SessionID) => Effect.Effect<void>
readonly ensureRunning: (
@@ -20,11 +19,11 @@ export namespace SessionRunState {
onInterrupt: Effect.Effect<MessageV2.WithParts>,
work: Effect.Effect<MessageV2.WithParts>,
) => Effect.Effect<MessageV2.WithParts>
}
}
export class Service extends Context.Service<Service, Interface>()("@opencode/SessionRunState") {}
export class Service extends Context.Service<Service, Interface>()("@opencode/SessionRunState") {}
export const layer = Layer.effect(
export const layer = Layer.effect(
Service,
Effect.gen(function* () {
const status = yield* SessionStatus.Service
@@ -102,7 +101,8 @@ export namespace SessionRunState {
return Service.of({ assertNotBusy, cancel, ensureRunning, startShell })
}),
)
)
export const defaultLayer = layer.pipe(Layer.provide(SessionStatus.defaultLayer))
}
export const defaultLayer = layer.pipe(Layer.provide(SessionStatus.defaultLayer))
export * as SessionRunState from "./run-state"
packages/opencode/src/session/status.ts
+12 -12
View File
@@ -5,8 +5,7 @@ import { SessionID } from "./schema"
import { Effect, Layer, Context } from "effect"
import z from "zod"
export namespace SessionStatus {
export const Info = z
export const Info = z
.union([
z.object({
type: z.literal("idle"),
@@ -24,9 +23,9 @@ export namespace SessionStatus {
.meta({
ref: "SessionStatus",
})
export type Info = z.infer<typeof Info>
export type Info = z.infer<typeof Info>
export const Event = {
export const Event = {
Status: BusEvent.define(
"session.status",
z.object({
@@ -41,17 +40,17 @@ export namespace SessionStatus {
sessionID: SessionID.zod,
}),
),
}
}
export interface Interface {
export interface Interface {
readonly get: (sessionID: SessionID) => Effect.Effect<Info>
readonly list: () => Effect.Effect<Map<SessionID, Info>>
readonly set: (sessionID: SessionID, status: Info) => Effect.Effect<void>
}
}
export class Service extends Context.Service<Service, Interface>()("@opencode/SessionStatus") {}
export class Service extends Context.Service<Service, Interface>()("@opencode/SessionStatus") {}
export const layer = Layer.effect(
export const layer = Layer.effect(
Service,
Effect.gen(function* () {
const bus = yield* Bus.Service
@@ -82,7 +81,8 @@ export namespace SessionStatus {
return Service.of({ get, list, set })
}),
)
)
export const defaultLayer = layer.pipe(Layer.provide(Bus.layer))
}
export const defaultLayer = layer.pipe(Layer.provide(Bus.layer))
export * as SessionStatus from "./status"
packages/opencode/src/session/summary.ts
+13 -13
View File
@@ -7,8 +7,7 @@ import * as Session from "./session"
import { MessageV2 } from "./message-v2"
import { SessionID, MessageID } from "./schema"
export namespace SessionSummary {
function unquoteGitPath(input: string) {
function unquoteGitPath(input: string) {
if (!input.startsWith('"')) return input
if (!input.endsWith('"')) return input
const body = input.slice(1, -1)
@@ -62,17 +61,17 @@ export namespace SessionSummary {
}
return Buffer.from(bytes).toString()
}
}
export interface Interface {
export interface Interface {
readonly summarize: (input: { sessionID: SessionID; messageID: MessageID }) => Effect.Effect<void>
readonly diff: (input: { sessionID: SessionID; messageID?: MessageID }) => Effect.Effect<Snapshot.FileDiff[]>
readonly computeDiff: (input: { messages: MessageV2.WithParts[] }) => Effect.Effect<Snapshot.FileDiff[]>
}
}
export class Service extends Context.Service<Service, Interface>()("@opencode/SessionSummary") {}
export class Service extends Context.Service<Service, Interface>()("@opencode/SessionSummary") {}
export const layer = Layer.effect(
export const layer = Layer.effect(
Service,
Effect.gen(function* () {
const sessions = yield* Session.Service
@@ -147,19 +146,20 @@ export namespace SessionSummary {
return Service.of({ summarize, diff, computeDiff })
}),
)
)
export const defaultLayer = Layer.suspend(() =>
export const defaultLayer = Layer.suspend(() =>
layer.pipe(
Layer.provide(Session.defaultLayer),
Layer.provide(Snapshot.defaultLayer),
Layer.provide(Storage.defaultLayer),
Layer.provide(Bus.layer),
),
)
)
export const DiffInput = z.object({
export const DiffInput = z.object({
sessionID: SessionID.zod,
messageID: MessageID.zod.optional(),
})
}
})
export * as SessionSummary from "./summary"
packages/opencode/src/session/system.ts
+10 -10
View File
@@ -16,8 +16,7 @@ import type { Agent } from "@/agent/agent"
import { Permission } from "@/permission"
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"))
return [PROMPT_BEAST]
if (model.api.id.includes("gpt")) {
@@ -31,16 +30,16 @@ export namespace SystemPrompt {
if (model.api.id.toLowerCase().includes("trinity")) return [PROMPT_TRINITY]
if (model.api.id.toLowerCase().includes("kimi")) return [PROMPT_KIMI]
return [PROMPT_DEFAULT]
}
}
export interface Interface {
export interface Interface {
readonly environment: (model: Provider.Model) => string[]
readonly skills: (agent: Agent.Info) => Effect.Effect<string | undefined>
}
}
export class Service extends Context.Service<Service, Interface>()("@opencode/SystemPrompt") {}
export class Service extends Context.Service<Service, Interface>()("@opencode/SystemPrompt") {}
export const layer = Layer.effect(
export const layer = Layer.effect(
Service,
Effect.gen(function* () {
const skill = yield* Skill.Service
@@ -78,7 +77,8 @@ 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"
packages/opencode/src/session/todo.ts
+12 -12
View File
@@ -6,17 +6,16 @@ import z from "zod"
import { Database, eq, asc } from "../storage"
import { TodoTable } from "./session.sql"
export namespace Todo {
export const Info = z
export const Info = z
.object({
content: z.string().describe("Brief description of the task"),
status: z.string().describe("Current status of the task: pending, in_progress, completed, cancelled"),
priority: z.string().describe("Priority level of the task: high, medium, low"),
})
.meta({ ref: "Todo" })
export type Info = z.infer<typeof Info>
export type Info = z.infer<typeof Info>
export const Event = {
export const Event = {
Updated: BusEvent.define(
"todo.updated",
z.object({
@@ -24,16 +23,16 @@ export namespace Todo {
todos: z.array(Info),
}),
),
}
}
export interface Interface {
export interface Interface {
readonly update: (input: { sessionID: SessionID; todos: Info[] }) => Effect.Effect<void>
readonly get: (sessionID: SessionID) => Effect.Effect<Info[]>
}
}
export class Service extends Context.Service<Service, Interface>()("@opencode/SessionTodo") {}
export class Service extends Context.Service<Service, Interface>()("@opencode/SessionTodo") {}
export const layer = Layer.effect(
export const layer = Layer.effect(
Service,
Effect.gen(function* () {
const bus = yield* Bus.Service
@@ -79,7 +78,8 @@ export namespace Todo {
return Service.of({ update, get })
}),
)
)
export const defaultLayer = layer.pipe(Layer.provide(Bus.layer))
}
export const defaultLayer = layer.pipe(Layer.provide(Bus.layer))
export * as Todo from "./todo"