新增工程信息、检索的知识片段节点回传、下一轮建议问题列表

backend/app/engine/__init__.py

@@ -8,9 +8,18 @@ from app.engine.engine import create_query_engine, create_summary_query_engine
 from app.engine.index import get_index
 #from app.engine.loaders.db import makeDescriptionByEngine
 from app.engine.tools import ToolFactory
+from app.api.routers.request.base import ProjectInfo
+
+def getPrjFalg(params:dict=None)->str:
+    prjFlag = ''
+    if params is not None:
+       inputs:dict = params.get('inputs')
+       if inputs is not None:
+           prjFlag = ProjectInfo.prjFalg(inputs.get('projectname'))
+    return prjFlag
 
 
-def get_chat_engine(filters=None, params=None,**args):
+def get_chat_engine(filters=None, params:dict=None):
     system_prompt = os.getenv("SYSTEM_PROMPT")
     top_k = int(os.getenv("TOP_K", "3"))
     use_reranker = os.getenv("RERANK_ENABLED")
@@ -24,7 +33,13 @@ def get_chat_engine(filters=None, params=None,**args):
     #tools.append(sql_query_tool)
 
     # Add query tool if index exists
-    index = get_index(**args)
+    prjFlag = ''
+    if params is not None:
+       inputs:dict = params.get('inputs')
+       if inputs is not None:
+           prjFlag = inputs.get('projectname')
+
+    index = get_index(prjFlag = getPrjFalg(params))
     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",
@@ -57,6 +72,7 @@ def get_chat_engine(filters=None, params=None,**args):
         verbose=True,
     )
     return agentrunner
+
     # create the function calling worker for reasoning
     # worker = FunctionCallingAgentWorker.from_tools(
     #     tools, verbose=True