chore: wrap openai client initialization

Signed-off-by: Francesco Fuggitti <francesco.fuggitti@gmail.com>
IBM · Feb 15, 2024 · ae0c243 · ae0c243
1 parent d404007
commit ae0c243
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/nl2ltl/engines/gpt/core.py b/nl2ltl/engines/gpt/core.py
@@ -18,7 +18,11 @@
 from nl2ltl.engines.gpt.output import GPTOutput, parse_gpt_output, parse_gpt_result
 from nl2ltl.filters.base import Filter
 
-client = OpenAI()
+try:
+    client = OpenAI()
+except Exception:
+    client = None
+
 engine_root = ENGINE_ROOT
 DATA_DIR = engine_root / "data"
 PROMPT_PATH = engine_root / DATA_DIR / "prompt.json"