合并Dev分支代码

backend/app/engine/index.py

@@ -2,22 +2,20 @@ 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")
 
-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
+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