fix: revert llamaindex tests

Author: cosmyo

packages/uipath-llamaindex/src/uipath_llamaindex/runtime/_sqlite.py

@@ -57,25 +57,7 @@ async def connect(self) -> None:
     async def close(self) -> None:
         """Close database connection."""
         if self.conn:
-            try:
-                # Commit any pending transactions before closing
-                await self.conn.commit()
-            except Exception:
-                pass  # Best effort
-
-            try:
-                # Force WAL checkpoint to release shared memory files
-                # This is especially important on Windows
-                await self.conn.execute("PRAGMA wal_checkpoint(TRUNCATE)")
-                await self.conn.commit()
-            except Exception:
-                pass  # Best effort
-
-            try:
-                await self.conn.close()
-            except Exception:
-                pass  # Best effort cleanup
-
+            await self.conn.close()
             self.conn = None
             self.is_setup = False
 

packages/uipath-llamaindex/tests/conftest.py

@@ -1,10 +1,4 @@
-import gc
-import os
-import platform
-import shutil
-import sqlite3
 import tempfile
-import time
 from typing import Generator
 
 import pytest
@@ -20,47 +14,5 @@ def runner() -> CliRunner:
 @pytest.fixture
 def temp_dir() -> Generator[str, None, None]:
     """Provide a temporary directory for test files."""
-    tmp_dir = tempfile.mkdtemp()
-    try:
+    with tempfile.TemporaryDirectory() as tmp_dir:
         yield tmp_dir
-    finally:
-        # Force garbage collection to close any open file handles
-        gc.collect()
-
-        # On Windows, SQLite files may remain locked even after closing connections
-        # Try to close any remaining SQLite connections to the state.db file
-        if platform.system() == "Windows":
-            state_db_path = os.path.join(tmp_dir, "__uipath", "state.db")
-            if os.path.exists(state_db_path):
-                # Try to force close SQLite connections
-                try:
-                    # Connect and immediately close to ensure WAL checkpoint
-                    conn = sqlite3.connect(state_db_path, timeout=1.0)
-                    conn.execute("PRAGMA wal_checkpoint(TRUNCATE)")
-                    conn.close()
-                except Exception:
-                    pass  # Best effort
-
-                # Give Windows time to release file handles
-                time.sleep(0.1)
-
-        # Attempt cleanup with retries on Windows
-        max_retries = 3
-        for attempt in range(max_retries):
-            try:
-                shutil.rmtree(tmp_dir)
-                break
-            except (PermissionError, OSError) as e:
-                if attempt < max_retries - 1:
-                    # Wait briefly and try again
-                    time.sleep(0.2)
-                    gc.collect()
-                else:
-                    # On final attempt, log warning but don't fail the test
-                    import warnings
-
-                    warnings.warn(
-                        f"Failed to clean up temporary directory {tmp_dir}: {e}",
-                        ResourceWarning,
-                        stacklevel=2,
-                    )