ChainGraph now includes node types

kreneskyp · Aug 8, 2023 · 7c67380 · 7c67380
1 parent 20c5c16
commit 7c67380
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/ix/api/chains/endpoints.py b/ix/api/chains/endpoints.py
@@ -289,6 +289,7 @@ class GraphModel(BaseModel):
     chain: ChainPydantic
     nodes: List[NodePydantic]
     edges: List[EdgePydantic]
+    types: List[NodeTypePydantic]
 
 
 @router.get("/chains/{chain_id}/graph", response_model=GraphModel, tags=["Chains"])
@@ -306,8 +307,12 @@ async def get_chain_graph(chain_id: UUID):
     async for edge in edge_queryset:
         edges.append(EdgePydantic.from_orm(edge))
 
+    types_in_chain = NodeType.objects.filter(chainnode__chain_id=chain_id)
+    types = [NodeTypePydantic.from_orm(node_type) async for node_type in types_in_chain]
+
     return GraphModel(
         chain=ChainPydantic.from_orm(chain),
         nodes=nodes,
         edges=edges,
+        types=types,
     )