diff --git a/tests/integration/database/migrations/test_request_log_indexes.py b/tests/integration/database/migrations/test_request_log_indexes.py index bb85eeb6605..10e1cc151ee 100644 --- a/tests/integration/database/migrations/test_request_log_indexes.py +++ b/tests/integration/database/migrations/test_request_log_indexes.py @@ -39,7 +39,7 @@ requires_db: Final = pytest.mark.skipif( ) REPO: Final = Path(__file__).resolve().parents[4] -PACKAGE: Final = REPO / "litellm-proxy-extras" / "litellm_proxy_extras" +PACKAGE: Final = REPO / "litellm" / "proxy-extras" / "litellm_proxy_extras" PARTITION_SCRIPT: Final = REPO / "litellm/scripts" / "db" / "partition_spend_logs.sql" API_KEY_INDEX_MIGRATION: Final = "20260823000000_add_spend_logs_api_key_starttime_index" CALL_ID_INDEX_MIGRATION: Final = "20260831120001_spend_logs_litellm_call_id_index" diff --git a/tests/integration/database/test_managed_file_flat_ids_index.py b/tests/integration/database/test_managed_file_flat_ids_index.py index b297887411f..7c956a81b8a 100644 --- a/tests/integration/database/test_managed_file_flat_ids_index.py +++ b/tests/integration/database/test_managed_file_flat_ids_index.py @@ -19,7 +19,7 @@ from psycopg import sql from pydantic import JsonValue, TypeAdapter REPO_ROOT: Final = Path(__file__).resolve().parents[3] -PRISMA_DIR: Final = REPO_ROOT / "litellm-proxy-extras" / "litellm_proxy_extras" +PRISMA_DIR: Final = REPO_ROOT / "litellm" / "proxy-extras" / "litellm_proxy_extras" GIN_MIGRATION: Final = "20261003000000_add_managed_file_flat_ids_gin_index" INDEX_NAME: Final = "LiteLLM_ManagedFileTable_flat_model_file_ids_idx" SHIPPED_MIGRATIONS: Final = tuple(sorted(path.name for path in (PRISMA_DIR / "migrations").iterdir() if path.is_dir())) @@ -57,8 +57,8 @@ def _leg_python_path() -> str: return os.pathsep.join( ( str(REPO_ROOT), - str(REPO_ROOT / "litellm-proxy-extras"), - str(REPO_ROOT / "enterprise"), + str(REPO_ROOT / "litellm" / "proxy-extras"), + str(REPO_ROOT / "litellm" / "enterprise"), os.environ.get("PYTHONPATH", ""), ) ) diff --git a/tests/integration/database/test_proxy_boot_without_general_settings.py b/tests/integration/database/test_proxy_boot_without_general_settings.py index 73254f3862f..2aa1b880083 100644 --- a/tests/integration/database/test_proxy_boot_without_general_settings.py +++ b/tests/integration/database/test_proxy_boot_without_general_settings.py @@ -12,7 +12,7 @@ from integration._support.process import DB_PUSH, LEGACY_MIGRATE_DEPLOY, MIGRATE pytestmark: Final = pytest.mark.timeout(300) REPO_ROOT: Final = Path(__file__).resolve().parents[3] -MIGRATIONS_DIR: Final = REPO_ROOT / "litellm-proxy-extras" / "litellm_proxy_extras" / "migrations" +MIGRATIONS_DIR: Final = REPO_ROOT / "litellm" / "proxy-extras" / "litellm_proxy_extras" / "migrations" SHIPPED_MIGRATIONS: Final = frozenset(path.name for path in MIGRATIONS_DIR.iterdir() if path.is_dir()) CANNED_REPLY: Final = "Hello! This is a mock response from the fake OpenAI endpoint." PROVIDER_KEY: Final = "integration-provider-key" diff --git a/tests/integration/database/test_proxy_database_lifespan.py b/tests/integration/database/test_proxy_database_lifespan.py index b80ff8acb30..c79c0ba4b35 100644 --- a/tests/integration/database/test_proxy_database_lifespan.py +++ b/tests/integration/database/test_proxy_database_lifespan.py @@ -121,7 +121,7 @@ def test_failed_startup_then_retry_releases_database_resources(tmp_path: Path, m "LITELLM_LOCAL_MODEL_COST_MAP": "True", "LITELLM_TELEMETRY_ENABLED": "false", "PRISMA_HEALTH_WATCHDOG_ENABLED": "true", - "PYTHONPATH": os.pathsep.join((str(ROOT), str(ROOT / "enterprise"))), + "PYTHONPATH": os.pathsep.join((str(ROOT), str(ROOT / "litellm" / "enterprise"))), **({"DATABASE_URL_READ_REPLICA": database_url} if replica else {}), } schema: Final = subprocess.run( diff --git a/tests/integration/database/test_request_log_indexes_at_boot.py b/tests/integration/database/test_request_log_indexes_at_boot.py index 3337393a52a..afcd1a34b7d 100644 --- a/tests/integration/database/test_request_log_indexes_at_boot.py +++ b/tests/integration/database/test_request_log_indexes_at_boot.py @@ -20,7 +20,7 @@ from psycopg.rows import class_row pytestmark: Final = pytest.mark.timeout(240) REPO_ROOT: Final = Path(__file__).resolve().parents[3] -PRISMA_DIR: Final = REPO_ROOT / "litellm-proxy-extras" / "litellm_proxy_extras" +PRISMA_DIR: Final = REPO_ROOT / "litellm" / "proxy-extras" / "litellm_proxy_extras" PARTITION_SCRIPT: Final = REPO_ROOT / "litellm/scripts" / "db" / "partition_spend_logs.sql" SHIPPED_MIGRATIONS: Final = tuple(sorted(path.name for path in (PRISMA_DIR / "migrations").iterdir() if path.is_dir())) API_KEY_INDEX_MIGRATION: Final = "20260823000000_add_spend_logs_api_key_starttime_index" diff --git a/tests/integration/database/test_scheduled_job_runs.py b/tests/integration/database/test_scheduled_job_runs.py index 913111600bf..a0d71eef9b3 100644 --- a/tests/integration/database/test_scheduled_job_runs.py +++ b/tests/integration/database/test_scheduled_job_runs.py @@ -29,7 +29,7 @@ _MIGRATION: Final = next( Path(__file__) .resolve() .parents[3] - .joinpath("litellm-proxy-extras/litellm_proxy_extras/migrations") + .joinpath("litellm/proxy-extras/litellm_proxy_extras/migrations") .glob("*_scheduled_job_runs/migration.sql") ) _PODS: Final = 4 diff --git a/tests/integration/database/test_v1_migration_error_logging.py b/tests/integration/database/test_v1_migration_error_logging.py index 3ca2179d983..1da45276e8e 100644 --- a/tests/integration/database/test_v1_migration_error_logging.py +++ b/tests/integration/database/test_v1_migration_error_logging.py @@ -29,7 +29,7 @@ from tests.integration._support.ports import allocate_port from tests.integration._support.process import stop_process_group REPO_ROOT: Final = Path(__file__).resolve().parents[3] -MIGRATIONS_DIR: Final = REPO_ROOT / "litellm-proxy-extras" / "litellm_proxy_extras" / "migrations" +MIGRATIONS_DIR: Final = REPO_ROOT / "litellm" / "proxy-extras" / "litellm_proxy_extras" / "migrations" MIGRATION_NAME: Final = "20260921190000_agent_identity" pytestmark: Final = pytest.mark.timeout(300) PASSWORD: Final = "wr ong'pw9" diff --git a/tests/integration/security/test_hide_secrets_tool_traffic.py b/tests/integration/security/test_hide_secrets_tool_traffic.py index 60b49520b1f..38044d18a49 100644 --- a/tests/integration/security/test_hide_secrets_tool_traffic.py +++ b/tests/integration/security/test_hide_secrets_tool_traffic.py @@ -161,7 +161,7 @@ def tool_rig( "ANTHROPIC_API_KEY": "synthetic-provider-key", "LITELLM_LOCAL_ANTHROPIC_BETA_HEADERS": "True", "PYTHONPATH": os.pathsep.join( - (str(product_root / "enterprise"), str(product_root), os.environ.get("PYTHONPATH", "")) + (str(product_root / "litellm" / "enterprise"), str(product_root), os.environ.get("PYTHONPATH", "")) ), **(environment or {}), }