Skip to content

Commit b57bae4

Browse files
author
Piotr Stachaczynski
committed
fix: tokencallback fix
1 parent 79adcb3 commit b57bae4

File tree

3 files changed

+17
-1
lines changed

3 files changed

+17
-1
lines changed

src/MaIN.Core/.nuspec

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
<package>
33
<metadata>
44
<id>MaIN.NET</id>
5-
<version>0.7.4</version>
5+
<version>0.7.5</version>
66
<authors>Wisedev</authors>
77
<owners>Wisedev</owners>
88
<icon>favicon.png</icon>

src/MaIN.Services/Services/LLMService/AnthropicService.cs

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,14 @@ private void ValidateApiKey()
7474
resultBuilder.Append(memoryResult!.Message.Content);
7575
lastMessage.MarkProcessed();
7676
UpdateSessionCache(chat.Id, resultBuilder.ToString(), options.CreateSession);
77+
if (options.TokenCallback != null)
78+
{
79+
await options.TokenCallback(new LLMTokenValue()
80+
{
81+
Text = resultBuilder.ToString(),
82+
Type = TokenType.FullAnswer
83+
});
84+
}
7785
return CreateChatResult(chat, resultBuilder.ToString(), tokens);
7886
}
7987

src/MaIN.Services/Services/LLMService/OpenAiCompatibleService.cs

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,14 @@ public abstract class OpenAiCompatibleService(
7272
resultBuilder.Append(memoryResult!.Message.Content);
7373
lastMessage.MarkProcessed();
7474
UpdateSessionCache(chat.Id, resultBuilder.ToString(), options.CreateSession);
75+
if (options.TokenCallback != null)
76+
{
77+
await InvokeTokenCallbackAsync(options.TokenCallback, new LLMTokenValue()
78+
{
79+
Text = resultBuilder.ToString(),
80+
Type = TokenType.FullAnswer
81+
});
82+
}
7583
return CreateChatResult(chat, resultBuilder.ToString(), tokens);
7684
}
7785

0 commit comments

Comments
 (0)