1212from .resource_tracker import PricingPlanId
1313from .utils .enums import StrAutoEnum
1414
15- LicensedItemID : TypeAlias = UUID
15+ LicensedItemID : TypeAlias = UUID # <-- LicensedItemID == LicenseID
16+ LicenseID : TypeAlias = UUID
17+ LicensedResourceID : TypeAlias = UUID
1618
1719
1820class LicensedResourceType (StrAutoEnum ):
@@ -64,8 +66,8 @@ class FeaturesDict(TypedDict):
6466#
6567
6668
67- class LicensedItemDB (BaseModel ):
68- licensed_item_id : LicensedItemID
69+ class LicenseDB (BaseModel ):
70+ licensed_resource_id : LicensedResourceID
6971 display_name : str
7072
7173 licensed_resource_name : str
@@ -83,15 +85,15 @@ class LicensedItemDB(BaseModel):
8385 model_config = ConfigDict (from_attributes = True )
8486
8587
86- class LicensedItemUpdateDB (BaseModel ):
88+ class LicenseUpdateDB (BaseModel ):
8789 display_name : str | None = None
8890 licensed_resource_name : str | None = None
8991 pricing_plan_id : PricingPlanId | None = None
9092 trash : bool | None = None
9193
9294
93- class LicensedItem (BaseModel ):
94- licensed_item_id : LicensedItemID
95+ class LicensedResource (BaseModel ):
96+ licensed_resource_id : LicensedResourceID
9597 display_name : str
9698 licensed_resource_name : str
9799 licensed_resource_type : LicensedResourceType
@@ -106,7 +108,7 @@ def _update_json_schema_extra(schema: JsonDict) -> None:
106108 {
107109 "examples" : [
108110 {
109- "licensed_item_id " : "0362b88b-91f8-4b41-867c-35544ad1f7a1" ,
111+ "licensed_resource_id " : "0362b88b-91f8-4b41-867c-35544ad1f7a1" ,
110112 "display_name" : "my best model" ,
111113 "licensed_resource_name" : "best-model" ,
112114 "licensed_resource_type" : f"{ LicensedResourceType .VIP_MODEL } " ,
@@ -129,6 +131,6 @@ def _update_json_schema_extra(schema: JsonDict) -> None:
129131 model_config = ConfigDict (json_schema_extra = _update_json_schema_extra )
130132
131133
132- class LicensedItemPage (NamedTuple ):
134+ class LicensedResourcePage (NamedTuple ):
133135 total : PositiveInt
134- items : list [LicensedItem ]
136+ items : list [LicensedResource ]
0 commit comments