From dd00cf2a970a6363556cd9204662073c9808902e Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 21 Aug 2024 14:25:47 -0700 Subject: [PATCH] add VertexMultimodalEmbeddingRequest type --- litellm/types/llms/vertex_ai.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/litellm/types/llms/vertex_ai.py b/litellm/types/llms/vertex_ai.py index 6758c356f57..5586d4861c2 100644 --- a/litellm/types/llms/vertex_ai.py +++ b/litellm/types/llms/vertex_ai.py @@ -1,6 +1,6 @@ import json from enum import Enum -from typing import Any, Dict, List, Literal, Optional, TypedDict, Union +from typing import Any, Dict, List, Literal, Optional, Tuple, TypedDict, Union from typing_extensions import ( Protocol, @@ -305,3 +305,18 @@ class ResponseTuningJob(TypedDict): ] createTime: Optional[str] updateTime: Optional[str] + + +class InstanceVideo(TypedDict, total=False): + gcsUri: str + videoSegmentConfig: Tuple[float, float, float] + + +class Instance(TypedDict, total=False): + text: str + image: Dict[str, str] + video: InstanceVideo + + +class VertexMultimodalEmbeddingRequest(TypedDict, total=False): + instances: List[Instance]