diff --git a/pkgs/google_generative_ai/CHANGELOG.md b/pkgs/google_generative_ai/CHANGELOG.md
index 1be7a5d..d648248 100644
--- 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
index 7520d4f..4ddd7f2 100644
--- 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
index 6651d1f..4c7d074 100644
--- 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
index 29a21c0..ac8c1fa 100644
--- 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).