chore(config): raise max_code_bytes for zip submissions

Author: echobt

src/prism_challenge/config.py

@@ -36,7 +36,7 @@ class PrismSettings(ChallengeSettings):
     allow_insecure_signatures: bool = False
     signature_ttl_seconds: int = 300
     epoch_seconds: int = 21_600
-    max_code_bytes: int = 200_000
+    max_code_bytes: int = 7_500_000
     max_parameters: int = 150_000_000
     max_layers: int = 96
     max_sequence_length: int = 512

tests/test_api.py

@@ -5,11 +5,52 @@
 import json
 import zipfile
 
+import pytest
 from conftest import VALID_CODE, signed_headers
+from fastapi.testclient import TestClient
 
+from prism_challenge.app import create_app
+from prism_challenge.config import PrismSettings
 from prism_challenge.sdk.executors.docker import DockerRunResult
 
 
+@pytest.fixture
+def small_cap_client(tmp_path) -> TestClient:
+    settings = PrismSettings(
+        database_url=f"sqlite+aiosqlite:///{tmp_path / 'prism.sqlite3'}",
+        shared_token="secret",
+        allow_insecure_signatures=True,
+        fineweb_sample_count=4,
+        max_code_bytes=2_000,
+    )
+    with TestClient(create_app(settings)) as test_client:
+        yield test_client
+
+
+def _post_code(client: TestClient, code: str):
+    payload = {"code": code, "filename": "model.py"}
+    body = json.dumps(payload, separators=(",", ":")).encode()
+    return client.post(
+        "/v1/submissions",
+        content=body,
+        headers={**signed_headers("secret", body), "Content-Type": "application/json"},
+    )
+
+
+def test_size_check_accepts_just_under_cap(small_cap_client):
+    cap = small_cap_client.app.state.settings.max_code_bytes
+    response = _post_code(small_cap_client, "A" * (cap - 1))
+    assert response.status_code == 200, response.text
+
+
+def test_size_check_rejects_just_over_cap(small_cap_client):
+    cap = small_cap_client.app.state.settings.max_code_bytes
+    response = _post_code(small_cap_client, "A" * (cap + 1))
+    assert response.status_code == 413, response.text
+    assert response.json()["detail"] == "submission too large"
+
+
+
 def test_health_version_and_internal_auth(client):
     assert client.get("/health").json()["slug"] == "prism"
     assert "nas" in client.get("/version").json()["capabilities"]

tests/test_config.py

@@ -73,6 +73,17 @@ def test_internal_token_requires_secret() -> None:
 
 
 
+def test_max_code_bytes_holds_five_mib_zip_base64() -> None:
+    # 5 MiB raw zip -> base64 length 6,990,508; cap must comfortably exceed it.
+    raw_five_mib = 5 * 1024 * 1024
+    base64_len = 4 * ((raw_five_mib + 2) // 3)
+
+    settings = PrismSettings()
+
+    assert settings.max_code_bytes == 7_500_000
+    assert settings.max_code_bytes > base64_len
+
+
 def test_example_config_parses_with_nas_defaults() -> None:
     payload = yaml.safe_load(Path("config.example.yaml").read_text(encoding="utf-8"))