Skip to content

Commit 01fceb2

Browse files
Fix asyncio loop issue in combination with enable_input event
1 parent 11294d9 commit 01fceb2

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

nemoguardrails/cli/chat.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@
3030
from nemoguardrails.logging import verbose
3131
from nemoguardrails.logging.verbose import console
3232
from nemoguardrails.streaming import StreamingHandler
33-
from nemoguardrails.utils import new_event_dict, new_uuid
33+
from nemoguardrails.utils import get_or_create_event_loop, new_event_dict, new_uuid
3434

3535
os.environ["TOKENIZERS_PARALLELISM"] = "false"
3636

@@ -662,6 +662,7 @@ def run_chat(
662662
)
663663
elif rails_config.colang_version == "2.x":
664664
rails_app = LLMRails(rails_config, verbose=verbose)
665-
asyncio.run(_run_chat_v2_x(rails_app))
665+
loop = get_or_create_event_loop()
666+
loop.run_until_complete(_run_chat_v2_x(rails_app))
666667
else:
667668
raise Exception(f"Invalid colang version: {rails_config.colang_version}")

0 commit comments

Comments
 (0)