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

Refactor file attachment handling to skip forbidden file types in DocumentAnalysisAssistant, ContentText, and ChatComponent

Browse Source
This commit is contained in:
Thorsten Sommer 2025-12-28 16:22:18 +01:00
parent cca5c4d3a4
commit ca76da65f2
Signed by: tsommer
GPG Key ID: 371BBA77A02C0108
3 changed files with 18 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
app/MindWork AI Studio/Assistants/DocumentAnalysis/DocumentAnalysisAssistant.razor.cs
View File

@ -332,6 +332,12 @@ public partial class DocumentAnalysisAssistant : AssistantBaseCore<SettingsDialo
foreach (var fileAttachment in this.loadedDocumentPaths) foreach (var fileAttachment in this.loadedDocumentPaths)
{ {
if (fileAttachment.IsForbidden)
{
this.Logger.LogWarning($"Skipping forbidden file: '{fileAttachment.FilePath}'.");
continue;
}
var fileContent = await this.RustService.ReadArbitraryFileData(fileAttachment.FilePath, int.MaxValue); var fileContent = await this.RustService.ReadArbitraryFileData(fileAttachment.FilePath, int.MaxValue);
documentSections.Add($""" documentSections.Add($"""

13
app/MindWork AI Studio/Chat/ContentText.cs
View File

@ -165,8 +165,11 @@ public sealed class ContentText : IContent
foreach (var missingFile in missingFiles) foreach (var missingFile in missingFiles)
LOGGER.LogWarning("File attachment no longer exists and will be skipped: '{MissingFile}'.", missingFile.FilePath); LOGGER.LogWarning("File attachment no longer exists and will be skipped: '{MissingFile}'.", missingFile.FilePath);
// Only proceed if there are existing files // Determine allowed attachments:
if (existingFiles.Count > 0) var numberAllowedExistingFiles = existingFiles.Count(x => x.IsValid);
// Only proceed if there are existing, allowed files:
if (numberAllowedExistingFiles > 0)
{ {
// Check Pandoc availability once before processing file attachments // Check Pandoc availability once before processing file attachments
var pandocState = await Pandoc.CheckAvailabilityAsync(Program.RUST_SERVICE, showMessages: true, showSuccessMessage: false); var pandocState = await Pandoc.CheckAvailabilityAsync(Program.RUST_SERVICE, showMessages: true, showSuccessMessage: false);
@ -181,6 +184,12 @@ public sealed class ContentText : IContent
sb.AppendLine("The following files are attached to this message:"); sb.AppendLine("The following files are attached to this message:");
foreach(var file in existingFiles) foreach(var file in existingFiles)
{ {
if (file.IsForbidden)
{
LOGGER.LogWarning("File attachment '{FilePath}' has a forbidden file type and will be skipped.", file.FilePath);
continue;
}
sb.AppendLine(); sb.AppendLine();
sb.AppendLine("---------------------------------------"); sb.AppendLine("---------------------------------------");
sb.AppendLine($"File path: {file.FilePath}"); sb.AppendLine($"File path: {file.FilePath}");

2
app/MindWork AI Studio/Components/ChatComponent.razor.cs
View File

@ -464,7 +464,7 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
lastUserPrompt = new ContentText lastUserPrompt = new ContentText
{ {
Text = this.userInput, Text = this.userInput,
FileAttachments = [..this.chatDocumentPaths], FileAttachments = [..this.chatDocumentPaths.Where(x => x.IsValid)],
}; };
// //