fix: ensure fact_embedding is removed from attributes in format_fact_… (#611)
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:
parent
25dfc10662
commit
b8e6762cb2
1 changed files with 3 additions and 1 deletions
|
|
@ -628,12 +628,14 @@ def format_fact_result(edge: EntityEdge) -> dict[str, Any]:
|
|||
Returns:
|
||||
A dictionary representation of the edge with serialized dates and excluded embeddings
|
||||
"""
|
||||
return edge.model_dump(
|
||||
result = edge.model_dump(
|
||||
mode='json',
|
||||
exclude={
|
||||
'fact_embedding',
|
||||
},
|
||||
)
|
||||
result.get('attributes', {}).pop('fact_embedding', None)
|
||||
return result
|
||||
|
||||
|
||||
# Dictionary to store queues for each group_id
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue