File size: 1,282 Bytes
e706de2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 | /**
* BasePromptTemplate
*
* Abstract prompt template class*
*/
import { Runnable } from '../core/runnable.js';
/**
* Base class for all prompt templates
*/
export class BasePromptTemplate extends Runnable {
constructor(options = {}) {
super();
this.inputVariables = options.inputVariables || [];
this.partialVariables = options.partialVariables || {};
}
/**
* Format the prompt with given values
* @abstract
*/
async format(values) {
throw new Error('Subclasses must implement format()');
}
/**
* Runnable interface: invoke returns formatted prompt
*/
async _call(input, config) {
return await this.format(input);
}
/**
* Validate that all required variables are provided
*/
_validateInput(values) {
const provided = { ...this.partialVariables, ...values };
const missing = this.inputVariables.filter(
key => !(key in provided)
);
if (missing.length > 0) {
throw new Error(
`Missing required input variables: ${missing.join(', ')}`
);
}
}
/**
* Merge partial variables with provided values
*/
_mergePartialAndUserVariables(values) {
return { ...this.partialVariables, ...values };
}
} |