refactor: retrieve GEMINI_API_KEY from config utility instead of environment variables.
Browse files
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 =
|
| 36 |
if not api_key:
|
| 37 |
-
logger.error("[%s] generate_video: GEMINI_API_KEY not found in
|
| 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 =
|
| 104 |
if not api_key:
|
| 105 |
-
logger.error("[%s] generate_image: GEMINI_API_KEY not found in
|
| 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 =
|
| 180 |
if not api_key:
|
| 181 |
-
logger.error("[%s] generate: GEMINI_API_KEY not found in
|
| 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:
|