Skip to content

Commit f712d19

Browse files
authored
Update API key references from GOOGLE to GEMINI
1 parent f13a000 commit f712d19

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

packages/memory/llm/providers/gemini.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,10 +29,10 @@ def __init__(
2929
super().__init__(model=model, temperature=temperature, max_tokens=max_tokens)
3030

3131
# 2. Auth: Try settings first, then fallback to OS environment
32-
self.api_key = api_key or getattr(settings, "GOOGLE_API_KEY", None) or os.getenv("GOOGLE_API_KEY")
32+
self.api_key = api_key or getattr(settings, "GEMINI_API_KEY", None) or os.getenv("GEMINI_API_KEY")
3333

3434
if not self.api_key:
35-
raise ValueError("GOOGLE_API_KEY is missing. Please add it to your .env file.")
35+
raise ValueError("GEMINI_API_KEY is missing. Please add it to your .env file.")
3636

3737
def get_provider_name(self) -> str:
3838
return "gemini"
@@ -85,4 +85,4 @@ def stream_text(self, prompt: str, system_prompt: Optional[str] = None) -> Itera
8585
Stream text using Gemini (Simplified to non-streaming for stability first).
8686
"""
8787
# For the initial setup, we simply yield the full result to avoid async complexities
88-
yield self.generate_text(prompt, system_prompt)
88+
yield self.generate_text(prompt, system_prompt)

0 commit comments

Comments
 (0)