Pass requestOptions from API surface into createGenerativeModel

firebase · Jan 22, 2025 · ff55e55 · ff55e55
1 parent 1637edc
commit ff55e55
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/packages/firebase_vertexai/firebase_vertexai/lib/src/firebase_vertexai.dart b/packages/firebase_vertexai/firebase_vertexai/lib/src/firebase_vertexai.dart
@@ -110,6 +110,7 @@ class FirebaseVertexAI extends FirebasePluginPlatform {
       tools: tools,
       toolConfig: toolConfig,
       systemInstruction: systemInstruction,
+      requestOptions: requestOptions,
     );
   }
 }
diff --git a/packages/firebase_vertexai/firebase_vertexai/lib/src/model.dart b/packages/firebase_vertexai/firebase_vertexai/lib/src/model.dart
@@ -313,6 +313,7 @@ GenerativeModel createGenerativeModel({
       tools: tools,
       toolConfig: toolConfig,
       systemInstruction: systemInstruction,
+      requestOptions: requestOptions,
     );
 
 /// Creates a model with an overridden [ApiClient] for testing.
@@ -330,6 +331,7 @@ GenerativeModel createModelWithClient({
   List<SafetySetting>? safetySettings,
   List<Tool>? tools,
   ToolConfig? toolConfig,
+  RequestOptions? requestOptions,
 }) =>
     GenerativeModel._constructTestModel(
         model: model,
@@ -342,4 +344,5 @@ GenerativeModel createModelWithClient({
         systemInstruction: systemInstruction,
         tools: tools,
         toolConfig: toolConfig,
+        requestOptions: requestOptions,
         apiClient: client);