87 lines
2.8 KiB
C#
87 lines
2.8 KiB
C#
using BotSharp.Abstraction.Functions;
|
|
|
|
namespace BotSharp.Core.Routing;
|
|
|
|
public partial class RoutingService
|
|
{
|
|
public async Task<bool> InvokeFunction(string name, RoleDialogModel message, Func<RoleDialogModel, Task>? onFunctionExecuting = null)
|
|
{
|
|
var function = _services.GetServices<IFunctionCallback>().FirstOrDefault(x => x.Name == name);
|
|
if (function == null)
|
|
{
|
|
message.StopCompletion = true;
|
|
message.Content = $"Can't find function implementation of {name}.";
|
|
_logger.LogError(message.Content);
|
|
return false;
|
|
}
|
|
|
|
// Clone message
|
|
var clonedMessage = RoleDialogModel.From(message);
|
|
clonedMessage.FunctionName = name;
|
|
|
|
var hooks = _services.GetServices<IConversationHook>()
|
|
.OrderBy(x => x.Priority)
|
|
.ToList();
|
|
|
|
// Before executing functions
|
|
clonedMessage.Indication = function.Indication;
|
|
if (onFunctionExecuting != null)
|
|
{
|
|
await onFunctionExecuting(clonedMessage);
|
|
}
|
|
|
|
foreach (var hook in hooks)
|
|
{
|
|
await hook.OnFunctionExecuting(clonedMessage);
|
|
}
|
|
|
|
bool result = false;
|
|
|
|
try
|
|
{
|
|
result = await function.Execute(clonedMessage);
|
|
|
|
// After functions have been executed
|
|
foreach (var hook in hooks)
|
|
{
|
|
await hook.OnFunctionExecuted(clonedMessage);
|
|
}
|
|
|
|
// Set result to original message
|
|
message.PostbackFunctionName = clonedMessage.PostbackFunctionName;
|
|
message.CurrentAgentId = clonedMessage.CurrentAgentId;
|
|
message.Content = clonedMessage.Content;
|
|
message.StopCompletion = clonedMessage.StopCompletion;
|
|
message.RichContent = clonedMessage.RichContent;
|
|
message.Data = clonedMessage.Data;
|
|
}
|
|
catch (JsonException ex)
|
|
{
|
|
_logger.LogError($"The input does not contain any JSON tokens:\r\n{message.Content}");
|
|
message.StopCompletion = true;
|
|
message.Content = ex.Message;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
message.StopCompletion = true;
|
|
message.Content = ex.Message;
|
|
_logger.LogError(ex.ToString());
|
|
}
|
|
|
|
// Make sure content has been populated
|
|
if (string.IsNullOrEmpty(message.Content) && message.Data != null)
|
|
{
|
|
message.Content = JsonSerializer.Serialize(message.Data);
|
|
}
|
|
|
|
// Save to Storage as well
|
|
/*if (!message.StopCompletion && message.FunctionName != "route_to_agent")
|
|
{
|
|
var storage = _services.GetRequiredService<IConversationStorage>();
|
|
storage.Append(Context.ConversationId, message);
|
|
}*/
|
|
|
|
return result;
|
|
}
|
|
}
|