新增工程数据

This commit is contained in:
wanyaokun
2024-08-30 19:12:40 +08:00
parent c4088fe963
commit ced3199550
37 changed files with 157 additions and 35 deletions
+2 -10
View File
@@ -13,9 +13,7 @@ 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'))
prjFlag = ProjectInfo().prjFalg(params.get('projectname'))
return prjFlag
@@ -33,13 +31,7 @@ def get_chat_engine(filters=None, params:dict=None):
#tools.append(sql_query_tool)
# Add query tool if index exists
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))
index = get_index(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",