mirror of
https://github.com/microsoft/autogen.git
synced 2025-07-23 00:42:54 +00:00
e1a823fb6d
* initial impl of topics and subscriptions * Update python/src/agnext/core/_agent_runtime.py Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com> * add topic in context * migrate * migrate code for topics * migrate team one * edit notebooks * formatting * fix imports * Build proto * Fix circular import --------- Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com>
50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
from dataclasses import dataclass
|
|
from typing import Any
|
|
|
|
from agnext.components import TypeRoutedAgent, message_handler
|
|
from agnext.core import BaseAgent
|
|
from agnext.core import MessageContext
|
|
|
|
|
|
@dataclass
|
|
class MessageType:
|
|
...
|
|
|
|
@dataclass
|
|
class CascadingMessageType:
|
|
round: int
|
|
|
|
class LoopbackAgent(TypeRoutedAgent):
|
|
def __init__(self) -> None:
|
|
super().__init__("A loop back agent.")
|
|
self.num_calls = 0
|
|
|
|
|
|
@message_handler
|
|
async def on_new_message(self, message: MessageType, ctx: MessageContext) -> MessageType:
|
|
self.num_calls += 1
|
|
return message
|
|
|
|
|
|
class CascadingAgent(TypeRoutedAgent):
|
|
|
|
def __init__(self, max_rounds: int) -> None:
|
|
super().__init__("A cascading agent.")
|
|
self.num_calls = 0
|
|
self.max_rounds = max_rounds
|
|
|
|
@message_handler
|
|
async def on_new_message(self, message: CascadingMessageType, ctx: MessageContext) -> None:
|
|
self.num_calls += 1
|
|
if message.round == self.max_rounds:
|
|
return
|
|
assert ctx.topic_id is not None
|
|
await self.publish_message(CascadingMessageType(round=message.round + 1), topic_id=ctx.topic_id)
|
|
|
|
class NoopAgent(BaseAgent):
|
|
def __init__(self) -> None:
|
|
super().__init__("A no op agent")
|
|
|
|
async def on_message(self, message: Any, ctx: MessageContext) -> Any:
|
|
raise NotImplementedError
|