from backend.database.schemas import Chunk
from backend.graph.dependency_graph import dependency_graph_store
from backend.retrieval.bm25 import tokenize


def graph_boost(repo_id: str, query: str, chunks: list[Chunk]) -> dict[str, float]:
    terms = tokenize(query)
    related = dependency_graph_store.neighbors_for_terms(repo_id, terms)
    boosts: dict[str, float] = {}
    for chunk in chunks:
        haystack = f"{chunk.path} {chunk.symbol} {chunk.content[:500]}".lower()
        if any(item.lower() in haystack for item in related):
            boosts[chunk.id] = 0.15
    return boosts