fix: apply provider flag before startup banner (#322)

This commit is contained in:
Vasanth T
2026-04-04 11:49:02 +05:30
committed by GitHub
parent bd4daa3ee7
commit cdbe016e6f
3 changed files with 50 additions and 17 deletions

View File

@@ -1,5 +1,10 @@
import { describe, expect, test, afterEach } from 'bun:test'
import { parseProviderFlag, applyProviderFlag, VALID_PROVIDERS } from './providerFlag.js'
import {
parseProviderFlag,
applyProviderFlag,
applyProviderFlagFromArgs,
VALID_PROVIDERS,
} from './providerFlag.js'
const originalEnv = { ...process.env }
@@ -137,3 +142,23 @@ describe('applyProviderFlag - invalid provider', () => {
expect(result.error).toContain(VALID_PROVIDERS.join(', '))
})
})
describe('applyProviderFlagFromArgs', () => {
test('applies ollama provider and model from argv in one step', () => {
const result = applyProviderFlagFromArgs([
'--provider',
'ollama',
'--model',
'qwen2.5:3b',
])
expect(result?.error).toBeUndefined()
expect(process.env.CLAUDE_CODE_USE_OPENAI).toBe('1')
expect(process.env.OPENAI_BASE_URL).toBe('http://localhost:11434/v1')
expect(process.env.OPENAI_MODEL).toBe('qwen2.5:3b')
})
test('returns undefined when --provider is absent', () => {
expect(applyProviderFlagFromArgs(['--model', 'gpt-4o'])).toBeUndefined()
})
})