@vibe-forge/core 0.7.4 → 0.8.0

package/src/controllers/task/generate-adapter-query-options.ts

@@ -1,218 +0,0 @@
-import process from 'node:process'
-import { basename, dirname } from 'node:path'
-
-import type { AdapterQueryOptions } from '#~/adapter/type.js'
-import { DefinitionLoader } from '#~/utils/definition-loader.js'
-import type { Definition, Entity, Filter, Skill, Spec } from '#~/utils/definition-loader.js'
-
-const filterSkills = (
-  skills: Definition<Skill>[],
-  selection?: AdapterQueryOptions['skills']
-) => {
-  if (selection == null) return skills
-
-  const include = selection.include != null && selection.include.length > 0
-    ? new Set(selection.include)
-    : undefined
-  const exclude = new Set(selection.exclude ?? [])
-
-  return skills.filter((skill) => {
-    const name = basename(dirname(skill.path))
-    return (include == null || include.has(name)) && !exclude.has(name)
-  })
-}
-
-const dedupeSkills = (skills: Definition<Skill>[]) => {
-  const seen = new Set<string>()
-  return skills.filter((skill) => {
-    if (seen.has(skill.path)) return false
-    seen.add(skill.path)
-    return true
-  })
-}
-
-type SkillSelectionInput =
-  | AdapterQueryOptions['skills']
-  | Entity['skills']
-  | Spec['skills']
-
-const toNormalizedSkillSelection = (
-  selection?: SkillSelectionInput
-): AdapterQueryOptions['skills'] | undefined => {
-  if (selection == null) return undefined
-
-  if (Array.isArray(selection)) {
-    return selection.length > 0
-      ? {
-          include: selection
-        }
-      : undefined
-  }
-
-  if ('type' in selection && Array.isArray(selection.list)) {
-    const list = selection.list.filter((item): item is string => typeof item === 'string')
-    return selection.type === 'include'
-      ? {
-          include: list
-        }
-      : {
-          exclude: list
-        }
-  }
-
-  return selection
-}
-
-const mergeSkillSelections = (
-  ...selections: Array<SkillSelectionInput | undefined>
-): AdapterQueryOptions['skills'] | undefined => {
-  let include: Set<string> | undefined
-  const exclude = new Set<string>()
-
-  for (const selection of selections) {
-    const normalized = toNormalizedSkillSelection(selection)
-    if (normalized == null) continue
-
-    if (normalized.include != null && normalized.include.length > 0) {
-      const current = new Set(normalized.include)
-      include = include == null
-        ? current
-        : new Set([...include].filter(item => current.has(item)))
-    }
-
-    for (const item of normalized.exclude ?? []) {
-      exclude.add(item)
-    }
-  }
-
-  if (include == null && exclude.size === 0) return undefined
-
-  return {
-    include: include == null ? undefined : [...include],
-    exclude: exclude.size === 0 ? undefined : [...exclude]
-  }
-}
-
-const getIncludedSkillNames = (selection?: SkillSelectionInput): string[] => {
-  const normalized = toNormalizedSkillSelection(selection)
-  return normalized?.include ?? []
-}
-
-export async function generateAdapterQueryOptions(
-  type: 'spec' | 'entity' | undefined,
-  name?: string,
-  cwd: string = process.cwd(),
-  input?: {
-    skills?: AdapterQueryOptions['skills']
-  }
-) {
-  const loader = new DefinitionLoader(cwd)
-  const options: Partial<AdapterQueryOptions> = {}
-  const systemPromptParts: string[] = []
-  let effectiveSkillSelection = toNormalizedSkillSelection(input?.skills)
-
-  // 1. 获取数据
-  // 1.1 获取默认数据
-  const entities = type !== 'entity'
-    ? await loader.loadDefaultEntities()
-    : []
-  const defaultSkills = await loader.loadDefaultSkills()
-  const rules = await loader.loadDefaultRules()
-  const specs = await loader.loadDefaultSpecs()
-
-  // 1.2 获取指定数据
-  const targetSkills: Definition<Skill>[] = []
-  let targetBody = ''
-  let targetToolsFilter: Filter | undefined
-  let targetMcpServersFilter: Filter | undefined
-  if (type && name) {
-    const data = {
-      spec: await loader.loadSpec(name),
-      entity: await loader.loadEntity(name)
-    }[type]
-    if (!data) {
-      throw new Error(`Failed to load ${type} ${name}`)
-    }
-    const { attributes, body } = data
-    if (
-      attributes.rules
-    ) {
-      // always load spec or entity tagged rules
-      rules.push(
-        ...(
-          await loader.loadRules(attributes.rules, {
-            baseDir: dirname(data.path)
-          })
-        ).map((rule) => ({
-          ...rule,
-          attributes: {
-            ...rule.attributes,
-            // 实体或流程中的规则为默认加载
-            always: true
-          }
-        }))
-      )
-    }
-    if (
-      attributes.skills
-    ) {
-      effectiveSkillSelection = mergeSkillSelections(
-        input?.skills,
-        attributes.skills
-      )
-      targetSkills.push(
-        ...filterSkills(
-          await loader.loadSkills(getIncludedSkillNames(attributes.skills)),
-          effectiveSkillSelection
-        )
-      )
-    }
-
-    targetBody = body
-    targetToolsFilter = attributes.tools
-    targetMcpServersFilter = attributes.mcpServers
-  }
-  const skills = filterSkills(defaultSkills, effectiveSkillSelection)
-  let selectedSkillsPrompt: Definition<Skill>[] = []
-  if (input?.skills?.include != null && input.skills.include.length > 0) {
-    selectedSkillsPrompt = dedupeSkills(
-      filterSkills(
-        await loader.loadSkills(input.skills.include),
-        effectiveSkillSelection
-      )
-    )
-  }
-
-  // 2. 基于数据生成上下文
-  // 2.1 加载关联上下文
-  systemPromptParts.push(loader.generateRulesPrompt(rules))
-  systemPromptParts.push(loader.generateSkillsPrompt(dedupeSkills(targetSkills)))
-  systemPromptParts.push(loader.generateSkillsPrompt(
-    selectedSkillsPrompt.filter(skill => !targetSkills.some(target => target.path === skill.path))
-  ))
-  // 2.2 加载上下文路由
-  systemPromptParts.push(loader.generateEntitiesRoutePrompt(entities))
-  systemPromptParts.push(loader.generateSkillsRoutePrompt(skills))
-  systemPromptParts.push(loader.generateSpecRoutePrompt(specs))
-  // 2.3 加载目标上下文与配置
-  systemPromptParts.push(targetBody)
-  targetToolsFilter && (
-    options.tools = targetToolsFilter
-  )
-  targetMcpServersFilter && (
-    options.mcpServers = targetMcpServersFilter
-  )
-
-  options.systemPrompt = systemPromptParts.join('\n\n')
-  return [
-    {
-      rules,
-      targetSkills,
-      entities,
-      skills,
-      specs,
-      targetBody
-    },
-    options
-  ] as const
-}

package/src/controllers/task/index.ts

@@ -1,2 +0,0 @@
-export { generateAdapterQueryOptions } from './generate-adapter-query-options'
-export { run } from './run'

package/src/controllers/task/prepare.ts

@@ -1,68 +0,0 @@
-import process from 'node:process'
-
-import { loadConfig } from '@vibe-forge/core'
-import type { AdapterCtx, AdapterQueryOptions } from '@vibe-forge/core/adapter'
-import { getCache, setCache } from '@vibe-forge/core/utils/cache'
-import { createLogger } from '@vibe-forge/core/utils/create-logger'
-import { uuid } from '@vibe-forge/core/utils/uuid'
-
-import { resolveServerLogLevel } from '#~/env.js'
-
-import type { RunTaskOptions } from './type'
-
-export const prepare = async (
-  options: RunTaskOptions,
-  adapterOptions: AdapterQueryOptions
-) => {
-  const cwd = options.cwd ?? process.env.__VF_PROJECT_WORKSPACE_FOLDER__ ?? process.cwd()
-
-  const {
-    sessionId = uuid()
-  } = adapterOptions
-  const {
-    ctxId = process.env.__VF_PROJECT_AI_CTX_ID__ ?? sessionId,
-    env: envFromOptions
-  } = options
-  const {
-    __IS_LOADER_CLI__: _0,
-    ...prevEnv
-  } = {
-    ...process.env,
-    ...envFromOptions
-  }
-  const env: Record<string, string | null | undefined> = {
-    ...prevEnv,
-    __VF_PROJECT_AI_CTX_ID__: ctxId,
-    __VF_PROJECT_AI_SESSION_ID__: sessionId,
-    __VF_PROJECT_AI_RUN_TYPE__: adapterOptions.runtime,
-    // 移除 NODE_OPTIONS 环境变量，防止干扰子进程的运行环境
-    NODE_OPTIONS: undefined
-  }
-  const logger = createLogger(
-    cwd,
-    ctxId,
-    sessionId,
-    env?.LOG_PREFIX ?? '',
-    resolveServerLogLevel(env)
-  )
-
-  const jsonVariables: Record<string, string | null | undefined> = {
-    ...env,
-    WORKSPACE_FOLDER: cwd,
-    __VF_PROJECT_WORKSPACE_FOLDER__: cwd
-  }
-  const [config, userConfig] = await loadConfig({ jsonVariables })
-  return [
-    {
-      ctxId,
-      cwd,
-      env,
-      cache: {
-        set: (key, value) => setCache(cwd, ctxId, sessionId, key, value),
-        get: (key) => getCache(cwd, ctxId, sessionId, key)
-      },
-      logger,
-      configs: [config, userConfig]
-    } satisfies AdapterCtx
-  ] as const
-}

package/src/controllers/task/run.ts

@@ -1,191 +0,0 @@
-import type { AdapterCtx, AdapterOutputEvent, AdapterQueryOptions } from '#~/adapter/index.js'
-import { loadAdapter } from '#~/adapter/index.js'
-import type { ModelServiceConfig } from '#~/config.js'
-import { callHook } from '#~/hooks/call.js'
-import type { TaskDetail } from '#~/types.js'
-
-import { prepare } from './prepare'
-import type { RunTaskOptions } from './type'
-
-const normalizeNonEmptyString = (value: unknown) => (
-  typeof value === 'string' && value.trim() !== '' ? value.trim() : undefined
-)
-
-const pickFirstNonEmptyString = (values: unknown[]) =>
-  values
-    .map(normalizeNonEmptyString)
-    .find((value): value is string => value != null)
-
-const resolveQueryModel = (params: {
-  config: AdapterCtx['configs'][0]
-  userConfig: AdapterCtx['configs'][1]
-  inputModel?: string
-}) => {
-  const inputModel = normalizeNonEmptyString(params.inputModel)
-  // User explicitly provided a model → pass through as-is.
-  // The adapter decides CCR vs native based on whether it contains ",".
-  if (inputModel != null) return inputModel
-
-  // No explicit model → auto-resolve from modelServices config.
-  // Produces "service,model" format when services are configured,
-  // which signals the adapter to route through CCR.
-  const mergedModelServices = {
-    ...(params.config?.modelServices ?? {}),
-    ...(params.userConfig?.modelServices ?? {})
-  }
-  const mergedDefaultModel = pickFirstNonEmptyString(
-    [
-      params.userConfig?.defaultModel,
-      params.config?.defaultModel
-    ]
-  )
-  const mergedDefaultModelService = pickFirstNonEmptyString(
-    [
-      params.userConfig?.defaultModelService,
-      params.config?.defaultModelService
-    ]
-  )
-
-  const serviceEntries = Object.entries(mergedModelServices)
-  const modelToService = new Map<string, string>()
-  const availableModels: string[] = []
-  for (const [serviceKey, serviceValue] of serviceEntries) {
-    const service = (serviceValue != null && typeof serviceValue === 'object')
-      ? serviceValue as ModelServiceConfig
-      : undefined
-    const models = Array.isArray(service?.models)
-      ? service?.models.filter(item => typeof item === 'string' && item.trim() !== '')
-      : []
-    for (const model of models) {
-      if (!modelToService.has(model)) modelToService.set(model, serviceKey)
-      availableModels.push(model)
-    }
-  }
-
-  if (availableModels.length === 0) return undefined
-
-  const resolveDefaultModel = () => {
-    if (mergedDefaultModel && modelToService.has(mergedDefaultModel)) return mergedDefaultModel
-    if (mergedDefaultModelService && mergedModelServices[mergedDefaultModelService]) {
-      const service = mergedModelServices[mergedDefaultModelService] as ModelServiceConfig | undefined
-      const models = Array.isArray(service?.models)
-        ? service?.models.filter((item: unknown) => typeof item === 'string' && (item as string).trim() !== '')
-        : []
-      if (models.length > 0) return models[0]
-    }
-    return availableModels[0]
-  }
-
-  const resolvedModel = resolveDefaultModel()
-  if (!resolvedModel) return undefined
-
-  const resolvedService = modelToService.get(resolvedModel) ??
-    mergedDefaultModelService ??
-    serviceEntries[0]?.[0]
-
-  return resolvedService ? `${resolvedService},${resolvedModel}` : resolvedModel
-}
-
-declare module '@vibe-forge/core' {
-  interface Cache {
-    base: Omit<AdapterCtx, 'logger' | 'cache'>
-    detail: TaskDetail
-  }
-}
-
-export const run = async (
-  options: RunTaskOptions,
-  adapterOptions: AdapterQueryOptions
-) => {
-  const [ctx] = await prepare(options, adapterOptions)
-  const {
-    configs: [config, userConfig]
-  } = ctx
-
-  const { logger, cache, ...base } = ctx
-
-  await cache.set('base', base)
-
-  const adapters = {
-    ...config?.adapters,
-    ...userConfig?.adapters
-  }
-  // dprint-ignore
-  const adapterType =
-    // 0. adapter from options
-    options.adapter ??
-    // 1. config default adapter
-    config?.defaultAdapter ??
-    // 2. user config default adapter
-    userConfig?.defaultAdapter ??
-    // 3. first adapter in config
-    (() => {
-      const adapterNames = Object.keys(adapters)
-      if (adapterNames.length === 0) {
-        throw new Error('No adapter found in config, please set adapters in config file')
-      }
-      return adapterNames[0]
-    })()
-
-  const originalOnEvent = adapterOptions.onEvent
-  const wrappedOnEvent = (event: AdapterOutputEvent) => {
-    if (event.type === 'init') {
-      originalOnEvent({
-        ...event,
-        data: {
-          ...event.data,
-          adapter: adapterType
-        }
-      })
-      return
-    }
-
-    if (event.type === 'exit') {
-      const { data } = event
-
-      void callHook('TaskStop', {
-        adapter: adapterType,
-        cwd: ctx.cwd,
-        sessionId: adapterOptions.sessionId,
-
-        options,
-        adapterOptions,
-
-        exitCode: data.exitCode,
-        stderr: data.stderr
-      }, ctx.env)
-        .catch((e) => {
-          logger.error('[Hook] TaskStop failed', e)
-        })
-    }
-    originalOnEvent(event)
-  }
-
-  const adapter = await loadAdapter(adapterType)
-  await adapter.init?.(ctx)
-
-  const resolvedModel = resolveQueryModel({
-    config,
-    userConfig,
-    inputModel: adapterOptions.model
-  })
-
-  await callHook('TaskStart', {
-    adapter: adapterType,
-    cwd: ctx.cwd,
-    sessionId: adapterOptions.sessionId,
-
-    options,
-    adapterOptions
-  }, ctx.env)
-  const session = await adapter.query(
-    ctx,
-    {
-      ...adapterOptions,
-      model: resolvedModel,
-      onEvent: wrappedOnEvent
-    }
-  )
-
-  return { session, ctx }
-}

package/src/controllers/task/schema.ts

@@ -1,131 +0,0 @@
-import z from 'zod'
-
-export const TaskOptions = z.object({
-  type: z
-    .union([
-      z.literal('entity'),
-      // 基础库 API 能力开发
-      z.literal('spec')
-    ])
-    .describe('任务模式'),
-  name: z.string().describe('垂类知识库基准目标'),
-  specParams: z.record(z.string()).describe('SPEC 执行时的参数').optional(),
-  runtime: z
-    .object({
-      type: z
-        .string()
-        .describe('选择特定的 AI CLI 类型'),
-      noDefaultSystemPrompt: z
-        .boolean()
-        .describe('是否禁用默认的系统提示')
-        .optional()
-    }),
-  description: z
-    .string()
-    .describe('本次任务的描述，介绍关于本次任务需要进行的工作')
-    .optional(),
-  sessionId: z
-    .string()
-    .describe(
-      '复用上次会话的历史消息作为本次任务的上下文。\n' +
-        '- 通常如果某个任务在执行的时候出现了非预期的错误，那么你可以通过传入相同的 sessionID 来继续这个会话\n' +
-        '- 如果有一个步骤需要间隔执行，比如说先执行任务 A 的 A-1，然后完成后执行 B 的 B-1，等 B-1 这个完成后回到 A-1 继续执行 A-2 时也可以使用'
-    )
-    .optional(),
-  frontendTimeout: z
-    .number()
-    .describe(
-      '前台等待时间的上限，单位秒，默认值为 8 分钟。' +
-        '在超过这个时间后该任务会被转化为一个后台任务，并直接返回当前的输出以及 sessionId，你可以通过 sessionId 来继续这个任务，或者查询任务的状态'
-    )
-    .default(8 * 60)
-    .optional(),
-  defaultModel: z.string().describe('默认的模型').optional()
-})
-
-export type TaskOptions = z.infer<typeof TaskOptions>
-
-export const MCPRunTasksOptions = z.object({
-  tasks: z
-    .array(TaskOptions)
-    .describe(
-      '子任务列表。传递多个子任务时会并发执行多个，可用于优化整体任务效率。'
-    ),
-
-  bashDefaultTimeoutMs: z.number().optional(),
-  bashMaxTimeoutMs: z.number().optional(),
-  maxMcpOutputTokens: z.number().optional(),
-  mcpTimeout: z.number().optional(),
-  mcpToolTimeout: z.number().optional()
-})
-
-export type MCPRunTasksOptions = z.infer<typeof MCPRunTasksOptions>
-
-export const Options = z
-  .object({
-    taskId: z
-      .string()
-      .describe(
-        '唯一 id，会用于关联多个任务相关信息。\n' +
-          '如果是第一次执行，则不需要传入，会在返回值中自动生成一个，在下次创建或复用的时候必须传入该值以供记录相关上下文。\n' +
-          '用户指定了 taskId，则以用户指定的 taskId 为准，在创建任务时则必须要指定对应的 taskId 参数。'
-      )
-      .optional()
-  })
-  .extend(MCPRunTasksOptions.shape)
-
-export type Options = z.infer<typeof Options>
-
-export const Entity = z.object({
-  prompt: z
-    .string()
-    .describe('实体的描述，简单介绍一下当前实体的作用。')
-    .optional(),
-  promptPath: z
-    .string()
-    .describe(
-      '实体的描述文件路径，文件内容为实体的描述。默认为当前目录下的 AGENTS.md 文件。'
-    )
-    .optional(),
-  rules: z
-    .array(
-      z.union([
-        z.string(),
-        z.discriminatedUnion('type', [
-          z.object({
-            type: z.literal('local').optional(),
-            path: z.string(),
-            desc: z.string().optional()
-          }),
-          z.object({
-            type: z.literal('remote'),
-            tags: z.array(z.string()).describe('关键 tag').optional(),
-            desc: z.string().describe('知识库的描述').optional()
-          })
-        ])
-      ])
-    )
-    .optional()
-    .describe('垂类 agent 的规则集合'),
-  skills: z
-    .object({
-      type: z.union([z.literal('include'), z.literal('exclude')]),
-      list: z.array(z.string()).describe('技能列表')
-    })
-    .optional(),
-  mcpServers: z
-    .object({
-      include: z.array(z.string()).describe('包含的服务名称列表'),
-      exclude: z.array(z.string()).describe('排除的服务名称列表')
-    })
-    .optional(),
-  tools: z
-    .object({
-      include: z.array(z.string()).describe('包含的工具名称列表'),
-      exclude: z.array(z.string()).describe('排除的工具名称列表')
-    })
-    .optional(),
-  defaultModel: z.string().optional()
-})
-
-export type Entity = z.infer<typeof Entity>

package/src/controllers/task/type.ts

@@ -1,6 +0,0 @@
-export interface RunTaskOptions {
-  ctxId?: string
-  adapter?: string
-  env?: Record<string, string | undefined | null>
-  cwd?: string
-}

package/src/hooks/call.ts

@@ -1,74 +0,0 @@
-import { Buffer } from 'node:buffer'
-import { spawn } from 'node:child_process'
-import path from 'node:path'
-import process from 'node:process'
-
-import type { HookInputs, HookOutputs } from './type'
-
-export type HookEventName = keyof HookInputs
-
-type HookInputPayload<K extends HookEventName> = Omit<HookInputs[K], 'hookEventName'>
-
-const pickHookEnv = (env: Record<string, unknown>): Record<string, string> => {
-  const result: Record<string, string> = {}
-  for (const [key, value] of Object.entries(env)) {
-    if (typeof value === 'string') {
-      result[key] = value
-    }
-  }
-  return result
-}
-
-const resolveHookCliJs = () => {
-  try {
-    const pkgJsonPath = require.resolve('@vibe-forge/cli/package.json')
-    return path.resolve(path.dirname(pkgJsonPath), 'call-hook.js')
-  } catch (error) {
-    throw new Error('Failed to resolve @vibe-forge/cli hook entry', { cause: error })
-  }
-}
-
-export const callHook = async <K extends HookEventName>(
-  hookEventName: K,
-  input: HookInputPayload<K>,
-  env: Record<string, unknown> = process.env
-): Promise<HookOutputs[K]> => {
-  const childEnv = pickHookEnv(env)
-  const child = spawn(process.execPath, [resolveHookCliJs()], {
-    cwd: typeof input.cwd === 'string' ? input.cwd : process.cwd(),
-    env: childEnv,
-    stdio: ['pipe', 'pipe', 'pipe']
-  })
-
-  const stdoutChunks: Buffer[] = []
-  const stderrChunks: Buffer[] = []
-
-  child.stdout.on('data', chunk => stdoutChunks.push(chunk))
-  child.stderr.on('data', chunk => stderrChunks.push(chunk))
-
-  const exitCode = await new Promise<number>((resolve, reject) => {
-    child.once('error', reject)
-    child.once('close', code => resolve(code ?? 0))
-    child.stdin.end(JSON.stringify({
-      ...input,
-      hookEventName
-    }))
-  })
-
-  const stdout = Buffer.concat(stdoutChunks).toString('utf-8').trim()
-  const stderr = Buffer.concat(stderrChunks).toString('utf-8').trim()
-
-  if (exitCode !== 0) {
-    throw new Error(`Failed to call hook: process exited with code ${exitCode}${stderr ? ` - ${stderr}` : ''}`)
-  }
-
-  if (stdout === '') {
-    return { continue: true } as HookOutputs[K]
-  }
-
-  try {
-    return JSON.parse(stdout) as HookOutputs[K]
-  } catch (error) {
-    throw new Error(`Failed to parse hook output: ${stdout}${stderr ? `\nstderr: ${stderr}` : ''}`, { cause: error })
-  }
-}