@neat.is/core 0.4.20-dev.20260619 → 0.4.20-dev.20260621

package/dist/index.cjs

@@ -1953,13 +1953,14 @@ var parentSpanCache = /* @__PURE__ */ new Map();
 function parentSpanKey(traceId, spanId) {
   return `${traceId}:${spanId}`;
 }
-function cacheSpanService(span, now) {
+function cacheSpanService(span, now, callSite) {
   if (!span.traceId || !span.spanId) return;
   const key = parentSpanKey(span.traceId, span.spanId);
   parentSpanCache.delete(key);
   parentSpanCache.set(key, {
     service: span.service,
     env: span.env ?? "unknown",
+    ...callSite ? { callSite } : {},
     expiresAt: now + PARENT_SPAN_CACHE_TTL_MS
   });
   while (parentSpanCache.size > PARENT_SPAN_CACHE_SIZE) {
@@ -1975,7 +1976,11 @@ function lookupParentSpan(traceId, parentSpanId, now) {
     parentSpanCache.delete(parentSpanKey(traceId, parentSpanId));
     return null;
   }
-  return { service: entry.service, env: entry.env };
+  return {
+    service: entry.service,
+    env: entry.env,
+    ...entry.callSite ? { callSite: entry.callSite } : {}
+  };
 }
 function resolveServiceId(graph, host, env) {
   const envTagged = (0, import_types3.serviceId)(host, env);
@@ -2248,9 +2253,9 @@ async function handleSpan(ctx, span) {
   }
   const sourceId = ensureServiceNode(ctx.graph, span.service, env);
   const isError = span.statusCode === 2;
-  cacheSpanService(span, nowMs);
   const sourceServiceNode = ctx.graph.getNodeAttributes(sourceId);
   const callSite = callSiteFromSpan(span, sourceServiceNode, ctx.scanPath);
+  cacheSpanService(span, nowMs, callSite);
   const observedSource = () => callSite ? ensureObservedFileNode(ctx.graph, span.service, sourceId, callSite) : sourceId;
   const callSiteEvidence = callSite ? { file: callSite.relPath, ...callSite.line !== void 0 ? { line: callSite.line } : {} } : void 0;
   let affectedNode = sourceId;
@@ -2307,13 +2312,19 @@ async function handleSpan(ctx, span) {
       const parent = lookupParentSpan(span.traceId, span.parentSpanId, nowMs);
       if (parent && parent.service !== span.service) {
         const parentId = ensureServiceNode(ctx.graph, parent.service, parent.env);
+        const fallbackSource = parent.callSite ? ensureObservedFileNode(ctx.graph, parent.service, parentId, parent.callSite) : parentId;
+        const fallbackEvidence = parent.callSite ? {
+          file: parent.callSite.relPath,
+          ...parent.callSite.line !== void 0 ? { line: parent.callSite.line } : {}
+        } : void 0;
         upsertObservedEdge(
           ctx.graph,
           import_types3.EdgeType.CALLS,
-          parentId,
+          fallbackSource,
           sourceId,
           ts,
-          isError
+          isError,
+          fallbackEvidence
         );
       }
     }