backend/.env.xinference

@@ -6,12 +6,13 @@ SQL_DATABASE_URL=mysql+pymysql://zjinfo1:Dy2Bcr53Hm5xRkba@110.42.234.166:3306/zj
 # The number of similar embeddings to return when retrieving documents.
 TOP_K=10
 #--------------------------
-# 是否启用检索重排功能
-RERANK_ENABLED=true
 # 是否启用混合检索
 HYBRID_ENABLED = true
 # 混合检索阈值
 HYBRID_ALPHA = 0.6
+#--------------------------
+# 是否启用检索重排功能
+RERANK_ENABLED=true
 # Rerank model
 RERANK_MODEL=bge-reranker-v2-m3
 RERANK_BASE_URL=http://10.1.16.39:9995

backend/app/engine/__init__.py

@@ -4,7 +4,7 @@ from llama_index.core.agent import AgentRunner, ReActChatFormatter
 from llama_index.core.settings import Settings
 from llama_index.core.tools.query_engine import QueryEngineTool
 
-from app.engine.engine import create_query_engine, create_summary_query_engine
+from app.engine.engine import create_query_engine, create_summary_query_engine, create_sql_query_engine
 from app.engine.index import get_index
 #from app.engine.loaders.db import makeDescriptionByEngine
 from app.engine.tools import ToolFactory
@@ -17,11 +17,11 @@ def get_chat_engine(filters=None, params=None):
     tools = []
 
     # 创建SQL查询工具
-#    sql_query_engine = create_summary_query_engine(index)
+    sql_query_engine = create_sql_query_engine()
-    # sql_query_tool = QueryEngineTool.from_defaults(query_engine=sql_query_engine,
+    sql_query_tool = QueryEngineTool.from_defaults(query_engine=sql_query_engine,
-    #                                                name="zjdata_query_tool",
+                                                   name="zjdata_query_tool",
-    #                                                description="来源于一个由博微公司电力造价软件编制的造价工程文件。该文件以多张表格的形式存储存储了整个工程的全部数据内容。适用于以详细的自然语言查询表格数据方式查询造价工程各项具体属性、费用的数值。请先使用“zj_query_tool”无法解决才使用本工具"
+                                                   description="来源于一个由博微公司电力造价软件编制的造价工程文件。该文件以多张表格的形式存储存储了整个工程的全部数据内容。适用于以详细的自然语言查询表格数据方式查询造价工程各项具体属性、费用的数值。请先使用“zj_query_tool”无法解决才使用本工具"
-    #                                                )
+                                                   )
     #tools.append(sql_query_tool)
 
     # Add query tool if index exists

backend/app/engine/engine.py

@@ -52,8 +52,8 @@ def get_Retriever(index,**kwargs):
 sql_database = None
 sql_obj_index = None
 
-# Create a summary query engine
+# Create a sql query engine
-def create_summary_query_engine(top_k=3, use_reranker=False, filters=None):
+def create_sql_query_engine(top_k=3, use_reranker=False, filters=None):
     global sql_obj_index
     global sql_database
     if sql_obj_index is None or sql_database is None: