import logging
from llama_index.core.indices import VectorStoreIndex
from app.engine.vectordb import get_vector_store
from app.engine.loaders import get_document_Types
from typing import Dict,Any
logger = logging.getLogger("uvicorn")

def get_index(**args):
    logger.info("Connecting vector store...")
    prjFlags = get_document_Types()
    if len(prjFlags)<=0:
        return None
    prjFlag = args.get('prjFlag','')
    flag = prjFlags[0] if prjFlag not in prjFlags else prjFlag
    store = get_vector_store(flag)
    # Load the index from the vector store
    # If you are using a vector store that doesn't store text,
    # you must load the index from both the vector store and the document store
    index = VectorStoreIndex.from_vector_store(store)
    logger.info("Finished load index from vector store.")
    return index