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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: graphiti-core
-Version: 0.18.2
+Version: 0.18.4
 Summary: A temporal graph building library
 Project-URL: Homepage, https://help.getzep.com/graphiti/graphiti/overview
 Project-URL: Repository, https://github.com/getzep/graphiti
@@ -369,7 +369,7 @@ graphiti = Graphiti(
     "neo4j",
     "password",
     llm_client=OpenAIClient(
-        llm_config=azure_llm_config,
+        config=azure_llm_config,
         client=llm_client_azure
     ),
     embedder=OpenAIEmbedder(
@@ -379,7 +379,7 @@ graphiti = Graphiti(
         client=embedding_client_azure
     ),
     cross_encoder=OpenAIRerankerClient(
-        llm_config=LLMConfig(
+        config=LLMConfig(
             model=azure_llm_config.small_model  # Use small model for reranking
         ),
         client=llm_client_azure

{graphiti_core-0.18.2 → graphiti_core-0.18.4}/README.md

@@ -317,7 +317,7 @@ graphiti = Graphiti(
     "neo4j",
     "password",
     llm_client=OpenAIClient(
-        llm_config=azure_llm_config,
+        config=azure_llm_config,
         client=llm_client_azure
     ),
     embedder=OpenAIEmbedder(
@@ -327,7 +327,7 @@ graphiti = Graphiti(
         client=embedding_client_azure
     ),
     cross_encoder=OpenAIRerankerClient(
-        llm_config=LLMConfig(
+        config=LLMConfig(
             model=azure_llm_config.small_model  # Use small model for reranking
         ),
         client=llm_client_azure

{graphiti_core-0.18.2 → graphiti_core-0.18.4}/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.4}/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.4}/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.4}/graphiti_core/nodes.py

@@ -118,7 +118,7 @@ class Node(BaseModel, ABC):
         return False
 
     @classmethod
-    async def delete_by_group_id(cls, driver: GraphDriver, group_id: str):
+    async def delete_by_group_id(cls, driver: GraphDriver, group_id: str, batch_size: int = 100):
         if driver.provider == GraphProvider.FALKORDB:
             for label in ['Entity', 'Episodic', 'Community']:
                 await driver.execute_query(
@@ -129,13 +129,18 @@ class Node(BaseModel, ABC):
                     group_id=group_id,
                 )
         else:
-            await driver.execute_query(
-                """
-                MATCH (n:Entity|Episodic|Community {group_id: $group_id})
-                DETACH DELETE n
-                """,
-                group_id=group_id,
-            )
+            async with driver.session() as session:
+                await session.run(
+                    """
+                    MATCH (n:Entity|Episodic|Community {group_id: $group_id})
+                    CALL {
+                        WITH n
+                        DETACH DELETE n
+                    } IN TRANSACTIONS OF $batch_size ROWS
+                    """,
+                    group_id=group_id,
+                    batch_size=batch_size,
+                )
 
     @classmethod
     async def get_by_uuid(cls, driver: GraphDriver, uuid: str): ...

{graphiti_core-0.18.2 → graphiti_core-0.18.4}/graphiti_core/search/search_filters.py

@@ -31,7 +31,7 @@ class ComparisonOperator(Enum):
 
 
 class DateFilter(BaseModel):
-    date: datetime = Field(description='A datetime to filter on')
+    date: datetime | None = Field(description='A datetime to filter on')
     comparison_operator: ComparisonOperator = Field(
         description='Comparison operator for date filter'
     )

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

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

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

@@ -263,6 +263,30 @@
       "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
+    },
+    {
+      "name": "mvanders",
+      "id": 758617,
+      "comment_id": 3160523661,
+      "created_at": "2025-08-06T14:56:21Z",
+      "repoId": 840056306,
+      "pullRequestNo": 808
+    },
+    {
+      "name": "v-khanna",
+      "id": 102773390,
+      "comment_id": 3162200130,
+      "created_at": "2025-08-07T02:23:09Z",
+      "repoId": 840056306,
+      "pullRequestNo": 812
     }
   ]
 }

{graphiti_core-0.18.2 → graphiti_core-0.18.4}/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.4}/uv.lock

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