Add apiVersion to RequestOptions

Sources/GoogleAI/CountTokensRequest.swift

@@ -30,7 +30,7 @@ extension CountTokensRequest: GenerativeAIRequest {
   typealias Response = CountTokensResponse
 
   var url: URL {
-    URL(string: "\(GenerativeAISwift.baseURL)/\(model):countTokens")!
+    URL(string: "\(GenerativeAISwift.baseURL)/\(options.apiVersion)/\(model):countTokens")!
   }
 }
 

Sources/GoogleAI/GenerateContentRequest.swift

@@ -36,10 +36,11 @@ extension GenerateContentRequest: GenerativeAIRequest {
   typealias Response = GenerateContentResponse
 
   var url: URL {
+    let modelURL = "\(GenerativeAISwift.baseURL)/\(options.apiVersion)/\(model)"
     if isStreaming {
-      URL(string: "\(GenerativeAISwift.baseURL)/\(model):streamGenerateContent?alt=sse")!
+      return URL(string: "\(modelURL):streamGenerateContent?alt=sse")!
     } else {
-      URL(string: "\(GenerativeAISwift.baseURL)/\(model):generateContent")!
+      return URL(string: "\(modelURL):generateContent")!
     }
   }
 }

Sources/GoogleAI/GenerativeAIRequest.swift

@@ -28,11 +28,17 @@ public struct RequestOptions {
   /// `URLRequest`.
   let timeout: TimeInterval?
 
+  /// The API version to use in requests to the backend.
+  let apiVersion: String
+
   /// Initializes a request options object.
   ///
-  /// - Parameter timeout The request’s timeout interval in seconds; if not specified uses the
-  /// default value for a `URLRequest`.
-  public init(timeout: TimeInterval? = nil) {
+  /// - Parameters:
+  ///   - timeout The request’s timeout interval in seconds; if not specified uses the default value
+  ///   for a `URLRequest`.
+  ///   - apiVersion The API version to use in requests to the backend; defaults to "v1".
+  public init(timeout: TimeInterval? = nil, apiVersion: String = "v1") {
     self.timeout = timeout
+    self.apiVersion = apiVersion
   }
 }

Sources/GoogleAI/GenerativeAISwift.swift

@@ -21,5 +21,5 @@ import Foundation
 public enum GenerativeAISwift {
   /// String value of the SDK version
   public static let version = "0.4.7"
-  static let baseURL = "https://generativelanguage.googleapis.com/v1"
+  static let baseURL = "https://generativelanguage.googleapis.com"
 }