perf(sdk): improve issue custom fields parsing speed

moneymeets · Jun 20, 2024 · 7454ada · 7454ada
1 parent 6e9a328
commit 7454ada
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/youtrack_sdk/entities.py b/youtrack_sdk/entities.py
@@ -1,4 +1,4 @@
-from typing import Literal, Optional, Sequence
+from typing import Annotated, Literal, Optional, Sequence
 
 from pydantic import AwareDatetime, ConfigDict, Field, StrictFloat, StrictInt, StrictStr
 from pydantic import BaseModel as PydanticBaseModel
@@ -239,7 +239,7 @@ class StateIssueCustomField(IssueCustomField):
     value: Optional[StateBundleElement] = None
 
 
-IssueCustomFieldType = (
+IssueCustomFieldType = Annotated[
     SingleEnumIssueCustomField
     | MultiEnumIssueCustomField
     | SingleBuildIssueCustomField
@@ -256,8 +256,9 @@ class StateIssueCustomField(IssueCustomField):
     | SimpleIssueCustomField
     | DateIssueCustomField
     | PeriodIssueCustomField
-    | TextIssueCustomField
-)
+    | TextIssueCustomField,
+    Field(discriminator="type"),
+]
 
 
 class Project(BaseModel):