@heroku/plugin-ai
Version:
Heroku CLI plugin for Heroku AI add-on
42 lines (41 loc) • 1.95 kB
TypeScript
import Command from '../../../lib/base';
export type ChatCompletionRequest = {
model: string;
messages: Array<{
role: string;
content: string;
}>;
} & {
prompt: string;
};
export default class Call extends Command {
static args: {
model_resource: import("@oclif/core/lib/interfaces/parser").Arg<string, Record<string, unknown>>;
};
static description: string;
static examples: string[];
static flags: {
app: import("@oclif/core/lib/interfaces/parser").OptionFlag<string | undefined, import("@oclif/core/lib/interfaces/parser").CustomOptions>;
json: import("@oclif/core/lib/interfaces/parser").BooleanFlag<boolean>;
optfile: import("@oclif/core/lib/interfaces/parser").OptionFlag<string | undefined, import("@oclif/core/lib/interfaces/parser").CustomOptions>;
opts: import("@oclif/core/lib/interfaces/parser").OptionFlag<string | undefined, import("@oclif/core/lib/interfaces/parser").CustomOptions>;
output: import("@oclif/core/lib/interfaces/parser").OptionFlag<string | undefined, import("@oclif/core/lib/interfaces/parser").CustomOptions>;
prompt: import("@oclif/core/lib/interfaces/parser").OptionFlag<string | undefined, import("@oclif/core/lib/interfaces/parser").CustomOptions>;
remote: import("@oclif/core/lib/interfaces/parser").OptionFlag<string | undefined, import("@oclif/core/lib/interfaces/parser").CustomOptions>;
};
run(): Promise<void>;
/**
* Parse the model call request options from the command flags.
*
* @param optfile Path to a JSON file containing options.
* @param opts JSON string containing options.
* @returns The parsed options as an object.
*/
private parseOptions;
private createChatCompletion;
private displayChatCompletion;
private generateImage;
private displayImageResult;
private createEmbedding;
private displayEmbedding;
}