SemianiakaVY/BotSharp
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Auto restore function name in InvokeFunction.

Browse source
This commit is contained in:
Haiping Chen 2024-02-27 14:59:08 -06:00
parent 2582ecd578
commit 18721a6b8f
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/Infrastructure/BotSharp.Core/Routing/RoutingService.InvokeFunction.cs
View file

@ -9,8 +9,17 @@ public partial class RoutingService
var function = _services.GetServices<IFunctionCallback>().FirstOrDefault(x => x.Name == name);
if (function == null) return false;
var originalFunctionName = message.FunctionName;
message.FunctionName = name;
message.Role = AgentRole.Function;
return await function.Execute(message);
var result = await function.Execute(message);
// restore original function name
if (!message.StopCompletion)
{
message.FunctionName = originalFunctionName;
}
return result;
}
}