fix lint

Author: dishaprakash

tests/test_async_chatmessagehistory.py

@@ -38,9 +38,11 @@
 # Helper to bridge the Main Test Loop and the Engine Background Loop
 async def run_on_background(engine: PostgresEngine, coro: Coroutine) -> Any:
     """Runs a coroutine on the engine's background loop."""
-    return await asyncio.wrap_future(
-        asyncio.run_coroutine_threadsafe(coro, engine._loop)
-    )
+    if engine._loop:
+        return await asyncio.wrap_future(
+            asyncio.run_coroutine_threadsafe(coro, engine._loop)
+        )
+    return await coro
 
 
 async def aexecute(engine: PostgresEngine, query: str) -> None:

tests/test_async_checkpoint.py

@@ -111,9 +111,11 @@ def _AnyIdToolMessage(**kwargs: Any) -> ToolMessage:
 # Helper to bridge the Main Test Loop and the Engine Background Loop
 async def run_on_background(engine: PostgresEngine, coro: Coroutine) -> Any:
     """Runs a coroutine on the engine's background loop."""
-    return await asyncio.wrap_future(
-        asyncio.run_coroutine_threadsafe(coro, engine._loop)
-    )
+    if engine._loop:
+        return await asyncio.wrap_future(
+            asyncio.run_coroutine_threadsafe(coro, engine._loop)
+        )
+    return await coro
 
 
 async def aexecute(engine: PostgresEngine, query: str) -> None:

tests/test_async_loader.py

@@ -39,9 +39,11 @@
 # Helper to bridge the Main Test Loop and the Engine Background Loop
 async def run_on_background(engine: PostgresEngine, coro: Coroutine) -> Any:
     """Runs a coroutine on the engine's background loop."""
-    return await asyncio.wrap_future(
-        asyncio.run_coroutine_threadsafe(coro, engine._loop)
-    )
+    if engine._loop:
+        return await asyncio.wrap_future(
+            asyncio.run_coroutine_threadsafe(coro, engine._loop)
+        )
+    return await coro
 
 
 async def aexecute(engine: PostgresEngine, query: str) -> None:

tests/test_async_vectorstore.py

@@ -54,9 +54,11 @@ def get_env_var(key: str, desc: str) -> str:
 # Helper to bridge the Main Test Loop and the Engine Background Loop
 async def run_on_background(engine: PostgresEngine, coro: Coroutine) -> Any:
     """Runs a coroutine on the engine's background loop."""
-    return await asyncio.wrap_future(
-        asyncio.run_coroutine_threadsafe(coro, engine._loop)
-    )
+    if engine._loop:
+        return await asyncio.wrap_future(
+            asyncio.run_coroutine_threadsafe(coro, engine._loop)
+        )
+    return await coro
 
 
 async def aexecute(engine: PostgresEngine, query: str) -> None:

tests/test_async_vectorstore_from_methods.py

@@ -55,9 +55,11 @@ def get_env_var(key: str, desc: str) -> str:
 # Helper to bridge the Main Test Loop and the Engine Background Loop
 async def run_on_background(engine: PostgresEngine, coro: Coroutine) -> Any:
     """Runs a coroutine on the engine's background loop."""
-    return await asyncio.wrap_future(
-        asyncio.run_coroutine_threadsafe(coro, engine._loop)
-    )
+    if engine._loop:
+        return await asyncio.wrap_future(
+            asyncio.run_coroutine_threadsafe(coro, engine._loop)
+        )
+    return await coro
 
 
 async def aexecute(engine: PostgresEngine, query: str) -> None:

tests/test_async_vectorstore_index.py

@@ -65,9 +65,11 @@ def get_env_var(key: str, desc: str) -> str:
 # Helper to bridge the Main Test Loop and the Engine Background Loop
 async def run_on_background(engine: PostgresEngine, coro: Coroutine) -> Any:
     """Runs a coroutine on the engine's background loop."""
-    return await asyncio.wrap_future(
-        asyncio.run_coroutine_threadsafe(coro, engine._loop)
-    )
+    if engine._loop:
+        return await asyncio.wrap_future(
+            asyncio.run_coroutine_threadsafe(coro, engine._loop)
+        )
+    return await coro
 
 
 async def aexecute(engine: PostgresEngine, query: str) -> None: