cd: infer done cli refactor in progress

Author: Piotr Stachaczynski

Examples/Examples/Utils/ImagePreviewer.cs

@@ -9,7 +9,7 @@ namespace Examples.Utils;
 
 public static class ImagePreview
 {
-    public static void ShowImage(byte[] imageData, string extension = "png")
+    public static void ShowImage(byte[]? imageData, string extension = "png")
     {
         // Validate extension
         if (string.IsNullOrWhiteSpace(extension) || extension.Contains("."))

src/MaIN.Core.UnitTests/AgentContextTests.cs

@@ -125,7 +125,7 @@ public void WithBehaviour_ShouldAddBehaviourAndSetCurrent()
     public async Task CreateAsync_ShouldCallAgentServiceCreateAgent()
     {
         // Arrange
-        var agent = new Agent();
+        var agent = new Agent() {Id = Guid.NewGuid().ToString()};
         _mockAgentService
             .Setup(s => s.CreateAgent(
                 It.IsAny<Agent>(), 

src/MaIN.Core/Hub/Contexts/AgentContext.cs

@@ -40,7 +40,7 @@ internal AgentContext(IAgentService agentService, Agent existingAgent)
         _agent = existingAgent;
     }
 
-    public AgentContext WithId(string id)
+    public AgentContext WithId(string? id)
     {
         _agent.Id = id;
         return this;
@@ -97,13 +97,13 @@ public AgentContext WithInitialPrompt(string prompt)
         return this;
     }
 
-    public AgentContext WithSteps(List<string> steps)
+    public AgentContext WithSteps(List<string>? steps)
     {
         _agent.Context.Steps = steps;
         return this;
     }
 
-    public AgentContext WithBehaviour(string name, string instruction)
+    public AgentContext WithBehaviour(string? name, string instruction)
     {
         _agent.Behaviours ??= new Dictionary<string, string>();
         _agent.Behaviours[name] = instruction;

src/MaIN.Core/Hub/Contexts/ChatContext.cs

@@ -112,7 +112,7 @@ public ChatContext EnableVisual()
     public async Task<ChatResult> CompleteAsync(
         bool translate = false,
         bool interactive = false,
-        Func<string, Task>? changeOfValue = null)
+        Func<string?, Task>? changeOfValue = null)
     {
         if (_chat.Id == null || !await ChatExists(_chat.Id))
         {

src/MaIN.Core/Hub/Utils/StepBuilder.cs

@@ -2,7 +2,7 @@ namespace MaIN.Core.Hub.Utils;
 
 public class StepBuilder
 {
-    public List<string> Steps = new();
+    public List<string>? Steps = new();
     public static StepBuilder Instance => new();
 
     public StepBuilder Answer()
@@ -35,7 +35,7 @@ public StepBuilder Redirect(string agentId, string output = "AS_Output", string
         return this;
     }
 
-    public List<string> Build()
+    public List<string>? Build()
     {
         return Steps;
     }

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

@@ -2,13 +2,13 @@ namespace MaIN.Domain.Entities.Agents;
 
 public class Agent
 {
-    public string Id { get; set; }
-    public string Name { get; set; }
-    public string Model { get; set; }
-    public string? Description { get; set; }
+    public required string Id { get; set; }
+    public string Name { get; set; } = null!;
+    public string Model { get; set; } = null!;
+    public string? Description { get; init; }
     public bool Started { get; set; }
     public bool Flow { get; set; }
-    public AgentData Context { get; set; }
+    public AgentData? Context { get; set; }
     public string? ChatId { get; set; }
     public int Order { get; set; }
     public Dictionary<string, string>? Behaviours { get; set; }

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

@@ -4,7 +4,7 @@ public class AgentData
 {
     public string Instruction { get; set; }
     public AgentSource.AgentSource? Source { get; set; }
-    public List<string> Steps { get; set; }
+    public List<string>? Steps { get; set; }
     public List<string>? Relations { get; set; }
 
 }

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

@@ -4,5 +4,5 @@ namespace MaIN.Domain.Entities.Agents.Commands;
 
 public class StartCommand : BaseCommand
 {
-    public string InitialPrompt { get; set; }
+    public string? InitialPrompt { get; init; }
 }

src/MaIN.Domain/Entities/Agents/Commands/StartCommand.cs

@@ -2,15 +2,15 @@ namespace MaIN.Domain.Entities;
 
 public class Chat
 {
-    public string Id { get; set; }
-    public string Name { get; set; }
-    public string Model { get; set; }
+    public string Id { get; init; } = null!;
+    public string Name { get; init; } = null!;
+    public string Model { get; set; } = null!;
     public List<Message> Messages { get; set; } = [];
     public ChatType Type { get; set; } = ChatType.Conversation;
     public bool Visual { get; set; } = false;
     public InferenceParams InterferenceParams { get; set; } = new();
-    public Dictionary<string, string> Properties { get; set; } = [];
-    public List<string> Memory { get; set; } = [];
+    public Dictionary<string, string> Properties { get; init; } = [];
+    public List<string> Memory { get; } = [];
 
     public bool Interactive = false;
     public bool Translate = false;