alihan/graphiti
1
0
Fork 0
mirror of https://github.com/getzep/graphiti.git synced 2024-09-08 19:13:11 +03:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Pavlo Paliychuk
2024-09-06 15:37:19 -04:00
committed by GitHub
parent 42fb590606
commit 39ad28d6db
3 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pyproject.toml
View File

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

2
server/graph_service/routers/ingest.py
View File

@@ -54,8 +54,8 @@ async def add_messages(
graphiti: ZepGraphitiDep, graphiti: ZepGraphitiDep,
): ):
async def add_messages_task(m: Message): 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( await graphiti.add_episode(
group_id=request.group_id,
name=m.name, name=m.name,
episode_body=f"{m.role or ''}({m.role_type}): {m.content}", episode_body=f"{m.role or ''}({m.role_type}): {m.content}",
reference_time=m.timestamp, reference_time=m.timestamp,

2
server/graph_service/routers/retrieve.py
View File

@@ -20,6 +20,7 @@ async def search(query: SearchQuery, graphiti: ZepGraphitiDep):
if user_node: if user_node:
center_node_uuid = user_node.uuid center_node_uuid = user_node.uuid
relevant_edges = await graphiti.search( relevant_edges = await graphiti.search(
group_ids=[query.group_id],
query=query.query, query=query.query,
num_results=query.max_facts, num_results=query.max_facts,
center_node_uuid=center_node_uuid, center_node_uuid=center_node_uuid,
@@ -37,6 +38,7 @@ async def get_memory(
): ):
combined_query = compose_query_from_messages(request.messages) combined_query = compose_query_from_messages(request.messages)
result = await graphiti.search( result = await graphiti.search(
group_ids=[request.group_id],
query=combined_query, query=combined_query,
num_results=request.max_facts, num_results=request.max_facts,
) )