restrict Boltz sequence length

ginkgo_ai_client/queries.py

@@ -277,9 +277,9 @@ class PromoterActivityQuery(QueryBase):
         The name of the query. It will appear in the API response and can be used to
         handle exceptions.
     inference_framework: Literal["promoter-0"] = "promoter-0"
-        The inference framework to use for the inference. Currently only supports 
+        The inference framework to use for the inference. Currently only supports
     borzoi_model: Literal["human-fold0"] = "human-fold0"
-        The model to use for the inference. Currently only supports the trained 
+        The model to use for the inference. Currently only supports the trained
         model of "human-fold0".
     Returns
     -------
@@ -511,6 +511,11 @@ class _Protein(pydantic.BaseModel):
 
     @pydantic.validator("sequence")
     def validate_sequence(cls, sequence):
+        if len(sequence) > 1000:
+            raise ValueError(
+                f"We currently only accept sequences of length 1000 or less for Boltz "
+                f"structure prediction (length: {len(sequence)})"
+            )
         sequence = sequence.upper()
         invalid_chars = [c for c in sequence if c not in "LAGVSERTIDPKQNFYMHWCXBUZO"]
         if len(invalid_chars) > 0:

test/test_query_creation.py

@@ -62,3 +62,11 @@ def test_boltz_structure_prediction_query_from_protein_sequence():
     query = BoltzStructurePredictionQuery.from_protein_sequence(sequence="MLLKP")
     sequences = query.model_dump(exclude_none=True)["sequences"]
     assert sequences == [{"protein": {"id": "A", "sequence": "MLLKP"}}]
+
+
+def test_boltz_structure_prediction_query_fails_on_sequence_too_long():
+    expected_error_message = re.escape(
+        "We currently only accept sequences of length 1000 or less"
+    )
+    with pytest.raises(ValueError, match=expected_error_message):
+        BoltzStructurePredictionQuery.from_protein_sequence(sequence=1100 * "A")