From e71da586e3aa73b8e63dbb77b2187d366bb292f6 Mon Sep 17 00:00:00 2001 From: paituo <330435863@qq.com> Date: Thu, 22 Aug 2024 16:02:07 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E7=BC=BA=E9=99=B7?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- backend/app/engine/__init__.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/backend/app/engine/__init__.py b/backend/app/engine/__init__.py index a224477..7a351cc 100644 --- a/backend/app/engine/__init__.py +++ b/backend/app/engine/__init__.py @@ -1,15 +1,13 @@ import os +from llama_index.core.agent import AgentRunner, ReActChatFormatter from llama_index.core.settings import Settings -from llama_index.core.agent import AgentRunner, StructuredPlannerAgent, FunctionCallingAgentWorker, ReActChatFormatter from llama_index.core.tools.query_engine import QueryEngineTool -from sqlalchemy import create_engine, Engine -from llama_index.core.response_synthesizers.type import ResponseMode 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.engine.index import get_index def get_chat_engine(filters=None, params=None): @@ -32,7 +30,7 @@ def get_chat_engine(filters=None, params=None): summary_query_tool = QueryEngineTool.from_defaults( query_engine=summary_query_engine, name="summary_query_tool", description="适用于任何需要进行全面总结、概括的要求。", ) - query_engine = create_query_engine() + query_engine = create_query_engine(index) query_engine_tool = QueryEngineTool.from_defaults(query_engine=query_engine, name="zj_query_tool", )