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

default to no pagination (#232)

Browse source 
* default to no pagination

* remove unused imports
This commit is contained in:
Preston Rasmussen 2024-12-06 12:46:50 -05:00 • committed by GitHub
parent 6a152ab91a
commit 8858be8cef
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 26 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
graphiti_core/edges.py
View file

@ -27,7 +27,7 @@ from typing_extensions import LiteralString
from graphiti_core.embedder import EmbedderClient from graphiti_core.embedder import EmbedderClient
from graphiti_core.errors import EdgeNotFoundError, GroupsEdgesNotFoundError from graphiti_core.errors import EdgeNotFoundError, GroupsEdgesNotFoundError
from graphiti_core.helpers import DEFAULT_DATABASE, DEFAULT_PAGE_LIMIT, parse_db_date from graphiti_core.helpers import DEFAULT_DATABASE, parse_db_date
from graphiti_core.models.edges.edge_db_queries import ( from graphiti_core.models.edges.edge_db_queries import (
COMMUNITY_EDGE_SAVE, COMMUNITY_EDGE_SAVE,
ENTITY_EDGE_SAVE, ENTITY_EDGE_SAVE,
@ -142,10 +142,11 @@ class EpisodicEdge(Edge):
cls, cls,
driver: AsyncDriver, driver: AsyncDriver,
group_ids: list[str], group_ids: list[str],
limit: int = DEFAULT_PAGE_LIMIT, limit: int | None = None,
created_at: datetime | None = None, created_at: datetime | None = None,
): ):
cursor_query: LiteralString = 'AND e.created_at < $created_at' if created_at else '' cursor_query: LiteralString = 'AND e.created_at < $created_at' if created_at else ''
limit_query: LiteralString = 'LIMIT $limit' if limit is not None else ''
records, _, _ = await driver.execute_query( records, _, _ = await driver.execute_query(
""" """
@ -161,8 +162,8 @@ class EpisodicEdge(Edge):
m.uuid AS target_node_uuid, m.uuid AS target_node_uuid,
e.created_at AS created_at e.created_at AS created_at
ORDER BY e.uuid DESC ORDER BY e.uuid DESC
LIMIT $limit """
""", + limit_query,
group_ids=group_ids, group_ids=group_ids,
created_at=created_at, created_at=created_at,
limit=limit, limit=limit,
@ -294,10 +295,11 @@ class EntityEdge(Edge):
cls, cls,
driver: AsyncDriver, driver: AsyncDriver,
group_ids: list[str], group_ids: list[str],
limit: int = DEFAULT_PAGE_LIMIT, limit: int | None = None,
created_at: datetime | None = None, created_at: datetime | None = None,
): ):
cursor_query: LiteralString = 'AND e.created_at < $created_at' if created_at else '' cursor_query: LiteralString = 'AND e.created_at < $created_at' if created_at else ''
limit_query: LiteralString = 'LIMIT $limit' if limit is not None else ''
records, _, _ = await driver.execute_query( records, _, _ = await driver.execute_query(
""" """
@ -320,8 +322,8 @@ class EntityEdge(Edge):
e.valid_at AS valid_at, e.valid_at AS valid_at,
e.invalid_at AS invalid_at e.invalid_at AS invalid_at
ORDER BY e.uuid DESC ORDER BY e.uuid DESC
LIMIT $limit """
""", + limit_query,
group_ids=group_ids, group_ids=group_ids,
created_at=created_at, created_at=created_at,
limit=limit, limit=limit,
@ -400,10 +402,11 @@ class CommunityEdge(Edge):
cls, cls,
driver: AsyncDriver, driver: AsyncDriver,
group_ids: list[str], group_ids: list[str],
limit: int = DEFAULT_PAGE_LIMIT, limit: int | None = None,
created_at: datetime | None = None, created_at: datetime | None = None,
): ):
cursor_query: LiteralString = 'AND e.created_at < $created_at' if created_at else '' cursor_query: LiteralString = 'AND e.created_at < $created_at' if created_at else ''
limit_query: LiteralString = 'LIMIT $limit' if limit is not None else ''
records, _, _ = await driver.execute_query( records, _, _ = await driver.execute_query(
""" """
@ -419,8 +422,8 @@ class CommunityEdge(Edge):
m.uuid AS target_node_uuid, m.uuid AS target_node_uuid,
e.created_at AS created_at e.created_at AS created_at
ORDER BY e.uuid DESC ORDER BY e.uuid DESC
LIMIT $limit """
""", + limit_query,
group_ids=group_ids, group_ids=group_ids,
created_at=created_at, created_at=created_at,
limit=limit, limit=limit,

23
graphiti_core/nodes.py
View file

@ -28,7 +28,7 @@ from typing_extensions import LiteralString
from graphiti_core.embedder import EmbedderClient from graphiti_core.embedder import EmbedderClient
from graphiti_core.errors import NodeNotFoundError from graphiti_core.errors import NodeNotFoundError
from graphiti_core.helpers import DEFAULT_DATABASE, DEFAULT_PAGE_LIMIT from graphiti_core.helpers import DEFAULT_DATABASE
from graphiti_core.models.nodes.node_db_queries import ( from graphiti_core.models.nodes.node_db_queries import (
COMMUNITY_NODE_SAVE, COMMUNITY_NODE_SAVE,
ENTITY_NODE_SAVE, ENTITY_NODE_SAVE,
@ -212,10 +212,11 @@ class EpisodicNode(Node):
cls, cls,
driver: AsyncDriver, driver: AsyncDriver,
group_ids: list[str], group_ids: list[str],
limit: int = DEFAULT_PAGE_LIMIT, limit: int | None = None,
created_at: datetime | None = None, created_at: datetime | None = None,
): ):
cursor_query: LiteralString = 'AND e.created_at < $created_at' if created_at else '' cursor_query: LiteralString = 'AND e.created_at < $created_at' if created_at else ''
limit_query: LiteralString = 'LIMIT $limit' if limit is not None else ''
records, _, _ = await driver.execute_query( records, _, _ = await driver.execute_query(
""" """
@ -233,8 +234,8 @@ class EpisodicNode(Node):
e.source_description AS source_description, e.source_description AS source_description,
e.source AS source e.source AS source
ORDER BY e.uuid DESC ORDER BY e.uuid DESC
LIMIT $limit """
""", + limit_query,
group_ids=group_ids, group_ids=group_ids,
created_at=created_at, created_at=created_at,
limit=limit, limit=limit,
@ -328,10 +329,11 @@ class EntityNode(Node):
cls, cls,
driver: AsyncDriver, driver: AsyncDriver,
group_ids: list[str], group_ids: list[str],
limit: int = DEFAULT_PAGE_LIMIT, limit: int | None = None,
created_at: datetime | None = None, created_at: datetime | None = None,
): ):
cursor_query: LiteralString = 'AND n.created_at < $created_at' if created_at else '' cursor_query: LiteralString = 'AND n.created_at < $created_at' if created_at else ''
limit_query: LiteralString = 'LIMIT $limit' if limit is not None else ''
records, _, _ = await driver.execute_query( records, _, _ = await driver.execute_query(
""" """
@ -347,8 +349,8 @@ class EntityNode(Node):
n.created_at AS created_at, n.created_at AS created_at,
n.summary AS summary n.summary AS summary
ORDER BY n.uuid DESC ORDER BY n.uuid DESC
LIMIT $limit """
""", + limit_query,
group_ids=group_ids, group_ids=group_ids,
created_at=created_at, created_at=created_at,
limit=limit, limit=limit,
@ -442,10 +444,11 @@ class CommunityNode(Node):
cls, cls,
driver: AsyncDriver, driver: AsyncDriver,
group_ids: list[str], group_ids: list[str],
limit: int = DEFAULT_PAGE_LIMIT, limit: int | None = None,
created_at: datetime | None = None, created_at: datetime | None = None,
): ):
cursor_query: LiteralString = 'AND n.created_at < $created_at' if created_at else '' cursor_query: LiteralString = 'AND n.created_at < $created_at' if created_at else ''
limit_query: LiteralString = 'LIMIT $limit' if limit is not None else ''
records, _, _ = await driver.execute_query( records, _, _ = await driver.execute_query(
""" """
@ -461,8 +464,8 @@ class CommunityNode(Node):
n.created_at AS created_at, n.created_at AS created_at,
n.summary AS summary n.summary AS summary
ORDER BY n.uuid DESC ORDER BY n.uuid DESC
LIMIT $limit """
""", + limit_query,
group_ids=group_ids, group_ids=group_ids,
created_at=created_at, created_at=created_at,
limit=limit, limit=limit,