From a93cfc80a9b38e2c6b79b5d9e719012dbbe9c76b Mon Sep 17 00:00:00 2001
From: Mark Daoust <markdaoust@google.com>
Date: Fri, 18 Aug 2023 17:20:34 -0700
Subject: [PATCH] Add 'import os'

---
 google/generativeai/__init__.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/google/generativeai/__init__.py b/google/generativeai/__init__.py
index 821d8aed7..6b74f83a8 100644
--- a/google/generativeai/__init__.py
+++ b/google/generativeai/__init__.py
@@ -22,6 +22,7 @@
 
 ```
 import google.generativeai as palm
+import os
 
 palm.configure(api_key=os.environ['API_KEY'])
 ```