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

early return

Browse source
This commit is contained in:
prestonrasmussen 2025-10-02 14:54:21 -04:00
parent c5ec136b1d
commit bb4406c5c5
2 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
graphiti_core/edges.py
View file

@ -644,10 +644,11 @@ def get_community_edge_from_record(record: Any):
async def create_entity_edge_embeddings(embedder: EmbedderClient, edges: list[EntityEdge]): async def create_entity_edge_embeddings(embedder: EmbedderClient, edges: list[EntityEdge]):
if len(edges) == 0:
return
# filter out falsey values from edges # filter out falsey values from edges
filtered_edges = [edge for edge in edges if edge.fact] filtered_edges = [edge for edge in edges if edge.fact]
if len(filtered_edges) == 0:
return
fact_embeddings = await embedder.create_batch([edge.fact for edge in filtered_edges]) fact_embeddings = await embedder.create_batch([edge.fact for edge in filtered_edges])
for edge, fact_embedding in zip(filtered_edges, fact_embeddings, strict=True): for edge, fact_embedding in zip(filtered_edges, fact_embeddings, strict=True):
edge.fact_embedding = fact_embedding edge.fact_embedding = fact_embedding

7
graphiti_core/nodes.py
View file

@ -868,11 +868,12 @@ def get_community_node_from_record(record: Any) -> CommunityNode:
async def create_entity_node_embeddings(embedder: EmbedderClient, nodes: list[EntityNode]): async def create_entity_node_embeddings(embedder: EmbedderClient, nodes: list[EntityNode]):
if not nodes: # Handle empty list case
return
# filter out falsey values from nodes # filter out falsey values from nodes
filtered_nodes = [node for node in nodes if node.name] filtered_nodes = [node for node in nodes if node.name]
if not filtered_nodes:
return
name_embeddings = await embedder.create_batch([node.name for node in filtered_nodes]) name_embeddings = await embedder.create_batch([node.name for node in filtered_nodes])
for node, name_embedding in zip(filtered_nodes, name_embeddings, strict=True): for node, name_embedding in zip(filtered_nodes, name_embeddings, strict=True):
node.name_embedding = name_embedding node.name_embedding = name_embedding