Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
24 changes: 12 additions & 12 deletions src/graphn/_generated/api/organizations/create_organization.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,8 +6,8 @@
from ... import errors
from ...client import AuthenticatedClient, Client
from ...models.error import Error
from ...models.organization import Organization
from ...models.organization_create import OrganizationCreate
from ...models.organization_created import OrganizationCreated
from ...types import Response


Expand All @@ -32,9 +32,9 @@ def _get_kwargs(

def _parse_response(
*, client: AuthenticatedClient | Client, response: httpx.Response
) -> Error | Organization | None:
) -> Error | OrganizationCreated | None:
if response.status_code == 201:
response_201 = Organization.from_dict(response.json())
response_201 = OrganizationCreated.from_dict(response.json())

return response_201

Expand All @@ -61,7 +61,7 @@ def _parse_response(

def _build_response(
*, client: AuthenticatedClient | Client, response: httpx.Response
) -> Response[Error | Organization]:
) -> Response[Error | OrganizationCreated]:
return Response(
status_code=HTTPStatus(response.status_code),
content=response.content,
Expand All @@ -74,7 +74,7 @@ def sync_detailed(
*,
client: AuthenticatedClient | Client,
body: OrganizationCreate,
) -> Response[Error | Organization]:
) -> Response[Error | OrganizationCreated]:
"""Create an organization

Args:
Expand All @@ -85,7 +85,7 @@ def sync_detailed(
httpx.TimeoutException: If the request takes longer than Client.timeout.

Returns:
Response[Error | Organization]
Response[Error | OrganizationCreated]
"""

kwargs = _get_kwargs(
Expand All @@ -103,7 +103,7 @@ def sync(
*,
client: AuthenticatedClient | Client,
body: OrganizationCreate,
) -> Error | Organization | None:
) -> Error | OrganizationCreated | None:
"""Create an organization

Args:
Expand All @@ -114,7 +114,7 @@ def sync(
httpx.TimeoutException: If the request takes longer than Client.timeout.

Returns:
Error | Organization
Error | OrganizationCreated
"""

return sync_detailed(
Expand All @@ -127,7 +127,7 @@ async def asyncio_detailed(
*,
client: AuthenticatedClient | Client,
body: OrganizationCreate,
) -> Response[Error | Organization]:
) -> Response[Error | OrganizationCreated]:
"""Create an organization

Args:
Expand All @@ -138,7 +138,7 @@ async def asyncio_detailed(
httpx.TimeoutException: If the request takes longer than Client.timeout.

Returns:
Response[Error | Organization]
Response[Error | OrganizationCreated]
"""

kwargs = _get_kwargs(
Expand All @@ -154,7 +154,7 @@ async def asyncio(
*,
client: AuthenticatedClient | Client,
body: OrganizationCreate,
) -> Error | Organization | None:
) -> Error | OrganizationCreated | None:
"""Create an organization

Args:
Expand All @@ -165,7 +165,7 @@ async def asyncio(
httpx.TimeoutException: If the request takes longer than Client.timeout.

Returns:
Error | Organization
Error | OrganizationCreated
"""

return (
Expand Down
4 changes: 4 additions & 0 deletions src/graphn/_generated/models/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -323,6 +323,8 @@
from .organization import Organization
from .organization_create import OrganizationCreate
from .organization_create_type import OrganizationCreateType
from .organization_created import OrganizationCreated
from .organization_created_type import OrganizationCreatedType
from .organization_list import OrganizationList
from .organization_type import OrganizationType
from .organization_update import OrganizationUpdate
Expand Down Expand Up @@ -745,6 +747,8 @@
"Organization",
"OrganizationCreate",
"OrganizationCreateType",
"OrganizationCreated",
"OrganizationCreatedType",
"OrganizationList",
"OrganizationType",
"OrganizationUpdate",
Expand Down
5 changes: 4 additions & 1 deletion src/graphn/_generated/models/custom_model_create.py
Original file line number Diff line number Diff line change
Expand Up @@ -81,7 +81,10 @@ class CustomModelCreate:
that serves its own weights and the adapters attached to it.
Requires `max_lora_rank`. Cannot be combined with
`base_model_id`, `quantization`, `disaggregated_pd`,
`speculative_decoding` or more than one GPU.
`speculative_decoding` or more than one GPU. The base must be
a dense Llama, Mistral, Qwen2, Qwen3, Qwen3.5, Qwen3.6, Qwen3.8
or Gemma 4 model (Gemma 4 31B and 26B-A4B excluded); any other
base fails to deploy.
Default: False.
max_lora_rank (CustomModelCreateMaxLoraRank | Unset): Preview. Required with `lora_enabled`; no default. The
highest
Expand Down
137 changes: 137 additions & 0 deletions src/graphn/_generated/models/organization_created.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,137 @@
from __future__ import annotations

import datetime
from collections.abc import Mapping
from typing import TYPE_CHECKING, Any, Self, TypeVar

from attrs import define as _attrs_define
from attrs import field as _attrs_field

from ..models.organization_created_type import OrganizationCreatedType
from ..types import UNSET, Unset

if TYPE_CHECKING:
from ..models.workspace import Workspace


T = TypeVar("T", bound="OrganizationCreated")


@_attrs_define
class OrganizationCreated:
"""The created organization, plus its default workspace when this request
created one.

Attributes:
id (str):
name (str):
slug (str):
type_ (OrganizationCreatedType):
owner_id (str):
created_at (datetime.datetime):
updated_at (datetime.datetime):
default_workspace (Workspace | Unset):
"""

id: str
name: str
slug: str
type_: OrganizationCreatedType
owner_id: str
created_at: datetime.datetime
updated_at: datetime.datetime
default_workspace: Workspace | Unset = UNSET
additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)

def to_dict(self) -> dict[str, Any]:
id = self.id

name = self.name

slug = self.slug

type_ = self.type_.value

owner_id = self.owner_id

created_at = self.created_at.isoformat()

updated_at = self.updated_at.isoformat()

default_workspace: dict[str, Any] | Unset = UNSET
if not isinstance(self.default_workspace, Unset):
default_workspace = self.default_workspace.to_dict()

field_dict: dict[str, Any] = {}
field_dict.update(self.additional_properties)
field_dict.update(
{
"id": id,
"name": name,
"slug": slug,
"type": type_,
"owner_id": owner_id,
"created_at": created_at,
"updated_at": updated_at,
}
)
if default_workspace is not UNSET:
field_dict["default_workspace"] = default_workspace

return field_dict

@classmethod
def from_dict(cls, src_dict: Mapping[str, Any]) -> Self:
from ..models.workspace import Workspace

d = dict(src_dict)
id = d.pop("id")

name = d.pop("name")

slug = d.pop("slug")

type_ = OrganizationCreatedType(d.pop("type"))

owner_id = d.pop("owner_id")

created_at = datetime.datetime.fromisoformat(d.pop("created_at"))

updated_at = datetime.datetime.fromisoformat(d.pop("updated_at"))

_default_workspace = d.pop("default_workspace", UNSET)
default_workspace: Workspace | Unset
if isinstance(_default_workspace, Unset):
default_workspace = UNSET
else:
default_workspace = Workspace.from_dict(_default_workspace)

organization_created = cls(
id=id,
name=name,
slug=slug,
type_=type_,
owner_id=owner_id,
created_at=created_at,
updated_at=updated_at,
default_workspace=default_workspace,
)

organization_created.additional_properties = d
return organization_created

@property
def additional_keys(self) -> list[str]:
return list(self.additional_properties.keys())

def __getitem__(self, key: str) -> Any:
return self.additional_properties[key]

def __setitem__(self, key: str, value: Any) -> None:
self.additional_properties[key] = value

def __delitem__(self, key: str) -> None:
del self.additional_properties[key]

def __contains__(self, key: str) -> bool:
return key in self.additional_properties
9 changes: 9 additions & 0 deletions src/graphn/_generated/models/organization_created_type.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
from enum import StrEnum


class OrganizationCreatedType(StrEnum):
PERSONAL = "personal"
TEAM = "team"

def __str__(self) -> str:
return str(self.value)
Loading