@ottocode/ai-sdk 0.1.3 → 0.1.4

package/package.json

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

package/src/auth.ts

@@ -9,8 +9,8 @@ export interface WalletContext {
   privateKeyBytes: Uint8Array;
 }
 
-export function createWalletContext(auth: SetuAuth): WalletContext {
-  const privateKeyBytes = bs58.decode(auth.privateKey);
+export function createWalletContext(auth: Required<SetuAuth>): WalletContext {
+  const privateKeyBytes = bs58.decode(auth.privateKey!);
   const keypair = Keypair.fromSecretKey(privateKeyBytes);
   const walletAddress = keypair.publicKey.toBase58();
   return { keypair, walletAddress, privateKeyBytes };
@@ -32,7 +32,8 @@ export function buildWalletHeaders(ctx: WalletContext): Record<string, string> {
   };
 }
 
-export function getPublicKeyFromPrivate(privateKey: string): string | null {
+export function getPublicKeyFromPrivate(privateKey?: string): string | null {
+  if (!privateKey) return null;
   try {
     const privateKeyBytes = bs58.decode(privateKey);
     const keypair = Keypair.fromSecretKey(privateKeyBytes);

package/src/balance.ts

@@ -13,11 +13,11 @@ function trimTrailingSlash(url: string) {
 }
 
 export async function fetchBalance(
-  auth: SetuAuth,
+  auth: Required<SetuAuth>,
   baseURL?: string,
 ): Promise<BalanceResponse | null> {
   try {
-    const privateKeyBytes = bs58.decode(auth.privateKey);
+    const privateKeyBytes = bs58.decode(auth.privateKey!);
     const keypair = Keypair.fromSecretKey(privateKeyBytes);
     const walletAddress = keypair.publicKey.toBase58();
     const url = trimTrailingSlash(baseURL ?? DEFAULT_BASE_URL);
@@ -60,11 +60,11 @@ export async function fetchBalance(
 }
 
 export async function fetchWalletUsdcBalance(
-  auth: SetuAuth,
+  auth: Required<SetuAuth>,
   network: 'mainnet' | 'devnet' = 'mainnet',
 ): Promise<WalletUsdcBalance | null> {
   try {
-    const privateKeyBytes = bs58.decode(auth.privateKey);
+    const privateKeyBytes = bs58.decode(auth.privateKey!);
     const keypair = Keypair.fromSecretKey(privateKeyBytes);
     const walletAddress = keypair.publicKey.toBase58();
     const rpcUrl = network === 'devnet' ? 'https://api.devnet.solana.com' : DEFAULT_RPC_URL;

package/src/setu.ts

@@ -27,7 +27,12 @@ export interface SetuInstance {
 
 export function createSetu(config: SetuConfig): SetuInstance {
   const baseURL = trimTrailingSlash(config.baseURL ?? DEFAULT_BASE_URL);
-  const wallet = createWalletContext(config.auth);
+  const privateKey = config.auth.privateKey || process.env.SETU_PRIVATE_KEY;
+  if (!privateKey) {
+    throw new Error('Setu: privateKey is required. Pass it via config.auth.privateKey or set SETU_PRIVATE_KEY env variable.');
+  }
+  const resolvedAuth = { ...config.auth, privateKey };
+  const wallet = createWalletContext(resolvedAuth);
   const registry = new ProviderRegistry(config.providers, config.modelMap);
 
   const setuFetch = createSetuFetch({
@@ -74,14 +79,14 @@ export function createSetu(config: SetuConfig): SetuInstance {
     },
 
     async balance() {
-      return fetchBalance(config.auth, baseURL);
+      return fetchBalance(resolvedAuth, baseURL);
     },
 
     async walletBalance(network?: 'mainnet' | 'devnet') {
-      return fetchWalletUsdcBalance(config.auth, network);
+      return fetchWalletUsdcBalance(resolvedAuth, network);
     },
 
-    walletAddress: getPublicKeyFromPrivate(config.auth.privateKey),
+    walletAddress: getPublicKeyFromPrivate(resolvedAuth.privateKey),
 
     registry,
   };

package/src/types.ts

@@ -21,7 +21,7 @@ export interface ProviderConfig {
 }
 
 export interface SetuAuth {
-  privateKey: string;
+  privateKey?: string;
 }
 
 export interface BalanceUpdate {