fix for passing tests

landing-ai · Apr 6, 2024 · 6cdb50f · 6cdb50f
1 parent c18f749
commit 6cdb50f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 9 deletions.
diff --git 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
@@ -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