graphiti-core 0.18.2__tar.gz → 0.18.3__tar.gz

{graphiti_core-0.18.2 → graphiti_core-0.18.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: graphiti-core
-Version: 0.18.2
+Version: 0.18.3
 Summary: A temporal graph building library
 Project-URL: Homepage, https://help.getzep.com/graphiti/graphiti/overview
 Project-URL: Repository, https://github.com/getzep/graphiti

{graphiti_core-0.18.2 → graphiti_core-0.18.3}/examples/podcast/podcast_runner.py

@@ -65,6 +65,12 @@ class Person(BaseModel):
     occupation: str | None = Field(..., description="The person's work occupation")
 
 
+class City(BaseModel):
+    """A city"""
+
+    country: str | None = Field(..., description='The country the city is in')
+
+
 class IsPresidentOf(BaseModel):
     """Relationship between a person and the entity they are a president of"""
 
@@ -92,7 +98,7 @@ async def main(use_bulk: bool = False):
         await client.add_episode_bulk(
             raw_episodes,
             group_id=group_id,
-            entity_types={'Person': Person},
+            entity_types={'Person': Person, 'City': City},
             edge_types={'IS_PRESIDENT_OF': IsPresidentOf},
             edge_type_map={('Person', 'Entity'): ['IS_PRESIDENT_OF']},
         )
@@ -109,7 +115,7 @@ async def main(use_bulk: bool = False):
                 reference_time=message.actual_timestamp,
                 source_description='Podcast Transcript',
                 group_id=group_id,
-                entity_types={'Person': Person},
+                entity_types={'Person': Person, 'City': City},
                 edge_types={'IS_PRESIDENT_OF': IsPresidentOf},
                 edge_type_map={('Person', 'Entity'): ['PRESIDENT_OF']},
                 previous_episode_uuids=episode_uuids,

{graphiti_core-0.18.2 → graphiti_core-0.18.3}/graphiti_core/graph_queries.py

@@ -44,6 +44,7 @@ def get_range_indices(provider: GraphProvider) -> list[LiteralString]:
         'CREATE INDEX has_member_uuid IF NOT EXISTS FOR ()-[e:HAS_MEMBER]-() ON (e.uuid)',
         'CREATE INDEX entity_group_id IF NOT EXISTS FOR (n:Entity) ON (n.group_id)',
         'CREATE INDEX episode_group_id IF NOT EXISTS FOR (n:Episodic) ON (n.group_id)',
+        'CREATE INDEX community_group_id IF NOT EXISTS FOR (n:Community) ON (n.group_id)',
         'CREATE INDEX relation_group_id IF NOT EXISTS FOR ()-[e:RELATES_TO]-() ON (e.group_id)',
         'CREATE INDEX mention_group_id IF NOT EXISTS FOR ()-[e:MENTIONS]-() ON (e.group_id)',
         'CREATE INDEX name_entity_index IF NOT EXISTS FOR (n:Entity) ON (n.name)',

{graphiti_core-0.18.2 → graphiti_core-0.18.3}/graphiti_core/graphiti.py

@@ -26,7 +26,12 @@ from graphiti_core.cross_encoder.client import CrossEncoderClient
 from graphiti_core.cross_encoder.openai_reranker_client import OpenAIRerankerClient
 from graphiti_core.driver.driver import GraphDriver
 from graphiti_core.driver.neo4j_driver import Neo4jDriver
-from graphiti_core.edges import CommunityEdge, EntityEdge, EpisodicEdge
+from graphiti_core.edges import (
+    CommunityEdge,
+    EntityEdge,
+    EpisodicEdge,
+    create_entity_edge_embeddings,
+)
 from graphiti_core.embedder import EmbedderClient, OpenAIEmbedder
 from graphiti_core.graphiti_types import GraphitiClients
 from graphiti_core.helpers import (
@@ -36,7 +41,13 @@ from graphiti_core.helpers import (
     validate_group_id,
 )
 from graphiti_core.llm_client import LLMClient, OpenAIClient
-from graphiti_core.nodes import CommunityNode, EntityNode, EpisodeType, EpisodicNode
+from graphiti_core.nodes import (
+    CommunityNode,
+    EntityNode,
+    EpisodeType,
+    EpisodicNode,
+    create_entity_node_embeddings,
+)
 from graphiti_core.search.search import SearchConfig, search
 from graphiti_core.search.search_config import DEFAULT_SEARCH_LIMIT, SearchResults
 from graphiti_core.search.search_config_recipes import (
@@ -984,7 +995,7 @@ class Graphiti:
         if edge.fact_embedding is None:
             await edge.generate_embedding(self.embedder)
 
-        resolved_nodes, uuid_map, _ = await resolve_extracted_nodes(
+        nodes, uuid_map, _ = await resolve_extracted_nodes(
             self.clients,
             [source_node, target_node],
         )
@@ -1012,9 +1023,12 @@ class Graphiti:
             ),
         )
 
-        await add_nodes_and_edges_bulk(
-            self.driver, [], [], resolved_nodes, [resolved_edge] + invalidated_edges, self.embedder
-        )
+        edges: list[EntityEdge] = [resolved_edge] + invalidated_edges
+
+        await create_entity_edge_embeddings(self.embedder, edges)
+        await create_entity_node_embeddings(self.embedder, nodes)
+
+        await add_nodes_and_edges_bulk(self.driver, [], [], nodes, edges, self.embedder)
 
     async def remove_episode(self, episode_uuid: str):
         # Find the episode to be deleted

{graphiti_core-0.18.2 → graphiti_core-0.18.3}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "graphiti-core"
 description = "A temporal graph building library"
-version = "0.18.2"
+version = "0.18.3"
 authors = [
     { name = "Paul Paliychuk", email = "paul@getzep.com" },
     { name = "Preston Rasmussen", email = "preston@getzep.com" },

{graphiti_core-0.18.2 → graphiti_core-0.18.3}/signatures/version1/cla.json

@@ -263,6 +263,14 @@
       "created_at": "2025-07-31T15:58:08Z",
       "repoId": 840056306,
       "pullRequestNo": 793
+    },
+    {
+      "name": "hugo-son",
+      "id": 141999572,
+      "comment_id": 3155009405,
+      "created_at": "2025-08-05T12:27:09Z",
+      "repoId": 840056306,
+      "pullRequestNo": 805
     }
   ]
 }

{graphiti_core-0.18.2 → graphiti_core-0.18.3}/tests/test_graphiti_int.py

@@ -64,7 +64,7 @@ async def test_graphiti_init(driver):
     await graphiti.build_indices_and_constraints()
 
     search_filter = SearchFilters(
-        node_labels=['Person'],
+        node_labels=['Person', 'City'],
         created_at=[[DateFilter(date=utc_now(), comparison_operator=ComparisonOperator.less_than)]],
     )
 

{graphiti_core-0.18.2 → graphiti_core-0.18.3}/uv.lock

@@ -746,7 +746,7 @@ wheels = [
 
 [[package]]
 name = "graphiti-core"
-version = "0.18.2"
+version = "0.18.3"
 source = { editable = "." }
 dependencies = [
     { name = "diskcache" },