Add tool and toolConfig to createModelWithBaseUri() (#137)

Allow forwarding from the Vertex model constructor.
google-gemini · Apr 29, 2024 · 8f6bb83 · 8f6bb83
1 parent 467d8c1
commit 8f6bb83
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 4 deletions.
diff --git a/pkgs/google_generative_ai/CHANGELOG.md b/pkgs/google_generative_ai/CHANGELOG.md
@@ -1,3 +1,5 @@
+## 0.3.3-wip
+
 ## 0.3.2
 
 - Use API version `v1beta` by default.

diff --git a/pkgs/google_generative_ai/lib/src/model.dart b/pkgs/google_generative_ai/lib/src/model.dart
@@ -343,7 +343,9 @@ GenerativeModel createModelWithBaseUri({
   FutureOr<Map<String, String>> Function()? requestHeaders,
   List<SafetySetting> safetySettings = const [],
   GenerationConfig? generationConfig,
+  List<Tool>? tools,
   Content? systemInstruction,
+  ToolConfig? toolConfig,
 }) =>
     GenerativeModel._withClient(
       client: HttpApiClient(apiKey: apiKey, requestHeaders: requestHeaders),
@@ -352,6 +354,6 @@ GenerativeModel createModelWithBaseUri({
       generationConfig: generationConfig,
       baseUri: baseUri,
       systemInstruction: systemInstruction,
-      tools: null,
-      toolConfig: null,
+      tools: tools,
+      toolConfig: toolConfig,
     );
diff --git a/pkgs/google_generative_ai/lib/src/version.dart b/pkgs/google_generative_ai/lib/src/version.dart
@@ -12,4 +12,4 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-const packageVersion = '0.3.2';
+const packageVersion = '0.3.3-wip';
diff --git a/pkgs/google_generative_ai/pubspec.yaml b/pkgs/google_generative_ai/pubspec.yaml
@@ -1,6 +1,6 @@
 name: google_generative_ai
 # Update `lib/version.dart` when changing version.
-version: 0.3.2
+version: 0.3.3-wip
 description: >-
   The Google AI Dart SDK enables developers to use Google's state-of-the-art
   generative AI models (like Gemini).