diff --git a/litellm/llms/vertex_ai/gemini_embeddings/batch_embed_content_handler.py b/litellm/llms/vertex_ai/gemini_embeddings/batch_embed_content_handler.py index 25c3465807c..68901340c7c 100644 --- a/litellm/llms/vertex_ai/gemini_embeddings/batch_embed_content_handler.py +++ b/litellm/llms/vertex_ai/gemini_embeddings/batch_embed_content_handler.py @@ -3,7 +3,7 @@ Google AI Studio /batchEmbedContents Embeddings Endpoint """ import json -from typing import Any, Dict, List, Literal, Optional, Union +from typing import Any, Dict, Literal, Optional, Union import httpx @@ -15,7 +15,6 @@ from litellm.llms.custom_httpx.http_handler import ( ) from litellm.types.llms.openai import EmbeddingInput from litellm.types.llms.vertex_ai import ( - GeminiEmbedContentResponseObject, VertexAIBatchEmbeddingsRequestBody, VertexAIBatchEmbeddingsResponseObject, )