import {ChatAdapterBuilder, ChatAdapterOptions, createChatAdapter} from '@nlux/hf';
import {NluxUsageError} from '@shared/types/error';

const source = 'hooks/getAdapterBuilder';

export const getAdapterBuilder = <AiMsg>(
    options: ChatAdapterOptions<AiMsg>,
): ChatAdapterBuilder<AiMsg> => {
    const {
        model,
        authToken,
        dataTransferMode,
        preProcessors,
        maxNewTokens,
        systemMessage,
    } = options || {};

    if (dataTransferMode && dataTransferMode !== 'stream' && dataTransferMode !== 'batch') {
        throw new NluxUsageError({
            source,
            message: 'Data transfer mode for Hugging Face Inference API must be either "stream" or "batch"',
        });
    }

    if (model === undefined) {
        throw new NluxUsageError({
            source,
            message: 'You must provide either a model or an endpoint to use Hugging Face Inference API.',
        });
    }

    let newAdapter = createChatAdapter<AiMsg>().withModel(model);

    if (authToken !== undefined) {
        newAdapter = newAdapter.withAuthToken(authToken);
    }

    if (dataTransferMode !== undefined) {
        newAdapter = newAdapter.withDataTransferMode(dataTransferMode);
    }

    if (preProcessors?.input !== undefined) {
        newAdapter = newAdapter.withInputPreProcessor(preProcessors.input);
    }

    if (preProcessors?.output !== undefined) {
        newAdapter = newAdapter.withOutputPreProcessor(preProcessors?.output);
    }

    if (systemMessage !== undefined) {
        newAdapter = newAdapter.withSystemMessage(systemMessage);
    }

    if (maxNewTokens !== undefined) {
        newAdapter = newAdapter.withMaxNewTokens(maxNewTokens);
    }

    return newAdapter;
};