scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-12 23:01:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 182

fix: allow batch embedding result reassignment

Browse source
This commit is contained in:
Syed Ahmed Mubasiruddin 2026-08-31 09:59:43 +00:00
parent 78b8a5f415
commit ebb34775fa
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/router_strategy/auto_router/litellm_encoder.py
View file

@ -153,7 +153,7 @@ class LiteLLMRouterEncoder(CustomDenseEncoder, AsymmetricDenseMixin):
batch = docs[i : i + EMBEDDING_BATCH_SIZE]
try:
embeds: Final = await self.litellm_router_instance.aembedding(
embeds = await self.litellm_router_instance.aembedding( # rebind-ok: one result per batch
input=self._clamp(batch),
model=self.model_name,
**kwargs,