Log server error responses without `-GoogleGenerativeAIDebugLogEnable…

…d` (google-gemini#177)
enefry · May 29, 2024 · 5478400 · 5478400
1 parent 97a81a2
commit 5478400
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/Sources/GoogleAI/GenerativeAIService.swift b/Sources/GoogleAI/GenerativeAIService.swift
@@ -41,9 +41,9 @@ struct GenerativeAIService {
 
     // Verify the status code is 200
     guard response.statusCode == 200 else {
-      Logging.default.error("[GoogleGenerativeAI] The server responded with an error: \(response)")
+      Logging.network.error("[GoogleGenerativeAI] The server responded with an error: \(response)")
       if let responseString = String(data: data, encoding: .utf8) {
-        Logging.network.error("[GoogleGenerativeAI] Response payload: \(responseString)")
+        Logging.default.error("[GoogleGenerativeAI] Response payload: \(responseString)")
       }
 
       throw parseError(responseData: data)
@@ -89,14 +89,14 @@ struct GenerativeAIService {
 
         // Verify the status code is 200
         guard response.statusCode == 200 else {
-          Logging.default
+          Logging.network
             .error("[GoogleGenerativeAI] The server responded with an error: \(response)")
           var responseBody = ""
           for try await line in stream.lines {
             responseBody += line + "\n"
           }
 
-          Logging.network.error("[GoogleGenerativeAI] Response payload: \(responseBody)")
+          Logging.default.error("[GoogleGenerativeAI] Response payload: \(responseBody)")
           continuation.finish(throwing: parseError(responseBody: responseBody))
 
           return