mirror of
https://github.com/getzep/graphiti.git
synced 2025-11-09 06:47:42 +00:00
7f7a17c926
* Bump version from 0.9.0 to 0.9.1 in pyproject.toml and update google-genai dependency to >=0.1.0 * Bump version from 0.9.1 to 0.9.2 in pyproject.toml * Update google-genai dependency version to >=0.8.0 in pyproject.toml * loc file * Update pyproject.toml to version 0.9.3, restructure dependencies, and modify author format. Remove outdated Google API key note from README.md. * upgrade poetry and ruff * Update README.md to include installation instructions for Graphiti with Google Gemini support * fix to deps since peotry doesn't fully implement PEP 735 * Refactor string formatting in various files to use single quotes for consistency and improve readability. This includes updates in agent.ipynb, quickstart.py, multiple prompt files, and ingest.py and retrieve.py modules. * Remove optional dependencies from pyproject.toml to streamline project requirements.
64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
from datetime import datetime, timezone
|
|
|
|
from fastapi import APIRouter, status
|
|
|
|
from graph_service.dto import (
|
|
GetMemoryRequest,
|
|
GetMemoryResponse,
|
|
Message,
|
|
SearchQuery,
|
|
SearchResults,
|
|
)
|
|
from graph_service.zep_graphiti import ZepGraphitiDep, get_fact_result_from_edge
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.post('/search', status_code=status.HTTP_200_OK)
|
|
async def search(query: SearchQuery, graphiti: ZepGraphitiDep):
|
|
relevant_edges = await graphiti.search(
|
|
group_ids=query.group_ids,
|
|
query=query.query,
|
|
num_results=query.max_facts,
|
|
)
|
|
facts = [get_fact_result_from_edge(edge) for edge in relevant_edges]
|
|
return SearchResults(
|
|
facts=facts,
|
|
)
|
|
|
|
|
|
@router.get('/entity-edge/{uuid}', status_code=status.HTTP_200_OK)
|
|
async def get_entity_edge(uuid: str, graphiti: ZepGraphitiDep):
|
|
entity_edge = await graphiti.get_entity_edge(uuid)
|
|
return get_fact_result_from_edge(entity_edge)
|
|
|
|
|
|
@router.get('/episodes/{group_id}', status_code=status.HTTP_200_OK)
|
|
async def get_episodes(group_id: str, last_n: int, graphiti: ZepGraphitiDep):
|
|
episodes = await graphiti.retrieve_episodes(
|
|
group_ids=[group_id], last_n=last_n, reference_time=datetime.now(timezone.utc)
|
|
)
|
|
return episodes
|
|
|
|
|
|
@router.post('/get-memory', status_code=status.HTTP_200_OK)
|
|
async def get_memory(
|
|
request: GetMemoryRequest,
|
|
graphiti: ZepGraphitiDep,
|
|
):
|
|
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,
|
|
)
|
|
facts = [get_fact_result_from_edge(edge) for edge in result]
|
|
return GetMemoryResponse(facts=facts)
|
|
|
|
|
|
def compose_query_from_messages(messages: list[Message]):
|
|
combined_query = ''
|
|
for message in messages:
|
|
combined_query += f'{message.role_type or ""}({message.role or ""}): {message.content}\n'
|
|
return combined_query
|