Make OpenAIEmbeddingService less strict about the response

Author: intitni

Tool/Sources/OpenAIService/APIs/EmbeddingAPIDefinitions.swift

@@ -1,6 +1,7 @@
 import AIModel
 import Foundation
 import Preferences
+import CodableWrappers
 
 protocol EmbeddingAPI {
     func embed(text: String) async throws -> EmbeddingResponse
@@ -12,17 +13,28 @@ public struct EmbeddingResponse: Decodable {
     public struct Object: Decodable {
         public var embedding: [Float]
         public var index: Int
+        @FallbackDecoding<EmptyString>
         public var object: String
     }
 
+    @FallbackDecoding<EmptyArray>
     public var data: [Object]
+    @FallbackDecoding<EmptyString>
     public var model: String
 
     public struct Usage: Decodable {
+        @FallbackDecoding<EmptyInt>
         public var prompt_tokens: Int
+        @FallbackDecoding<EmptyInt>
         public var total_tokens: Int
+        
+        public struct Fallback: FallbackValueProvider {
+            public static var defaultValue: Usage { Usage(prompt_tokens: 0, total_tokens: 0) }
+        }
     }
 
+    @FallbackDecoding<Usage.Fallback>
     public var usage: Usage
 }
 
+

Tool/Sources/OpenAIService/APIs/OpenAIEmbeddingService.swift

@@ -50,15 +50,6 @@ struct OpenAIEmbeddingService: EmbeddingAPI {
         }
 
         let embeddingResponse = try JSONDecoder().decode(EmbeddingResponse.self, from: result)
-        #if DEBUG
-        Logger.service.info("""
-        Embedding usage
-        - number of strings: \(text.count)
-        - prompt tokens: \(embeddingResponse.usage.prompt_tokens)
-        - total tokens: \(embeddingResponse.usage.total_tokens)
-
-        """)
-        #endif
         return embeddingResponse
     }
 
@@ -92,15 +83,6 @@ struct OpenAIEmbeddingService: EmbeddingAPI {
         }
 
         let embeddingResponse = try JSONDecoder().decode(EmbeddingResponse.self, from: result)
-        #if DEBUG
-        Logger.service.info("""
-        Embedding usage
-        - number of strings: \(tokens.count)
-        - prompt tokens: \(embeddingResponse.usage.prompt_tokens)
-        - total tokens: \(embeddingResponse.usage.total_tokens)
-
-        """)
-        #endif
         return embeddingResponse
     }