Merge pull request #94 from astuto-ai/pu/update-python-client-2026-04-28

updated models for insights tracker

Author: pranjal-astuto

onelens_backend_client_v2/models.py

@@ -22759,3 +22759,46 @@ class GetS3TicketMetadataResponse(BaseModel):
 class GetS3TicketResponse(BaseModel):
     s3_ticket: S3TicketMixin
     assigned_to_email: Optional[str] = Field(None, title="Assigned To Email")
+
+
+class InsightTrackerState(str, Enum):
+    OPEN = "OPEN"
+    CLOSED = "CLOSED"
+
+
+class CostDetailBreakDownMetrics(BaseModel):
+    cost: float = Field(
+        ..., description="The cost of the cost detail breakdown metrics.", title="Cost"
+    )
+    value: Optional[float] = Field(
+        None,
+        description="The value of the cost detail breakdown metrics.",
+        title="Value",
+    )
+    unit: Optional[str] = Field(
+        None, description="The unit of the cost detail breakdown metrics.", title="Unit"
+    )
+
+
+class InsightTrackerMixin(BaseModel):
+    insight_id: UUID = Field(..., title="Insight Id")
+    state: InsightTrackerState
+
+
+class BulkUpsertInsightsTrackerResponse(BaseModel):
+    upserted_count: int = Field(..., title="Upserted Count")
+
+
+class BulkUpsertInsightsTrackerRequest(BaseModel):
+    tenant_id: UUID = Field(..., title="Tenant Id")
+    items: List[InsightTrackerMixin] = Field(..., title="Items")
+
+
+class S3OptimisationCostDetailsResponse(BaseModel):
+    cost_details_breakdown: Dict[str, Dict[str, CostDetailBreakDownMetrics]] = Field(
+        ..., title="Cost Details Breakdown"
+    )
+    total_cost: float = Field(..., title="Total Cost")
+    total_value: None = Field(..., title="Total Value")
+    other_costs: float = Field(..., title="Other Costs")
+    other_costs_value: None = Field(..., title="Other Costs Value")

onelens_backend_client_v2/rpc/s3_optimisation_service_rpc_handler.py

@@ -5,6 +5,12 @@
 from typing_extensions import Annotated
 
 
+from onelens_backend_client_v2.models import BulkUpsertInsightsTrackerRequest
+
+
+from onelens_backend_client_v2.models import BulkUpsertInsightsTrackerResponse
+
+
 from onelens_backend_client_v2.models import S3BucketDetailsExportAPIRequest
 
 
@@ -17,6 +23,9 @@
 from onelens_backend_client_v2.models import GetS3CostDetailsRequest
 
 
+from onelens_backend_client_v2.models import S3OptimisationCostDetailsResponse
+
+
 from onelens_backend_client_v2.models import S3OptimisationGridDataRequest
 
 
@@ -46,6 +55,127 @@ def __init__(self, api_client=None) -> None:
             api_client = ApiClient.get_default()
         self.api_client = api_client
 
+    @validate_call
+    def bulk_upsert_s3_insights_tracker(
+        self,
+        request: BulkUpsertInsightsTrackerRequest,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)], Annotated[StrictFloat, Field(gt=0)]
+            ],
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> BulkUpsertInsightsTrackerResponse:
+        """Bulk Upsert S3 Insights Tracker
+
+
+
+        :param request: (required)
+        :type request: BulkUpsertInsightsTrackerRequest
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """
+
+        _param = self._bulk_upsert_s3_insights_tracker_serialize(
+            request=request,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            "200": "BulkUpsertInsightsTrackerResponse",
+            "422": "HTTPValidationError",
+        }
+        response_data = self.api_client.call_api(
+            *_param, _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+
+    def _bulk_upsert_s3_insights_tracker_serialize(
+        self,
+        request: BulkUpsertInsightsTrackerRequest,
+        _request_auth,
+        _content_type,
+        _headers,
+        _host_index,
+    ) -> RequestSerialized:
+        _host = None
+
+        _collection_formats: Dict[str, str] = {}
+
+        _path_params: Dict[str, str] = {}
+        _query_params: List[Tuple[str, str]] = []
+        _header_params: Dict[str, Optional[str]] = _headers or {}
+        _form_params: List[Tuple[str, str]] = []
+        _files: Dict[str, Union[str, bytes]] = {}
+        _body_params: Optional[bytes] = None
+
+        # process the body parameter
+        if request is not None:
+            _body_params = request
+
+        # set the HTTP header `Accept`
+        _header_params["Accept"] = self.api_client.select_header_accept(
+            ["application/json"]
+        )
+
+        # set the HTTP header `Content-Type`
+        if _content_type:
+            _header_params["Content-Type"] = _content_type
+        else:
+            _default_content_type = self.api_client.select_header_content_type(
+                ["application/json"]
+            )
+            if _default_content_type is not None:
+                _header_params["Content-Type"] = _default_content_type
+
+        # authentication setting
+        _auth_settings: List[str] = []
+
+        return self.api_client.param_serialize(
+            method="POST",
+            resource_path="/rpc/s3_optimisation_service/bulk_upsert_s3_insights_tracker",
+            path_params=_path_params,
+            query_params=_query_params,
+            header_params=_header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            auth_settings=_auth_settings,
+            collection_formats=_collection_formats,
+            _host=_host,
+            _request_auth=_request_auth,
+        )
+
     @validate_call
     def export_s3_bucket_details(
         self,
@@ -303,7 +433,7 @@ def get_s3_optimisation_cost_details(
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> Any:
+    ) -> S3OptimisationCostDetailsResponse:
         """Get S3 Optimisation Cost Details
 
 
@@ -341,7 +471,7 @@ def get_s3_optimisation_cost_details(
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            "200": "Any",
+            "200": "S3OptimisationCostDetailsResponse",
             "422": "HTTPValidationError",
         }
         response_data = self.api_client.call_api(