github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-03 17:39:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(Agent): ChatAgent And AgentHub

Browse Source 
1.Upgrade sqlalchemy to version 2.0
This commit is contained in:
yhjun1026 2023-10-16 12:00:46 +08:00
parent 1c90cbc64c
commit 0e8fe405b1
5 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pilot/base_modules/meta_data/meta_data.py
View File

@ -72,7 +72,7 @@ else:
Session = sessionmaker(autocommit=False, autoflush=False, bind=engine) Session = sessionmaker(autocommit=False, autoflush=False, bind=engine)
session = Session() session = Session()
Base = declarative_base(bind=engine) Base = declarative_base()
# Base.metadata.create_all() # Base.metadata.create_all()
@ -84,6 +84,8 @@ alembic_cfg = AlembicConfig(alembic_ini_path)
alembic_cfg.set_main_option('sqlalchemy.url', str(engine.url)) alembic_cfg.set_main_option('sqlalchemy.url', str(engine.url))
os.makedirs(default_db_path + "/alembic", exist_ok=True) os.makedirs(default_db_path + "/alembic", exist_ok=True)
os.makedirs(default_db_path + "/alembic/versions", exist_ok=True)
alembic_cfg.set_main_option('script_location', default_db_path + "/alembic") alembic_cfg.set_main_option('script_location', default_db_path + "/alembic")
# 将模型和会话传递给Alembic配置 # 将模型和会话传递给Alembic配置

2
pilot/scene/chat_agent/chat.py
View File

@ -31,7 +31,7 @@ class ChatAgent(BaseChat):
self.plugins_prompt_generator.command_registry = CFG.command_registry self.plugins_prompt_generator.command_registry = CFG.command_registry
# load select plugin # load select plugin
agent_module = CFG.SYSTEM_APP.get_componet(ComponentType.AGENT_HUB, ModuleAgent) agent_module = CFG.SYSTEM_APP.get_component(ComponentType.AGENT_HUB, ModuleAgent)
self.plugins_prompt_generator = agent_module.load_select_plugin(self.plugins_prompt_generator, self.select_plugins) self.plugins_prompt_generator = agent_module.load_select_plugin(self.plugins_prompt_generator, self.select_plugins)
self.api_call = ApiCall(plugin_generator=self.plugins_prompt_generator) self.api_call = ApiCall(plugin_generator=self.plugins_prompt_generator)

4
pilot/server/component_configs.py
View File

@ -26,9 +26,11 @@ def initialize_components(
# Register global default executor factory first # Register global default executor factory first
system_app.register(DefaultExecutorFactory) system_app.register(DefaultExecutorFactory)
system_app.register_instance(controller) system_app.register_instance(controller)
from pilot.base_modules.agent.controller import module_agent
system_app.register_instance(module_agent)
_initialize_embedding_model( _initialize_embedding_model(
param, system_app, embedding_model_name, embedding_model_path param, system_app, embedding_model_name, embedding_model_path
) )

3
pilot/server/dbgpt_server.py
View File

@ -81,9 +81,8 @@ app.include_router(api_fb_v1, prefix="/api", tags=["FeedBack"])
app.include_router(knowledge_router, tags=["Knowledge"]) app.include_router(knowledge_router, tags=["Knowledge"])
app.include_router(prompt_router, tags=["Prompt"]) app.include_router(prompt_router, tags=["Prompt"])
# app.include_router(api_editor_route_v1)
@app.get("/openapi.json") @app.get("/openapi.json")
async def get_openapi_endpoint(): async def get_openapi_endpoint():

2
setup.py
View File

@ -308,7 +308,7 @@ def core_requires():
"auto-gpt-plugin-template", "auto-gpt-plugin-template",
"gTTS==2.3.1", "gTTS==2.3.1",
"langchain>=0.0.286", "langchain>=0.0.286",
"SQLAlchemy", "SQLAlchemy==2.0.22",
"pymysql", "pymysql",
"duckdb==0.8.1", "duckdb==0.8.1",
"duckdb-engine", "duckdb-engine",