Update text_generation.py

google-gemini · Sep 15, 2024 · 2e2db45 · 2e2db45
1 parent b0779ad
commit 2e2db45
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/samples/text_generation.py b/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)