robosystems-client 0.2.11__py3-none-any.whl → 0.2.12__py3-none-any.whl

robosystems_client/models/org_member_response.py

@@ -0,0 +1,104 @@
+import datetime
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+from dateutil.parser import isoparse
+
+from ..models.org_role import OrgRole
+
+T = TypeVar("T", bound="OrgMemberResponse")
+
+
+@_attrs_define
+class OrgMemberResponse:
+  """Organization member response.
+
+  Attributes:
+      user_id (str):
+      name (str):
+      email (str):
+      role (OrgRole):
+      joined_at (datetime.datetime):
+      is_active (bool):
+  """
+
+  user_id: str
+  name: str
+  email: str
+  role: OrgRole
+  joined_at: datetime.datetime
+  is_active: bool
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    user_id = self.user_id
+
+    name = self.name
+
+    email = self.email
+
+    role = self.role.value
+
+    joined_at = self.joined_at.isoformat()
+
+    is_active = self.is_active
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "user_id": user_id,
+        "name": name,
+        "email": email,
+        "role": role,
+        "joined_at": joined_at,
+        "is_active": is_active,
+      }
+    )
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    user_id = d.pop("user_id")
+
+    name = d.pop("name")
+
+    email = d.pop("email")
+
+    role = OrgRole(d.pop("role"))
+
+    joined_at = isoparse(d.pop("joined_at"))
+
+    is_active = d.pop("is_active")
+
+    org_member_response = cls(
+      user_id=user_id,
+      name=name,
+      email=email,
+      role=role,
+      joined_at=joined_at,
+      is_active=is_active,
+    )
+
+    org_member_response.additional_properties = d
+    return org_member_response
+
+  @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

robosystems_client/models/org_response.py

@@ -0,0 +1,121 @@
+import datetime
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+from dateutil.parser import isoparse
+
+from ..models.org_role import OrgRole
+from ..models.org_type import OrgType
+
+T = TypeVar("T", bound="OrgResponse")
+
+
+@_attrs_define
+class OrgResponse:
+  """Organization summary response.
+
+  Attributes:
+      id (str):
+      name (str):
+      org_type (OrgType):
+      role (OrgRole):
+      member_count (int):
+      graph_count (int):
+      created_at (datetime.datetime):
+      joined_at (datetime.datetime):
+  """
+
+  id: str
+  name: str
+  org_type: OrgType
+  role: OrgRole
+  member_count: int
+  graph_count: int
+  created_at: datetime.datetime
+  joined_at: datetime.datetime
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    id = self.id
+
+    name = self.name
+
+    org_type = self.org_type.value
+
+    role = self.role.value
+
+    member_count = self.member_count
+
+    graph_count = self.graph_count
+
+    created_at = self.created_at.isoformat()
+
+    joined_at = self.joined_at.isoformat()
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "id": id,
+        "name": name,
+        "org_type": org_type,
+        "role": role,
+        "member_count": member_count,
+        "graph_count": graph_count,
+        "created_at": created_at,
+        "joined_at": joined_at,
+      }
+    )
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    id = d.pop("id")
+
+    name = d.pop("name")
+
+    org_type = OrgType(d.pop("org_type"))
+
+    role = OrgRole(d.pop("role"))
+
+    member_count = d.pop("member_count")
+
+    graph_count = d.pop("graph_count")
+
+    created_at = isoparse(d.pop("created_at"))
+
+    joined_at = isoparse(d.pop("joined_at"))
+
+    org_response = cls(
+      id=id,
+      name=name,
+      org_type=org_type,
+      role=role,
+      member_count=member_count,
+      graph_count=graph_count,
+      created_at=created_at,
+      joined_at=joined_at,
+    )
+
+    org_response.additional_properties = d
+    return org_response
+
+  @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

robosystems_client/models/org_role.py

@@ -0,0 +1,10 @@
+from enum import Enum
+
+
+class OrgRole(str, Enum):
+  ADMIN = "admin"
+  MEMBER = "member"
+  OWNER = "owner"
+
+  def __str__(self) -> str:
+    return str(self.value)

robosystems_client/models/org_type.py

@@ -0,0 +1,10 @@
+from enum import Enum
+
+
+class OrgType(str, Enum):
+  ENTERPRISE = "enterprise"
+  PERSONAL = "personal"
+  TEAM = "team"
+
+  def __str__(self) -> str:
+    return str(self.value)

robosystems_client/models/org_usage_response.py

@@ -0,0 +1,146 @@
+import datetime
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+from dateutil.parser import isoparse
+
+if TYPE_CHECKING:
+  from ..models.org_usage_response_daily_trend_item import (
+    OrgUsageResponseDailyTrendItem,
+  )
+  from ..models.org_usage_response_graph_details_item import (
+    OrgUsageResponseGraphDetailsItem,
+  )
+  from ..models.org_usage_summary import OrgUsageSummary
+
+
+T = TypeVar("T", bound="OrgUsageResponse")
+
+
+@_attrs_define
+class OrgUsageResponse:
+  """Organization usage response.
+
+  Attributes:
+      org_id (str):
+      period_days (int):
+      start_date (datetime.datetime):
+      end_date (datetime.datetime):
+      summary (OrgUsageSummary): Organization usage summary.
+      graph_details (list['OrgUsageResponseGraphDetailsItem']):
+      daily_trend (list['OrgUsageResponseDailyTrendItem']):
+  """
+
+  org_id: str
+  period_days: int
+  start_date: datetime.datetime
+  end_date: datetime.datetime
+  summary: "OrgUsageSummary"
+  graph_details: list["OrgUsageResponseGraphDetailsItem"]
+  daily_trend: list["OrgUsageResponseDailyTrendItem"]
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    org_id = self.org_id
+
+    period_days = self.period_days
+
+    start_date = self.start_date.isoformat()
+
+    end_date = self.end_date.isoformat()
+
+    summary = self.summary.to_dict()
+
+    graph_details = []
+    for graph_details_item_data in self.graph_details:
+      graph_details_item = graph_details_item_data.to_dict()
+      graph_details.append(graph_details_item)
+
+    daily_trend = []
+    for daily_trend_item_data in self.daily_trend:
+      daily_trend_item = daily_trend_item_data.to_dict()
+      daily_trend.append(daily_trend_item)
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "org_id": org_id,
+        "period_days": period_days,
+        "start_date": start_date,
+        "end_date": end_date,
+        "summary": summary,
+        "graph_details": graph_details,
+        "daily_trend": daily_trend,
+      }
+    )
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    from ..models.org_usage_response_daily_trend_item import (
+      OrgUsageResponseDailyTrendItem,
+    )
+    from ..models.org_usage_response_graph_details_item import (
+      OrgUsageResponseGraphDetailsItem,
+    )
+    from ..models.org_usage_summary import OrgUsageSummary
+
+    d = dict(src_dict)
+    org_id = d.pop("org_id")
+
+    period_days = d.pop("period_days")
+
+    start_date = isoparse(d.pop("start_date"))
+
+    end_date = isoparse(d.pop("end_date"))
+
+    summary = OrgUsageSummary.from_dict(d.pop("summary"))
+
+    graph_details = []
+    _graph_details = d.pop("graph_details")
+    for graph_details_item_data in _graph_details:
+      graph_details_item = OrgUsageResponseGraphDetailsItem.from_dict(
+        graph_details_item_data
+      )
+
+      graph_details.append(graph_details_item)
+
+    daily_trend = []
+    _daily_trend = d.pop("daily_trend")
+    for daily_trend_item_data in _daily_trend:
+      daily_trend_item = OrgUsageResponseDailyTrendItem.from_dict(daily_trend_item_data)
+
+      daily_trend.append(daily_trend_item)
+
+    org_usage_response = cls(
+      org_id=org_id,
+      period_days=period_days,
+      start_date=start_date,
+      end_date=end_date,
+      summary=summary,
+      graph_details=graph_details,
+      daily_trend=daily_trend,
+    )
+
+    org_usage_response.additional_properties = d
+    return org_usage_response
+
+  @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

robosystems_client/models/org_usage_response_daily_trend_item.py

@@ -0,0 +1,44 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="OrgUsageResponseDailyTrendItem")
+
+
+@_attrs_define
+class OrgUsageResponseDailyTrendItem:
+  """ """
+
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    org_usage_response_daily_trend_item = cls()
+
+    org_usage_response_daily_trend_item.additional_properties = d
+    return org_usage_response_daily_trend_item
+
+  @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

robosystems_client/models/org_usage_response_graph_details_item.py

@@ -0,0 +1,44 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="OrgUsageResponseGraphDetailsItem")
+
+
+@_attrs_define
+class OrgUsageResponseGraphDetailsItem:
+  """ """
+
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    org_usage_response_graph_details_item = cls()
+
+    org_usage_response_graph_details_item.additional_properties = d
+    return org_usage_response_graph_details_item
+
+  @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

robosystems_client/models/org_usage_summary.py

@@ -0,0 +1,158 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="OrgUsageSummary")
+
+
+@_attrs_define
+class OrgUsageSummary:
+  """Organization usage summary.
+
+  Attributes:
+      total_credits_used (float):
+      total_ai_operations (int):
+      total_storage_gb (float):
+      total_api_calls (int):
+      daily_avg_credits (float):
+      daily_avg_api_calls (float):
+      projected_monthly_credits (float):
+      projected_monthly_api_calls (int):
+      credits_limit (Union[None, int]):
+      api_calls_limit (Union[None, int]):
+      storage_limit_gb (Union[None, int]):
+  """
+
+  total_credits_used: float
+  total_ai_operations: int
+  total_storage_gb: float
+  total_api_calls: int
+  daily_avg_credits: float
+  daily_avg_api_calls: float
+  projected_monthly_credits: float
+  projected_monthly_api_calls: int
+  credits_limit: Union[None, int]
+  api_calls_limit: Union[None, int]
+  storage_limit_gb: Union[None, int]
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    total_credits_used = self.total_credits_used
+
+    total_ai_operations = self.total_ai_operations
+
+    total_storage_gb = self.total_storage_gb
+
+    total_api_calls = self.total_api_calls
+
+    daily_avg_credits = self.daily_avg_credits
+
+    daily_avg_api_calls = self.daily_avg_api_calls
+
+    projected_monthly_credits = self.projected_monthly_credits
+
+    projected_monthly_api_calls = self.projected_monthly_api_calls
+
+    credits_limit: Union[None, int]
+    credits_limit = self.credits_limit
+
+    api_calls_limit: Union[None, int]
+    api_calls_limit = self.api_calls_limit
+
+    storage_limit_gb: Union[None, int]
+    storage_limit_gb = self.storage_limit_gb
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "total_credits_used": total_credits_used,
+        "total_ai_operations": total_ai_operations,
+        "total_storage_gb": total_storage_gb,
+        "total_api_calls": total_api_calls,
+        "daily_avg_credits": daily_avg_credits,
+        "daily_avg_api_calls": daily_avg_api_calls,
+        "projected_monthly_credits": projected_monthly_credits,
+        "projected_monthly_api_calls": projected_monthly_api_calls,
+        "credits_limit": credits_limit,
+        "api_calls_limit": api_calls_limit,
+        "storage_limit_gb": storage_limit_gb,
+      }
+    )
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    total_credits_used = d.pop("total_credits_used")
+
+    total_ai_operations = d.pop("total_ai_operations")
+
+    total_storage_gb = d.pop("total_storage_gb")
+
+    total_api_calls = d.pop("total_api_calls")
+
+    daily_avg_credits = d.pop("daily_avg_credits")
+
+    daily_avg_api_calls = d.pop("daily_avg_api_calls")
+
+    projected_monthly_credits = d.pop("projected_monthly_credits")
+
+    projected_monthly_api_calls = d.pop("projected_monthly_api_calls")
+
+    def _parse_credits_limit(data: object) -> Union[None, int]:
+      if data is None:
+        return data
+      return cast(Union[None, int], data)
+
+    credits_limit = _parse_credits_limit(d.pop("credits_limit"))
+
+    def _parse_api_calls_limit(data: object) -> Union[None, int]:
+      if data is None:
+        return data
+      return cast(Union[None, int], data)
+
+    api_calls_limit = _parse_api_calls_limit(d.pop("api_calls_limit"))
+
+    def _parse_storage_limit_gb(data: object) -> Union[None, int]:
+      if data is None:
+        return data
+      return cast(Union[None, int], data)
+
+    storage_limit_gb = _parse_storage_limit_gb(d.pop("storage_limit_gb"))
+
+    org_usage_summary = cls(
+      total_credits_used=total_credits_used,
+      total_ai_operations=total_ai_operations,
+      total_storage_gb=total_storage_gb,
+      total_api_calls=total_api_calls,
+      daily_avg_credits=daily_avg_credits,
+      daily_avg_api_calls=daily_avg_api_calls,
+      projected_monthly_credits=projected_monthly_credits,
+      projected_monthly_api_calls=projected_monthly_api_calls,
+      credits_limit=credits_limit,
+      api_calls_limit=api_calls_limit,
+      storage_limit_gb=storage_limit_gb,
+    )
+
+    org_usage_summary.additional_properties = d
+    return org_usage_summary
+
+  @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

robosystems_client/models/repository_subscriptions.py

@@ -15,13 +15,18 @@ T = TypeVar("T", bound="RepositorySubscriptions")
 class RepositorySubscriptions:
   """Repository subscription offerings.
 
-  Attributes:
-      description (str): Description of repository subscriptions
-      repositories (list['RepositoryInfo']): Available repositories
-      notes (list[str]): Important notes
+  Repository subscriptions are per-organization, not per-graph. All members
+  of an organization share access to subscribed repositories.
+
+      Attributes:
+          description (str): Description of repository subscriptions
+          pricing_model (str): Pricing model type (per_graph or per_organization)
+          repositories (list['RepositoryInfo']): Available repositories
+          notes (list[str]): Important notes
   """
 
   description: str
+  pricing_model: str
   repositories: list["RepositoryInfo"]
   notes: list[str]
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -29,6 +34,8 @@ class RepositorySubscriptions:
   def to_dict(self) -> dict[str, Any]:
     description = self.description
 
+    pricing_model = self.pricing_model
+
     repositories = []
     for repositories_item_data in self.repositories:
       repositories_item = repositories_item_data.to_dict()
@@ -41,6 +48,7 @@ class RepositorySubscriptions:
     field_dict.update(
       {
         "description": description,
+        "pricing_model": pricing_model,
         "repositories": repositories,
         "notes": notes,
       }
@@ -55,6 +63,8 @@ class RepositorySubscriptions:
     d = dict(src_dict)
     description = d.pop("description")
 
+    pricing_model = d.pop("pricing_model")
+
     repositories = []
     _repositories = d.pop("repositories")
     for repositories_item_data in _repositories:
@@ -66,6 +76,7 @@ class RepositorySubscriptions:
 
     repository_subscriptions = cls(
       description=description,
+      pricing_model=pricing_model,
       repositories=repositories,
       notes=notes,
     )