Spaces:
Sleeping
Sleeping
File size: 2,029 Bytes
cc54e11 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
# coding: utf-8
# Copyright (c) 2025 inclusionAI.
from typing import List, Dict, Any, Callable
from aworld.agents.llm_agent import Agent
from aworld.core.common import Observation, ActionModel, Config
from aworld.logs.util import logger
class SerialableAgent(Agent):
"""Support for serial execution of agents based on dependency relationships in the swarm."""
def __init__(self,
conf: Config,
resp_parse_func: Callable[..., Any] = None,
agents: List[Agent] = [],
**kwargs):
super().__init__(conf=conf, resp_parse_func=resp_parse_func, **kwargs)
self.agents = agents
async def async_policy(self, observation: Observation, info: Dict[str, Any] = {}, **kwargs) -> List[ActionModel]:
from aworld.config import RunConfig
from aworld.core.task import Task
from aworld.runners.utils import choose_runners, execute_runner
action = ActionModel(agent_name=self.id(), policy_info=observation.content)
for agent in self.agents:
task = Task(input=observation, agent=agent, context=self.context)
runners = await choose_runners([task])
res = await execute_runner(runners, RunConfig(reuse_process=False))
if res:
v = res.get(task.id)
action = ActionModel(agent_name=self.id(), policy_info=v.answer)
observation = self._action_to_observation(action, agent.name())
else:
raise Exception(f"{agent.id()} execute fail.")
return [action]
def _action_to_observation(self, policy: ActionModel, agent_name: str):
if not policy:
logger.warning("no agent policy, will use default error info.")
return Observation(content=f"{agent_name} no policy")
logger.debug(f"{policy.policy_info}")
return Observation(content=policy.policy_info)
def finished(self) -> bool:
return all([agent.finished for agent in self.agents])
|