from langchain_core.tools import tool
from rag_pipeline.hybrid_retriever import get_hybrid_retriever


@tool
def search_budget_rules(query: str, program_name: str = "") -> str:
    """
    Wyszukuje informacje w bazie wiedzy (RAG) na temat limitów kosztów kwalifikowanych,
    zasad finansowania, stawek ryczałtowych oraz dozwolonych budżetów dla programów dotacyjnych.

    Args:
        query (str): Pytanie dotyczące zasad budżetowych np. "Jaki jest limit kosztów pośrednich dla Ścieżki SMART?"
        program_name (str): Opcjonalnie nazwa programu, np. "FENG Ścieżka SMART".
    """
    retriever = get_hybrid_retriever()
    search_query = f"[Koszty, Budżet, Ewaluacja Finansowa] Program: {program_name}. Zapytanie: {query}"

    docs = retriever.invoke(search_query)
    if not docs:
        return "Nie znaleziono dokumentów precyzujących to zapytanie budżetowe w bazie wiedzy."

    result = "\n".join(
        [
            f"- Zródło: {d.metadata.get('source', 'nieznane')}\n{d.page_content}"
            for d in docs
        ]
    )
    return f"Wyniki wyszukiwania zasad budżetowych:\n{result}"