@vikasitai/vikasit-code 2.0.3 → 2.0.5

package/src/session/processor.ts

@@ -0,0 +1,697 @@
+import { Cause, Effect, Layer, ServiceMap } from "effect"
+import * as Stream from "effect/Stream"
+import { Agent } from "@/agent/agent"
+import { Bus } from "@/bus"
+import { Config } from "@/config/config"
+import { Permission } from "@/permission"
+import { Plugin } from "@/plugin"
+import { Snapshot } from "@/snapshot"
+import { Log } from "@/util/log"
+import { Session } from "."
+import { LLM } from "./llm"
+import { MessageV2 } from "./message-v2"
+import { isOverflow } from "./overflow"
+import { PartID } from "./schema"
+import type { SessionID } from "./schema"
+import { SessionRetry } from "./retry"
+import { SessionStatus } from "./status"
+import { SessionSummary } from "./summary"
+import { RateLimit } from "./rate-limit"
+import type { Provider } from "@/provider/provider"
+import { Question } from "@/question"
+
+declare const VIKASIT_CONSOLE_URL_DEFAULT: string | undefined
+
+export namespace SessionProcessor {
+  const DOOM_LOOP_THRESHOLD = 3
+  const log = Log.create({ service: "session.processor" })
+
+  export type Result = "compact" | "stop" | "continue"
+
+  export type Event = LLM.Event
+
+  export interface Handle {
+    readonly message: MessageV2.Assistant
+    readonly partFromToolCall: (toolCallID: string) => MessageV2.ToolPart | undefined
+    readonly abort: () => Effect.Effect<void>
+    readonly process: (streamInput: LLM.StreamInput) => Effect.Effect<Result>
+  }
+
+  type Input = {
+    assistantMessage: MessageV2.Assistant
+    sessionID: SessionID
+    model: Provider.Model
+  }
+
+  export interface Interface {
+    readonly create: (input: Input) => Effect.Effect<Handle>
+  }
+
+  interface ProcessorContext extends Input {
+    toolcalls: Record<string, MessageV2.ToolPart>
+    shouldBreak: boolean
+    snapshot: string | undefined
+    blocked: boolean
+    needsCompaction: boolean
+    goBalanceActivated: boolean
+    currentText: MessageV2.TextPart | undefined
+    reasoningMap: Record<string, MessageV2.ReasoningPart>
+  }
+
+  type StreamEvent = Event
+
+  export class Service extends ServiceMap.Service<Service, Interface>()("@opencode/SessionProcessor") {}
+
+  export const layer: Layer.Layer<
+    Service,
+    never,
+    | Session.Service
+    | Config.Service
+    | Bus.Service
+    | Snapshot.Service
+    | Agent.Service
+    | LLM.Service
+    | Permission.Service
+    | Plugin.Service
+    | SessionStatus.Service
+  > = Layer.effect(
+    Service,
+    Effect.gen(function* () {
+      const session = yield* Session.Service
+      const config = yield* Config.Service
+      const bus = yield* Bus.Service
+      const snapshot = yield* Snapshot.Service
+      const agents = yield* Agent.Service
+      const llm = yield* LLM.Service
+      const permission = yield* Permission.Service
+      const plugin = yield* Plugin.Service
+      const status = yield* SessionStatus.Service
+
+      const create = Effect.fn("SessionProcessor.create")(function* (input: Input) {
+        const ctx: ProcessorContext = {
+          assistantMessage: input.assistantMessage,
+          sessionID: input.sessionID,
+          model: input.model,
+          toolcalls: {},
+          shouldBreak: false,
+          snapshot: undefined,
+          blocked: false,
+          needsCompaction: false,
+          goBalanceActivated: false,
+          currentText: undefined,
+          reasoningMap: {},
+        }
+        let aborted = false
+
+        const parse = (e: unknown) =>
+          MessageV2.fromError(e, {
+            providerID: input.model.providerID,
+            aborted,
+          })
+
+        const handleEvent = Effect.fn("SessionProcessor.handleEvent")(function* (value: StreamEvent) {
+          switch (value.type) {
+            case "start":
+              yield* status.set(ctx.sessionID, { type: "busy" })
+              return
+
+            case "reasoning-start":
+              if (value.id in ctx.reasoningMap) return
+              ctx.reasoningMap[value.id] = {
+                id: PartID.ascending(),
+                messageID: ctx.assistantMessage.id,
+                sessionID: ctx.assistantMessage.sessionID,
+                type: "reasoning",
+                text: "",
+                time: { start: Date.now() },
+                metadata: value.providerMetadata,
+              }
+              yield* session.updatePart(ctx.reasoningMap[value.id])
+              return
+
+            case "reasoning-delta":
+              if (!(value.id in ctx.reasoningMap)) return
+              ctx.reasoningMap[value.id].text += value.text
+              if (value.providerMetadata) ctx.reasoningMap[value.id].metadata = value.providerMetadata
+              yield* session.updatePartDelta({
+                sessionID: ctx.reasoningMap[value.id].sessionID,
+                messageID: ctx.reasoningMap[value.id].messageID,
+                partID: ctx.reasoningMap[value.id].id,
+                field: "text",
+                delta: value.text,
+              })
+              return
+
+            case "reasoning-end":
+              if (!(value.id in ctx.reasoningMap)) return
+              ctx.reasoningMap[value.id].text = ctx.reasoningMap[value.id].text.trimEnd()
+              ctx.reasoningMap[value.id].time = { ...ctx.reasoningMap[value.id].time, end: Date.now() }
+              if (value.providerMetadata) ctx.reasoningMap[value.id].metadata = value.providerMetadata
+              yield* session.updatePart(ctx.reasoningMap[value.id])
+              delete ctx.reasoningMap[value.id]
+              return
+
+            case "tool-input-start":
+              if (ctx.assistantMessage.summary) {
+                throw new Error(`Tool call not allowed while generating summary: ${value.toolName}`)
+              }
+              ctx.toolcalls[value.id] = yield* session.updatePart({
+                id: ctx.toolcalls[value.id]?.id ?? PartID.ascending(),
+                messageID: ctx.assistantMessage.id,
+                sessionID: ctx.assistantMessage.sessionID,
+                type: "tool",
+                tool: value.toolName,
+                callID: value.id,
+                state: { status: "pending", input: {}, raw: "" },
+              } satisfies MessageV2.ToolPart)
+              return
+
+            case "tool-input-delta":
+              return
+
+            case "tool-input-end":
+              return
+
+            case "tool-call": {
+              if (ctx.assistantMessage.summary) {
+                throw new Error(`Tool call not allowed while generating summary: ${value.toolName}`)
+              }
+              const match = ctx.toolcalls[value.toolCallId]
+              if (!match) return
+              ctx.toolcalls[value.toolCallId] = yield* session.updatePart({
+                ...match,
+                tool: value.toolName,
+                state: { status: "running", input: value.input, time: { start: Date.now() } },
+                metadata: value.providerMetadata,
+              } satisfies MessageV2.ToolPart)
+
+              const parts = yield* Effect.promise(() => MessageV2.parts(ctx.assistantMessage.id))
+              const recentParts = parts.slice(-DOOM_LOOP_THRESHOLD)
+
+              if (
+                recentParts.length !== DOOM_LOOP_THRESHOLD ||
+                !recentParts.every(
+                  (part) =>
+                    part.type === "tool" &&
+                    part.tool === value.toolName &&
+                    part.state.status !== "pending" &&
+                    JSON.stringify(part.state.input) === JSON.stringify(value.input),
+                )
+              ) {
+                return
+              }
+
+              const agent = yield* agents.get(ctx.assistantMessage.agent)
+              yield* permission.ask({
+                permission: "doom_loop",
+                patterns: [value.toolName],
+                sessionID: ctx.assistantMessage.sessionID,
+                metadata: { tool: value.toolName, input: value.input },
+                always: [value.toolName],
+                ruleset: agent.permission,
+              })
+              return
+            }
+
+            case "tool-result": {
+              const match = ctx.toolcalls[value.toolCallId]
+              if (!match || match.state.status !== "running") return
+              yield* session.updatePart({
+                ...match,
+                state: {
+                  status: "completed",
+                  input: value.input ?? match.state.input,
+                  output: value.output.output,
+                  metadata: value.output.metadata,
+                  title: value.output.title,
+                  time: { start: match.state.time.start, end: Date.now() },
+                  attachments: value.output.attachments,
+                },
+              })
+              delete ctx.toolcalls[value.toolCallId]
+              return
+            }
+
+            case "tool-error": {
+              const match = ctx.toolcalls[value.toolCallId]
+              if (!match || match.state.status !== "running") return
+              yield* session.updatePart({
+                ...match,
+                state: {
+                  status: "error",
+                  input: value.input ?? match.state.input,
+                  error: value.error instanceof Error ? value.error.message : String(value.error),
+                  time: { start: match.state.time.start, end: Date.now() },
+                },
+              })
+              if (value.error instanceof Permission.RejectedError || value.error instanceof Question.RejectedError) {
+                ctx.blocked = ctx.shouldBreak
+              }
+              delete ctx.toolcalls[value.toolCallId]
+              return
+            }
+
+            case "error":
+              throw value.error
+
+            case "start-step":
+              ctx.snapshot = yield* snapshot.track()
+              yield* session.updatePart({
+                id: PartID.ascending(),
+                messageID: ctx.assistantMessage.id,
+                sessionID: ctx.sessionID,
+                snapshot: ctx.snapshot,
+                type: "step-start",
+              })
+              return
+
+            case "finish-step": {
+              const usage = Session.getUsage({
+                model: ctx.model,
+                usage: value.usage,
+                metadata: value.providerMetadata,
+              })
+              ctx.assistantMessage.finish = value.finishReason
+              ctx.assistantMessage.cost += usage.cost
+              ctx.assistantMessage.tokens = usage.tokens
+              yield* session.updatePart({
+                id: PartID.ascending(),
+                reason: value.finishReason,
+                snapshot: yield* snapshot.track(),
+                messageID: ctx.assistantMessage.id,
+                sessionID: ctx.assistantMessage.sessionID,
+                type: "step-finish",
+                tokens: usage.tokens,
+                cost: usage.cost,
+              })
+              yield* session.updateMessage(ctx.assistantMessage)
+              if (ctx.snapshot) {
+                const patch = yield* snapshot.patch(ctx.snapshot)
+                if (patch.files.length) {
+                  yield* session.updatePart({
+                    id: PartID.ascending(),
+                    messageID: ctx.assistantMessage.id,
+                    sessionID: ctx.sessionID,
+                    type: "patch",
+                    hash: patch.hash,
+                    files: patch.files,
+                  })
+                }
+                ctx.snapshot = undefined
+              }
+              // Check rate limit headers from Vikasit API proxy
+              if (ctx.model.providerID.startsWith("vikasit")) {
+                const rateLimitHeaders = RateLimit.parse(value.response?.headers)
+                if (rateLimitHeaders) {
+                  yield* RateLimit.notify(bus, rateLimitHeaders)
+                }
+              }
+              yield* Effect.promise(() =>
+                SessionSummary.summarize({
+                  sessionID: ctx.sessionID,
+                  messageID: ctx.assistantMessage.parentID,
+                }),
+              ).pipe(Effect.ignoreCause({ log: true, message: "session summary failed" }), Effect.forkDetach)
+              if (
+                !ctx.assistantMessage.summary &&
+                isOverflow({ cfg: yield* config.get(), tokens: usage.tokens, model: ctx.model })
+              ) {
+                ctx.needsCompaction = true
+              }
+              return
+            }
+
+            case "text-start":
+              ctx.currentText = {
+                id: PartID.ascending(),
+                messageID: ctx.assistantMessage.id,
+                sessionID: ctx.assistantMessage.sessionID,
+                type: "text",
+                text: "",
+                time: { start: Date.now() },
+                metadata: value.providerMetadata,
+              }
+              yield* session.updatePart(ctx.currentText)
+              return
+
+            case "text-delta":
+              if (!ctx.currentText) return
+              ctx.currentText.text += value.text
+              if (value.providerMetadata) ctx.currentText.metadata = value.providerMetadata
+              yield* session.updatePartDelta({
+                sessionID: ctx.currentText.sessionID,
+                messageID: ctx.currentText.messageID,
+                partID: ctx.currentText.id,
+                field: "text",
+                delta: value.text,
+              })
+              return
+
+            case "text-end":
+              if (!ctx.currentText) return
+              ctx.currentText.text = ctx.currentText.text.trimEnd()
+              ctx.currentText.text = (yield* plugin.trigger(
+                "experimental.text.complete",
+                {
+                  sessionID: ctx.sessionID,
+                  messageID: ctx.assistantMessage.id,
+                  partID: ctx.currentText.id,
+                },
+                { text: ctx.currentText.text },
+              )).text
+              ctx.currentText.time = { start: Date.now(), end: Date.now() }
+              if (value.providerMetadata) ctx.currentText.metadata = value.providerMetadata
+              yield* session.updatePart(ctx.currentText)
+              ctx.currentText = undefined
+              return
+
+            case "finish":
+              return
+
+            default:
+              log.info("unhandled", { ...value })
+              return
+          }
+        })
+
+        const cleanup = Effect.fn("SessionProcessor.cleanup")(function* () {
+          if (ctx.snapshot) {
+            const patch = yield* snapshot.patch(ctx.snapshot)
+            if (patch.files.length) {
+              yield* session.updatePart({
+                id: PartID.ascending(),
+                messageID: ctx.assistantMessage.id,
+                sessionID: ctx.sessionID,
+                type: "patch",
+                hash: patch.hash,
+                files: patch.files,
+              })
+            }
+            ctx.snapshot = undefined
+          }
+
+          if (ctx.currentText) {
+            const end = Date.now()
+            ctx.currentText.time = { start: ctx.currentText.time?.start ?? end, end }
+            yield* session.updatePart(ctx.currentText)
+            ctx.currentText = undefined
+          }
+
+          for (const part of Object.values(ctx.reasoningMap)) {
+            const end = Date.now()
+            yield* session.updatePart({
+              ...part,
+              time: { start: part.time.start ?? end, end },
+            })
+          }
+          ctx.reasoningMap = {}
+
+          const parts = yield* Effect.promise(() => MessageV2.parts(ctx.assistantMessage.id))
+          for (const part of parts) {
+            if (part.type !== "tool" || part.state.status === "completed" || part.state.status === "error") continue
+            yield* session.updatePart({
+              ...part,
+              state: {
+                ...part.state,
+                status: "error",
+                error: "Tool execution aborted",
+                time: { start: Date.now(), end: Date.now() },
+              },
+            })
+          }
+          ctx.assistantMessage.time.completed = Date.now()
+          yield* session.updateMessage(ctx.assistantMessage)
+        })
+
+        const halt = Effect.fn("SessionProcessor.halt")(function* (e: unknown) {
+          log.error("process", { error: e, stack: e instanceof Error ? e.stack : undefined })
+          const error = parse(e)
+          // Handle 429 from Vikasit API proxy
+          if (
+            ctx.model.providerID.startsWith("vikasit") &&
+            MessageV2.APIError.isInstance(error) &&
+            error.data.statusCode === 429
+          ) {
+            const rateLimitHeaders = RateLimit.parse(error.data.responseHeaders)
+            const balanceAvailable =
+              error.data.responseHeaders?.["x-vikasit-balance-available"] === "true" ||
+              error.data.responseBody?.includes("balanceAvailable")
+
+            // Helper to call consent endpoint and activate Go Balance
+            const activateGoBalance = async () => {
+              // Get API key: try env var, then read from auth.json
+              let apiKey = globalThis?.process?.env?.["VIKASIT_API_KEY"] ?? ""
+              if (!apiKey) {
+                try {
+                  const fs = await import("fs")
+                  const pathMod = await import("path")
+                  const { Global } = await import("@/global")
+                  const authFile = pathMod.default.join(Global.Path.data, "auth.json")
+                  const authData = JSON.parse(fs.default.readFileSync(authFile, "utf-8"))
+                  const providerAuth = authData[ctx.model.providerID]
+                  if (providerAuth?.type === "api" && providerAuth?.key) apiKey = providerAuth.key
+                  else if (providerAuth?.type === "oauth" && providerAuth?.access) apiKey = providerAuth.access
+                } catch {}
+              }
+              // Build consent URL from the error's request URL or model config
+              const requestUrl = MessageV2.APIError.isInstance(error) ? error.data.metadata?.url : undefined
+              const modelUrl = requestUrl ?? input.model.api?.url ?? ""
+              const consoleUrl = typeof VIKASIT_CONSOLE_URL_DEFAULT === "string" ? VIKASIT_CONSOLE_URL_DEFAULT : "https://vikasit.ai"
+              const consentUrl = modelUrl
+                ? modelUrl.replace(/\/zen(\/go)?\/v1\/?.*$/, "/zen/go/v1/consent")
+                : `${consoleUrl}/zen/go/v1/consent`
+              log.info("activating Go Balance", { consentUrl, apiKeyPresent: !!apiKey })
+              try {
+                const res = await fetch(consentUrl, {
+                  method: "POST",
+                  headers: { "x-api-key": apiKey },
+                })
+                const body = await res.text()
+                log.info("consent response", { status: res.status, ok: res.ok, body })
+                if (res.ok) {
+                  log.info("Go Balance activated via consent endpoint")
+                  ctx.goBalanceActivated = true
+                }
+                return res.ok
+              } catch (err) {
+                log.error("consent call failed", { error: err })
+                return false
+              }
+            }
+
+            if (balanceAvailable) {
+              // Ask user if they want to activate Go Balance
+              yield* Effect.tryPromise({
+                try: async () => {
+                  const answers = await Question.ask({
+                    sessionID: ctx.sessionID,
+                    questions: [
+                      {
+                        question:
+                          "You've hit your subscription rate limit. Would you like to activate Go balance? (Pay-as-you-go charges will apply until the rate limit window resets)",
+                        header: "Activate Go Balance?",
+                        options: [
+                          { label: "Yes", description: "Activate Go balance now" },
+                          { label: "No", description: "Wait for rate limit to reset" },
+                        ],
+                        custom: false,
+                      },
+                    ],
+                  })
+                  if (answers[0]?.[0] === "Yes") {
+                    await activateGoBalance()
+                  }
+                },
+                catch: (e) => { log.warn("question prompt failed", { error: String(e) }) },
+              })
+            } else {
+              // No Go balance available — offer to open billing page to add funds, then continue
+              const consoleUrl = typeof VIKASIT_CONSOLE_URL_DEFAULT === "string" ? VIKASIT_CONSOLE_URL_DEFAULT : "https://vikasit.ai"
+              let resolved = false
+              while (!resolved) {
+                resolved = yield* Effect.tryPromise({
+                  try: async () => {
+                    const answers = await Question.ask({
+                      sessionID: ctx.sessionID,
+                      questions: [
+                        {
+                          question:
+                            "You've hit your subscription rate limit and have no Go balance. Add balance on the website, then select 'Continue with Go' to resume.",
+                          header: "Add Go Balance?",
+                          options: [
+                            { label: "Open billing", description: "Open billing page in browser to add Go balance" },
+                            { label: "Continue with Go", description: "I've added balance — activate Go and continue" },
+                            { label: "Wait", description: "Wait for rate limit to reset" },
+                          ],
+                          custom: false,
+                        },
+                      ],
+                    })
+                    if (answers[0]?.[0] === "Open billing") {
+                      const open = globalThis?.process?.platform === "darwin" ? "open" : "xdg-open"
+                      const { exec } = await import("child_process")
+                      exec(`${open} "${consoleUrl}/auth"`)
+                      return false // keep looping — show prompt again after opening browser
+                    } else if (answers[0]?.[0] === "Continue with Go") {
+                      await activateGoBalance()
+                      return true
+                    }
+                    return true // "Wait" selected
+                  },
+                  catch: (e) => { log.warn("question prompt failed", { error: String(e) }); return true },
+                })
+              }
+            }
+
+            // If Go Balance was activated, skip the error — let the retry loop handle it
+            if (ctx.goBalanceActivated) {
+              ctx.goBalanceActivated = false
+              return
+            }
+
+            // Show rate limit notification (with retry time)
+            if (rateLimitHeaders) {
+              yield* RateLimit.notify(bus, { ...rateLimitHeaders, utilization: 1.0 })
+            }
+          }
+          // Handle insufficient balance / no payment method (401 CreditsError)
+          if (
+            ctx.model.providerID.startsWith("vikasit") &&
+            MessageV2.APIError.isInstance(error) &&
+            error.data.statusCode === 401 &&
+            error.data.responseBody?.includes("CreditsError")
+          ) {
+            const consoleUrl = typeof VIKASIT_CONSOLE_URL_DEFAULT === "string" ? VIKASIT_CONSOLE_URL_DEFAULT : "https://vikasit.ai"
+            // Extract billing URL from error message if present
+            const billingMatch = error.data.responseBody?.match(/https?:\/\/[^\s"]+\/billing/)
+            const billingPageUrl = billingMatch?.[0] ?? `${consoleUrl}/auth`
+            yield* Effect.tryPromise({
+              try: async () => {
+                const answers = await Question.ask({
+                  sessionID: ctx.sessionID,
+                  questions: [
+                    {
+                      question: error.data.message,
+                      header: "Insufficient Balance",
+                      options: [
+                        { label: "Open billing", description: "Open billing page in browser to add funds" },
+                        { label: "OK", description: "Dismiss" },
+                      ],
+                      custom: false,
+                    },
+                  ],
+                })
+                if (answers[0]?.[0] === "Open billing") {
+                  const open = globalThis?.process?.platform === "darwin" ? "open" : "xdg-open"
+                  const { exec } = await import("child_process")
+                  exec(`${open} "${billingPageUrl}"`)
+                }
+              },
+              catch: (e) => { log.warn("question prompt failed", { error: String(e) }) },
+            })
+          }
+          if (MessageV2.ContextOverflowError.isInstance(error)) {
+            ctx.needsCompaction = true
+            yield* bus.publish(Session.Event.Error, { sessionID: ctx.sessionID, error })
+            return
+          }
+          ctx.assistantMessage.error = error
+          yield* bus.publish(Session.Event.Error, {
+            sessionID: ctx.assistantMessage.sessionID,
+            error: ctx.assistantMessage.error,
+          })
+          yield* status.set(ctx.sessionID, { type: "idle" })
+        })
+
+        const abort = Effect.fn("SessionProcessor.abort")(() =>
+          Effect.gen(function* () {
+            if (!ctx.assistantMessage.error) {
+              yield* halt(new DOMException("Aborted", "AbortError"))
+            }
+            if (!ctx.assistantMessage.time.completed) {
+              yield* cleanup()
+              return
+            }
+            yield* session.updateMessage(ctx.assistantMessage)
+          }),
+        )
+
+        const process = Effect.fn("SessionProcessor.process")(function* (streamInput: LLM.StreamInput) {
+          log.info("process")
+          ctx.needsCompaction = false
+          ctx.shouldBreak = (yield* config.get()).experimental?.continue_loop_on_deny !== true
+
+          return yield* Effect.gen(function* () {
+            yield* Effect.gen(function* () {
+              ctx.currentText = undefined
+              ctx.reasoningMap = {}
+              const stream = llm.stream(streamInput)
+
+              yield* stream.pipe(
+                Stream.tap((event) => handleEvent(event)),
+                Stream.takeUntil(() => ctx.needsCompaction),
+                Stream.runDrain,
+              )
+            }).pipe(
+              Effect.onInterrupt(() => Effect.sync(() => void (aborted = true))),
+              Effect.catchCauseIf(
+                (cause) => !Cause.hasInterruptsOnly(cause),
+                (cause) => Effect.fail(Cause.squash(cause)),
+              ),
+              Effect.retry(
+                SessionRetry.policy({
+                  parse,
+                  set: (info) =>
+                    status.set(ctx.sessionID, {
+                      type: "retry",
+                      attempt: info.attempt,
+                      message: info.message,
+                      next: info.next,
+                    }),
+                }),
+              ),
+              Effect.catch(halt),
+              Effect.ensuring(cleanup()),
+            )
+
+            if (aborted && !ctx.assistantMessage.error) {
+              yield* abort()
+            }
+            if (ctx.needsCompaction) return "compact"
+            if (ctx.blocked || ctx.assistantMessage.error || aborted) return "stop"
+            return "continue"
+          }).pipe(Effect.onInterrupt(() => abort().pipe(Effect.asVoid)))
+        })
+
+        return {
+          get message() {
+            return ctx.assistantMessage
+          },
+          partFromToolCall(toolCallID: string) {
+            return ctx.toolcalls[toolCallID]
+          },
+          abort,
+          process,
+        } as unknown as Handle
+      })
+
+      return Service.of({ create } as Interface)
+    }),
+  )
+
+  export const defaultLayer = Layer.unwrap(
+    Effect.sync(() =>
+      layer.pipe(
+        Layer.provide(Session.defaultLayer),
+        Layer.provide(Snapshot.defaultLayer),
+        Layer.provide(Agent.defaultLayer),
+        Layer.provide(LLM.defaultLayer),
+        Layer.provide(Permission.layer),
+        Layer.provide(Plugin.defaultLayer),
+        Layer.provide(SessionStatus.layer.pipe(Layer.provide(Bus.layer))),
+        Layer.provide(Bus.layer),
+        Layer.provide(Config.defaultLayer),
+      ),
+    ),
+  )
+}