-
-
Notifications
You must be signed in to change notification settings - Fork 551
Add new email reader utility to read emails using IMAP. #568
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Merged
Merged
Changes from 1 commit
Commits
Show all changes
3 commits
Select commit
Hold shift + click to select a range
File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
37 changes: 37 additions & 0 deletions
37
src/Plugins/BotSharp.Plugin.EmailReader/BotSharp.Plugin.EmailReader.csproj
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,37 @@ | ||
<Project Sdk="Microsoft.NET.Sdk"> | ||
|
||
<PropertyGroup> | ||
<TargetFramework>$(TargetFramework)</TargetFramework> | ||
<Nullable>enable</Nullable> | ||
<LangVersion>$(LangVersion)</LangVersion> | ||
<VersionPrefix>$(BotSharpVersion)</VersionPrefix> | ||
<GeneratePackageOnBuild>$(GeneratePackageOnBuild)</GeneratePackageOnBuild> | ||
<GenerateDocumentationFile>$(GenerateDocumentationFile)</GenerateDocumentationFile> | ||
<OutputPath>$(SolutionDir)packages</OutputPath> | ||
</PropertyGroup> | ||
|
||
<ItemGroup> | ||
<None Remove="data\agents\6745151e-6d46-4a02-8de4-1c4f21c7da95\functions\handle_email_reader.json" /> | ||
<None Remove="data\agents\6745151e-6d46-4a02-8de4-1c4f21c7da95\templates\handle_email_reader.fn.liquid" /> | ||
</ItemGroup> | ||
|
||
<ItemGroup> | ||
<Content Include="data\agents\6745151e-6d46-4a02-8de4-1c4f21c7da95\functions\handle_email_reader.json"> | ||
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory> | ||
</Content> | ||
<Content Include="data\agents\6745151e-6d46-4a02-8de4-1c4f21c7da95\templates\handle_email_reader.fn.liquid"> | ||
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory> | ||
</Content> | ||
</ItemGroup> | ||
|
||
<ItemGroup> | ||
<PackageReference Include="MailKit" Version="4.7.0" /> | ||
</ItemGroup> | ||
|
||
<ItemGroup> | ||
<ProjectReference Include="..\..\..\..\onebrain\BusinessCore\BusinessCore.csproj" /> | ||
<ProjectReference Include="..\..\Infrastructure\BotSharp.Abstraction\BotSharp.Abstraction.csproj" /> | ||
<ProjectReference Include="..\..\Infrastructure\BotSharp.Core\BotSharp.Core.csproj" /> | ||
</ItemGroup> | ||
|
||
</Project> |
41 changes: 41 additions & 0 deletions
41
src/Plugins/BotSharp.Plugin.EmailReader/EmailReaderPlugin.cs
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,41 @@ | ||
using BotSharp.Abstraction.Agents; | ||
using BotSharp.Abstraction.Repositories.Enums; | ||
using BotSharp.Abstraction.Repositories; | ||
using BotSharp.Abstraction.Settings; | ||
using BotSharp.Core.Repository; | ||
using BotSharp.Plugin.EmailReader.Hooks; | ||
using BotSharp.Plugin.EmailReader.Settings; | ||
using EntityFrameworkCore.BootKit; | ||
using Microsoft.Extensions.Configuration; | ||
using System; | ||
using System.Collections.Generic; | ||
using System.Linq; | ||
using System.Text; | ||
using System.Threading.Tasks; | ||
using BotSharp.Core.Infrastructures; | ||
using BotSharp.Plugin.EmailReader.Providers; | ||
|
||
namespace BotSharp.Plugin.EmailReader; | ||
|
||
public class EmailReaderPlugin : IBotSharpPlugin | ||
{ | ||
public string Id => "c88d27c8-127e-4aff-9cf4-74b49eec2926"; | ||
public string Name => "Email Reader"; | ||
public string Description => "Empower agent to read messages from email"; | ||
public string IconUrl => "https://cdn-icons-png.freepik.com/512/6711/6711567.png"; | ||
|
||
public void RegisterDI(IServiceCollection services, IConfiguration config) | ||
{ | ||
var emailReaderSettings = new EmailReaderSettings(); | ||
config.Bind("EmailReader", emailReaderSettings); | ||
services.AddScoped(provider => | ||
{ | ||
var settingService = provider.GetRequiredService<ISettingService>(); | ||
return settingService.Bind<EmailReaderSettings>("EmailReader"); | ||
}); | ||
services.AddSingleton(provider => emailReaderSettings); | ||
services.AddScoped<IAgentHook, EmailReaderHook>(); | ||
services.AddScoped<IAgentUtilityHook, EmailReaderUtilityHook>(); | ||
services.AddScoped<IEmailReader, DefaultEmailReader>(); | ||
} | ||
} |
12 changes: 12 additions & 0 deletions
12
src/Plugins/BotSharp.Plugin.EmailReader/Enums/UtilityName.cs
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,12 @@ | ||
using System; | ||
using System.Collections.Generic; | ||
using System.Linq; | ||
using System.Text; | ||
using System.Threading.Tasks; | ||
|
||
namespace BotSharp.Plugin.EmailReader.Enums; | ||
|
||
public class UtilityName | ||
{ | ||
public const string EmailReader = "email-reader"; | ||
} |
196 changes: 196 additions & 0 deletions
196
src/Plugins/BotSharp.Plugin.EmailReader/Functions/HandleEmailReaderFn.cs
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,196 @@ | ||
using BotSharp.Abstraction.Agents.Enums; | ||
using BotSharp.Abstraction.Files; | ||
using BotSharp.Abstraction.Messaging.Models.RichContent.Template; | ||
using BotSharp.Abstraction.MLTasks; | ||
using BotSharp.Core.Infrastructures; | ||
using BotSharp.Plugin.EmailReader.LlmContexts; | ||
using BotSharp.Plugin.EmailReader.Models; | ||
using BotSharp.Plugin.EmailReader.Providers; | ||
using BotSharp.Plugin.EmailReader.Settings; | ||
using BotSharp.Plugin.EmailReader.Templates; | ||
using BusinessCore.Utils; | ||
using MailKit; | ||
using MailKit.Net.Imap; | ||
using MailKit.Search; | ||
using MailKit.Security; | ||
using Microsoft.AspNetCore.Http; | ||
using Microsoft.Extensions.Logging; | ||
using MimeKit; | ||
|
||
namespace BotSharp.Plugin.EmailReader.Functions; | ||
|
||
public class HandleEmailReaderFn : IFunctionCallback | ||
{ | ||
public string Name => "handle_email_reader"; | ||
public readonly static string PROMPT_SUMMARY = "Provide a text summary of the following content."; | ||
public readonly static string RICH_CONTENT_SUMMARIZE = "Summarize the particular email by messageId"; | ||
public readonly static string RICH_CONTENT_READ_EMAIL = "Read the email by messageId"; | ||
public readonly static string RICH_CONTENT_MARK_READ = "Mark the email message as read by messageId"; | ||
public string Indication => "Handling email read"; | ||
private readonly IServiceProvider _services; | ||
private readonly ILogger<HandleEmailReaderFn> _logger; | ||
private readonly IHttpContextAccessor _context; | ||
private readonly BotSharpOptions _options; | ||
private readonly EmailReaderSettings _emailSettings; | ||
private readonly IConversationStateService _state; | ||
private readonly IEmailReader _emailProvider; | ||
|
||
public HandleEmailReaderFn(IServiceProvider services, | ||
ILogger<HandleEmailReaderFn> logger, | ||
IHttpContextAccessor context, | ||
BotSharpOptions options, | ||
EmailReaderSettings emailPluginSettings, | ||
IConversationStateService state, | ||
IEmailReader emailProvider) | ||
{ | ||
_services = services; | ||
_logger = logger; | ||
_context = context; | ||
_options = options; | ||
_emailSettings = emailPluginSettings; | ||
_state = state; | ||
_emailProvider = emailProvider; | ||
} | ||
public async Task<bool> Execute(RoleDialogModel message) | ||
{ | ||
var args = JsonSerializer.Deserialize<LlmContextIn>(message.FunctionArgs, _options.JsonSerializerOptions); | ||
var isMarkRead = args?.IsMarkRead ?? false; | ||
var isSummarize = args?.IsSummarize ?? false; | ||
var messageId = args?.MessageId; | ||
try | ||
{ | ||
if (!string.IsNullOrEmpty(messageId)) | ||
{ | ||
if (isMarkRead) | ||
{ | ||
await _emailProvider.MarkEmailAsReadById(messageId); | ||
message.Content = $"The email message has been marked as read."; | ||
return true; | ||
} | ||
var emailMessage = await _emailProvider.GetEmailById(messageId); | ||
if (isSummarize) | ||
{ | ||
var prompt = $"{PROMPT_SUMMARY} The content was sent by {emailMessage.From.ToString()}. Details: {emailMessage.TextBody}"; | ||
var agent = new Agent | ||
{ | ||
Id = BuiltInAgentId.UtilityAssistant, | ||
Name = "Utility Assistant", | ||
Instruction = prompt | ||
}; | ||
|
||
var llmProviderService = _services.GetRequiredService<ILlmProviderService>(); | ||
var provider = llmProviderService.GetProviders().FirstOrDefault(x => x == "openai"); | ||
var model = llmProviderService.GetProviderModel(provider: provider, id: "gpt-4"); | ||
var completion = CompletionProvider.GetChatCompletion(_services, provider: provider, model: model.Name); | ||
var convService = _services.GetService<IConversationService>(); | ||
var conversationId = convService.ConversationId; | ||
var dialogs = convService.GetDialogHistory(fromBreakpoint: false); | ||
var response = await completion.GetChatCompletions(agent, dialogs); | ||
var content = response?.Content ?? string.Empty; | ||
message.Content = content; | ||
message.RichContent = BuildRichContent.TextPostBackRichContent(_state.GetConversationId(), message.Content); | ||
return true; | ||
} | ||
UniqueId.TryParse(messageId, out UniqueId uid); | ||
message.RichContent = BuildRichContentForEmail(emailMessage, uid.ToString()); | ||
return true; | ||
} | ||
var emails = await _emailProvider.GetUnreadEmails(); | ||
message.Content = "Please choose which one to read for you."; | ||
message.RichContent = BuildRichContentForSubject(emails.OrderByDescending(x => x.CreateDate).ToList()); | ||
return true; | ||
} | ||
catch (Exception ex) | ||
{ | ||
var msg = $"Failed to read the emails. {ex.Message}"; | ||
_logger.LogError($"{msg}\n(Error: {ex.Message})"); | ||
message.Content = msg; | ||
return false; | ||
} | ||
} | ||
private RichContent<IRichMessage> BuildRichContentForSubject(List<EmailModel> emailSubjects) | ||
{ | ||
var text = "Please let me know which message I need to read?"; | ||
|
||
return new RichContent<IRichMessage> | ||
{ | ||
FillPostback = true, | ||
Editor = EditorTypeEnum.None, | ||
Recipient = new Recipient | ||
{ | ||
Id = _state.GetConversationId() | ||
}, | ||
Message = new GenericTemplateMessage<EmailSubjectElement> | ||
{ | ||
Text = text, | ||
Elements = GetElements(emailSubjects) | ||
} | ||
}; | ||
} | ||
private RichContent<IRichMessage> BuildRichContentForEmail(EmailModel email, string uid) | ||
{ | ||
var text = "The email details are given below. \n"; | ||
|
||
return new RichContent<IRichMessage> | ||
{ | ||
FillPostback = true, | ||
Editor = EditorTypeEnum.None, | ||
Recipient = new Recipient | ||
{ | ||
Id = _state.GetConversationId() | ||
}, | ||
Message = new GenericTemplateMessage<EmailSubjectElement> | ||
{ | ||
Text = $"{text}<b>From</b>: {email.From.ToString()}\n<b>Subject</b>: {email.Subject}\n{email.Body}", | ||
Elements = GetElements(uid) | ||
} | ||
}; | ||
} | ||
private static List<EmailSubjectElement> GetElements(string uid) | ||
{ | ||
var element = new EmailSubjectElement() | ||
{ | ||
Buttons = new ElementButton[] | ||
{ | ||
BuildMarkReadElementButton(uid) | ||
} | ||
}; | ||
return new List<EmailSubjectElement>() { element }; | ||
} | ||
private static List<EmailSubjectElement> GetElements(List<EmailModel> emails) | ||
{ | ||
var elements = emails.Select(e => new EmailSubjectElement | ||
{ | ||
Title = $"Subject: {e.Subject}", | ||
Subtitle = $"From: {e.From}<br/> Date: {e.CreateDate}", | ||
Buttons = BuildElementButton(e) | ||
}).ToList(); | ||
return elements; | ||
} | ||
private static ElementButton[] BuildElementButton(EmailModel email) | ||
{ | ||
var elements = new List<ElementButton>() { }; | ||
elements.Add(new ElementButton | ||
{ | ||
Title = "Read", | ||
Payload = $"{RICH_CONTENT_READ_EMAIL}: {email.UId}.", | ||
Type = "text", | ||
}); | ||
elements.Add(new ElementButton | ||
{ | ||
Title = "Summarize", | ||
Payload = $"{RICH_CONTENT_SUMMARIZE}: {email.UId}.", | ||
Type = "text", | ||
}); | ||
return elements.ToArray(); | ||
} | ||
private static ElementButton BuildMarkReadElementButton(string uId) | ||
{ | ||
return new ElementButton | ||
{ | ||
Title = "Mark as read", | ||
Payload = $"{RICH_CONTENT_MARK_READ}: {uId}.", | ||
Type = "text", | ||
}; | ||
} | ||
} |
63 changes: 63 additions & 0 deletions
63
src/Plugins/BotSharp.Plugin.EmailReader/Hooks/EmailReaderHook.cs
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,63 @@ | ||
using BotSharp.Abstraction.Agents; | ||
using BotSharp.Abstraction.Agents.Enums; | ||
using BotSharp.Abstraction.Agents.Settings; | ||
using BotSharp.Abstraction.Functions.Models; | ||
using BotSharp.Abstraction.Repositories; | ||
using BotSharp.Plugin.EmailReader.Enums; | ||
using System; | ||
using System.Collections.Generic; | ||
using System.Linq; | ||
using System.Text; | ||
using System.Threading.Tasks; | ||
|
||
namespace BotSharp.Plugin.EmailReader.Hooks; | ||
|
||
public class EmailReaderHook : AgentHookBase | ||
{ | ||
private static string FUNCTION_NAME = "handle_email_reader"; | ||
|
||
public override string SelfId => string.Empty; | ||
|
||
public EmailReaderHook(IServiceProvider services, AgentSettings settings) | ||
: base(services, settings) | ||
{ | ||
} | ||
public override void OnAgentLoaded(Agent agent) | ||
{ | ||
var conv = _services.GetRequiredService<IConversationService>(); | ||
var isConvMode = conv.IsConversationMode(); | ||
var isEnabled = !agent.Utilities.IsNullOrEmpty() && agent.Utilities.Contains(UtilityName.EmailReader); | ||
|
||
if (isConvMode && isEnabled) | ||
{ | ||
var (prompt, fn) = GetPromptAndFunction(); | ||
if (fn != null) | ||
{ | ||
if (!string.IsNullOrWhiteSpace(prompt)) | ||
{ | ||
agent.Instruction += $"\r\n\r\n{prompt}\r\n\r\n"; | ||
} | ||
|
||
if (agent.Functions == null) | ||
{ | ||
agent.Functions = new List<FunctionDef> { fn }; | ||
} | ||
else | ||
{ | ||
agent.Functions.Add(fn); | ||
} | ||
} | ||
} | ||
|
||
base.OnAgentLoaded(agent); | ||
} | ||
|
||
private (string, FunctionDef?) GetPromptAndFunction() | ||
{ | ||
var db = _services.GetRequiredService<IBotSharpRepository>(); | ||
var agent = db.GetAgent(BuiltInAgentId.UtilityAssistant); | ||
var prompt = agent?.Templates?.FirstOrDefault(x => x.Name.IsEqualTo($"{FUNCTION_NAME}.fn"))?.Content ?? string.Empty; | ||
var loadAttachmentFn = agent?.Functions?.FirstOrDefault(x => x.Name.IsEqualTo(FUNCTION_NAME)); | ||
return (prompt, loadAttachmentFn); | ||
} | ||
} |
17 changes: 17 additions & 0 deletions
17
src/Plugins/BotSharp.Plugin.EmailReader/Hooks/EmailReaderUtilityHook.cs
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,17 @@ | ||
using BotSharp.Abstraction.Agents; | ||
using BotSharp.Plugin.EmailReader.Enums; | ||
using System; | ||
using System.Collections.Generic; | ||
using System.Linq; | ||
using System.Text; | ||
using System.Threading.Tasks; | ||
|
||
namespace BotSharp.Plugin.EmailReader.Hooks; | ||
|
||
public class EmailReaderUtilityHook : IAgentUtilityHook | ||
{ | ||
public void AddUtilities(List<string> utilities) | ||
{ | ||
utilities.Add(UtilityName.EmailReader); | ||
} | ||
} |
18 changes: 18 additions & 0 deletions
18
src/Plugins/BotSharp.Plugin.EmailReader/LlmContexts/LlmContextIn.cs
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,18 @@ | ||
using System; | ||
using System.Collections.Generic; | ||
using System.Linq; | ||
using System.Text; | ||
using System.Text.Json.Serialization; | ||
using System.Threading.Tasks; | ||
|
||
namespace BotSharp.Plugin.EmailReader.LlmContexts; | ||
|
||
public class LlmContextIn | ||
{ | ||
[JsonPropertyName("mark_as_read")] | ||
public bool? IsMarkRead { get; set; } | ||
[JsonPropertyName("message_id")] | ||
public string? MessageId { get; set; } | ||
[JsonPropertyName("is_email_summarize")] | ||
public bool? IsSummarize { get; set; } | ||
} |
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Is it possible to place
EmailReader
in theEmailHandler
plugin?