Handle optional import

guidance-ai · May 3, 2024 · 4281d7f · 4281d7f
1 parent cdf679c
commit 4281d7f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/guidance/models/_azureai_studio.py b/guidance/models/_azureai_studio.py
@@ -3,13 +3,16 @@
 import urllib.parse
 
 import diskcache as dc
-import openai
 import platformdirs
 import requests
 
 from ._model import Chat
 from ._grammarless import GrammarlessEngine, Grammarless
 
+try:
+    import openai
+except ImportError:
+    openai = None
 
 class AzureAIStudioChatEngine(GrammarlessEngine):
     def __init__(