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

Solve the problem of resource duplication and solve the problem of slow recall of table structure (#1381)

Browse Source 
Co-authored-by: ChenXinye <chenxinye@weihaizixun.com>
This commit is contained in:
Evan Chen 2024-04-08 19:57:51 +08:00 committed by GitHub
parent 5a0c20a864
commit 01a8a0fff6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
dbgpt/serve/agent/team/plan/planner_agent.py
View File

@ -4,7 +4,7 @@ from pydantic import Field
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()
@ -70,11 +70,19 @@ class PlannerAgent(ConversableAgent):
}
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]: