mirror of
https://github.com/getzep/graphiti.git
synced 2025-06-27 02:00:02 +00:00
ba48f64492
* chore: Folder rearrangement * chore: Remove unused deps, and add mypy step in CI for graph-service * fix: Mypy errors * fix: linter * fix mypy * fix mypy * chore: Update docker setup * chore: Reduce graph service image size * chore: Install graph service deps on CI * remove cache from typecheck * chore: install graph-service deps on typecheck action * update graph service mypy direction * feat: Add release service image step * chore: Update depot configuration * chore: Update release image job to run on releases * chore: Test depot multiplatform build * update release action tag * chore: Update action to be in accordance with zep image publish * test * test * revert * chore: Update python slim image used in service docker * chore: Remove unused endpoints and dtos
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
from typing import Annotated
|
|
|
|
from fastapi import Depends
|
|
from graphiti_core import Graphiti # type: ignore
|
|
from graphiti_core.edges import EntityEdge # type: ignore
|
|
from graphiti_core.llm_client import LLMClient # type: ignore
|
|
from graphiti_core.nodes import EntityNode # type: ignore
|
|
|
|
from graph_service.config import ZepEnvDep
|
|
from graph_service.dto import FactResult
|
|
|
|
|
|
class ZepGraphiti(Graphiti):
|
|
def __init__(
|
|
self, uri: str, user: str, password: str, user_id: str, llm_client: LLMClient | None = None
|
|
):
|
|
super().__init__(uri, user, password, llm_client)
|
|
self.user_id = user_id
|
|
|
|
async def get_user_node(self, user_id: str) -> EntityNode | None: ...
|
|
|
|
|
|
async def get_graphiti(settings: ZepEnvDep):
|
|
client = ZepGraphiti(
|
|
uri=settings.neo4j_uri,
|
|
user=settings.neo4j_user,
|
|
password=settings.neo4j_password,
|
|
user_id='test1234',
|
|
)
|
|
try:
|
|
yield client
|
|
finally:
|
|
client.close()
|
|
|
|
|
|
def get_fact_result_from_edge(edge: EntityEdge):
|
|
return FactResult(
|
|
uuid=edge.uuid,
|
|
name=edge.name,
|
|
fact=edge.fact,
|
|
valid_at=edge.valid_at,
|
|
invalid_at=edge.invalid_at,
|
|
created_at=edge.created_at,
|
|
expired_at=edge.expired_at,
|
|
)
|
|
|
|
|
|
ZepGraphitiDep = Annotated[ZepGraphiti, Depends(get_graphiti)]
|