Skip to content

Commit

Permalink
Solve the problem of resource duplication and solve the problem of sl…
Browse files Browse the repository at this point in the history
…ow recall of table structure
  • Loading branch information
ChenXinye committed Apr 7, 2024
1 parent 62f90c9 commit 7cd9752
Showing 1 changed file with 10 additions and 2 deletions.
12 changes: 10 additions & 2 deletions dbgpt/serve/agent/team/plan/planner_agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@

from dbgpt._private.config import Config
from dbgpt.agent.agents.base_agent_new import ConversableAgent

from dbgpt.agent.resource.resource_api import AgentResource
from .plan_action import PlanAction

CFG = Config()
Expand Down Expand Up @@ -70,11 +70,19 @@ def _init_reply_message(self, recive_message):
}
return reply_message

@staticmethod
def get_unique_resources_codes(resource: AgentResource) -> str:
return resource.name + "_" + resource.type.value + "_" + resource.value

def bind_agents(self, agents: List[ConversableAgent]) -> ConversableAgent:
self.agents = agents
unique_resources = set()
for agent in self.agents:
if agent.resources and len(agent.resources) > 0:
self.resources.extend(agent.resources)
for resource in agent.resources:
if self.get_unique_resources_codes(resource) not in unique_resources:
unique_resources.add(self.get_unique_resources_codes(resource))
self.resources.append(resource)
return self

def prepare_act_param(self) -> Optional[Dict]:
Expand Down

0 comments on commit 7cd9752

Please sign in to comment.