fix: add IndexError guard and error handling test

This commit is contained in:
2026-04-13 14:33:26 -04:00
parent b2a42e5fe6
commit d0a201476e
2 changed files with 29 additions and 1 deletions

View File

@@ -67,7 +67,12 @@ class SearchEngine:
k = top_k or self.default_top_k
try:
query_embedding = self.embedder.embed([query])[0]
embeddings = self.embedder.embed([query])
if not embeddings:
raise RuntimeError(
"Failed to generate embedding for query: embedder returned empty result"
)
query_embedding = embeddings[0]
except RuntimeError as e:
raise RuntimeError(f"Failed to generate embedding for query: {e}") from e