yujunjun/graphiti
1
0
Fork 0
mirror of https://github.com/getzep/graphiti.git synced 2025-06-27 02:00:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

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.

Browse Source
This commit is contained in:
Daniel Chalef 2025-05-21 19:37:05 -07:00
parent b6c0bb9c54
commit 7a2111b809
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