Improve error handling in GeminiEmbedder by raising ValueError for empty embedding values. This change enhances robustness by ensuring that all returned embeddings contain valid data.
This commit is contained in:
parent
b6c0bb9c54
commit
7a2111b809
1 changed files with 6 additions and 1 deletions
|
|
@ -81,4 +81,9 @@ class GeminiEmbedder(EmbedderClient):
|
||||||
if not result.embeddings or len(result.embeddings) == 0:
|
if not result.embeddings or len(result.embeddings) == 0:
|
||||||
raise Exception('No embeddings returned')
|
raise Exception('No embeddings returned')
|
||||||
|
|
||||||
return [embedding.values if embedding.values else [] for embedding in result.embeddings]
|
embeddings = []
|
||||||
|
for embedding in result.embeddings:
|
||||||
|
if not embedding.values:
|
||||||
|
raise ValueError('Empty embedding values returned')
|
||||||
|
embeddings.append(embedding.values)
|
||||||
|
return embeddings
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue