@blockrun/llm 0.1.1 → 0.2.0

package/dist/index.d.ts

@@ -14,6 +14,7 @@ interface ChatUsage {
     prompt_tokens: number;
     completion_tokens: number;
     total_tokens: number;
+    num_sources_used?: number;
 }
 interface ChatResponse {
     id: string;
@@ -22,6 +23,7 @@ interface ChatResponse {
     model: string;
     choices: ChatChoice[];
     usage?: ChatUsage;
+    citations?: string[];
 }
 interface Model {
     id: string;
@@ -33,6 +35,83 @@ interface Model {
     contextWindow: number;
     maxOutput: number;
     available: boolean;
+    type?: "llm" | "image";
+}
+interface ImageData {
+    url: string;
+    revised_prompt?: string;
+    b64_json?: string;
+}
+interface ImageResponse {
+    created: number;
+    data: ImageData[];
+}
+interface ImageModel {
+    id: string;
+    name: string;
+    provider: string;
+    description: string;
+    pricePerImage: number;
+    supportedSizes?: string[];
+    maxPromptLength?: number;
+    available: boolean;
+    type?: "llm" | "image";
+}
+interface ImageClientOptions {
+    /** EVM wallet private key (hex string starting with 0x) */
+    privateKey?: `0x${string}` | string;
+    /** API endpoint URL (default: https://blockrun.ai/api) */
+    apiUrl?: string;
+    /** Request timeout in milliseconds (default: 120000 for images) */
+    timeout?: number;
+}
+interface ImageGenerateOptions {
+    /** Model ID (default: "google/nano-banana") */
+    model?: string;
+    /** Image size (default: "1024x1024") */
+    size?: string;
+    /** Number of images to generate (default: 1) */
+    n?: number;
+    /** Image quality (for supported models) */
+    quality?: "standard" | "hd";
+}
+interface WebSearchSource {
+    type: "web";
+    country?: string;
+    excludedWebsites?: string[];
+    allowedWebsites?: string[];
+    safeSearch?: boolean;
+}
+interface XSearchSource {
+    type: "x";
+    includedXHandles?: string[];
+    excludedXHandles?: string[];
+    postFavoriteCount?: number;
+    postViewCount?: number;
+}
+interface NewsSearchSource {
+    type: "news";
+    country?: string;
+    excludedWebsites?: string[];
+    allowedWebsites?: string[];
+    safeSearch?: boolean;
+}
+interface RssSearchSource {
+    type: "rss";
+    links: string[];
+}
+type SearchSource = WebSearchSource | XSearchSource | NewsSearchSource | RssSearchSource;
+interface SearchParameters {
+    mode?: "off" | "auto" | "on";
+    sources?: SearchSource[];
+    returnCitations?: boolean;
+    fromDate?: string;
+    toDate?: string;
+    maxSearchResults?: number;
+}
+interface Spending {
+    totalUsd: number;
+    calls: number;
 }
 interface LLMClientOptions {
     /** EVM wallet private key (hex string starting with 0x). Optional if BASE_CHAIN_WALLET_KEY env var is set. */
@@ -51,6 +130,10 @@ interface ChatOptions {
     temperature?: number;
     /** Nucleus sampling parameter */
     topP?: number;
+    /** Enable xAI Live Search (shortcut for searchParameters.mode = "on") */
+    search?: boolean;
+    /** Full xAI Live Search configuration (for Grok models) */
+    searchParameters?: SearchParameters;
 }
 interface ChatCompletionOptions {
     /** Max tokens to generate */
@@ -59,6 +142,10 @@ interface ChatCompletionOptions {
     temperature?: number;
     /** Nucleus sampling parameter */
     topP?: number;
+    /** Enable xAI Live Search (shortcut for searchParameters.mode = "on") */
+    search?: boolean;
+    /** Full xAI Live Search configuration (for Grok models) */
+    searchParameters?: SearchParameters;
 }
 declare class BlockrunError extends Error {
     constructor(message: string);
@@ -99,6 +186,8 @@ declare class LLMClient {
     private privateKey;
     private apiUrl;
     private timeout;
+    private sessionTotalUsd;
+    private sessionCalls;
     /**
      * Initialize the BlockRun LLM client.
      *
@@ -140,13 +229,117 @@ declare class LLMClient {
      */
     private fetchWithTimeout;
     /**
-     * List available models with pricing.
+     * List available LLM models with pricing.
      */
     listModels(): Promise<Model[]>;
+    /**
+     * List available image generation models with pricing.
+     */
+    listImageModels(): Promise<ImageModel[]>;
+    /**
+     * List all available models (both LLM and image) with pricing.
+     *
+     * @returns Array of all models with 'type' field ('llm' or 'image')
+     *
+     * @example
+     * const models = await client.listAllModels();
+     * for (const model of models) {
+     *   if (model.type === 'llm') {
+     *     console.log(`LLM: ${model.id} - $${model.inputPrice}/M input`);
+     *   } else {
+     *     console.log(`Image: ${model.id} - $${model.pricePerImage}/image`);
+     *   }
+     * }
+     */
+    listAllModels(): Promise<(Model | ImageModel)[]>;
+    /**
+     * Get current session spending.
+     *
+     * @returns Object with totalUsd and calls count
+     *
+     * @example
+     * const spending = client.getSpending();
+     * console.log(`Spent $${spending.totalUsd.toFixed(4)} across ${spending.calls} calls`);
+     */
+    getSpending(): Spending;
+    /**
+     * Get the wallet address being used for payments.
+     */
+    getWalletAddress(): string;
+}
+
+/**
+ * BlockRun Image Client - Generate images via x402 micropayments.
+ *
+ * SECURITY NOTE - Private Key Handling:
+ * Your private key NEVER leaves your machine. Here's what happens:
+ * 1. Key stays local - only used to sign an EIP-712 typed data message
+ * 2. Only the SIGNATURE is sent in the PAYMENT-SIGNATURE header
+ * 3. BlockRun verifies the signature on-chain via Coinbase CDP facilitator
+ *
+ * Usage:
+ *   import { ImageClient } from '@blockrun/llm';
+ *
+ *   const client = new ImageClient({ privateKey: '0x...' });
+ *   const result = await client.generate('A cute cat in space');
+ *   console.log(result.data[0].url);
+ */
+
+/**
+ * BlockRun Image Generation Client.
+ *
+ * Generate images using Nano Banana (Google Gemini), DALL-E 3, or GPT Image
+ * with automatic x402 micropayments on Base chain.
+ */
+declare class ImageClient {
+    private account;
+    private privateKey;
+    private apiUrl;
+    private timeout;
+    private sessionTotalUsd;
+    private sessionCalls;
+    /**
+     * Initialize the BlockRun Image client.
+     *
+     * @param options - Client configuration options
+     */
+    constructor(options?: ImageClientOptions);
+    /**
+     * Generate an image from a text prompt.
+     *
+     * @param prompt - Text description of the image to generate
+     * @param options - Optional generation parameters
+     * @returns ImageResponse with generated image URLs
+     *
+     * @example
+     * const result = await client.generate('A sunset over mountains');
+     * console.log(result.data[0].url);
+     */
+    generate(prompt: string, options?: ImageGenerateOptions): Promise<ImageResponse>;
+    /**
+     * List available image generation models with pricing.
+     */
+    listImageModels(): Promise<ImageModel[]>;
+    /**
+     * Make a request with automatic x402 payment handling.
+     */
+    private requestWithPayment;
+    /**
+     * Handle 402 response: parse requirements, sign payment, retry.
+     */
+    private handlePaymentAndRetry;
+    /**
+     * Fetch with timeout.
+     */
+    private fetchWithTimeout;
     /**
      * Get the wallet address being used for payments.
      */
     getWalletAddress(): string;
+    /**
+     * Get session spending information.
+     */
+    getSpending(): Spending;
 }
 
 /**
@@ -159,6 +352,105 @@ declare class LLMClient {
 declare const BASE_CHAIN_ID = 8453;
 declare const USDC_BASE: "0x833589fCD6eDb6E08f4c7C32D4f71b54bdA02913";
 
+/**
+ * BlockRun Wallet Management - Auto-create and manage wallets.
+ *
+ * Provides frictionless wallet setup for new users:
+ * - Auto-creates wallet if none exists
+ * - Stores key securely at ~/.blockrun/.session
+ * - Generates EIP-681 URIs for easy MetaMask funding
+ */
+declare const USDC_BASE_CONTRACT = "0x833589fCD6eDb6E08f4c7C32D4f71b54bdA02913";
+interface WalletInfo {
+    privateKey: string;
+    address: string;
+    isNew: boolean;
+}
+interface PaymentLinks {
+    basescan: string;
+    walletLink: string;
+    ethereum: string;
+    blockrun: string;
+}
+/**
+ * Create a new Ethereum wallet.
+ *
+ * @returns Object with address and privateKey
+ */
+declare function createWallet(): {
+    address: string;
+    privateKey: string;
+};
+/**
+ * Save wallet private key to ~/.blockrun/.session
+ *
+ * @param privateKey - Private key string (with 0x prefix)
+ * @returns Path to saved wallet file
+ */
+declare function saveWallet(privateKey: string): string;
+/**
+ * Load wallet private key from file.
+ *
+ * @returns Private key string or null if not found
+ */
+declare function loadWallet(): string | null;
+/**
+ * Get existing wallet or create new one.
+ *
+ * Priority:
+ * 1. BLOCKRUN_WALLET_KEY environment variable
+ * 2. ~/.blockrun/.session file
+ * 3. ~/.blockrun/wallet.key file (legacy)
+ * 4. Create new wallet
+ *
+ * @returns WalletInfo with address, privateKey, and isNew flag
+ */
+declare function getOrCreateWallet(): WalletInfo;
+/**
+ * Get wallet address without exposing private key.
+ *
+ * @returns Wallet address or null if no wallet configured
+ */
+declare function getWalletAddress(): string | null;
+/**
+ * Generate EIP-681 URI for USDC transfer on Base.
+ *
+ * @param address - Recipient Ethereum address
+ * @param amountUsdc - Amount in USDC (default 1.0)
+ * @returns EIP-681 URI string for MetaMask/wallet scanning
+ */
+declare function getEip681Uri(address: string, amountUsdc?: number): string;
+/**
+ * Generate payment links for the wallet address.
+ *
+ * @param address - Ethereum address
+ * @returns Object with various payment links
+ */
+declare function getPaymentLinks(address: string): PaymentLinks;
+/**
+ * Format the message shown when a new wallet is created.
+ *
+ * @param address - New wallet address
+ * @returns Formatted message string
+ */
+declare function formatWalletCreatedMessage(address: string): string;
+/**
+ * Format the message shown when wallet needs more funds.
+ *
+ * @param address - Wallet address
+ * @returns Formatted message string
+ */
+declare function formatNeedsFundingMessage(address: string): string;
+/**
+ * Compact funding message (no QR) for repeated displays.
+ *
+ * @param address - Wallet address
+ * @returns Short formatted message string
+ */
+declare function formatFundingMessageCompact(address: string): string;
+declare const WALLET_FILE_PATH: string;
+declare const WALLET_DIR_PATH: string;
+
 /**
  * OpenAI-compatible API wrapper for BlockRun LLM SDK.
  *
@@ -292,4 +584,4 @@ declare class OpenAI {
     getWalletAddress(): string;
 }
 
-export { APIError, BASE_CHAIN_ID, BlockrunError, type ChatChoice, type ChatCompletionOptions, type ChatMessage, type ChatOptions, type ChatResponse, type ChatUsage, LLMClient, type LLMClientOptions, type Model, OpenAI, type OpenAIChatCompletionChoice, type OpenAIChatCompletionChunk, type OpenAIChatCompletionParams, type OpenAIChatCompletionResponse, type OpenAIClientOptions, PaymentError, USDC_BASE, LLMClient as default };
+export { APIError, BASE_CHAIN_ID, BlockrunError, type ChatChoice, type ChatCompletionOptions, type ChatMessage, type ChatOptions, type ChatResponse, type ChatUsage, ImageClient, type ImageClientOptions, type ImageData, type ImageGenerateOptions, type ImageModel, type ImageResponse, LLMClient, type LLMClientOptions, type Model, type NewsSearchSource, OpenAI, type OpenAIChatCompletionChoice, type OpenAIChatCompletionChunk, type OpenAIChatCompletionParams, type OpenAIChatCompletionResponse, type OpenAIClientOptions, PaymentError, type PaymentLinks, type RssSearchSource, type SearchParameters, type SearchSource, type Spending, USDC_BASE, USDC_BASE_CONTRACT, WALLET_DIR_PATH, WALLET_FILE_PATH, type WalletInfo, type WebSearchSource, type XSearchSource, createWallet, LLMClient as default, formatFundingMessageCompact, formatNeedsFundingMessage, formatWalletCreatedMessage, getEip681Uri, getOrCreateWallet, getPaymentLinks, getWalletAddress, loadWallet, saveWallet };

package/dist/{index.esm-FLTVMT4C.mjs → index.esm-DGL5QIKD.js}

@@ -75,8 +75,8 @@ import {
   clusterApiUrl,
   sendAndConfirmRawTransaction,
   sendAndConfirmTransaction
-} from "./chunk-R7MQBF5Y.mjs";
-import "./chunk-HEBXNMVQ.mjs";
+} from "./chunk-4LEERQOK.js";
+import "./chunk-2ESYSVXG.js";
 export {
   Account,
   AddressLookupTableAccount,