Make llm related testing optional in CI.

Author: chinganc

tests/unit_tests/test_llm.py

@@ -1,7 +1,7 @@
 from opto.utils.llm import LLM
 from opto.optimizers.utils import print_color
 
-try:
+if os.path.exists("OAI_CONFIG_LIST") or os.environ.get("DEFAULT_LITELLM_MODEL") or os.environ.get("OPENAI_API_KEY"):
     llm = LLM()
     system_prompt = 'You are a helpful assistant.'
     user_prompt = "Hello world."
@@ -20,7 +20,3 @@
     print_color(f'System: {system_prompt}', 'red')
     print_color(f'User: {user_prompt}', 'blue')
     print_color(f'LLM: {response}', 'green')
-
-except Exception as e:
-    print_color(f'Error: {e}', 'red')
-    print_color('Omit the test.', 'yellow')

tests/unit_tests/test_optimizer.py

@@ -34,7 +34,7 @@ def user(x):
     else:
         return "Success."
 
-if os.path.exists("OAI_CONFIG_LIST") or os.environ.get("DEFAULT_LITELLM_MODEL"):
+if os.path.exists("OAI_CONFIG_LIST") or os.environ.get("DEFAULT_LITELLM_MODEL") or os.environ.get("OPENAI_API_KEY"):
     # One-step optimization example
     x = node(-1.0, trainable=True)
     optimizer = OptoPrime([x])
@@ -124,7 +124,7 @@ def foobar_text(x):
 GRAPH.clear()
 x = node("negative point one", trainable=True)
 
-if os.path.exists("OAI_CONFIG_LIST") or os.environ.get("DEFAULT_LITELLM_MODEL"):
+if os.path.exists("OAI_CONFIG_LIST") or os.environ.get("DEFAULT_LITELLM_MODEL") or os.environ.get("OPENAI_API_KEY"):
     optimizer = OptoPrime([x])
     output = foobar_text(x)
     feedback = user(output.data)