fix: update parse_obj to model_validate (#511)

Author: stephenkilbourn

ingest_api/runtime/src/services.py

@@ -4,7 +4,7 @@
 import src.schemas as schemas
 from boto3.dynamodb import conditions
 from boto3.dynamodb.types import DYNAMODB_CONTEXT
-from pydantic import parse_obj_as
+from pydantic import TypeAdapter
 
 if TYPE_CHECKING:
     from mypy_boto3_dynamodb.service_resource import Table
@@ -25,7 +25,7 @@ def fetch_one(self, username: str, ingestion_id: str):
             Key={"created_by": username, "id": ingestion_id},
         )
         try:
-            return schemas.Ingestion.parse_obj(response["Item"])
+            return schemas.Ingestion.model_validate(response["Item"])
         except KeyError:
             raise NotInDb("Record not found")
 
@@ -38,8 +38,9 @@ def fetch_many(
             **{"Limit": limit} if limit else {},
             **{"ExclusiveStartKey": next} if next else {},
         )
+        list_of_ingestions = TypeAdapter(List[schemas.Ingestion])
         return {
-            "items": parse_obj_as(List[schemas.Ingestion], response["Items"]),
+            "items": list_of_ingestions.validate_python(response["Items"]),
             "next": response.get("LastEvaluatedKey"),
         }
 

ingest_api/runtime/tests/conftest.py

@@ -162,6 +162,6 @@ def example_ingestion(example_stac_item):
         id=example_stac_item["id"],
         created_by="test-user",
         status=schemas.Status.queued,
-        item=Item.parse_obj(example_stac_item),
+        item=Item.model_validate(example_stac_item),
         created_at=datetime.now(),
     )

scripts/run-local-tests.sh

@@ -1,6 +1,14 @@
 #!/bin/bash
 set -e
 
+# =================================================================
+#  Ensure all Python dependencies are installed
+# =================================================================
+echo "--- Installing all development dependencies ---"
+pip install -r ingest_api/runtime/requirements_dev.txt
+echo "--- Dependency installation complete ---"
+# =================================================================
+
 # Lint
 pre-commit run --all-files