switch between content generation remotes based on ai source

Author: iamgabrielma

Modules/Sources/Yosemite/Actions/ProductAction.swift

@@ -1,6 +1,11 @@
 import Foundation
 import Networking
 
+public enum AISource {
+    case jetpack
+    case merchant
+}
+
 public enum ItemIdentifierSearchResult {
     case product(Product)
     case variation(ProductVariation)
@@ -130,6 +135,7 @@ public enum ProductAction: Action {
     ///
     case identifyLanguage(siteID: Int64,
                           string: String,
+                          aiSource: AISource,
                           feature: GenerativeContentRemoteFeature,
                           completion: (Result<String, Error>) -> Void)
 

Modules/Sources/Yosemite/Stores/ProductStore.swift

@@ -122,9 +122,11 @@ public class ProductStore: Store {
             replaceProductLocally(product: product, onCompletion: onCompletion)
         case let .checkIfStoreHasProducts(siteID, status, onCompletion):
             checkIfStoreHasProducts(siteID: siteID, status: status, onCompletion: onCompletion)
-        case let .identifyLanguage(siteID, string, feature, completion):
+        case let .identifyLanguage(siteID, string, aiSource, feature, completion):
             identifyLanguage(siteID: siteID,
-                             string: string, feature: feature,
+                             string: string,
+                             aisource: aiSource,
+                             feature: feature,
                              completion: completion)
         case let .generateProductDescription(siteID, name, features, language, completion):
             generateProductDescription(siteID: siteID, name: name, features: features, language: language, completion: completion)
@@ -586,15 +588,28 @@ private extension ProductStore {
 
     func identifyLanguage(siteID: Int64,
                           string: String,
+                          aisource: AISource,
                           feature: GenerativeContentRemoteFeature,
                           completion: @escaping (Result<String, Error>) -> Void) {
         Task { @MainActor in
-            let result = await Result {
-                try await generativeContentRemote.identifyLanguage(siteID: siteID,
-                                                                   string: string,
-                                                                   feature: feature)
+            switch aisource {
+            case .jetpack:
+                let result = await Result {
+                    try await generativeContentRemote.identifyLanguage(siteID: siteID,
+                                                                       string: string,
+                                                                       feature: feature)
+                }
+                completion(result)
+            case .merchant:
+                let result = await Result {
+                    // Temporary. This will come from the KeyChain rather than the environment
+                    let key = ProcessInfo.processInfo.environment["openai-hack-key"] ?? "api key not found"
+                    return try await MerchantGenerativeContentRemote(apiKey: key).identifyLanguage(siteID: siteID,
+                                                                                                   string: string,
+                                                                                                   feature: feature)
+                }
+                completion(result)
             }
-            completion(result)
         }
     }
 

WooCommerce/Classes/ViewRelated/Products/AI/ProductDescriptionGenerationViewModel.swift

@@ -49,6 +49,10 @@ final class ProductDescriptionGenerationViewModel: ObservableObject {
     /// Language used in product identified by AI
     ///
     private var languageIdentifiedUsingAI: String?
+    
+    private var aiSource: AISource {
+        ServiceLocator.featureFlagService.isFeatureFlagEnabled(.allowMerchantAIAPIKey) ? .merchant : .jetpack
+    }
 
     init(siteID: Int64,
          name: String,
@@ -139,6 +143,7 @@ private extension ProductDescriptionGenerationViewModel {
             let language = try await withCheckedThrowingContinuation { continuation in
                 stores.dispatch(ProductAction.identifyLanguage(siteID: siteID,
                                                                string: name + " " + features,
+                                                               aiSource: aiSource,
                                                                feature: .productDescription,
                                                                completion: { result in
                     continuation.resume(with: result)

WooCommerce/Classes/ViewRelated/Products/AI/ProductSharingMessageGenerationViewModel.swift

@@ -51,6 +51,10 @@ final class ProductSharingMessageGenerationViewModel: ObservableObject {
     /// Language used in product identified by AI
     ///
     private var languageIdentifiedUsingAI: String?
+    
+    private var aiSource: AISource {
+        ServiceLocator.featureFlagService.isFeatureFlagEnabled(.allowMerchantAIAPIKey) ? .merchant : .jetpack
+    }
 
     init(siteID: Int64,
          url: String,
@@ -152,6 +156,7 @@ private extension ProductSharingMessageGenerationViewModel {
             let language = try await withCheckedThrowingContinuation { continuation in
                 stores.dispatch(ProductAction.identifyLanguage(siteID: siteID,
                                                                string: productName + " " + productDescription,
+                                                               aiSource: aiSource,
                                                                feature: .productSharing,
                                                                completion: { result in
                     continuation.resume(with: result)

WooCommerce/Classes/ViewRelated/Products/Add Product/AddProductWithAI/Preview/ProductDetailPreviewViewModel.swift

@@ -153,6 +153,10 @@ final class ProductDetailPreviewViewModel: ObservableObject {
         return productImageUploader.actionHandler(key: key, originalStatuses: [])
     }()
 
+    private var aiSource: AISource {
+        ServiceLocator.featureFlagService.isFeatureFlagEnabled(.allowMerchantAIAPIKey) ? .merchant : .jetpack
+    }
+
     init(siteID: Int64,
          productFeatures: String,
          imageState: ImageState,
@@ -529,6 +533,7 @@ private extension ProductDetailPreviewViewModel {
             let language = try await withCheckedThrowingContinuation { continuation in
                 stores.dispatch(ProductAction.identifyLanguage(siteID: siteID,
                                                                string: productInfo,
+                                                               aiSource: aiSource,
                                                                feature: .productCreation,
                                                                completion: { result in
                     continuation.resume(with: result)

WooCommerce/Classes/ViewRelated/Products/Add Product/AddProductWithAI/ProductCreationAIEligibilityChecker.swift

@@ -20,6 +20,11 @@ final class ProductCreationAIEligibilityChecker: ProductCreationAIEligibilityChe
             return false
         }
 
-        return site.isWordPressComStore || site.isAIAssistantFeatureActive
+        if ServiceLocator.featureFlagService.isFeatureFlagEnabled(.allowMerchantAIAPIKey) {
+            // Temporary: Always allow for AI usage if the flag is enabled, bypassing eligibility criteria
+            return true
+        } else {
+            return site.isWordPressComStore || site.isAIAssistantFeatureActive
+        }
     }
 }