合并Dev分支代码

backend/app/engine/__init__.py

@@ -10,12 +10,11 @@ from app.engine.index import get_index
 from app.engine.tools import ToolFactory
 
 
-def get_chat_engine(filters=None, params=None):
+def get_chat_engine(filters=None, params=None,**args):
     system_prompt = os.getenv("SYSTEM_PROMPT")
     top_k = int(os.getenv("TOP_K", "3"))
     use_reranker = os.getenv("RERANK_ENABLED")
     tools = []
-
     # 创建SQL查询工具
 #    sql_query_engine = create_summary_query_engine(index)
     # sql_query_tool = QueryEngineTool.from_defaults(query_engine=sql_query_engine,
@@ -25,9 +24,7 @@ def get_chat_engine(filters=None, params=None):
     #tools.append(sql_query_tool)
 
     # Add query tool if index exists
-    indexs = get_index()
-    if len(indexs) > 0:
-        index = list(indexs.values())[0]
+    index = get_index(**args)
     if index is not None:
         summary_query_engine = create_summary_query_engine(index,top_k,use_reranker,filters)
         summary_query_tool = QueryEngineTool.from_defaults( query_engine=summary_query_engine, name="summary_query_tool",