From 5ff7e1bcb36bf461a17a5e44558b0e4f6ff8cf8e Mon Sep 17 00:00:00 2001 From: chinganc Date: Wed, 12 Feb 2025 12:40:07 -0800 Subject: [PATCH] Fix missing import --- tests/unit_tests/test_llm.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/unit_tests/test_llm.py b/tests/unit_tests/test_llm.py index b4ad90c..d660602 100644 --- a/tests/unit_tests/test_llm.py +++ b/tests/unit_tests/test_llm.py @@ -1,5 +1,6 @@ from opto.utils.llm import LLM from opto.optimizers.utils import print_color +import os if os.path.exists("OAI_CONFIG_LIST") or os.environ.get("DEFAULT_LITELLM_MODEL") or os.environ.get("OPENAI_API_KEY"): llm = LLM()