Coverage for node / src / stigmem_node / models / graph.py: 100%

1"""Graph route wire-format models.""" 

2 

3from __future__ import annotations 

4 

5from pydantic import BaseModel, Field 

6 

7 

8class NeighborItem(BaseModel): 

9 entity: str 

10 relation: str 

11 hops: int 

12 confidence: float 

13 source_trust: float | None = None 

14 path: list[str] = Field(default_factory=list) 

15 

16 

17class NeighborsResponse(BaseModel): 

18 entity: str 

19 depth: int 

20 neighbors: list[NeighborItem] 

21 next_cursor: str | None = None 

22 total_hint: int