feat: Add group id support to service (#92)

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 <paul@getzep.com>",

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,

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,
     )