fix streamlit issue for good?????

approximatelabs · Jan 18, 2024 · b7e62cf · b7e62cf
1 parent 3efdca9
commit b7e62cf
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lambdaprompt/__init__.py b/lambdaprompt/__init__.py
@@ -1,11 +1,16 @@
 import nest_asyncio
+import asyncio
 
 from .gpt3 import AsyncGPT3Chat, AsyncGPT3Prompt, GPT3Chat, GPT3Prompt
 from .prompt import (AsyncPrompt, Prompt, prompt, Completion, AsyncCompletion, Chat, AsyncChat, register_call_callback,
                      register_creation_callback)
 from . import backends
 
-nest_asyncio.apply()
+try:
+    asyncio.get_running_loop()
+    nest_asyncio.apply()
+except RuntimeError:
+    pass
 
 __all__ = [
     "AsyncGPT3Prompt",