@sconedev/ai_toolkit/dist/core/providers.js

Simplify AI integration in web apps with local and offline model support

import { getConfig } from './config';
import { OpenAIProvider } from '../providers/openai';
import { HuggingFaceProvider } from '../providers/huggingface';
import { GeminiProvider } from '../providers/gemini';
import { MistralProvider } from '../providers/mistral';
import { OnnxProvider } from '../providers/onnx';
export function getProvider() {
    const config = getConfig();
    const model = config.model;
    switch (config.provider) {
        case 'openai':
            if (!model) {
                throw new Error('Model must be specified for OpenAI provider');
            }
            return new OpenAIProvider(config.apiKey, model);
        case 'huggingface':
            return new HuggingFaceProvider(config.apiKey, model);
        case 'gemini':
            return new GeminiProvider(config.apiKey, model);
        case 'mistral':
            return new MistralProvider(config.apiKey, model);
        case 'onnx': // Add this case
            return new OnnxProvider(config.apiKey, model);
        default:
            throw new Error(`Unsupported provider: ${config.provider}`);
    }
}