Function registerGraphFunction

src/functions/graph.ts:453–997

(
  sdk: ISdk,
  kv: StateKV,
  provider: MemoryProvider,
)

}

export function registerGraphFunction(
  sdk: ISdk,
  kv: StateKV,
  provider: MemoryProvider,
): void {
  sdk.registerFunction("mem::graph-extract", 
    async (data: { observations: CompressedObservation[] }) => {
      if (!data.observations || data.observations.length === 0) {
        return { success: false, error: "No observations provided" };
      }

      const prompt = buildGraphExtractionPrompt(
        data.observations.map((o) => ({
          title: o.title,
          narrative: o.narrative,
          concepts: o.concepts,
          files: o.files,
          type: o.type,
        })),
      );

      try {
        const response = await provider.compress(
          GRAPH_EXTRACTION_SYSTEM,
          prompt,
        );

        const obsIds = data.observations.map((o) => o.id);
        const { nodes, edges } = parseGraphXml(response, obsIds);

        // #814 v2: targeted name-index lookups replace the O(n) scan
        // over `kv.list<GraphNode>(KV.graphNodes)`. At 75K nodes the
        // list payload exceeds the iii heartbeat budget and the worker
        // dies before merge can complete. Each name-index entry is a
        // single small kv.get/set pair.
        const snap = (await readSnapshot(kv)) ?? emptySnapshot();
        const capturedAt = new Date().toISOString();
        let newNodeCount = 0;
        let newEdgeCount = 0;
        const newEdgesForTopCheck: GraphEdge[] = [];

        for (const node of nodes) {
          const indexKey = nameIndexKey(node.type, node.name);
          const existingId = await kv.get<string>(
            KV.graphNameIndex,
            indexKey,
          );

          let existing: GraphNode | null = null;
          if (existingId) {
            existing = await kv.get<GraphNode>(KV.graphNodes, existingId);
            // #825 follow-up: name-index lookups can resolve into
            // pre-reset rows. Drop them so extract writes a fresh
            // node + index entry instead of silently reconnecting
            // to a legacy orphan (which would keep the snapshot at
            // 0 forever after a reset).
            if (
              existing &&