velobase-billing 0.1.9

Velobase Billing SDK for Python

velobase-billing

Official Velobase Billing SDK for Python.

• Synchronous and asynchronous clients
• Type-safe with Pydantic v2 models and py.typed support
• Automatic retries with exponential backoff
• Python 3.9+

Installation

pip install velobase-billing

Quick Start

import uuid
from velobase_billing import Velobase

vb = Velobase(api_key="vb_live_xxx")

# 1. Deposit credits to a customer (creates the customer if new)
deposit = vb.customers.deposit(customer_id="user_123", amount=1000)

# 2. Check balance
customer = vb.customers.get("user_123")
print(customer.balance.available)  # 1000.0

# 3. Generate transaction_id once and persist before freezing
transaction_id = f"user_123_{uuid.uuid4().hex}"

# 4. Freeze credits before doing work
freeze = vb.billing.freeze(
    customer_id="user_123",
    amount=50,
    transaction_id=transaction_id,
)

# 5a. Job succeeded — consume (supports partial)
consume = vb.billing.consume(
    transaction_id=transaction_id,
    actual_amount=32,  # only charge 32, return 18
)

# 5b. Or if the job failed — unfreeze to return all
unfreeze = vb.billing.unfreeze(transaction_id=transaction_id)

How It Works

Velobase Billing uses a freeze-then-consume pattern to safely manage credits:

deposit -> freeze -> consume   (normal flow)
                  -> unfreeze  (failure/cancellation)

It also supports a direct deduct pattern for immediate deduction without freezing:

deposit -> deduct  (immediate deduction)

1. Deposit — Add credits to a customer's account. Creates the customer automatically on first deposit. Supports credit_type to specify the credit category, and starts_at/expires_at for time-limited credits.
2. Freeze — Pre-authorize an amount before performing work. The frozen credits are deducted from available but not yet used. Each freeze is identified by a unique transaction_id you provide. Supports credit_types to freeze from specific credit categories.
3. Consume — After the work is done, settle the frozen amount. You can pass actual_amount to charge less than what was frozen; the difference is automatically returned.
4. Unfreeze — If the work fails or is cancelled, release the full frozen amount back to the customer.
5. Deduct — Directly deduct credits from a customer without freezing first. Useful for immediate charges. Supports credit_types to deduct from specific credit categories.

All write operations are idempotent — repeating the same transaction_id (freeze/consume/unfreeze/deduct) or idempotency_key (deposit) returns the original result without double-charging.

Async Usage

Every method is available in an async version via AsyncVelobase:

from velobase_billing import AsyncVelobase

async def main():
    vb = AsyncVelobase(api_key="vb_live_xxx")

    deposit = await vb.customers.deposit(customer_id="user_123", amount=1000)
    customer = await vb.customers.get("user_123")
    print(customer.balance.available)

    await vb.close()

Context Manager

Both clients support context managers to ensure the underlying HTTP connection is closed:

# Sync
with Velobase(api_key="vb_live_xxx") as vb:
    customer = vb.customers.get("user_123")

# Async
async with AsyncVelobase(api_key="vb_live_xxx") as vb:
    customer = await vb.customers.get("user_123")

Configuration

vb = Velobase(
    api_key="vb_live_xxx",               # Required. Your Velobase API key.
    base_url="https://api.velobase.io",   # Optional. Override the API endpoint.
    timeout=30.0,                         # Optional. Request timeout in seconds (default: 30).
    max_retries=2,                        # Optional. Retry count on 5xx/network errors (default: 2).
)

Usage Examples

Deposit with idempotency

# Safe to retry — the second call returns the same result without double-charging
result = vb.customers.deposit(
    customer_id="user_123",
    amount=500,
    idempotency_key="order_abc_payment",
    description="Purchase of 500 credits",
)

print(result.added_amount)         # 500.0
print(result.is_idempotent_replay) # False on first call, True on retries

Deposit with credit type and expiry

result = vb.customers.deposit(
    customer_id="user_123",
    amount=1000,
    credit_type="BONUS",
    starts_at="2025-01-01T00:00:00Z",
    expires_at="2025-12-31T23:59:59Z",
    description="Annual bonus credits",
)
print(result.credit_type)   # "BONUS"
print(result.starts_at)     # "2025-01-01T00:00:00.000Z"
print(result.expires_at)    # "2025-12-31T23:59:59.000Z"

Deposit with customer metadata

result = vb.customers.deposit(
    customer_id="user_123",
    amount=1000,
    name="Alice",
    email="[e-mail]",
    metadata={"plan": "pro", "source": "stripe"},
)

Full billing flow (freeze-then-consume)

CUSTOMER = "user_123"
JOB_ID = "video_gen_001"

# Check balance before starting
before = vb.customers.get(CUSTOMER)
print(f"Available: {before.balance.available}")

# Freeze the estimated cost
vb.billing.freeze(
    customer_id=CUSTOMER,
    amount=100,
    transaction_id=JOB_ID,
    business_type="TASK",
    description="1080p video, ~60s",
)

# ... do the work ...

# Settle with the actual cost (partial consumption)
result = vb.billing.consume(
    transaction_id=JOB_ID,
    actual_amount=73,
)
print(f"Charged: {result.consumed_amount}")   # 73.0
print(f"Returned: {result.returned_amount}")  # 27.0

# Verify final balance
after = vb.customers.get(CUSTOMER)
print(f"Available: {after.balance.available}")

Direct deduct (without freezing)

CUSTOMER = "user_123"
TXN_ID = "api_call_001"

result = vb.billing.deduct(
    customer_id=CUSTOMER,
    amount=5,
    transaction_id=TXN_ID,
    business_type="TASK",
    description="API call charge",
)
print(f"Deducted: {result.deducted_amount}")  # 5.0
print(f"At: {result.deducted_at}")

Freeze with credit_types filter

# Only freeze from specific credit categories
freeze = vb.billing.freeze(
    customer_id="user_123",
    amount=50,
    transaction_id="job_abc",
    credit_types=["BONUS", "DEFAULT"],
)

Customer balance structure

customer = vb.customers.get("user_123")

# Aggregate balance across all accounts
customer.balance.total      # total deposited
customer.balance.used       # total consumed
customer.balance.frozen     # currently frozen (pending)
customer.balance.available  # total - used - frozen

# Individual accounts (e.g., different credit types/expiry)
for account in customer.accounts:
    print(account.account_type)  # "CREDIT"
    print(account.credit_type)   # "DEFAULT", "BONUS", etc.
    print(account.available)
    print(account.starts_at)     # None or ISO date string
    print(account.expires_at)    # None or ISO date string

API Reference

vb.customers.deposit(...) -> DepositResponse

Deposit credits. Creates the customer if they don't exist.

Parameter	Type	Required	Description
customer_id	str	Yes	Your unique customer identifier
amount	float	Yes	Amount to deposit (must be > 0)
credit_type	str	No	Credit category (e.g. "DEFAULT", "BONUS"). Defaults to "DEFAULT" on server.
starts_at	str	No	ISO datetime string. When the credits become active.
expires_at	str	No	ISO datetime string. When the credits expire. Must be after starts_at.
idempotency_key	str	No	Prevents duplicate deposits on retry
name	str	No	Customer display name
email	str	No	Customer email
metadata	dict	No	Arbitrary key-value metadata
description	str	No	Description for the deposit

Returns: DepositResponse with fields customer_id, account_id, credit_type, total_amount, added_amount, starts_at, expires_at, record_id, is_idempotent_replay

vb.customers.get(customer_id) -> CustomerResponse

Retrieve a customer's balance and account details.

Returns: CustomerResponse with fields id, name, email, metadata, balance, accounts, created_at

vb.billing.freeze(...) -> FreezeResponse

Freeze credits before performing work.

Parameter	Type	Required	Description
customer_id	str	Yes	Customer identifier
amount	float	Yes	Amount to freeze (must be > 0)
transaction_id	str	Yes	Your unique ID for this operation (idempotency key)
credit_types	list[str]	No	Restrict freeze to specific credit categories
business_type	BusinessType	No	Business category. See business_type for accepted values.
description	str	No	Human-readable description

Returns: FreezeResponse with fields transaction_id, frozen_amount, freeze_details, is_idempotent_replay

vb.billing.consume(...) -> ConsumeResponse

Settle a frozen amount. Supports partial consumption.

Parameter	Type	Required	Description
transaction_id	str	Yes	The transaction_id from the freeze
actual_amount	float	No	Actual amount to charge. Defaults to full frozen amount.

Returns: ConsumeResponse with fields transaction_id, consumed_amount, returned_amount, consume_details, consumed_at, is_idempotent_replay

vb.billing.unfreeze(...) -> UnfreezeResponse

Release a frozen amount back to the customer.

Parameter	Type	Required	Description
transaction_id	str	Yes	The transaction_id from the freeze

Returns: UnfreezeResponse with fields transaction_id, unfrozen_amount, unfreeze_details, unfrozen_at, is_idempotent_replay

vb.billing.deduct(...) -> DeductResponse

Directly deduct credits without freezing first. Useful for immediate charges.

Parameter	Type	Required	Description
customer_id	str	Yes	Customer identifier
amount	float	Yes	Amount to deduct (must be > 0)
transaction_id	str	Yes	Your unique ID for this operation (idempotency key)
credit_types	list[str]	No	Restrict deduction to specific credit categories
business_type	BusinessType	No	Business category. See business_type for accepted values.
description	str	No	Human-readable description

Returns: DeductResponse with fields transaction_id, deducted_amount, deduct_details, deducted_at, is_idempotent_replay

transaction_id

transaction_id uniquely identifies one billing operation (freeze → consume/unfreeze cycle, or a single deduct) and acts as its idempotency key. The server uses it to prevent double-charging on retries.

Recommended format: {customer_id}_{uuid}

import uuid

# Generate once per billing operation, then persist it
transaction_id = f"{customer_id}_{uuid.uuid4().hex}"
# e.g. "user_123_a3f8c21d4e0b4a9f8c1d2e3f4a5b6c7d"

Rules:

• Generate once and store — create the UUID before calling freeze() or deduct(), save it to your database, and reuse the same value on retries
• Never regenerate at the call site — calling uuid.uuid4() inside freeze() produces a different ID on every attempt, breaking idempotency
• Unique within your project — two different billing operations must not share the same transaction_id

# Wrong — new UUID on every call, idempotency broken on retry
vb.billing.freeze(
    customer_id=customer_id,
    amount=50,
    transaction_id=f"{customer_id}_{uuid.uuid4().hex}",  # ❌ regenerated each time
)

# Correct — UUID generated once and persisted before calling freeze
transaction_id = db.get_or_create_transaction_id(operation_id, customer_id)
# e.g. db.get_or_create_transaction_id returns an existing ID or
#      stores f"{customer_id}_{uuid.uuid4().hex}" on first call

vb.billing.freeze(customer_id=customer_id, amount=50, transaction_id=transaction_id)
# Safe to retry — same transaction_id returns the original result
vb.billing.freeze(customer_id=customer_id, amount=50, transaction_id=transaction_id)

business_type

business_type is an optional parameter on freeze() and deduct() that categorises the billing operation for analytics and reconciliation. The SDK validates the value client-side and raises ValueError before sending any network request.

Accepted values:

Value	Description
"UNDEFINED"	Default / unclassified (server default when omitted)
"TASK"	Async task execution (e.g. video generation, image processing)
"ORDER"	One-time purchase or order fulfilment
"MEMBERSHIP"	Membership plan credit grant
"SUBSCRIPTION"	Subscription renewal credit grant
"FREE_TRIAL"	Free-trial credit grant
"ADMIN_GRANT"	Manually granted credits by an admin

from velobase_billing import Velobase, BusinessType

vb = Velobase(api_key="vb_live_xxx")

vb.billing.freeze(
    customer_id="user_123",
    amount=50,
    transaction_id="job_abc",
    business_type="TASK",         # ✅ IDE autocomplete + client-side validation
)

vb.billing.freeze(
    customer_id="user_123",
    amount=50,
    transaction_id="job_abc",
    business_type="INVALID_VAL",  # ❌ raises ValueError before making a network call
)

Error Handling

All API errors raise typed exceptions that inherit from VelobaseError:

from velobase_billing import VelobaseError, ValidationError, AuthenticationError, NotFoundError

try:
    vb.billing.freeze(
        customer_id="user_123",
        amount=999999,
        transaction_id="job_xyz",
    )
except ValidationError as e:
    # 400 — bad request or insufficient balance
    print(e.message)  # "insufficient balance"
except AuthenticationError as e:
    # 401 — invalid or missing API key
    pass
except NotFoundError as e:
    # 404 — customer not found
    pass
except VelobaseError as e:
    # catch-all for other API errors
    print(e.status, e.type, e.message)

Exception	HTTP Status	When
AuthenticationError	401	Invalid or missing API key
ValidationError	400	Bad params, insufficient balance
NotFoundError	404	Customer or resource not found
ConflictError	409	Conflicting operation
InternalError	500	Server-side error (auto-retried)

Retries

The SDK automatically retries on 5xx errors and network failures with exponential backoff (0.5s, 1s, 2s..., capped at 5s). Retries are safe because all Velobase write operations are idempotent.

4xx errors (validation, auth, not found) are never retried.

License

MIT