LLM Chat API Examples

Basic example showing synchronous usage of the AIME API client interface.

chat_example.js

const { ModelAPI, doAPIRequest } = require('../model_api');

// Configuration
const CONFIG = {
    BASE_URL: 'https://api.aime.info',
    ENDPOINT: 'llama4_chat', 
    API_EMAIL: 'apiexample@aime.info',
    API_KEY: '181e35ac-7b7d-4bfe-9f12-153757ec3952'
};

// Set up global fetch to handle base URL
const { fetch: originalFetch } = global;
global.fetch = (input, init = {}) => {
    const url = typeof input === 'string' 
        ? input.startsWith('http') ? input : `${CONFIG.BASE_URL}${input.startsWith('/') ? '' : '/'}${input}`
        : input;
    return originalFetch(url, init);
};

// Chat parameters
const chatContext = [
    { role: 'user', content: 'Hi! How are you?' },
    { role: 'assistant', content: 'I\'m doing well, thank you! How can I help you today?' }
];

const params = {
    prompt_input: 'Tell me a joke',
    chat_context: JSON.stringify(chatContext),
    top_k: 40,
    top_p: 0.9,
    temperature: 0.8,
    max_gen_tokens: 1000
};

console.log('💬 Starting chat...');

// Make the API request
doAPIRequest(
    CONFIG.ENDPOINT,
    params,
    (result, error) => {
        if (error) {
            console.error('❌ Error:', error.message || 'Unknown error');
            if (error.response) {
                console.error('Response status:', error.response.status);
                console.error('Response data:', JSON.stringify(error.response.data, null, 2));
            }
            return;
        }

        console.log('\n📄 Full API response:', JSON.stringify(result, null, 2));

        if (!result) {
            console.log('⚠️  No response received from the API');
            return;
        }

        // Handle the response
        if (result && result.success === false) {
            console.error('❌ API Error:', result.error || 'Unknown error');
            return;
        }

        if (result && result.text) {
            console.log('\n🤖 Assistant:', result.text);
            console.log('\n📊 Stats:', {
                'Generated Tokens': result.num_generated_tokens,
                'Compute Duration': `${result.compute_duration.toFixed(2)}s`,
                'Total Duration': `${result.total_duration.toFixed(2)}s`
            });
        } else {
            console.log('⚠️  Unexpected response format:');
            console.log(JSON.stringify(result, null, 2));
        }
        
        console.log('\n✨ Chat completed!');
    },
    CONFIG.API_EMAIL,
    CONFIG.API_KEY,
    (progress) => {
        const msg = progress.progress >= 0 
            ? `Progress: ${progress.progress}%` 
            : 'Thinking...';
        process.stdout.write(`\r${msg}${' '.repeat(20)}`);
    }
);

Image Generation Example

Example showing how to generate images using the AIME API client interface.

image_generation_example.js

const { doAPIRequest } = require('../model_api');
const fs = require('fs');

// Configuration
const CONFIG = {
    BASE_URL: 'https://api.aime.info',
    ENDPOINT: 'flux-dev',
    API_EMAIL: 'apiexample@aime.info',
    API_KEY: '181e35ac-7b7d-4bfe-9f12-153757ec3952'
};

// Set up global fetch to handle base URL
const { fetch: originalFetch } = global;
global.fetch = (input, init = {}) => {
    const url = typeof input === 'string' 
        ? input.startsWith('http') ? input : `${CONFIG.BASE_URL}${input.startsWith('/') ? '' : '/'}${input}`
        : input;
    return originalFetch(url, init);
};

// Image generation parameters
const params = {
    prompt: 'A beautiful sunset over mountains',
    height: 1024,
    width: 1024,
    steps: 30,
    guidance: 7.5,
    wait_for_result: true
};

console.log('🚀 Starting image generation...');

// Make the API request
doAPIRequest(
    CONFIG.ENDPOINT,
    params,
    (result, error) => {
        if (error) {
            console.error('❌ Error:', error.message || 'Unknown error');
            return;
        }

        const images = result.images || [];
        if (images.length === 0) {
            console.log('⚠️  No images were generated');
            return;
        }

        // Save the first image
        const imageData = images[0].includes(',') ? images[0].split(',')[1] : images[0];
        const outputPath = 'generated_image.png';
        fs.writeFileSync(outputPath, Buffer.from(imageData, 'base64'));
        
        console.log(`✅ Image saved as ${outputPath}`);
        console.log('✨ All done!');
    },
    CONFIG.API_EMAIL,
    CONFIG.API_KEY,
    (progress) => {
        const msg = progress.progress >= 0 
            ? `Progress: ${progress.progress}%` 
            : 'Starting...';
        process.stdout.write(`\r${msg}${' '.repeat(20)}`);
    }
);

Text-to-Speech Generation Example

Example showing text-to-speech generation using the AIME API client interface.

tts_example.js

const { doAPIRequest } = require('../model_api');
const fs = require('fs');

// Configuration
const CONFIG = {
    BASE_URL: 'https://api.aime.info',
    ENDPOINT: 'tts_tortoise',
    API_EMAIL: 'apiexample@aime.info',
    API_KEY: '181e35ac-7b7d-4bfe-9f12-153757ec3952',
    OUTPUT_FILE: 'generated_speech.mp3'
};

// Set up global fetch to handle base URL
const { fetch: originalFetch } = global;
global.fetch = (input, init = {}) => {
    const url = typeof input === 'string' 
        ? input.startsWith('http') ? input : `${CONFIG.BASE_URL}${input.startsWith('/') ? '' : '/'}${input}`
        : input;
    return originalFetch(url, init);
};

// TTS parameters
const params = {
    text: "Hello, this is a test of the text-to-speech system. How are you doing today?",
    voice: "emma",
    preset: "fast",
};

console.log('🔊 Starting text-to-speech generation...');

// Make the API request
doAPIRequest(
    CONFIG.ENDPOINT,
    params,
    (result, error) => {
        if (error) {
            console.error('❌ Error:', error.message || 'Unknown error');
            return;
        }

        // Check for audio data in both audio_output and audio_data fields
        const audioData = result.audio_output || result.audio_data;
        
        if (audioData) {
            // Extract base64 data if it's a data URL
            const base64Data = audioData.includes(',') 
                ? audioData.split(',')[1] 
                : audioData;
            
            try {
                fs.writeFileSync(CONFIG.OUTPUT_FILE, Buffer.from(base64Data, 'base64'));
                console.log(`✅ Audio saved as ${CONFIG.OUTPUT_FILE}`);
                return;
            } catch (writeError) {
                console.error('❌ Error saving audio file:', writeError.message);
            }
        }
        
        console.log('⚠️  No audio data received in the response');
        if (result) {
            console.log('API response:', JSON.stringify(result, null, 2));
        }
        
        console.log('✨ All done!');
    },
    CONFIG.API_EMAIL,
    CONFIG.API_KEY,
    (progress) => {
        const msg = progress.progress >= 0 
            ? `Progress: ${progress.progress}%` 
            : 'Processing...';
        process.stdout.write(`\r${msg}${' '.repeat(20)}`);
    }
);