Skip to content

Commit 7316c49

Browse files
authored
Merge pull request #239 from iceljc/bugfix/fix-file-llm-prompt-log
fix file llm prompt log
2 parents 759e116 + 694a3c2 commit 7316c49

File tree

2 files changed

+9
-3
lines changed

2 files changed

+9
-3
lines changed

src/Infrastructure/BotSharp.Core/Repository/FileRepository.cs

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -860,8 +860,10 @@ public List<string> GetExecutionLogs(string conversationId)
860860
#region LLM Completion Log
861861
public void SaveLlmCompletionLog(LlmCompletionLog log)
862862
{
863+
if (log == null || string.IsNullOrEmpty(log.ConversationId)) return;
864+
863865
var convDir = FindConversationDirectory(log.ConversationId);
864-
if (!Directory.Exists(convDir)) return;
866+
if (string.IsNullOrEmpty(convDir)) return;
865867

866868
var logDir = Path.Combine(convDir, "llm_prompt_log");
867869
if (!Directory.Exists(logDir))
@@ -870,7 +872,7 @@ public void SaveLlmCompletionLog(LlmCompletionLog log)
870872
}
871873

872874
log.Id = Guid.NewGuid().ToString();
873-
var index = GetLlmCompletionLogIndex(logDir, log.MessageId);
875+
var index = GetNextLlmCompletionLogIndex(logDir, log.MessageId);
874876
var file = Path.Combine(logDir, $"{log.MessageId}.{index}.log");
875877
File.WriteAllText(file, JsonSerializer.Serialize(log, _options));
876878
}
@@ -969,6 +971,8 @@ private List<AgentResponse> FetchResponses(string fileDir)
969971

970972
private string? FindConversationDirectory(string conversationId)
971973
{
974+
if (string.IsNullOrEmpty(conversationId)) return null;
975+
972976
var dir = Path.Combine(_dbSettings.FileRepository, _conversationSettings.DataDir, conversationId);
973977
if (!Directory.Exists(dir)) return null;
974978

@@ -1024,7 +1028,7 @@ private List<StateKeyValue> CollectConversationStates(string stateDir)
10241028
return states;
10251029
}
10261030

1027-
private int GetLlmCompletionLogIndex(string logDir, string id)
1031+
private int GetNextLlmCompletionLogIndex(string logDir, string id)
10281032
{
10291033
var files = Directory.GetFiles(logDir);
10301034
if (files.IsNullOrEmpty())

src/Plugins/BotSharp.Plugin.MongoStorage/Repository/MongoRepository.cs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -892,6 +892,8 @@ public List<string> GetExecutionLogs(string conversationId)
892892
#region LLM Completion Log
893893
public void SaveLlmCompletionLog(LlmCompletionLog log)
894894
{
895+
if (log == null || string.IsNullOrEmpty(log.ConversationId)) return;
896+
895897
var completiongLog = new LlmCompletionLogDocument
896898
{
897899
Id = string.IsNullOrEmpty(log.Id) ? Guid.NewGuid().ToString() : log.Id,

0 commit comments

Comments
 (0)