wanyaokun
22 lines
723 B
Python
22 lines
723 B
Python
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...")
|
|
if 'prjFlag' in args:
|
|
prjFlags = get_document_Types()
|
|
if len(prjFlags)<=0:
|
|
return None
|
|
prjFlag = args.get('prjFlag','')
|
|
flag = prjFlags[0] if prjFlag not in prjFlags else prjFlag
|
|
else:
|
|
flag = ''
|
|
store = get_vector_store(flag)
|
|
index = VectorStoreIndex.from_vector_store(store)
|
|
logger.info("Finished load index from vector store.")
|
|
return index
|