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

fix: ensure fact_embedding is removed from attributes in format_fact_… (#611)

Browse source 
fix: ensure fact_embedding is removed from attributes in format_fact_result

This ensures that fact_embedding is completely cleaned from the result,
both from the main level and from the attributes dictionary.
This commit is contained in:
robrichardson13 2025-06-23 22:01:26 -07:00 • committed by GitHub
parent 25dfc10662
commit b8e6762cb2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
mcp_server/graphiti_mcp_server.py
View file

@ -628,12 +628,14 @@ def format_fact_result(edge: EntityEdge) -> dict[str, Any]:
Returns: Returns:
A dictionary representation of the edge with serialized dates and excluded embeddings A dictionary representation of the edge with serialized dates and excluded embeddings
""" """
return edge.model_dump( result = edge.model_dump(
mode='json', mode='json',
exclude={ exclude={
'fact_embedding', 'fact_embedding',
}, },
) )
result.get('attributes', {}).pop('fact_embedding', None)
return result
# Dictionary to store queues for each group_id # Dictionary to store queues for each group_id