UNPKG

@smythos/sdk

Version:
72 lines (58 loc) 1.99 kB
//!!! 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 TClassifierSettings { name?: string; model: string; /** Prompt */ prompt?: string; } export type TClassifierInputs = { Input?: any; [key: string]: any; }; export type TClassifierOutputs = { [key: string]: any; }; export function Classifier(settings?: TClassifierSettings, agent?: Agent) { const { name, ...settingsWithoutName } = settings || {}; const dataObject: any = { name: settings?.name || 'Classifier', settings: { ...settingsWithoutName } }; const component = new ComponentWrapper(dataObject, agent); if (agent) { (agent.structure.components as ComponentWrapper[]).push(component); } const _out: TClassifierOutputs = createSafeAccessor({ // No outputs defined }, component, ''); const _in: { [key: string]: ComponentInput } = { Input: { component, type: 'Any', optional: false, default: true, }, }; 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: TClassifierInputs) => void, }; return wrapper; }