Preserve Claude SDK child environment

This commit is contained in:
pku-xht
2026-08-05 01:07:58 +08:00
parent 34b6cb91ed
commit 96d6853a96
12 changed files with 72 additions and 50 deletions

View File

@@ -10,9 +10,10 @@ import type {
SpawnedProcess,
SpawnOptions,
} from '@anthropic-ai/claude-agent-sdk'
import type {
SubprocessHandle,
SubprocessSpawnSpec,
import {
scrubbedParentEnv,
type SubprocessHandle,
type SubprocessSpawnSpec,
} from '@deepseek-ai/dsh-subprocess'
function thrown(value: unknown): Error {
@@ -21,19 +22,18 @@ function thrown(value: unknown): Error {
}
/**
* Convert the SDK environment to the shared subprocess seam's defined-value
* overlay without changing the effective child environment.
* @param env - SDK-composed child environment.
* @returns entries whose values survive Node's subprocess environment.
* Encode the SDK's complete child environment as a subprocess overlay.
* @param env - SDK-composed child environment after its removals and replacements.
* @returns explicit values plus tombstones for surviving ambient names the SDK removed.
*/
export function definedEnvironment(
export function sdkEnvironmentOverlay(
env: SpawnOptions['env'],
): Record<string, string> {
const defined: Record<string, string> = {}
for (const [name, value] of Object.entries(env)) {
if (value !== undefined) defined[name] = value
): NodeJS.ProcessEnv {
const overlay: NodeJS.ProcessEnv = { ...env }
for (const name of Object.keys(scrubbedParentEnv())) {
if (!(name in env)) overlay[name] = undefined
}
return defined
return overlay
}
/**
@@ -55,7 +55,7 @@ export function claudeSpawnSpec(
stdio: { stdin: 'pipe', stdout: 'pipe', stderr: 'inherit' },
graceMs,
signal: options.signal,
env: definedEnvironment(options.env),
env: sdkEnvironmentOverlay(options.env),
}
}

View File

@@ -31,8 +31,8 @@ import * as claudeCode from '../src/index.ts'
import * as invariant from '../src/invariant.ts'
import {
claudeSpawnSpec,
definedEnvironment,
ManagedClaudeCodeProcess,
sdkEnvironmentOverlay,
} from '../src/process.ts'
import {
claudeQueryOptions,
@@ -386,6 +386,7 @@ describe('task admission and package contracts', () => {
describe('official spawn projection', () => {
it('forwards command, arguments, cwd, environment, and signal exactly', () => {
vi.stubEnv('SDK_REMOVED_AMBIENT', 'ambient-value')
const signal = new AbortController().signal
const options = sdkSpawnOptions({
command: '/official/claude',
@@ -394,15 +395,26 @@ describe('official spawn projection', () => {
env: { A: 'one', B: undefined, C: 'three' },
signal,
})
expect(definedEnvironment(options.env)).toEqual({ A: 'one', C: 'three' })
expect(claudeSpawnSpec(options, 321)).toEqual({
expect(sdkEnvironmentOverlay(options.env)).toEqual(expect.objectContaining({
A: 'one',
B: undefined,
C: 'three',
SDK_REMOVED_AMBIENT: undefined,
}))
const spawnSpec = claudeSpawnSpec(options, 321)
expect(spawnSpec).toMatchObject({
argv: ['/official/claude', '--one', 'two'],
cwd: '/parent/workspace',
stdio: { stdin: 'pipe', stdout: 'pipe', stderr: 'inherit' },
graceMs: 321,
signal,
env: { A: 'one', C: 'three' },
})
expect(spawnSpec.env).toEqual(expect.objectContaining({
A: 'one',
B: undefined,
C: 'three',
SDK_REMOVED_AMBIENT: undefined,
}))
const missingCwd = sdkSpawnOptions()
delete missingCwd.cwd
expect(() => claudeSpawnSpec(