remove src_code from node

FalkorDB · Jan 15, 2024 · 81eed7d · 81eed7d
1 parent cf60d35
commit 81eed7d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/app/api/repo/[graph]/[node]/route.ts b/app/api/repo/[graph]/[node]/route.ts
@@ -25,7 +25,10 @@ export async function GET(request: NextRequest, { params }: { params: { graph: s
     let nodes = res.data[0]['nodes'];
     let edges = res.data[0]['edges'];
 
-    nodes.forEach((node: any) => { delete node.properties?.src_embeddings});
+    nodes.forEach((node: any) => {
+      delete node.properties?.src_code;
+      delete node.properties?.src_embeddings;
+    });
 
     return NextResponse.json({ id: graphId, nodes: nodes, edges: edges }, { status: 200 })
 }
diff --git a/app/api/repo/route.ts b/app/api/repo/route.ts
@@ -364,7 +364,10 @@ export async function POST(request: NextRequest) {
 
 	let code_graph = await GraphOps.projectGraph(graph, 600);
 
-	code_graph.nodes.forEach((node: any) => { delete node.properties?.src_embeddings});
+	code_graph.nodes.forEach((node: any) => {
+		delete node.properties?.src_code;
+		delete node.properties?.src_embeddings;
+	});
 
 	return NextResponse.json({ id: graphId, nodes: code_graph.nodes, edges: code_graph.edges }, { status: 201 })
 }