@llumiverse/drivers
Version:
LLM driver implementations. Currently supported are: openai, huggingface, bedrock, replicate.
30 lines • 1.28 kB
TypeScript
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