🐛 Fix: not supported discriminators and repo check run status (#149)

yanyongyu · Oct 9, 2024 · 7f9a5c2 · 7f9a5c2
1 parent 5558799
commit 7f9a5c2
Show file tree

Hide file tree

Showing 17 changed files with 104 additions and 108 deletions.
diff --git a/githubkit/versions/ghec_v2022_11_28/models/group_0051.py b/githubkit/versions/ghec_v2022_11_28/models/group_0051.py
@@ -9,7 +9,7 @@
 
 from __future__ import annotations
 
-from typing import Union
+from typing import Union, Literal
 from datetime import date, datetime
 
 from pydantic import Field
@@ -55,6 +55,10 @@ class CopilotSeatDetails(GitHubModel):
         default=UNSET,
         description="Timestamp of when the assignee's GitHub Copilot access was last updated, in ISO 8601 format.",
     )
+    plan_type: Missing[Literal["business", "enterprise", "unknown"]] = Field(
+        default=UNSET,
+        description="The Copilot plan of the organization, or the parent enterprise, when applicable.",
+    )
 
 
 class EnterpriseTeam(GitHubModel):

diff --git a/githubkit/versions/ghec_v2022_11_28/models/group_0098.py b/githubkit/versions/ghec_v2022_11_28/models/group_0098.py
@@ -49,6 +49,10 @@ class CopilotOrganizationDetails(ExtraGitHubModel):
     seat_management_setting: Literal[
         "assign_all", "assign_selected", "disabled", "unconfigured"
     ] = Field(description="The mode of assigning new seats.")
+    plan_type: Missing[Literal["business", "enterprise", "unknown"]] = Field(
+        default=UNSET,
+        description="The Copilot plan of the organization, or the parent enterprise, when applicable.",
+    )
 
 
 class CopilotSeatBreakdown(GitHubModel):

diff --git a/githubkit/versions/ghec_v2022_11_28/models/group_1010.py b/githubkit/versions/ghec_v2022_11_28/models/group_1010.py
@@ -38,7 +38,9 @@ class ReposOwnerRepoCheckRunsPostBodyOneof1(ExtraGitHubModel):
     external_id: Missing[str] = Field(
         default=UNSET, description="A reference for the run on the integrator's system."
     )
-    status: Missing[Literal["queued", "in_progress"]] = Field(default=UNSET)
+    status: Missing[
+        Literal["queued", "in_progress", "waiting", "requested", "pending"]
+    ] = Field(default=UNSET)
     started_at: Missing[datetime] = Field(
         default=UNSET,
         description="The time that the check run began. This is a timestamp in [ISO 8601](https://en.wikipedia.org/wiki/ISO_8601) format: `YYYY-MM-DDTHH:MM:SSZ`.",

diff --git a/githubkit/versions/ghec_v2022_11_28/rest/checks.py b/githubkit/versions/ghec_v2022_11_28/rest/checks.py
@@ -10,10 +10,9 @@
 from __future__ import annotations
 
 from weakref import ref
-from typing_extensions import Annotated
 from typing import TYPE_CHECKING, Dict, Literal, Optional, overload
 
-from pydantic import Field, BaseModel
+from pydantic import BaseModel
 
 from githubkit.typing import Missing, UnsetType
 from githubkit.utils import UNSET, exclude_unset
@@ -124,7 +123,9 @@ def create(
         head_sha: str,
         details_url: Missing[str] = UNSET,
         external_id: Missing[str] = UNSET,
-        status: Missing[Literal["queued", "in_progress"]] = UNSET,
+        status: Missing[
+            Literal["queued", "in_progress", "waiting", "requested", "pending"]
+        ] = UNSET,
         started_at: Missing[datetime] = UNSET,
         conclusion: Missing[
             Literal[
@@ -178,12 +179,9 @@ def create(
 
         json = kwargs if data is UNSET else data
         json = type_validate_python(
-            Annotated[
-                Union[
-                    ReposOwnerRepoCheckRunsPostBodyOneof0,
-                    ReposOwnerRepoCheckRunsPostBodyOneof1,
-                ],
-                Field(discriminator="status"),
+            Union[
+                ReposOwnerRepoCheckRunsPostBodyOneof0,
+                ReposOwnerRepoCheckRunsPostBodyOneof1,
             ],
             json,
         )
@@ -253,7 +251,9 @@ async def async_create(
         head_sha: str,
         details_url: Missing[str] = UNSET,
         external_id: Missing[str] = UNSET,
-        status: Missing[Literal["queued", "in_progress"]] = UNSET,
+        status: Missing[
+            Literal["queued", "in_progress", "waiting", "requested", "pending"]
+        ] = UNSET,
         started_at: Missing[datetime] = UNSET,
         conclusion: Missing[
             Literal[
@@ -307,12 +307,9 @@ async def async_create(
 
         json = kwargs if data is UNSET else data
         json = type_validate_python(
-            Annotated[
-                Union[
-                    ReposOwnerRepoCheckRunsPostBodyOneof0,
-                    ReposOwnerRepoCheckRunsPostBodyOneof1,
-                ],
-                Field(discriminator="status"),
+            Union[
+                ReposOwnerRepoCheckRunsPostBodyOneof0,
+                ReposOwnerRepoCheckRunsPostBodyOneof1,
             ],
             json,
         )

diff --git a/githubkit/versions/ghec_v2022_11_28/rest/repos.py b/githubkit/versions/ghec_v2022_11_28/rest/repos.py
@@ -10,10 +10,9 @@
 from __future__ import annotations
 
 from weakref import ref
-from typing_extensions import Annotated
 from typing import TYPE_CHECKING, Dict, Literal, Optional, overload
 
-from pydantic import Field, BaseModel
+from pydantic import BaseModel
 
 from githubkit.typing import Missing, UnsetType
 from githubkit.utils import UNSET, exclude_unset
@@ -7142,14 +7141,8 @@ def get_content(
         *,
         headers: Optional[Dict[str, str]] = None,
     ) -> Response[
-        Annotated[
-            Union[
-                List[ContentDirectoryItems],
-                ContentFile,
-                ContentSymlink,
-                ContentSubmodule,
-            ],
-            Field(discriminator="type"),
+        Union[
+            List[ContentDirectoryItems], ContentFile, ContentSymlink, ContentSubmodule
         ]
     ]:
         """See also: https://docs.github.com/enterprise-cloud@latest//rest/repos/contents#get-repository-content"""
@@ -7177,14 +7170,11 @@ def get_content(
             url,
             params=exclude_unset(params),
             headers=exclude_unset(headers),
-            response_model=Annotated[
-                Union[
-                    List[ContentDirectoryItems],
-                    ContentFile,
-                    ContentSymlink,
-                    ContentSubmodule,
-                ],
-                Field(discriminator="type"),
+            response_model=Union[
+                List[ContentDirectoryItems],
+                ContentFile,
+                ContentSymlink,
+                ContentSubmodule,
             ],
             error_models={
                 "404": BasicError,
@@ -7201,14 +7191,8 @@ async def async_get_content(
         *,
         headers: Optional[Dict[str, str]] = None,
     ) -> Response[
-        Annotated[
-            Union[
-                List[ContentDirectoryItems],
-                ContentFile,
-                ContentSymlink,
-                ContentSubmodule,
-            ],
-            Field(discriminator="type"),
+        Union[
+            List[ContentDirectoryItems], ContentFile, ContentSymlink, ContentSubmodule
         ]
     ]:
         """See also: https://docs.github.com/enterprise-cloud@latest//rest/repos/contents#get-repository-content"""
@@ -7236,14 +7220,11 @@ async def async_get_content(
             url,
             params=exclude_unset(params),
             headers=exclude_unset(headers),
-            response_model=Annotated[
-                Union[
-                    List[ContentDirectoryItems],
-                    ContentFile,
-                    ContentSymlink,
-                    ContentSubmodule,
-                ],
-                Field(discriminator="type"),
+            response_model=Union[
+                List[ContentDirectoryItems],
+                ContentFile,
+                ContentSymlink,
+                ContentSubmodule,
             ],
             error_models={
                 "404": BasicError,

diff --git a/githubkit/versions/ghec_v2022_11_28/types/group_0051.py b/githubkit/versions/ghec_v2022_11_28/types/group_0051.py
@@ -9,7 +9,7 @@
 
 from __future__ import annotations
 
-from typing import Union
+from typing import Union, Literal
 from datetime import date, datetime
 from typing_extensions import TypedDict, NotRequired
 
@@ -33,6 +33,7 @@ class CopilotSeatDetailsType(TypedDict):
     last_activity_editor: NotRequired[Union[str, None]]
     created_at: datetime
     updated_at: NotRequired[datetime]
+    plan_type: NotRequired[Literal["business", "enterprise", "unknown"]]
 
 
 class EnterpriseTeamType(TypedDict):

diff --git a/githubkit/versions/ghec_v2022_11_28/types/group_0098.py b/githubkit/versions/ghec_v2022_11_28/types/group_0098.py
@@ -28,6 +28,7 @@ class CopilotOrganizationDetailsType(TypedDict):
     seat_management_setting: Literal[
         "assign_all", "assign_selected", "disabled", "unconfigured"
     ]
+    plan_type: NotRequired[Literal["business", "enterprise", "unknown"]]
 
 
 class CopilotSeatBreakdownType(TypedDict):

diff --git a/githubkit/versions/ghec_v2022_11_28/types/group_1010.py b/githubkit/versions/ghec_v2022_11_28/types/group_1010.py
@@ -26,7 +26,9 @@ class ReposOwnerRepoCheckRunsPostBodyOneof1Type(TypedDict):
     head_sha: str
     details_url: NotRequired[str]
     external_id: NotRequired[str]
-    status: NotRequired[Literal["queued", "in_progress"]]
+    status: NotRequired[
+        Literal["queued", "in_progress", "waiting", "requested", "pending"]
+    ]
     started_at: NotRequired[datetime]
     conclusion: NotRequired[
         Literal[

diff --git a/githubkit/versions/v2022_11_28/models/group_0030.py b/githubkit/versions/v2022_11_28/models/group_0030.py
@@ -9,7 +9,7 @@
 
 from __future__ import annotations
 
-from typing import Union
+from typing import Union, Literal
 from datetime import date, datetime
 
 from pydantic import Field
@@ -55,6 +55,10 @@ class CopilotSeatDetails(GitHubModel):
         default=UNSET,
         description="Timestamp of when the assignee's GitHub Copilot access was last updated, in ISO 8601 format.",
     )
+    plan_type: Missing[Literal["business", "enterprise", "unknown"]] = Field(
+        default=UNSET,
+        description="The Copilot plan of the organization, or the parent enterprise, when applicable.",
+    )
 
 
 class EnterpriseTeam(GitHubModel):

diff --git a/githubkit/versions/v2022_11_28/models/group_0082.py b/githubkit/versions/v2022_11_28/models/group_0082.py
@@ -49,6 +49,10 @@ class CopilotOrganizationDetails(ExtraGitHubModel):
     seat_management_setting: Literal[
         "assign_all", "assign_selected", "disabled", "unconfigured"
     ] = Field(description="The mode of assigning new seats.")
+    plan_type: Missing[Literal["business", "enterprise", "unknown"]] = Field(
+        default=UNSET,
+        description="The Copilot plan of the organization, or the parent enterprise, when applicable.",
+    )
 
 
 class CopilotSeatBreakdown(GitHubModel):

diff --git a/githubkit/versions/v2022_11_28/models/group_0936.py b/githubkit/versions/v2022_11_28/models/group_0936.py
@@ -38,7 +38,9 @@ class ReposOwnerRepoCheckRunsPostBodyOneof1(ExtraGitHubModel):
     external_id: Missing[str] = Field(
         default=UNSET, description="A reference for the run on the integrator's system."
     )
-    status: Missing[Literal["queued", "in_progress"]] = Field(default=UNSET)
+    status: Missing[
+        Literal["queued", "in_progress", "waiting", "requested", "pending"]
+    ] = Field(default=UNSET)
     started_at: Missing[datetime] = Field(
         default=UNSET,
         description="The time that the check run began. This is a timestamp in [ISO 8601](https://en.wikipedia.org/wiki/ISO_8601) format: `YYYY-MM-DDTHH:MM:SSZ`.",

diff --git a/githubkit/versions/v2022_11_28/rest/checks.py b/githubkit/versions/v2022_11_28/rest/checks.py
@@ -10,10 +10,9 @@
 from __future__ import annotations
 
 from weakref import ref
-from typing_extensions import Annotated
 from typing import TYPE_CHECKING, Dict, Literal, Optional, overload
 
-from pydantic import Field, BaseModel
+from pydantic import BaseModel
 
 from githubkit.typing import Missing, UnsetType
 from githubkit.utils import UNSET, exclude_unset
@@ -124,7 +123,9 @@ def create(
         head_sha: str,
         details_url: Missing[str] = UNSET,
         external_id: Missing[str] = UNSET,
-        status: Missing[Literal["queued", "in_progress"]] = UNSET,
+        status: Missing[
+            Literal["queued", "in_progress", "waiting", "requested", "pending"]
+        ] = UNSET,
         started_at: Missing[datetime] = UNSET,
         conclusion: Missing[
             Literal[
@@ -178,12 +179,9 @@ def create(
 
         json = kwargs if data is UNSET else data
         json = type_validate_python(
-            Annotated[
-                Union[
-                    ReposOwnerRepoCheckRunsPostBodyOneof0,
-                    ReposOwnerRepoCheckRunsPostBodyOneof1,
-                ],
-                Field(discriminator="status"),
+            Union[
+                ReposOwnerRepoCheckRunsPostBodyOneof0,
+                ReposOwnerRepoCheckRunsPostBodyOneof1,
             ],
             json,
         )
@@ -253,7 +251,9 @@ async def async_create(
         head_sha: str,
         details_url: Missing[str] = UNSET,
         external_id: Missing[str] = UNSET,
-        status: Missing[Literal["queued", "in_progress"]] = UNSET,
+        status: Missing[
+            Literal["queued", "in_progress", "waiting", "requested", "pending"]
+        ] = UNSET,
         started_at: Missing[datetime] = UNSET,
         conclusion: Missing[
             Literal[
@@ -307,12 +307,9 @@ async def async_create(
 
         json = kwargs if data is UNSET else data
         json = type_validate_python(
-            Annotated[
-                Union[
-                    ReposOwnerRepoCheckRunsPostBodyOneof0,
-                    ReposOwnerRepoCheckRunsPostBodyOneof1,
-                ],
-                Field(discriminator="status"),
+            Union[
+                ReposOwnerRepoCheckRunsPostBodyOneof0,
+                ReposOwnerRepoCheckRunsPostBodyOneof1,
             ],
             json,
         )