fix(curation_set): add discriminated union types for curation sets

Author: tharropoulos

src/typesense/types/curation_set.py

@@ -25,18 +25,47 @@ class CurationExcludeSchema(typing.TypedDict):
     id: str
 
 
-class CurationRuleSchema(typing.TypedDict, total=False):
+class CurationRuleTagsSchema(typing.TypedDict):
     """
-    Schema representing rule conditions for a curation item.
+    Schema for a curation rule using tags.
     """
 
-    query: typing.NotRequired[str]
-    match: typing.NotRequired[typing.Literal["exact", "contains"]]
-    filter_by: typing.NotRequired[str]
-    tags: typing.NotRequired[typing.List[str]]
+    tags: typing.List[str]
+
+
+class CurationRuleQuerySchema(typing.TypedDict):
+    """
+    Schema for a curation rule using query and match.
+    """
+
+    query: str
+    match: typing.Literal["exact", "contains"]
 
 
-class CurationItemSchema(typing.TypedDict, total=False):
+class CurationRuleFilterBySchema(typing.TypedDict):
+    """
+    Schema for a curation rule using filter_by.
+    """
+
+    filter_by: str
+
+
+CurationRuleSchema = typing.Union[
+    CurationRuleTagsSchema,
+    CurationRuleQuerySchema,
+    CurationRuleFilterBySchema,
+]
+"""
+Schema representing rule conditions for a curation item.
+
+A curation rule must be exactly one of:
+- A tags-based rule: `{ tags: string[] }`
+- A query-based rule: `{ query: string; match: "exact" | "contains" }`
+- A filter_by-based rule: `{ filter_by: string }`
+"""
+
+
+class CurationItemSchema(typing.TypedDict):
     """
     Schema for a single curation item (aka CurationObject in the API).
     """
@@ -51,7 +80,9 @@ class CurationItemSchema(typing.TypedDict, total=False):
     remove_matched_tokens: typing.NotRequired[bool]
     filter_curated_hits: typing.NotRequired[bool]
     stop_processing: typing.NotRequired[bool]
-    metadata: typing.Dict[str, typing.Any]
+    effective_from_ts: typing.NotRequired[int]
+    effective_to_ts: typing.NotRequired[int]
+    metadata: typing.NotRequired[typing.Dict[str, typing.Any]]
 
 
 class CurationSetUpsertSchema(typing.TypedDict):
@@ -62,12 +93,12 @@ class CurationSetUpsertSchema(typing.TypedDict):
     items: typing.List[CurationItemSchema]
 
 
-class CurationSetSchema(CurationSetUpsertSchema):
+class CurationSetSchema(CurationSetUpsertSchema, total=False):
     """
     Response schema for a curation set.
     """
 
-    name: str
+    name: typing.NotRequired[str]
 
 
 class CurationSetsListEntrySchema(typing.TypedDict):