@truefoundry/ai-gateway-sdk 0.0.0-stage → 0.0.1

package/dist/providers/google-vertex-ai/types.d.ts

@@ -0,0 +1,365 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * But needs some custom changes for TFY
+ * 1. Add VertexQwenChatCompleteStreamChunk
+ * ================================
+ */
+import { Message, ToolCall } from '../../types/requestBody';
+import { ChatCompletionResponse, GroundingMetadata } from '../types';
+export interface GoogleErrorResponse {
+    error: {
+        code: number;
+        message: string;
+        status: string;
+        details: Array<Record<string, any>>;
+    };
+}
+export interface GoogleGenerateFunctionCall {
+    name: string;
+    args: Record<string, any>;
+}
+export interface GoogleResponseCandidate {
+    content: {
+        parts: {
+            text?: string;
+            thought?: boolean;
+            functionCall?: GoogleGenerateFunctionCall;
+            inlineData?: {
+                mimeType: string;
+                data: string;
+            };
+            thoughtSignature?: string;
+        }[];
+    };
+    logprobsResult?: {
+        topCandidates: [
+            {
+                candidates: [
+                    {
+                        token: string;
+                        logProbability: number;
+                    }
+                ];
+            }
+        ];
+        chosenCandidates: [
+            {
+                token: string;
+                logProbability: number;
+            }
+        ];
+    };
+    finishReason: string;
+    index: 0;
+    safetyRatings: {
+        category: string;
+        probability: string;
+    }[];
+    groundingMetadata?: GroundingMetadata;
+}
+export interface GoogleGenerateContentResponse {
+    modelVersion: string;
+    candidates: GoogleResponseCandidate[];
+    promptFeedback: {
+        safetyRatings: {
+            category: string;
+            probability: string;
+            probabilityScore: number;
+            severity: string;
+            severityScore: number;
+        }[];
+    };
+    usageMetadata: {
+        promptTokenCount: number;
+        candidatesTokenCount: number;
+        totalTokenCount: number;
+        thoughtsTokenCount?: number;
+        cachedContentTokenCount?: number;
+        promptTokensDetails: {
+            modality: VERTEX_MODALITY;
+            tokenCount: number;
+        }[];
+        candidatesTokensDetails: {
+            modality: VERTEX_MODALITY;
+            tokenCount: number;
+        }[];
+    };
+}
+export interface VertexLLamaChatCompleteResponse extends Omit<ChatCompletionResponse, 'id' | 'created'> {
+}
+export interface VertexLlamaChatCompleteStreamChunk {
+    choices: {
+        delta: {
+            content: string;
+            role: string;
+        };
+        finish_reason?: string;
+        index: 0;
+    }[];
+    model: string;
+    object: string;
+    usage?: {
+        completion_tokens: number;
+        prompt_tokens: number;
+        total_tokens: number;
+    };
+    id?: string;
+    created?: number;
+    provider?: string;
+}
+/**
+ * ================================
+ * NOTE: Add VertexQwenChatCompleteStreamChunk
+ * ================================
+ */
+export interface VertexQwenChatCompleteStreamChunk {
+    choices: {
+        delta: {
+            content: string;
+            role: string;
+        };
+        finish_reason?: string;
+        index: 0;
+    }[];
+    model: string;
+    object: string;
+    usage?: {
+        completion_tokens: number;
+        prompt_tokens: number;
+        total_tokens: number;
+    };
+    id?: string;
+    created?: number;
+    provider?: string;
+}
+export type EmbedInstancesData = TextEmbedInstance | MultimodalEmbedInstance;
+export interface TextEmbedInstance {
+    task_type: string;
+    content: string;
+}
+export interface MultimodalEmbedInstance {
+    image?: {
+        gcsUri?: string;
+        bytesBase64Encoded?: string;
+    };
+    text?: string;
+    video?: {
+        gcsUri?: string;
+        bytesBase64Encoded?: string;
+        videoSegmentConfig?: {
+            startOffsetSec?: number;
+            endOffsetSec?: number;
+            intervalSec?: number;
+        };
+    };
+}
+interface EmbedPredictionsResponse {
+    embeddings: {
+        values: number[];
+        statistics: {
+            truncated: string;
+            token_count: number;
+        };
+    };
+    imageEmbedding?: number[];
+    textEmbedding?: number[];
+    videoEmbeddings?: {
+        embedding: number[];
+        endOffsetSec: number;
+        startOffsetSec: number;
+    }[];
+}
+export interface GoogleEmbedResponse {
+    predictions: EmbedPredictionsResponse[];
+    metadata: {
+        billableCharacterCount: number;
+    };
+}
+/**
+ * Response shape for Vertex AI Gemini `:embedContent` endpoint
+ * (used by gemini-embedding-2* models via `isEmbedContentModel`).
+ * Returns a single embedding per call.
+ *
+ * @see https://docs.cloud.google.com/vertex-ai/generative-ai/docs/reference/rest/v1beta1/projects.locations.publishers.models/embedContent
+ */
+export interface GoogleEmbedContentResponse {
+    embedding: {
+        values: number[];
+    };
+    usageMetadata?: {
+        promptTokenCount?: number;
+        totalTokenCount?: number;
+        candidatesTokenCount?: number;
+        cachedContentTokenCount?: number;
+    };
+    truncated?: boolean;
+}
+export interface GoogleSearchRetrievalTool {
+    googleSearchRetrieval: {
+        dynamicRetrievalConfig?: {
+            mode: string;
+            dynamicThreshold?: string;
+        };
+    };
+}
+type GoogleBatchJobStatus = 'JOB_STATE_UNSPECIFIED' | 'JOB_STATE_QUEUED' | 'JOB_STATE_PENDING' | 'JOB_STATE_RUNNING' | 'JOB_STATE_SUCCEEDED' | 'JOB_STATE_FAILED' | 'JOB_STATE_CANCELLING' | 'JOB_STATE_CANCELLED' | 'JOB_STATE_PAUSED' | 'JOB_STATE_EXPIRED' | 'JOB_STATE_UPDATING' | 'JOB_STATE_PARTIALLY_SUCCEEDED';
+export interface GoogleBatchRecord {
+    /**
+     * @example projects/562188160088/locations/us-east4/batchPredictionJobs/{id}
+     */
+    name: string;
+    displayName: string;
+    /**
+     * @example projects/562188160088/locations/us-east4/models/{model}
+     */
+    model: string;
+    inputConfig: {
+        instancesFormat: 'jsonl';
+        gcsSource: {
+            uris: string;
+        };
+    };
+    outputConfig: {
+        predictionsFormat: 'jsonl';
+        gcsDestination: {
+            outputUriPrefix: string;
+        };
+    };
+    outputInfo?: {
+        gcsOutputDirectory: string;
+    };
+    state: GoogleBatchJobStatus;
+    createTime: string;
+    updateTime: string;
+    modelVersionId: string;
+    error?: {
+        code: string;
+        message: string;
+    };
+    startTime: string;
+    endTime: string;
+    completionStats?: {
+        successfulCount: string;
+        failedCount: string;
+        incompleteCount: string;
+        successfulForecastPointCount: string;
+    };
+}
+export interface GoogleFinetuneRecord {
+    name: string;
+    state: GoogleBatchJobStatus;
+    tunedModelDisplayName: string;
+    description: string;
+    createTime: string;
+    startTime: string;
+    endTime: string;
+    updateTime: string;
+    error: string;
+    tunedModel?: {
+        model: string;
+        endpoint: string;
+    };
+    tuningDataStats?: {
+        supervisedTuningDataStats: {
+            tuningDatasetExampleCount: number;
+            totalTuningCharacterCount: number;
+            totalBillableTokenCount: number;
+            tuningStepCount: number;
+            userInputTokenDistribution: number;
+        };
+    };
+    baseModel: string;
+    source_model?: {
+        baseModel: string;
+    };
+    supervisedTuningSpec: {
+        trainingDatasetUri: string;
+        validationDatasetUri: string;
+        hyperParameters: {
+            learningRateMultiplier: number;
+            epochCount: number;
+            adapterSize: number;
+        };
+    };
+}
+export declare enum VERTEX_GEMINI_GENERATE_CONTENT_FINISH_REASON {
+    FINISH_REASON_UNSPECIFIED = "FINISH_REASON_UNSPECIFIED",
+    STOP = "STOP",
+    MAX_TOKENS = "MAX_TOKENS",
+    SAFETY = "SAFETY",
+    RECITATION = "RECITATION",
+    OTHER = "OTHER",
+    BLOCKLIST = "BLOCKLIST",
+    PROHIBITED_CONTENT = "PROHIBITED_CONTENT",
+    SPII = "SPII"
+}
+export declare enum VERTEX_MODALITY {
+    MODALITY_UNSPECIFIED = "MODALITY_UNSPECIFIED",
+    TEXT = "TEXT",
+    IMAGE = "IMAGE",
+    AUDIO = "AUDIO"
+}
+/** Error response shape returned by Vertex AI's OpenAI-compatible endpoint. */
+export type OpenAICompatibleErrorResponse = Array<{
+    error: {
+        message: string;
+        status: string;
+        code: string | number;
+    };
+}>;
+/** Success response shape returned by Vertex AI's OpenAI-compatible endpoint.
+ *  Extends ChatCompletionResponse to allow pass-through of extra provider fields. */
+export type OpenAICompatibleSuccessResponse = ChatCompletionResponse & Record<string, unknown>;
+/**
+ * Gateway-extended ToolCall that carries optional thinking-signature fields used
+ * for multi-turn round-trips with Gemini and Bedrock thinking models.
+ *
+ * All extra fields are optional so any plain ToolCall is structurally compatible.
+ */
+export interface GatewayToolCall extends ToolCall {
+    /** Thought signature stored at the tool-call level (primary location). */
+    provider_specific_fields?: {
+        thought_signature?: string;
+    };
+    /** Extended function metadata including per-function thought signature fields. */
+    function: ToolCall['function'] & {
+        /** Thought signature stored at the function level (secondary location). */
+        provider_specific_fields?: {
+            thought_signature?: string;
+        };
+        /** Legacy: thought_signature stored directly on function (previous gateway format). */
+        thought_signature?: string;
+    };
+}
+/**
+ * Gateway-extended Message that carries optional thinking-block fields used for
+ * multi-turn round-trips with Gemini and Bedrock thinking models.
+ *
+ * All extra fields are optional so any plain Message is structurally compatible.
+ */
+export interface GatewayMessage extends Message {
+    /**
+     * Thinking blocks from the previous assistant turn.
+     * Each block is either a visible thinking block (with text + signature) or a
+     * redacted block (with opaque data). Used to reconstruct thoughtSignature /
+     * reasoningContent in the next request.
+     */
+    thinking_blocks?: Array<{
+        type: 'thinking' | 'redacted_thinking';
+        thinking?: string;
+        signature?: string;
+        data?: string;
+    }>;
+    /**
+     * Provider-specific metadata attached to the message.
+     * For Gemini: `thought_signatures` is an array of base64 signatures, one per
+     * function-call part, used when provider_specific_fields is preserved by the client.
+     */
+    provider_specific_fields?: {
+        thought_signatures?: string[];
+        [key: string]: unknown;
+    };
+}
+export {};
+//# sourceMappingURL=types.d.ts.map

package/dist/providers/google-vertex-ai/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/providers/google-vertex-ai/types.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EAAE,sBAAsB,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAErE,MAAM,WAAW,mBAAmB;IAClC,KAAK,EAAE;QACL,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,EAAE,MAAM,CAAC;QACf,OAAO,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC;KACrC,CAAC;CACH;AAED,MAAM,WAAW,0BAA0B;IACzC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC3B;AAED,MAAM,WAAW,uBAAuB;IACtC,OAAO,EAAE;QACP,KAAK,EAAE;YACL,IAAI,CAAC,EAAE,MAAM,CAAC;YACd,OAAO,CAAC,EAAE,OAAO,CAAC;YAClB,YAAY,CAAC,EAAE,0BAA0B,CAAC;YAC1C,UAAU,CAAC,EAAE;gBACX,QAAQ,EAAE,MAAM,CAAC;gBACjB,IAAI,EAAE,MAAM,CAAC;aACd,CAAC;YACF,gBAAgB,CAAC,EAAE,MAAM,CAAC;SAC3B,EAAE,CAAC;KACL,CAAC;IACF,cAAc,CAAC,EAAE;QACf,aAAa,EAAE;YACb;gBACE,UAAU,EAAE;oBACV;wBACE,KAAK,EAAE,MAAM,CAAC;wBACd,cAAc,EAAE,MAAM,CAAC;qBACxB;iBACF,CAAC;aACH;SACF,CAAC;QACF,gBAAgB,EAAE;YAChB;gBACE,KAAK,EAAE,MAAM,CAAC;gBACd,cAAc,EAAE,MAAM,CAAC;aACxB;SACF,CAAC;KACH,CAAC;IACF,YAAY,EAAE,MAAM,CAAC;IACrB,KAAK,EAAE,CAAC,CAAC;IACT,aAAa,EAAE;QACb,QAAQ,EAAE,MAAM,CAAC;QACjB,WAAW,EAAE,MAAM,CAAC;KACrB,EAAE,CAAC;IACJ,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;CACvC;AAED,MAAM,WAAW,6BAA6B;IAC5C,YAAY,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,uBAAuB,EAAE,CAAC;IACtC,cAAc,EAAE;QACd,aAAa,EAAE;YACb,QAAQ,EAAE,MAAM,CAAC;YACjB,WAAW,EAAE,MAAM,CAAC;YACpB,gBAAgB,EAAE,MAAM,CAAC;YACzB,QAAQ,EAAE,MAAM,CAAC;YACjB,aAAa,EAAE,MAAM,CAAC;SACvB,EAAE,CAAC;KACL,CAAC;IACF,aAAa,EAAE;QACb,gBAAgB,EAAE,MAAM,CAAC;QACzB,oBAAoB,EAAE,MAAM,CAAC;QAC7B,eAAe,EAAE,MAAM,CAAC;QACxB,kBAAkB,CAAC,EAAE,MAAM,CAAC;QAC5B,uBAAuB,CAAC,EAAE,MAAM,CAAC;QACjC,mBAAmB,EAAE;YACnB,QAAQ,EAAE,eAAe,CAAC;YAC1B,UAAU,EAAE,MAAM,CAAC;SACpB,EAAE,CAAC;QACJ,uBAAuB,EAAE;YACvB,QAAQ,EAAE,eAAe,CAAC;YAC1B,UAAU,EAAE,MAAM,CAAC;SACpB,EAAE,CAAC;KACL,CAAC;CACH;AAED,MAAM,WAAW,+BAAgC,SAAQ,IAAI,CAAC,sBAAsB,EAAE,IAAI,GAAG,SAAS,CAAC;CAAG;AAE1G,MAAM,WAAW,kCAAkC;IACjD,OAAO,EAAE;QACP,KAAK,EAAE;YACL,OAAO,EAAE,MAAM,CAAC;YAChB,IAAI,EAAE,MAAM,CAAC;SACd,CAAC;QACF,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,KAAK,EAAE,CAAC,CAAC;KACV,EAAE,CAAC;IACJ,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE;QACN,iBAAiB,EAAE,MAAM,CAAC;QAC1B,aAAa,EAAE,MAAM,CAAC;QACtB,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;IACF,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED;;;;GAIG;AACH,MAAM,WAAW,iCAAiC;IAChD,OAAO,EAAE;QACP,KAAK,EAAE;YACL,OAAO,EAAE,MAAM,CAAC;YAChB,IAAI,EAAE,MAAM,CAAC;SACd,CAAC;QACF,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,KAAK,EAAE,CAAC,CAAC;KACV,EAAE,CAAC;IACJ,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE;QACN,iBAAiB,EAAE,MAAM,CAAC;QAC1B,aAAa,EAAE,MAAM,CAAC;QACtB,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;IACF,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,MAAM,kBAAkB,GAAG,iBAAiB,GAAG,uBAAuB,CAAC;AAE7E,MAAM,WAAW,iBAAiB;IAChC,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;CACjB;AACD,MAAM,WAAW,uBAAuB;IACtC,KAAK,CAAC,EAAE;QACN,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,kBAAkB,CAAC,EAAE,MAAM,CAAC;KAC7B,CAAC;IACF,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE;QACN,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,kBAAkB,CAAC,EAAE,MAAM,CAAC;QAC5B,kBAAkB,CAAC,EAAE;YACnB,cAAc,CAAC,EAAE,MAAM,CAAC;YACxB,YAAY,CAAC,EAAE,MAAM,CAAC;YACtB,WAAW,CAAC,EAAE,MAAM,CAAC;SACtB,CAAC;KACH,CAAC;CACH;AAED,UAAU,wBAAwB;IAChC,UAAU,EAAE;QACV,MAAM,EAAE,MAAM,EAAE,CAAC;QACjB,UAAU,EAAE;YACV,SAAS,EAAE,MAAM,CAAC;YAClB,WAAW,EAAE,MAAM,CAAC;SACrB,CAAC;KACH,CAAC;IACF,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1B,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,eAAe,CAAC,EAAE;QAChB,SAAS,EAAE,MAAM,EAAE,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,cAAc,EAAE,MAAM,CAAC;KACxB,EAAE,CAAC;CACL;AAED,MAAM,WAAW,mBAAmB;IAClC,WAAW,EAAE,wBAAwB,EAAE,CAAC;IACxC,QAAQ,EAAE;QACR,sBAAsB,EAAE,MAAM,CAAC;KAChC,CAAC;CACH;AAED;;;;;;GAMG;AACH,MAAM,WAAW,0BAA0B;IACzC,SAAS,EAAE;QACT,MAAM,EAAE,MAAM,EAAE,CAAC;KAClB,CAAC;IACF,aAAa,CAAC,EAAE;QACd,gBAAgB,CAAC,EAAE,MAAM,CAAC;QAC1B,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,oBAAoB,CAAC,EAAE,MAAM,CAAC;QAC9B,uBAAuB,CAAC,EAAE,MAAM,CAAC;KAClC,CAAC;IACF,SAAS,CAAC,EAAE,OAAO,CAAC;CACrB;AAED,MAAM,WAAW,yBAAyB;IACxC,qBAAqB,EAAE;QACrB,sBAAsB,CAAC,EAAE;YACvB,IAAI,EAAE,MAAM,CAAC;YACb,gBAAgB,CAAC,EAAE,MAAM,CAAC;SAC3B,CAAC;KACH,CAAC;CACH;AAED,KAAK,oBAAoB,GACrB,uBAAuB,GACvB,kBAAkB,GAClB,mBAAmB,GACnB,mBAAmB,GACnB,qBAAqB,GACrB,kBAAkB,GAClB,sBAAsB,GACtB,qBAAqB,GACrB,kBAAkB,GAClB,mBAAmB,GACnB,oBAAoB,GACpB,+BAA+B,CAAC;AAEpC,MAAM,WAAW,iBAAiB;IAChC;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE;QACX,eAAe,EAAE,OAAO,CAAC;QACzB,SAAS,EAAE;YACT,IAAI,EAAE,MAAM,CAAC;SACd,CAAC;KACH,CAAC;IACF,YAAY,EAAE;QACZ,iBAAiB,EAAE,OAAO,CAAC;QAC3B,cAAc,EAAE;YACd,eAAe,EAAE,MAAM,CAAC;SACzB,CAAC;KACH,CAAC;IACF,UAAU,CAAC,EAAE;QACX,kBAAkB,EAAE,MAAM,CAAC;KAC5B,CAAC;IACF,KAAK,EAAE,oBAAoB,CAAC;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,cAAc,EAAE,MAAM,CAAC;IACvB,KAAK,CAAC,EAAE;QACN,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;KACjB,CAAC;IACF,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,eAAe,CAAC,EAAE;QAChB,eAAe,EAAE,MAAM,CAAC;QACxB,WAAW,EAAE,MAAM,CAAC;QACpB,eAAe,EAAE,MAAM,CAAC;QACxB,4BAA4B,EAAE,MAAM,CAAC;KACtC,CAAC;CACH;AAED,MAAM,WAAW,oBAAoB;IACnC,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,oBAAoB,CAAC;IAC5B,qBAAqB,EAAE,MAAM,CAAC;IAC9B,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE;QACX,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,MAAM,CAAC;KAClB,CAAC;IACF,eAAe,CAAC,EAAE;QAChB,yBAAyB,EAAE;YACzB,yBAAyB,EAAE,MAAM,CAAC;YAClC,yBAAyB,EAAE,MAAM,CAAC;YAClC,uBAAuB,EAAE,MAAM,CAAC;YAChC,eAAe,EAAE,MAAM,CAAC;YACxB,0BAA0B,EAAE,MAAM,CAAC;SACpC,CAAC;KACH,CAAC;IACF,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,CAAC,EAAE;QACb,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;IACF,oBAAoB,EAAE;QACpB,kBAAkB,EAAE,MAAM,CAAC;QAC3B,oBAAoB,EAAE,MAAM,CAAC;QAC7B,eAAe,EAAE;YACf,sBAAsB,EAAE,MAAM,CAAC;YAC/B,UAAU,EAAE,MAAM,CAAC;YACnB,WAAW,EAAE,MAAM,CAAC;SACrB,CAAC;KACH,CAAC;CACH;AAED,oBAAY,4CAA4C;IACtD,yBAAyB,8BAA8B;IACvD,IAAI,SAAS;IACb,UAAU,eAAe;IACzB,MAAM,WAAW;IACjB,UAAU,eAAe;IACzB,KAAK,UAAU;IACf,SAAS,cAAc;IACvB,kBAAkB,uBAAuB;IACzC,IAAI,SAAS;CACd;AAED,oBAAY,eAAe;IACzB,oBAAoB,yBAAyB;IAC7C,IAAI,SAAS;IACb,KAAK,UAAU;IACf,KAAK,UAAU;CAChB;AAED,+EAA+E;AAC/E,MAAM,MAAM,6BAA6B,GAAG,KAAK,CAAC;IAChD,KAAK,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAAA;KAAE,CAAC;CACnE,CAAC,CAAC;AAEH;qFACqF;AACrF,MAAM,MAAM,+BAA+B,GAAG,sBAAsB,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAE/F;;;;;GAKG;AACH,MAAM,WAAW,eAAgB,SAAQ,QAAQ;IAC/C,0EAA0E;IAC1E,wBAAwB,CAAC,EAAE;QACzB,iBAAiB,CAAC,EAAE,MAAM,CAAC;KAC5B,CAAC;IACF,kFAAkF;IAClF,QAAQ,EAAE,QAAQ,CAAC,UAAU,CAAC,GAAG;QAC/B,2EAA2E;QAC3E,wBAAwB,CAAC,EAAE;YACzB,iBAAiB,CAAC,EAAE,MAAM,CAAC;SAC5B,CAAC;QACF,uFAAuF;QACvF,iBAAiB,CAAC,EAAE,MAAM,CAAC;KAC5B,CAAC;CACH;AAED;;;;;GAKG;AACH,MAAM,WAAW,cAAe,SAAQ,OAAO;IAC7C;;;;;OAKG;IACH,eAAe,CAAC,EAAE,KAAK,CAAC;QACtB,IAAI,EAAE,UAAU,GAAG,mBAAmB,CAAC;QACvC,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,IAAI,CAAC,EAAE,MAAM,CAAC;KACf,CAAC,CAAC;IACH;;;;OAIG;IACH,wBAAwB,CAAC,EAAE;QACzB,kBAAkB,CAAC,EAAE,MAAM,EAAE,CAAC;QAC9B,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;KACxB,CAAC;CACH"}

package/dist/providers/google-vertex-ai/types.js

@@ -0,0 +1,20 @@
+export var VERTEX_GEMINI_GENERATE_CONTENT_FINISH_REASON;
+(function (VERTEX_GEMINI_GENERATE_CONTENT_FINISH_REASON) {
+    VERTEX_GEMINI_GENERATE_CONTENT_FINISH_REASON["FINISH_REASON_UNSPECIFIED"] = "FINISH_REASON_UNSPECIFIED";
+    VERTEX_GEMINI_GENERATE_CONTENT_FINISH_REASON["STOP"] = "STOP";
+    VERTEX_GEMINI_GENERATE_CONTENT_FINISH_REASON["MAX_TOKENS"] = "MAX_TOKENS";
+    VERTEX_GEMINI_GENERATE_CONTENT_FINISH_REASON["SAFETY"] = "SAFETY";
+    VERTEX_GEMINI_GENERATE_CONTENT_FINISH_REASON["RECITATION"] = "RECITATION";
+    VERTEX_GEMINI_GENERATE_CONTENT_FINISH_REASON["OTHER"] = "OTHER";
+    VERTEX_GEMINI_GENERATE_CONTENT_FINISH_REASON["BLOCKLIST"] = "BLOCKLIST";
+    VERTEX_GEMINI_GENERATE_CONTENT_FINISH_REASON["PROHIBITED_CONTENT"] = "PROHIBITED_CONTENT";
+    VERTEX_GEMINI_GENERATE_CONTENT_FINISH_REASON["SPII"] = "SPII";
+})(VERTEX_GEMINI_GENERATE_CONTENT_FINISH_REASON || (VERTEX_GEMINI_GENERATE_CONTENT_FINISH_REASON = {}));
+export var VERTEX_MODALITY;
+(function (VERTEX_MODALITY) {
+    VERTEX_MODALITY["MODALITY_UNSPECIFIED"] = "MODALITY_UNSPECIFIED";
+    VERTEX_MODALITY["TEXT"] = "TEXT";
+    VERTEX_MODALITY["IMAGE"] = "IMAGE";
+    VERTEX_MODALITY["AUDIO"] = "AUDIO";
+})(VERTEX_MODALITY || (VERTEX_MODALITY = {}));
+//# sourceMappingURL=types.js.map

package/dist/providers/google-vertex-ai/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/providers/google-vertex-ai/types.ts"],"names":[],"mappings":"AAyTA,MAAM,CAAN,IAAY,4CAUX;AAVD,WAAY,4CAA4C;IACtD,uGAAuD,CAAA;IACvD,6DAAa,CAAA;IACb,yEAAyB,CAAA;IACzB,iEAAiB,CAAA;IACjB,yEAAyB,CAAA;IACzB,+DAAe,CAAA;IACf,uEAAuB,CAAA;IACvB,yFAAyC,CAAA;IACzC,6DAAa,CAAA;AACf,CAAC,EAVW,4CAA4C,KAA5C,4CAA4C,QAUvD;AAED,MAAM,CAAN,IAAY,eAKX;AALD,WAAY,eAAe;IACzB,gEAA6C,CAAA;IAC7C,gCAAa,CAAA;IACb,kCAAe,CAAA;IACf,kCAAe,CAAA;AACjB,CAAC,EALW,eAAe,KAAf,eAAe,QAK1B"}

package/dist/providers/google-vertex-ai/uploadFile.d.ts

@@ -0,0 +1,8 @@
+import { BatchEndpoints } from '../../globals';
+import { ProviderConfig, RequestHandler } from '../types';
+/** Embeddings JSONL uses the embed config; everything else stays on chat completions. */
+export declare function resolveVertexUploadBatchEndpoint(modelName?: string): BatchEndpoints;
+export declare function vertexUploadProviderConfig(provider: string, modelName?: string): ProviderConfig | undefined;
+export declare const GoogleFileUploadRequestHandler: RequestHandler<ReadableStream>;
+export declare const GoogleFileUploadResponseTransform: (response: Response) => Response;
+//# sourceMappingURL=uploadFile.d.ts.map

package/dist/providers/google-vertex-ai/uploadFile.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"uploadFile.d.ts","sourceRoot":"","sources":["../../../src/providers/google-vertex-ai/uploadFile.ts"],"names":[],"mappings":"AAWA,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAO/C,OAAO,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAmC1D,yFAAyF;AACzF,wBAAgB,gCAAgC,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,cAAc,CAEnF;AAED,wBAAgB,0BAA0B,CAAC,QAAQ,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,cAAc,GAAG,SAAS,CAI3G;AAED,eAAO,MAAM,8BAA8B,EAAE,cAAc,CAAC,cAAc,CAsLzE,CAAC;AAEF,eAAO,MAAM,iCAAiC,GAAI,UAAU,QAAQ,aAEnE,CAAC"}

package/dist/providers/google-vertex-ai/uploadFile.js

@@ -0,0 +1,193 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * But needs some custom changes for TFY
+ * 1. By default purpose is set to batch
+ * 2. Add custom_id to the request body
+ * 3. Use tfy- prefix for ids instead of the legacy prefix in the response
+ * 4. Remove unused vars for linting
+ * ================================
+ */
+import { ValidationError } from '../../errors.js';
+import { BatchEndpoints } from '../../globals.js';
+import { createLineSplitter } from '../../handlers/streamHandler.js';
+import { safeFetch } from '../../http/index.js';
+import { transformUsingProviderConfig } from '../../services/transformToProviderRequest.js';
+import { buildErrorBody, errorResponse } from '../../utils/errorHandlingUtils.js';
+import { chatCompleteParams, embedParams } from '../open-ai-base/index.js';
+import { providerApiArgsFromHandler } from '../requestHandlerApiArgs.js';
+import GoogleApiConfig, { resolveVertexFields } from './api.js';
+import { VertexAnthropicChatCompleteConfig, VertexGoogleChatCompleteConfig, VertexLlamaChatCompleteConfig, } from './chatComplete.js';
+import { VertexBatchEmbedConfig } from './embed.js';
+import { generateSignedURL, getModelAndProvider, GoogleResponseHandler, isEmbeddingModel, vertexRequestLineHandler, } from './utils.js';
+const PROVIDER_CONFIG = {
+    google: {
+        [BatchEndpoints.CHAT_COMPLETIONS]: VertexGoogleChatCompleteConfig,
+        [BatchEndpoints.EMBEDDINGS]: VertexBatchEmbedConfig,
+    },
+    anthropic: {
+        [BatchEndpoints.CHAT_COMPLETIONS]: VertexAnthropicChatCompleteConfig,
+    },
+    meta: {
+        [BatchEndpoints.CHAT_COMPLETIONS]: VertexLlamaChatCompleteConfig,
+    },
+    endpoints: {
+        [BatchEndpoints.CHAT_COMPLETIONS]: chatCompleteParams(['model']),
+        [BatchEndpoints.EMBEDDINGS]: embedParams(['model']),
+    },
+};
+const encoder = new TextEncoder();
+/** Embeddings JSONL uses the embed config; everything else stays on chat completions. */
+export function resolveVertexUploadBatchEndpoint(modelName) {
+    return isEmbeddingModel(modelName ?? '') ? BatchEndpoints.EMBEDDINGS : BatchEndpoints.CHAT_COMPLETIONS;
+}
+export function vertexUploadProviderConfig(provider, modelName) {
+    const vertexBatchEndpoint = resolveVertexUploadBatchEndpoint(modelName);
+    const providerConfigMap = PROVIDER_CONFIG[provider];
+    return providerConfigMap?.[vertexBatchEndpoint] ?? PROVIDER_CONFIG['endpoints'][vertexBatchEndpoint];
+}
+export const GoogleFileUploadRequestHandler = async (args) => {
+    const requestBody = args.request.body;
+    const requestHeaders = args.request.headers;
+    const vertex = args.headerOptions.vertex;
+    const vertexStorageBucketName = vertex.vertexStorageBucketName;
+    const filename = vertex.filename;
+    const vertexModelName = vertex.vertexModelName;
+    const vertexBatchEndpoint = resolveVertexUploadBatchEndpoint(vertexModelName);
+    let purpose = requestHeaders['x-tfy-file-purpose'] ?? '';
+    if ((purpose === 'upload' ? false : !vertexModelName) || !vertexStorageBucketName) {
+        return GoogleResponseHandler('Invalid request, please provide `x-tfy-provider-model` and `x-tfy-vertex-storage-bucket-name` in the request headers', 400);
+    }
+    const objectKey = filename ?? `${crypto.randomUUID()}.jsonl`;
+    const bytes = requestHeaders['content-length'];
+    const { provider } = getModelAndProvider(vertexModelName ?? '');
+    const providerConfig = vertexUploadProviderConfig(provider, vertexModelName);
+    if (!providerConfig) {
+        throw new ValidationError(`Endpoint ${vertexBatchEndpoint} not supported for provider ${provider}`);
+    }
+    let isPurposeHeader = false;
+    let transformStream = requestBody;
+    let uploadMethod = 'PUT';
+    // Create a reusable line splitter stream
+    const lineSplitter = createLineSplitter();
+    if (purpose === 'upload') {
+        uploadMethod = 'POST';
+    }
+    else {
+        // Transform stream to process each complete line.
+        transformStream = new TransformStream({
+            transform: function (chunk, controller) {
+                let buffer;
+                try {
+                    const _chunk = chunk.toString();
+                    const match = _chunk.match(/name="([^"]+)"/);
+                    const headerKey = match ? match[1] : null;
+                    if (headerKey && headerKey === 'purpose') {
+                        isPurposeHeader = true;
+                        return;
+                    }
+                    if (isPurposeHeader && _chunk?.length > 0 && !purpose) {
+                        isPurposeHeader = false;
+                        purpose = _chunk.trim();
+                        return;
+                    }
+                    if (!_chunk) {
+                        return;
+                    }
+                    const json = JSON.parse(chunk.toString());
+                    if (json && !purpose) {
+                        // Close the stream.
+                        controller.terminate();
+                    }
+                    const toTranspose = purpose === 'batch' ? json.body : json;
+                    const transformedBody = transformUsingProviderConfig(providerConfig, toTranspose, {
+                        callConfig: args.callConfig,
+                        headerOptions: args.headerOptions,
+                        providerOptions: args.providerOptions,
+                    });
+                    delete transformedBody['model'];
+                    const bufferTransposed = vertexRequestLineHandler(purpose, vertexBatchEndpoint, transformedBody, json['custom_id']);
+                    buffer = JSON.stringify(bufferTransposed);
+                }
+                catch {
+                    buffer = null;
+                }
+                finally {
+                    if (buffer) {
+                        controller.enqueue(encoder.encode(buffer + '\n'));
+                    }
+                }
+            },
+            flush(controller) {
+                controller.terminate();
+            },
+        });
+        requestBody.pipeThrough(lineSplitter).pipeTo(transformStream.writable);
+    }
+    // Pipe the node stream through our line splitter and into the transform stream.
+    const providerHeaders = await GoogleApiConfig.headers({
+        ...providerApiArgsFromHandler(args, { endpoint: 'uploadFile' }),
+        transformedRequestBody: {},
+        transformedRequestUrl: '',
+    });
+    const encodedFile = encodeURIComponent(objectKey ?? '');
+    let url;
+    if (uploadMethod !== 'POST') {
+        url = `https://storage.googleapis.com/${vertexStorageBucketName}/${encodedFile}`;
+    }
+    else {
+        const serviceAccountJson = resolveVertexFields(args.providerOptions, args.headerOptions).vertexServiceAccountJson;
+        if (!serviceAccountJson) {
+            throw new ValidationError('Vertex resumable upload (POST) requires a service account key (auth_data type key-file) to sign the URL. WIF (workload-identity-federation-file) is not supported for this upload path yet.');
+        }
+        const queryParameters = Object.fromEntries(new URLSearchParams(args.request.search));
+        url = await generateSignedURL(serviceAccountJson, vertexStorageBucketName, objectKey, 10 * 60, 'POST', queryParameters, {});
+    }
+    const options = {
+        body: uploadMethod === 'POST'
+            ? transformStream
+            : transformStream.readable,
+        headers: {
+            ...(uploadMethod !== 'POST' ? { Authorization: providerHeaders.Authorization } : {}),
+            'Content-Type': uploadMethod === 'POST' ? requestHeaders['content-type'] : 'application/octet-stream',
+        },
+        method: uploadMethod,
+        duplex: 'half',
+        signal: args.signal,
+    };
+    try {
+        const request = await safeFetch(url, { ssrf: 'guard', ...options });
+        if (!request.ok) {
+            const error = await request.text();
+            return GoogleResponseHandler(error, request.status);
+        }
+        const response = {
+            filename: filename,
+            id: encodeURIComponent(`gs://${vertexStorageBucketName}/${objectKey}`),
+            object: 'file',
+            create_at: Date.now(),
+            purpose: purpose,
+            bytes: Number.parseInt(bytes ?? '0'),
+            status: 'processed',
+        };
+        return GoogleResponseHandler(response, 200);
+    }
+    catch ( /* eslint-disable @typescript-eslint/no-unused-vars */error) {
+        if (!purpose) {
+            return errorResponse(buildErrorBody({
+                message: 'Purpose is not set',
+                statusCode: 400,
+                type: 'invalid_request_error',
+            }), 400);
+        }
+        return errorResponse(buildErrorBody({
+            message: 'Something went wrong',
+            statusCode: 500,
+            type: null,
+        }), 500);
+    }
+};
+export const GoogleFileUploadResponseTransform = (response) => {
+    return response;
+};
+//# sourceMappingURL=uploadFile.js.map

package/dist/providers/google-vertex-ai/uploadFile.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"uploadFile.js","sourceRoot":"","sources":["../../../src/providers/google-vertex-ai/uploadFile.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AACH,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,EAAE,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AAClE,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AACvC,OAAO,EAAE,4BAA4B,EAAE,MAAM,2CAA2C,CAAC;AACzF,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AAC/E,OAAO,EAAE,kBAAkB,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAClE,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AAEtE,OAAO,eAAe,EAAE,EAAE,mBAAmB,EAAE,MAAM,OAAO,CAAC;AAC7D,OAAO,EACL,iCAAiC,EACjC,8BAA8B,EAC9B,6BAA6B,GAC9B,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,sBAAsB,EAAE,MAAM,SAAS,CAAC;AACjD,OAAO,EACL,iBAAiB,EACjB,mBAAmB,EACnB,qBAAqB,EACrB,gBAAgB,EAChB,wBAAwB,GACzB,MAAM,SAAS,CAAC;AAEjB,MAAM,eAAe,GAAoE;IACvF,MAAM,EAAE;QACN,CAAC,cAAc,CAAC,gBAAgB,CAAC,EAAE,8BAA8B;QACjE,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE,sBAAsB;KACpD;IACD,SAAS,EAAE;QACT,CAAC,cAAc,CAAC,gBAAgB,CAAC,EAAE,iCAAiC;KACrE;IACD,IAAI,EAAE;QACJ,CAAC,cAAc,CAAC,gBAAgB,CAAC,EAAE,6BAA6B;KACjE;IACD,SAAS,EAAE;QACT,CAAC,cAAc,CAAC,gBAAgB,CAAC,EAAE,kBAAkB,CAAC,CAAC,OAAO,CAAC,CAAC;QAChE,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE,WAAW,CAAC,CAAC,OAAO,CAAC,CAAC;KACpD;CACF,CAAC;AAEF,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;AAElC,yFAAyF;AACzF,MAAM,UAAU,gCAAgC,CAAC,SAAkB;IACjE,OAAO,gBAAgB,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC;AACzG,CAAC;AAED,MAAM,UAAU,0BAA0B,CAAC,QAAgB,EAAE,SAAkB;IAC7E,MAAM,mBAAmB,GAAG,gCAAgC,CAAC,SAAS,CAAC,CAAC;IACxE,MAAM,iBAAiB,GAAG,eAAe,CAAC,QAAwC,CAAC,CAAC;IACpF,OAAO,iBAAiB,EAAE,CAAC,mBAAmB,CAAC,IAAI,eAAe,CAAC,WAAW,CAAC,CAAC,mBAAmB,CAAC,CAAC;AACvG,CAAC;AAED,MAAM,CAAC,MAAM,8BAA8B,GAAmC,KAAK,EAAC,IAAI,EAAC,EAAE;IACzF,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;IACtC,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC;IAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC;IACzC,MAAM,uBAAuB,GAAG,MAAM,CAAC,uBAAuB,CAAC;IAC/D,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;IACjC,MAAM,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;IAC/C,MAAM,mBAAmB,GAAG,gCAAgC,CAAC,eAAe,CAAC,CAAC;IAE9E,IAAI,OAAO,GAAG,cAAc,CAAC,oBAAoB,CAAC,IAAI,EAAE,CAAC;IACzD,IAAI,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAClF,OAAO,qBAAqB,CAC1B,sHAAsH,EACtH,GAAG,CACJ,CAAC;IACJ,CAAC;IAED,MAAM,SAAS,GAAG,QAAQ,IAAI,GAAG,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC;IAC7D,MAAM,KAAK,GAAG,cAAc,CAAC,gBAAgB,CAAC,CAAC;IAC/C,MAAM,EAAE,QAAQ,EAAE,GAAG,mBAAmB,CAAC,eAAe,IAAI,EAAE,CAAC,CAAC;IAChE,MAAM,cAAc,GAAG,0BAA0B,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;IAE7E,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,MAAM,IAAI,eAAe,CAAC,YAAY,mBAAmB,+BAA+B,QAAQ,EAAE,CAAC,CAAC;IACtG,CAAC;IAED,IAAI,eAAe,GAAG,KAAK,CAAC;IAC5B,IAAI,eAAe,GAAoD,WAAW,CAAC;IACnF,IAAI,YAAY,GAAG,KAAK,CAAC;IACzB,yCAAyC;IACzC,MAAM,YAAY,GAAG,kBAAkB,EAAE,CAAC;IAE1C,IAAI,OAAO,KAAK,QAAQ,EAAE,CAAC;QACzB,YAAY,GAAG,MAAM,CAAC;IACxB,CAAC;SAAM,CAAC;QACN,kDAAkD;QAClD,eAAe,GAAG,IAAI,eAAe,CAAC;YACpC,SAAS,EAAE,UAAU,KAAK,EAAE,UAAU;gBACpC,IAAI,MAAM,CAAC;gBACX,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;oBAEhC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;oBAC7C,MAAM,SAAS,GAAG,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;oBAE1C,IAAI,SAAS,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;wBACzC,eAAe,GAAG,IAAI,CAAC;wBACvB,OAAO;oBACT,CAAC;oBAED,IAAI,eAAe,IAAI,MAAM,EAAE,MAAM,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;wBACtD,eAAe,GAAG,KAAK,CAAC;wBACxB,OAAO,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;wBACxB,OAAO;oBACT,CAAC;oBAED,IAAI,CAAC,MAAM,EAAE,CAAC;wBACZ,OAAO;oBACT,CAAC;oBAED,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;oBAE1C,IAAI,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;wBACrB,oBAAoB;wBACpB,UAAU,CAAC,SAAS,EAAE,CAAC;oBACzB,CAAC;oBAED,MAAM,WAAW,GAAG,OAAO,KAAK,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;oBAC3D,MAAM,eAAe,GAAG,4BAA4B,CAAC,cAAc,EAAE,WAAW,EAAE;wBAChF,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,aAAa,EAAE,IAAI,CAAC,aAAa;wBACjC,eAAe,EAAE,IAAI,CAAC,eAAe;qBACtC,CAAC,CAAC;oBAEH,OAAO,eAAe,CAAC,OAAO,CAAC,CAAC;oBAEhC,MAAM,gBAAgB,GAAG,wBAAwB,CAC/C,OAAO,EACP,mBAAmB,EACnB,eAAe,EACf,IAAI,CAAC,WAAW,CAAC,CAClB,CAAC;oBAEF,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;gBAC5C,CAAC;gBAAC,MAAM,CAAC;oBACP,MAAM,GAAG,IAAI,CAAC;gBAChB,CAAC;wBAAS,CAAC;oBACT,IAAI,MAAM,EAAE,CAAC;wBACX,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC;oBACpD,CAAC;gBACH,CAAC;YACH,CAAC;YACD,KAAK,CAAC,UAAU;gBACd,UAAU,CAAC,SAAS,EAAE,CAAC;YACzB,CAAC;SACF,CAAC,CAAC;QACH,WAAW,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;IACzE,CAAC;IAED,gFAAgF;IAEhF,MAAM,eAAe,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC;QACpD,GAAG,0BAA0B,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC;QAC/D,sBAAsB,EAAE,EAAE;QAC1B,qBAAqB,EAAE,EAAE;KAC1B,CAAC,CAAC;IAEH,MAAM,WAAW,GAAG,kBAAkB,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC;IACxD,IAAI,GAAG,CAAC;IACR,IAAI,YAAY,KAAK,MAAM,EAAE,CAAC;QAC5B,GAAG,GAAG,kCAAkC,uBAAuB,IAAI,WAAW,EAAE,CAAC;IACnF,CAAC;SAAM,CAAC;QACN,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC,wBAAwB,CAAC;QAClH,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACxB,MAAM,IAAI,eAAe,CACvB,6LAA6L,CAC9L,CAAC;QACJ,CAAC;QACD,MAAM,eAAe,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;QACrF,GAAG,GAAG,MAAM,iBAAiB,CAC3B,kBAAkB,EAClB,uBAAuB,EACvB,SAAS,EACT,EAAE,GAAG,EAAE,EACP,MAAM,EACN,eAAe,EACf,EAAE,CACH,CAAC;IACJ,CAAC;IAED,MAAM,OAAO,GAAG;QACd,IAAI,EACF,YAAY,KAAK,MAAM;YACrB,CAAC,CAAE,eAAuC;YAC1C,CAAC,CAAE,eAAmC,CAAC,QAAQ;QACnD,OAAO,EAAE;YACP,GAAG,CAAC,YAAY,KAAK,MAAM,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,eAAe,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACpF,cAAc,EAAE,YAAY,KAAK,MAAM,CAAC,CAAC,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,0BAA0B;SACtG;QACD,MAAM,EAAE,YAAY;QACpB,MAAM,EAAE,MAAM;QACd,MAAM,EAAE,IAAI,CAAC,MAAM;KACpB,CAAC;IAEF,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,GAAG,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;QACpE,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,CAAC;YAChB,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;YACnC,OAAO,qBAAqB,CAAC,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QACtD,CAAC;QAED,MAAM,QAAQ,GAAG;YACf,QAAQ,EAAE,QAAQ;YAClB,EAAE,EAAE,kBAAkB,CAAC,QAAQ,uBAAuB,IAAI,SAAS,EAAE,CAAC;YACtE,MAAM,EAAE,MAAM;YACd,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,OAAO,EAAE,OAAO;YAChB,KAAK,EAAE,MAAM,CAAC,QAAQ,CAAC,KAAK,IAAI,GAAG,CAAC;YACpC,MAAM,EAAE,WAAW;SACpB,CAAC;QAEF,OAAO,qBAAqB,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;IAC9C,CAAC;IAAC,QAAO,sDAAuD,KAAK,EAAE,CAAC;QACtE,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,aAAa,CAClB,cAAc,CAAC;gBACb,OAAO,EAAE,oBAAoB;gBAC7B,UAAU,EAAE,GAAG;gBACf,IAAI,EAAE,uBAAuB;aAC9B,CAAC,EACF,GAAG,CACJ,CAAC;QACJ,CAAC;QACD,OAAO,aAAa,CAClB,cAAc,CAAC;YACb,OAAO,EAAE,sBAAsB;YAC/B,UAAU,EAAE,GAAG;YACf,IAAI,EAAE,IAAI;SACX,CAAC,EACF,GAAG,CACJ,CAAC;IACJ,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,iCAAiC,GAAG,CAAC,QAAkB,EAAE,EAAE;IACtE,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC"}