diff --git a/vertexai/_genai/datasets.py b/vertexai/_genai/datasets.py
index 77b3cd97f0..d90bcdab2a 100644
--- a/vertexai/_genai/datasets.py
+++ b/vertexai/_genai/datasets.py
@@ -15,6 +15,7 @@
 
 # Code generated by the Google Gen AI SDK generator DO NOT EDIT.
 
+import asyncio
 import asyncio
 import json
 import logging
diff --git a/vertexai/_genai/prompts.py b/vertexai/_genai/prompts.py
index 6ed6da8bca..8888c35319 100644
--- a/vertexai/_genai/prompts.py
+++ b/vertexai/_genai/prompts.py
@@ -15,6 +15,7 @@
 
 # Code generated by the Google Gen AI SDK generator DO NOT EDIT.
 
+import asyncio
 import asyncio
 import datetime
 import json