Add support for additional model name prefixes (#98)

Sources/GoogleAI/GenerativeModel.swift

@@ -21,8 +21,11 @@ public final class GenerativeModel {
   // The prefix for a model resource in the Gemini API.
   private static let modelResourcePrefix = "models/"
 
+  // The prefix for a tuned model resource in the Gemini API.
+  private static let tunedModelResourcePrefix = "tunedModels/"
+
   /// The resource name of the model in the backend; has the format "models/model-name".
-  private let modelResourceName: String
+  let modelResourceName: String
 
   /// The backing service responsible for sending and receiving model requests to the backend.
   let generativeAIService: GenerativeAIService
@@ -246,7 +249,7 @@ public final class GenerativeModel {
 
   /// Returns a model resource name of the form "models/model-name" based on `name`.
   private static func modelResourceName(name: String) -> String {
-    if name.hasPrefix(modelResourcePrefix) {
+    if name.contains("/") {
       return name
     } else {
       return modelResourcePrefix + name

Tests/GoogleAITests/GenerativeModelTests.swift

@@ -845,6 +845,33 @@ final class GenerativeModelTests: XCTestCase {
     XCTAssertEqual(response.totalTokens, 6)
   }
 
+  // MARK: - Model Resource Name
+
+  func testModelResourceName_noPrefix() async throws {
+    let modelName = "my-model"
+    let modelResourceName = "models/\(modelName)"
+
+    model = GenerativeModel(name: modelName, apiKey: "API_KEY")
+
+    XCTAssertEqual(model.modelResourceName, modelResourceName)
+  }
+
+  func testModelResourceName_modelsPrefix() async throws {
+    let modelResourceName = "models/my-model"
+
+    model = GenerativeModel(name: modelResourceName, apiKey: "API_KEY")
+
+    XCTAssertEqual(model.modelResourceName, modelResourceName)
+  }
+
+  func testModelResourceName_tunedModelsPrefix() async throws {
+    let tunedModelResourceName = "tunedModels/my-model"
+
+    model = GenerativeModel(name: tunedModelResourceName, apiKey: "API_KEY")
+
+    XCTAssertEqual(model.modelResourceName, tunedModelResourceName)
+  }
+
   // MARK: - Helpers
 
   private func nonHTTPRequestHandler() throws -> ((URLRequest) -> (