Make use of the ChatId property of the Agent

Author: srebrek

MaIN.Core.IntegrationTests/ChatTests.cs

@@ -47,12 +47,11 @@ public async Task Should_AnswerDifferences_BetweenDocuments_ChatWithFiles()
     public async Task Should_AnswerQuestion_FromExistingChat()
     {
         var result = AIHub.Chat()
-            .WithModel(Models.Local.Qwen2_5_0_5b);
+            .WithModel(Models.Local.Gemma2_2b);
 
         await result.WithMessage("What do you think about math theories?")
             .CompleteAsync();
 
-
         await result.WithMessage("And about physics?")
             .CompleteAsync();
 

src/MaIN.Domain/Entities/Agents/Agent.cs

@@ -12,10 +12,10 @@ public class Agent
     public bool Started { get; set; }
     public bool Flow { get; set; }
     public required AgentData Context { get; init; }
-    public string ChatId => string.Empty;
+    public string ChatId { get; set; } = string.Empty;
     public int Order { get; set; }
     public BackendType? Backend { get; set; }
     public Dictionary<string, string> Behaviours { get; set; } = [];
     public required string CurrentBehaviour { get; set; }
     public ToolsConfiguration? ToolsConfiguration { get; set; }
-}
+}

src/MaIN.Services/Services/AgentService.cs

@@ -76,7 +76,10 @@ await notificationService.DispatchNotification(
             await agentRepository.UpdateAgent(agent.Id, agent);
 
             await notificationService.DispatchNotification(
-                NotificationMessageBuilder.ProcessingComplete(agentId, agent.CurrentBehaviour, "COMPLETED"), "ReceiveAgentUpdate");
+                NotificationMessageBuilder.ProcessingComplete(
+                    agentId,
+                    agent.CurrentBehaviour,
+                    "COMPLETED"), "ReceiveAgentUpdate");
 
             //normalize message before returning it to user
             chat.Messages.Last().Content = Replace(
@@ -90,7 +93,8 @@ await notificationService.DispatchNotification(
         catch (Exception ex)
         {
             await notificationService.DispatchNotification(
-                NotificationMessageBuilder.ProcessingFailed(agentId, agent.CurrentBehaviour, ex.Message), "ReceiveAgentUpdate");
+                NotificationMessageBuilder.ProcessingFailed(agentId, agent.CurrentBehaviour, ex.Message),
+                "ReceiveAgentUpdate");
             throw;
         }
     }
@@ -128,15 +132,13 @@ public async Task<Agent> CreateAgent(Agent agent, bool flow = false, bool intera
 
         agent.Started = true;
         agent.Flow = flow;
+        agent.ChatId = chat.Id;
         agent.Behaviours ??= [];
         agent.Behaviours.Add("Default", agent.Context.Instruction!);
         agent.CurrentBehaviour = "Default";
 
-        var agentDocument = agent.ToDocument();
-        agentDocument.ChatId = chat.Id;
-
         await chatRepository.AddChat(chat.ToDocument());
-        await agentRepository.AddAgent(agentDocument);
+        await agentRepository.AddAgent(agent.ToDocument());
 
         return agent;
     }