Skip to content

Commit aae5528

Browse files
author
Piotr Stachaczynski
committed
feat: cleanup
1 parent e90edf4 commit aae5528

2 files changed

Lines changed: 10 additions & 3 deletions

File tree

src/MaIN.Services/Services/Steps/Commands/FetchCommandHandler.cs

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,8 @@ public class FetchCommandHandler(
2020
{
2121
var properties = new Dictionary<string, string>
2222
{
23-
{ "agent_internal", "true" }
23+
{ "agent_internal", "true" },
24+
{ Message.UnprocessedMessageProperty, string.Empty }
2425
};
2526

2627
Message? response;
@@ -134,7 +135,11 @@ private async Task<Message> ProcessJsonResponse(Message response, FetchCommand c
134135

135136
result!.Message.Role = command.ResponseType == FetchResponseType.AS_System ? "System" : "Assistant";
136137
var newMessage = result!.Message;
137-
newMessage.Properties = new() { { "agent_internal", "true" } };
138+
newMessage.Properties = new()
139+
{
140+
{ "agent_internal", "true" },
141+
{ Message.UnprocessedMessageProperty, string.Empty}
142+
};
138143
return newMessage;
139144
}
140145

src/MaIN.Services/Services/Steps/Commands/RedirectCommandHandler.cs

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
using MaIN.Domain.Configuration;
22
using MaIN.Domain.Entities;
3+
using MaIN.Services.Constants;
34
using MaIN.Services.Services.Abstract;
45
using MaIN.Services.Services.Models.Commands;
56

@@ -35,7 +36,8 @@ public class RedirectCommandHandler(IAgentService agentService) : ICommandHandle
3536
Type = chat.Backend != BackendType.Self ? MessageType.CloudLLM : MessageType.LocalLLM,
3637
Properties = new Dictionary<string, string>()
3738
{
38-
{ "agent_internal", "true" }
39+
{ "agent_internal", "true" },
40+
{ Message.UnprocessedMessageProperty, string.Empty}
3941
}
4042
};
4143
}

0 commit comments

Comments
 (0)