Add support for request options

Author: daymxn

FirebaseAI/Sources/FirebaseAI.swift

@@ -151,14 +151,15 @@ public final class FirebaseAI: Sendable {
   ///
   /// - Parameters:
   ///   - modelName: The name of the Livemodel to use, for example
-  /// `"gemini-live-2.5-flash-preview"`;
+  ///     `"gemini-live-2.5-flash-preview"`;
   ///     see [model versions](https://firebase.google.com/docs/ai-logic/live-api?api=dev#models-that-support-capability)
   ///     for a list of supported Live models.
   ///   - generationConfig: The content generation parameters your model should use.
   ///   - tools: A list of ``Tool`` objects that the model may use to generate the next response.
   ///   - toolConfig: Tool configuration for any ``Tool`` specified in the request.
   ///   - systemInstruction: Instructions that direct the model to behave a certain way; currently
   ///     only text content is supported.
+  ///   - requestOptions: Configuration parameters for sending requests to the backend.
   public func liveModel(modelName: String,
                         generationConfig: LiveGenerationConfig? = nil,
                         tools: [Tool]? = nil,
@@ -172,7 +173,8 @@ public final class FirebaseAI: Sendable {
       generationConfig: generationConfig,
       tools: tools,
       toolConfig: toolConfig,
-      systemInstruction: systemInstruction
+      systemInstruction: systemInstruction,
+      requestOptions: requestOptions
     )
   }
 

FirebaseAI/Sources/Types/Internal/Live/LiveSessionService.swift

@@ -44,6 +44,7 @@ actor LiveSessionService {
   let urlSession: URLSession
   let apiConfig: APIConfig
   let firebaseInfo: FirebaseInfo
+  let requestOptions: RequestOptions
   let tools: [Tool]?
   let toolConfig: ToolConfig?
   let systemInstruction: ModelContent?
@@ -73,7 +74,8 @@ actor LiveSessionService {
        firebaseInfo: FirebaseInfo,
        tools: [Tool]?,
        toolConfig: ToolConfig?,
-       systemInstruction: ModelContent?) {
+       systemInstruction: ModelContent?,
+       requestOptions: RequestOptions) {
     (responses, responseContinuation) = AsyncThrowingStream.makeStream()
     (messageQueue, messageQueueContinuation) = AsyncStream.makeStream()
     self.modelResourceName = modelResourceName
@@ -84,6 +86,7 @@ actor LiveSessionService {
     self.tools = tools
     self.toolConfig = toolConfig
     self.systemInstruction = systemInstruction
+    self.requestOptions = requestOptions
     setupTask = Task {}
   }
 
@@ -301,6 +304,7 @@ actor LiveSessionService {
       )
     }
     var urlRequest = URLRequest(url: url)
+    urlRequest.timeoutInterval = requestOptions.timeout
     urlRequest.setValue(firebaseInfo.apiKey, forHTTPHeaderField: "x-goog-api-key")
     urlRequest.setValue(
       "\(GenerativeAIService.languageTag) \(GenerativeAIService.firebaseVersionTag)",

FirebaseAI/Sources/Types/Public/Live/LiveGenerativeModel.swift

@@ -28,6 +28,7 @@ public final class LiveGenerativeModel {
   let toolConfig: ToolConfig?
   let systemInstruction: ModelContent?
   let urlSession: URLSession
+  let requestOptions: RequestOptions
 
   init(modelResourceName: String,
        firebaseInfo: FirebaseInfo,
@@ -36,7 +37,8 @@ public final class LiveGenerativeModel {
        tools: [Tool]? = nil,
        toolConfig: ToolConfig? = nil,
        systemInstruction: ModelContent? = nil,
-       urlSession: URLSession = GenAIURLSession.default) {
+       urlSession: URLSession = GenAIURLSession.default,
+       requestOptions: RequestOptions) {
     self.modelResourceName = modelResourceName
     self.firebaseInfo = firebaseInfo
     self.apiConfig = apiConfig
@@ -45,6 +47,7 @@ public final class LiveGenerativeModel {
     self.toolConfig = toolConfig
     self.systemInstruction = systemInstruction
     self.urlSession = urlSession
+    self.requestOptions = requestOptions
   }
 
   /// Start a ``LiveSession`` with the server for bidirectional streaming.
@@ -60,6 +63,7 @@ public final class LiveGenerativeModel {
       tools: tools,
       toolConfig: toolConfig,
       systemInstruction: systemInstruction,
+      requestOptions: requestOptions
     )
 
     await service.connect()