diff --git a/cmd/picoclaw/internal/auth/helpers.go b/cmd/picoclaw/internal/auth/helpers.go index e3db7f333..271859128 100644 --- a/cmd/picoclaw/internal/auth/helpers.go +++ b/cmd/picoclaw/internal/auth/helpers.go @@ -16,8 +16,11 @@ import ( "github.com/sipeed/picoclaw/pkg/providers" ) -const supportedProvidersMsg = "supported providers: openai, anthropic, google-antigravity" -const defaultAnthropicModel = "claude-sonnet-4.6" +const ( + supportedProvidersMsg = "supported providers: openai, anthropic, google-antigravity" + defaultAnthropicModel = "claude-sonnet-4.6" +) + func authLoginCmd(provider string, useDeviceCode bool, useOauth bool) error { switch provider { diff --git a/cmd/picoclaw/internal/auth/login.go b/cmd/picoclaw/internal/auth/login.go index 223823489..afbe098aa 100644 --- a/cmd/picoclaw/internal/auth/login.go +++ b/cmd/picoclaw/internal/auth/login.go @@ -20,7 +20,10 @@ func newLoginCommand() *cobra.Command { cmd.Flags().StringVarP(&provider, "provider", "p", "", "Provider to login with (openai, anthropic)") cmd.Flags().BoolVar(&useDeviceCode, "device-code", false, "Use device code flow (for headless environments)") - cmd.Flags().BoolVar(&useOauth, "setup-token", false, "Use setup-token flow for Anthropic (from `claude setup-token`)") + cmd.Flags().BoolVar( + &useOauth, "setup-token", false, + "Use setup-token flow for Anthropic (from `claude setup-token`)", + ) _ = cmd.MarkFlagRequired("provider") return cmd diff --git a/pkg/auth/anthropic_usage.go b/pkg/auth/anthropic_usage.go index 3fffc3aa8..716b2908e 100644 --- a/pkg/auth/anthropic_usage.go +++ b/pkg/auth/anthropic_usage.go @@ -30,8 +30,8 @@ func FetchAnthropicUsage(token string) (*AnthropicUsage, error) { return nil, err } req.Header.Set("Authorization", "Bearer "+token) - req.Header.Set("anthropic-version", anthropicAPIVersion) - req.Header.Set("anthropic-beta", anthropicBetaHeader) + req.Header.Set("Anthropic-Version", anthropicAPIVersion) + req.Header.Set("Anthropic-Beta", anthropicBetaHeader) client := &http.Client{Timeout: 10 * time.Second} resp, err := client.Do(req) diff --git a/pkg/auth/anthropic_usage_test.go b/pkg/auth/anthropic_usage_test.go index 2c11ac05a..ef4a35364 100644 --- a/pkg/auth/anthropic_usage_test.go +++ b/pkg/auth/anthropic_usage_test.go @@ -12,8 +12,8 @@ func TestFetchAnthropicUsage_Success(t *testing.T) { if got := r.Header.Get("Authorization"); got != "Bearer test-token" { t.Errorf("Authorization = %q, want %q", got, "Bearer test-token") } - if got := r.Header.Get("anthropic-beta"); got != anthropicBetaHeader { - t.Errorf("anthropic-beta = %q, want %q", got, anthropicBetaHeader) + if got := r.Header.Get("Anthropic-Beta"); got != anthropicBetaHeader { + t.Errorf("Anthropic-Beta = %q, want %q", got, anthropicBetaHeader) } w.WriteHeader(http.StatusOK) w.Write([]byte(`{"five_hour":{"utilization":0.42},"seven_day":{"utilization":0.85}}`))