@smythos/sdk
Version:
72 lines (58 loc) • 1.99 kB
text/typescript
//!!! 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;
}