Skip to content

Commit a0524a1

Browse files
committed
fix: type checking
1 parent ccd0a7f commit a0524a1

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

integrations/ollama/src/haystack_integrations/components/embedders/ollama/document_embedder.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -152,6 +152,9 @@ async def _embed_batch_async(
152152
results = await asyncio.gather(*tasks, return_exceptions=True)
153153

154154
for _idx, res in enumerate(results):
155+
if isinstance(res, Exception):
156+
raise RuntimeError(f"Error in batch {_idx}: {res}")
157+
continue
155158
all_embeddings.extend(res["embeddings"])
156159

157160
return all_embeddings

0 commit comments

Comments
 (0)