app/models/graph_context.py

from pydantic import BaseModel


class GraphNode(BaseModel):

    qualified_name: str | None = None

    labels: list[str]

    properties: dict


class GraphEdge(BaseModel):

    source: str

    target: str

    type: str


class GraphContext(BaseModel):

    nodes: list[GraphNode]

    edges: list[GraphEdge]