Skip to content

Commit 81eed7d

Browse files
committed
remove src_code from node
1 parent cf60d35 commit 81eed7d

2 files changed

Lines changed: 8 additions & 2 deletions

File tree

app/api/repo/[graph]/[node]/route.ts

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,10 @@ export async function GET(request: NextRequest, { params }: { params: { graph: s
2525
let nodes = res.data[0]['nodes'];
2626
let edges = res.data[0]['edges'];
2727

28-
nodes.forEach((node: any) => { delete node.properties?.src_embeddings});
28+
nodes.forEach((node: any) => {
29+
delete node.properties?.src_code;
30+
delete node.properties?.src_embeddings;
31+
});
2932

3033
return NextResponse.json({ id: graphId, nodes: nodes, edges: edges }, { status: 200 })
3134
}

app/api/repo/route.ts

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -364,7 +364,10 @@ export async function POST(request: NextRequest) {
364364

365365
let code_graph = await GraphOps.projectGraph(graph, 600);
366366

367-
code_graph.nodes.forEach((node: any) => { delete node.properties?.src_embeddings});
367+
code_graph.nodes.forEach((node: any) => {
368+
delete node.properties?.src_code;
369+
delete node.properties?.src_embeddings;
370+
});
368371

369372
return NextResponse.json({ id: graphId, nodes: code_graph.nodes, edges: code_graph.edges }, { status: 201 })
370373
}

0 commit comments

Comments
 (0)