chat.ts

import { LLMGateway } from "../src/index";
import { EnumLLMProvider } from "../src/types";

const llmGatewayTest = async () => {
    //ANTHROPIC 
    const anthropicGateway = new LLMGateway({
        provider: EnumLLMProvider.ANTHROPIC, // or ANTHROPIC, AZUREOPENAI, OPENAI
        apiKey: process.env['ANTHROPIC_API_KEY'],
    });
    const anthropicResponse = await anthropicGateway.chatCompletion({
        messages: [
            { role: 'user', content: 'Write a one sentence story about a cat.' }
        ],
        model: 'claude-3-5-sonnet-latest',
        max_tokens:200
    })
    console.log('Response:', JSON.stringify(anthropicResponse.llmGatewayOutput, null, 2));


    // OPEN AI
    const openAIGateway = new LLMGateway({
        provider: EnumLLMProvider.OPENAI, // or ANTHROPIC, AZUREOPENAI, OPENAI
        apiKey: process.env['OPENAI_API_KEY'],
    });
    const openAIresponse = await openAIGateway.chatCompletion({
        messages: [
            { role: 'user', content: 'Write a one sentence story about a cat.' }
        ],
        model: 'gpt-4o-mini',
        max_completion_tokens:200
    })
    console.log('Response:', JSON.stringify(openAIresponse.llmGatewayOutput, null, 2));
}
llmGatewayTest()