fal 0.11.2__py3-none-any.whl → 0.11.4__py3-none-any.whl

openapi_fal_rest/api/billing/get_user_spending.py

@@ -47,7 +47,9 @@ def _get_kwargs(
     }
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Union[HTTPValidationError, float]]:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Optional[Union[HTTPValidationError, float]]:
     if response.status_code == HTTPStatus.OK:
         response_200 = cast(float, response.json())
         return response_200
@@ -61,7 +63,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Uni
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[HTTPValidationError, float]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[HTTPValidationError, float]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

openapi_fal_rest/api/billing/handle_stripe_webhook.py

@@ -37,9 +37,13 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: Client, response: httpx.Response
-) -> Optional[Union[HTTPValidationError, HandleStripeWebhookResponseHandleStripeWebhook]]:
+) -> Optional[
+    Union[HTTPValidationError, HandleStripeWebhookResponseHandleStripeWebhook]
+]:
     if response.status_code == HTTPStatus.OK:
-        response_200 = HandleStripeWebhookResponseHandleStripeWebhook.from_dict(response.json())
+        response_200 = HandleStripeWebhookResponseHandleStripeWebhook.from_dict(
+            response.json()
+        )
 
         return response_200
     if response.status_code == HTTPStatus.UNPROCESSABLE_ENTITY:
@@ -54,7 +58,9 @@ def _parse_response(
 
 def _build_response(
     *, client: Client, response: httpx.Response
-) -> Response[Union[HTTPValidationError, HandleStripeWebhookResponseHandleStripeWebhook]]:
+) -> Response[
+    Union[HTTPValidationError, HandleStripeWebhookResponseHandleStripeWebhook]
+]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -67,7 +73,9 @@ def sync_detailed(
     *,
     client: Client,
     stripe_signature: Union[Unset, str] = UNSET,
-) -> Response[Union[HTTPValidationError, HandleStripeWebhookResponseHandleStripeWebhook]]:
+) -> Response[
+    Union[HTTPValidationError, HandleStripeWebhookResponseHandleStripeWebhook]
+]:
     """Handle Stripe Webhook
 
     Args:
@@ -98,7 +106,9 @@ def sync(
     *,
     client: Client,
     stripe_signature: Union[Unset, str] = UNSET,
-) -> Optional[Union[HTTPValidationError, HandleStripeWebhookResponseHandleStripeWebhook]]:
+) -> Optional[
+    Union[HTTPValidationError, HandleStripeWebhookResponseHandleStripeWebhook]
+]:
     """Handle Stripe Webhook
 
     Args:
@@ -122,7 +132,9 @@ async def asyncio_detailed(
     *,
     client: Client,
     stripe_signature: Union[Unset, str] = UNSET,
-) -> Response[Union[HTTPValidationError, HandleStripeWebhookResponseHandleStripeWebhook]]:
+) -> Response[
+    Union[HTTPValidationError, HandleStripeWebhookResponseHandleStripeWebhook]
+]:
     """Handle Stripe Webhook
 
     Args:
@@ -151,7 +163,9 @@ async def asyncio(
     *,
     client: Client,
     stripe_signature: Union[Unset, str] = UNSET,
-) -> Optional[Union[HTTPValidationError, HandleStripeWebhookResponseHandleStripeWebhook]]:
+) -> Optional[
+    Union[HTTPValidationError, HandleStripeWebhookResponseHandleStripeWebhook]
+]:
     """Handle Stripe Webhook
 
     Args:

openapi_fal_rest/api/billing/upcoming_invoice.py

@@ -29,7 +29,9 @@ def _get_kwargs(
     }
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Union[HTTPValidationError, Invoice]]:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Optional[Union[HTTPValidationError, Invoice]]:
     if response.status_code == HTTPStatus.OK:
         response_200 = Invoice.from_dict(response.json())
 
@@ -44,7 +46,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Uni
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[HTTPValidationError, Invoice]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[HTTPValidationError, Invoice]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

openapi_fal_rest/api/billing/update_customer_budget.py

@@ -38,7 +38,9 @@ def _get_kwargs(
     }
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Optional[Union[Any, HTTPValidationError]]:
     if response.status_code == HTTPStatus.OK:
         response_200 = cast(Any, response.json())
         return response_200
@@ -52,7 +54,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Uni
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[Any, HTTPValidationError]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

openapi_fal_rest/api/files/check_dir_hash.py

@@ -16,7 +16,9 @@ def _get_kwargs(
     client: Client,
     json_body: HashCheck,
 ) -> Dict[str, Any]:
-    url = "{}/files/dir/check_hash/{target_path}".format(client.base_url, target_path=target_path)
+    url = "{}/files/dir/check_hash/{target_path}".format(
+        client.base_url, target_path=target_path
+    )
 
     headers: Dict[str, str] = client.get_headers()
     cookies: Dict[str, Any] = client.get_cookies()
@@ -34,7 +36,9 @@ def _get_kwargs(
     }
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Union[HTTPValidationError, bool]]:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Optional[Union[HTTPValidationError, bool]]:
     if response.status_code == HTTPStatus.OK:
         response_200 = cast(bool, response.json())
         return response_200
@@ -48,7 +52,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Uni
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[HTTPValidationError, bool]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[HTTPValidationError, bool]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

openapi_fal_rest/api/files/delete.py

@@ -29,7 +29,9 @@ def _get_kwargs(
     }
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Union[HTTPValidationError, bool]]:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Optional[Union[HTTPValidationError, bool]]:
     if response.status_code == HTTPStatus.OK:
         response_200 = cast(bool, response.json())
         return response_200
@@ -43,7 +45,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Uni
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[HTTPValidationError, bool]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[HTTPValidationError, bool]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

openapi_fal_rest/api/files/download.py

@@ -29,7 +29,9 @@ def _get_kwargs(
     }
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Optional[Union[Any, HTTPValidationError]]:
     if response.status_code == HTTPStatus.OK:
         response_200 = cast(Any, response.json())
         return response_200
@@ -43,7 +45,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Uni
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[Any, HTTPValidationError]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

openapi_fal_rest/api/files/file_exists.py

@@ -37,7 +37,9 @@ def _get_kwargs(
     }
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Union[FileSpec, HTTPValidationError]]:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Optional[Union[FileSpec, HTTPValidationError]]:
     if response.status_code == HTTPStatus.OK:
         response_200 = FileSpec.from_dict(response.json())
 
@@ -52,7 +54,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Uni
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[FileSpec, HTTPValidationError]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[FileSpec, HTTPValidationError]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

openapi_fal_rest/api/files/upload_from_url.py

@@ -34,7 +34,9 @@ def _get_kwargs(
     }
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Union[HTTPValidationError, bool]]:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Optional[Union[HTTPValidationError, bool]]:
     if response.status_code == HTTPStatus.OK:
         response_200 = cast(bool, response.json())
         return response_200
@@ -48,7 +50,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Uni
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[HTTPValidationError, bool]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[HTTPValidationError, bool]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

openapi_fal_rest/api/files/upload_local_file.py

@@ -17,7 +17,9 @@ def _get_kwargs(
     multipart_data: BodyUploadLocalFile,
     unzip: Union[Unset, None, bool] = False,
 ) -> Dict[str, Any]:
-    url = "{}/files/file/local/{target_path}".format(client.base_url, target_path=target_path)
+    url = "{}/files/file/local/{target_path}".format(
+        client.base_url, target_path=target_path
+    )
 
     headers: Dict[str, str] = client.get_headers()
     cookies: Dict[str, Any] = client.get_cookies()
@@ -41,7 +43,9 @@ def _get_kwargs(
     }
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Union[HTTPValidationError, bool]]:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Optional[Union[HTTPValidationError, bool]]:
     if response.status_code == HTTPStatus.OK:
         response_200 = cast(bool, response.json())
         return response_200
@@ -55,7 +59,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Uni
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[HTTPValidationError, bool]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[HTTPValidationError, bool]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

openapi_fal_rest/api/keys/create_key.py

@@ -42,7 +42,9 @@ def _get_kwargs(
     }
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Union[HTTPValidationError, NewUserKey]]:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Optional[Union[HTTPValidationError, NewUserKey]]:
     if response.status_code == HTTPStatus.CREATED:
         response_201 = NewUserKey.from_dict(response.json())
 
@@ -57,7 +59,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Uni
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[HTTPValidationError, NewUserKey]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[HTTPValidationError, NewUserKey]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

openapi_fal_rest/api/keys/delete_key.py

@@ -29,7 +29,9 @@ def _get_kwargs(
     }
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Union[Any, HTTPValidationError]]:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Optional[Union[Any, HTTPValidationError]]:
     if response.status_code == HTTPStatus.NO_CONTENT:
         response_204 = cast(Any, None)
         return response_204
@@ -43,7 +45,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Uni
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Any, HTTPValidationError]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[Any, HTTPValidationError]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

openapi_fal_rest/api/tokens/create_token.py

@@ -33,7 +33,9 @@ def _get_kwargs(
     }
 
 
-def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Union[HTTPValidationError, str]]:
+def _parse_response(
+    *, client: Client, response: httpx.Response
+) -> Optional[Union[HTTPValidationError, str]]:
     if response.status_code == HTTPStatus.CREATED:
         response_201 = cast(str, response.json())
         return response_201
@@ -47,7 +49,9 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Uni
         return None
 
 
-def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[HTTPValidationError, str]]:
+def _build_response(
+    *, client: Client, response: httpx.Response
+) -> Response[Union[HTTPValidationError, str]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

openapi_fal_rest/api/usage/get_gateway_request_stats_by_time.py

@@ -64,9 +64,18 @@ def _get_kwargs(
 
 def _parse_response(
     *, client: Client, response: httpx.Response
-) -> Optional[Union[GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime, HTTPValidationError]]:
+) -> Optional[
+    Union[
+        GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime,
+        HTTPValidationError,
+    ]
+]:
     if response.status_code == HTTPStatus.OK:
-        response_200 = GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime.from_dict(response.json())
+        response_200 = (
+            GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime.from_dict(
+                response.json()
+            )
+        )
 
         return response_200
     if response.status_code == HTTPStatus.UNPROCESSABLE_ENTITY:
@@ -81,7 +90,12 @@ def _parse_response(
 
 def _build_response(
     *, client: Client, response: httpx.Response
-) -> Response[Union[GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime, HTTPValidationError]]:
+) -> Response[
+    Union[
+        GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime,
+        HTTPValidationError,
+    ]
+]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,
@@ -97,7 +111,12 @@ def sync_detailed(
     end_time: datetime.datetime,
     timeframe: StatsTimeframe,
     app_aliases: Union[Unset, None, List[str]] = UNSET,
-) -> Response[Union[GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime, HTTPValidationError]]:
+) -> Response[
+    Union[
+        GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime,
+        HTTPValidationError,
+    ]
+]:
     """Get Gateway Request Stats By Time
 
     Args:
@@ -137,7 +156,12 @@ def sync(
     end_time: datetime.datetime,
     timeframe: StatsTimeframe,
     app_aliases: Union[Unset, None, List[str]] = UNSET,
-) -> Optional[Union[GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime, HTTPValidationError]]:
+) -> Optional[
+    Union[
+        GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime,
+        HTTPValidationError,
+    ]
+]:
     """Get Gateway Request Stats By Time
 
     Args:
@@ -170,7 +194,12 @@ async def asyncio_detailed(
     end_time: datetime.datetime,
     timeframe: StatsTimeframe,
     app_aliases: Union[Unset, None, List[str]] = UNSET,
-) -> Response[Union[GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime, HTTPValidationError]]:
+) -> Response[
+    Union[
+        GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime,
+        HTTPValidationError,
+    ]
+]:
     """Get Gateway Request Stats By Time
 
     Args:
@@ -208,7 +237,12 @@ async def asyncio(
     end_time: datetime.datetime,
     timeframe: StatsTimeframe,
     app_aliases: Union[Unset, None, List[str]] = UNSET,
-) -> Optional[Union[GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime, HTTPValidationError]]:
+) -> Optional[
+    Union[
+        GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime,
+        HTTPValidationError,
+    ]
+]:
     """Get Gateway Request Stats By Time
 
     Args:

openapi_fal_rest/api/usage/per_machine_usage_details.py

@@ -15,7 +15,9 @@ def _get_kwargs(
     *,
     client: Client,
 ) -> Dict[str, Any]:
-    url = "{}/usage/machine_type/{machine_type}".format(client.base_url, machine_type=machine_type)
+    url = "{}/usage/machine_type/{machine_type}".format(
+        client.base_url, machine_type=machine_type
+    )
 
     headers: Dict[str, str] = client.get_headers()
     cookies: Dict[str, Any] = client.get_cookies()

openapi_fal_rest/models/__init__.py

@@ -12,7 +12,9 @@ from .get_gateway_request_stats_by_time_response_get_gateway_request_stats_by_ti
     GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime,
 )
 from .grouped_usage_detail import GroupedUsageDetail
-from .handle_stripe_webhook_response_handle_stripe_webhook import HandleStripeWebhookResponseHandleStripeWebhook
+from .handle_stripe_webhook_response_handle_stripe_webhook import (
+    HandleStripeWebhookResponseHandleStripeWebhook,
+)
 from .hash_check import HashCheck
 from .http_validation_error import HTTPValidationError
 from .initiate_upload_info import InitiateUploadInfo

openapi_fal_rest/models/body_upload_file.py

@@ -36,7 +36,10 @@ class BodyUploadFile:
 
         field_dict: Dict[str, Any] = {}
         field_dict.update(
-            {key: (None, str(value).encode(), "text/plain") for key, value in self.additional_properties.items()}
+            {
+                key: (None, str(value).encode(), "text/plain")
+                for key, value in self.additional_properties.items()
+            }
         )
         field_dict.update(
             {

openapi_fal_rest/models/body_upload_local_file.py

@@ -36,7 +36,10 @@ class BodyUploadLocalFile:
 
         field_dict: Dict[str, Any] = {}
         field_dict.update(
-            {key: (None, str(value).encode(), "text/plain") for key, value in self.additional_properties.items()}
+            {
+                key: (None, str(value).encode(), "text/plain")
+                for key, value in self.additional_properties.items()
+            }
         )
         field_dict.update(
             {

openapi_fal_rest/models/get_gateway_request_stats_by_time_response_get_gateway_request_stats_by_time.py

@@ -6,14 +6,18 @@ if TYPE_CHECKING:
     from ..models.gateway_stats_by_time import GatewayStatsByTime
 
 
-T = TypeVar("T", bound="GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime")
+T = TypeVar(
+    "T", bound="GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime"
+)
 
 
 @attr.s(auto_attribs=True)
 class GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime:
     """ """
 
-    additional_properties: Dict[str, List["GatewayStatsByTime"]] = attr.ib(init=False, factory=dict)
+    additional_properties: Dict[str, List["GatewayStatsByTime"]] = attr.ib(
+        init=False, factory=dict
+    )
 
     def to_dict(self) -> Dict[str, Any]:
         pass
@@ -35,14 +39,18 @@ class GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime:
         from ..models.gateway_stats_by_time import GatewayStatsByTime
 
         d = src_dict.copy()
-        get_gateway_request_stats_by_time_response_get_gateway_request_stats_by_time = cls()
+        get_gateway_request_stats_by_time_response_get_gateway_request_stats_by_time = (
+            cls()
+        )
 
         additional_properties = {}
         for prop_name, prop_dict in d.items():
             additional_property = []
             _additional_property = prop_dict
             for additional_property_item_data in _additional_property:
-                additional_property_item = GatewayStatsByTime.from_dict(additional_property_item_data)
+                additional_property_item = GatewayStatsByTime.from_dict(
+                    additional_property_item_data
+                )
 
                 additional_property.append(additional_property_item)
 
@@ -51,7 +59,9 @@ class GetGatewayRequestStatsByTimeResponseGetGatewayRequestStatsByTime:
         get_gateway_request_stats_by_time_response_get_gateway_request_stats_by_time.additional_properties = (
             additional_properties
         )
-        return get_gateway_request_stats_by_time_response_get_gateway_request_stats_by_time
+        return (
+            get_gateway_request_stats_by_time_response_get_gateway_request_stats_by_time
+        )
 
     @property
     def additional_keys(self) -> List[str]:

openapi_fal_rest/models/grouped_usage_detail.py

@@ -12,14 +12,14 @@ class GroupedUsageDetail:
         model_id (str):
         machine_type (str):
         request_count (int):
-        avg_duration (float):
+        median_duration (float):
         total_duration (float):
     """
 
     model_id: str
     machine_type: str
     request_count: int
-    avg_duration: float
+    median_duration: float
     total_duration: float
     additional_properties: Dict[str, Any] = attr.ib(init=False, factory=dict)
 
@@ -27,7 +27,7 @@ class GroupedUsageDetail:
         model_id = self.model_id
         machine_type = self.machine_type
         request_count = self.request_count
-        avg_duration = self.avg_duration
+        median_duration = self.median_duration
         total_duration = self.total_duration
 
         field_dict: Dict[str, Any] = {}
@@ -37,7 +37,7 @@ class GroupedUsageDetail:
                 "model_id": model_id,
                 "machine_type": machine_type,
                 "request_count": request_count,
-                "avg_duration": avg_duration,
+                "median_duration": median_duration,
                 "total_duration": total_duration,
             }
         )
@@ -53,7 +53,7 @@ class GroupedUsageDetail:
 
         request_count = d.pop("request_count")
 
-        avg_duration = d.pop("avg_duration")
+        median_duration = d.pop("median_duration")
 
         total_duration = d.pop("total_duration")
 
@@ -61,7 +61,7 @@ class GroupedUsageDetail:
             model_id=model_id,
             machine_type=machine_type,
             request_count=request_count,
-            avg_duration=avg_duration,
+            median_duration=median_duration,
             total_duration=total_duration,
         )