routers/pinecone.py

import logging
from fastapi import APIRouter, Request, HTTPException

logger = logging.getLogger(__name__)

router = APIRouter()


@router.get("/stats")
def stats(request: Request):
    pinecone_service = request.app.state.pinecone_service
    logger.info("Pinecone — stats requested")
    return pinecone_service.index.describe_index_stats()


@router.get("/fetch/{record_id}")
def fetch(record_id: str, request: Request):
    pinecone_service = request.app.state.pinecone_service
    logger.info(f"Pinecone — fetch: \"{record_id}\"")
    result = pinecone_service.fetch_by_id(record_id)
    if not result:
        raise HTTPException(status_code=404, detail=f"Record \"{record_id}\" not found")
    return result