From 6cdb50f02604fb2541089d67285a323e9da04b2e Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Fri, 5 Apr 2024 17:29:56 -0700 Subject: [PATCH] fix for passing tests --- vision_agent/llm/llm.py | 8 +++----- vision_agent/lmm/lmm.py | 7 +++---- 2 files changed, 6 insertions(+), 9 deletions(-) diff --git a/vision_agent/llm/llm.py b/vision_agent/llm/llm.py index 1ccc5809..9022ef73 100644 --- a/vision_agent/llm/llm.py +++ b/vision_agent/llm/llm.py @@ -39,12 +39,10 @@ def __init__( **kwargs: Any ): if not api_key: - api_key = os.getenv("OPENAI_API_KEY") + self.client = OpenAI() + else: + self.client = OpenAI(api_key=api_key) - if not api_key: - raise ValueError("OpenAI API key is required.") - - self.client = OpenAI(api_key=api_key) self.model_name = model_name self.kwargs = kwargs if json_mode: diff --git a/vision_agent/lmm/lmm.py b/vision_agent/lmm/lmm.py index 3eee8766..99e5b01b 100644 --- a/vision_agent/lmm/lmm.py +++ b/vision_agent/lmm/lmm.py @@ -105,10 +105,9 @@ def __init__( **kwargs: Any, ): if not api_key: - api_key = os.getenv("OPENAI_API_KEY") - - if not api_key: - raise ValueError("OpenAI API key is required.") + self.client = OpenAI() + else: + self.client = OpenAI(api_key=api_key) self.client = OpenAI(api_key=api_key) self.model_name = model_name