File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change 30
30
from nemoguardrails .logging import verbose
31
31
from nemoguardrails .logging .verbose import console
32
32
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
34
34
35
35
os .environ ["TOKENIZERS_PARALLELISM" ] = "false"
36
36
@@ -662,6 +662,7 @@ def run_chat(
662
662
)
663
663
elif rails_config .colang_version == "2.x" :
664
664
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 ))
666
667
else :
667
668
raise Exception (f"Invalid colang version: { rails_config .colang_version } " )
You can’t perform that action at this time.
0 commit comments