修改答案合成的策略

This commit is contained in:
wanyaokun
2024-09-06 08:23:17 +08:00
parent 9a09e9f79f
commit 64ba7efcdc
3 changed files with 8 additions and 7 deletions
+3 -2
View File
@@ -11,6 +11,7 @@ from app.engine.prompt import ReActChatFormatter_messages, tree_summary_query_en
#from app.engine.loaders.db import makeDescriptionByEngine
from app.engine.tools import ToolFactory
from app.api.routers.request.base import ProjectInfo
from llama_index.core.response_synthesizers import ResponseMode
def getPrjFalg(params:dict=None)->str:
prjFlag = ''
@@ -42,11 +43,11 @@ def get_chat_engine(filters=None, params:dict=None):
description=summary_query_tool_messages,
)
query_engine = create_query_engine(index,top_k,use_reranker,filters,response_mode = "COMPACT")
query_engine = create_query_engine(index,top_k,use_reranker,filters,response_mode = ResponseMode.TREE_SUMMARIZE)
query_engine_tool = QueryEngineTool.from_defaults(query_engine=query_engine, name="zj_query_tool",
description=query_engine_tool_messages)
query_engine = create_query_engine(index,top_k,use_reranker,filters,response_mode = "TREE_SUMMARIZE")
query_engine = create_query_engine(index,top_k,use_reranker,filters,response_mode = ResponseMode.TREE_SUMMARIZE)
query_engine_tool_1 = QueryEngineTool.from_defaults(query_engine=query_engine, name="zj_query_tool_1",
description=tree_summary_query_engine_tool_messages)