fix: content streaming on macOS 12+

tisfeng · Jan 26, 2024 · fcc3b11 · fcc3b11
1 parent 7ca612c
commit fcc3b11
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/Easydict/Feature/Service/Gemini/GeminiService.swift b/Easydict/Feature/Service/Gemini/GeminiService.swift
@@ -107,7 +107,10 @@ public final class GeminiService: QueryService {
 
                         print("gemini response: \(line)")
                         resultString += line
+                        result.translatedResults = [resultString]
+                        completion(result, nil)
                     }
+
                 } else {
                     let outputContent = try await model.generateContent(prompt)
                     guard let line = outputContent.text else {
@@ -116,16 +119,15 @@ public final class GeminiService: QueryService {
 
                     print("gemini response: \(line)")
                     resultString += line
+                    result.translatedResults = [resultString]
+                    completion(result, nil)
                 }
-
-                result.translatedResults = [resultString]
-                completion(result, nil)
             } catch {
                 /**
                  https://github.com/google/generative-ai-swift/issues/89
-                 
+
                  String(describing: error)
-                 
+
                  "internalError(underlying: GoogleGenerativeAI.RPCError(httpResponseCode: 400, message: \"API key not valid. Please pass a valid API key.\", status: GoogleGenerativeAI.RPCStatus.invalidArgument))"
                  */
                 let ezError = EZError(nsError: error)