import { createAGIProxySDK } from '@aerocorp/agi-proxy-sdk';

// Initialize the SDK
const agi = createAGIProxySDK({
  baseUrl: 'https://agi.aerocorpindustries.org',
  apiKey: 'your-api-key', // Optional
  defaultModel: 'google/gemini-2.5-flash'
});

// Create a chat completion
const response = await agi.createChatCompletion({
  messages: [
    { role: 'user', content: 'Hello, how are you?' }
  ]
});

console.log(response.choices[0].message.content);

import { AGIProxySDK } from '@aerocorp/agi-proxy-sdk';

const agi = new AGIProxySDK({
  baseUrl: 'https://agi.aerocorpindustries.org',
  apiKey: 'your-api-key',
  defaultModel: 'google/gemini-2.5-flash',
  timeout: 120000, // 120 seconds
  retries: 3 // Number of retry attempts
});

// Chat completion with custom parameters
const response = await agi.createChatCompletion({
  messages: [
    { role: 'system', content: 'You are a helpful assistant.' },
    { role: 'user', content: 'Explain quantum computing in simple terms.' }
  ],
  model: 'google/gemini-2.5-pro', // Override default model
  temperature: 0.7,
  max_tokens: 1024,
  top_p: 1.0
});

const stream = agi.streamChatCompletion({
  messages: [
    { role: 'user', content: 'Write a short story about AI.' }
  ],
  stream: true
});

for await (const chunk of stream) {
  const content = chunk.choices[0]?.delta?.content;
  if (content) {
    process.stdout.write(content);
  }
}

const models = await agi.listModels();
console.log('Available models:', models.data);

const health = await agi.getHealth();
console.log('Service status:', health);

interface AGIProxyConfig {
  baseUrl?: string;        // Default: 'https://agi.aerocorpindustries.org'
  apiKey?: string;         // Optional API key
  defaultModel?: string;   // Default: 'google/gemini-2.5-flash'
  timeout?: number;        // Default: 120000 (120 seconds)
  retries?: number;        // Default: 3
}

try {
  const response = await agi.createChatCompletion({
    messages: [{ role: 'user', content: 'Hello!' }]
  });
} catch (error) {
  if (error instanceof Error) {
    console.error('Error:', error.message);
  }
}

import type {
  ChatMessage,
  ChatCompletionRequest,
  ChatCompletionResponse,
  Model,
  ModelsResponse
} from '@aerocorp/agi-proxy-sdk';

const agi = createAGIProxySDK();

async function chat(userMessage: string) {
  const response = await agi.createChatCompletion({
    messages: [
      { role: 'system', content: 'You are a helpful AI assistant.' },
      { role: 'user', content: userMessage }
    ]
  });
  
  return response.choices[0].message.content;
}

const answer = await chat('What is the meaning of life?');
console.log(answer);

const conversation: ChatMessage[] = [
  { role: 'system', content: 'You are a helpful assistant.' }
];

async function sendMessage(message: string) {
  conversation.push({ role: 'user', content: message });
  
  const response = await agi.createChatCompletion({
    messages: conversation
  });
  
  const assistantMessage = response.choices[0].message;
  conversation.push(assistantMessage);
  
  return assistantMessage.content;
}

await sendMessage('Hello!');
await sendMessage('Tell me a joke.');