Adding annotated from main.

stac-utils · Aug 20, 2024 · 0c5de64 · 0c5de64
1 parent 14d18f9
commit 0c5de64
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/stac_fastapi/types/stac_fastapi/types/transaction.py b/stac_fastapi/types/stac_fastapi/types/transaction.py
@@ -6,6 +6,7 @@
 from fastapi import Body
 from pydantic import BaseModel
 from stac_pydantic import Collection, Item, ItemCollection
+from typing_extensions import Annotated
 
 from stac_fastapi.api.models import CollectionUri, ItemUri
 
@@ -14,14 +15,14 @@
 class PostItem(CollectionUri):
     """Create Item."""
 
-    item: Union[Item, ItemCollection] = attr.ib(default=Body(None))
+    item: Annotated[Union[Item, ItemCollection], Body()] = attr.ib(default=None)
 
 
 @attr.s
 class PutPatchItem(ItemUri):
     """Update Item."""
 
-    item: Item = attr.ib(default=Body(None))
+    item: Annotated[Item, Body()] = attr.ib(default=None)
 
 
 @attr.s
@@ -62,4 +63,4 @@ def __init__(self, *args, **kwargs):
 class PutPatchCollection(CollectionUri):
     """Update Collection."""
 
-    collection: Collection = attr.ib(default=Body(None))
+    collection: Annotated[Collection, Body()] = attr.ib(default=None)