handle name changes in helicone

StampyAI · Aug 19, 2023 · 04d0cb2 · 04d0cb2
1 parent ba83761
commit 04d0cb2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/api/openai.py b/api/openai.py
@@ -16,7 +16,10 @@
 from utilities import utilities, Utilities
 from utilities import discordutils
 if use_helicone:
-    from helicone import openai
+    try:
+        from helicone import openai
+    except ImportError:
+        from helicone import openai_proxy as openai
 else:
     import openai
 import discord

diff --git a/modules/chatgpt.py b/modules/chatgpt.py
@@ -21,7 +21,10 @@
 from servicemodules.serviceConstants import service_italics_marks, default_italics_mark
 
 if use_helicone:
-    from helicone import openai
+    try:
+        from helicone import openai
+    except ImportError:
+        from helicone import openai_proxy as openai
 else:
     import openai