github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-02-12 00:34:56 -05:00
Code Issues Packages Projects Releases Wiki Activity

Simplify handler decorator (#50)

Browse Source 
* Simplify handler decorator

* add more tests

* mypy

* formatting

* fix 3.10 and improve type handling of decorator

* test fix

* format
This commit is contained in:
Jack Gerrits
2024-06-05 08:51:49 -04:00
committed by GitHub
parent ad513d5017
commit 8cb530f65e
10 changed files with 191 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/futures.py
View File

@@ -16,7 +16,7 @@ class Inner(TypeRoutedAgent):
def __init__(self, name: str, router: AgentRuntime) -> None:
super().__init__(name, router)
@message_handler(MessageType)
@message_handler()
async def on_new_message(self, message: MessageType, cancellation_token: CancellationToken) -> MessageType:
return MessageType(body=f"Inner: {message.body}", sender=self.name)
@@ -26,7 +26,7 @@ class Outer(TypeRoutedAgent):
super().__init__(name, router)
self._inner = inner
@message_handler(MessageType)
@message_handler()
async def on_new_message(self, message: MessageType, cancellation_token: CancellationToken) -> MessageType:
inner_response = self._send_message(message, self._inner)
inner_message = await inner_response