新增工程数据

backend/app/engine/index.py

@@ -5,17 +5,11 @@ from app.engine.loaders import get_document_Types
 from typing import Dict,Any
 logger = logging.getLogger("uvicorn")
 
-def get_index(**args):
+def get_index(prjFlag:str):
+    if prjFlag is None or prjFlag == '':
+        raise ValueError('无效的工程标识')
     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)
+    store = get_vector_store(prjFlag)
     index = VectorStoreIndex.from_vector_store(store)
     logger.info("Finished load index from vector store.")
     return index