@ottocode/ai-sdk 0.1.6 → 0.1.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ottocode/ai-sdk",
-  "version": "0.1.6",
+  "version": "0.1.8",
   "type": "module",
   "exports": {
     ".": "./src/index.ts",

package/src/balance.ts

@@ -45,9 +45,36 @@ export async function fetchBalance(
 			request_count: number;
 			created_at?: string;
 			last_request?: string;
+			scope?: 'wallet' | 'account';
+			payg?: {
+				wallet_balance_usd: number;
+				account_balance_usd: number;
+				raw_pool_usd: number;
+				effective_spendable_usd: number;
+			};
+			limits?: {
+				enabled: boolean;
+				daily_limit_usd: number | null;
+				daily_spent_usd: number;
+				daily_remaining_usd: number | null;
+				monthly_limit_usd: number | null;
+				monthly_spent_usd: number;
+				monthly_remaining_usd: number | null;
+				cap_remaining_usd: number | null;
+			} | null;
+			subscription?: {
+				active: boolean;
+				tier_id?: string;
+				tier_name?: string;
+				credits_included?: number;
+				credits_used?: number;
+				credits_remaining?: number;
+				period_start?: string;
+				period_end?: string;
+			} | null;
 		};
 
-		return {
+		const result: BalanceResponse = {
 			walletAddress: data.wallet_address,
 			balance: data.balance_usd,
 			totalSpent: data.total_spent,
@@ -55,7 +82,49 @@ export async function fetchBalance(
 			requestCount: data.request_count,
 			createdAt: data.created_at,
 			lastRequest: data.last_request,
+			scope: data.scope,
 		};
+
+		if (data.payg) {
+			result.payg = {
+				walletBalanceUsd: data.payg.wallet_balance_usd,
+				accountBalanceUsd: data.payg.account_balance_usd,
+				rawPoolUsd: data.payg.raw_pool_usd,
+				effectiveSpendableUsd: data.payg.effective_spendable_usd,
+			};
+		}
+
+		if (data.limits !== undefined) {
+			result.limits = data.limits
+				? {
+						enabled: data.limits.enabled,
+						dailyLimitUsd: data.limits.daily_limit_usd,
+						dailySpentUsd: data.limits.daily_spent_usd,
+						dailyRemainingUsd: data.limits.daily_remaining_usd,
+						monthlyLimitUsd: data.limits.monthly_limit_usd,
+						monthlySpentUsd: data.limits.monthly_spent_usd,
+						monthlyRemainingUsd: data.limits.monthly_remaining_usd,
+						capRemainingUsd: data.limits.cap_remaining_usd,
+					}
+				: null;
+		}
+
+		if (data.subscription !== undefined) {
+			result.subscription = data.subscription
+				? {
+						active: data.subscription.active,
+						tierId: data.subscription.tier_id,
+						tierName: data.subscription.tier_name,
+						creditsIncluded: data.subscription.credits_included,
+						creditsUsed: data.subscription.credits_used,
+						creditsRemaining: data.subscription.credits_remaining,
+						periodStart: data.subscription.period_start,
+						periodEnd: data.subscription.period_end,
+					}
+				: null;
+		}
+
+		return result;
 	} catch {
 		return null;
 	}

package/src/fetch.ts

@@ -4,6 +4,7 @@ import type {
 	PaymentOptions,
 	CacheOptions,
 	BalanceUpdate,
+	FetchFunction,
 } from './types.ts';
 import { pickPaymentRequirement, handlePayment } from './payment.ts';
 import { addAnthropicCacheControl } from './cache.ts';
@@ -80,34 +81,41 @@ function wrapResponseWithBalanceSniffing(
 	}
 
 	if (!response.body) return response;
+	if (typeof TransformStream === 'undefined') return response;
+	if (typeof response.body.pipeThrough !== 'function') return response;
 
 	const onBalanceUpdate = callbacks.onBalanceUpdate;
 	let partial = '';
 	const decoder = new TextDecoder();
-	const transform = new TransformStream<Uint8Array, Uint8Array>({
-		transform(chunk, controller) {
-			controller.enqueue(chunk);
-			partial += decoder.decode(chunk, { stream: true });
-			let nlIndex = partial.indexOf('\n');
-			while (nlIndex !== -1) {
-				const line = partial.slice(0, nlIndex);
-				partial = partial.slice(nlIndex + 1);
-				tryParseSetuComment(line, onBalanceUpdate);
-				nlIndex = partial.indexOf('\n');
-			}
-		},
-		flush() {
-			if (partial.trim()) {
-				tryParseSetuComment(partial, onBalanceUpdate);
-			}
-		},
-	});
 
-	return new Response(response.body.pipeThrough(transform), {
-		status: response.status,
-		statusText: response.statusText,
-		headers: response.headers,
-	});
+	try {
+		const transform = new TransformStream<Uint8Array, Uint8Array>({
+			transform(chunk, controller) {
+				controller.enqueue(chunk);
+				partial += decoder.decode(chunk, { stream: true });
+				let nlIndex = partial.indexOf('\n');
+				while (nlIndex !== -1) {
+					const line = partial.slice(0, nlIndex);
+					partial = partial.slice(nlIndex + 1);
+					tryParseSetuComment(line, onBalanceUpdate);
+					nlIndex = partial.indexOf('\n');
+				}
+			},
+			flush() {
+				if (partial.trim()) {
+					tryParseSetuComment(partial, onBalanceUpdate);
+				}
+			},
+		});
+
+		return new Response(response.body.pipeThrough(transform), {
+			status: response.status,
+			statusText: response.statusText,
+			headers: response.headers,
+		});
+	} catch {
+		return response;
+	}
 }
 
 async function getWalletUsdcBalance(
@@ -154,6 +162,7 @@ async function getWalletUsdcBalance(
 export interface CreateSetuFetchOptions {
 	wallet: WalletContext;
 	baseURL: string;
+	fetch?: FetchFunction;
 	rpcURL?: string;
 	callbacks?: PaymentCallbacks;
 	cache?: CacheOptions;
@@ -164,6 +173,7 @@ export function createSetuFetch(options: CreateSetuFetchOptions) {
 	const {
 		wallet,
 		baseURL,
+		fetch: customFetch,
 		rpcURL = DEFAULT_RPC_URL,
 		callbacks = {},
 		cache,
@@ -175,7 +185,7 @@ export function createSetuFetch(options: CreateSetuFetchOptions) {
 		payment?.maxPaymentAttempts ?? DEFAULT_MAX_PAYMENT_ATTEMPTS;
 	const topupApprovalMode = payment?.topupApprovalMode ?? 'auto';
 	const autoPayThresholdUsd = payment?.autoPayThresholdUsd ?? 0;
-	const baseFetch = globalThis.fetch.bind(globalThis);
+	const baseFetch = customFetch ?? globalThis.fetch.bind(globalThis);
 
 	return async (
 		input: Parameters<typeof fetch>[0],

package/src/payment.ts

@@ -7,6 +7,7 @@ import type {
 	ExactPaymentRequirement,
 	PaymentPayload,
 	PaymentCallbacks,
+	FetchFunction,
 } from './types.ts';
 import {
 	address,
@@ -136,7 +137,7 @@ export async function processSinglePayment(args: {
 	wallet: WalletContext;
 	rpcURL: string;
 	baseURL: string;
-	baseFetch: typeof fetch;
+	baseFetch: FetchFunction;
 	callbacks: PaymentCallbacks;
 }): Promise<{ attempts: number; balance?: number | string }> {
 	args.callbacks.onPaymentSigning?.();
@@ -208,7 +209,7 @@ export async function handlePayment(args: {
 	wallet: WalletContext;
 	rpcURL: string;
 	baseURL: string;
-	baseFetch: typeof fetch;
+	baseFetch: FetchFunction;
 	maxAttempts: number;
 	callbacks: PaymentCallbacks;
 }): Promise<{ attemptsUsed: number }> {

package/src/setu.ts

@@ -62,6 +62,7 @@ export function createSetu(config: SetuConfig): SetuInstance {
 	const setuFetch = createSetuFetch({
 		wallet,
 		baseURL,
+		fetch: config.fetch,
 		rpcURL: config.rpcURL,
 		callbacks: config.callbacks,
 		cache: config.cache,

package/src/types.ts

@@ -91,6 +91,7 @@ export interface PaymentOptions {
 export interface SetuConfig {
 	auth: SetuAuth;
 	baseURL?: string;
+	fetch?: FetchFunction;
 	rpcURL?: string;
 	providers?: ProviderConfig[];
 	modelMap?: Record<string, ProviderId>;
@@ -127,6 +128,33 @@ export interface BalanceResponse {
 	requestCount: number;
 	createdAt?: string;
 	lastRequest?: string;
+	scope?: 'wallet' | 'account';
+	payg?: {
+		walletBalanceUsd: number;
+		accountBalanceUsd: number;
+		rawPoolUsd: number;
+		effectiveSpendableUsd: number;
+	};
+	limits?: {
+		enabled: boolean;
+		dailyLimitUsd: number | null;
+		dailySpentUsd: number;
+		dailyRemainingUsd: number | null;
+		monthlyLimitUsd: number | null;
+		monthlySpentUsd: number;
+		monthlyRemainingUsd: number | null;
+		capRemainingUsd: number | null;
+	} | null;
+	subscription?: {
+		active: boolean;
+		tierId?: string;
+		tierName?: string;
+		creditsIncluded?: number;
+		creditsUsed?: number;
+		creditsRemaining?: number;
+		periodStart?: string;
+		periodEnd?: string;
+	} | null;
 }
 
 export interface WalletUsdcBalance {