chore(spec): sync generated client from upstream openapi.yaml

src/graphn/_generated/client.py

@@ -100,7 +100,7 @@ def __enter__(self) -> "Client":
         self.get_httpx_client().__enter__()
         return self
 
-    def __exit__(self, *args: Any, **kwargs: Any) -> None:
+    def __exit__(self, *args: object, **kwargs: Any) -> None:
         """Exit a context manager for internal httpx.Client (see httpx docs)"""
         self.get_httpx_client().__exit__(*args, **kwargs)
 
@@ -131,7 +131,7 @@ async def __aenter__(self) -> "Client":
         await self.get_async_httpx_client().__aenter__()
         return self
 
-    async def __aexit__(self, *args: Any, **kwargs: Any) -> None:
+    async def __aexit__(self, *args: object, **kwargs: Any) -> None:
         """Exit a context manager for underlying httpx.AsyncClient (see httpx docs)"""
         await self.get_async_httpx_client().__aexit__(*args, **kwargs)
 
@@ -241,7 +241,7 @@ def __enter__(self) -> "AuthenticatedClient":
         self.get_httpx_client().__enter__()
         return self
 
-    def __exit__(self, *args: Any, **kwargs: Any) -> None:
+    def __exit__(self, *args: object, **kwargs: Any) -> None:
         """Exit a context manager for internal httpx.Client (see httpx docs)"""
         self.get_httpx_client().__exit__(*args, **kwargs)
 
@@ -277,6 +277,6 @@ async def __aenter__(self) -> "AuthenticatedClient":
         await self.get_async_httpx_client().__aenter__()
         return self
 
-    async def __aexit__(self, *args: Any, **kwargs: Any) -> None:
+    async def __aexit__(self, *args: object, **kwargs: Any) -> None:
         """Exit a context manager for underlying httpx.AsyncClient (see httpx docs)"""
         await self.get_async_httpx_client().__aexit__(*args, **kwargs)

src/graphn/_generated/models/__init__.py

@@ -49,6 +49,7 @@
 from .tts_request_response_format import TTSRequestResponseFormat
 from .validate_model_request import ValidateModelRequest
 from .validate_model_request_quantization import ValidateModelRequestQuantization
+from .validate_model_request_weight_source import ValidateModelRequestWeightSource
 from .validate_model_response import ValidateModelResponse
 from .validate_model_response_artifact_type import ValidateModelResponseArtifactType
 from .weight_source import WeightSource
@@ -76,9 +77,9 @@
     "CustomModelQuantization",
     "CustomModelStatus",
     "CustomModelUpdate",
-    "DiscoveredImportedModel",
     "DiscoverImportedModelsRequest",
     "DiscoverImportedModelsResponse",
+    "DiscoveredImportedModel",
     "Error",
     "GpuHoursResponse",
     "InferenceError",
@@ -94,13 +95,14 @@
     "SecretList",
     "SecretUpdate",
     "SupportedArchitectures",
+    "TTSRequest",
+    "TTSRequestResponseFormat",
     "TestConnectionRequest",
     "TestConnectionResponse",
     "TestConnectionResponseUsage",
-    "TTSRequest",
-    "TTSRequestResponseFormat",
     "ValidateModelRequest",
     "ValidateModelRequestQuantization",
+    "ValidateModelRequestWeightSource",
     "ValidateModelResponse",
     "ValidateModelResponseArtifactType",
     "WeightSource",

src/graphn/_generated/models/architecture_info.py

@@ -4,6 +4,7 @@
 from typing import Any, TypeVar, cast
 
 from attrs import define as _attrs_define
+from typing_extensions import Self
 
 T = TypeVar("T", bound="ArchitectureInfo")
 
@@ -39,7 +40,7 @@ def to_dict(self) -> dict[str, Any]:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
         d = dict(src_dict)
         name = d.pop("name")
 

src/graphn/_generated/models/chat_completion_request.py

@@ -5,6 +5,7 @@
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
+from typing_extensions import Self
 
 from ..types import UNSET, Unset
 
@@ -157,7 +158,7 @@ def to_dict(self) -> dict[str, Any]:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
         from ..models.chat_completion_request_response_format import (
             ChatCompletionRequestResponseFormat,
         )

src/graphn/_generated/models/chat_completion_request_response_format.py

@@ -5,6 +5,7 @@
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
+from typing_extensions import Self
 
 T = TypeVar("T", bound="ChatCompletionRequestResponseFormat")
 
@@ -23,7 +24,7 @@ def to_dict(self) -> dict[str, Any]:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
         d = dict(src_dict)
         chat_completion_request_response_format = cls()
 

src/graphn/_generated/models/chat_completion_request_tool_choice_type_1.py

@@ -5,6 +5,7 @@
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
+from typing_extensions import Self
 
 T = TypeVar("T", bound="ChatCompletionRequestToolChoiceType1")
 
@@ -23,7 +24,7 @@ def to_dict(self) -> dict[str, Any]:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
         d = dict(src_dict)
         chat_completion_request_tool_choice_type_1 = cls()
 

src/graphn/_generated/models/chat_completion_request_tools_item.py

@@ -5,6 +5,7 @@
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
+from typing_extensions import Self
 
 T = TypeVar("T", bound="ChatCompletionRequestToolsItem")
 
@@ -23,7 +24,7 @@ def to_dict(self) -> dict[str, Any]:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
         d = dict(src_dict)
         chat_completion_request_tools_item = cls()
 

src/graphn/_generated/models/chat_completion_response.py

@@ -5,6 +5,7 @@
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
+from typing_extensions import Self
 
 from ..models.chat_completion_response_object import ChatCompletionResponseObject
 from ..types import UNSET, Unset
@@ -76,7 +77,7 @@ def to_dict(self) -> dict[str, Any]:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
         from ..models.chat_completion_response_choices_item import (
             ChatCompletionResponseChoicesItem,
         )

src/graphn/_generated/models/chat_completion_response_choices_item.py

@@ -5,6 +5,7 @@
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
+from typing_extensions import Self
 
 from ..types import UNSET, Unset
 
@@ -57,7 +58,7 @@ def to_dict(self) -> dict[str, Any]:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
         from ..models.chat_message import ChatMessage
 
         d = dict(src_dict)

src/graphn/_generated/models/chat_completion_response_usage.py

@@ -5,6 +5,7 @@
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
+from typing_extensions import Self
 
 from ..types import UNSET, Unset
 
@@ -45,7 +46,7 @@ def to_dict(self) -> dict[str, Any]:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
         d = dict(src_dict)
         prompt_tokens = d.pop("prompt_tokens", UNSET)
 

src/graphn/_generated/models/chat_message.py

@@ -5,6 +5,7 @@
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
+from typing_extensions import Self
 
 from ..models.chat_message_role import ChatMessageRole
 from ..types import UNSET, Unset
@@ -67,7 +68,7 @@ def to_dict(self) -> dict[str, Any]:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
         from ..models.chat_message_tool_calls_item import ChatMessageToolCallsItem
 
         d = dict(src_dict)

src/graphn/_generated/models/chat_message_tool_calls_item.py

@@ -5,6 +5,7 @@
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
+from typing_extensions import Self
 
 T = TypeVar("T", bound="ChatMessageToolCallsItem")
 
@@ -23,7 +24,7 @@ def to_dict(self) -> dict[str, Any]:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
         d = dict(src_dict)
         chat_message_tool_calls_item = cls()
 

src/graphn/_generated/models/custom_model.py

@@ -6,7 +6,7 @@
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
-from dateutil.parser import isoparse
+from typing_extensions import Self
 
 from ..models.capability import Capability
 from ..models.custom_model_artifact_type import CustomModelArtifactType
@@ -288,7 +288,7 @@ def to_dict(self) -> dict[str, Any]:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
         d = dict(src_dict)
         id = d.pop("id")
 
@@ -317,9 +317,9 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
 
         cooldown_seconds = d.pop("cooldown_seconds")
 
-        created_at = isoparse(d.pop("created_at"))
+        created_at = datetime.datetime.fromisoformat(d.pop("created_at"))
 
-        updated_at = isoparse(d.pop("updated_at"))
+        updated_at = datetime.datetime.fromisoformat(d.pop("updated_at"))
 
         display_name = d.pop("display_name", UNSET)
 

src/graphn/_generated/models/custom_model_access.py

@@ -4,6 +4,7 @@
 from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
+from typing_extensions import Self
 
 T = TypeVar("T", bound="CustomModelAccess")
 
@@ -31,7 +32,7 @@ def to_dict(self) -> dict[str, Any]:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
         d = dict(src_dict)
         allowed = d.pop("allowed")
 

src/graphn/_generated/models/custom_model_create.py

@@ -4,6 +4,7 @@
 from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
+from typing_extensions import Self
 
 from ..models.capability import Capability
 from ..models.custom_model_create_quantization import CustomModelCreateQuantization
@@ -176,7 +177,7 @@ def to_dict(self) -> dict[str, Any]:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
         d = dict(src_dict)
         name = d.pop("name")
 

src/graphn/_generated/models/custom_model_list.py

@@ -5,6 +5,7 @@
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
+from typing_extensions import Self
 
 from ..types import UNSET, Unset
 
@@ -53,7 +54,7 @@ def to_dict(self) -> dict[str, Any]:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
         from ..models.custom_model import CustomModel
 
         d = dict(src_dict)

src/graphn/_generated/models/custom_model_update.py

@@ -4,6 +4,7 @@
 from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
+from typing_extensions import Self
 
 from ..types import UNSET, Unset
 
@@ -63,7 +64,7 @@ def to_dict(self) -> dict[str, Any]:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
         d = dict(src_dict)
         name = d.pop("name", UNSET)
 

src/graphn/_generated/models/discover_imported_models_request.py

@@ -4,6 +4,7 @@
 from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
+from typing_extensions import Self
 
 T = TypeVar("T", bound="DiscoverImportedModelsRequest")
 
@@ -36,7 +37,7 @@ def to_dict(self) -> dict[str, Any]:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
         d = dict(src_dict)
         endpoint = d.pop("endpoint")
 

src/graphn/_generated/models/discover_imported_models_response.py

@@ -4,6 +4,7 @@
 from typing import TYPE_CHECKING, Any, TypeVar
 
 from attrs import define as _attrs_define
+from typing_extensions import Self
 
 if TYPE_CHECKING:
     from ..models.discovered_imported_model import DiscoveredImportedModel
@@ -38,7 +39,7 @@ def to_dict(self) -> dict[str, Any]:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
         from ..models.discovered_imported_model import DiscoveredImportedModel
 
         d = dict(src_dict)

src/graphn/_generated/models/discovered_imported_model.py

@@ -4,6 +4,7 @@
 from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
+from typing_extensions import Self
 
 T = TypeVar("T", bound="DiscoveredImportedModel")
 
@@ -36,7 +37,7 @@ def to_dict(self) -> dict[str, Any]:
         return field_dict
 
     @classmethod
-    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
         d = dict(src_dict)
         id = d.pop("id")