fix(agent): Remove resources of planner agent (#1540)

This commit is contained in:
Fangyin Cheng
2024-05-22 15:38:49 +08:00
committed by GitHub
parent 04af30e3db
commit 47430f2a0b
2 changed files with 12 additions and 11 deletions

View File

@@ -1,6 +1,6 @@
"""Planner Agent."""
from typing import Any, Dict, List
from typing import Any, Dict, List, Optional
from dbgpt._private.pydantic import Field
@@ -160,6 +160,15 @@ assistants:[
self.resource = ResourcePack(resources)
return self
async def generate_resource_variables(
self, question: Optional[str] = None
) -> Dict[str, Any]:
"""Generate the resource variables."""
out_schema: Optional[str] = None
if self.actions and len(self.actions) > 0:
out_schema = self.actions[0].ai_out_schema
return {"out_schema": out_schema}
def prepare_act_param(self) -> Dict[str, Any]:
"""Prepare the parameters for the act method."""
return {

View File

@@ -291,21 +291,13 @@ class MultiAgents(BaseComponent, ABC):
else:
raise ValueError(f"Unknown Agent Team Mode!{team_mode}")
manager = (
await manager.bind(context)
.bind(llm_config)
.bind(depend_resource)
.bind(agent_memory)
.build()
await manager.bind(context).bind(llm_config).bind(agent_memory).build()
)
manager.hire(employees)
recipient = manager
user_proxy: UserProxyAgent = (
await UserProxyAgent()
.bind(context)
.bind(depend_resource)
.bind(agent_memory)
.build()
await UserProxyAgent().bind(context).bind(agent_memory).build()
)
if is_retry_chat:
# retry chat