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

feat!: Add message context to signature of intervention handler, add more to docs (#4882)

Browse Source 
* Add message context to signature of intervention handler, add more to docs

* example

* Add to test

* Fix pyright

* mypy
This commit is contained in:
Jack Gerrits
2025-01-07 12:51:35 -05:00
committed by GitHub
parent f4382f01c8
commit 5b9be79fba
6 changed files with 113 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
python/packages/autogen-core/samples/slow_human_in_loop.py
View File

@@ -31,7 +31,6 @@ from dataclasses import dataclass
from typing import Any, Mapping, Optional
from autogen_core import (
AgentId,
CancellationToken,
DefaultInterventionHandler,
DefaultTopicId,
@@ -211,7 +210,7 @@ class NeedsUserInputHandler(DefaultInterventionHandler):
def __init__(self):
self.question_for_user: GetSlowUserMessage | None = None
async def on_publish(self, message: Any, *, sender: AgentId | None) -> Any:
async def on_publish(self, message: Any, *, message_context: MessageContext) -> Any:
if isinstance(message, GetSlowUserMessage):
self.question_for_user = message
return message
@@ -231,7 +230,7 @@ class TerminationHandler(DefaultInterventionHandler):
def __init__(self):
self.terminateMessage: TerminateMessage | None = None
async def on_publish(self, message: Any, *, sender: AgentId | None) -> Any:
async def on_publish(self, message: Any, *, message_context: MessageContext) -> Any:
if isinstance(message, TerminateMessage):
self.terminateMessage = message
return message