wangtao
/
autogen
mirror of https://github.com/microsoft/autogen.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

fix handler selection

This commit is contained in:
Jack Gerrits 2025-01-27 16:09:05 -05:00 committed by Jack Gerrits
parent 2fd6eee080
commit fd6a369422
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
dotnet/src/Microsoft.AutoGen/Core/BaseAgent.cs
View File

@ -71,8 +71,12 @@ public abstract class BaseAgent : IAgent, IHostableAgent
if (_handlersByMessageType.TryGetValue(messageType, out var handlerMethod))
{
// Determine if this is a IHandle<T> or IHandle<T, U> method
var genericArguments = handlerMethod.GetParameters();
if (genericArguments.Length == 1)
// We need to check if return type is a bare ValueTask or ValueTask<T>
var ret = handlerMethod.ReturnType;
var genericArguments = ret.GetGenericArguments();
// The non-returning type uses ValueTask
if (genericArguments.Length == 0)
{
// This is a IHandle<T> method
var return_value = handlerMethod.Invoke(this, new object[] { message, messageContext });
@ -82,7 +86,8 @@ public abstract class BaseAgent : IAgent, IHostableAgent
return ValueTask.CompletedTask;
}
else if (genericArguments.Length == 2)
// The returning type uses ValueTask<T>
else if (genericArguments.Length == 1)
{
// This is a IHandle<T, U> method
// var _messageType = genericArguments[0];