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", )