bedrock: Fix unchecked error when stream channel is closed (#998)

fix(bedrock): Handle unchecked error when stream channel is closed
tmc · Sep 13, 2024 · 17d9e48 · 17d9e48
1 parent 71f5789
commit 17d9e48
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/llms/bedrock/internal/bedrockclient/provider_anthropic.go b/llms/bedrock/internal/bedrockclient/provider_anthropic.go
@@ -267,6 +267,9 @@ func parseStreamingCompletionResponse(ctx context.Context, client *bedrockruntim
 			}
 		}
 	}
+	if err = stream.Err(); err != nil {
+		return nil, err
+	}
 
 	return &llms.ContentResponse{
 		Choices: contentchoices,