diff --git a/pyproject.toml b/pyproject.toml index 1101867..3976e44 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "graphiti-core" -version = "0.2.2" +version = "0.2.3" description = "A temporal graph building library" authors = [ "Paul Paliychuk ", diff --git a/server/graph_service/routers/ingest.py b/server/graph_service/routers/ingest.py index f1f1ae4..e98013c 100644 --- a/server/graph_service/routers/ingest.py +++ b/server/graph_service/routers/ingest.py @@ -54,8 +54,8 @@ async def add_messages( graphiti: ZepGraphitiDep, ): async def add_messages_task(m: Message): - # Will pass a group_id to the add_episode call once it is implemented await graphiti.add_episode( + group_id=request.group_id, name=m.name, episode_body=f"{m.role or ''}({m.role_type}): {m.content}", reference_time=m.timestamp, diff --git a/server/graph_service/routers/retrieve.py b/server/graph_service/routers/retrieve.py index fbc2698..8f449ed 100644 --- a/server/graph_service/routers/retrieve.py +++ b/server/graph_service/routers/retrieve.py @@ -20,6 +20,7 @@ async def search(query: SearchQuery, graphiti: ZepGraphitiDep): if user_node: center_node_uuid = user_node.uuid relevant_edges = await graphiti.search( + group_ids=[query.group_id], query=query.query, num_results=query.max_facts, center_node_uuid=center_node_uuid, @@ -37,6 +38,7 @@ async def get_memory( ): combined_query = compose_query_from_messages(request.messages) result = await graphiti.search( + group_ids=[request.group_id], query=combined_query, num_results=request.max_facts, )