routes/api/query.py

from fastapi import APIRouter, HTTPException
from llama_index.core.indices.base import BaseIndex
from llama_index.core.query_engine import BaseQueryEngine
from components.indexers.news_indexer import get_or_build_index
import os

router = APIRouter()

DATA_PATH = "data/raw"
INDEX_PATH = "storage/index"

_index: BaseIndex = None

def get_index() -> BaseIndex:
    global _index
    if _index is None:
        if not os.path.exists(DATA_PATH):
            raise RuntimeError("❌ `data/raw/` is missing! Add documents for indexing.")
        _index = get_or_build_index(DATA_PATH, INDEX_PATH)
    return _index

@router.get("/query-news")
def query_news(q: str):
    index = get_index()
    query_engine: BaseQueryEngine = index.as_query_engine()
    response = query_engine.query(q)
    return {"response": str(response)}