usky-human/digitalHuman/agent/core/repeaterAgent.py

# -*- coding: utf-8 -*-

from ..builder import AGENTS
from ..agentBase import BaseAgent
from digitalHuman.protocol import *

__all__ = ["Repeater"]


@AGENTS.register("Repeater")
class RepeaterAgent(BaseAgent):
    async def run(
        self, 
        input: TextMessage, 
        **kwargs
    ):
        yield eventStreamText(input.data)
        yield eventStreamDone()