feat(ui/circuits): add isFromQkTracing field for filtering (untested)

Author: dest1n1s

ui-ssr/src/components/circuits/link-graph/link-graph.tsx

@@ -98,7 +98,7 @@ function topologicalSort(
 }
 
 const LinkGraphComponent: React.FC<LinkGraphProps> = ({
-  data,
+  data: rawData,
   visState,
   onNodeClick,
   onNodeHover,
@@ -112,6 +112,15 @@ const LinkGraphComponent: React.FC<LinkGraphProps> = ({
     },
   )
 
+  const data = useMemo(() => {
+    const nodes = rawData.nodes.filter((n) => !n.isFromQkTracing)
+    const nodeIds = new Set(nodes.map((n) => n.nodeId))
+    const edges = rawData.edges.filter(
+      (e) => nodeIds.has(e.source) && nodeIds.has(e.target),
+    )
+    return { ...rawData, nodes, edges }
+  }, [rawData])
+
   // 1. Calculate stats about context counts and total units needed
   const { calculatedCtxCounts, totalUnits } = useMemo(() => {
     if (!data.nodes.length) {

ui-ssr/src/types/circuit.ts

@@ -19,6 +19,7 @@ export const FeatureNodeSchema = z.object({
   activation: z.number(),
   feature: FeatureSchema,
   qkTracingResults: QKTracingResultsSchema.nullish(),
+  isFromQkTracing: z.boolean().default(false),
 })
 
 export type FeatureNode = z.infer<typeof FeatureNodeSchema>
@@ -29,6 +30,7 @@ export const TokenNodeSchema = z.object({
   layer: z.number(),
   ctxIdx: z.number(),
   token: z.string(),
+  isFromQkTracing: z.boolean().default(false),
 })
 
 export type TokenNode = z.infer<typeof TokenNodeSchema>
@@ -38,6 +40,7 @@ export const ErrorNodeSchema = z.object({
   nodeId: z.string(),
   layer: z.number(),
   ctxIdx: z.number(),
+  isFromQkTracing: z.boolean().default(false),
 })
 
 export type ErrorNode = z.infer<typeof ErrorNodeSchema>
@@ -49,6 +52,7 @@ export const LogitNodeSchema = z.object({
   ctxIdx: z.number(),
   tokenProb: z.number(),
   token: z.string(),
+  isFromQkTracing: z.boolean().default(false),
 })
 
 export type LogitNode = z.infer<typeof LogitNodeSchema>