Mark DocumentClasses as internal

Author: srebrek

Frontend/MainFE/Components/Models/AgentConfigDto.cs

@@ -2,7 +2,7 @@
 
 namespace MaIN.Models.Rag;
 
-public class AgentContextDto
+public class AgentConfigDto
 {
     [JsonPropertyName("instruction")]
     public string Instruction { get; set; }
@@ -13,4 +13,4 @@ public class AgentContextDto
     [JsonPropertyName("relations")]
     public List<string>? Relations { get; set; }
 
-}
+}

Frontend/MainFE/Components/Models/AgentDto.cs

@@ -22,7 +22,7 @@ public class AgentDto
     [JsonPropertyName("flow")]
     public bool Flow { get; set; }
     [JsonPropertyName("context")]
-    public AgentContextDto Context { get; set; }
+    public AgentConfigDto Context { get; set; }
     public AgentProcessingState State { get; set; }
     public bool IsProcessing { get; set; }
     public List<string>? AgentDependencies { get; set; } = [];
@@ -36,4 +36,4 @@ public string ProgressMessage
         set => _progressMessage = value;
     }
     private string? _progressMessage;
-}
+}

src/MaIN.Infrastructure/Mappers/AgentDocumentMapper.cs

@@ -35,6 +35,7 @@ internal static class AgentDocumentMapper
         Flow = agent.Flow,
         ToolsConfiguration = agent.ToolsConfiguration,
         Backend = agent.Backend,
+        ChatId = agent.ChatId,
         Description = agent.Description,
         Behaviours = agent.Behaviours,
         CurrentBehaviour = agent.CurrentBehaviour,

src/MaIN.Infrastructure/Models/AgentConfigDocument.cs

@@ -2,11 +2,11 @@
 
 namespace MaIN.Infrastructure.Models;
 
-public class AgentConfigDocument
+internal class AgentConfigDocument
 {
     public string? Instruction { get; init; }
     public AgentSourceDocument? Source { get; init; }
     public List<string>? Steps { get; init; }
     public List<string>? Relations { get; init; }
-    public Mcp? McpConfig { get; set; }
+    public Mcp? McpConfig { get; init; }
 }

src/MaIN.Infrastructure/Models/AgentDocument.cs

@@ -4,7 +4,7 @@
 
 namespace MaIN.Infrastructure.Models;
 
-public class AgentDocument
+internal class AgentDocument
 {
     [BsonId]
     public required string Id { get; init; }

src/MaIN.Infrastructure/Models/AgentFlowDocument.cs

@@ -2,11 +2,11 @@
 
 namespace MaIN.Infrastructure.Models;
 
-public class AgentFlowDocument
+internal class AgentFlowDocument
 {
     [BsonId]
     public required string Id { get; init; }
     public required string Name { get; init; }
     public required List<AgentDocument> Agents { get; init; }
     public required string Description { get; init; }
-}
+}

src/MaIN.Infrastructure/Models/AgentSourceDocument.cs

@@ -1,8 +1,8 @@
 namespace MaIN.Infrastructure.Models;
 
-public class AgentSourceDocument
+internal class AgentSourceDocument
 {
     public string? DetailsSerialized { get; init; }
     public AgentSourceTypeDocument Type { get; init; }
     public string? AdditionalMessage { get; init; }
-}
+}

src/MaIN.Infrastructure/Models/ChatDocument.cs

@@ -3,7 +3,7 @@
 
 namespace MaIN.Infrastructure.Models;
 
-public class ChatDocument
+internal class ChatDocument
 {
     [BsonId]
     public required string Id { get; init; }

src/MaIN.Infrastructure/Models/InterferenceParamsDocument.cs

@@ -2,20 +2,20 @@
 
 namespace MaIN.Infrastructure.Models;
 
-public class InferenceParamsDocument
+internal class InferenceParamsDocument
 {
     public float Temperature { get; set; }
     public int ContextSize { get; set; }
-    public int GpuLayerCount { get; init; } 
-    public uint SeqMax { get; init; } 
+    public int GpuLayerCount { get; init; }
+    public uint SeqMax { get; init; }
     public uint BatchSize { get; init; }
-    public uint UBatchSize { get; init; } 
-    public bool Embeddings { get; init; } 
-    public int TypeK { get; init; } 
-    public int TypeV { get; init; } 
+    public uint UBatchSize { get; init; }
+    public bool Embeddings { get; init; }
+    public int TypeK { get; init; }
+    public int TypeV { get; init; }
     public int TokensKeep { get; set; }
     public int MaxTokens { get; set; }
     public int TopK { get; init; }
     public float TopP { get; init; }
     public Grammar? Grammar { get; set; }
-}
+}

src/MaIN.Infrastructure/Models/LLMTokenValueDocument.cs

@@ -2,8 +2,8 @@
 
 namespace MaIN.Infrastructure.Models;
 
-public class LLMTokenValueDocument
+internal class LLMTokenValueDocument
 {
     public required string Text { get; set; }
     public TokenType Type { get; set; } //TODO add document representation of this domain enum
-}
+}