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

update

Browse source
This commit is contained in:
prestonrasmussen 2025-10-02 14:49:24 -04:00
parent dd0d42855e
commit c5ec136b1d
2 changed files with 22 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
graphiti_core/edges.py
View file

@ -649,5 +649,5 @@ async def create_entity_edge_embeddings(embedder: EmbedderClient, edges: list[En
# 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]
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(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

46
graphiti_core/nodes.py
View file

@ -96,8 +96,7 @@ class Node(BaseModel, ABC):
created_at: datetime = Field(default_factory=lambda: utc_now()) created_at: datetime = Field(default_factory=lambda: utc_now())
@abstractmethod @abstractmethod
async def save(self, driver: GraphDriver): async def save(self, driver: GraphDriver): ...
...
async def delete(self, driver: GraphDriver): async def delete(self, driver: GraphDriver):
match driver.provider: match driver.provider:
@ -345,12 +344,10 @@ class Node(BaseModel, ABC):
await driver.aoss_client.bulk(body=actions) await driver.aoss_client.bulk(body=actions)
@classmethod @classmethod
async def get_by_uuid(cls, driver: GraphDriver, uuid: str): async def get_by_uuid(cls, driver: GraphDriver, uuid: str): ...
...
@classmethod @classmethod
async def get_by_uuids(cls, driver: GraphDriver, uuids: list[str]): async def get_by_uuids(cls, driver: GraphDriver, uuids: list[str]): ...
...
class EpisodicNode(Node): class EpisodicNode(Node):
@ -438,11 +435,11 @@ class EpisodicNode(Node):
@classmethod @classmethod
async def get_by_group_ids( async def get_by_group_ids(
cls, cls,
driver: GraphDriver, driver: GraphDriver,
group_ids: list[str], group_ids: list[str],
limit: int | None = None, limit: int | None = None,
uuid_cursor: str | None = None, uuid_cursor: str | None = None,
): ):
cursor_query: LiteralString = 'AND e.uuid < $uuid' if uuid_cursor else '' cursor_query: LiteralString = 'AND e.uuid < $uuid' if uuid_cursor else ''
limit_query: LiteralString = 'LIMIT $limit' if limit is not None else '' limit_query: LiteralString = 'LIMIT $limit' if limit is not None else ''
@ -572,8 +569,7 @@ class EntityNode(Node):
labels = ':'.join(self.labels + ['Entity']) labels = ':'.join(self.labels + ['Entity'])
if driver.aoss_client: if driver.aoss_client:
await driver.save_to_aoss(ENTITY_INDEX_NAME, await driver.save_to_aoss(ENTITY_INDEX_NAME, [entity_data]) # pyright: ignore reportAttributeAccessIssue
[entity_data]) # pyright: ignore reportAttributeAccessIssue
result = await driver.execute_query( result = await driver.execute_query(
get_entity_node_save_query(driver.provider, labels, bool(driver.aoss_client)), get_entity_node_save_query(driver.provider, labels, bool(driver.aoss_client)),
@ -622,12 +618,12 @@ class EntityNode(Node):
@classmethod @classmethod
async def get_by_group_ids( async def get_by_group_ids(
cls, cls,
driver: GraphDriver, driver: GraphDriver,
group_ids: list[str], group_ids: list[str],
limit: int | None = None, limit: int | None = None,
uuid_cursor: str | None = None, uuid_cursor: str | None = None,
with_embeddings: bool = False, with_embeddings: bool = False,
): ):
cursor_query: LiteralString = 'AND n.uuid < $uuid' if uuid_cursor else '' cursor_query: LiteralString = 'AND n.uuid < $uuid' if uuid_cursor else ''
limit_query: LiteralString = 'LIMIT $limit' if limit is not None else '' limit_query: LiteralString = 'LIMIT $limit' if limit is not None else ''
@ -767,11 +763,11 @@ class CommunityNode(Node):
@classmethod @classmethod
async def get_by_group_ids( async def get_by_group_ids(
cls, cls,
driver: GraphDriver, driver: GraphDriver,
group_ids: list[str], group_ids: list[str],
limit: int | None = None, limit: int | None = None,
uuid_cursor: str | None = None, uuid_cursor: str | None = None,
): ):
cursor_query: LiteralString = 'AND c.uuid < $uuid' if uuid_cursor else '' cursor_query: LiteralString = 'AND c.uuid < $uuid' if uuid_cursor else ''
limit_query: LiteralString = 'LIMIT $limit' if limit is not None else '' limit_query: LiteralString = 'LIMIT $limit' if limit is not None else ''
@ -878,5 +874,5 @@ async def create_entity_node_embeddings(embedder: EmbedderClient, nodes: list[En
# 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]
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(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