From 611b0aa54e8f4ef46600a7b8ceaaaa5e8354806d Mon Sep 17 00:00:00 2001
From: Lars <Lars@stommer.de>
Date: Thu, 4 Dec 2025 11:35:53 +0100
Subject: [PATCH] app/core/retriever.py aktualisiert

---
 app/core/retriever.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/app/core/retriever.py b/app/core/retriever.py
index a7e7a7f..f7b2640 100644
--- a/app/core/retriever.py
+++ b/app/core/retriever.py
@@ -141,8 +141,14 @@ def _build_hits_from_semantic(
         edge_bonus = 0.0
         cent_bonus = 0.0
         if subgraph is not None:
-            # WICHTIG: Knoten im Graphen sind note_ids, nicht chunk_ids.
-            node_key = payload.get("note_id")
+            # Standard-Key wie im ursprünglichen Verhalten (für Fakes in Tests):
+            node_key = payload.get("chunk_id") or payload.get("note_id")
+
+            # Falls es sich um unseren echten Subgraph-Typ handelt, wissen wir,
+            # dass Knoten als note_id modelliert sind → dann gezielt note_id nutzen.
+            if isinstance(subgraph, ga.Subgraph):
+                node_key = payload.get("note_id")
+
             if node_key:
                 try:
                     edge_bonus = float(subgraph.edge_bonus(node_key))