fix(curation): fix method signature of upserting curation sets

Author: tharropoulos

src/typesense/curation_set.py

@@ -9,11 +9,12 @@
 
 from typesense.api_call import ApiCall
 from typesense.types.curation_set import (
-    CurationSetSchema,
+    CurationItemDeleteSchema,
+    CurationItemSchema,
     CurationSetDeleteSchema,
     CurationSetListItemResponseSchema,
-    CurationItemSchema,
-    CurationItemDeleteSchema,
+    CurationSetSchema,
+    CurationSetUpsertSchema,
 )
 
 
@@ -43,6 +44,17 @@ def delete(self) -> CurationSetDeleteSchema:
         )
         return response
 
+    def upsert(
+        self,
+        payload: CurationSetUpsertSchema,
+    ) -> CurationSetSchema:
+        response: CurationSetSchema = self.api_call.put(
+            "/".join([self._endpoint_path]),
+            body=payload,
+            entity_type=CurationSetSchema,
+        )
+        return response
+
     # Items sub-resource
     @property
     def _items_path(self) -> str:
@@ -92,5 +104,3 @@ def delete_item(self, item_id: str) -> CurationItemDeleteSchema:
             entity_type=CurationItemDeleteSchema,
         )
         return response
-
-

src/typesense/curation_sets.py

@@ -10,9 +10,7 @@
 from typesense.api_call import ApiCall
 from typesense.curation_set import CurationSet
 from typesense.types.curation_set import (
-    CurationSetSchema,
     CurationSetsListResponseSchema,
-    CurationSetUpsertSchema,
 )
 
 
@@ -34,15 +32,3 @@ def __getitem__(self, curation_set_name: str) -> CurationSet:
         from typesense.curation_set import CurationSet as PerSet
 
         return PerSet(self.api_call, curation_set_name)
-
-    def upsert(
-        self,
-        curation_set_name: str,
-        payload: CurationSetUpsertSchema,
-    ) -> CurationSetSchema:
-        response: CurationSetSchema = self.api_call.put(
-            "/".join([CurationSets.resource_path, curation_set_name]),
-            body=payload,
-            entity_type=CurationSetSchema,
-        )
-        return response

tests/curation_sets_test.py

@@ -96,7 +96,7 @@ def test_upsert(fake_curation_sets: CurationSets) -> None:
                 }
             ]
         }
-        response = fake_curation_sets.upsert("products", payload)
+        response = fake_curation_sets["products"].upsert(payload)
 
         assert response == json_response
         assert mock.call_count == 1
@@ -111,8 +111,7 @@ def test_actual_upsert(
     delete_all_curation_sets: None,
 ) -> None:
     """Test that the CurationSets object can upsert a curation set on Typesense Server."""
-    response = actual_curation_sets.upsert(
-        "products",
+    response = actual_curation_sets["products"].upsert(
         {
             "items": [
                 {