pass api_key to genai.configure

joshbickett · joshbickett · commit d0a972fd4601 · 2024-01-19T14:31:38.000-08:00
diff --git a/operate/config.py b/operate/config.py
@@ -69,7 +69,7 @@ def initialize_google(self):
                 "[Config][initialize_google] no cached google_api_key, try to get from env."
             )
             api_key = os.getenv("GOOGLE_API_KEY")
-        genai.configure(api_key=os.getenv("GOOGLE_API_KEY"), transport="rest")
+        genai.configure(api_key=api_key, transport="rest")
         model = genai.GenerativeModel("gemini-pro-vision")
 
         return model

Original file line number	Diff line number	Diff line change
`@@ -69,7 +69,7 @@ def initialize_google(self):`
`69`	`69`	`"[Config][initialize_google] no cached google_api_key, try to get from env."`
`70`	`70`	`)`
`71`	`71`	`api_key = os.getenv("GOOGLE_API_KEY")`
`72`		`- genai.configure(api_key=os.getenv("GOOGLE_API_KEY"), transport="rest")`
	`72`	`+ genai.configure(api_key=api_key, transport="rest")`
`73`	`73`	`model = genai.GenerativeModel("gemini-pro-vision")`
`74`	`74`
`75`	`75`	`return model`