File tree Expand file tree Collapse file tree 2 files changed +2
-7
lines changed
src/KernelMemory.DashScope
test/KernelMemory.DashScope.UnitTests Expand file tree Collapse file tree 2 files changed +2
-7
lines changed Original file line number Diff line number Diff line change @@ -72,11 +72,6 @@ public async IAsyncEnumerable<GeneratedTextContent> GenerateTextAsync(
72
72
var tokens = dashScopeClient . GetTextCompletionStreamAsync ( request , cancellationToken ) ;
73
73
await foreach ( var token in tokens )
74
74
{
75
- if ( cancellationToken . IsCancellationRequested )
76
- {
77
- break ;
78
- }
79
-
80
75
yield return new GeneratedTextContent (
81
76
token . Output . Text ?? string . Empty ,
82
77
token . Usage . ToKernelMemoryTokenUsage ( modelId ) ) ;
Original file line number Diff line number Diff line change @@ -26,7 +26,7 @@ public async Task TextGenerator_GenerateText_IncrementalGenerateAsync()
26
26
var response = await generator . GenerateTextAsync ( Cases . Text , Cases . TextGenerationOptions ) . ToListAsync ( ) ;
27
27
28
28
// Assert
29
- response [ 0 ] . Should ( ) . BeSameAs (
29
+ response [ 0 ] . Text . Should ( ) . BeSameAs (
30
30
Cases . TextGenerationResponse . Output . Text ,
31
31
"generated text should mapped from output.text" ) ;
32
32
captured . Should ( ) . BeEquivalentTo (
@@ -51,7 +51,7 @@ public async Task TextGenerator_DefaultsToZero_MapZeroToNullAsync()
51
51
var response = await generator . GenerateTextAsync ( Cases . Text , Cases . TextGenerationOptions ) . ToListAsync ( ) ;
52
52
53
53
// Assert
54
- response [ 0 ] . Should ( ) . BeSameAs (
54
+ response [ 0 ] . Text . Should ( ) . BeSameAs (
55
55
Cases . TextGenerationResponse . Output . Text ,
56
56
"generated text should mapped from output.text" ) ;
57
57
captured . Should ( ) . BeEquivalentTo (
You can’t perform that action at this time.
0 commit comments