wanyaokun
24 lines
853 B
Python
24 lines
853 B
Python
import logging
|
|
from llama_index.core.indices import VectorStoreIndex
|
|
from app.engine.vectordb import get_vector_store
|
|
from app.engine.generate import get_document_Types
|
|
|
|
logger = logging.getLogger("uvicorn")
|
|
|
|
indexs = {}
|
|
|
|
def get_index(params=None):
|
|
global indexs
|
|
if len(index) <= 0:
|
|
logger.info("Connecting vector store...")
|
|
docTypes = get_document_Types()
|
|
for docType in docTypes:
|
|
store = get_vector_store(docType)
|
|
# 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.")
|
|
indexs[docType] = index
|
|
return indexs
|