graphiti-core 0.8.4__tar.gz → 0.8.6__tar.gz

{graphiti_core-0.8.4 → graphiti_core-0.8.6}/PKG-INFO

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

{graphiti_core-0.8.4 → graphiti_core-0.8.6}/graphiti_core/edges.py

@@ -26,7 +26,7 @@ from pydantic import BaseModel, Field
 from typing_extensions import LiteralString
 
 from graphiti_core.embedder import EmbedderClient
-from graphiti_core.errors import EdgeNotFoundError, EdgesNotFoundError, GroupsEdgesNotFoundError
+from graphiti_core.errors import EdgeNotFoundError, GroupsEdgesNotFoundError
 from graphiti_core.helpers import DEFAULT_DATABASE, parse_db_date
 from graphiti_core.models.edges.edge_db_queries import (
     COMMUNITY_EDGE_SAVE,
@@ -289,8 +289,6 @@ class EntityEdge(Edge):
 
         edges = [get_entity_edge_from_record(record) for record in records]
 
-        if len(edges) == 0:
-            raise EdgesNotFoundError(uuids)
         return edges
 
     @classmethod

{graphiti_core-0.8.4 → graphiti_core-0.8.6}/graphiti_core/llm_client/config.py

@@ -14,7 +14,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-DEFAULT_MAX_TOKENS = 2048
+DEFAULT_MAX_TOKENS = 8192
 DEFAULT_TEMPERATURE = 0
 
 

{graphiti_core-0.8.4 → graphiti_core-0.8.6}/graphiti_core/utils/maintenance/graph_data_operations.py

@@ -92,13 +92,22 @@ async def build_indices_and_constraints(driver: AsyncDriver, delete_existing: bo
     )
 
 
-async def clear_data(driver: AsyncDriver):
+async def clear_data(driver: AsyncDriver, group_ids: list[str] | None = None):
     async with driver.session() as session:
 
         async def delete_all(tx):
             await tx.run('MATCH (n) DETACH DELETE n')
 
-        await session.execute_write(delete_all)
+        async def delete_group_ids(tx):
+            await tx.run(
+                'MATCH (n:Entity|Episodic|Community) WHERE n.group_id IN $group_ids DETACH DELETE n',
+                group_ids=group_ids,
+            )
+
+        if group_ids is None:
+            await session.execute_write(delete_all)
+        else:
+            await session.execute_write(delete_group_ids)
 
 
 async def retrieve_episodes(

{graphiti_core-0.8.4 → graphiti_core-0.8.6}/graphiti_core/utils/maintenance/node_operations.py

@@ -15,6 +15,7 @@ limitations under the License.
 """
 
 import logging
+from contextlib import suppress
 from time import time
 
 import pydantic
@@ -365,9 +366,13 @@ async def resolve_extracted_node(
 
     extracted_node.summary = node_attributes_response.get('summary', '')
     node_attributes = {
-        key: value if value != 'None' else None for key, value in node_attributes_response.items()
+        key: value if (value != 'None' or key == 'summary') else None
+        for key, value in node_attributes_response.items()
     }
 
+    with suppress(KeyError):
+        del node_attributes['summary']
+
     extracted_node.attributes.update(node_attributes)
 
     is_duplicate: bool = llm_response.get('is_duplicate', False)

{graphiti_core-0.8.4 → graphiti_core-0.8.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "graphiti-core"
-version = "0.8.4"
+version = "0.8.6"
 description = "A temporal graph building library"
 authors = [
     "Paul Paliychuk <paul@getzep.com>",