jebin2 commited on
Commit
fccc73c
·
1 Parent(s): ff14cb7

refactor: retrieve GEMINI_API_KEY from config utility instead of environment variables.

Browse files
Files changed (1) hide show
  1. src/google_src/ai_studio_sdk.py +7 -6
src/google_src/ai_studio_sdk.py CHANGED
@@ -5,6 +5,7 @@ import uuid
5
  import time
6
  from src.utils import logger
7
  import utils
 
8
 
9
  def _get_request_id() -> str:
10
  return uuid.uuid4().hex[:8]
@@ -32,9 +33,9 @@ def generate_video(prompt: str, output_path: str, image: str = None) -> str | No
32
  logger.warning("[%s] generate_video: empty prompt provided", req_id)
33
  return None
34
 
35
- api_key = os.environ.get("GEMINI_API_KEY")
36
  if not api_key:
37
- logger.error("[%s] generate_video: GEMINI_API_KEY not found in environment", req_id)
38
  return None
39
 
40
  try:
@@ -100,9 +101,9 @@ def generate_image(prompt: str) -> str | None:
100
  logger.warning("[%s] generate_image: empty prompt provided", req_id)
101
  return None
102
 
103
- api_key = os.environ.get("GEMINI_API_KEY")
104
  if not api_key:
105
- logger.error("[%s] generate_image: GEMINI_API_KEY not found in environment", req_id)
106
  return None
107
 
108
  try:
@@ -176,9 +177,9 @@ def generate(prompt: str, model: str = "gemini-flash-latest") -> str | None:
176
  logger.warning("[%s] generate: empty prompt provided", req_id)
177
  return None
178
 
179
- api_key = os.environ.get("GEMINI_API_KEY")
180
  if not api_key:
181
- logger.error("[%s] generate: GEMINI_API_KEY not found in environment", req_id)
182
  return None
183
 
184
  try:
 
5
  import time
6
  from src.utils import logger
7
  import utils
8
+ from src.config import get_config_value
9
 
10
  def _get_request_id() -> str:
11
  return uuid.uuid4().hex[:8]
 
33
  logger.warning("[%s] generate_video: empty prompt provided", req_id)
34
  return None
35
 
36
+ api_key = get_config_value("GEMINI_API_KEY")
37
  if not api_key:
38
+ logger.error("[%s] generate_video: GEMINI_API_KEY not found in config", req_id)
39
  return None
40
 
41
  try:
 
101
  logger.warning("[%s] generate_image: empty prompt provided", req_id)
102
  return None
103
 
104
+ api_key = get_config_value("GEMINI_API_KEY")
105
  if not api_key:
106
+ logger.error("[%s] generate_image: GEMINI_API_KEY not found in config", req_id)
107
  return None
108
 
109
  try:
 
177
  logger.warning("[%s] generate: empty prompt provided", req_id)
178
  return None
179
 
180
+ api_key = get_config_value("GEMINI_API_KEY")
181
  if not api_key:
182
+ logger.error("[%s] generate: GEMINI_API_KEY not found in config", req_id)
183
  return None
184
 
185
  try: