@llumiverse/drivers 0.20.0 → 0.22.0-dev.1

package/lib/types/src/togetherai/interfaces.d.ts

@@ -0,0 +1,95 @@
+interface ModelInstanceConfig {
+    appearsIn: any[];
+    order: number;
+}
+interface Config {
+    stop: string[];
+    prompt_format: string;
+    chat_template: string;
+}
+interface Pricing {
+    input: number;
+    output: number;
+    hourly: number;
+}
+interface Instance {
+    av_zone: string;
+    cluster: string;
+}
+interface Ask {
+    [key: string]: number;
+}
+interface Gpu {
+    [key: string]: number;
+}
+interface Price {
+    base: number;
+    fine_tune: number;
+    hourly: number;
+    input: number;
+    output: number;
+}
+interface Stat {
+    av_zone: string;
+    cluster: string;
+    capacity: number;
+    qps: number;
+    throughput_in: number;
+    throughput_out: number;
+    error_rate: number;
+    retry_rate: number;
+}
+interface Depth {
+    num_asks: number;
+    num_bids: number;
+    num_running: number;
+    asks: Ask;
+    asks_updated: string;
+    gpus: Gpu;
+    qps: number;
+    permit_required: boolean;
+    price: Price;
+    throughput_in: number;
+    throughput_out: number;
+    stats: Stat[];
+}
+export interface TogetherModelInfo {
+    modelInstanceConfig: ModelInstanceConfig;
+    _id: string;
+    name: string;
+    display_name: string;
+    display_type: string;
+    description: string;
+    license: string;
+    creator_organization: string;
+    hardware_label: string;
+    num_parameters: number;
+    show_in_playground: boolean;
+    isFeaturedModel: boolean;
+    context_length: number;
+    config: Config;
+    pricing: Pricing;
+    created_at: string;
+    update_at: string;
+    instances: Instance[];
+    access: string;
+    link: string;
+    descriptionLink: string;
+    depth: Depth;
+}
+export interface TextCompletion {
+    id: string;
+    choices: {
+        text: string;
+        finish_reason: string;
+    }[];
+    usage: {
+        prompt_tokens: number;
+        completion_tokens: number;
+        total_tokens: number;
+    };
+    created: number;
+    model: string;
+    object: string;
+}
+export {};

package/lib/types/src/vertexai/debug.d.ts

@@ -0,0 +1 @@
+export declare function logObject(prefix: string, obj: any): void;

package/lib/types/src/vertexai/embeddings/embeddings-image.d.ts

@@ -0,0 +1,10 @@
+import { EmbeddingsOptions, EmbeddingsResult } from '@llumiverse/core';
+import { VertexAIDriver } from '../index.js';
+export interface ImageEmbeddingsOptions {
+    model?: string;
+    image?: {
+        bytesBase64Encoded?: string;
+    };
+    text: string;
+}
+export declare function getEmbeddingsForImages(driver: VertexAIDriver, options: EmbeddingsOptions): Promise<EmbeddingsResult>;

package/lib/types/src/vertexai/embeddings/embeddings-text.d.ts

@@ -0,0 +1,9 @@
+import { EmbeddingsResult } from '@llumiverse/core';
+import { VertexAIDriver } from '../index.js';
+export interface TextEmbeddingsOptions {
+    model?: string;
+    task_type?: "RETRIEVAL_QUERY" | "RETRIEVAL_DOCUMENT" | "SEMANTIC_SIMILARITY" | "CLASSIFICATION" | "CLUSTERING";
+    title?: string;
+    content: string;
+}
+export declare function getEmbeddingsForText(driver: VertexAIDriver, options: TextEmbeddingsOptions): Promise<EmbeddingsResult>;

package/lib/types/src/vertexai/index.d.ts

@@ -0,0 +1,52 @@
+import { AIModel, AbstractDriver, Completion, CompletionChunkObject, DriverOptions, EmbeddingsResult, ExecutionOptions, ModelSearchPayload, PromptSegment } from "@llumiverse/core";
+import { FetchClient } from "@vertesia/api-fetch-client";
+import { GoogleAuth, GoogleAuthOptions } from "google-auth-library";
+import { JSONClient } from "google-auth-library/build/src/auth/googleauth.js";
+import { EmbeddingsOptions } from "@llumiverse/core";
+import { PredictionServiceClient, v1beta1 } from "@google-cloud/aiplatform";
+import { AnthropicVertex } from "@anthropic-ai/vertex-sdk";
+import { ImagenPrompt } from "./models/imagen.js";
+import { GoogleGenAI, Content, Model } from "@google/genai";
+export interface VertexAIDriverOptions extends DriverOptions {
+    project: string;
+    region: string;
+    googleAuthOptions?: GoogleAuthOptions;
+}
+export interface GenerateContentPrompt {
+    contents: Content[];
+    system?: Content;
+}
+export type VertexAIPrompt = ImagenPrompt | GenerateContentPrompt;
+export declare function trimModelName(model: string): string;
+export declare class VertexAIDriver extends AbstractDriver<VertexAIDriverOptions, VertexAIPrompt> {
+    static PROVIDER: string;
+    provider: string;
+    aiplatform: v1beta1.ModelServiceClient | undefined;
+    anthropicClient: AnthropicVertex | undefined;
+    fetchClient: FetchClient | undefined;
+    googleGenAI: GoogleGenAI | undefined;
+    llamaClient: FetchClient & {
+        region?: string;
+    } | undefined;
+    modelGarden: v1beta1.ModelGardenServiceClient | undefined;
+    imagenClient: PredictionServiceClient | undefined;
+    authClient: JSONClient | GoogleAuth<JSONClient>;
+    constructor(options: VertexAIDriverOptions);
+    getGoogleGenAIClient(region?: string): GoogleGenAI;
+    getFetchClient(): FetchClient;
+    getLLamaClient(region?: string): FetchClient;
+    getAnthropicClient(): AnthropicVertex;
+    getAIPlatformClient(): v1beta1.ModelServiceClient;
+    getModelGardenClient(): v1beta1.ModelGardenServiceClient;
+    getImagenClient(): PredictionServiceClient;
+    validateResult(result: Completion, options: ExecutionOptions): void;
+    protected canStream(options: ExecutionOptions): Promise<boolean>;
+    createPrompt(segments: PromptSegment[], options: ExecutionOptions): Promise<VertexAIPrompt>;
+    requestTextCompletion(prompt: VertexAIPrompt, options: ExecutionOptions): Promise<Completion>;
+    requestTextCompletionStream(prompt: VertexAIPrompt, options: ExecutionOptions): Promise<AsyncIterable<CompletionChunkObject>>;
+    requestImageGeneration(_prompt: ImagenPrompt, _options: ExecutionOptions): Promise<Completion>;
+    getGenAIModelsArray(client: GoogleGenAI): Promise<Model[]>;
+    listModels(_params?: ModelSearchPayload): Promise<AIModel<string>[]>;
+    validateConnection(): Promise<boolean>;
+    generateEmbeddings(options: EmbeddingsOptions): Promise<EmbeddingsResult>;
+}

package/lib/types/src/vertexai/models/claude.d.ts

@@ -0,0 +1,17 @@
+import { ContentBlock, MessageParam, TextBlockParam } from "@anthropic-ai/sdk/resources/index.js";
+import { AIModel, Completion, CompletionChunkObject, ExecutionOptions, PromptSegment, ToolUse } from "@llumiverse/core";
+import { VertexAIDriver } from "../index.js";
+import { ModelDefinition } from "../models.js";
+interface ClaudePrompt {
+    messages: MessageParam[];
+    system?: TextBlockParam[];
+}
+export declare function collectTools(content: ContentBlock[]): ToolUse[] | undefined;
+export declare class ClaudeModelDefinition implements ModelDefinition<ClaudePrompt> {
+    model: AIModel;
+    constructor(modelId: string);
+    createPrompt(_driver: VertexAIDriver, segments: PromptSegment[], options: ExecutionOptions): Promise<ClaudePrompt>;
+    requestTextCompletion(driver: VertexAIDriver, prompt: ClaudePrompt, options: ExecutionOptions): Promise<Completion>;
+    requestTextCompletionStream(driver: VertexAIDriver, prompt: ClaudePrompt, options: ExecutionOptions): Promise<AsyncIterable<CompletionChunkObject>>;
+}
+export {};

package/lib/types/src/vertexai/models/gemini.d.ts

@@ -0,0 +1,17 @@
+import { Content, GenerateContentResponseUsageMetadata } from "@google/genai";
+import { AIModel, Completion, CompletionChunkObject, ExecutionOptions, ExecutionTokenUsage, PromptSegment } from "@llumiverse/core";
+import { VertexAIDriver, GenerateContentPrompt } from "../index.js";
+import { ModelDefinition } from "../models.js";
+export declare function mergeConsecutiveRole(contents: Content[] | undefined): Content[];
+export declare class GeminiModelDefinition implements ModelDefinition<GenerateContentPrompt> {
+    model: AIModel;
+    constructor(modelId: string);
+    preValidationProcessing(result: Completion, options: ExecutionOptions): {
+        result: Completion;
+        options: ExecutionOptions;
+    };
+    createPrompt(_driver: VertexAIDriver, segments: PromptSegment[], options: ExecutionOptions): Promise<GenerateContentPrompt>;
+    usageMetadataToTokenUsage(usageMetadata: GenerateContentResponseUsageMetadata | undefined): ExecutionTokenUsage;
+    requestTextCompletion(driver: VertexAIDriver, prompt: GenerateContentPrompt, options: ExecutionOptions): Promise<Completion>;
+    requestTextCompletionStream(driver: VertexAIDriver, prompt: GenerateContentPrompt, options: ExecutionOptions): Promise<AsyncIterable<CompletionChunkObject>>;
+}

package/lib/types/src/vertexai/models/imagen.d.ts

@@ -0,0 +1,74 @@
+import { AIModel, Completion, ExecutionOptions, PromptSegment } from "@llumiverse/core";
+import { VertexAIDriver } from "../index.js";
+interface ImagenBaseReference {
+    referenceType: "REFERENCE_TYPE_RAW" | "REFERENCE_TYPE_MASK" | "REFERENCE_TYPE_SUBJECT" | "REFERENCE_TYPE_CONTROL" | "REFERENCE_TYPE_STYLE";
+    referenceId: number;
+    referenceImage: {
+        bytesBase64Encoded: string;
+    };
+}
+export declare enum ImagenTaskType {
+    TEXT_IMAGE = "TEXT_IMAGE",
+    EDIT_MODE_INPAINT_REMOVAL = "EDIT_MODE_INPAINT_REMOVAL",
+    EDIT_MODE_INPAINT_INSERTION = "EDIT_MODE_INPAINT_INSERTION",
+    EDIT_MODE_BGSWAP = "EDIT_MODE_BGSWAP",
+    EDIT_MODE_OUTPAINT = "EDIT_MODE_OUTPAINT",
+    CUSTOMIZATION_SUBJECT = "CUSTOMIZATION_SUBJECT",
+    CUSTOMIZATION_STYLE = "CUSTOMIZATION_STYLE",
+    CUSTOMIZATION_CONTROLLED = "CUSTOMIZATION_CONTROLLED",
+    CUSTOMIZATION_INSTRUCT = "CUSTOMIZATION_INSTRUCT"
+}
+export declare enum ImagenMaskMode {
+    MASK_MODE_USER_PROVIDED = "MASK_MODE_USER_PROVIDED",
+    MASK_MODE_BACKGROUND = "MASK_MODE_BACKGROUND",
+    MASK_MODE_FOREGROUND = "MASK_MODE_FOREGROUND",
+    MASK_MODE_SEMANTIC = "MASK_MODE_SEMANTIC"
+}
+interface ImagenReferenceRaw extends ImagenBaseReference {
+    referenceType: "REFERENCE_TYPE_RAW";
+}
+interface ImagenReferenceMask extends Omit<ImagenBaseReference, "referenceImage"> {
+    referenceType: "REFERENCE_TYPE_MASK";
+    maskImageConfig: {
+        maskMode?: ImagenMaskMode;
+        maskClasses?: number[];
+        dilation?: number;
+    };
+    referenceImage?: {
+        bytesBase64Encoded: string;
+    };
+}
+interface ImagenReferenceSubject extends ImagenBaseReference {
+    referenceType: "REFERENCE_TYPE_SUBJECT";
+    subjectImageConfig: {
+        subjectDescription: string;
+        subjectType: "SUBJECT_TYPE_PERSON" | "SUBJECT_TYPE_ANIMAL" | "SUBJECT_TYPE_PRODUCT" | "SUBJECT_TYPE_DEFAULT";
+    };
+}
+interface ImagenReferenceControl extends ImagenBaseReference {
+    referenceType: "REFERENCE_TYPE_CONTROL";
+    controlImageConfig: {
+        controlType: "CONTROL_TYPE_FACE_MESH" | "CONTROL_TYPE_CANNY" | "CONTROL_TYPE_SCRIBBLE";
+        enableControlImageComputation?: boolean;
+    };
+}
+interface ImagenReferenceStyle extends ImagenBaseReference {
+    referenceType: "REFERENCE_TYPE_STYLE";
+    styleImageConfig: {
+        styleDescription?: string;
+    };
+}
+type ImagenMessage = ImagenReferenceRaw | ImagenReferenceMask | ImagenReferenceSubject | ImagenReferenceControl | ImagenReferenceStyle;
+export interface ImagenPrompt {
+    prompt: string;
+    referenceImages?: ImagenMessage[];
+    subjectDescription?: string;
+    negativePrompt?: string;
+}
+export declare class ImagenModelDefinition {
+    model: AIModel;
+    constructor(modelId: string);
+    createPrompt(_driver: VertexAIDriver, segments: PromptSegment[], options: ExecutionOptions): Promise<ImagenPrompt>;
+    requestImageGeneration(driver: VertexAIDriver, prompt: ImagenPrompt, options: ExecutionOptions): Promise<Completion>;
+}
+export {};

package/lib/types/src/vertexai/models/llama.d.ts

@@ -0,0 +1,19 @@
+import { AIModel, Completion, CompletionChunkObject, ExecutionOptions, PromptOptions, PromptSegment } from "@llumiverse/core";
+import { VertexAIDriver } from "../index.js";
+import { ModelDefinition } from "../models.js";
+interface LLamaMessage {
+    role: string;
+    content: string;
+}
+interface LLamaPrompt {
+    messages: LLamaMessage[];
+}
+export declare class LLamaModelDefinition implements ModelDefinition<LLamaPrompt> {
+    model: AIModel;
+    constructor(modelId: string);
+    getLlamaModelRegion(modelName: string): string;
+    createPrompt(_driver: VertexAIDriver, segments: PromptSegment[], options: PromptOptions): Promise<LLamaPrompt>;
+    requestTextCompletion(driver: VertexAIDriver, prompt: LLamaPrompt, options: ExecutionOptions): Promise<Completion>;
+    requestTextCompletionStream(driver: VertexAIDriver, prompt: LLamaPrompt, options: ExecutionOptions): Promise<AsyncIterable<CompletionChunkObject>>;
+}
+export {};

package/lib/types/src/vertexai/models.d.ts

@@ -0,0 +1,14 @@
+import { AIModel, Completion, PromptSegment, ExecutionOptions, CompletionChunkObject } from "@llumiverse/core";
+import { VertexAIDriver } from "./index.js";
+export interface ModelDefinition<PromptT = any> {
+    model: AIModel;
+    versions?: string[];
+    createPrompt: (driver: VertexAIDriver, segments: PromptSegment[], options: ExecutionOptions) => Promise<PromptT>;
+    requestTextCompletion: (driver: VertexAIDriver, prompt: PromptT, options: ExecutionOptions) => Promise<Completion>;
+    requestTextCompletionStream: (driver: VertexAIDriver, prompt: PromptT, options: ExecutionOptions) => Promise<AsyncIterable<CompletionChunkObject>>;
+    preValidationProcessing?(result: Completion, options: ExecutionOptions): {
+        result: Completion;
+        options: ExecutionOptions;
+    };
+}
+export declare function getModelDefinition(model: string): ModelDefinition;

package/lib/types/src/watsonx/index.d.ts

@@ -0,0 +1,26 @@
+import { AIModel, AbstractDriver, Completion, CompletionChunkObject, DriverOptions, EmbeddingsOptions, EmbeddingsResult, ExecutionOptions } from "@llumiverse/core";
+import { FetchClient } from "@vertesia/api-fetch-client";
+import { WatsonAuthToken } from "./interfaces.js";
+interface WatsonxDriverOptions extends DriverOptions {
+    apiKey: string;
+    projectId: string;
+    endpointUrl: string;
+}
+export declare class WatsonxDriver extends AbstractDriver<WatsonxDriverOptions, string> {
+    static PROVIDER: string;
+    provider: string;
+    apiKey: string;
+    endpoint_url: string;
+    projectId: string;
+    authToken?: WatsonAuthToken;
+    fetcher?: FetchClient;
+    fetchClient: FetchClient;
+    constructor(options: WatsonxDriverOptions);
+    requestTextCompletion(prompt: string, options: ExecutionOptions): Promise<Completion>;
+    requestTextCompletionStream(prompt: string, options: ExecutionOptions): Promise<AsyncIterable<CompletionChunkObject>>;
+    listModels(): Promise<AIModel<string>[]>;
+    getAuthToken(): Promise<string>;
+    validateConnection(): Promise<boolean>;
+    generateEmbeddings(options: EmbeddingsOptions): Promise<EmbeddingsResult>;
+}
+export {};

package/lib/types/src/watsonx/interfaces.d.ts

@@ -0,0 +1,64 @@
+export interface WatsonxTextGenerationPayload {
+    model_id: string;
+    input: string;
+    parameters: {
+        max_new_tokens?: number;
+        time_limit?: number;
+        stop_sequences?: string[];
+        temperature?: number;
+        top_k?: number;
+        top_p?: number;
+    };
+    project_id: string;
+}
+export interface WatsonxTextGenerationResponse {
+    model_id: string;
+    created_at: string;
+    results: {
+        generated_text: string;
+        generated_token_count: number;
+        input_token_count: number;
+        stop_reason: string;
+    }[];
+}
+export interface GenerateEmbeddingPayload {
+    model_id: string;
+    inputs: string[];
+    project_id: string;
+}
+export interface GenerateEmbeddingResponse {
+    model_id: string;
+    created_at: string;
+    results: {
+        embedding: number[];
+    }[];
+    input_token_count: number;
+}
+export interface WatsonxModelSpec {
+    model_id: string;
+    label: string;
+    provider: string;
+    source: string;
+    short_description: string;
+    tasks: {
+        id: string;
+        ratings: {
+            quality: number;
+        };
+    }[];
+    min_shot_size: number;
+    tier: string;
+    number_params: string;
+}
+export interface WatsonxListModelResponse {
+    total_count: number;
+    limit: number;
+    resources: WatsonxModelSpec[];
+}
+export interface WatsonAuthToken {
+    access_token: string;
+    refresh_token: string;
+    token_type: string;
+    expire_in: number;
+    expiration: number;
+}

package/lib/types/src/xai/index.d.ts

@@ -0,0 +1,18 @@
+import { AIModel, Completion, DriverOptions, ExecutionOptions, PromptOptions, PromptSegment } from "@llumiverse/core";
+import { FetchClient } from "@vertesia/api-fetch-client";
+import OpenAI from "openai";
+import { BaseOpenAIDriver } from "../openai/index.js";
+export interface xAiDriverOptions extends DriverOptions {
+    apiKey: string;
+    endpoint?: string;
+}
+export declare class xAIDriver extends BaseOpenAIDriver {
+    service: OpenAI;
+    provider: "xai";
+    xai_service: FetchClient;
+    DEFAULT_ENDPOINT: string;
+    constructor(opts: xAiDriverOptions);
+    _formatPrompt(segments: PromptSegment[], opts: PromptOptions): Promise<OpenAI.Chat.Completions.ChatCompletionMessageParam[]>;
+    extractDataFromResponse(_options: ExecutionOptions, result: OpenAI.Chat.Completions.ChatCompletion): Completion;
+    listModels(): Promise<AIModel[]>;
+}

package/lib/types/test-driver/TestErrorCompletionStream.d.ts

@@ -0,0 +1,9 @@
+import { CompletionStream, ExecutionOptions, ExecutionResponse, PromptSegment } from "@llumiverse/core";
+export declare class TestErrorCompletionStream implements CompletionStream<PromptSegment[]> {
+    segments: PromptSegment[];
+    options: ExecutionOptions;
+    completion: ExecutionResponse<PromptSegment[]> | undefined;
+    constructor(segments: PromptSegment[], options: ExecutionOptions);
+    [Symbol.asyncIterator](): AsyncGenerator<string, void, unknown>;
+}
+//# sourceMappingURL=TestErrorCompletionStream.d.ts.map

package/lib/types/test-driver/TestErrorCompletionStream.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"TestErrorCompletionStream.d.ts","sourceRoot":"","sources":["../../../src/test-driver/TestErrorCompletionStream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAGxG,qBAAa,yBAA0B,YAAW,gBAAgB,CAAC,aAAa,EAAE,CAAC;IAI5D,QAAQ,EAAE,aAAa,EAAE;IACjC,OAAO,EAAE,gBAAgB;IAHpC,UAAU,EAAE,iBAAiB,CAAC,aAAa,EAAE,CAAC,GAAG,SAAS,CAAC;gBAExC,QAAQ,EAAE,aAAa,EAAE,EACjC,OAAO,EAAE,gBAAgB;IAE7B,CAAC,MAAM,CAAC,aAAa,CAAC;CAKhC"}

package/lib/types/test-driver/TestValidationErrorCompletionStream.d.ts

@@ -0,0 +1,9 @@
+import { CompletionStream, ExecutionOptions, ExecutionResponse, PromptSegment } from "@llumiverse/core";
+export declare class TestValidationErrorCompletionStream implements CompletionStream<PromptSegment[]> {
+    segments: PromptSegment[];
+    options: ExecutionOptions;
+    completion: ExecutionResponse<PromptSegment[]> | undefined;
+    constructor(segments: PromptSegment[], options: ExecutionOptions);
+    [Symbol.asyncIterator](): AsyncGenerator<"Started TestValidationError.\n" | "chunk1\n" | "chunk2\n", void, unknown>;
+}
+//# sourceMappingURL=TestValidationErrorCompletionStream.d.ts.map

package/lib/types/test-driver/TestValidationErrorCompletionStream.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"TestValidationErrorCompletionStream.d.ts","sourceRoot":"","sources":["../../../src/test-driver/TestValidationErrorCompletionStream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAIxG,qBAAa,mCAAoC,YAAW,gBAAgB,CAAC,aAAa,EAAE,CAAC;IAItE,QAAQ,EAAE,aAAa,EAAE;IACjC,OAAO,EAAE,gBAAgB;IAHpC,UAAU,EAAE,iBAAiB,CAAC,aAAa,EAAE,CAAC,GAAG,SAAS,CAAC;gBAExC,QAAQ,EAAE,aAAa,EAAE,EACjC,OAAO,EAAE,gBAAgB;IAE7B,CAAC,MAAM,CAAC,aAAa,CAAC;CAShC"}

package/lib/types/test-driver/index.d.ts

@@ -0,0 +1,24 @@
+import { AIModel, CompletionStream, Driver, EmbeddingsResult, ExecutionOptions, ExecutionResponse, PromptSegment, TrainingJob } from "@llumiverse/core";
+export * from "./TestErrorCompletionStream.js";
+export * from "./TestValidationErrorCompletionStream.js";
+export declare enum TestDriverModels {
+    executionError = "execution-error",
+    validationError = "validation-error"
+}
+export declare class TestDriver implements Driver<PromptSegment[]> {
+    provider: string;
+    createTrainingPrompt(): Promise<string>;
+    startTraining(): Promise<TrainingJob>;
+    cancelTraining(): Promise<TrainingJob>;
+    getTrainingJob(_jobId: string): Promise<TrainingJob>;
+    createPrompt(segments: PromptSegment[], _opts: ExecutionOptions): Promise<PromptSegment[]>;
+    execute(segments: PromptSegment[], options: ExecutionOptions): Promise<ExecutionResponse<PromptSegment[]>>;
+    stream(segments: PromptSegment[], options: ExecutionOptions): Promise<CompletionStream<PromptSegment[]>>;
+    listTrainableModels(): Promise<AIModel<string>[]>;
+    listModels(): Promise<AIModel<string>[]>;
+    validateConnection(): Promise<boolean>;
+    generateEmbeddings(): Promise<EmbeddingsResult>;
+    executeError(segments: PromptSegment[], _options: ExecutionOptions): Promise<ExecutionResponse<PromptSegment[]>>;
+    executeValidationError(segments: PromptSegment[], _options: ExecutionOptions): Promise<ExecutionResponse<PromptSegment[]>>;
+}
+//# sourceMappingURL=index.d.ts.map

package/lib/types/test-driver/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/test-driver/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAiB,gBAAgB,EAAE,MAAM,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,iBAAiB,EAAa,aAAa,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAKlL,cAAc,gCAAgC,CAAC;AAC/C,cAAc,0CAA0C,CAAC;AAEzD,oBAAY,gBAAgB;IACxB,cAAc,oBAAoB;IAClC,eAAe,qBAAqB;CACvC;AAED,qBAAa,UAAW,YAAW,MAAM,CAAC,aAAa,EAAE,CAAC;IACtD,QAAQ,SAAU;IAElB,oBAAoB,IAAI,OAAO,CAAC,MAAM,CAAC;IAIvC,aAAa,IAAI,OAAO,CAAC,WAAW,CAAC;IAIrC,cAAc,IAAI,OAAO,CAAC,WAAW,CAAC;IAItC,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC;IAI9C,YAAY,CAAC,QAAQ,EAAE,aAAa,EAAE,EAAE,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC;IAIhG,OAAO,CAAC,QAAQ,EAAE,aAAa,EAAE,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,iBAAiB,CAAC,aAAa,EAAE,CAAC,CAAC;IAUpG,MAAM,CAAC,QAAQ,EAAE,aAAa,EAAE,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC;IAWxG,mBAAmB,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;IAIjD,UAAU,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;IAsB9C,kBAAkB,IAAI,OAAO,CAAC,OAAO,CAAC;IAGtC,kBAAkB,IAAI,OAAO,CAAC,gBAAgB,CAAC;IAKzC,YAAY,CAAC,QAAQ,EAAE,aAAa,EAAE,EAAE,QAAQ,EAAE,gBAAgB,GAAG,OAAO,CAAC,iBAAiB,CAAC,aAAa,EAAE,CAAC,CAAC;IAKhH,sBAAsB,CAAC,QAAQ,EAAE,aAAa,EAAE,EAAE,QAAQ,EAAE,gBAAgB,GAAG,OAAO,CAAC,iBAAiB,CAAC,aAAa,EAAE,CAAC,CAAC;CAKnI"}

package/lib/types/test-driver/utils.d.ts

@@ -0,0 +1,5 @@
+import { ExecutionResponse, PromptSegment } from "@llumiverse/core";
+export declare function throwError(message: string, prompt: PromptSegment[]): never;
+export declare function createValidationErrorCompletion(segments: PromptSegment[]): ExecutionResponse<PromptSegment[]>;
+export declare function sleep(ms: number): Promise<unknown>;
+//# sourceMappingURL=utils.d.ts.map

package/lib/types/test-driver/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/test-driver/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAEpE,wBAAgB,UAAU,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,aAAa,EAAE,GAAG,KAAK,CAI1E;AAED,wBAAgB,+BAA+B,CAAC,QAAQ,EAAE,aAAa,EAAE,GAchE,iBAAiB,CAAC,aAAa,EAAE,CAAC,CAC1C;AAED,wBAAgB,KAAK,CAAC,EAAE,EAAE,MAAM,oBAE/B"}

package/lib/types/togetherai/index.d.ts

@@ -1,4 +1,4 @@
-import { AIModel, AbstractDriver, Completion, CompletionChunk, DriverOptions, EmbeddingsResult, ExecutionOptions } from "@llumiverse/core";
+import { AIModel, AbstractDriver, Completion, CompletionChunkObject, DriverOptions, EmbeddingsResult, ExecutionOptions } from "@llumiverse/core";
 import { FetchClient } from "@vertesia/api-fetch-client";
 interface TogetherAIDriverOptions extends DriverOptions {
     apiKey: string;
@@ -13,8 +13,8 @@ export declare class TogetherAIDriver extends AbstractDriver<TogetherAIDriverOpt
         type: string;
         schema: any;
     } | undefined;
-    requestTextCompletion(prompt: string, options: ExecutionOptions): Promise<Completion<any>>;
-    requestTextCompletionStream(prompt: string, options: ExecutionOptions): Promise<AsyncIterable<CompletionChunk>>;
+    requestTextCompletion(prompt: string, options: ExecutionOptions): Promise<Completion>;
+    requestTextCompletionStream(prompt: string, options: ExecutionOptions): Promise<AsyncIterable<CompletionChunkObject>>;
     listModels(): Promise<AIModel<string>[]>;
     validateConnection(): Promise<boolean>;
     generateEmbeddings(): Promise<EmbeddingsResult>;

package/lib/types/togetherai/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/togetherai/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,eAAe,EAAE,aAAa,EAAE,gBAAgB,EAAE,gBAAgB,EAAuB,MAAM,kBAAkB,CAAC;AAEhK,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAGzD,UAAU,uBAAwB,SAAQ,aAAa;IACnD,MAAM,EAAE,MAAM,CAAC;CAClB;AAED,qBAAa,gBAAiB,SAAQ,cAAc,CAAC,uBAAuB,EAAE,MAAM,CAAC;IACjF,MAAM,CAAC,QAAQ,SAAgB;IAC/B,QAAQ,SAA6B;IACrC,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,WAAW,CAAC;gBAEb,OAAO,EAAE,uBAAuB;IAQ5C,iBAAiB,GAAI,SAAS,gBAAgB,KAAG;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,GAAG,CAAA;KAAE,GAAG,SAAS,CAMzF;IAEK,qBAAqB,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;IA0C1F,2BAA2B,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,aAAa,CAAC,eAAe,CAAC,CAAC;IA4C/G,UAAU,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;IAiB9C,kBAAkB,IAAI,OAAO,CAAC,OAAO,CAAC;IAGtC,kBAAkB,IAAI,OAAO,CAAC,gBAAgB,CAAC;CAIlD"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/togetherai/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,qBAAqB,EAAE,aAAa,EAAE,gBAAgB,EAAE,gBAAgB,EAAuB,MAAM,kBAAkB,CAAC;AAEtK,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAGzD,UAAU,uBAAwB,SAAQ,aAAa;IACnD,MAAM,EAAE,MAAM,CAAC;CAClB;AAED,qBAAa,gBAAiB,SAAQ,cAAc,CAAC,uBAAuB,EAAE,MAAM,CAAC;IACjF,MAAM,CAAC,QAAQ,SAAgB;IAC/B,QAAQ,SAA6B;IACrC,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,WAAW,CAAC;gBAEb,OAAO,EAAE,uBAAuB;IAQ5C,iBAAiB,GAAI,SAAS,gBAAgB,KAAG;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,GAAG,CAAA;KAAE,GAAG,SAAS,CAMzF;IAEK,qBAAqB,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC;IA0CrF,2BAA2B,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,aAAa,CAAC,qBAAqB,CAAC,CAAC;IA4CrH,UAAU,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;IAiB9C,kBAAkB,IAAI,OAAO,CAAC,OAAO,CAAC;IAGtC,kBAAkB,IAAI,OAAO,CAAC,gBAAgB,CAAC;CAIlD"}

package/lib/types/vertexai/index.d.ts

@@ -1,12 +1,12 @@
-import { AIModel, AbstractDriver, Completion, CompletionChunk, DriverOptions, EmbeddingsResult, ExecutionOptions, ImageGeneration, ModelSearchPayload, PromptSegment } from "@llumiverse/core";
+import { AIModel, AbstractDriver, Completion, CompletionChunkObject, DriverOptions, EmbeddingsResult, ExecutionOptions, ModelSearchPayload, PromptSegment } from "@llumiverse/core";
 import { FetchClient } from "@vertesia/api-fetch-client";
 import { GoogleAuth, GoogleAuthOptions } from "google-auth-library";
 import { JSONClient } from "google-auth-library/build/src/auth/googleauth.js";
 import { EmbeddingsOptions } from "@llumiverse/core";
-import { v1beta1 } from "@google-cloud/aiplatform";
+import { PredictionServiceClient, v1beta1 } from "@google-cloud/aiplatform";
 import { AnthropicVertex } from "@anthropic-ai/vertex-sdk";
 import { ImagenPrompt } from "./models/imagen.js";
-import { GoogleGenAI, Content } from "@google/genai";
+import { GoogleGenAI, Content, Model } from "@google/genai";
 export interface VertexAIDriverOptions extends DriverOptions {
     project: string;
     region: string;
@@ -29,20 +29,23 @@ export declare class VertexAIDriver extends AbstractDriver<VertexAIDriverOptions
         region?: string;
     } | undefined;
     modelGarden: v1beta1.ModelGardenServiceClient | undefined;
+    imagenClient: PredictionServiceClient | undefined;
     authClient: JSONClient | GoogleAuth<JSONClient>;
     constructor(options: VertexAIDriverOptions);
-    getGoogleGenAIClient(): GoogleGenAI;
+    getGoogleGenAIClient(region?: string): GoogleGenAI;
     getFetchClient(): FetchClient;
     getLLamaClient(region?: string): FetchClient;
     getAnthropicClient(): AnthropicVertex;
     getAIPlatformClient(): v1beta1.ModelServiceClient;
     getModelGardenClient(): v1beta1.ModelGardenServiceClient;
+    getImagenClient(): PredictionServiceClient;
     validateResult(result: Completion, options: ExecutionOptions): void;
     protected canStream(options: ExecutionOptions): Promise<boolean>;
     createPrompt(segments: PromptSegment[], options: ExecutionOptions): Promise<VertexAIPrompt>;
-    requestTextCompletion(prompt: VertexAIPrompt, options: ExecutionOptions): Promise<Completion<any>>;
-    requestTextCompletionStream(prompt: VertexAIPrompt, options: ExecutionOptions): Promise<AsyncIterable<CompletionChunk>>;
-    requestImageGeneration(_prompt: ImagenPrompt, _options: ExecutionOptions): Promise<Completion<ImageGeneration>>;
+    requestTextCompletion(prompt: VertexAIPrompt, options: ExecutionOptions): Promise<Completion>;
+    requestTextCompletionStream(prompt: VertexAIPrompt, options: ExecutionOptions): Promise<AsyncIterable<CompletionChunkObject>>;
+    requestImageGeneration(_prompt: ImagenPrompt, _options: ExecutionOptions): Promise<Completion>;
+    getGenAIModelsArray(client: GoogleGenAI): Promise<Model[]>;
     listModels(_params?: ModelSearchPayload): Promise<AIModel<string>[]>;
     validateConnection(): Promise<boolean>;
     generateEmbeddings(options: EmbeddingsOptions): Promise<EmbeddingsResult>;

package/lib/types/vertexai/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/vertexai/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,OAAO,EACP,cAAc,EACd,UAAU,EACV,eAAe,EACf,aAAa,EACb,gBAAgB,EAChB,gBAAgB,EAChB,eAAe,EAEf,kBAAkB,EAClB,aAAa,EAGhB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAE,UAAU,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACpE,OAAO,EAAE,UAAU,EAAE,MAAM,kDAAkD,CAAC;AAG9E,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAErD,OAAO,EAAE,OAAO,EAAE,MAAM,0BAA0B,CAAC;AACnD,OAAO,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAC3D,OAAO,EAAyB,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACzE,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAErD,MAAM,WAAW,qBAAsB,SAAQ,aAAa;IACxD,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;CACzC;AAED,MAAM,WAAW,qBAAqB;IAClC,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,MAAM,CAAC,EAAE,OAAO,CAAC;CACpB;AAGD,MAAM,MAAM,cAAc,GAAG,YAAY,GAAG,qBAAqB,CAAC;AAElE,wBAAgB,aAAa,CAAC,KAAK,EAAE,MAAM,UAG1C;AAED,qBAAa,cAAe,SAAQ,cAAc,CAAC,qBAAqB,EAAE,cAAc,CAAC;IACrF,MAAM,CAAC,QAAQ,SAAc;IAC7B,QAAQ,SAA2B;IAEnC,UAAU,EAAE,OAAO,CAAC,kBAAkB,GAAG,SAAS,CAAC;IACnD,eAAe,EAAE,eAAe,GAAG,SAAS,CAAC;IAC7C,WAAW,EAAE,WAAW,GAAG,SAAS,CAAC;IACrC,WAAW,EAAE,WAAW,GAAG,SAAS,CAAC;IACrC,WAAW,EAAE,WAAW,GAAG;QAAE,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,SAAS,CAAC;IAC3D,WAAW,EAAE,OAAO,CAAC,wBAAwB,GAAG,SAAS,CAAC;IAE1D,UAAU,EAAE,UAAU,GAAG,UAAU,CAAC,UAAU,CAAC,CAAC;gBAEpC,OAAO,EAAE,qBAAqB;IAanC,oBAAoB,IAAI,WAAW;IAenC,cAAc,IAAI,WAAW;IAe7B,cAAc,CAAC,MAAM,GAAE,MAAsB,GAAG,WAAW;IAkB3D,kBAAkB,IAAI,eAAe;IAYrC,mBAAmB,IAAI,OAAO,CAAC,kBAAkB;IAYjD,oBAAoB,IAAI,OAAO,CAAC,wBAAwB;IAY/D,cAAc,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,gBAAgB;IAY5D,SAAS,CAAC,SAAS,CAAC,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,OAAO,CAAC;IAOzD,YAAY,CAAC,QAAQ,EAAE,aAAa,EAAE,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,cAAc,CAAC;IAO5F,qBAAqB,CAAC,MAAM,EAAE,cAAc,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;IAGlG,2BAA2B,CAC7B,MAAM,EAAE,cAAc,EACtB,OAAO,EAAE,gBAAgB,GAC1B,OAAO,CAAC,aAAa,CAAC,eAAe,CAAC,CAAC;IAIpC,sBAAsB,CACxB,OAAO,EAAE,YAAY,EACrB,QAAQ,EAAE,gBAAgB,GAC3B,OAAO,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;IAMjC,UAAU,CAAC,OAAO,CAAC,EAAE,kBAAkB,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;IAiH1E,kBAAkB,IAAI,OAAO,CAAC,OAAO,CAAC;IAIhC,kBAAkB,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAalF"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/vertexai/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,OAAO,EACP,cAAc,EACd,UAAU,EACV,qBAAqB,EACrB,aAAa,EACb,gBAAgB,EAChB,gBAAgB,EAEhB,kBAAkB,EAClB,aAAa,EAGhB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAE,UAAU,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACpE,OAAO,EAAE,UAAU,EAAE,MAAM,kDAAkD,CAAC;AAG9E,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAErD,OAAO,EAAE,uBAAuB,EAAE,OAAO,EAAE,MAAM,0BAA0B,CAAC;AAC5E,OAAO,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAC3D,OAAO,EAAyB,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACzE,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,eAAe,CAAC;AAE5D,MAAM,WAAW,qBAAsB,SAAQ,aAAa;IACxD,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;CACzC;AAED,MAAM,WAAW,qBAAqB;IAClC,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,MAAM,CAAC,EAAE,OAAO,CAAC;CACpB;AAGD,MAAM,MAAM,cAAc,GAAG,YAAY,GAAG,qBAAqB,CAAC;AAElE,wBAAgB,aAAa,CAAC,KAAK,EAAE,MAAM,UAG1C;AAED,qBAAa,cAAe,SAAQ,cAAc,CAAC,qBAAqB,EAAE,cAAc,CAAC;IACrF,MAAM,CAAC,QAAQ,SAAc;IAC7B,QAAQ,SAA2B;IAEnC,UAAU,EAAE,OAAO,CAAC,kBAAkB,GAAG,SAAS,CAAC;IACnD,eAAe,EAAE,eAAe,GAAG,SAAS,CAAC;IAC7C,WAAW,EAAE,WAAW,GAAG,SAAS,CAAC;IACrC,WAAW,EAAE,WAAW,GAAG,SAAS,CAAC;IACrC,WAAW,EAAE,WAAW,GAAG;QAAE,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,SAAS,CAAC;IAC3D,WAAW,EAAE,OAAO,CAAC,wBAAwB,GAAG,SAAS,CAAC;IAC1D,YAAY,EAAE,uBAAuB,GAAG,SAAS,CAAC;IAElD,UAAU,EAAE,UAAU,GAAG,UAAU,CAAC,UAAU,CAAC,CAAC;gBAEpC,OAAO,EAAE,qBAAqB;IAcnC,oBAAoB,CAAC,MAAM,GAAE,MAA4B,GAAG,WAAW;IA0BvE,cAAc,IAAI,WAAW;IAe7B,cAAc,CAAC,MAAM,GAAE,MAAsB,GAAG,WAAW;IAkB3D,kBAAkB,IAAI,eAAe;IAiBrC,mBAAmB,IAAI,OAAO,CAAC,kBAAkB;IAYjD,oBAAoB,IAAI,OAAO,CAAC,wBAAwB;IAYxD,eAAe,IAAI,uBAAuB;IAajD,cAAc,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,gBAAgB;IAY5D,SAAS,CAAC,SAAS,CAAC,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,OAAO,CAAC;IAOzD,YAAY,CAAC,QAAQ,EAAE,aAAa,EAAE,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,cAAc,CAAC;IAO5F,qBAAqB,CAAC,MAAM,EAAE,cAAc,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC;IAG7F,2BAA2B,CAC7B,MAAM,EAAE,cAAc,EACtB,OAAO,EAAE,gBAAgB,GAC1B,OAAO,CAAC,aAAa,CAAC,qBAAqB,CAAC,CAAC;IAI1C,sBAAsB,CACxB,OAAO,EAAE,YAAY,EACrB,QAAQ,EAAE,gBAAgB,GAC3B,OAAO,CAAC,UAAU,CAAC;IAMhB,mBAAmB,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC;IAS1D,UAAU,CAAC,OAAO,CAAC,EAAE,kBAAkB,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;IA4I1E,kBAAkB,IAAI,OAAO,CAAC,OAAO,CAAC;IAIhC,kBAAkB,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAalF"}

package/lib/types/vertexai/models/claude.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"claude.d.ts","sourceRoot":"","sources":["../../../../src/vertexai/models/claude.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAmE,YAAY,EAAE,cAAc,EAAwB,MAAM,sCAAsC,CAAC;AACzL,OAAO,EACH,OAAO,EAAE,UAAU,EAAE,qBAAqB,EAAE,gBAAgB,EAChD,aAAa,EAAqE,OAAO,EACxG,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAI/C,UAAU,YAAY;IAClB,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,MAAM,CAAC,EAAE,cAAc,EAAE,CAAC;CAC7B;AAWD,wBAAgB,YAAY,CAAC,OAAO,EAAE,YAAY,EAAE,GAAG,OAAO,EAAE,GAAG,SAAS,CAc3E;AA6FD,qBAAa,qBAAsB,YAAW,eAAe,CAAC,YAAY,CAAC;IAEvE,KAAK,EAAE,OAAO,CAAA;gBAEF,OAAO,EAAE,MAAM;IAUrB,YAAY,CAAC,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,aAAa,EAAE,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,YAAY,CAAC;IAuGlH,qBAAqB,CAAC,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,YAAY,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC;IAqCnH,2BAA2B,CAAC,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,YAAY,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,aAAa,CAAC,qBAAqB,CAAC,CAAC;CAiF5J"}
+{"version":3,"file":"claude.d.ts","sourceRoot":"","sources":["../../../../src/vertexai/models/claude.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAmE,YAAY,EAAE,cAAc,EAAwB,MAAM,sCAAsC,CAAC;AACzL,OAAO,EACH,OAAO,EAAE,UAAU,EAAE,qBAAqB,EAAE,gBAAgB,EAChD,aAAa,EAAqE,OAAO,EACxG,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAI/C,UAAU,YAAY;IAClB,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,MAAM,CAAC,EAAE,cAAc,EAAE,CAAC;CAC7B;AAWD,wBAAgB,YAAY,CAAC,OAAO,EAAE,YAAY,EAAE,GAAG,OAAO,EAAE,GAAG,SAAS,CAc3E;AA+FD,qBAAa,qBAAsB,YAAW,eAAe,CAAC,YAAY,CAAC;IAEvE,KAAK,EAAE,OAAO,CAAA;gBAEF,OAAO,EAAE,MAAM;IAUrB,YAAY,CAAC,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,aAAa,EAAE,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,YAAY,CAAC;IAuGlH,qBAAqB,CAAC,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,YAAY,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC;IAqCnH,2BAA2B,CAAC,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,YAAY,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,aAAa,CAAC,qBAAqB,CAAC,CAAC;CAiF5J"}

package/lib/types/vertexai/models/gemini.d.ts

@@ -1,5 +1,5 @@
-import { Content } from "@google/genai";
-import { AIModel, Completion, CompletionChunkObject, ExecutionOptions, PromptSegment } from "@llumiverse/core";
+import { Content, GenerateContentResponseUsageMetadata } from "@google/genai";
+import { AIModel, Completion, CompletionChunkObject, ExecutionOptions, ExecutionTokenUsage, PromptSegment } from "@llumiverse/core";
 import { VertexAIDriver, GenerateContentPrompt } from "../index.js";
 import { ModelDefinition } from "../models.js";
 export declare function mergeConsecutiveRole(contents: Content[] | undefined): Content[];
@@ -11,6 +11,7 @@ export declare class GeminiModelDefinition implements ModelDefinition<GenerateCo
         options: ExecutionOptions;
     };
     createPrompt(_driver: VertexAIDriver, segments: PromptSegment[], options: ExecutionOptions): Promise<GenerateContentPrompt>;
+    usageMetadataToTokenUsage(usageMetadata: GenerateContentResponseUsageMetadata | undefined): ExecutionTokenUsage;
     requestTextCompletion(driver: VertexAIDriver, prompt: GenerateContentPrompt, options: ExecutionOptions): Promise<Completion>;
     requestTextCompletionStream(driver: VertexAIDriver, prompt: GenerateContentPrompt, options: ExecutionOptions): Promise<AsyncIterable<CompletionChunkObject>>;
 }

package/lib/types/vertexai/models/gemini.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../../../src/vertexai/models/gemini.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,OAAO,EAEV,MAAM,eAAe,CAAC;AACvB,OAAO,EACH,OAAO,EAAE,UAAU,EAAE,qBAAqB,EAAE,gBAAgB,EAE5D,aAAa,EAChB,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAE,cAAc,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AACpE,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AA2a/C,wBAAgB,oBAAoB,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,SAAS,GAAG,OAAO,EAAE,CA2B/E;AAED,qBAAa,qBAAsB,YAAW,eAAe,CAAC,qBAAqB,CAAC;IAEhF,KAAK,EAAE,OAAO,CAAA;gBAEF,OAAO,EAAE,MAAM;IAU3B,uBAAuB,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,gBAAgB,GAAG;QAAE,MAAM,EAAE,UAAU,CAAC;QAAC,OAAO,EAAE,gBAAgB,CAAA;KAAE;IAiBnH,YAAY,CAAC,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,aAAa,EAAE,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,qBAAqB,CAAC;IAgH3H,qBAAqB,CAAC,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,qBAAqB,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC;IAsD5H,2BAA2B,CAAC,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,qBAAqB,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,aAAa,CAAC,qBAAqB,CAAC,CAAC;CAmDrK"}
+{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../../../src/vertexai/models/gemini.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,OAAO,EACP,oCAAoC,EAEvC,MAAM,eAAe,CAAC;AACvB,OAAO,EACH,OAAO,EAAE,UAAU,EAAE,qBAAqB,EAAoB,gBAAgB,EAC9E,mBAAmB,EACnB,aAAa,EAEhB,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAE,cAAc,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AACpE,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AA2c/C,wBAAgB,oBAAoB,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,SAAS,GAAG,OAAO,EAAE,CA2B/E;AAqCD,qBAAa,qBAAsB,YAAW,eAAe,CAAC,qBAAqB,CAAC;IAEhF,KAAK,EAAE,OAAO,CAAA;gBAEF,OAAO,EAAE,MAAM;IAU3B,uBAAuB,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,gBAAgB,GAAG;QAAE,MAAM,EAAE,UAAU,CAAC;QAAC,OAAO,EAAE,gBAAgB,CAAA;KAAE;IAyBnH,YAAY,CAAC,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,aAAa,EAAE,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,qBAAqB,CAAC;IAgHjI,yBAAyB,CAAC,aAAa,EAAE,oCAAoC,GAAG,SAAS,GAAG,mBAAmB;IA0BzG,qBAAqB,CAAC,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,qBAAqB,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC;IAoE5H,2BAA2B,CAAC,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,qBAAqB,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,aAAa,CAAC,qBAAqB,CAAC,CAAC;CA8DrK"}

package/lib/types/vertexai/models/imagen.d.ts

@@ -1,4 +1,4 @@
-import { AIModel, Completion, ExecutionOptions, ImageGeneration, PromptSegment } from "@llumiverse/core";
+import { AIModel, Completion, ExecutionOptions, PromptSegment } from "@llumiverse/core";
 import { VertexAIDriver } from "../index.js";
 interface ImagenBaseReference {
     referenceType: "REFERENCE_TYPE_RAW" | "REFERENCE_TYPE_MASK" | "REFERENCE_TYPE_SUBJECT" | "REFERENCE_TYPE_CONTROL" | "REFERENCE_TYPE_STYLE";
@@ -69,7 +69,7 @@ export declare class ImagenModelDefinition {
     model: AIModel;
     constructor(modelId: string);
     createPrompt(_driver: VertexAIDriver, segments: PromptSegment[], options: ExecutionOptions): Promise<ImagenPrompt>;
-    requestImageGeneration(driver: VertexAIDriver, prompt: ImagenPrompt, options: ExecutionOptions): Promise<Completion<ImageGeneration>>;
+    requestImageGeneration(driver: VertexAIDriver, prompt: ImagenPrompt, options: ExecutionOptions): Promise<Completion>;
 }
 export {};
 //# sourceMappingURL=imagen.d.ts.map