graphiti-core 0.12.0rc1__py3-none-any.whl → 0.24.3__py3-none-any.whl

graphiti_core/utils/maintenance/temporal_operations.py

@@ -43,7 +43,9 @@ async def extract_edge_dates(
         'reference_timestamp': current_episode.valid_at.isoformat(),
     }
     llm_response = await llm_client.generate_response(
-        prompt_library.extract_edge_dates.v1(context), response_model=EdgeDates
+        prompt_library.extract_edge_dates.v1(context),
+        response_model=EdgeDates,
+        prompt_name='extract_edge_dates.v1',
     )
 
     valid_at = llm_response.get('valid_at')
@@ -70,7 +72,9 @@ async def extract_edge_dates(
 
 
 async def get_edge_contradictions(
-    llm_client: LLMClient, new_edge: EntityEdge, existing_edges: list[EntityEdge]
+    llm_client: LLMClient,
+    new_edge: EntityEdge,
+    existing_edges: list[EntityEdge],
 ) -> list[EntityEdge]:
     start = time()
 
@@ -79,12 +83,16 @@ async def get_edge_contradictions(
         {'id': i, 'fact': existing_edge.fact} for i, existing_edge in enumerate(existing_edges)
     ]
 
-    context = {'new_edge': new_edge_context, 'existing_edges': existing_edge_context}
+    context = {
+        'new_edge': new_edge_context,
+        'existing_edges': existing_edge_context,
+    }
 
     llm_response = await llm_client.generate_response(
         prompt_library.invalidate_edges.v2(context),
         response_model=InvalidatedEdges,
         model_size=ModelSize.small,
+        prompt_name='invalidate_edges.v2',
     )
 
     contradicted_facts: list[int] = llm_response.get('contradicted_facts', [])

graphiti_core/utils/ontology_utils/entity_types_utils.py

@@ -21,7 +21,7 @@ from graphiti_core.nodes import EntityNode
 
 
 def validate_entity_types(
-    entity_types: dict[str, BaseModel] | None,
+    entity_types: dict[str, type[BaseModel]] | None,
 ) -> bool:
     if entity_types is None:
         return True

graphiti_core/utils/text_utils.py

@@ -0,0 +1,53 @@
+"""
+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.
+"""
+
+import re
+
+# Maximum length for entity/node summaries
+MAX_SUMMARY_CHARS = 500
+
+
+def truncate_at_sentence(text: str, max_chars: int) -> str:
+    """
+    Truncate text at or about max_chars while respecting sentence boundaries.
+
+    Attempts to truncate at the last complete sentence before max_chars.
+    If no sentence boundary is found before max_chars, truncates at max_chars.
+
+    Args:
+        text: The text to truncate
+        max_chars: Maximum number of characters
+
+    Returns:
+        Truncated text
+    """
+    if not text or len(text) <= max_chars:
+        return text
+
+    # Find all sentence boundaries (., !, ?) up to max_chars
+    truncated = text[:max_chars]
+
+    # Look for sentence boundaries: period, exclamation, or question mark followed by space or end
+    sentence_pattern = r'[.!?](?:\s|$)'
+    matches = list(re.finditer(sentence_pattern, truncated))
+
+    if matches:
+        # Truncate at the last sentence boundary found
+        last_match = matches[-1]
+        return text[: last_match.end()].rstrip()
+
+    # No sentence boundary found, truncate at max_chars
+    return truncated.rstrip()