MindWorkAI/AI-Studio
MindWorkAI/AI-Studio
1
0
Fork 0
mirror of https://github.com/MindWorkAI/AI-Studio.git synced 2026-02-12 20:01:37 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Refactor BuildMessagesAsync to use explicit task creation for improved readability and performance

Browse Source
This commit is contained in:
Thorsten Sommer 2025-12-29 12:09:37 +01:00
parent 6abe0374ea
commit ddb1677bc5
Signed by: tsommer
GPG Key ID: 371BBA77A02C0108
1 changed files with 24 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
app/MindWork AI Studio/Chat/ListContentBlockExtensions.cs
View File

@ -13,24 +13,32 @@ public static class ListContentBlockExtensions
/// <returns>An asynchronous task that resolves to a list of transformed results.</returns>
public static async Task<IList<IMessageBase>> BuildMessagesAsync(this List<ContentBlock> blocks, Func<ChatRole, string> roleTransformer)
{
var messages = blocks
.Where(n => n.ContentType is ContentType.TEXT && !string.IsNullOrWhiteSpace((n.Content as ContentText)?.Text))
.Select(async n => new TextMessage
{
Role = roleTransformer(n.Role),
Content = n.Content switch
{
ContentText text => await text.PrepareTextContentForAI(),
_ => string.Empty,
}
})
.ToList();
// Await all messages:
await Task.WhenAll(messages);
var messageTaskList = new List<Task<IMessageBase>>(blocks.Count);
foreach (var block in blocks)
{
switch (block.Content)
{
case ContentText text when block.ContentType is ContentType.TEXT && !string.IsNullOrWhiteSpace(text.Text):
messageTaskList.Add(CreateTextMessageAsync(block, text));
break;
}
}
// Await all messages:
await Task.WhenAll(messageTaskList);
// Select all results:
return messages.Select(n => n.Result).Cast<IMessageBase>().ToList();
return messageTaskList.Select(n => n.Result).ToList();
// Local function to create a text message asynchronously.
Task<IMessageBase> CreateTextMessageAsync(ContentBlock block, ContentText text)
{
return Task.Run(async () => new TextMessage
{
Role = roleTransformer(block.Role),
Content = await text.PrepareTextContentForAI(),
} as IMessageBase);
}
}
/// <summary>