Handle FastEmbed generator results

This commit is contained in:
george
2026-06-06 12:52:04 +01:00
parent 7707a6306d
commit 30fe050182
2 changed files with 38 additions and 2 deletions
+9 -2
View File
@@ -86,8 +86,15 @@ def embed_text(text: str) -> List[float]:
return [0.0] * get_embedding_size()
model = get_embedding_model()
embedding = model.embed([text])
return embedding[0].tolist()
embeddings = iter(model.embed([text]))
try:
embedding = next(embeddings)
except StopIteration as exc:
raise RuntimeError("Embedding model returned no vector") from exc
if hasattr(embedding, "tolist"):
return embedding.tolist()
return list(embedding)
def embed_texts(texts: List[str]) -> List[List[float]]: