SemianiakaVY/BotSharp
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

fix file save

Browse source
This commit is contained in:
Jicheng Lu 2024-06-19 11:58:27 -05:00
parent f12550a440
commit eb679701bd
2 changed files with 10 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/Infrastructure/BotSharp.Core/Conversations/Services/ConversationService.SendMessage.cs
View file

@ -16,6 +16,11 @@ public partial class ConversationService
{
var conversation = await GetConversationRecord(agentId);
// Save message files
var fileService = _services.GetRequiredService<IBotSharpFileService>();
fileService.SaveMessageFiles(_conversationId, message.MessageId, FileSourceType.User, message.Files);
message.Files?.Clear();
var agentService = _services.GetRequiredService<IAgentService>();
Agent agent = await agentService.LoadAgent(agentId);
@ -46,11 +51,6 @@ public partial class ConversationService
routing.Context.SetMessageId(_conversationId, message.MessageId);
routing.Context.Push(agent.Id);
// Save message files
var fileService = _services.GetRequiredService<IBotSharpFileService>();
fileService.SaveMessageFiles(_conversationId, message.MessageId, FileSourceType.User, message.Files);
message.Files?.Clear();
// Save payload
if (replyMessage != null && !string.IsNullOrEmpty(replyMessage.Payload))
{

7
src/Infrastructure/BotSharp.Core/Files/BotSharpFileService.Conversation.cs
View file

@ -1,3 +1,4 @@
using Amazon.Runtime.Internal.Auth;
using BotSharp.Abstraction.Browsing;
using BotSharp.Abstraction.Browsing.Models;
using Microsoft.EntityFrameworkCore;
@ -221,14 +222,16 @@ public partial class BotSharpFileService
}
var (_, bytes) = GetFileInfoFromData(file.FileData);
Thread.Sleep(100);
var subDir = Path.Combine(dir, source, $"{i + 1}");
if (!ExistDirectory(subDir))
{
Directory.CreateDirectory(subDir);
}
File.WriteAllBytes(Path.Combine(subDir, file.FileName), bytes);
using var fs = new FileStream(Path.Combine(subDir, file.FileName), FileMode.OpenOrCreate);
fs.Write(bytes, 0, bytes.Length);
fs.Flush();
Thread.Sleep(100);
}
return true;