portkey-ai
Version:
Node client library for the Portkey API
42 lines (36 loc) • 1.18 kB
text/typescript
import { ApiClientInterface } from '../_types/generalTypes';
import { ApiResource } from '../apiResource';
import { RequestOptions } from '../baseClient';
import { finalResponse, initOpenAIClient, overrideConfig } from '../utils';
import { createHeaders } from './createHeaders';
export interface ModerationCreateParams {
input: string | Array<string> | Array<ModerationMultiModalInput>;
model?: any;
[key: string]: any;
}
interface ModerationMultiModalInput {
type: string;
text?: string;
image_url?: object;
}
export class Moderations extends ApiResource {
async create(
_body: ModerationCreateParams,
params?: ApiClientInterface,
opts?: RequestOptions
): Promise<any> {
const body: ModerationCreateParams = _body;
if (params) {
const config = overrideConfig(this.client.config, params.config);
this.client.customHeaders = {
...this.client.customHeaders,
...createHeaders({ ...params, config }),
};
}
const OAIclient = initOpenAIClient(this.client);
const result = await OAIclient.moderations
.create(body as any, opts)
.withResponse();
return finalResponse(result);
}
}