index.ts 5.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157
  1. /**
  2. * Profile-named Claude Code one-shot subagent provider. Every accepted run
  3. * invokes the official Agent SDK in the delegating Session's workspace and
  4. * places the SDK-spawned real CLI under the shared subprocess owner.
  5. *
  6. * @module @deepseek-ai/dsh-subagent-claude-code
  7. */
  8. import type { Context } from '@deepseek-ai/cordis'
  9. import z from '@deepseek-ai/schemastery'
  10. import { MAX_TIMER_DELAY_MS } from '@deepseek-ai/dsh-timeout'
  11. import {
  12. assertPositiveFinite,
  13. NO_START_CAPABILITIES,
  14. resolveChildCwd,
  15. type ResolvedSubagentStartRequest,
  16. type SubagentCapabilities,
  17. type SubagentProvider,
  18. } from '@deepseek-ai/dsh-subagent'
  19. import {
  20. CLAUDE_CODE_PERMISSION_MODES,
  21. DEFAULT_CLAUDE_CODE_PERMISSION_MODE,
  22. DEFAULT_DISPOSE_GRACE_MS,
  23. claudeCodeStartupFailure,
  24. startClaudeCodeRun,
  25. type ClaudeCodePermissionMode,
  26. type ClaudeCodeRunSpec,
  27. } from './run.ts'
  28. export const name = 'subagent-claude-code'
  29. export const inject = ['subagents', 'subprocess']
  30. const DEFAULT_PROVIDER_NAME = 'claude-code'
  31. /* jscpd:ignore-start -- sibling product providers intentionally expose
  32. * overlapping deployment-owned fields without adding a shared config owner. */
  33. /** Deployment-owned model, permission, environment, and process-release settings. */
  34. export interface Config {
  35. /** Provider name on `ctx.subagents` (default `claude-code`). */
  36. providerName?: string
  37. /** Native Claude model fixed for this instance; omitted to inherit Claude settings. */
  38. model?: string
  39. /**
  40. * Explicit environment entries layered over the subprocess seam's
  41. * credential-scrubbed parent environment.
  42. */
  43. env?: Record<string, string>
  44. /**
  45. * Native non-interactive mode fixed for this Provider instance. Defaults to
  46. * `dontAsk`; `acceptEdits` accepts edits, `auto` uses the native classifier,
  47. * `plan` returns a plan without approving execution, and
  48. * `bypassPermissions` explicitly skips permission checks.
  49. */
  50. permissionMode?: ClaudeCodePermissionMode
  51. /** Grace in milliseconds between Claude Code managed-range termination tiers. */
  52. disposeGraceMs?: number
  53. }
  54. export const Config: z<Config> = z.object({
  55. providerName: z.string().min(1).default(DEFAULT_PROVIDER_NAME),
  56. model: z.string().min(1),
  57. env: z.dict(z.string()).default({}),
  58. permissionMode: z.union([...CLAUDE_CODE_PERMISSION_MODES])
  59. .default(DEFAULT_CLAUDE_CODE_PERMISSION_MODE),
  60. disposeGraceMs: z.number().default(DEFAULT_DISPOSE_GRACE_MS),
  61. })
  62. type ResolvedConfig = Omit<Required<Config>, 'model'> & Pick<Config, 'model'>
  63. /* jscpd:ignore-end */
  64. /* jscpd:ignore-start -- Cordis registration and shared-seam plumbing mirror
  65. * the Codex sibling; each product's lifecycle remains package-private. */
  66. class ClaudeCodeProvider implements SubagentProvider {
  67. readonly capabilities: SubagentCapabilities = NO_START_CAPABILITIES
  68. readonly inheritsParentContext = false
  69. constructor(
  70. readonly name: string,
  71. private readonly ctx: Context,
  72. private readonly config: ResolvedConfig,
  73. ) {}
  74. async start(request: ResolvedSubagentStartRequest) {
  75. const parentCwd = request.parent.session.header.cwd
  76. if (parentCwd === undefined) {
  77. throw new Error(
  78. 'subagent-claude-code: no working directory for the child — delegate from a parent session that has one',
  79. )
  80. }
  81. let cwd: string
  82. try {
  83. cwd = resolveChildCwd(
  84. 'subagent-claude-code',
  85. undefined,
  86. parentCwd,
  87. )
  88. } catch (error: unknown) {
  89. if (request.signal.aborted) {
  90. throw new Error(
  91. 'subagent-claude-code: request was aborted before SDK startup',
  92. )
  93. }
  94. const failure = claudeCodeStartupFailure(error)
  95. this.ctx.logger.warn(
  96. `subagent-claude-code "${this.name}": child start failed: %o`,
  97. failure,
  98. )
  99. throw failure
  100. }
  101. const spec: ClaudeCodeRunSpec = {
  102. cwd,
  103. ...this.config.model === undefined ? {} : { model: this.config.model },
  104. permissionMode: this.config.permissionMode,
  105. env: this.config.env,
  106. disposeGraceMs: this.config.disposeGraceMs,
  107. spawn: spawnSpec => this.ctx.subprocess.spawn(spawnSpec),
  108. onError: (error, stopReason) => {
  109. this.ctx.logger.warn(
  110. `subagent-claude-code "${this.name}": child run failed (${stopReason}): %o`,
  111. error,
  112. )
  113. },
  114. }
  115. return startClaudeCodeRun(request, spec)
  116. }
  117. }
  118. /**
  119. * Register one Profile-named Claude Code provider.
  120. * @param ctx - context carrying shared subagent and subprocess services.
  121. * @param config - registry name, optional model, permission mode, child environment, and disposal grace.
  122. */
  123. export function apply(ctx: Context, config: Config): void {
  124. const resolved: ResolvedConfig = {
  125. providerName: config.providerName ?? DEFAULT_PROVIDER_NAME,
  126. ...config.model === undefined ? {} : { model: config.model },
  127. env: config.env as Record<string, string>,
  128. permissionMode: config.permissionMode ?? DEFAULT_CLAUDE_CODE_PERMISSION_MODE,
  129. disposeGraceMs: config.disposeGraceMs as number,
  130. }
  131. assertPositiveFinite(
  132. 'subagent-claude-code',
  133. 'disposeGraceMs',
  134. resolved.disposeGraceMs,
  135. )
  136. if (resolved.disposeGraceMs > MAX_TIMER_DELAY_MS) {
  137. throw new Error(
  138. `subagent-claude-code: disposeGraceMs must be no greater than ${MAX_TIMER_DELAY_MS}`,
  139. )
  140. }
  141. ctx.subagents.registerProvider(new ClaudeCodeProvider(
  142. resolved.providerName,
  143. ctx,
  144. resolved,
  145. ))
  146. }
  147. /* jscpd:ignore-end */