[azopenai] Make sure we defer close the eventreader in our examples. (#…

…21312)
Azure · Aug 2, 2023 · 4060121 · 4060121
1 parent 6e977ba
commit 4060121
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/sdk/ai/azopenai/example_client_getchatcompletions_test.go b/sdk/ai/azopenai/example_client_getchatcompletions_test.go
@@ -229,11 +229,12 @@ func ExampleClient_GetChatCompletionsStream() {
 		// TODO: handle error
 	}
 
-	streamReader := resp.ChatCompletionsStream
+	defer resp.ChatCompletionsStream.Close()
+
 	gotReply := false
 
 	for {
-		chatCompletions, err := streamReader.Read()
+		chatCompletions, err := resp.ChatCompletionsStream.Read()
 
 		if errors.Is(err, io.EOF) {
 			break

diff --git a/sdk/ai/azopenai/example_client_getcompletions_test.go b/sdk/ai/azopenai/example_client_getcompletions_test.go
@@ -95,6 +95,8 @@ func ExampleClient_GetCompletionsStream() {
 		// TODO: handle error
 	}
 
+	defer resp.CompletionsStream.Close()
+
 	for {
 		entry, err := resp.CompletionsStream.Read()