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

feat(dbgpts): fix dbgpts bug

Browse Source
This commit is contained in:
yhjun1026
2024-08-29 20:38:23 +08:00
parent fb47392701
commit d34a11eb75
54 changed files with 61 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
dbgpt/agent/core/base_agent.py
View File

@@ -341,6 +341,9 @@ class ConversableAgent(Role, Agent):
received_message=received_message,
rely_messages=rely_messages,
)
retry_message.rounds = reply_message.rounds + 1
retry_message.content = fail_reason
retry_message.current_goal = received_message.current_goal
@@ -351,6 +354,7 @@ class ConversableAgent(Role, Agent):
await sender.send(
retry_message, self, reviewer, request_reply=False
)
reply_message.rounds = retry_message.rounds + 1
# In manual retry mode, load all messages of the last speaker as dependent messages # noqa
logger.info(

6
dbgpt/agent/core/plan/team_auto_plan.py
View File

@@ -168,6 +168,7 @@ class AutoPlanChatManager(ManagerAgent):
)
speaker: Agent = sender
final_message = message.content
rounds = message.rounds
for i in range(self.max_round):
if not self.memory:
return ActionOutput(
@@ -196,11 +197,12 @@ class AutoPlanChatManager(ManagerAgent):
plan_message = await planner.generate_reply(
received_message=AgentMessage.from_llm_message(
{"content": message.content}
{"content": message.content, "rounds": rounds}
),
sender=self,
reviewer=reviewer,
)
rounds = plan_message.rounds
await planner.send(
message=plan_message, recipient=self, request_reply=False
)
@@ -228,6 +230,7 @@ class AutoPlanChatManager(ManagerAgent):
"plan_task": now_plan.sub_task_content,
"plan_task_num": now_plan.sub_task_num,
},
rounds=rounds + 1,
)
# select the next speaker
speaker, model = await self.select_speaker(
@@ -264,6 +267,7 @@ class AutoPlanChatManager(ManagerAgent):
await speaker.send(
agent_reply_message, self, reviewer, request_reply=False
)
rounds = agent_reply_message.rounds
plan_result = ""
final_message = reply_message["content"]