diff --git a/docs/my-website/docs/providers/vertex.md b/docs/my-website/docs/providers/vertex.md
index baa5e3623..8e48780b0 100644
--- a/docs/my-website/docs/providers/vertex.md
+++ b/docs/my-website/docs/providers/vertex.md
@@ -2129,6 +2129,9 @@ Here's the code for storing your service account credentials as `GOOGLE_APPLICAT
 
 
 ```python
+import os 
+import tempfile
+
 def load_vertex_ai_credentials():
   # Define the path to the vertex_key.json file
   print("loading vertex ai credentials")