gmakstutis/graphiti
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

update episode override (#219)

Browse source 
* update episode override

* remove unused import
This commit is contained in:
Preston Rasmussen 2024-12-02 11:47:06 -05:00 • committed by GitHub
parent 0fbe5c0704
commit e42d3ae46c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 14 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
graphiti_core/graphiti.py
View file

@ -318,7 +318,11 @@ class Graphiti:
previous_episodes = await self.retrieve_episodes( previous_episodes = await self.retrieve_episodes(
reference_time, last_n=RELEVANT_SCHEMA_LIMIT, group_ids=[group_id] reference_time, last_n=RELEVANT_SCHEMA_LIMIT, group_ids=[group_id]
) )
episode = EpisodicNode(
episode = (
await EpisodicNode.get_by_uuid(self.driver, uuid)
if uuid is not None
else EpisodicNode(
name=name, name=name,
group_id=group_id, group_id=group_id,
labels=[], labels=[],
@ -328,7 +332,7 @@ class Graphiti:
created_at=now, created_at=now,
valid_at=reference_time, valid_at=reference_time,
) )
episode.uuid = uuid if uuid is not None else episode.uuid )
# Extract entities as nodes # Extract entities as nodes