Retry cosmos put (#119)

Author: Tom Augspurger

pctasks/core/pctasks/core/cosmos/container.py

@@ -248,14 +248,16 @@ def put(self, model: T) -> None:
             )
         else:
             # Otherwise upsert the item
-            self._container_client.upsert_item(
-                item,
-                pre_trigger_include=self.get_trigger(
-                    ContainerOperation.PUT, TriggerType.PRE
-                ),
-                post_trigger_include=self.get_trigger(
-                    ContainerOperation.PUT, TriggerType.POST
-                ),
+            with_backoff(
+                lambda: self._container_client.upsert_item(
+                    item,
+                    pre_trigger_include=self.get_trigger(
+                        ContainerOperation.PUT, TriggerType.PRE
+                    ),
+                    post_trigger_include=self.get_trigger(
+                        ContainerOperation.PUT, TriggerType.POST
+                    ),
+                )
             )
 
     def bulk_put(self, models: Iterable[T]) -> None: