@sleepy-ai/cli 0.1.5 → 0.1.7

package/src/session/compaction.ts

@@ -1,545 +0,0 @@
-import { BusEvent } from "@/bus/bus-event"
-import { Bus } from "@/bus"
-import * as Session from "./session"
-import { SessionID, MessageID, PartID } from "./schema"
-import { Provider } from "../provider"
-import { MessageV2 } from "./message-v2"
-import z from "zod"
-import { Token } from "../util"
-import { Log } from "../util"
-import { SessionProcessor } from "./processor"
-import { Agent } from "@/agent/agent"
-import { Plugin } from "@/plugin"
-import { Config } from "@/config"
-import { NotFoundError } from "@/storage"
-import { ModelID, ProviderID } from "@/provider/schema"
-import { Effect, Layer, Context } from "effect"
-import { InstanceState } from "@/effect"
-import { isOverflow as overflow, usable } from "./overflow"
-import { makeRuntime } from "@/effect/run-service"
-import { fn } from "@/util/fn"
-
-const log = Log.create({ service: "session.compaction" })
-
-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"]
-const DEFAULT_TAIL_TURNS = 2
-const MIN_PRESERVE_RECENT_TOKENS = 2_000
-const MAX_PRESERVE_RECENT_TOKENS = 8_000
-type Turn = {
-  start: number
-  end: number
-  id: MessageID
-}
-
-function preserveRecentBudget(input: { cfg: Config.Info; model: Provider.Model }) {
-  return (
-    input.cfg.compaction?.preserve_recent_tokens ??
-    Math.min(MAX_PRESERVE_RECENT_TOKENS, Math.max(MIN_PRESERVE_RECENT_TOKENS, Math.floor(usable(input) * 0.25)))
-  )
-}
-
-function turns(messages: MessageV2.WithParts[]) {
-  const result: Turn[] = []
-  for (let i = 0; i < messages.length; i++) {
-    const msg = messages[i]
-    if (msg.info.role !== "user") continue
-    if (msg.parts.some((part) => part.type === "compaction")) continue
-    result.push({
-      start: i,
-      end: messages.length,
-      id: msg.info.id,
-    })
-  }
-  for (let i = 0; i < result.length - 1; i++) {
-    result[i].end = result[i + 1].start
-  }
-  return result
-}
-
-export interface Interface {
-  readonly isOverflow: (input: {
-    tokens: MessageV2.Assistant["tokens"]
-    model: Provider.Model
-  }) => Effect.Effect<boolean>
-  readonly prune: (input: { sessionID: SessionID; agentID?: string }) => Effect.Effect<void>
-  readonly process: (input: {
-    parentID: MessageID
-    messages: MessageV2.WithParts[]
-    sessionID: SessionID
-    auto: boolean
-    overflow?: boolean
-    agentID?: string
-  }) => Effect.Effect<"continue" | "stop" | "text-repeat">
-  readonly create: (input: {
-    sessionID: SessionID
-    agent: string
-    model: { providerID: ProviderID; modelID: ModelID }
-    auto: boolean
-    overflow?: boolean
-    agentID?: string
-  }) => Effect.Effect<void>
-}
-
-export class Service extends Context.Service<Service, Interface>()("@opencode/SessionCompaction") {}
-
-export const layer: Layer.Layer<
-  Service,
-  never,
-  | Bus.Service
-  | Config.Service
-  | Session.Service
-  | Agent.Service
-  | Plugin.Service
-  | SessionProcessor.Service
-  | Provider.Service
-> = Layer.effect(
-  Service,
-  Effect.gen(function* () {
-    const bus = yield* Bus.Service
-    const config = yield* Config.Service
-    const session = yield* Session.Service
-    const agents = yield* Agent.Service
-    const plugin = yield* Plugin.Service
-    const processors = yield* SessionProcessor.Service
-    const provider = yield* Provider.Service
-
-    const isOverflow = Effect.fn("SessionCompaction.isOverflow")(function* (input: {
-      tokens: MessageV2.Assistant["tokens"]
-      model: Provider.Model
-    }) {
-      return overflow({ cfg: yield* config.get(), tokens: input.tokens, model: input.model })
-    })
-
-    const estimate = Effect.fn("SessionCompaction.estimate")(function* (input: {
-      messages: MessageV2.WithParts[]
-      model: Provider.Model
-    }) {
-      const msgs = yield* MessageV2.toModelMessagesEffect(input.messages, input.model)
-      return Token.estimate(JSON.stringify(msgs))
-    })
-
-    const select = Effect.fn("SessionCompaction.select")(function* (input: {
-      messages: MessageV2.WithParts[]
-      cfg: Config.Info
-      model: Provider.Model
-    }) {
-      const limit = input.cfg.compaction?.tail_turns ?? DEFAULT_TAIL_TURNS
-      if (limit <= 0) return { head: input.messages, tail_start_id: undefined }
-      const budget = preserveRecentBudget({ cfg: input.cfg, model: input.model })
-      const all = turns(input.messages)
-      if (!all.length) return { head: input.messages, tail_start_id: undefined }
-      const recent = all.slice(-limit)
-      const sizes = yield* Effect.forEach(
-        recent,
-        (turn) =>
-          estimate({
-            messages: input.messages.slice(turn.start, turn.end),
-            model: input.model,
-          }),
-        { concurrency: 1 },
-      )
-      if (sizes.at(-1)! > budget) {
-        log.info("tail fallback", { budget, size: sizes.at(-1) })
-        return { head: input.messages, tail_start_id: undefined }
-      }
-
-      let total = 0
-      let keep: Turn | undefined
-      for (let i = recent.length - 1; i >= 0; i--) {
-        const size = sizes[i]
-        if (total + size > budget) break
-        total += size
-        keep = recent[i]
-      }
-
-      if (!keep || keep.start === 0) return { head: input.messages, tail_start_id: undefined }
-      return {
-        head: input.messages.slice(0, keep.start),
-        tail_start_id: keep.id,
-      }
-    })
-
-    // goes backwards through parts until there are PRUNE_PROTECT tokens worth of tool
-    // calls, then erases output of older tool calls to free context space.
-    // Scoped to (sessionID, agentID): only inspects messages belonging to the
-    // given actor — main-agent messages stay untouched when agentID is set.
-    const prune = Effect.fn("SessionCompaction.prune")(function* (input: {
-      sessionID: SessionID
-      agentID?: string
-    }) {
-      const cfg = yield* config.get()
-      if (!cfg.compaction?.prune) return
-      log.info("pruning", { agentID: input.agentID ?? "main" })
-
-      const msgs = yield* MessageV2.filterCompactedEffect(input.sessionID, { agentID: input.agentID }).pipe(
-        Effect.catchIf(NotFoundError.isInstance, () => Effect.succeed(undefined)),
-      )
-      if (!msgs) return
-
-      let total = 0
-      let pruned = 0
-      const toPrune: MessageV2.ToolPart[] = []
-      let turns = 0
-
-      loop: for (let msgIndex = msgs.length - 1; msgIndex >= 0; msgIndex--) {
-        const msg = msgs[msgIndex]
-        if (msg.info.role === "user") turns++
-        if (turns < 2) continue
-        if (msg.info.role === "assistant" && msg.info.summary) break loop
-        for (let partIndex = msg.parts.length - 1; partIndex >= 0; partIndex--) {
-          const part = msg.parts[partIndex]
-          if (part.type === "tool")
-            if (part.state.status === "completed") {
-              if (PRUNE_PROTECTED_TOOLS.includes(part.tool)) continue
-              if (part.state.time.compacted) break loop
-              const estimate = Token.estimate(part.state.output)
-              total += estimate
-              if (total > PRUNE_PROTECT) {
-                pruned += estimate
-                toPrune.push(part)
-              }
-            }
-        }
-      }
-
-      log.info("found", { pruned, total })
-      if (pruned > PRUNE_MINIMUM) {
-        for (const part of toPrune) {
-          if (part.state.status === "completed") {
-            part.state.time.compacted = Date.now()
-            yield* session.updatePart(part)
-          }
-        }
-        log.info("pruned", { count: toPrune.length })
-      }
-    })
-
-    const processCompaction = Effect.fn("SessionCompaction.process")(function* (input: {
-      parentID: MessageID
-      messages: MessageV2.WithParts[]
-      sessionID: SessionID
-      auto: boolean
-      overflow?: boolean
-      agentID?: string
-    }) {
-      const parent = input.messages.findLast((m) => m.info.id === input.parentID)
-      if (!parent || parent.info.role !== "user") {
-        throw new Error(`Compaction parent must be a user message: ${input.parentID}`)
-      }
-      const userMessage = parent.info
-      const compactionPart = parent.parts.find((part): part is MessageV2.CompactionPart => part.type === "compaction")
-
-      let messages = input.messages
-      let replay:
-        | {
-            info: MessageV2.User
-            parts: MessageV2.Part[]
-          }
-        | undefined
-      if (input.overflow) {
-        const idx = input.messages.findIndex((m) => m.info.id === input.parentID)
-        for (let i = idx - 1; i >= 0; i--) {
-          const msg = input.messages[i]
-          if (msg.info.role === "user" && !msg.parts.some((p) => p.type === "compaction")) {
-            replay = { info: msg.info, parts: msg.parts }
-            messages = input.messages.slice(0, i)
-            break
-          }
-        }
-        const hasContent =
-          replay && messages.some((m) => m.info.role === "user" && !m.parts.some((p) => p.type === "compaction"))
-        if (!hasContent) {
-          replay = undefined
-          messages = input.messages
-        }
-      }
-
-      const agent = yield* agents.get("compaction")
-      const model = agent.model
-        ? yield* provider.getModel(agent.model.providerID, agent.model.modelID)
-        : yield* provider.getModel(userMessage.model.providerID, userMessage.model.modelID)
-      const cfg = yield* config.get()
-      const history = compactionPart && messages.at(-1)?.info.id === input.parentID ? messages.slice(0, -1) : messages
-      const selected = yield* select({
-        messages: history,
-        cfg,
-        model,
-      })
-      // Allow plugins to inject context or replace compaction prompt.
-      const compacting = yield* plugin.trigger(
-        "experimental.session.compacting",
-        { sessionID: input.sessionID },
-        { context: [], prompt: undefined },
-      )
-      const defaultPrompt = `When constructing the summary, try to stick to this template:
----
-## Goal
-
-[What goal(s) is the user trying to accomplish?]
-
-## Instructions
-
-- [What important instructions did the user give you that are relevant]
-- [If there is a plan or spec, include information about it so next agent can continue using it]
-
-## Discoveries
-
-[What notable things were learned during this conversation that would be useful for the next agent to know when continuing the work]
-
-## Accomplished
-
-[What work has been completed, what work is still in progress, and what work is left?]
-
-## Relevant files / directories
-
-[Construct a structured list of relevant files that have been read, edited, or created that pertain to the task at hand. If all the files in a directory are relevant, include the path to the directory.]
----`
-
-      const prompt = compacting.prompt ?? [defaultPrompt, ...compacting.context].join("\n\n")
-      const msgs = structuredClone(selected.head)
-      yield* plugin.trigger("experimental.chat.messages.transform", {}, { messages: msgs })
-      const modelMessages = yield* MessageV2.toModelMessagesEffect(msgs, model, { stripMedia: true })
-      const ctx = yield* InstanceState.context
-      const msg: MessageV2.Assistant = {
-        id: MessageID.ascending(),
-        role: "assistant",
-        parentID: input.parentID,
-        sessionID: input.sessionID,
-        agentID: input.agentID ?? undefined,
-        mode: "compaction",
-        agent: "compaction",
-        variant: userMessage.model.variant,
-        summary: true,
-        path: {
-          cwd: ctx.directory,
-          root: ctx.worktree,
-        },
-        cost: 0,
-        tokens: {
-          output: 0,
-          input: 0,
-          reasoning: 0,
-          cache: { read: 0, write: 0 },
-        },
-        modelID: model.id,
-        providerID: model.providerID,
-        time: {
-          created: Date.now(),
-        },
-      }
-      yield* session.updateMessage(msg)
-      const processor = yield* processors.create({
-        assistantMessage: msg,
-        sessionID: input.sessionID,
-        model,
-      })
-      const result = yield* processor.process({
-        user: userMessage,
-        agent,
-        sessionID: input.sessionID,
-        tools: {},
-        system: [],
-        messages: [
-          ...modelMessages,
-          {
-            role: "user",
-            content: [{ type: "text", text: prompt }],
-          },
-        ],
-        model,
-      })
-
-      if (result === "overflow") {
-        processor.message.error = new MessageV2.ContextOverflowError({
-          message: replay
-            ? "Conversation history too large to compact - exceeds model context limit"
-            : "Session too large to compact - context exceeds model limit even after stripping media",
-        }).toObject()
-        processor.message.finish = "error"
-        yield* session.updateMessage(processor.message)
-        return "stop"
-      }
-
-      if (result === "text-repeat") return "stop"
-
-      if (compactionPart && selected.tail_start_id && compactionPart.tail_start_id !== selected.tail_start_id) {
-        yield* session.updatePart({
-          ...compactionPart,
-          tail_start_id: selected.tail_start_id,
-        })
-      }
-
-      if (result === "continue" && input.auto) {
-        if (replay) {
-          const original = replay.info
-          const replayMsg = yield* session.updateMessage({
-            id: MessageID.ascending(),
-            role: "user",
-            sessionID: input.sessionID,
-            agentID: input.agentID ?? undefined,
-            time: { created: Date.now() },
-            agent: original.agent,
-            model: original.model,
-            format: original.format,
-            tools: original.tools,
-            system: original.system,
-          })
-          for (const part of replay.parts) {
-            if (part.type === "compaction") continue
-            const replayPart =
-              part.type === "file" && MessageV2.isMedia(part.mime)
-                ? { type: "text" as const, text: `[Attached ${part.mime}: ${part.filename ?? "file"}]` }
-                : part
-            yield* session.updatePart({
-              ...replayPart,
-              id: PartID.ascending(),
-              messageID: replayMsg.id,
-              sessionID: input.sessionID,
-            })
-          }
-        }
-
-        if (!replay) {
-          const info = yield* provider.getProvider(userMessage.model.providerID)
-          if (
-            (yield* plugin.trigger(
-              "experimental.compaction.autocontinue",
-              {
-                sessionID: input.sessionID,
-                agent: userMessage.agent,
-                model: yield* provider.getModel(userMessage.model.providerID, userMessage.model.modelID),
-                provider: {
-                  source: info.source,
-                  info,
-                  options: info.options,
-                },
-                message: userMessage,
-                overflow: input.overflow === true,
-              },
-              { enabled: true },
-            )).enabled
-          ) {
-            const continueMsg = yield* session.updateMessage({
-              id: MessageID.ascending(),
-              role: "user",
-              sessionID: input.sessionID,
-              agentID: input.agentID ?? undefined,
-              time: { created: Date.now() },
-              agent: userMessage.agent,
-              model: userMessage.model,
-            })
-            const text =
-              (input.overflow
-                ? "The previous request exceeded the provider's size limit due to large media attachments. The conversation was compacted and media files were removed from context. If the user was asking about attached images or files, explain that the attachments were too large to process and suggest they try again with smaller or fewer files.\n\n"
-                : "") +
-              "Continue if you have next steps, or stop and ask for clarification if you are unsure how to proceed."
-            yield* session.updatePart({
-              id: PartID.ascending(),
-              messageID: continueMsg.id,
-              sessionID: input.sessionID,
-              type: "text",
-              // Internal marker for auto-compaction followups so provider plugins
-              // can distinguish them from manual post-compaction user prompts.
-              // This is not a stable plugin contract and may change or disappear.
-              metadata: { compaction_continue: true },
-              synthetic: true,
-              text,
-              time: {
-                start: Date.now(),
-                end: Date.now(),
-              },
-            })
-          }
-        }
-      }
-
-      if (processor.message.error) return "stop"
-      if (result === "continue") yield* bus.publish(Event.Compacted, { sessionID: input.sessionID })
-      return result
-    })
-
-    const create = Effect.fn("SessionCompaction.create")(function* (input: {
-      sessionID: SessionID
-      agent: string
-      model: { providerID: ProviderID; modelID: ModelID }
-      auto: boolean
-      overflow?: boolean
-      agentID?: string
-    }) {
-      // Tag the synthetic boundary message with agent_id so per-actor
-      // filterCompactedEffect lookups stop at this row when scoping by the
-      // same agent_id (subagent compaction stays inside its own scope).
-      const msg = yield* session.updateMessage({
-        id: MessageID.ascending(),
-        role: "user",
-        model: input.model,
-        sessionID: input.sessionID,
-        agentID: input.agentID ?? undefined,
-        agent: input.agent,
-        time: { created: Date.now() },
-      })
-      yield* session.updatePart({
-        id: PartID.ascending(),
-        messageID: msg.id,
-        sessionID: msg.sessionID,
-        type: "compaction",
-        auto: input.auto,
-        overflow: input.overflow,
-      })
-    })
-
-    return Service.of({
-      isOverflow,
-      prune,
-      process: processCompaction,
-      create,
-    })
-  }),
-)
-
-export const defaultLayer = Layer.suspend(() =>
-  layer.pipe(
-    Layer.provide(Provider.defaultLayer),
-    Layer.provide(Session.defaultLayer),
-    Layer.provide(SessionProcessor.defaultLayer),
-    Layer.provide(Agent.defaultLayer),
-    Layer.provide(Plugin.defaultLayer),
-    Layer.provide(Bus.layer),
-    Layer.provide(Config.defaultLayer),
-  ),
-)
-
-const { runPromise } = makeRuntime(Service, defaultLayer)
-
-export async function isOverflow(input: { tokens: MessageV2.Assistant["tokens"]; model: Provider.Model }) {
-  return runPromise((svc) => svc.isOverflow(input))
-}
-
-export async function prune(input: { sessionID: SessionID; agentID?: string }) {
-  return runPromise((svc) => svc.prune(input))
-}
-
-export const create = fn(
-  z.object({
-    sessionID: SessionID.zod,
-    agent: z.string(),
-    model: z.object({ providerID: ProviderID.zod, modelID: ModelID.zod }),
-    auto: z.boolean(),
-    overflow: z.boolean().optional(),
-    agentID: z.string().optional(),
-  }),
-  (input) => runPromise((svc) => svc.create(input)),
-)
-
-export * as SessionCompaction from "./compaction"

package/src/session/external-import.sql.ts

@@ -1,18 +0,0 @@
-import { sqliteTable, text, integer, primaryKey } from "drizzle-orm/sqlite-core"
-import type { SessionID, MessageID } from "./schema"
-
-export type ExternalSource = "cc" | "codex" | "opencode"
-
-export const ExternalImportTable = sqliteTable(
-  "external_import",
-  {
-    source: text().$type<ExternalSource>().notNull(),
-    source_key: text().notNull(),
-    session_id: text().$type<SessionID>().notNull(),
-    source_path: text().notNull(),
-    source_mtime: integer().notNull(),
-    time_imported: integer().notNull(),
-    message_ids: text({ mode: "json" }).$type<MessageID[]>(),
-  },
-  (table) => [primaryKey({ columns: [table.source, table.source_key] })],
-)

package/src/session/external-import.ts

@@ -1,136 +0,0 @@
-import path from "path"
-import { existsSync } from "fs"
-import { Glob } from "@sleepy-ai/shared/util/glob"
-import { Global } from "../global"
-import { Log } from "../util"
-import { Database, eq, count } from "../storage"
-import { openReadonly } from "../storage/read-sqlite"
-import { ExternalImportTable } from "./external-import.sql"
-import { ClaudeImport } from "./claude-import"
-import { CodexImport } from "./codex-import"
-import { OpencodeImport, DEFAULT_DB_PATH } from "./opencode-import"
-
-const log = Log.create({ service: "external-import" })
-
-export type Source = "cc" | "codex" | "opencode"
-export const ALL_SOURCES: readonly Source[] = ["cc", "codex", "opencode"] as const
-
-export type ImportStats = {
-  scanned: number
-  imported: number
-  resynced: number
-  skipped: number
-  errors: string[]
-}
-
-export type RunAllResult = {
-  [K in Source]: ImportStats
-}
-
-export type SourceScan = {
-  available: boolean
-  sessions: number
-  imported: number
-}
-
-export type ScanResult = {
-  [K in Source]: SourceScan
-}
-
-/** Pre-scan: per-source availability + total/already-imported session counts. Read-only. */
-export async function scan(): Promise<ScanResult> {
-  const importedBySource = (source: Source) =>
-    Database.use((db) =>
-      db.select({ n: count() }).from(ExternalImportTable).where(eq(ExternalImportTable.source, source)).get(),
-    )?.n ?? 0
-
-  const result: ScanResult = {
-    cc: { available: false, sessions: 0, imported: 0 },
-    codex: { available: false, sessions: 0, imported: 0 },
-    opencode: { available: false, sessions: 0, imported: 0 },
-  }
-
-  // 每源独立 try:单源失败(权限/坏文件)不让整个 scan 端点 500,降级为 available:false
-  const ccRoot = path.join(Global.Path.home, ".claude", "projects")
-  if (existsSync(ccRoot)) {
-    try {
-      const files = await Glob.scan("*/*.jsonl", { cwd: ccRoot, absolute: false })
-      result.cc = { available: true, sessions: files.length, imported: importedBySource("cc") }
-    } catch (e) {
-      log.warn("cc scan failed", { error: String(e) })
-    }
-  }
-
-  const codexRoot = path.join(Global.Path.home, ".codex", "sessions")
-  if (existsSync(codexRoot)) {
-    try {
-      const files = await Glob.scan("**/*.jsonl", { cwd: codexRoot, absolute: false })
-      result.codex = { available: true, sessions: files.length, imported: importedBySource("codex") }
-    } catch (e) {
-      log.warn("codex scan failed", { error: String(e) })
-    }
-  }
-
-  if (existsSync(DEFAULT_DB_PATH)) {
-    try {
-      const db = openReadonly(DEFAULT_DB_PATH)
-      try {
-        const row = db.get("SELECT count(*) AS n FROM session") as { n: number } | null
-        result.opencode = { available: true, sessions: row?.n ?? 0, imported: importedBySource("opencode") }
-      } finally {
-        db.close()
-      }
-    } catch (e) {
-      log.warn("opencode scan failed", { error: String(e) })
-    }
-  }
-
-  return result
-}
-
-function emptyStats(): ImportStats {
-  return { scanned: 0, imported: 0, resynced: 0, skipped: 0, errors: [] }
-}
-
-export async function runAll(opts?: {
-  sources?: Source[]
-  force?: boolean
-}): Promise<RunAllResult> {
-  const sources = new Set(opts?.sources ?? ALL_SOURCES)
-
-  const result: RunAllResult = {
-    cc: emptyStats(),
-    codex: emptyStats(),
-    opencode: emptyStats(),
-  }
-
-  for (const source of ALL_SOURCES) {
-    if (!sources.has(source)) continue
-    try {
-      if (source === "cc") {
-        result.cc = await ClaudeImport.run({ force: opts?.force })
-      } else if (source === "codex") {
-        result.codex = await CodexImport.run({ force: opts?.force })
-      } else {
-        result.opencode = await OpencodeImport.run({ force: opts?.force })
-      }
-    } catch (e) {
-      log.warn(`${source} import failed`, { error: String(e) })
-      result[source].errors.push(String(e))
-    }
-  }
-
-  const totals = {
-    scanned: result.cc.scanned + result.codex.scanned + result.opencode.scanned,
-    imported: result.cc.imported + result.codex.imported + result.opencode.imported,
-    resynced: result.cc.resynced + result.codex.resynced + result.opencode.resynced,
-    errors: result.cc.errors.length + result.codex.errors.length + result.opencode.errors.length,
-  }
-  if (totals.imported + totals.resynced > 0 || totals.errors > 0) {
-    log.info("external import", totals)
-  }
-
-  return result
-}
-
-export * as ExternalImport from "./external-import"