From 217e8d7d448212d69612586ccfc822b805edec4a Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 29 Mar 2025 08:23:18 -0700 Subject: [PATCH] test: make script to run clearer --- tests/proxy_unit_tests/test_db_schema_migration.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/proxy_unit_tests/test_db_schema_migration.py b/tests/proxy_unit_tests/test_db_schema_migration.py index 47125e86a7..089c50ebf8 100644 --- a/tests/proxy_unit_tests/test_db_schema_migration.py +++ b/tests/proxy_unit_tests/test_db_schema_migration.py @@ -39,7 +39,9 @@ def test_aaaasschema_migration_check(schema_setup, monkeypatch): if not temp_migrations_dir.exists() or not any(temp_migrations_dir.iterdir()): print("No existing migrations found - first migration needed") - pytest.fail("No existing migrations found - first migration needed") + pytest.fail( + "No existing migrations found - first migration needed. Run `litellm/ci_cd/baseline_db.py` to create new migration -E.g. `python litellm/ci_cd/baseline_db_migration.py`." + ) # Apply all existing migrations subprocess.run( @@ -72,7 +74,7 @@ def test_aaaasschema_migration_check(schema_setup, monkeypatch): print("Schema differences:") print(diff_result.stdout) pytest.fail( - "Schema changes detected - new migration required. Run 'prisma migrate dev' to create new migration." + "Schema changes detected - new migration required. Run `litellm/ci_cd/run_migration.py` to create new migration -E.g. `python litellm/ci_cd/run_migration.py `." ) else: print("No schema changes detected. Migration not needed.")