gmakstutis/graphiti
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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. (#515)

Browse source
This commit is contained in:
Daniel Chalef 2025-05-21 20:32:18 -07:00 • committed by GitHub
parent 9ab8abf9b4
commit 1b54b45144
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
graphiti_core/embedder/gemini.py
View file

@ -81,4 +81,9 @@ class GeminiEmbedder(EmbedderClient):
if not result.embeddings or len(result.embeddings) == 0:
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