Fixing failing tests

tests/local_testing/test_add_update_models.py

@@ -16,6 +16,8 @@ sys.path.insert(
 )  # Adds the parent directory to the system path
 import pytest, logging, asyncio
 import litellm
+import litellm.proxy
+import litellm.proxy.proxy_server
 from litellm.proxy.management_endpoints.model_management_endpoints import (
     add_new_model,
     update_model,

tests/test_proxy_server_non_root.py

@@ -1,6 +1,7 @@
 from unittest.mock import patch
 import pytest
-@pytest.mark.skip(reason="Very Flaky in CI, will debug later")
+import litellm.proxy
+import litellm.proxy.proxy_server
 def test_restructure_ui_html_files_skipped_in_non_root(monkeypatch):
     """
     Test that _restructure_ui_html_files is SKIPPED when:
@@ -36,7 +37,6 @@ def test_restructure_ui_html_files_skipped_in_non_root(monkeypatch):
         # Verify it was NOT called
         mock_restructure.assert_not_called()
 
-@pytest.mark.skip(reason="Very Flaky in CI, will debug later")
 def test_restructure_ui_html_files_NOT_skipped_locally(monkeypatch):
     """
     Test that _restructure_ui_html_files is NOT skipped for local development