fix: ensure module is importable before patching in initialize tests

Author: T-DevH

tests/unit/test_document_action_tools.py

@@ -71,6 +71,12 @@ async def test_initialize_success(self):
         mock_nim_client = AsyncMock()
         mock_db_service = AsyncMock()
 
+        # Import the module first to ensure it's available for patching
+        try:
+            import src.api.services.document
+        except ImportError:
+            pass  # Module might not be importable in test environment
+        
         # Patch the module import - need to patch where it's imported from
         with patch("src.api.agents.document.action_tools.get_nim_client", return_value=mock_nim_client), \
              patch("src.api.services.document.get_document_db_service", return_value=mock_db_service), \
@@ -89,9 +95,15 @@ async def test_initialize_without_database_fallback(self):
 
         mock_nim_client = AsyncMock()
 
+        # Import the module first to ensure it's available for patching
+        try:
+            import src.api.services.document
+        except ImportError:
+            pass  # Module might not be importable in test environment
+        
         # Patch the module import - need to patch where it's imported from
         with patch("src.api.agents.document.action_tools.get_nim_client", return_value=mock_nim_client), \
-             patch("src.api.services.document.get_document_db_service", side_effect=ImportError("DB unavailable")), \
+             patch("src.api.services.document.get_document_db_service", side_effect=Exception("DB unavailable")), \
              patch.object(tools, "_load_status_data"):
 
             await tools.initialize()