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

24
graphiti_core/graphiti.py
View file

@ -318,17 +318,21 @@ 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(
name=name, episode = (
group_id=group_id, await EpisodicNode.get_by_uuid(self.driver, uuid)
labels=[], if uuid is not None
source=source, else EpisodicNode(
content=episode_body, name=name,
source_description=source_description, group_id=group_id,
created_at=now, labels=[],
valid_at=reference_time, source=source,
content=episode_body,
source_description=source_description,
created_at=now,
valid_at=reference_time,
)
) )
episode.uuid = uuid if uuid is not None else episode.uuid
# Extract entities as nodes # Extract entities as nodes