codex_usage.go RawHistoryFolder

1package agent
2
3import (
4	"context"
5	"fmt"
6	"strings"
7
8	"github.com/bnema/zut/packages/agent/modes"
9	"github.com/bnema/zut/packages/core"
10	"github.com/bnema/zut/packages/provider"
11)
12
13func (r Resolved) newInteractiveAgent() *core.Agent {
14	ag := r.NewAgent()
15	modes.BindCodexWeeklyUsageFetcher(ag, r.codexWeeklyUsageFetcher())
16	return ag
17}
18
19// Build the optional meter from the resolved route, not the TUI's startup auth
20// metadata. Rebuilds replace it alongside the model client (including rescue).
21func (r Resolved) codexWeeklyUsageFetcher() func(context.Context) (*provider.CodexWeeklyUsage, error) {
22	if r.Provider != "openai-codex" || r.AuthMethod != "oauth" || r.AccountID == "" ||
23		(r.BaseURL != "" && strings.TrimRight(r.BaseURL, "/") != "https://chatgpt.com/backend-api/codex/responses") {
24		return nil
25	}
26	accountID := r.AccountID
27	return func(ctx context.Context) (*provider.CodexWeeklyUsage, error) {
28		tok := loadOAuthToken("openai")
29		if tok == nil || tok.AccountID != accountID {
30			return nil, fmt.Errorf("codex usage: subscription account changed")
31		}
32		// A passive meter must not rotate credentials alongside an inference
33		// request or write them back after logout. Normal model requests/login
34		// own refresh; the next poll picks up their newly stored token.
35		if tok.Expired() {
36			return nil, fmt.Errorf("codex usage: subscription token expired")
37		}
38		return provider.FetchCodexWeeklyUsage(ctx, nil, tok.AccessToken, accountID)
39	}
40}