Update text_generation.py

Author: logankilpatrick

samples/text_generation.py

@@ -26,7 +26,7 @@ def test_text_gen_text_only_prompt(self):
         import google.generativeai as genai
         import os
 
-        genai.configure(api_key=os.environ["API_KEY"])
+        genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
         model = genai.GenerativeModel("gemini-1.5-flash")
         response = model.generate_content("Explain how AI works")
         print(response.text)