graphiti-core 0.8.7__py3-none-any.whl → 0.8.8__py3-none-any.whl

graphiti_core/errors.py

@@ -65,3 +65,11 @@ class SearchRerankerError(GraphitiError):
     def __init__(self, text: str):
         self.message = text
         super().__init__(self.message)
+
+
+class EntityTypeValidationError(GraphitiError):
+    """Raised when an entity type uses protected attribute names."""
+
+    def __init__(self, entity_type: str, entity_type_attribute: str):
+        self.message = f'{entity_type_attribute} cannot be used as an attribute for {entity_type} as it is a protected attribute name.'
+        super().__init__(self.message)

graphiti_core/graphiti.py

@@ -76,6 +76,7 @@ from graphiti_core.utils.maintenance.node_operations import (
     resolve_extracted_nodes,
 )
 from graphiti_core.utils.maintenance.temporal_operations import get_edge_contradictions
+from graphiti_core.utils.ontology_utils.entity_types_utils import validate_entity_types
 
 logger = logging.getLogger(__name__)
 
@@ -319,6 +320,8 @@ class Graphiti:
             entity_edges: list[EntityEdge] = []
             now = utc_now()
 
+            validate_entity_types(entity_types)
+
             previous_episodes = (
                 await self.retrieve_episodes(
                     reference_time, last_n=RELEVANT_SCHEMA_LIMIT, group_ids=[group_id]

graphiti_core/utils/ontology_utils/entity_types_utils.py

@@ -0,0 +1,37 @@
+"""
+Copyright 2024, Zep Software, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+from pydantic import BaseModel
+
+from graphiti_core.errors import EntityTypeValidationError
+from graphiti_core.nodes import EntityNode
+
+
+def validate_entity_types(
+    entity_types: dict[str, BaseModel] | None,
+) -> bool:
+    if entity_types is None:
+        return True
+
+    entity_node_field_names = EntityNode.model_fields.keys()
+
+    for entity_type_name, entity_type_model in entity_types.items():
+        entity_type_field_names = entity_type_model.model_fields.keys()
+        for entity_type_field_name in entity_type_field_names:
+            if entity_type_field_name in entity_node_field_names:
+                raise EntityTypeValidationError(entity_type_name, entity_type_field_name)
+
+    return True

{graphiti_core-0.8.7.dist-info → graphiti_core-0.8.8.dist-info}/METADATA

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

{graphiti_core-0.8.7.dist-info → graphiti_core-0.8.8.dist-info}/RECORD

@@ -8,8 +8,8 @@ graphiti_core/embedder/__init__.py,sha256=eWd-0sPxflnYXLoWNT9sxwCIFun5JNO9Fk4E-Z
 graphiti_core/embedder/client.py,sha256=HKIlpPLnzFT81jurPkry6z8F8nxfZVfejdcfxHVUSFU,995
 graphiti_core/embedder/openai.py,sha256=23BnPA10eiaa1HkxHKYSj75-0PymczPK2FNNIz8Txbc,1910
 graphiti_core/embedder/voyage.py,sha256=7kqrLG75J3Q6cdA2Nlx1JSYtpk2141ckdl3OtDDw0vU,1882
-graphiti_core/errors.py,sha256=cSOXXkydihNd6OHXzkwJvciRmR7EoFMq57AzUYVg0gc,2040
-graphiti_core/graphiti.py,sha256=FXkfgaib_JcVPHCBnJP__a4kJBlDs00ctvrtDCLHyoE,29588
+graphiti_core/errors.py,sha256=Nib1uQx2cO_VOizupmRjpFfmuRg-hFAVqTtZAuBehR8,2405
+graphiti_core/graphiti.py,sha256=KaDlOIhaAPskmVbR-hQf5XHt98gY3zGGa6BjcNc5xSg,29725
 graphiti_core/helpers.py,sha256=7BQzUBFmoBDA2OIDdFtoN4W-vXOhPRIsF0uDb7PsNi0,2913
 graphiti_core/llm_client/__init__.py,sha256=PA80TSMeX-sUXITXEAxMDEt3gtfZgcJrGJUcyds1mSo,207
 graphiti_core/llm_client/anthropic_client.py,sha256=dTM8rKhk9TZAU4O-0jFMivOwJvWM-gHpp5gLmuJHiGQ,2723
@@ -55,7 +55,8 @@ graphiti_core/utils/maintenance/graph_data_operations.py,sha256=UJlIAawswJ7eO70o
 graphiti_core/utils/maintenance/node_operations.py,sha256=yo5Qt7i5WN3J3rB_9RJ0TJ1ZG1cGFWUKGrktMDwmXyE,15521
 graphiti_core/utils/maintenance/temporal_operations.py,sha256=RdNtubCyYhOVrvcOIq2WppHls1Q-BEjtsN8r38l-Rtc,3691
 graphiti_core/utils/maintenance/utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-graphiti_core-0.8.7.dist-info/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
-graphiti_core-0.8.7.dist-info/METADATA,sha256=XrGUFhTTqF1IyDIuugdQeUTPaxyXEk-FkuyhqbBQmMw,14351
-graphiti_core-0.8.7.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-graphiti_core-0.8.7.dist-info/RECORD,,
+graphiti_core/utils/ontology_utils/entity_types_utils.py,sha256=QJX5cG0GSSNF_Mm_yrldr69wjVAbN_MxLhOSznz85Hk,1279
+graphiti_core-0.8.8.dist-info/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
+graphiti_core-0.8.8.dist-info/METADATA,sha256=q-Vnd7mLYOooY2-ZveX_Klw_ofinQGxGcc-Td8-P2Hk,14351
+graphiti_core-0.8.8.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+graphiti_core-0.8.8.dist-info/RECORD,,