From 38012d04717c49b388bc9f2185ebfe78ce4a4194 Mon Sep 17 00:00:00 2001 From: michaelfeil Date: Sat, 22 Jun 2024 20:41:47 -0700 Subject: [PATCH] fix: sync engine fails on startup --- libs/infinity_emb/infinity_emb/sync_engine.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/libs/infinity_emb/infinity_emb/sync_engine.py b/libs/infinity_emb/infinity_emb/sync_engine.py index cf49405d..59c430cb 100644 --- a/libs/infinity_emb/infinity_emb/sync_engine.py +++ b/libs/infinity_emb/infinity_emb/sync_engine.py @@ -101,8 +101,13 @@ def async_run( class SyncEngineArray(AsyncLifeMixin): def __init__(self, _engine_args_array: list[EngineArgs]): super().__init__() - self.async_engine_array = AsyncEngineArray.from_args(_engine_args_array) - self.async_run(self.async_engine_array.astart).result() + try: + self.async_engine_array = AsyncEngineArray.from_args(_engine_args_array) + self.async_run(self.async_engine_array.astart).result() + except Exception as e: + self.async_close_loop() + raise e + @classmethod def from_args(cls, engine_args_array: list[EngineArgs]) -> "SyncEngineArray":