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

src/utils/providerFlag.test.ts

@@ -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()
+  })
+})

src/utils/providerFlag.ts

@@ -35,6 +35,18 @@ export function parseProviderFlag(args: string[]): string | null {
   return value
 }
 
+/**
+ * Parse and apply --provider from argv in one step.
+ * Returns undefined when the flag is absent.
+ */
+export function applyProviderFlagFromArgs(
+  args: string[],
+): { error?: string } | undefined {
+  const provider = parseProviderFlag(args)
+  if (!provider) return undefined
+  return applyProviderFlag(provider, args)
+}
+
 /**
  * Extract the value of --model from argv.
  * Returns null if absent.