Fixed a bug where streaming would not be enabled where needed

SwiftedMind · May 11, 2023 · c90d3fc · c90d3fc
1 parent 534b178
commit c90d3fc
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/Sources/ChatGPT/ChatGPT+StreamedAnswer.swift b/Sources/ChatGPT/ChatGPT+StreamedAnswer.swift
@@ -86,6 +86,10 @@ extension ChatGPT {
         /// - Throws: A `GPTSwiftError`.
         @available(macOS 12.0, iOS 15.0, watchOS 8.0, tvOS 15.0, *)
         public func ask(request chatRequest: ChatRequest) async throws -> AsyncThrowingStream<String, Swift.Error> {
+
+            var chatRequest = chatRequest
+            chatRequest.stream = true
+
             let request = Request(path: API.v1ChatCompletion, method: .post, body: chatRequest)
             var urlRequest = try await client.makeURLRequest(for: request)
             _addHeaders(to: &urlRequest, apiKey: apiKey)

diff --git a/Sources/GPT/GPT+StreamedAnswer.swift b/Sources/GPT/GPT+StreamedAnswer.swift
@@ -69,6 +69,10 @@ extension GPT {
         /// - Throws: A `Swift.Error` if the request fails or the server returns an unauthorized status code.
         @available(macOS 12.0, iOS 15.0, watchOS 8.0, tvOS 15.0, *)
         public func complete(request completionRequest: CompletionRequest) async throws -> AsyncThrowingStream<String, Swift.Error> {
+
+            var completionRequest = completionRequest
+            completionRequest.stream = true
+
             let request = Request(path: API.v1Completion, method: .post, body: completionRequest)
             var urlRequest = try await client.makeURLRequest(for: request)
             _addHeaders(to: &urlRequest, apiKey: apiKey)