This commit is contained in:
Timothy Jaeryang Baek
2026-02-21 15:35:34 -06:00
parent c114fd6876
commit 631e30e22d
26 changed files with 177 additions and 164 deletions
+4 -2
View File
@@ -856,12 +856,14 @@ def get_embedding_function(
batch, prefix=prefix, user=user
)
tasks = [generate_batch_with_semaphore(batch) for batch in batches]
tasks = [
generate_batch_with_semaphore(batch) for batch in batches
]
else:
tasks = [
embedding_function(batch, prefix=prefix, user=user)
for batch in batches
]
]
batch_results = await asyncio.gather(*tasks)
else:
log.debug(