diff --git a/promptlayer/__init__.py b/promptlayer/__init__.py index c3e4852..bd1a6b9 100644 --- a/promptlayer/__init__.py +++ b/promptlayer/__init__.py @@ -1,4 +1,4 @@ from .promptlayer import AsyncPromptLayer, PromptLayer -__version__ = "1.0.46" +__version__ = "1.0.47" __all__ = ["PromptLayer", "AsyncPromptLayer", "__version__"] diff --git a/promptlayer/utils.py b/promptlayer/utils.py index 9daf193..d60df50 100644 --- a/promptlayer/utils.py +++ b/promptlayer/utils.py @@ -1767,7 +1767,9 @@ def google_stream_completion(results: list): def google_request(request: GetPromptTemplateResponse, **kwargs): from google import genai - client = genai.Client() + # First look for env variable "GOOGLE_API_KEY". Default to "GEMINI_API_KEY" if that doesn't exist + api_key = os.environ.get("GOOGLE_API_KEY") or os.environ.get("GEMINI_API_KEY") + client = genai.Client(api_key=api_key) request_to_make = MAP_TYPE_TO_GOOGLE_FUNCTION[request["prompt_template"]["type"]] return request_to_make(client, **kwargs) diff --git a/pyproject.toml b/pyproject.toml index 14c673f..1f5d374 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "promptlayer" -version = "1.0.46" +version = "1.0.47" description = "PromptLayer is a platform for prompt engineering and tracks your LLM requests." authors = ["Magniv "] license = "Apache-2.0"