工程名称下拉项获取兼容.md文件,同时新增自定义答案合成类

This commit is contained in:
wanyaokun
2024-09-10 09:59:00 +08:00
parent 54f19a20fc
commit cb34fde995
6 changed files with 308 additions and 17 deletions
+10
View File
@@ -10,6 +10,8 @@ from sqlalchemy import create_engine
from util.register import *
from app.engine.prompt import text_qa_template, refine_template, summary_template, simple_template
from app.engine.retriever.HybridRetriever import HybridRetriever
from app.engine.response.treeSummResponse import CustomTreeResponse
from llama_index.core.settings import Settings
ModelPlateCategory = '模型平台'
@@ -65,6 +67,14 @@ def get_Retriever(index,**kwargs):
return retriever
def get_synthesizer():
return CustomTreeResponse(
llm=Settings.llm,
summary_template=summary_template,
use_async=True,
streaming=False,
)
sql_database = None
sql_obj_index = None