wehub-resource-sync
45 lines
1.0 KiB
TypeScript
45 lines
1.0 KiB
TypeScript
import { post } from '../utils';
|
|
import { GuardResult, PIIResult, HarmResult, PromptfooResult } from './types';
|
|
|
|
export const PROMPTFOO_BASE_URL = 'https://api.promptfoo.dev/v1';
|
|
|
|
export const postPromptfoo = async <
|
|
T extends GuardResult | PIIResult | HarmResult,
|
|
>(
|
|
endpoint: string,
|
|
data: any,
|
|
timeout?: number
|
|
): Promise<PromptfooResult<T>> => {
|
|
const options = {
|
|
headers: {
|
|
'Content-Type': 'application/json',
|
|
},
|
|
};
|
|
|
|
switch (endpoint) {
|
|
case 'guard':
|
|
return post(
|
|
`${PROMPTFOO_BASE_URL}/guard`,
|
|
data,
|
|
options,
|
|
timeout
|
|
) as Promise<PromptfooResult<T>>;
|
|
case 'pii':
|
|
return post(
|
|
`${PROMPTFOO_BASE_URL}/pii`,
|
|
data,
|
|
options,
|
|
timeout
|
|
) as Promise<PromptfooResult<T>>;
|
|
case 'harm':
|
|
return post(
|
|
`${PROMPTFOO_BASE_URL}/harm`,
|
|
data,
|
|
options,
|
|
timeout
|
|
) as Promise<PromptfooResult<T>>;
|
|
default:
|
|
throw new Error(`Unknown Promptfoo endpoint: ${endpoint}`);
|
|
}
|
|
};
|