@smythos/sdk/src/Components/generated/HuggingFace.ts

SRE SDK

//!!! DO NOT EDIT THIS FILE, IT IS AUTO-GENERATED !!!//

import { Agent } from '../../Agent/Agent.class';
import { createSafeAccessor } from '../utils';
import { ComponentWrapper } from '../ComponentWrapper.class';
import { InputSettings, ComponentInput } from '../../types/SDKTypes';

export interface THuggingFaceSettings {
    /** Access Token */
    accessToken: string;
    modelName: string;
    modelTask: string;
    inputConfig?: string;
    parameters?: string;
    name: string;
    displayName: string;
    desc: string;
    logoUrl?: string;
    disableCache?: boolean;
}

export type THuggingFaceInputs = {
    [key: string]: InputSettings;
};

export type THuggingFaceOutputs = {
    [key: string]: any;
};

export function HuggingFace(settings?: THuggingFaceSettings, agent?: Agent) {    
    const { name, ...settingsWithoutName } = settings || {};
    const dataObject: any = { 
        name: settings?.name || 'HuggingFace', 
        settings: {
            ...settingsWithoutName 
        }
    };
    const component = new ComponentWrapper(dataObject, agent);

    if (agent) {
        (agent.structure.components as ComponentWrapper[]).push(component);
    }
    
    const _out: THuggingFaceOutputs = createSafeAccessor({
        // No outputs defined
    }, component, '');

    const _in: { [key: string]: ComponentInput } = {
        // No inputs defined
    };

    dataObject.outputs = _out;
    dataObject.inputs = _in;

    component.inputs(_in);

    const wrapper = {
        /** Component outputs - access via .out.OutputName */
        out: _out,        

        /** 
         * Create or Connect the component inputs 
         * if the input does not exist, it will be created
         * @examples 
         *    - component.in({ Input: source.out.data })
         *    - component.in({ Input: { type: 'string', source:source.out.data } })
         */        
        in: component.inputs.bind(component) as (inputs: THuggingFaceInputs) => void,
    };

    return wrapper;
}