47 lines
1.4 KiB
C#
47 lines
1.4 KiB
C#
|
using BotSharp.Core.Crontab.Abstraction;
|
||
|
|
using BotSharp.Core.Crontab.Models;
|
||
|
|
using Microsoft.AspNetCore.SignalR;
|
||
|
|
|
||
|
|
namespace BotSharp.Plugin.ChatHub.Hooks;
|
||
|
|
|
||
|
|
public class ChatHubCrontabHook : ICrontabHook
|
||
|
|
{
|
||
|
|
private readonly IServiceProvider _services;
|
||
|
|
private readonly IHubContext<SignalRHub> _chatHub;
|
||
|
|
private readonly IUserIdentity _user;
|
||
|
|
private readonly IConversationStorage _storage;
|
||
|
|
private readonly BotSharpOptions _options;
|
||
|
|
|
||
|
|
public ChatHubCrontabHook(IServiceProvider services,
|
||
|
|
IHubContext<SignalRHub> chatHub,
|
||
|
|
IUserIdentity user,
|
||
|
|
IConversationStorage storage,
|
||
|
|
BotSharpOptions options)
|
||
|
|
{
|
||
|
|
_services = services;
|
||
|
|
_chatHub = chatHub;
|
||
|
|
_user = user;
|
||
|
|
_storage = storage;
|
||
|
|
_options = options;
|
||
|
|
}
|
||
|
|
|
||
|
|
public async Task OnCronTriggered(CrontabItem item)
|
||
|
|
{
|
||
|
|
var json = JsonSerializer.Serialize(new ChatResponseModel()
|
||
|
|
{
|
||
|
|
ConversationId = item.ConversationId,
|
||
|
|
MessageId = Guid.NewGuid().ToString(),
|
||
|
|
Text = item.ExecutionResult,
|
||
|
|
Function = "",
|
||
|
|
Sender = new UserViewModel()
|
||
|
|
{
|
||
|
|
FirstName = "Crontab",
|
||
|
|
LastName = "AI",
|
||
|
|
Role = AgentRole.Assistant
|
||
|
|
}
|
||
|
|
}, _options.JsonSerializerOptions);
|
||
|
|
|
||
|
|
await _chatHub.Clients.User(item.UserId).SendAsync("OnNotificationGenerated", json);
|
||
|
|
}
|
||
|
|
}
|