github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-30 07:11:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'plugin_init' of https://github.com/csunny/DB-GPT into plugin_init

Browse Source
This commit is contained in:
csunny 2023-05-14 22:17:48 +08:00
commit 197a6fe33a
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pilot/commands/command.py
View File

@ -108,9 +108,8 @@ def execute_command(
try: try:
# 删除非定义参数 # 删除非定义参数
diff_ags = list(set(arguments.keys()).difference(set(command['args'].keys()))) diff_ags = list(set(arguments.keys()).difference(set(command['args'].keys())))
for arg_name in arguments.keys(): for arg_name in diff_ags:
if arg_name in diff_ags: del arguments[arg_name]
del arguments[arg_name]
print(str(arguments)) print(str(arguments))
return command["function"](**arguments) return command["function"](**arguments)
except Exception as e: except Exception as e:

3
pilot/prompts/auto_mode_prompt.py
View File

@ -7,7 +7,7 @@ from pathlib import Path
import distro import distro
import yaml import yaml
from pilot.configs.config import Config from pilot.configs.config import Config
from pilot.prompts.prompt import build_default_prompt_generator, DEFAULT_TRIGGERING_PROMPT from pilot.prompts.prompt import build_default_prompt_generator, DEFAULT_PROMPT_OHTER
class AutoModePrompt: class AutoModePrompt:
@ -51,6 +51,7 @@ class AutoModePrompt:
""" """
prompt_start = ( prompt_start = (
DEFAULT_PROMPT_OHTER
) )
if prompt_generator is None: if prompt_generator is None:
prompt_generator = build_default_prompt_generator() prompt_generator = build_default_prompt_generator()