basic-usage.js

/**
 * Basic Usage Example - AetherGuard SDK
 *
 * Prerequisites:
 *   1. Sign up at https://portal.aetherguard.ai
 *   2. Add a provider, register a model, and generate an API key
 *   3. Set AETHERGUARD_API_KEY and AETHERGUARD_BASE_URL env vars
 */

const { AetherGuardClient } = require('@aetherguard/sdk');

async function basicExample() {
  const client = new AetherGuardClient({
    apiKey: process.env.AETHERGUARD_API_KEY || 'your-api-key-here',
    baseUrl: process.env.AETHERGUARD_BASE_URL || 'http://localhost:8080',
    debug: true
  });

  try {
    // Test connection to the proxy
    console.log('Testing connection...');
    const isConnected = await client.testConnection();
    console.log('Connected:', isConnected);

    if (!isConnected) {
      console.error('Cannot reach AetherGuard proxy. Check your AETHERGUARD_BASE_URL.');
      return;
    }

    // Simple chat completion
    console.log('\n--- Chat Completion ---');
    const response = await client.createChatCompletion({
      model: 'gpt-4',
      messages: [
        { role: 'system', content: 'You are a helpful assistant.' },
        { role: 'user', content: 'What is artificial intelligence?' }
      ],
      max_tokens: 150,
      temperature: 0.7
    });

    console.log('Response:', response.choices[0].message.content);
    console.log('Tokens used:', response.usage);
    console.log('Finish reason:', response.choices[0].finish_reason);

    // Simple completion (non-chat)
    console.log('\n--- Simple Completion ---');
    const simple = await client.createCompletion({
      model: 'gpt-4',
      prompt: 'Explain machine learning in one sentence.',
      max_tokens: 50
    });

    console.log('Response:', simple.choices[0].message.content);

    // Check proxy status
    console.log('\n--- Proxy Status ---');
    const status = await client.getStatus();
    console.log('Pipeline:', status.pipeline);
    console.log('Stages:', status.stages.join(' → '));

  } catch (error) {
    if (error.code === 'CONTENT_BLOCKED') {
      console.log('Request blocked by security pipeline:', error.message);
    } else {
      console.error('Error:', error);
    }
  }
}

basicExample();