Files
llm-promptfoo/test/__mocks__/tempCustomModule.ts
2023-05-03 16:39:02 -07:00

15 lines
251 B
TypeScript

class CustomApiProvider {
id() {
return 'custom-api-provider';
}
async callApi(prompt) {
return {
output: 'Custom output',
tokenUsage: { total: 10, prompt: 5, completion: 5 },
};
}
}
export default CustomApiProvider;