30 lines
1.3 KiB
Python
30 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
from app.modules.rag.contracts import EvidenceLink, EvidenceType, RagDocument, RagLayer, RagSource, RagSpan
|
|
from app.modules.rag.indexing.code.edges.extractor import PyEdge
|
|
|
|
|
|
class EdgeDocumentBuilder:
|
|
def build(self, source: RagSource, edge: PyEdge) -> RagDocument:
|
|
dst = edge.dst_ref or edge.dst_symbol_id or "unknown"
|
|
return RagDocument(
|
|
layer=RagLayer.CODE_DEPENDENCY_GRAPH,
|
|
lang="python",
|
|
source=source,
|
|
title=f"{edge.src_qname}:{edge.edge_type}",
|
|
text=f"{edge.src_qname} {edge.edge_type} {dst}",
|
|
span=RagSpan(edge.start_line, edge.end_line),
|
|
metadata={
|
|
"edge_id": edge.edge_id,
|
|
"edge_type": edge.edge_type,
|
|
"src_symbol_id": edge.src_symbol_id,
|
|
"src_qname": edge.src_qname,
|
|
"dst_symbol_id": edge.dst_symbol_id,
|
|
"dst_ref": edge.dst_ref,
|
|
"resolution": edge.resolution,
|
|
"lang_payload": edge.metadata,
|
|
"artifact_type": "CODE",
|
|
},
|
|
links=[EvidenceLink(type=EvidenceType.EDGE, target_id=edge.edge_id, path=source.path, start_line=edge.start_line, end_line=edge.end_line)],
|
|
)
|