Skip to content

Commit 9f31f84

Browse files
authored
Merge pull request #313 from iceljc/features/refine-content-log
refine content log after call bot
2 parents 5e1f5c0 + 41ff234 commit 9f31f84

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

src/Plugins/BotSharp.Plugin.ChatHub/Hooks/StreamingLogHook.cs

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,18 @@ public async Task AfterGenerated(RoleDialogModel message, TokenStatsModel tokenS
8282
var conversationId = _state.GetConversationId();
8383
var agent = await agentService.LoadAgent(message.CurrentAgentId);
8484

85-
await _chatHub.Clients.User(_user.Id).SendAsync("OnConversationContentLogGenerated", BuildContentLog(conversationId, agent?.Name, tokenStats.Prompt, message));
85+
string log;
86+
if (message.Role == AgentRole.Function)
87+
{
88+
log = $"[{agent?.Name}]: {message.FunctionName}({message.FunctionArgs}) => {message.Content}";
89+
log += $"\r\n<== MessageId: {message.MessageId}";
90+
}
91+
else
92+
{
93+
log = tokenStats.Prompt;
94+
}
95+
96+
await _chatHub.Clients.User(_user.Id).SendAsync("OnConversationContentLogGenerated", BuildContentLog(conversationId, agent?.Name, log, message));
8697
}
8798

8899
/// <summary>

0 commit comments

Comments
 (0)