Add Venice API balance provider by clawSean · Pull Request #865 · steipete/CodexBar

clawSean · 2026-05-08T06:41:18Z

Add Venice API balance provider

Summary

This PR adds Venice as a first-class CodexBar provider, enabling users to monitor their Venice API balance (DIEM or USD) directly from the menu bar.

Scope: Stable balance endpoint only (GET /api/v1/billing/balance). No beta history or usage-analytics endpoints in this PR.

Changes

Core provider setup

Added .venice case to UsageProvider and IconStyle enums
Created VeniceProviderDescriptor with macro registration and fetch strategy
Created VeniceSettingsReader for env key resolution (VENICE_API_KEY / VENICE_KEY)

API integration

Implemented VeniceUsageFetcher for /api/v1/billing/balance endpoint with bearer auth (15s timeout)
Decoded response: canConsume, consumptionCurrency, balances.diem, balances.usd, diemEpochAllocation
Honest balance display: no fake quota percentages unless DIEM epoch allocation is present
Balance display copy: "DIEM 90.50 / 100.00 epoch allocation" or "$25.00 USD remaining" or "Balance unavailable"

Token & settings integration

Added veniceToken() and veniceResolution() to ProviderTokenResolver
Added Venice to TokenAccountSupportCatalog+Data with environment injection
Added veniceUsage log category
Updated ProviderConfigEnvironment for API key injection

App-side implementation

Created VeniceProviderImplementation following DeepSeek pattern (available when key exists or token accounts exist)

Testing

VeniceSettingsReaderTests: 10 tests covering env key resolution, quoting, trimming, and precedence
VeniceUsageFetcherTests: 19 tests covering DIEM/USD parsing, epoch allocation, canConsume=false, edge cases, and error handling

Documentation

Created docs/venice.md with setup instructions, balance types, and troubleshooting
Updated README.md to include Venice in provider list
Updated docs/providers.md with Venice details and API endpoint info

Implementation notes

Provider identity is siloed to Venice (no sharing with OpenRouter/Codex/others)
Balance display uses RateWindow with honest remaining amounts (prefers DIEM with allocation percentage when available; USD or DIEM remaining otherwise)
No real API keys, account data, screenshots, or local paths included
Follows CodexBar AGENTS.md contribution guidelines and existing provider patterns (DeepSeek/OpenRouter)

Testing

Run locally (on macOS):

swift test --filter Venice
swift test --filter ProviderRegistry
swift test
pnpm check
swiftformat Sources Tests
swiftlint --strict

Dashboard & Setup

Venice API settings: https://venice.ai/settings/api
Env keys: VENICE_API_KEY (primary), VENICE_KEY (fallback)
Endpoint: https://api.venice.ai/api/v1/billing/balance (stable, v1)

chatgpt-codex-connector

💡 Codex Review

Here are some automated review suggestions for this pull request.

Reviewed commit: b14d485f73

ℹ️ About Codex in GitHub

Codex has been enabled to automatically review pull requests in this repo. Reviews are triggered when you

Open a pull request for review
Mark a draft as ready
Comment "@codex review".

If Codex has suggestions, it will comment; otherwise it will react with 👍.

When you sign up for Codex through ChatGPT, Codex can also answer questions or update the PR, like "@codex address that feedback".

chatgpt-codex-connector · 2026-05-08T06:43:45Z

    case mistral
    case deepseek
    case codebuff
+    case venice


Update exhaustive provider switches for Venice

Adding the .venice enum case makes every exhaustive switch UsageProvider require a Venice branch. I checked Sources/CodexBar/Providers/Shared/ProviderImplementationRegistry.swift and the widget switches in Sources/CodexBarWidget/CodexBarWidgetProvider.swift / CodexBarWidgetViews.swift; they still stop at .codebuff with no default, so the app/widget targets will fail to compile as soon as this case is introduced.

Useful? React with 👍 / 👎.

chatgpt-codex-connector · 2026-05-08T06:43:45Z

+        if !self.canConsume {
+            balanceDetail = "Balance unavailable for API calls"
+            usedPercent = 100
+        } else if let diem = self.diemBalance, let allocation = self.diemEpochAllocation, allocation > 0 {


Use the active Venice consumption currency

When the balance response contains both DIEM and USD but consumptionCurrency is USD, this branch still reports DIEM first and can show DIEM allocation progress even though Venice says API calls will consume USD. Since the API includes consumptionCurrency specifically to identify the active billing currency, users with both balances can see the wrong remaining balance in the menu bar.

Useful? React with 👍 / 👎.

clawSean · 2026-05-08T07:08:55Z

Clean follow-up (previous comment had shell-escaped markdown issues):

Addressed both review items in commit 1c7c3d2.

Exhaustive switch coverage for new Venice provider:

Sources/CodexBar/Providers/Shared/ProviderImplementationRegistry.swift
Sources/CodexBarWidget/CodexBarWidgetProvider.swift
Sources/CodexBarWidget/CodexBarWidgetViews.swift (short label + widget color switch)

Active billing currency display:

Sources/CodexBarCore/Providers/Venice/VeniceUsageFetcher.swift now honors consumptionCurrency when both DIEM and USD balances are present.
Added regression test in Tests/CodexBarTests/VeniceUsageFetcherTests.swift:
uses USD display when consumptionCurrency is USD and both balances exist

clawSean · 2026-05-08T17:15:08Z

@codex review

chatgpt-codex-connector

💡 Codex Review

Here are some automated review suggestions for this pull request.

Reviewed commit: 1c7c3d25ee

ℹ️ About Codex in GitHub

Codex has been enabled to automatically review pull requests in this repo. Reviews are triggered when you

Open a pull request for review
Mark a draft as ready
Comment "@codex review".

If Codex has suggestions, it will comment; otherwise it will react with 👍.

When you sign up for Codex through ChatGPT, Codex can also answer questions or update the PR, like "@codex address that feedback".

chatgpt-codex-connector · 2026-05-08T17:18:45Z

    case mistral
    case deepseek
    case codebuff
+    case venice


Add Venice to remaining exhaustive provider switches

Adding a new UsageProvider case requires updating every exhaustive switch; this commit updates several call sites but leaves others, including Sources/CodexBarCLI/TokenAccountCLI.swift:81, Sources/CodexBar/UsageStore.swift:836, and Sources/CodexBarCore/Vendored/CostUsage/CostUsageScanner.swift:225, without a .venice branch or default. As soon as the app/CLI targets are built with this enum case, those switches are non-exhaustive and the package cannot compile.

Useful? React with 👍 / 👎.

clawSean · 2026-05-08T17:25:06Z

Checked — new Codex P1 was valid. Addressed in 0b5d0e90.

Added .venice to the remaining exhaustive UsageProvider switches called out by review:

Sources/CodexBarCLI/TokenAccountCLI.swift
Sources/CodexBar/UsageStore.swift
Sources/CodexBarCore/Vendored/CostUsage/CostUsageScanner.swift

Branch is updated and pushed.

clawSean · 2026-05-08T21:22:17Z

@codex review

chatgpt-codex-connector

💡 Codex Review

Here are some automated review suggestions for this pull request.

Reviewed commit: 0b5d0e901f

ℹ️ About Codex in GitHub

Codex has been enabled to automatically review pull requests in this repo. Reviews are triggered when you

Open a pull request for review
Mark a draft as ready
Comment "@codex review".

If Codex has suggestions, it will comment; otherwise it will react with 👍.

When you sign up for Codex through ChatGPT, Codex can also answer questions or update the PR, like "@codex address that feedback".

chatgpt-codex-connector · 2026-05-08T21:24:58Z

+    private static func parseSnapshot(data: Data) throws -> VeniceUsageSnapshot {
+        let decoded: VeniceBalanceResponse
+        do {
+            decoded = try JSONDecoder().decode(VeniceBalanceResponse.self, from: data)


Accept string-encoded Venice balances

When Venice returns balance numbers as JSON strings, this strict decode into Double? fields fails and the provider reports parseFailed instead of showing the balance. Venice’s API docs note that numeric balance values may be returned as strings to avoid precision loss (https://docs.venice.ai/api-reference/api-spec), so the balance response should decode both numeric and string forms for balances and diemEpochAllocation.

Useful? React with 👍 / 👎.

clawSean · 2026-05-08T21:47:26Z

Status update: a new Codex P2 landed (Accept string-encoded Venice balances).

Fixed in 2b506fda:

VeniceBalanceResponse + VeniceBalances now decode numeric fields from either JSON number or numeric string.
Added regression test: parses string-encoded balances and allocation.

Branch pushed and ready for re-review.

clawSean · 2026-05-08T22:39:15Z

@codex review

chatgpt-codex-connector

💡 Codex Review

Here are some automated review suggestions for this pull request.

Reviewed commit: 2b506fdaf6

ℹ️ About Codex in GitHub

Codex has been enabled to automatically review pull requests in this repo. Reviews are triggered when you

Open a pull request for review
Mark a draft as ready
Comment "@codex review".

If Codex has suggestions, it will comment; otherwise it will react with 👍.

When you sign up for Codex through ChatGPT, Codex can also answer questions or update the PR, like "@codex address that feedback".

chatgpt-codex-connector · 2026-05-08T22:42:06Z

+            let usdStr = String(format: "%.2f", usd)
+            balanceDetail = "$\(usdStr) USD remaining"
+            usedPercent = 0
+        } else if activeCurrency == "DIEM", let diem = self.diemBalance, let allocation = self.diemEpochAllocation, allocation > 0 {


Handle bundled-credit DIEM allocations

For accounts where Venice returns consumptionCurrency: "BUNDLED_CREDITS" with balances.diem and diemEpochAllocation (a documented response shape), this condition is skipped because it only accepts DIEM; the code then falls through to the generic DIEM balance branch and reports usedPercent = 0 instead of the allocation progress. This makes staked/bundled-credit accounts with partial usage look completely unused in the menu bar.

Useful? React with 👍 / 👎.

clawSean · 2026-05-08T22:59:51Z

Checked now — latest Codex P2 was valid (Handle bundled-credit DIEM allocations).

Fixed in 4abdf58b:

Treat non-USD consumption currencies (including BUNDLED_CREDITS) with diemEpochAllocation as DIEM allocation progress.
Added regression test: uses DIEM allocation progress for bundled credits currency.

Branch pushed.

clawSean · 2026-05-09T00:00:46Z

@codex review

chatgpt-codex-connector · 2026-05-09T00:04:53Z

Codex Review: Didn't find any major issues. Keep it up!

ℹ️ About Codex in GitHub

Codex has been enabled to automatically review pull requests in this repo. Reviews are triggered when you

Open a pull request for review
Mark a draft as ready
Comment "@codex review".

If Codex has suggestions, it will comment; otherwise it will react with 👍.

When you sign up for Codex through ChatGPT, Codex can also answer questions or update the PR, like "@codex address that feedback".

- Add Venice as first-class UsageProvider with .venice enum case - Implement VeniceSettingsReader for VENICE_API_KEY / VENICE_KEY env resolution - Implement VeniceUsageFetcher for GET /api/v1/billing/balance endpoint - Add VeniceProviderDescriptor with macro registration and API fetch strategy - Add token account support in TokenAccountSupportCatalog+Data - Add config environment injection in ProviderConfigEnvironment - Implement VeniceProviderImplementation for menu bar integration - Add VeniceSettingsReaderTests with env key resolution coverage - Add VeniceUsageFetcherTests with DIEM/USD/epoch allocation scenarios - Display honest balance (no fake quota %) - Add docs/venice.md and update README / providers documentation API: stable /api/v1/billing/balance endpoint only (no beta history endpoints) Scope: balance display with canConsume flag and optional epoch allocation Dashboard: https://venice.ai/settings/api

…play

chatgpt-codex-connector

💡 Codex Review

Here are some automated review suggestions for this pull request.

Reviewed commit: 48f34749ab

ℹ️ About Codex in GitHub

Your team has set up Codex to review pull requests in this repo. Reviews are triggered when you

Open a pull request for review
Mark a draft as ready
Comment "@codex review".

If Codex has suggestions, it will comment; otherwise it will react with 👍.

Codex can also answer questions or update the PR. Try commenting "@codex address that feedback".

chatgpt-codex-connector · 2026-05-10T01:32:15Z

+            Self.scheduleAppGroupMigration()
+            appGroupMigration = AppGroupSupport.MigrationResult(status: .targetUnavailable)
+        }


Run app-group migration before reading settings

Calling scheduleAppGroupMigration() here makes migration asynchronous, but loadDefaultsState runs immediately afterward and reads defaults before legacy app-group keys are copied. On first launch after upgrade, users who only have legacy values (e.g. debugDisableKeychainAccess / widgetSelectedProvider) will run with stale settings until a restart, because this initializer never reloads defaults after the detached task finishes.

Useful? React with 👍 / 👎.

chatgpt-codex-connector · 2026-05-10T01:32:15Z

+        if try self.decodeNil(forKey: key) {
+            return nil
+        }


Treat omitted Venice balance fields as optional

decodeFlexibleDoubleIfPresent begins with decodeNil(forKey:), which throws keyNotFound when the key is absent, so these optional numeric fields become effectively required. If Venice omits diem, usd, or diemEpochAllocation (instead of sending explicit null), parsing fails and the provider reports parseFailed for an otherwise valid response.

Useful? React with 👍 / 👎.

steipete

Reviewed provider wiring, decoder behavior, token-account/env routing, Linux compatibility, and live app smoke. Local + CI green.

chatgpt-codex-connector Bot reviewed May 8, 2026

View reviewed changes

clawSean and others added 8 commits May 10, 2026 02:07

Fix Venice provider registration and icon asset

565a8af

Address PR feedback: exhaustive Venice switches + active currency dis…

bb6cbfb

…play

Add Venice to remaining exhaustive UsageProvider switches

415a269

Support string-encoded Venice balances in decoder

134456f

Handle bundled-credit DIEM allocation progress

3f72e9f

fix: parse Venice string balances

56d227b

fix: avoid startup defaults stalls

48f3474

steipete force-pushed the feat/venice-usage-provider branch from 4abdf58 to 48f3474 Compare May 10, 2026 01:26

chatgpt-codex-connector Bot reviewed May 10, 2026

View reviewed changes

steipete added 2 commits May 10, 2026 02:37

fix: keep cost scanner Linux-compatible

dcb4f78

docs: thank Venice provider contributor

da7ea85

steipete approved these changes May 10, 2026

View reviewed changes

steipete merged commit bf1472e into steipete:main May 10, 2026
4 checks passed

Conversation

clawSean commented May 8, 2026

Add Venice API balance provider

Summary

Changes

Core provider setup

API integration

Token & settings integration

App-side implementation

Testing

Documentation

Implementation notes

Testing

Dashboard & Setup

Uh oh!

chatgpt-codex-connector Bot left a comment

Choose a reason for hiding this comment

💡 Codex Review

Uh oh!

chatgpt-codex-connector Bot May 8, 2026

Choose a reason for hiding this comment

Uh oh!

chatgpt-codex-connector Bot May 8, 2026

Choose a reason for hiding this comment

Uh oh!

clawSean commented May 8, 2026

Uh oh!

clawSean commented May 8, 2026

Uh oh!

chatgpt-codex-connector Bot left a comment

Choose a reason for hiding this comment

💡 Codex Review

Uh oh!

chatgpt-codex-connector Bot May 8, 2026

Choose a reason for hiding this comment

Uh oh!

clawSean commented May 8, 2026

Uh oh!

clawSean commented May 8, 2026

Uh oh!

chatgpt-codex-connector Bot left a comment

Choose a reason for hiding this comment

💡 Codex Review

Uh oh!

chatgpt-codex-connector Bot May 8, 2026

Choose a reason for hiding this comment

Uh oh!

clawSean commented May 8, 2026

Uh oh!

clawSean commented May 8, 2026

Uh oh!

chatgpt-codex-connector Bot left a comment

Choose a reason for hiding this comment

💡 Codex Review

Uh oh!

chatgpt-codex-connector Bot May 8, 2026

Choose a reason for hiding this comment

Uh oh!

clawSean commented May 8, 2026

Uh oh!

clawSean commented May 9, 2026

Uh oh!

chatgpt-codex-connector Bot commented May 9, 2026

Uh oh!

chatgpt-codex-connector Bot left a comment

Choose a reason for hiding this comment

💡 Codex Review

Uh oh!

chatgpt-codex-connector Bot May 10, 2026

Choose a reason for hiding this comment

Uh oh!

chatgpt-codex-connector Bot May 10, 2026

Choose a reason for hiding this comment

Uh oh!

steipete left a comment

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Reviewers

Assignees