Merge pull request #895 from iceljc/features/refine-llm-completion-log

refine llm completion log

Author: iceljc

src/Infrastructure/BotSharp.Abstraction/Loggers/Models/LlmCompletionLog.cs

@@ -7,5 +7,5 @@ public class LlmCompletionLog
     public string AgentId { get; set; } = string.Empty;
     public string Prompt { get; set; } = string.Empty;
     public string? Response { get; set; }
-    public DateTime CreateDateTime { get; set; } = DateTime.UtcNow;
+    public DateTime CreatedTime { get; set; } = DateTime.UtcNow;
 }

src/Infrastructure/BotSharp.Logger/Hooks/CommonContentGeneratingHook.cs

@@ -30,7 +30,7 @@ private void SaveLlmCompletionLog(RoleDialogModel message, TokenStatsModel token
             AgentId = message.CurrentAgentId,
             Prompt = tokenStats.Prompt,
             Response = message.Content,
-            CreateDateTime = DateTime.UtcNow
+            CreatedTime = DateTime.UtcNow
         };
 
         db.SaveLlmCompletionLog(completionLog);

src/Plugins/BotSharp.Plugin.MongoStorage/Collections/LlmCompletionLogDocument.cs

@@ -3,5 +3,9 @@ namespace BotSharp.Plugin.MongoStorage.Collections;
 public class LlmCompletionLogDocument : MongoBase
 {
     public string ConversationId { get; set; } = default!;
-    public List<PromptLogMongoElement> Logs { get; set; } = [];
+    public string MessageId { get; set; } = default!;
+    public string AgentId { get; set; } = default!;
+    public string Prompt { get; set; } = default!;
+    public string? Response { get; set; }
+    public DateTime CreatedTime { get; set; }
 }

src/Plugins/BotSharp.Plugin.MongoStorage/Models/PromptLogMongoElement.cs

@@ -12,21 +12,17 @@ public void SaveLlmCompletionLog(LlmCompletionLog log)
         var conversationId = log.ConversationId.IfNullOrEmptyAs(Guid.NewGuid().ToString());
         var messageId = log.MessageId.IfNullOrEmptyAs(Guid.NewGuid().ToString());
 
-        var logElement = new PromptLogMongoElement
+        var data = new LlmCompletionLogDocument
         {
+            Id = Guid.NewGuid().ToString(),
+            ConversationId = conversationId,
             MessageId = messageId,
             AgentId = log.AgentId,
             Prompt = log.Prompt,
             Response = log.Response,
-            CreateDateTime = log.CreateDateTime
+            CreatedTime = log.CreatedTime
         };
-
-        var filter = Builders<LlmCompletionLogDocument>.Filter.Eq(x => x.ConversationId, conversationId);
-        var update = Builders<LlmCompletionLogDocument>.Update
-                                                       .SetOnInsert(x => x.Id, Guid.NewGuid().ToString())
-                                                       .Push(x => x.Logs, logElement);
-
-        _dc.LlmCompletionLogs.UpdateOne(filter, update, _options);
+        _dc.LlmCompletionLogs.InsertOne(data);
     }
 
     #endregion