graphiti-core 0.8.5__tar.gz → 0.8.7__tar.gz

{graphiti_core-0.8.5 → graphiti_core-0.8.7}/PKG-INFO

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

{graphiti_core-0.8.5 → graphiti_core-0.8.7}/graphiti_core/graphiti.py

@@ -262,6 +262,7 @@ class Graphiti:
         uuid: str | None = None,
         update_communities: bool = False,
         entity_types: dict[str, BaseModel] | None = None,
+        previous_episode_uuids: list[str] | None = None,
     ) -> AddEpisodeResults:
         """
         Process an episode and update the graph.
@@ -287,6 +288,9 @@ class Graphiti:
             Optional uuid of the episode.
         update_communities : bool
             Optional. Whether to update communities with new node information
+        previous_episode_uuids : list[str] | None
+            Optional.  list of episode uuids to use as the previous episodes. If this is not provided,
+            the most recent episodes by created_at date will be used.
 
         Returns
         -------
@@ -315,8 +319,12 @@ class Graphiti:
             entity_edges: list[EntityEdge] = []
             now = utc_now()
 
-            previous_episodes = await self.retrieve_episodes(
-                reference_time, last_n=RELEVANT_SCHEMA_LIMIT, group_ids=[group_id]
+            previous_episodes = (
+                await self.retrieve_episodes(
+                    reference_time, last_n=RELEVANT_SCHEMA_LIMIT, group_ids=[group_id]
+                )
+                if previous_episode_uuids is None
+                else await EpisodicNode.get_by_uuids(self.driver, previous_episode_uuids)
             )
 
             episode = (

{graphiti_core-0.8.5 → graphiti_core-0.8.7}/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.5 → graphiti_core-0.8.7}/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.5 → graphiti_core-0.8.7}/pyproject.toml

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