github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-10-22 09:28:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(agent): Multi agent sdk (#976)

Browse Source 
Co-authored-by: xtyuns <xtyuns@163.com>
Co-authored-by: Fangyin Cheng <staneyffer@gmail.com>
Co-authored-by: csunny <cfqsunny@163.com>
Co-authored-by: qidanrui <qidanrui@gmail.com>
This commit is contained in:
明天
2023-12-27 16:25:55 +08:00
committed by GitHub
parent 69fb97e508
commit 9aec636b02
79 changed files with 6359 additions and 121 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
dbgpt/app/base.py
View File

@@ -32,7 +32,7 @@ def async_db_summary(system_app: SystemApp):
def server_init(param: "WebServerParameters", system_app: SystemApp):
from dbgpt.agent.commands.command_mange import CommandRegistry
from dbgpt.agent.plugin.commands.command_mange import CommandRegistry
# logger.info(f"args: {args}")
@@ -47,8 +47,8 @@ def server_init(param: "WebServerParameters", system_app: SystemApp):
# Loader plugins and commands
command_categories = [
"dbgpt.agent.commands.built_in.audio_text",
"dbgpt.agent.commands.built_in.image_gen",
"dbgpt.agent.plugin.commands.built_in.audio_text",
"dbgpt.agent.plugin.commands.built_in.image_gen",
]
# exclude commands
command_categories = [
@@ -61,9 +61,9 @@ def server_init(param: "WebServerParameters", system_app: SystemApp):
cfg.command_registry = command_registry
command_disply_commands = [
"dbgpt.agent.commands.disply_type.show_chart_gen",
"dbgpt.agent.commands.disply_type.show_table_gen",
"dbgpt.agent.commands.disply_type.show_text_gen",
"dbgpt.agent.plugin.commands.built_in.disply_type.show_chart_gen",
"dbgpt.agent.plugin.commands.built_in.disply_type.show_table_gen",
"dbgpt.agent.plugin.commands.built_in.disply_type.show_text_gen",
]
command_disply_registry = CommandRegistry()
for command in command_disply_commands:
@@ -133,7 +133,7 @@ def _initialize_db(try_to_create_db: Optional[bool] = False) -> str:
f"{urlquote(CFG.LOCAL_DB_PASSWORD)}@"
f"{CFG.LOCAL_DB_HOST}:"
f"{str(CFG.LOCAL_DB_PORT)}/"
f"{db_name}?charset=utf8mb4&collation=utf8mb4_unicode_ci"
f"{db_name}?charset=utf8mb4"
)
# Try to create database, if failed, will raise exception
_create_mysql_database(db_name, db_url, try_to_create_db)