scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

add BGE handling

Browse source
This commit is contained in:
Ishaan Jaffer 2025-10-28 18:11:01 -07:00 • committed by Sameer Kankute
parent 39e750d3b2
commit 3293ac8a3d
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/llms/vertex_ai/vertex_embeddings/transformation.py
View file

@ -5,6 +5,7 @@ from pydantic import BaseModel
from litellm.types.utils import EmbeddingResponse, Usage
from .bge import VertexBGEConfig
from .types import *
@ -109,6 +110,11 @@ class VertexAITextEmbeddingConfig(BaseModel):
return self._transform_openai_request_to_fine_tuned_embedding_request(
input, optional_params, model
)
if VertexBGEConfig.is_bge_model(model):
return VertexBGEConfig.transform_request(
input=input, optional_params=optional_params, model=model
)
vertex_request: VertexEmbeddingRequest = VertexEmbeddingRequest()
vertex_text_embedding_input_list: List[TextEmbeddingInput] = []
@ -186,8 +192,8 @@ class VertexAITextEmbeddingConfig(BaseModel):
Args:
content (str): The content to be embedded.
task_type (Optional[TaskType]): The type of task to be performed".
title (Optional[str]): The title of the document to be embedded
task_type (Optional[TaskType]): The type of task to be performed.
title (Optional[str]): The title of the document to be embedded.
Returns:
TextEmbeddingInput: A TextEmbeddingInput object.