Add timeouts for engine deletion

RelationalAI · Aug 1, 2023 · 858fb4f · 858fb4f
1 parent 7c97760
commit 858fb4f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/testpool.jl b/src/testpool.jl
@@ -78,7 +78,7 @@ function replace_engine(name::String)
     end
     # If the engine could not be deleted, notify and continue
     try
-        delete_engine(get_context(), name)
+        delete_engine(get_context(), name, readtimeout=30)
     catch
         @warn("Could not delete engine: ", name)
         name = TEST_ENGINE_POOL.generator(TEST_ENGINE_POOL.next_id)
@@ -174,7 +174,7 @@ function resize_test_engine_pool(size::Int64, generator::Option{Function}=nothin
         Threads.@sync for engine in engines_to_delete
             @info("Deleting engine", engine)
             @async try
-                delete_engine(get_context(), engine)
+                delete_engine(get_context(), engine, readtimeout=30)
             catch e
                 # The engine may not exist if it hasn't been used yet
                 # For other errors, we just report the error and delete what we can