UNPKG

@llumiverse/drivers

Version:

LLM driver implementations. Currently supported are: openai, huggingface, bedrock, replicate.

30 lines 1.28 kB
import { AbstractDriver, AIModel, Completion, CompletionChunkObject, DriverOptions, EmbeddingsOptions, EmbeddingsResult, ExecutionOptions, ModelSearchPayload, PromptSegment } from "@llumiverse/core"; interface FireflySize { width: number; height: number; } export interface FireflyDriverOptions extends DriverOptions { /** * Adobe Firefly API key */ apiKey: string; /** * Optional API endpoint override */ endpoint?: string; } export declare class FireflyDriver extends AbstractDriver<FireflyDriverOptions> { static PROVIDER: string; provider: string; private readonly endpoint; constructor(options: FireflyDriverOptions); requestTextCompletion(_prompt: string, _options: ExecutionOptions): Promise<Completion>; requestTextCompletionStream(_prompt: string, _options: ExecutionOptions): Promise<AsyncIterable<CompletionChunkObject>>; requestImageGeneration(segments: PromptSegment[], options: ExecutionOptions): Promise<Completion>; mapSize(size?: string): FireflySize; listModels(_params?: ModelSearchPayload): Promise<AIModel[]>; validateConnection(): Promise<boolean>; generateEmbeddings(_options: EmbeddingsOptions): Promise<EmbeddingsResult>; } export {}; //# sourceMappingURL=firefly.d.ts.map