kilocode/packages/opencode/script/seed-e2e.ts

import { AppRuntime } from "@/effect/app-runtime"

const dir = process.env.KILO_E2E_PROJECT_DIR ?? process.cwd()
const title = process.env.KILO_E2E_SESSION_TITLE ?? "E2E Session"
const text = process.env.KILO_E2E_MESSAGE ?? "Seeded for UI e2e"
const model = process.env.KILO_E2E_MODEL ?? "kilo/kilo-auto/frontier"
const parts = model.split("/")
const providerID = parts[0] ?? "kilo" // kilocode_change
const modelID = parts.slice(1).join("/") || "kilo-auto/frontier" // kilocode_change
const now = Date.now()

const seed = async () => {
  const { Instance } = await import("../src/project/instance")
  const { InstanceBootstrap } = await import("../src/project/bootstrap")
  const { Config } = await import("../src/config/config")
  const { Session } = await import("../src/session")
  const { MessageID, PartID } = await import("../src/session/schema")
  const { Project } = await import("../src/project/project")
  const { ModelID, ProviderID } = await import("../src/provider/schema")
  const { ToolRegistry } = await import("../src/tool/registry")
  const { Effect } = await import("effect")

  try {
    await Instance.provide({
      directory: dir,
      init: () => AppRuntime.runPromise(InstanceBootstrap),
      fn: async () => {
        await AppRuntime.runPromise(Config.Service.use((cfg) => cfg.waitForDependencies()))
        await AppRuntime.runPromise(
          Effect.gen(function* () {
            const registry = yield* ToolRegistry.Service
            yield* registry.ids()
          }),
        )

        await AppRuntime.runPromise(
          Effect.gen(function* () {
            const session = yield* Session.Service
            const result = yield* session.create({ title })
            const messageID = MessageID.ascending()
            const partID = PartID.ascending()
            const message = {
              id: messageID,
              sessionID: result.id,
              role: "user" as const,
              time: { created: now },
              agent: "code", // kilocode_change - renamed from "build" to "code"
              model: {
                providerID: ProviderID.make(providerID),
                modelID: ModelID.make(modelID),
              },
            }
            const part = {
              id: partID,
              sessionID: result.id,
              messageID,
              type: "text" as const,
              text,
              time: { start: now },
            }
            yield* session.updateMessage(message)
            yield* session.updatePart(part)
          }),
        )
        await AppRuntime.runPromise(
          Project.Service.use((svc) => svc.update({ projectID: Instance.project.id, name: "E2E Project" })),
        )
      },
    })
  } finally {
    await Instance.disposeAll().catch(() => {})
    await AppRuntime.dispose().catch(() => {})
  }
}

await seed()