diff --git a/faust_bootstrap/core/app/app_faust.py b/faust_bootstrap/core/app/app_faust.py index 6ac357a..dd6a855 100644 --- a/faust_bootstrap/core/app/app_faust.py +++ b/faust_bootstrap/core/app/app_faust.py @@ -40,7 +40,7 @@ def __init__(self): self._cli = FaustCli() def create_agent(self, agent_function: AgentFun[Callable], channel: Union[str, ChannelT] = None, **kwargs): - self.app.agent(channel, **kwargs)(agent_function) + return self.app.agent(channel, **kwargs)(agent_function) def register_parameter(self, name: str, required: bool, help: str, type: Any, **kwargs): self._cli.add_option(name, required, help, type, **kwargs)