@ottocode/server 0.1.204 → 0.1.205

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@ottocode/server",
-	"version": "0.1.204",
+	"version": "0.1.205",
 	"description": "HTTP API server for ottocode",
 	"type": "module",
 	"main": "./src/index.ts",
@@ -49,8 +49,8 @@
 		"typecheck": "tsc --noEmit"
 	},
 	"dependencies": {
-		"@ottocode/sdk": "0.1.204",
-		"@ottocode/database": "0.1.204",
+		"@ottocode/sdk": "0.1.205",
+		"@ottocode/database": "0.1.205",
 		"drizzle-orm": "^0.44.5",
 		"hono": "^4.9.9",
 		"zod": "^4.1.8"

package/src/routes/setu.ts

@@ -4,7 +4,6 @@ import {
 	getPublicKeyFromPrivate,
 	getAuth,
 	loadConfig,
-	fetchSolanaUsdcBalance,
 } from '@ottocode/sdk';
 import { logger } from '@ottocode/sdk';
 import { serializeError } from '../runtime/errors/api-error.ts';
@@ -117,18 +116,44 @@ export function registerSetuRoutes(app: Hono) {
 				return c.json({ error: 'Setu wallet not configured' }, 401);
 			}
 
-			const network =
-				(c.req.query('network') as 'mainnet' | 'devnet') || 'mainnet';
+			const publicKey = getPublicKeyFromPrivate(privateKey);
+			if (!publicKey) {
+				return c.json({ error: 'Invalid private key' }, 400);
+			}
 
-			const balance = await fetchSolanaUsdcBalance({ privateKey }, network);
-			if (!balance) {
-				return c.json(
-					{ error: 'Failed to fetch USDC balance from Solana' },
-					502,
-				);
+			const baseUrl = getSetuBaseUrl();
+			const response = await fetch(
+				`${baseUrl}/v1/wallet/${publicKey}/balances?limit=100&showNative=false&showNfts=false&showZeroBalance=false`,
+				{
+					method: 'GET',
+					headers: { 'Content-Type': 'application/json' },
+				},
+			);
+
+			if (!response.ok) {
+				return c.json({ error: 'Failed to fetch wallet balances' }, 502);
 			}
 
-			return c.json(balance);
+			const data = (await response.json()) as {
+				balances: Array<{
+					mint: string;
+					symbol: string;
+					name: string;
+					balance: number;
+					decimals: number;
+					pricePerToken: number | null;
+					usdValue: number | null;
+				}>;
+				totalUsdValue: number;
+			};
+
+			const usdcEntry = data.balances.find((b) => b.symbol === 'USDC');
+
+			return c.json({
+				walletAddress: publicKey,
+				usdcBalance: usdcEntry?.balance ?? 0,
+				network: 'mainnet' as const,
+			});
 		} catch (error) {
 			logger.error('Failed to fetch USDC balance', error);
 			const errorResponse = serializeError(error);

package/src/runtime/provider/index.ts

@@ -21,6 +21,7 @@ export async function resolveModel(
 		sessionId?: string;
 		messageId?: string;
 		topupApprovalMode?: ResolveSetuModelOptions['topupApprovalMode'];
+		autoPayThresholdUsd?: ResolveSetuModelOptions['autoPayThresholdUsd'];
 	},
 ) {
 	if (provider === 'openai') {
@@ -46,6 +47,7 @@ export async function resolveModel(
 		return await resolveSetuModel(model, options?.sessionId, {
 			messageId: options?.messageId,
 			topupApprovalMode: options?.topupApprovalMode,
+			autoPayThresholdUsd: options?.autoPayThresholdUsd,
 		});
 	}
 	if (provider === 'zai') {

package/src/runtime/provider/setu.ts

@@ -21,6 +21,7 @@ function getProviderNpm(model: string): string | undefined {
 export interface ResolveSetuModelOptions {
 	messageId?: string;
 	topupApprovalMode?: 'auto' | 'approval';
+	autoPayThresholdUsd?: number;
 }
 
 async function getSetuPrivateKey(): Promise<string> {
@@ -50,7 +51,11 @@ export async function resolveSetuModel(
 	}
 	const baseURL = process.env.SETU_BASE_URL;
 	const rpcURL = process.env.SETU_SOLANA_RPC_URL;
-	const { messageId, topupApprovalMode = 'approval' } = options;
+	const {
+		messageId,
+		topupApprovalMode = 'approval',
+		autoPayThresholdUsd = MIN_TOPUP_USD,
+	} = options;
 
 	const callbacks: SetuPaymentCallbacks = sessionId
 		? {
@@ -128,6 +133,7 @@ export async function resolveSetuModel(
 			callbacks,
 			providerNpm,
 			topupApprovalMode,
+			autoPayThresholdUsd,
 		},
 	);
 }