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

feat(agent):Fix agent bug (#1953)

Browse Source 
Co-authored-by: aries_ckt <916701291@qq.com>
This commit is contained in:
明天
2024-09-04 10:59:03 +08:00
committed by GitHub
parent d72bfb2f5f
commit b951b50689
18 changed files with 67 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
dbgpt/serve/agent/agents/expand/actions/app_start_action.py
View File

@@ -50,6 +50,7 @@ class StartAppAction(Action[LinkAppInput]):
**kwargs,
) -> ActionOutput:
conv_id = kwargs.get("conv_id")
user_input = kwargs.get("user_input")
paren_agent = kwargs.get("paren_agent")
init_message_rounds = kwargs.get("init_message_rounds")
@@ -83,7 +84,7 @@ class StartAppAction(Action[LinkAppInput]):
from dbgpt.serve.agent.agents.controller import multi_agents
await multi_agents.agent_team_chat_new(
new_user_input,
new_user_input if new_user_input else user_input,
conv_id,
gpts_app,
paren_agent.memory,

24
dbgpt/serve/agent/agents/expand/actions/intent_recognition_action.py
View File

@@ -54,18 +54,28 @@ class IntentRecognitionAction(Action[IntentRecognitionInput]):
@property
def ai_out_schema(self) -> Optional[str]:
out_put_schema = {
"intent": "[The recognized intent is placed here]",
"app_code": "[App code in selected intent]",
"slots": {"意图定义中槽位属性1": "具体值", "意图定义中槽位属性2": "具体值"},
"ask_user": "If you want the user to supplement slot data, ask the user a question",
"user_input": "[Complete instructions generated based on intent and slot]",
}
if self.language == "en":
out_put_schema = {
"intent": "[The recognized intent is placed here]",
"app_code": "[App code in selected intent]",
"slots": {
"Slot attribute 1 in intent definition": "value",
"Slot attribute 2 in intent definition": "value",
},
"ask_user": "[If you want the user to supplement slot data, ask the user a question]",
"user_input": "[Complete instructions generated based on intent and slot]",
}
return f"""Please reply in the following json format:
{json.dumps(out_put_schema, indent=2, ensure_ascii=False)}
Make sure the output is only json and can be parsed by Python json.loads.""" # noqa: E501
else:
out_put_schema = {
"intent": "选择的意图放在这里",
"app_code": "选择意图对应的Appcode值",
"slots": {"意图定义中槽位属性1": "具体值", "意图定义中槽位属性2": "具体值"},
"ask_user": "如果需要用户补充槽位属性的具体值,请向用户进行提问",
"user_input": "根据意图和槽位生成完整指令问题",
}
return f"""请按如下JSON格式输出:
{json.dumps(out_put_schema, indent=2, ensure_ascii=False)}
确保输出只有json，且可以被python json.loads加载."""