SemianiakaVY/BotSharp
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

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

Browse source 
refine llm completion log
This commit is contained in:
iceljc 2025-02-20 19:27:56 -06:00 committed by GitHub
parent 29a3565a52 6e9237ca4b
commit 8a18780f83
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 12 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/Infrastructure/BotSharp.Abstraction/Loggers/Models/LlmCompletionLog.cs
View file

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

2
src/Infrastructure/BotSharp.Logger/Hooks/CommonContentGeneratingHook.cs
View file

@ -30,7 +30,7 @@ public class CommonContentGeneratingHook : IContentGeneratingHook
AgentId = message.CurrentAgentId, AgentId = message.CurrentAgentId,
Prompt = tokenStats.Prompt, Prompt = tokenStats.Prompt,
Response = message.Content, Response = message.Content,
CreateDateTime = DateTime.UtcNow CreatedTime = DateTime.UtcNow
}; };
db.SaveLlmCompletionLog(completionLog); db.SaveLlmCompletionLog(completionLog);

6
src/Plugins/BotSharp.Plugin.MongoStorage/Collections/LlmCompletionLogDocument.cs
View file

@ -3,5 +3,9 @@ namespace BotSharp.Plugin.MongoStorage.Collections;
public class LlmCompletionLogDocument : MongoBase public class LlmCompletionLogDocument : MongoBase
{ {
public string ConversationId { get; set; } = default!; 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; }
} }

11
src/Plugins/BotSharp.Plugin.MongoStorage/Models/PromptLogMongoElement.cs
View file

@ -1,11 +0,0 @@
namespace BotSharp.Plugin.MongoStorage.Models;
[BsonIgnoreExtraElements(Inherited = true)]
public class PromptLogMongoElement
{
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 CreateDateTime { get; set; }
}

14
src/Plugins/BotSharp.Plugin.MongoStorage/Repository/MongoRepository.Log.cs
View file

@ -12,21 +12,17 @@ public partial class MongoRepository
var conversationId = log.ConversationId.IfNullOrEmptyAs(Guid.NewGuid().ToString()); var conversationId = log.ConversationId.IfNullOrEmptyAs(Guid.NewGuid().ToString());
var messageId = log.MessageId.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, MessageId = messageId,
AgentId = log.AgentId, AgentId = log.AgentId,
Prompt = log.Prompt, Prompt = log.Prompt,
Response = log.Response, Response = log.Response,
CreateDateTime = log.CreateDateTime CreatedTime = log.CreatedTime
}; };
_dc.LlmCompletionLogs.InsertOne(data);
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);
} }
#endregion #endregion