graphiti-core 0.9.0__tar.gz → 0.9.2__tar.gz

{graphiti_core-0.9.0 → graphiti_core-0.9.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: graphiti-core
-Version: 0.9.0
+Version: 0.9.2
 Summary: A temporal graph building library
 License: Apache-2.0
 Author: Paul Paliychuk

{graphiti_core-0.9.0 → graphiti_core-0.9.2}/graphiti_core/edges.py

@@ -143,9 +143,9 @@ class EpisodicEdge(Edge):
         driver: AsyncDriver,
         group_ids: list[str],
         limit: int | None = None,
-        created_at: datetime | None = None,
+        uuid_cursor: str | None = None,
     ):
-        cursor_query: LiteralString = 'AND e.created_at < $created_at' if created_at else ''
+        cursor_query: LiteralString = 'AND e.uuid < $uuid' if uuid_cursor else ''
         limit_query: LiteralString = 'LIMIT $limit' if limit is not None else ''
 
         records, _, _ = await driver.execute_query(
@@ -165,7 +165,7 @@ class EpisodicEdge(Edge):
         """
             + limit_query,
             group_ids=group_ids,
-            created_at=created_at,
+            uuid=uuid_cursor,
             limit=limit,
             database_=DEFAULT_DATABASE,
             routing_='r',
@@ -297,9 +297,9 @@ class EntityEdge(Edge):
         driver: AsyncDriver,
         group_ids: list[str],
         limit: int | None = None,
-        created_at: datetime | None = None,
+        uuid_cursor: str | None = None,
     ):
-        cursor_query: LiteralString = 'AND e.created_at < $created_at' if created_at else ''
+        cursor_query: LiteralString = 'AND e.uuid < $uuid' if uuid_cursor else ''
         limit_query: LiteralString = 'LIMIT $limit' if limit is not None else ''
 
         records, _, _ = await driver.execute_query(
@@ -326,7 +326,7 @@ class EntityEdge(Edge):
         """
             + limit_query,
             group_ids=group_ids,
-            created_at=created_at,
+            uuid=uuid_cursor,
             limit=limit,
             database_=DEFAULT_DATABASE,
             routing_='r',
@@ -430,9 +430,9 @@ class CommunityEdge(Edge):
         driver: AsyncDriver,
         group_ids: list[str],
         limit: int | None = None,
-        created_at: datetime | None = None,
+        uuid_cursor: str | None = None,
     ):
-        cursor_query: LiteralString = 'AND e.created_at < $created_at' if created_at else ''
+        cursor_query: LiteralString = 'AND e.uuid < $uuid' if uuid_cursor else ''
         limit_query: LiteralString = 'LIMIT $limit' if limit is not None else ''
 
         records, _, _ = await driver.execute_query(
@@ -452,7 +452,7 @@ class CommunityEdge(Edge):
         """
             + limit_query,
             group_ids=group_ids,
-            created_at=created_at,
+            uuid=uuid_cursor,
             limit=limit,
             database_=DEFAULT_DATABASE,
             routing_='r',

{graphiti_core-0.9.0 → graphiti_core-0.9.2}/graphiti_core/llm_client/openai_client.py

@@ -87,6 +87,8 @@ class OpenAIClient(LLMClient):
         else:
             self.client = client
 
+        self.max_tokens = max_tokens
+
     async def _generate_response(
         self,
         messages: list[Message],

{graphiti_core-0.9.0 → graphiti_core-0.9.2}/graphiti_core/nodes.py

@@ -216,9 +216,9 @@ class EpisodicNode(Node):
         driver: AsyncDriver,
         group_ids: list[str],
         limit: int | None = None,
-        created_at: datetime | None = None,
+        uuid_cursor: str | None = None,
     ):
-        cursor_query: LiteralString = 'AND e.created_at < $created_at' if created_at else ''
+        cursor_query: LiteralString = 'AND e.uuid < $uuid' if uuid_cursor else ''
         limit_query: LiteralString = 'LIMIT $limit' if limit is not None else ''
 
         records, _, _ = await driver.execute_query(
@@ -241,7 +241,7 @@ class EpisodicNode(Node):
         """
             + limit_query,
             group_ids=group_ids,
-            created_at=created_at,
+            uuid=uuid_cursor,
             limit=limit,
             database_=DEFAULT_DATABASE,
             routing_='r',
@@ -348,9 +348,9 @@ class EntityNode(Node):
         driver: AsyncDriver,
         group_ids: list[str],
         limit: int | None = None,
-        created_at: datetime | None = None,
+        uuid_cursor: str | None = None,
     ):
-        cursor_query: LiteralString = 'AND n.created_at < $created_at' if created_at else ''
+        cursor_query: LiteralString = 'AND n.uuid < $uuid' if uuid_cursor else ''
         limit_query: LiteralString = 'LIMIT $limit' if limit is not None else ''
 
         records, _, _ = await driver.execute_query(
@@ -372,7 +372,7 @@ class EntityNode(Node):
         """
             + limit_query,
             group_ids=group_ids,
-            created_at=created_at,
+            uuid=uuid_cursor,
             limit=limit,
             database_=DEFAULT_DATABASE,
             routing_='r',
@@ -465,9 +465,9 @@ class CommunityNode(Node):
         driver: AsyncDriver,
         group_ids: list[str],
         limit: int | None = None,
-        created_at: datetime | None = None,
+        uuid_cursor: str | None = None,
     ):
-        cursor_query: LiteralString = 'AND n.created_at < $created_at' if created_at else ''
+        cursor_query: LiteralString = 'AND n.uuid < $uuid' if uuid_cursor else ''
         limit_query: LiteralString = 'LIMIT $limit' if limit is not None else ''
 
         records, _, _ = await driver.execute_query(
@@ -487,7 +487,7 @@ class CommunityNode(Node):
         """
             + limit_query,
             group_ids=group_ids,
-            created_at=created_at,
+            uuid=uuid_cursor,
             limit=limit,
             database_=DEFAULT_DATABASE,
             routing_='r',

{graphiti_core-0.9.0 → graphiti_core-0.9.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "graphiti-core"
-version = "0.9.0"
+version = "0.9.2"
 description = "A temporal graph building library"
 authors = [
     "Paul Paliychuk <paul@getzep.com>",
@@ -31,6 +31,7 @@ google-genai = ["google-genai"]
 mypy = "^1.11.1"
 groq = ">=0.9,<0.12"
 anthropic = "~0.49.0"
+google-genai = ">=0.8.0"
 ipykernel = "^6.29.5"
 jupyterlab = "^4.2.4"
 diskcache-stubs = "^5.6.3.6.20240818"