Reviewing agent

Reviewing Agent class.

ReviewingAgent

Bases: Agent

Reviewing Agent module.

Source code in agentuniverse/agent/default/reviewing_agent/reviewing_agent.py

Python

class ReviewingAgent(Agent):
    """Reviewing Agent module."""

    def input_keys(self) -> list[str]:
        """Return the input keys of the Agent."""
        return ['input', 'expressing_result']

    def output_keys(self) -> list[str]:
        """Return the output keys of the Agent."""
        return ['output']

    def parse_input(self, input_object: InputObject, agent_input: dict) -> dict:
        """Agent parameter parsing.

        Args:
            input_object(InputObject): input parameters passed by the user.
            agent_input(dict): agent input preparsed by the agent.
        Returns:
            dict: agent input parsed from `input_object` by the user.
        """
        agent_input['input'] = input_object.get_data('input')
        agent_input['expressing_result'] = input_object.get_data('expressing_result').get_data('output')
        self.agent_model.profile.setdefault('prompt_version', 'default_reviewing_agent.cn')
        return agent_input

    def parse_result(self, planner_result: dict) -> dict:
        """Planner result parser.

        Args:
            planner_result(dict): Planner result
        Returns:
            dict: Agent result object.
        """
        agent_result = dict()

        output = planner_result.get('output')
        output = json.loads(output)
        is_useful = output.get('is_useful')
        if is_useful is None:
            is_useful = False
        is_useful = bool(is_useful)
        if is_useful:
            score = 80
        else:
            score = 0

        agent_result['output'] = output
        agent_result['score'] = score
        agent_result['suggestion'] = output.get('suggestion')
        return agent_result

input_keys()

Return the input keys of the Agent.

Source code in agentuniverse/agent/default/reviewing_agent/reviewing_agent.py

Python

def input_keys(self) -> list[str]:
    """Return the input keys of the Agent."""
    return ['input', 'expressing_result']

output_keys()

Return the output keys of the Agent.

Source code in agentuniverse/agent/default/reviewing_agent/reviewing_agent.py

Python

def output_keys(self) -> list[str]:
    """Return the output keys of the Agent."""
    return ['output']

parse_input(input_object, agent_input)

Agent parameter parsing.

Parameters:

Name	Type	Description	Default
input_object(InputObject)		input parameters passed by the user.	required
agent_input(dict)		agent input preparsed by the agent.	required

Source code in agentuniverse/agent/default/reviewing_agent/reviewing_agent.py

Python

def parse_input(self, input_object: InputObject, agent_input: dict) -> dict:
    """Agent parameter parsing.

    Args:
        input_object(InputObject): input parameters passed by the user.
        agent_input(dict): agent input preparsed by the agent.
    Returns:
        dict: agent input parsed from `input_object` by the user.
    """
    agent_input['input'] = input_object.get_data('input')
    agent_input['expressing_result'] = input_object.get_data('expressing_result').get_data('output')
    self.agent_model.profile.setdefault('prompt_version', 'default_reviewing_agent.cn')
    return agent_input

parse_result(planner_result)

Planner result parser.

Parameters:

Name	Type	Description	Default
planner_result(dict)		Planner result	required

Source code in agentuniverse/agent/default/reviewing_agent/reviewing_agent.py

Python

def parse_result(self, planner_result: dict) -> dict:
    """Planner result parser.

    Args:
        planner_result(dict): Planner result
    Returns:
        dict: Agent result object.
    """
    agent_result = dict()

    output = planner_result.get('output')
    output = json.loads(output)
    is_useful = output.get('is_useful')
    if is_useful is None:
        is_useful = False
    is_useful = bool(is_useful)
    if is_useful:
        score = 80
    else:
        score = 0

    agent_result['output'] = output
    agent_result['score'] = score
    agent_result['suggestion'] = output.get('suggestion')
    return agent_result