diff --git a/.pylintrc b/.pylintrc index 339e71cb..d0498311 100644 --- a/.pylintrc +++ b/.pylintrc @@ -26,6 +26,5 @@ reports=no max-line-length=79 [MASTER] -# Ignore anything inside launch/clientlib (since it's documentation) -ignore=clientlib,api_client +ignore=api_client,openapi_client extension-pkg-whitelist=pydantic diff --git a/launch/api_client/__init__.py b/launch/api_client/__init__.py index f6709b0c..99e0c1e4 100644 --- a/launch/api_client/__init__.py +++ b/launch/api_client/__init__.py @@ -1,30 +1,16 @@ -# coding: utf-8 +import inspect -# flake8: noqa +from pydantic import BaseModel -""" - launch - - No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator) # noqa: E501 - - The version of the OpenAPI document: 1.0.0 - Generated by: https://openapi-generator.tech -""" - -__version__ = "1.0.5" - -# import ApiClient -from launch.api_client.api_client import ApiClient - -# import Configuration -from launch.api_client.configuration import Configuration - -# import exceptions -from launch.api_client.exceptions import ( - ApiAttributeError, - ApiException, - ApiKeyError, - ApiTypeError, - ApiValueError, - OpenApiException, +from launch.api_client import models +from launch.api_client.api_client import ( # noqa F401 + ApiClient, + AsyncApis, + SyncApis, ) + +for model in inspect.getmembers(models, inspect.isclass): + if model[1].__module__ == "launch.api_client.models": + model_class = model[1] + if isinstance(model_class, BaseModel): + model_class.update_forward_refs() diff --git a/launch/api_client/api/__init__.py b/launch/api_client/api/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/launch/api_client/api/default_api.py b/launch/api_client/api/default_api.py new file mode 100644 index 00000000..2acb77b2 --- /dev/null +++ b/launch/api_client/api/default_api.py @@ -0,0 +1,717 @@ +# flake8: noqa E501 +from asyncio import get_event_loop +from typing import TYPE_CHECKING, Awaitable + +from fastapi.encoders import jsonable_encoder + +from launch.api_client import models as m + +if TYPE_CHECKING: + from launch.api_client.api_client import ApiClient + + +class _DefaultApi: + def __init__(self, api_client: "ApiClient"): + self.api_client = api_client + + def _build_for_clone_model_bundle_with_changes_v1_model_bundles_clone_with_changes_post( + self, clone_model_bundle_request: m.CloneModelBundleRequest + ) -> Awaitable[m.CreateModelBundleResponse]: + """ + Creates a ModelBundle by cloning an existing one and then applying changes on top. + """ + body = jsonable_encoder(clone_model_bundle_request) + + return self.api_client.request( + type_=m.CreateModelBundleResponse, + method="POST", + url="/v1/model-bundles/clone-with-changes", + json=body, + ) + + def _build_for_create_async_inference_task_v1_async_tasks_post( + self, model_endpoint_id: str, endpoint_predict_request: m.EndpointPredictRequest + ) -> Awaitable[m.CreateAsyncTaskResponse]: + """ + Runs an async inference prediction. + """ + query_params = {"model_endpoint_id": str(model_endpoint_id)} + + body = jsonable_encoder(endpoint_predict_request) + + return self.api_client.request( + type_=m.CreateAsyncTaskResponse, + method="POST", + url="/v1/async-tasks", + params=query_params, + json=body, + ) + + def _build_for_create_batch_job_v1_batch_jobs_post( + self, create_batch_job_request: m.CreateBatchJobRequest + ) -> Awaitable[m.CreateBatchJobResponse]: + """ + Runs a sync inference prediction. + """ + body = jsonable_encoder(create_batch_job_request) + + return self.api_client.request(type_=m.CreateBatchJobResponse, method="POST", url="/v1/batch-jobs", json=body) + + def _build_for_create_model_bundle_v1_model_bundles_post( + self, create_model_bundle_request: m.CreateModelBundleRequest + ) -> Awaitable[m.CreateModelBundleResponse]: + """ + Creates a ModelBundle for the current user. + """ + body = jsonable_encoder(create_model_bundle_request) + + return self.api_client.request( + type_=m.CreateModelBundleResponse, method="POST", url="/v1/model-bundles", json=body + ) + + def _build_for_create_model_endpoint_v1_model_endpoints_post( + self, create_model_endpoint_request: m.CreateModelEndpointRequest + ) -> Awaitable[m.CreateModelEndpointResponse]: + """ + Creates a Model for the current user. + """ + body = jsonable_encoder(create_model_endpoint_request) + + return self.api_client.request( + type_=m.CreateModelEndpointResponse, method="POST", url="/v1/model-endpoints", json=body + ) + + def _build_for_create_sync_inference_task_v1_sync_tasks_post( + self, model_endpoint_id: str, endpoint_predict_request: m.EndpointPredictRequest + ) -> Awaitable[m.SyncEndpointPredictResponse]: + """ + Runs a sync inference prediction. + """ + query_params = {"model_endpoint_id": str(model_endpoint_id)} + + body = jsonable_encoder(endpoint_predict_request) + + return self.api_client.request( + type_=m.SyncEndpointPredictResponse, + method="POST", + url="/v1/sync-tasks", + params=query_params, + json=body, + ) + + def _build_for_delete_model_endpoint_v1_model_endpoints_model_endpoint_id_delete( + self, model_endpoint_id: str + ) -> Awaitable[m.DeleteModelEndpointResponse]: + """ + Lists the Models owned by the current owner. + """ + path_params = {"model_endpoint_id": str(model_endpoint_id)} + + return self.api_client.request( + type_=m.DeleteModelEndpointResponse, + method="DELETE", + url="/v1/model-endpoints/{model_endpoint_id}", + path_params=path_params, + ) + + def _build_for_get_async_inference_task_v1_async_tasks_task_id_get( + self, task_id: str + ) -> Awaitable[m.GetAsyncTaskResponse]: + """ + Gets the status of an async inference task. + """ + path_params = {"task_id": str(task_id)} + + return self.api_client.request( + type_=m.GetAsyncTaskResponse, + method="GET", + url="/v1/async-tasks/{task_id}", + path_params=path_params, + ) + + def _build_for_get_batch_job_v1_batch_jobs_batch_job_id_get( + self, batch_job_id: str + ) -> Awaitable[m.GetBatchJobResponse]: + """ + Runs a sync inference prediction. + """ + path_params = {"batch_job_id": str(batch_job_id)} + + return self.api_client.request( + type_=m.GetBatchJobResponse, + method="GET", + url="/v1/batch-jobs/{batch_job_id}", + path_params=path_params, + ) + + def _build_for_get_latest_model_bundle_v1_model_bundles_latest_get( + self, model_name: str + ) -> Awaitable[m.ModelBundleResponse]: + """ + Gets the the latest Model Bundle with the given name owned by the current owner. + """ + query_params = {"model_name": str(model_name)} + + return self.api_client.request( + type_=m.ModelBundleResponse, + method="GET", + url="/v1/model-bundles/latest", + params=query_params, + ) + + def _build_for_get_model_bundle_v1_model_bundles_model_bundle_id_get( + self, model_bundle_id: str + ) -> Awaitable[m.ModelBundleResponse]: + """ + Gets the details for a given ModelBundle owned by the current owner. + """ + path_params = {"model_bundle_id": str(model_bundle_id)} + + return self.api_client.request( + type_=m.ModelBundleResponse, + method="GET", + url="/v1/model-bundles/{model_bundle_id}", + path_params=path_params, + ) + + def _build_for_get_model_endpoint_v1_model_endpoints_model_endpoint_id_get( + self, model_endpoint_id: str + ) -> Awaitable[m.GetModelEndpointResponse]: + """ + Lists the Models owned by the current owner. + """ + path_params = {"model_endpoint_id": str(model_endpoint_id)} + + return self.api_client.request( + type_=m.GetModelEndpointResponse, + method="GET", + url="/v1/model-endpoints/{model_endpoint_id}", + path_params=path_params, + ) + + def _build_for_get_model_endpoints_api_v1_model_endpoints_api_get( + self, + ) -> Awaitable[m.Any]: + """ + Shows the API of the Model Endpoints owned by the current owner. + """ + return self.api_client.request( + type_=m.Any, + method="GET", + url="/v1/model-endpoints-api", + ) + + def _build_for_get_model_endpoints_schema_v1_model_endpoints_schema_json_get( + self, + ) -> Awaitable[m.Any]: + """ + Lists the schemas of the Model Endpoints owned by the current owner. + """ + return self.api_client.request( + type_=m.Any, + method="GET", + url="/v1/model-endpoints-schema.json", + ) + + def _build_for_healthcheck_healthcheck_get( + self, + ) -> Awaitable[m.Any]: + """ + Returns 200 if the app is healthy. + """ + return self.api_client.request( + type_=m.Any, + method="GET", + url="/healthcheck", + ) + + def _build_for_healthcheck_healthz_get( + self, + ) -> Awaitable[m.Any]: + """ + Returns 200 if the app is healthy. + """ + return self.api_client.request( + type_=m.Any, + method="GET", + url="/healthz", + ) + + def _build_for_healthcheck_readyz_get( + self, + ) -> Awaitable[m.Any]: + """ + Returns 200 if the app is healthy. + """ + return self.api_client.request( + type_=m.Any, + method="GET", + url="/readyz", + ) + + def _build_for_list_model_bundles_v1_model_bundles_get( + self, model_name: str = None, order_by: m.ModelBundleOrderBy = None + ) -> Awaitable[m.ListModelBundlesResponse]: + """ + Lists the ModelBundles owned by the current owner. + """ + query_params = {} + if model_name is not None: + query_params["model_name"] = str(model_name) + if order_by is not None: + query_params["order_by"] = str(order_by) + + return self.api_client.request( + type_=m.ListModelBundlesResponse, + method="GET", + url="/v1/model-bundles", + params=query_params, + ) + + def _build_for_list_model_endpoints_v1_model_endpoints_get( + self, name: str = None, order_by: m.ModelEndpointOrderBy = None + ) -> Awaitable[m.ListModelEndpointsResponse]: + """ + Lists the Models owned by the current owner. + """ + query_params = {} + if name is not None: + query_params["name"] = str(name) + if order_by is not None: + query_params["order_by"] = str(order_by) + + return self.api_client.request( + type_=m.ListModelEndpointsResponse, + method="GET", + url="/v1/model-endpoints", + params=query_params, + ) + + def _build_for_update_batch_job_v1_batch_jobs_batch_job_id_put( + self, batch_job_id: str, update_batch_job_request: m.UpdateBatchJobRequest + ) -> Awaitable[m.UpdateBatchJobResponse]: + """ + Runs a sync inference prediction. + """ + path_params = {"batch_job_id": str(batch_job_id)} + + body = jsonable_encoder(update_batch_job_request) + + return self.api_client.request( + type_=m.UpdateBatchJobResponse, + method="PUT", + url="/v1/batch-jobs/{batch_job_id}", + path_params=path_params, + json=body, + ) + + def _build_for_update_model_endpoint_v1_model_endpoints_model_endpoint_id_put( + self, model_endpoint_id: str, update_model_endpoint_request: m.UpdateModelEndpointRequest + ) -> Awaitable[m.UpdateModelEndpointResponse]: + """ + Lists the Models owned by the current owner. + """ + path_params = {"model_endpoint_id": str(model_endpoint_id)} + + body = jsonable_encoder(update_model_endpoint_request) + + return self.api_client.request( + type_=m.UpdateModelEndpointResponse, + method="PUT", + url="/v1/model-endpoints/{model_endpoint_id}", + path_params=path_params, + json=body, + ) + + +class AsyncDefaultApi(_DefaultApi): + async def clone_model_bundle_with_changes_v1_model_bundles_clone_with_changes_post( + self, clone_model_bundle_request: m.CloneModelBundleRequest + ) -> m.CreateModelBundleResponse: + """ + Creates a ModelBundle by cloning an existing one and then applying changes on top. + """ + return await self._build_for_clone_model_bundle_with_changes_v1_model_bundles_clone_with_changes_post( + clone_model_bundle_request=clone_model_bundle_request + ) + + async def create_async_inference_task_v1_async_tasks_post( + self, model_endpoint_id: str, endpoint_predict_request: m.EndpointPredictRequest + ) -> m.CreateAsyncTaskResponse: + """ + Runs an async inference prediction. + """ + return await self._build_for_create_async_inference_task_v1_async_tasks_post( + model_endpoint_id=model_endpoint_id, endpoint_predict_request=endpoint_predict_request + ) + + async def create_batch_job_v1_batch_jobs_post( + self, create_batch_job_request: m.CreateBatchJobRequest + ) -> m.CreateBatchJobResponse: + """ + Runs a sync inference prediction. + """ + return await self._build_for_create_batch_job_v1_batch_jobs_post( + create_batch_job_request=create_batch_job_request + ) + + async def create_model_bundle_v1_model_bundles_post( + self, create_model_bundle_request: m.CreateModelBundleRequest + ) -> m.CreateModelBundleResponse: + """ + Creates a ModelBundle for the current user. + """ + return await self._build_for_create_model_bundle_v1_model_bundles_post( + create_model_bundle_request=create_model_bundle_request + ) + + async def create_model_endpoint_v1_model_endpoints_post( + self, create_model_endpoint_request: m.CreateModelEndpointRequest + ) -> m.CreateModelEndpointResponse: + """ + Creates a Model for the current user. + """ + return await self._build_for_create_model_endpoint_v1_model_endpoints_post( + create_model_endpoint_request=create_model_endpoint_request + ) + + async def create_sync_inference_task_v1_sync_tasks_post( + self, model_endpoint_id: str, endpoint_predict_request: m.EndpointPredictRequest + ) -> m.SyncEndpointPredictResponse: + """ + Runs a sync inference prediction. + """ + return await self._build_for_create_sync_inference_task_v1_sync_tasks_post( + model_endpoint_id=model_endpoint_id, endpoint_predict_request=endpoint_predict_request + ) + + async def delete_model_endpoint_v1_model_endpoints_model_endpoint_id_delete( + self, model_endpoint_id: str + ) -> m.DeleteModelEndpointResponse: + """ + Lists the Models owned by the current owner. + """ + return await self._build_for_delete_model_endpoint_v1_model_endpoints_model_endpoint_id_delete( + model_endpoint_id=model_endpoint_id + ) + + async def get_async_inference_task_v1_async_tasks_task_id_get(self, task_id: str) -> m.GetAsyncTaskResponse: + """ + Gets the status of an async inference task. + """ + return await self._build_for_get_async_inference_task_v1_async_tasks_task_id_get(task_id=task_id) + + async def get_batch_job_v1_batch_jobs_batch_job_id_get(self, batch_job_id: str) -> m.GetBatchJobResponse: + """ + Runs a sync inference prediction. + """ + return await self._build_for_get_batch_job_v1_batch_jobs_batch_job_id_get(batch_job_id=batch_job_id) + + async def get_latest_model_bundle_v1_model_bundles_latest_get(self, model_name: str) -> m.ModelBundleResponse: + """ + Gets the the latest Model Bundle with the given name owned by the current owner. + """ + return await self._build_for_get_latest_model_bundle_v1_model_bundles_latest_get(model_name=model_name) + + async def get_model_bundle_v1_model_bundles_model_bundle_id_get( + self, model_bundle_id: str + ) -> m.ModelBundleResponse: + """ + Gets the details for a given ModelBundle owned by the current owner. + """ + return await self._build_for_get_model_bundle_v1_model_bundles_model_bundle_id_get( + model_bundle_id=model_bundle_id + ) + + async def get_model_endpoint_v1_model_endpoints_model_endpoint_id_get( + self, model_endpoint_id: str + ) -> m.GetModelEndpointResponse: + """ + Lists the Models owned by the current owner. + """ + return await self._build_for_get_model_endpoint_v1_model_endpoints_model_endpoint_id_get( + model_endpoint_id=model_endpoint_id + ) + + async def get_model_endpoints_api_v1_model_endpoints_api_get( + self, + ) -> m.Any: + """ + Shows the API of the Model Endpoints owned by the current owner. + """ + return await self._build_for_get_model_endpoints_api_v1_model_endpoints_api_get() + + async def get_model_endpoints_schema_v1_model_endpoints_schema_json_get( + self, + ) -> m.Any: + """ + Lists the schemas of the Model Endpoints owned by the current owner. + """ + return await self._build_for_get_model_endpoints_schema_v1_model_endpoints_schema_json_get() + + async def healthcheck_healthcheck_get( + self, + ) -> m.Any: + """ + Returns 200 if the app is healthy. + """ + return await self._build_for_healthcheck_healthcheck_get() + + async def healthcheck_healthz_get( + self, + ) -> m.Any: + """ + Returns 200 if the app is healthy. + """ + return await self._build_for_healthcheck_healthz_get() + + async def healthcheck_readyz_get( + self, + ) -> m.Any: + """ + Returns 200 if the app is healthy. + """ + return await self._build_for_healthcheck_readyz_get() + + async def list_model_bundles_v1_model_bundles_get( + self, model_name: str = None, order_by: m.ModelBundleOrderBy = None + ) -> m.ListModelBundlesResponse: + """ + Lists the ModelBundles owned by the current owner. + """ + return await self._build_for_list_model_bundles_v1_model_bundles_get(model_name=model_name, order_by=order_by) + + async def list_model_endpoints_v1_model_endpoints_get( + self, name: str = None, order_by: m.ModelEndpointOrderBy = None + ) -> m.ListModelEndpointsResponse: + """ + Lists the Models owned by the current owner. + """ + return await self._build_for_list_model_endpoints_v1_model_endpoints_get(name=name, order_by=order_by) + + async def update_batch_job_v1_batch_jobs_batch_job_id_put( + self, batch_job_id: str, update_batch_job_request: m.UpdateBatchJobRequest + ) -> m.UpdateBatchJobResponse: + """ + Runs a sync inference prediction. + """ + return await self._build_for_update_batch_job_v1_batch_jobs_batch_job_id_put( + batch_job_id=batch_job_id, update_batch_job_request=update_batch_job_request + ) + + async def update_model_endpoint_v1_model_endpoints_model_endpoint_id_put( + self, model_endpoint_id: str, update_model_endpoint_request: m.UpdateModelEndpointRequest + ) -> m.UpdateModelEndpointResponse: + """ + Lists the Models owned by the current owner. + """ + return await self._build_for_update_model_endpoint_v1_model_endpoints_model_endpoint_id_put( + model_endpoint_id=model_endpoint_id, + update_model_endpoint_request=update_model_endpoint_request, + ) + + +class SyncDefaultApi(_DefaultApi): + def clone_model_bundle_with_changes_v1_model_bundles_clone_with_changes_post( + self, clone_model_bundle_request: m.CloneModelBundleRequest + ) -> m.CreateModelBundleResponse: + """ + Creates a ModelBundle by cloning an existing one and then applying changes on top. + """ + coroutine = self._build_for_clone_model_bundle_with_changes_v1_model_bundles_clone_with_changes_post( + clone_model_bundle_request=clone_model_bundle_request + ) + return get_event_loop().run_until_complete(coroutine) + + def create_async_inference_task_v1_async_tasks_post( + self, model_endpoint_id: str, endpoint_predict_request: m.EndpointPredictRequest + ) -> m.CreateAsyncTaskResponse: + """ + Runs an async inference prediction. + """ + coroutine = self._build_for_create_async_inference_task_v1_async_tasks_post( + model_endpoint_id=model_endpoint_id, endpoint_predict_request=endpoint_predict_request + ) + return get_event_loop().run_until_complete(coroutine) + + def create_batch_job_v1_batch_jobs_post( + self, create_batch_job_request: m.CreateBatchJobRequest + ) -> m.CreateBatchJobResponse: + """ + Runs a sync inference prediction. + """ + coroutine = self._build_for_create_batch_job_v1_batch_jobs_post( + create_batch_job_request=create_batch_job_request + ) + return get_event_loop().run_until_complete(coroutine) + + def create_model_bundle_v1_model_bundles_post( + self, create_model_bundle_request: m.CreateModelBundleRequest + ) -> m.CreateModelBundleResponse: + """ + Creates a ModelBundle for the current user. + """ + coroutine = self._build_for_create_model_bundle_v1_model_bundles_post( + create_model_bundle_request=create_model_bundle_request + ) + return get_event_loop().run_until_complete(coroutine) + + def create_model_endpoint_v1_model_endpoints_post( + self, create_model_endpoint_request: m.CreateModelEndpointRequest + ) -> m.CreateModelEndpointResponse: + """ + Creates a Model for the current user. + """ + coroutine = self._build_for_create_model_endpoint_v1_model_endpoints_post( + create_model_endpoint_request=create_model_endpoint_request + ) + return get_event_loop().run_until_complete(coroutine) + + def create_sync_inference_task_v1_sync_tasks_post( + self, model_endpoint_id: str, endpoint_predict_request: m.EndpointPredictRequest + ) -> m.SyncEndpointPredictResponse: + """ + Runs a sync inference prediction. + """ + coroutine = self._build_for_create_sync_inference_task_v1_sync_tasks_post( + model_endpoint_id=model_endpoint_id, endpoint_predict_request=endpoint_predict_request + ) + return get_event_loop().run_until_complete(coroutine) + + def delete_model_endpoint_v1_model_endpoints_model_endpoint_id_delete( + self, model_endpoint_id: str + ) -> m.DeleteModelEndpointResponse: + """ + Lists the Models owned by the current owner. + """ + coroutine = self._build_for_delete_model_endpoint_v1_model_endpoints_model_endpoint_id_delete( + model_endpoint_id=model_endpoint_id + ) + return get_event_loop().run_until_complete(coroutine) + + def get_async_inference_task_v1_async_tasks_task_id_get(self, task_id: str) -> m.GetAsyncTaskResponse: + """ + Gets the status of an async inference task. + """ + coroutine = self._build_for_get_async_inference_task_v1_async_tasks_task_id_get(task_id=task_id) + return get_event_loop().run_until_complete(coroutine) + + def get_batch_job_v1_batch_jobs_batch_job_id_get(self, batch_job_id: str) -> m.GetBatchJobResponse: + """ + Runs a sync inference prediction. + """ + coroutine = self._build_for_get_batch_job_v1_batch_jobs_batch_job_id_get(batch_job_id=batch_job_id) + return get_event_loop().run_until_complete(coroutine) + + def get_latest_model_bundle_v1_model_bundles_latest_get(self, model_name: str) -> m.ModelBundleResponse: + """ + Gets the the latest Model Bundle with the given name owned by the current owner. + """ + coroutine = self._build_for_get_latest_model_bundle_v1_model_bundles_latest_get(model_name=model_name) + return get_event_loop().run_until_complete(coroutine) + + def get_model_bundle_v1_model_bundles_model_bundle_id_get(self, model_bundle_id: str) -> m.ModelBundleResponse: + """ + Gets the details for a given ModelBundle owned by the current owner. + """ + coroutine = self._build_for_get_model_bundle_v1_model_bundles_model_bundle_id_get( + model_bundle_id=model_bundle_id + ) + return get_event_loop().run_until_complete(coroutine) + + def get_model_endpoint_v1_model_endpoints_model_endpoint_id_get( + self, model_endpoint_id: str + ) -> m.GetModelEndpointResponse: + """ + Lists the Models owned by the current owner. + """ + coroutine = self._build_for_get_model_endpoint_v1_model_endpoints_model_endpoint_id_get( + model_endpoint_id=model_endpoint_id + ) + return get_event_loop().run_until_complete(coroutine) + + def get_model_endpoints_api_v1_model_endpoints_api_get( + self, + ) -> m.Any: + """ + Shows the API of the Model Endpoints owned by the current owner. + """ + coroutine = self._build_for_get_model_endpoints_api_v1_model_endpoints_api_get() + return get_event_loop().run_until_complete(coroutine) + + def get_model_endpoints_schema_v1_model_endpoints_schema_json_get( + self, + ) -> m.Any: + """ + Lists the schemas of the Model Endpoints owned by the current owner. + """ + coroutine = self._build_for_get_model_endpoints_schema_v1_model_endpoints_schema_json_get() + return get_event_loop().run_until_complete(coroutine) + + def healthcheck_healthcheck_get( + self, + ) -> m.Any: + """ + Returns 200 if the app is healthy. + """ + coroutine = self._build_for_healthcheck_healthcheck_get() + return get_event_loop().run_until_complete(coroutine) + + def healthcheck_healthz_get( + self, + ) -> m.Any: + """ + Returns 200 if the app is healthy. + """ + coroutine = self._build_for_healthcheck_healthz_get() + return get_event_loop().run_until_complete(coroutine) + + def healthcheck_readyz_get( + self, + ) -> m.Any: + """ + Returns 200 if the app is healthy. + """ + coroutine = self._build_for_healthcheck_readyz_get() + return get_event_loop().run_until_complete(coroutine) + + def list_model_bundles_v1_model_bundles_get( + self, model_name: str = None, order_by: m.ModelBundleOrderBy = None + ) -> m.ListModelBundlesResponse: + """ + Lists the ModelBundles owned by the current owner. + """ + coroutine = self._build_for_list_model_bundles_v1_model_bundles_get(model_name=model_name, order_by=order_by) + return get_event_loop().run_until_complete(coroutine) + + def list_model_endpoints_v1_model_endpoints_get( + self, name: str = None, order_by: m.ModelEndpointOrderBy = None + ) -> m.ListModelEndpointsResponse: + """ + Lists the Models owned by the current owner. + """ + coroutine = self._build_for_list_model_endpoints_v1_model_endpoints_get(name=name, order_by=order_by) + return get_event_loop().run_until_complete(coroutine) + + def update_batch_job_v1_batch_jobs_batch_job_id_put( + self, batch_job_id: str, update_batch_job_request: m.UpdateBatchJobRequest + ) -> m.UpdateBatchJobResponse: + """ + Runs a sync inference prediction. + """ + coroutine = self._build_for_update_batch_job_v1_batch_jobs_batch_job_id_put( + batch_job_id=batch_job_id, update_batch_job_request=update_batch_job_request + ) + return get_event_loop().run_until_complete(coroutine) + + def update_model_endpoint_v1_model_endpoints_model_endpoint_id_put( + self, model_endpoint_id: str, update_model_endpoint_request: m.UpdateModelEndpointRequest + ) -> m.UpdateModelEndpointResponse: + """ + Lists the Models owned by the current owner. + """ + coroutine = self._build_for_update_model_endpoint_v1_model_endpoints_model_endpoint_id_put( + model_endpoint_id=model_endpoint_id, + update_model_endpoint_request=update_model_endpoint_request, + ) + return get_event_loop().run_until_complete(coroutine) diff --git a/launch/api_client/api_client.py b/launch/api_client/api_client.py index 61d57dbd..181745a4 100644 --- a/launch/api_client/api_client.py +++ b/launch/api_client/api_client.py @@ -1,1543 +1,133 @@ -# coding: utf-8 -""" - launch - - No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator) # noqa: E501 - - The version of the OpenAPI document: 1.0.0 - Generated by: https://openapi-generator.tech -""" - -import atexit -import email -import enum -import io -import json -import os -import re -import tempfile -import typing -from dataclasses import dataclass -from decimal import Decimal -from multiprocessing.pool import ThreadPool -from urllib.parse import quote, urlparse - -import frozendict -import typing_extensions -import urllib3 -from urllib3._collections import HTTPHeaderDict -from urllib3.fields import RequestField as RequestFieldBase - -from launch.api_client import rest -from launch.api_client.configuration import Configuration -from launch.api_client.exceptions import ApiTypeError, ApiValueError -from launch.api_client.schemas import ( - BinarySchema, - BoolClass, - FileIO, - NoneClass, - Schema, - Unset, - date, - datetime, - none_type, - unset, +from asyncio import get_event_loop +from typing import ( + Any, + Awaitable, + Callable, + Dict, + Generic, + Type, + TypeVar, + overload, ) +from httpx import AsyncClient, Request, Response +from pydantic import ValidationError, parse_obj_as -class RequestField(RequestFieldBase): - def __eq__(self, other): - if not isinstance(other, RequestField): - return False - return self.__dict__ == other.__dict__ - - -class JSONEncoder(json.JSONEncoder): - compact_separators = (",", ":") - - def default(self, obj): - if isinstance(obj, str): - return str(obj) - elif isinstance(obj, float): - return float(obj) - elif isinstance(obj, int): - return int(obj) - elif isinstance(obj, Decimal): - if obj.as_tuple().exponent >= 0: - return int(obj) - return float(obj) - elif isinstance(obj, NoneClass): - return None - elif isinstance(obj, BoolClass): - return bool(obj) - elif isinstance(obj, (dict, frozendict.frozendict)): - return {key: self.default(val) for key, val in obj.items()} - elif isinstance(obj, (list, tuple)): - return [self.default(item) for item in obj] - raise ApiValueError("Unable to prepare type {} for serialization".format(obj.__class__.__name__)) - - -class ParameterInType(enum.Enum): - QUERY = "query" - HEADER = "header" - PATH = "path" - COOKIE = "cookie" - - -class ParameterStyle(enum.Enum): - MATRIX = "matrix" - LABEL = "label" - FORM = "form" - SIMPLE = "simple" - SPACE_DELIMITED = "spaceDelimited" - PIPE_DELIMITED = "pipeDelimited" - DEEP_OBJECT = "deepObject" - - -class PrefixSeparatorIterator: - # A class to store prefixes and separators for rfc6570 expansions - - def __init__(self, prefix: str, separator: str): - self.prefix = prefix - self.separator = separator - self.first = True - if separator in {".", "|", "%20"}: - item_separator = separator - else: - item_separator = "," - self.item_separator = item_separator - - def __iter__(self): - return self - - def __next__(self): - if self.first: - self.first = False - return self.prefix - return self.separator - - -class ParameterSerializerBase: - @classmethod - def _get_default_explode(cls, style: ParameterStyle) -> bool: - return False - - @staticmethod - def __ref6570_item_value(in_data: typing.Any, percent_encode: bool): - """ - Get representation if str/float/int/None/items in list/ values in dict - None is returned if an item is undefined, use cases are value= - - None - - [] - - {} - - [None, None None] - - {'a': None, 'b': None} - """ - if type(in_data) in {str, float, int}: - if percent_encode: - return quote(str(in_data)) - return str(in_data) - elif isinstance(in_data, none_type): - # ignored by the expansion process https://datatracker.ietf.org/doc/html/rfc6570#section-3.2.1 - return None - elif isinstance(in_data, list) and not in_data: - # ignored by the expansion process https://datatracker.ietf.org/doc/html/rfc6570#section-3.2.1 - return None - elif isinstance(in_data, dict) and not in_data: - # ignored by the expansion process https://datatracker.ietf.org/doc/html/rfc6570#section-3.2.1 - return None - raise ApiValueError("Unable to generate a ref6570 item representation of {}".format(in_data)) - - @staticmethod - def _to_dict(name: str, value: str): - return {name: value} - - @classmethod - def __ref6570_str_float_int_expansion( - cls, - variable_name: str, - in_data: typing.Any, - explode: bool, - percent_encode: bool, - prefix_separator_iterator: PrefixSeparatorIterator, - var_name_piece: str, - named_parameter_expansion: bool, - ) -> str: - item_value = cls.__ref6570_item_value(in_data, percent_encode) - if item_value is None or (item_value == "" and prefix_separator_iterator.separator == ";"): - return next(prefix_separator_iterator) + var_name_piece - value_pair_equals = "=" if named_parameter_expansion else "" - return next(prefix_separator_iterator) + var_name_piece + value_pair_equals + item_value - - @classmethod - def __ref6570_list_expansion( - cls, - variable_name: str, - in_data: typing.Any, - explode: bool, - percent_encode: bool, - prefix_separator_iterator: PrefixSeparatorIterator, - var_name_piece: str, - named_parameter_expansion: bool, - ) -> str: - item_values = [cls.__ref6570_item_value(v, percent_encode) for v in in_data] - item_values = [v for v in item_values if v is not None] - if not item_values: - # ignored by the expansion process https://datatracker.ietf.org/doc/html/rfc6570#section-3.2.1 - return "" - value_pair_equals = "=" if named_parameter_expansion else "" - if not explode: - return ( - next(prefix_separator_iterator) - + var_name_piece - + value_pair_equals - + prefix_separator_iterator.item_separator.join(item_values) - ) - # exploded - return next(prefix_separator_iterator) + next(prefix_separator_iterator).join( - [var_name_piece + value_pair_equals + val for val in item_values] - ) - - @classmethod - def __ref6570_dict_expansion( - cls, - variable_name: str, - in_data: typing.Any, - explode: bool, - percent_encode: bool, - prefix_separator_iterator: PrefixSeparatorIterator, - var_name_piece: str, - named_parameter_expansion: bool, - ) -> str: - in_data_transformed = {key: cls.__ref6570_item_value(val, percent_encode) for key, val in in_data.items()} - in_data_transformed = {key: val for key, val in in_data_transformed.items() if val is not None} - if not in_data_transformed: - # ignored by the expansion process https://datatracker.ietf.org/doc/html/rfc6570#section-3.2.1 - return "" - value_pair_equals = "=" if named_parameter_expansion else "" - if not explode: - return ( - next(prefix_separator_iterator) - + var_name_piece - + value_pair_equals - + prefix_separator_iterator.item_separator.join( - prefix_separator_iterator.item_separator.join(item_pair) - for item_pair in in_data_transformed.items() - ) - ) - # exploded - return next(prefix_separator_iterator) + next(prefix_separator_iterator).join( - [key + "=" + val for key, val in in_data_transformed.items()] - ) - - @classmethod - def _ref6570_expansion( - cls, - variable_name: str, - in_data: typing.Any, - explode: bool, - percent_encode: bool, - prefix_separator_iterator: PrefixSeparatorIterator, - ) -> str: - """ - Separator is for separate variables like dict with explode true, not for array item separation - """ - named_parameter_expansion = prefix_separator_iterator.separator in { - "&", - ";", - } - var_name_piece = variable_name if named_parameter_expansion else "" - if type(in_data) in {str, float, int}: - return cls.__ref6570_str_float_int_expansion( - variable_name, - in_data, - explode, - percent_encode, - prefix_separator_iterator, - var_name_piece, - named_parameter_expansion, - ) - elif isinstance(in_data, none_type): - # ignored by the expansion process https://datatracker.ietf.org/doc/html/rfc6570#section-3.2.1 - return "" - elif isinstance(in_data, list): - return cls.__ref6570_list_expansion( - variable_name, - in_data, - explode, - percent_encode, - prefix_separator_iterator, - var_name_piece, - named_parameter_expansion, - ) - elif isinstance(in_data, dict): - return cls.__ref6570_dict_expansion( - variable_name, - in_data, - explode, - percent_encode, - prefix_separator_iterator, - var_name_piece, - named_parameter_expansion, - ) - # bool, bytes, etc - raise ApiValueError("Unable to generate a ref6570 representation of {}".format(in_data)) - - -class StyleFormSerializer(ParameterSerializerBase): - @classmethod - def _get_default_explode(cls, style: ParameterStyle) -> bool: - if style is ParameterStyle.FORM: - return True - return super()._get_default_explode(style) - - def _serialize_form( - self, - in_data: typing.Union[None, int, float, str, bool, dict, list], - name: str, - explode: bool, - percent_encode: bool, - prefix_separator_iterator: typing.Optional[PrefixSeparatorIterator] = None, - ) -> str: - if prefix_separator_iterator is None: - prefix_separator_iterator = PrefixSeparatorIterator("", "&") - return self._ref6570_expansion( - variable_name=name, - in_data=in_data, - explode=explode, - percent_encode=percent_encode, - prefix_separator_iterator=prefix_separator_iterator, - ) - - -class StyleSimpleSerializer(ParameterSerializerBase): - def _serialize_simple( - self, - in_data: typing.Union[None, int, float, str, bool, dict, list], - name: str, - explode: bool, - percent_encode: bool, - ) -> str: - prefix_separator_iterator = PrefixSeparatorIterator("", ",") - return self._ref6570_expansion( - variable_name=name, - in_data=in_data, - explode=explode, - percent_encode=percent_encode, - prefix_separator_iterator=prefix_separator_iterator, - ) - - -class JSONDetector: - """ - Works for: - application/json - application/json; charset=UTF-8 - application/json-patch+json - application/geo+json - """ - - __json_content_type_pattern = re.compile("application/[^+]*[+]?(json);?.*") - - @classmethod - def _content_type_is_json(cls, content_type: str) -> bool: - if cls.__json_content_type_pattern.match(content_type): - return True - return False - - -@dataclass -class ParameterBase(JSONDetector): - name: str - in_type: ParameterInType - required: bool - style: typing.Optional[ParameterStyle] - explode: typing.Optional[bool] - allow_reserved: typing.Optional[bool] - schema: typing.Optional[typing.Type[Schema]] - content: typing.Optional[typing.Dict[str, typing.Type[Schema]]] - - __style_to_in_type = { - ParameterStyle.MATRIX: {ParameterInType.PATH}, - ParameterStyle.LABEL: {ParameterInType.PATH}, - ParameterStyle.FORM: {ParameterInType.QUERY, ParameterInType.COOKIE}, - ParameterStyle.SIMPLE: {ParameterInType.PATH, ParameterInType.HEADER}, - ParameterStyle.SPACE_DELIMITED: {ParameterInType.QUERY}, - ParameterStyle.PIPE_DELIMITED: {ParameterInType.QUERY}, - ParameterStyle.DEEP_OBJECT: {ParameterInType.QUERY}, - } - __in_type_to_default_style = { - ParameterInType.QUERY: ParameterStyle.FORM, - ParameterInType.PATH: ParameterStyle.SIMPLE, - ParameterInType.HEADER: ParameterStyle.SIMPLE, - ParameterInType.COOKIE: ParameterStyle.FORM, - } - __disallowed_header_names = {"Accept", "Content-Type", "Authorization"} - _json_encoder = JSONEncoder() - - @classmethod - def __verify_style_to_in_type(cls, style: typing.Optional[ParameterStyle], in_type: ParameterInType): - if style is None: - return - in_type_set = cls.__style_to_in_type[style] - if in_type not in in_type_set: - raise ValueError( - "Invalid style and in_type combination. For style={} only in_type={} are allowed".format( - style, in_type_set - ) - ) - - def __init__( - self, - name: str, - in_type: ParameterInType, - required: bool = False, - style: typing.Optional[ParameterStyle] = None, - explode: bool = False, - allow_reserved: typing.Optional[bool] = None, - schema: typing.Optional[typing.Type[Schema]] = None, - content: typing.Optional[typing.Dict[str, typing.Type[Schema]]] = None, - ): - if schema is None and content is None: - raise ValueError("Value missing; Pass in either schema or content") - if schema and content: - raise ValueError("Too many values provided. Both schema and content were provided. Only one may be input") - if name in self.__disallowed_header_names and in_type is ParameterInType.HEADER: - raise ValueError("Invalid name, name may not be one of {}".format(self.__disallowed_header_names)) - self.__verify_style_to_in_type(style, in_type) - if content is None and style is None: - style = self.__in_type_to_default_style[in_type] - if content is not None and in_type in self.__in_type_to_default_style and len(content) != 1: - raise ValueError("Invalid content length, content length must equal 1") - self.in_type = in_type - self.name = name - self.required = required - self.style = style - self.explode = explode - self.allow_reserved = allow_reserved - self.schema = schema - self.content = content - - def _serialize_json( - self, - in_data: typing.Union[None, int, float, str, bool, dict, list], - eliminate_whitespace: bool = False, - ) -> str: - if eliminate_whitespace: - return json.dumps(in_data, separators=self._json_encoder.compact_separators) - return json.dumps(in_data) - - -class PathParameter(ParameterBase, StyleSimpleSerializer): - def __init__( - self, - name: str, - required: bool = False, - style: typing.Optional[ParameterStyle] = None, - explode: bool = False, - allow_reserved: typing.Optional[bool] = None, - schema: typing.Optional[typing.Type[Schema]] = None, - content: typing.Optional[typing.Dict[str, typing.Type[Schema]]] = None, - ): - super().__init__( - name, - in_type=ParameterInType.PATH, - required=required, - style=style, - explode=explode, - allow_reserved=allow_reserved, - schema=schema, - content=content, - ) - - def __serialize_label( - self, in_data: typing.Union[None, int, float, str, bool, dict, list] - ) -> typing.Dict[str, str]: - prefix_separator_iterator = PrefixSeparatorIterator(".", ".") - value = self._ref6570_expansion( - variable_name=self.name, - in_data=in_data, - explode=self.explode, - percent_encode=True, - prefix_separator_iterator=prefix_separator_iterator, - ) - return self._to_dict(self.name, value) - - def __serialize_matrix( - self, in_data: typing.Union[None, int, float, str, bool, dict, list] - ) -> typing.Dict[str, str]: - prefix_separator_iterator = PrefixSeparatorIterator(";", ";") - value = self._ref6570_expansion( - variable_name=self.name, - in_data=in_data, - explode=self.explode, - percent_encode=True, - prefix_separator_iterator=prefix_separator_iterator, - ) - return self._to_dict(self.name, value) - - def __serialize_simple( - self, - in_data: typing.Union[None, int, float, str, bool, dict, list], - ) -> typing.Dict[str, str]: - value = self._serialize_simple( - in_data=in_data, - name=self.name, - explode=self.explode, - percent_encode=True, - ) - return self._to_dict(self.name, value) - - def serialize( - self, - in_data: typing.Union[ - Schema, - Decimal, - int, - float, - str, - date, - datetime, - None, - bool, - list, - tuple, - dict, - frozendict.frozendict, - ], - ) -> typing.Dict[str, str]: - if self.schema: - cast_in_data = self.schema(in_data) - cast_in_data = self._json_encoder.default(cast_in_data) - """ - simple -> path - path: - returns path_params: dict - label -> path - returns path_params - matrix -> path - returns path_params - """ - if self.style: - if self.style is ParameterStyle.SIMPLE: - return self.__serialize_simple(cast_in_data) - elif self.style is ParameterStyle.LABEL: - return self.__serialize_label(cast_in_data) - elif self.style is ParameterStyle.MATRIX: - return self.__serialize_matrix(cast_in_data) - # self.content will be length one - for content_type, schema in self.content.items(): - cast_in_data = schema(in_data) - cast_in_data = self._json_encoder.default(cast_in_data) - if self._content_type_is_json(content_type): - value = self._serialize_json(cast_in_data) - return self._to_dict(self.name, value) - raise NotImplementedError("Serialization of {} has not yet been implemented".format(content_type)) - - -class QueryParameter(ParameterBase, StyleFormSerializer): - def __init__( - self, - name: str, - required: bool = False, - style: typing.Optional[ParameterStyle] = None, - explode: typing.Optional[bool] = None, - allow_reserved: typing.Optional[bool] = None, - schema: typing.Optional[typing.Type[Schema]] = None, - content: typing.Optional[typing.Dict[str, typing.Type[Schema]]] = None, - ): - used_style = ParameterStyle.FORM if style is None else style - used_explode = self._get_default_explode(used_style) if explode is None else explode - - super().__init__( - name, - in_type=ParameterInType.QUERY, - required=required, - style=used_style, - explode=used_explode, - allow_reserved=allow_reserved, - schema=schema, - content=content, - ) - - def __serialize_space_delimited( - self, - in_data: typing.Union[None, int, float, str, bool, dict, list], - prefix_separator_iterator: typing.Optional[PrefixSeparatorIterator], - ) -> typing.Dict[str, str]: - if prefix_separator_iterator is None: - prefix_separator_iterator = self.get_prefix_separator_iterator() - value = self._ref6570_expansion( - variable_name=self.name, - in_data=in_data, - explode=self.explode, - percent_encode=True, - prefix_separator_iterator=prefix_separator_iterator, - ) - return self._to_dict(self.name, value) - - def __serialize_pipe_delimited( - self, - in_data: typing.Union[None, int, float, str, bool, dict, list], - prefix_separator_iterator: typing.Optional[PrefixSeparatorIterator], - ) -> typing.Dict[str, str]: - if prefix_separator_iterator is None: - prefix_separator_iterator = self.get_prefix_separator_iterator() - value = self._ref6570_expansion( - variable_name=self.name, - in_data=in_data, - explode=self.explode, - percent_encode=True, - prefix_separator_iterator=prefix_separator_iterator, - ) - return self._to_dict(self.name, value) - - def __serialize_form( - self, - in_data: typing.Union[None, int, float, str, bool, dict, list], - prefix_separator_iterator: typing.Optional[PrefixSeparatorIterator], - ) -> typing.Dict[str, str]: - if prefix_separator_iterator is None: - prefix_separator_iterator = self.get_prefix_separator_iterator() - value = self._serialize_form( - in_data, - name=self.name, - explode=self.explode, - percent_encode=True, - prefix_separator_iterator=prefix_separator_iterator, - ) - return self._to_dict(self.name, value) - - def get_prefix_separator_iterator( - self, - ) -> typing.Optional[PrefixSeparatorIterator]: - if self.style is ParameterStyle.FORM: - return PrefixSeparatorIterator("?", "&") - elif self.style is ParameterStyle.SPACE_DELIMITED: - return PrefixSeparatorIterator("", "%20") - elif self.style is ParameterStyle.PIPE_DELIMITED: - return PrefixSeparatorIterator("", "|") - - def serialize( - self, - in_data: typing.Union[ - Schema, - Decimal, - int, - float, - str, - date, - datetime, - None, - bool, - list, - tuple, - dict, - frozendict.frozendict, - ], - prefix_separator_iterator: typing.Optional[PrefixSeparatorIterator] = None, - ) -> typing.Dict[str, str]: - if self.schema: - cast_in_data = self.schema(in_data) - cast_in_data = self._json_encoder.default(cast_in_data) - """ - form -> query - query: - - GET/HEAD/DELETE: could use fields - - PUT/POST: must use urlencode to send parameters - returns fields: tuple - spaceDelimited -> query - returns fields - pipeDelimited -> query - returns fields - deepObject -> query, https://github.com/OAI/OpenAPI-Specification/issues/1706 - returns fields - """ - if self.style: - # TODO update query ones to omit setting values when [] {} or None is input - if self.style is ParameterStyle.FORM: - return self.__serialize_form(cast_in_data, prefix_separator_iterator) - elif self.style is ParameterStyle.SPACE_DELIMITED: - return self.__serialize_space_delimited(cast_in_data, prefix_separator_iterator) - elif self.style is ParameterStyle.PIPE_DELIMITED: - return self.__serialize_pipe_delimited(cast_in_data, prefix_separator_iterator) - # self.content will be length one - if prefix_separator_iterator is None: - prefix_separator_iterator = self.get_prefix_separator_iterator() - for content_type, schema in self.content.items(): - cast_in_data = schema(in_data) - cast_in_data = self._json_encoder.default(cast_in_data) - if self._content_type_is_json(content_type): - value = self._serialize_json(cast_in_data, eliminate_whitespace=True) - return self._to_dict( - self.name, - next(prefix_separator_iterator) + self.name + "=" + quote(value), - ) - raise NotImplementedError("Serialization of {} has not yet been implemented".format(content_type)) - - -class CookieParameter(ParameterBase, StyleFormSerializer): - def __init__( - self, - name: str, - required: bool = False, - style: typing.Optional[ParameterStyle] = None, - explode: typing.Optional[bool] = None, - allow_reserved: typing.Optional[bool] = None, - schema: typing.Optional[typing.Type[Schema]] = None, - content: typing.Optional[typing.Dict[str, typing.Type[Schema]]] = None, - ): - used_style = ParameterStyle.FORM if style is None and content is None and schema else style - used_explode = self._get_default_explode(used_style) if explode is None else explode - - super().__init__( - name, - in_type=ParameterInType.COOKIE, - required=required, - style=used_style, - explode=used_explode, - allow_reserved=allow_reserved, - schema=schema, - content=content, - ) - - def serialize( - self, - in_data: typing.Union[ - Schema, - Decimal, - int, - float, - str, - date, - datetime, - None, - bool, - list, - tuple, - dict, - frozendict.frozendict, - ], - ) -> typing.Dict[str, str]: - if self.schema: - cast_in_data = self.schema(in_data) - cast_in_data = self._json_encoder.default(cast_in_data) - """ - form -> cookie - returns fields: tuple - """ - if self.style: - """ - TODO add escaping of comma, space, equals - or turn encoding on - """ - value = self._serialize_form( - cast_in_data, - explode=self.explode, - name=self.name, - percent_encode=False, - prefix_separator_iterator=PrefixSeparatorIterator("", "&"), - ) - return self._to_dict(self.name, value) - # self.content will be length one - for content_type, schema in self.content.items(): - cast_in_data = schema(in_data) - cast_in_data = self._json_encoder.default(cast_in_data) - if self._content_type_is_json(content_type): - value = self._serialize_json(cast_in_data) - return self._to_dict(self.name, value) - raise NotImplementedError("Serialization of {} has not yet been implemented".format(content_type)) - - -class HeaderParameter(ParameterBase, StyleSimpleSerializer): - def __init__( - self, - name: str, - required: bool = False, - style: typing.Optional[ParameterStyle] = None, - explode: bool = False, - allow_reserved: typing.Optional[bool] = None, - schema: typing.Optional[typing.Type[Schema]] = None, - content: typing.Optional[typing.Dict[str, typing.Type[Schema]]] = None, - ): - super().__init__( - name, - in_type=ParameterInType.HEADER, - required=required, - style=style, - explode=explode, - allow_reserved=allow_reserved, - schema=schema, - content=content, - ) - - @staticmethod - def __to_headers(in_data: typing.Tuple[typing.Tuple[str, str], ...]) -> HTTPHeaderDict: - data = tuple(t for t in in_data if t) - headers = HTTPHeaderDict() - if not data: - return headers - headers.extend(data) - return headers - - def serialize( - self, - in_data: typing.Union[ - Schema, - Decimal, - int, - float, - str, - date, - datetime, - None, - bool, - list, - tuple, - dict, - frozendict.frozendict, - ], - ) -> HTTPHeaderDict: - if self.schema: - cast_in_data = self.schema(in_data) - cast_in_data = self._json_encoder.default(cast_in_data) - """ - simple -> header - headers: PoolManager needs a mapping, tuple is close - returns headers: dict - """ - if self.style: - value = self._serialize_simple(cast_in_data, self.name, self.explode, False) - return self.__to_headers(((self.name, value),)) - # self.content will be length one - for content_type, schema in self.content.items(): - cast_in_data = schema(in_data) - cast_in_data = self._json_encoder.default(cast_in_data) - if self._content_type_is_json(content_type): - value = self._serialize_json(cast_in_data) - return self.__to_headers(((self.name, value),)) - raise NotImplementedError("Serialization of {} has not yet been implemented".format(content_type)) - - -class Encoding: - def __init__( - self, - content_type: str, - headers: typing.Optional[typing.Dict[str, HeaderParameter]] = None, - style: typing.Optional[ParameterStyle] = None, - explode: bool = False, - allow_reserved: bool = False, - ): - self.content_type = content_type - self.headers = headers - self.style = style - self.explode = explode - self.allow_reserved = allow_reserved - - -@dataclass -class MediaType: - """ - Used to store request and response body schema information - encoding: - A map between a property name and its encoding information. - The key, being the property name, MUST exist in the schema as a property. - The encoding object SHALL only apply to requestBody objects when the media type is - multipart or application/x-www-form-urlencoded. - """ - - schema: typing.Optional[typing.Type[Schema]] = None - encoding: typing.Optional[typing.Dict[str, Encoding]] = None - - -@dataclass -class ApiResponse: - response: urllib3.HTTPResponse - body: typing.Union[Unset, Schema] - headers: typing.Union[Unset, typing.List[HeaderParameter]] - - def __init__( - self, - response: urllib3.HTTPResponse, - body: typing.Union[Unset, typing.Type[Schema]], - headers: typing.Union[Unset, typing.List[HeaderParameter]], - ): - """ - pycharm needs this to prevent 'Unexpected argument' warnings - """ - self.response = response - self.body = body - self.headers = headers - - -@dataclass -class ApiResponseWithoutDeserialization(ApiResponse): - response: urllib3.HTTPResponse - body: typing.Union[Unset, typing.Type[Schema]] = unset - headers: typing.Union[Unset, typing.List[HeaderParameter]] = unset - - -class OpenApiResponse(JSONDetector): - __filename_content_disposition_pattern = re.compile('filename="(.+?)"') - - def __init__( - self, - response_cls: typing.Type[ApiResponse] = ApiResponse, - content: typing.Optional[typing.Dict[str, MediaType]] = None, - headers: typing.Optional[typing.List[HeaderParameter]] = None, - ): - self.headers = headers - if content is not None and len(content) == 0: - raise ValueError("Invalid value for content, the content dict must have >= 1 entry") - self.content = content - self.response_cls = response_cls - - @staticmethod - def __deserialize_json(response: urllib3.HTTPResponse) -> typing.Any: - # python must be >= 3.9 so we can pass in bytes into json.loads - return json.loads(response.data) - - @staticmethod - def __file_name_from_response_url( - response_url: typing.Optional[str], - ) -> typing.Optional[str]: - if response_url is None: - return None - url_path = urlparse(response_url).path - if url_path: - path_basename = os.path.basename(url_path) - if path_basename: - _filename, ext = os.path.splitext(path_basename) - if ext: - return path_basename - return None - - @classmethod - def __file_name_from_content_disposition(cls, content_disposition: typing.Optional[str]) -> typing.Optional[str]: - if content_disposition is None: - return None - match = cls.__filename_content_disposition_pattern.search(content_disposition) - if not match: - return None - return match.group(1) +from launch.api_client.api.default_api import AsyncDefaultApi, SyncDefaultApi +from launch.api_client.exceptions import ( + ResponseHandlingException, + UnexpectedResponse, +) - def __deserialize_application_octet_stream( - self, response: urllib3.HTTPResponse - ) -> typing.Union[bytes, io.BufferedReader]: - """ - urllib3 use cases: - 1. when preload_content=True (stream=False) then supports_chunked_reads is False and bytes are returned - 2. when preload_content=False (stream=True) then supports_chunked_reads is True and - a file will be written and returned - """ - if response.supports_chunked_reads(): - file_name = self.__file_name_from_content_disposition( - response.headers.get("content-disposition") - ) or self.__file_name_from_response_url(response.geturl()) +ClientT = TypeVar("ClientT", bound="ApiClient") - if file_name is None: - _fd, path = tempfile.mkstemp() - else: - path = os.path.join(tempfile.gettempdir(), file_name) - with open(path, "wb") as new_file: - chunk_size = 1024 - while True: - data = response.read(chunk_size) - if not data: - break - new_file.write(data) - # release_conn is needed for streaming connections only - response.release_conn() - new_file = open(path, "rb") - return new_file - else: - return response.data +class AsyncApis(Generic[ClientT]): + def __init__(self, client: ClientT): + self.client = client - @staticmethod - def __deserialize_multipart_form_data( - response: urllib3.HTTPResponse, - ) -> typing.Dict[str, typing.Any]: - msg = email.message_from_bytes(response.data) - return { - part.get_param("name", header="Content-Disposition"): part.get_payload(decode=True).decode( - part.get_content_charset() - ) - if part.get_content_charset() - else part.get_payload() - for part in msg.get_payload() - } + self.default_api = AsyncDefaultApi(self.client) - def deserialize(self, response: urllib3.HTTPResponse, configuration: Configuration) -> ApiResponse: - content_type = response.getheader("content-type") - deserialized_body = unset - streamed = response.supports_chunked_reads() - deserialized_headers = unset - if self.headers is not None: - # TODO add header deserialiation here - pass +class SyncApis(Generic[ClientT]): + def __init__(self, client: ClientT): + self.client = client - if self.content is not None: - if content_type not in self.content: - raise ApiValueError( - f"Invalid content_type returned. Content_type='{content_type}' was returned " - f"when only {str(set(self.content))} are defined for status_code={str(response.status)}" - ) - body_schema = self.content[content_type].schema - if body_schema is None: - # some specs do not define response content media type schemas - return self.response_cls(response=response, headers=deserialized_headers, body=unset) + self.default_api = SyncDefaultApi(self.client) - if self._content_type_is_json(content_type): - body_data = self.__deserialize_json(response) - elif content_type == "application/octet-stream": - body_data = self.__deserialize_application_octet_stream(response) - elif content_type.startswith("multipart/form-data"): - body_data = self.__deserialize_multipart_form_data(response) - content_type = "multipart/form-data" - else: - raise NotImplementedError("Deserialization of {} has not yet been implemented".format(content_type)) - deserialized_body = body_schema.from_openapi_data_oapg(body_data, _configuration=configuration) - elif streamed: - response.release_conn() - return self.response_cls( - response=response, - headers=deserialized_headers, - body=deserialized_body, - ) +T = TypeVar("T") +Send = Callable[[Request], Awaitable[Response]] +MiddlewareT = Callable[[Request, Send], Awaitable[Response]] class ApiClient: - """Generic API client for OpenAPI client library builds. - - OpenAPI generic API client. This client handles the client- - server communication, and is invariant across implementations. Specifics of - the methods and models for each application are generated from the OpenAPI - templates. - - NOTE: This class is auto generated by OpenAPI Generator. - Ref: https://openapi-generator.tech - Do not edit the class manually. - - :param configuration: .Configuration object for this client - :param header_name: a header to pass when making calls to the API. - :param header_value: a header value to pass when making calls to - the API. - :param cookie: a cookie to include in the header when making calls - to the API - :param pool_threads: The number of threads to use for async requests - to the API. More threads means more concurrent API requests. - """ - - _pool = None + def __init__(self, host: str = None, **kwargs: Any) -> None: + self.host = host + self.middleware: MiddlewareT = BaseMiddleware() + self._async_client = AsyncClient(**kwargs) - def __init__( + @overload + async def request( self, - configuration: typing.Optional[Configuration] = None, - header_name: typing.Optional[str] = None, - header_value: typing.Optional[str] = None, - cookie: typing.Optional[str] = None, - pool_threads: int = 1, - ): - if configuration is None: - configuration = Configuration() - self.configuration = configuration - self.pool_threads = pool_threads - - self.rest_client = rest.RESTClientObject(configuration) - self.default_headers = HTTPHeaderDict() - if header_name is not None: - self.default_headers[header_name] = header_value - self.cookie = cookie - # Set default User-Agent. - self.user_agent = "OpenAPI-Generator/1.0.5/python" - - def __enter__(self): - return self - - def __exit__(self, exc_type, exc_value, traceback): - self.close() - - def close(self): - if self._pool: - self._pool.close() - self._pool.join() - self._pool = None - if hasattr(atexit, "unregister"): - atexit.unregister(self.close) - - @property - def pool(self): - """Create thread pool on first request - avoids instantiating unused threadpool for blocking clients. - """ - if self._pool is None: - atexit.register(self.close) - self._pool = ThreadPool(self.pool_threads) - return self._pool - - @property - def user_agent(self): - """User agent for this API client""" - return self.default_headers["User-Agent"] - - @user_agent.setter - def user_agent(self, value): - self.default_headers["User-Agent"] = value - - def set_default_header(self, header_name, header_value): - self.default_headers[header_name] = header_value - - def __call_api( - self, - resource_path: str, + *, + type_: Type[T], method: str, - headers: typing.Optional[HTTPHeaderDict] = None, - body: typing.Optional[typing.Union[str, bytes]] = None, - fields: typing.Optional[typing.Tuple[typing.Tuple[str, str], ...]] = None, - auth_settings: typing.Optional[typing.List[str]] = None, - stream: bool = False, - timeout: typing.Optional[typing.Union[int, typing.Tuple]] = None, - host: typing.Optional[str] = None, - ) -> urllib3.HTTPResponse: - # header parameters - used_headers = HTTPHeaderDict(self.default_headers) - if self.cookie: - headers["Cookie"] = self.cookie - - # auth setting - self.update_params_for_auth(used_headers, auth_settings, resource_path, method, body) - - # must happen after cookie setting and auth setting in case user is overriding those - if headers: - used_headers.update(headers) - - # request url - if host is None: - url = self.configuration.host + resource_path - else: - # use server/host defined in path or operation instead - url = host + resource_path - - # perform request and return response - response = self.request( - method, - url, - headers=used_headers, - fields=fields, - body=body, - stream=stream, - timeout=timeout, - ) - return response + url: str, + path_params: Dict[str, Any] = None, + **kwargs: Any, + ) -> T: + ... - def call_api( + @overload # noqa F811 + async def request( self, - resource_path: str, + *, + type_: None, method: str, - headers: typing.Optional[HTTPHeaderDict] = None, - body: typing.Optional[typing.Union[str, bytes]] = None, - fields: typing.Optional[typing.Tuple[typing.Tuple[str, str], ...]] = None, - auth_settings: typing.Optional[typing.List[str]] = None, - async_req: typing.Optional[bool] = None, - stream: bool = False, - timeout: typing.Optional[typing.Union[int, typing.Tuple]] = None, - host: typing.Optional[str] = None, - ) -> urllib3.HTTPResponse: - """Makes the HTTP request (synchronous) and returns deserialized data. - - To make an async_req request, set the async_req parameter. - - :param resource_path: Path to method endpoint. - :param method: Method to call. - :param headers: Header parameters to be - placed in the request header. - :param body: Request body. - :param fields: Request post form parameters, - for `application/x-www-form-urlencoded`, `multipart/form-data`. - :param auth_settings: Auth Settings names for the request. - :param async_req: execute request asynchronously - :type async_req: bool, optional TODO remove, unused - :param stream: if True, the urllib3.HTTPResponse object will - be returned without reading/decoding response - data. Also when True, if the openapi spec describes a file download, - the data will be written to a local filesystme file and the BinarySchema - instance will also inherit from FileSchema and FileIO - Default is False. - :type stream: bool, optional - :param 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. - :param host: api endpoint host - :return: - If async_req parameter is True, - the request will be called asynchronously. - The method will return the request thread. - If parameter async_req is False or missing, - then the method will return the response directly. - """ - - if not async_req: - return self.__call_api( - resource_path, - method, - headers, - body, - fields, - auth_settings, - stream, - timeout, - host, - ) - - return self.pool.apply_async( - self.__call_api, - ( - resource_path, - method, - headers, - body, - json, - fields, - auth_settings, - stream, - timeout, - host, - ), - ) + url: str, + path_params: Dict[str, Any] = None, + **kwargs: Any, + ) -> None: + ... - def request( + async def request( # noqa F811 self, + *, + type_: Any, method: str, url: str, - headers: typing.Optional[HTTPHeaderDict] = None, - fields: typing.Optional[typing.Tuple[typing.Tuple[str, str], ...]] = None, - body: typing.Optional[typing.Union[str, bytes]] = None, - stream: bool = False, - timeout: typing.Optional[typing.Union[int, typing.Tuple]] = None, - ) -> urllib3.HTTPResponse: - """Makes the HTTP request using RESTClient.""" - if method == "GET": - return self.rest_client.GET(url, stream=stream, timeout=timeout, headers=headers) - elif method == "HEAD": - return self.rest_client.HEAD(url, stream=stream, timeout=timeout, headers=headers) - elif method == "OPTIONS": - return self.rest_client.OPTIONS( - url, - headers=headers, - fields=fields, - stream=stream, - timeout=timeout, - body=body, - ) - elif method == "POST": - return self.rest_client.POST( - url, - headers=headers, - fields=fields, - stream=stream, - timeout=timeout, - body=body, - ) - elif method == "PUT": - return self.rest_client.PUT( - url, - headers=headers, - fields=fields, - stream=stream, - timeout=timeout, - body=body, - ) - elif method == "PATCH": - return self.rest_client.PATCH( - url, - headers=headers, - fields=fields, - stream=stream, - timeout=timeout, - body=body, - ) - elif method == "DELETE": - return self.rest_client.DELETE(url, headers=headers, stream=stream, timeout=timeout, body=body) - else: - raise ApiValueError("http method must be `GET`, `HEAD`, `OPTIONS`," " `POST`, `PATCH`, `PUT` or `DELETE`.") - - def update_params_for_auth(self, headers, auth_settings, resource_path, method, body): - """Updates header and query params based on authentication setting. - - :param headers: Header parameters dict to be updated. - :param auth_settings: Authentication setting identifiers list. - :param resource_path: A string representation of the HTTP request resource path. - :param method: A string representation of the HTTP request method. - :param body: A object representing the body of the HTTP request. - The object type is the return value of _encoder.default(). - """ - if not auth_settings: - return - - for auth in auth_settings: - auth_setting = self.configuration.auth_settings().get(auth) - if not auth_setting: - continue - if auth_setting["in"] == "cookie": - headers.add("Cookie", auth_setting["value"]) - elif auth_setting["in"] == "header": - if auth_setting["type"] != "http-signature": - headers.add(auth_setting["key"], auth_setting["value"]) - elif auth_setting["in"] == "query": - """TODO implement auth in query - need to pass in prefix_separator_iterator - and need to output resource_path with query params added - """ - raise ApiValueError("Auth in query not yet implemented") - else: - raise ApiValueError("Authentication token must be in `query` or `header`") - - -class Api: - """NOTE: This class is auto generated by OpenAPI Generator - Ref: https://openapi-generator.tech - - Do not edit the class manually. - """ - - def __init__(self, api_client: typing.Optional[ApiClient] = None): - if api_client is None: - api_client = ApiClient() - self.api_client = api_client - - @staticmethod - def _verify_typed_dict_inputs_oapg( - cls: typing.Type[typing_extensions.TypedDict], - data: typing.Dict[str, typing.Any], - ): - """ - Ensures that: - - required keys are present - - additional properties are not input - - value stored under required keys do not have the value unset - Note: detailed value checking is done in schema classes - """ - missing_required_keys = [] - required_keys_with_unset_values = [] - for required_key in cls.__required_keys__: - if required_key not in data: - missing_required_keys.append(required_key) - continue - value = data[required_key] - if value is unset: - required_keys_with_unset_values.append(required_key) - if missing_required_keys: - raise ApiTypeError( - "{} missing {} required arguments: {}".format( - cls.__name__, - len(missing_required_keys), - missing_required_keys, - ) - ) - if required_keys_with_unset_values: - raise ApiValueError( - "{} contains invalid unset values for {} required keys: {}".format( - cls.__name__, - len(required_keys_with_unset_values), - required_keys_with_unset_values, - ) - ) - - disallowed_additional_keys = [] - for key in data: - if key in cls.__required_keys__ or key in cls.__optional_keys__: - continue - disallowed_additional_keys.append(key) - if disallowed_additional_keys: - raise ApiTypeError( - "{} got {} unexpected keyword arguments: {}".format( - cls.__name__, - len(disallowed_additional_keys), - disallowed_additional_keys, - ) - ) - - def _get_host_oapg( - self, - operation_id: str, - servers: typing.Tuple[typing.Dict[str, str], ...] = tuple(), - host_index: typing.Optional[int] = None, - ) -> typing.Optional[str]: - configuration = self.api_client.configuration + path_params: Dict[str, Any] = None, + **kwargs: Any, + ) -> Any: + if path_params is None: + path_params = {} + url = (self.host or "") + url.format(**path_params) + request = Request(method, url, **kwargs) + return await self.send(request, type_) + + @overload + def request_sync(self, *, type_: Type[T], **kwargs: Any) -> T: + ... + + @overload # noqa F811 + def request_sync(self, *, type_: None, **kwargs: Any) -> None: + ... + + def request_sync(self, *, type_: Any, **kwargs: Any) -> Any: # noqa F811 + """ + This method is not used by the generated apis, but is included for convenience + """ + return get_event_loop().run_until_complete(self.request(type_=type_, **kwargs)) + + async def send(self, request: Request, type_: Type[T]) -> T: + response = await self.middleware(request, self.send_inner) + if response.status_code in [200, 201]: + try: + return parse_obj_as(type_, response.json()) + except ValidationError as e: + raise ResponseHandlingException(e) + raise UnexpectedResponse.for_response(response) + + async def send_inner(self, request: Request) -> Response: try: - if host_index is None: - index = configuration.server_operation_index.get(operation_id, configuration.server_index) - else: - index = host_index - server_variables = configuration.server_operation_variables.get( - operation_id, configuration.server_variables - ) - host = configuration.get_host_from_settings(index, variables=server_variables, servers=servers) - except IndexError: - if servers: - raise ApiValueError("Invalid host index. Must be 0 <= index < %s" % len(servers)) - host = None - return host - - -class SerializedRequestBody(typing_extensions.TypedDict, total=False): - body: typing.Union[str, bytes] - fields: typing.Tuple[typing.Union[RequestField, typing.Tuple[str, str]], ...] - - -class RequestBody(StyleFormSerializer, JSONDetector): - """ - A request body parameter - content: content_type to MediaType Schema info - """ - - __json_encoder = JSONEncoder() - - def __init__( - self, - content: typing.Dict[str, MediaType], - required: bool = False, - ): - self.required = required - if len(content) == 0: - raise ValueError("Invalid value for content, the content dict must have >= 1 entry") - self.content = content - - def __serialize_json(self, in_data: typing.Any) -> typing.Dict[str, bytes]: - in_data = self.__json_encoder.default(in_data) - json_str = json.dumps(in_data, separators=(",", ":"), ensure_ascii=False).encode("utf-8") - return dict(body=json_str) - - @staticmethod - def __serialize_text_plain(in_data: typing.Any) -> typing.Dict[str, str]: - if isinstance(in_data, frozendict.frozendict): - raise ValueError("Unable to serialize type frozendict.frozendict to text/plain") - elif isinstance(in_data, tuple): - raise ValueError("Unable to serialize type tuple to text/plain") - elif isinstance(in_data, NoneClass): - raise ValueError("Unable to serialize type NoneClass to text/plain") - elif isinstance(in_data, BoolClass): - raise ValueError("Unable to serialize type BoolClass to text/plain") - return dict(body=str(in_data)) - - def __multipart_json_item(self, key: str, value: Schema) -> RequestField: - json_value = self.__json_encoder.default(value) - return RequestField( - name=key, - data=json.dumps(json_value), - headers={"Content-Type": "application/json"}, - ) - - def __multipart_form_item(self, key: str, value: Schema) -> RequestField: - if isinstance(value, str): - return RequestField( - name=key, - data=str(value), - headers={"Content-Type": "text/plain"}, - ) - elif isinstance(value, bytes): - return RequestField( - name=key, - data=value, - headers={"Content-Type": "application/octet-stream"}, - ) - elif isinstance(value, FileIO): - request_field = RequestField( - name=key, - data=value.read(), - filename=os.path.basename(value.name), - headers={"Content-Type": "application/octet-stream"}, - ) - value.close() - return request_field - else: - return self.__multipart_json_item(key=key, value=value) - - def __serialize_multipart_form_data(self, in_data: Schema) -> typing.Dict[str, typing.Tuple[RequestField, ...]]: - if not isinstance(in_data, frozendict.frozendict): - raise ValueError(f"Unable to serialize {in_data} to multipart/form-data because it is not a dict of data") - """ - In a multipart/form-data request body, each schema property, or each element of a schema array property, - takes a section in the payload with an internal header as defined by RFC7578. The serialization strategy - for each property of a multipart/form-data request body can be specified in an associated Encoding Object. - - When passing in multipart types, boundaries MAY be used to separate sections of the content being - transferred – thus, the following default Content-Types are defined for multipart: + response = await self._async_client.send(request) + except Exception as e: + raise ResponseHandlingException(e) + return response - If the (object) property is a primitive, or an array of primitive values, the default Content-Type is text/plain - If the property is complex, or an array of complex values, the default Content-Type is application/json - Question: how is the array of primitives encoded? - If the property is a type: string with a contentEncoding, the default Content-Type is application/octet-stream - """ - fields = [] - for key, value in in_data.items(): - if isinstance(value, tuple): - if value: - # values use explode = True, so the code makes a RequestField for each item with name=key - for item in value: - request_field = self.__multipart_form_item(key=key, value=item) - fields.append(request_field) - else: - # send an empty array as json because exploding will not send it - request_field = self.__multipart_json_item(key=key, value=value) - fields.append(request_field) - else: - request_field = self.__multipart_form_item(key=key, value=value) - fields.append(request_field) + def add_middleware(self, middleware: MiddlewareT) -> None: + current_middleware = self.middleware - return dict(fields=tuple(fields)) + async def new_middleware(request: Request, call_next: Send) -> Response: + async def inner_send(request: Request) -> Response: + return await current_middleware(request, call_next) - def __serialize_application_octet_stream(self, in_data: BinarySchema) -> typing.Dict[str, bytes]: - if isinstance(in_data, bytes): - return dict(body=in_data) - # FileIO type - result = dict(body=in_data.read()) - in_data.close() - return result + return await middleware(request, inner_send) - def __serialize_application_x_www_form_data(self, in_data: typing.Any) -> SerializedRequestBody: - """ - POST submission of form data in body - """ - if not isinstance(in_data, frozendict.frozendict): - raise ValueError( - f"Unable to serialize {in_data} to application/x-www-form-urlencoded because it is not a dict of data" - ) - cast_in_data = self.__json_encoder.default(in_data) - value = self._serialize_form(cast_in_data, name="", explode=True, percent_encode=True) - return dict(body=value) + self.middleware = new_middleware - def serialize(self, in_data: typing.Any, content_type: str) -> SerializedRequestBody: - """ - If a str is returned then the result will be assigned to data when making the request - If a tuple is returned then the result will be used as fields input in encode_multipart_formdata - Return a tuple of - The key of the return dict is - - body for application/json - - encode_multipart and fields for multipart/form-data - """ - media_type = self.content[content_type] - if isinstance(in_data, media_type.schema): - cast_in_data = in_data - elif isinstance(in_data, (dict, frozendict.frozendict)) and in_data: - cast_in_data = media_type.schema(**in_data) - else: - cast_in_data = media_type.schema(in_data) - # TODO check for and use encoding if it exists - # and content_type is multipart or application/x-www-form-urlencoded - if self._content_type_is_json(content_type): - return self.__serialize_json(cast_in_data) - elif content_type == "text/plain": - return self.__serialize_text_plain(cast_in_data) - elif content_type == "multipart/form-data": - return self.__serialize_multipart_form_data(cast_in_data) - elif content_type == "application/x-www-form-urlencoded": - return self.__serialize_application_x_www_form_data(cast_in_data) - elif content_type == "application/octet-stream": - return self.__serialize_application_octet_stream(cast_in_data) - raise NotImplementedError("Serialization has not yet been implemented for {}".format(content_type)) +class BaseMiddleware: + async def __call__(self, request: Request, call_next: Send) -> Response: + return await call_next(request) diff --git a/launch/api_client/apis/paths/healthcheck.py b/launch/api_client/apis/paths/healthcheck.py deleted file mode 100644 index 6f11f7bb..00000000 --- a/launch/api_client/apis/paths/healthcheck.py +++ /dev/null @@ -1,7 +0,0 @@ -from launch.api_client.paths.healthcheck.get import ApiForget - - -class Healthcheck( - ApiForget, -): - pass diff --git a/launch/api_client/apis/paths/healthz.py b/launch/api_client/apis/paths/healthz.py deleted file mode 100644 index 62f2a817..00000000 --- a/launch/api_client/apis/paths/healthz.py +++ /dev/null @@ -1,7 +0,0 @@ -from launch.api_client.paths.healthz.get import ApiForget - - -class Healthz( - ApiForget, -): - pass diff --git a/launch/api_client/apis/paths/readyz.py b/launch/api_client/apis/paths/readyz.py deleted file mode 100644 index 35b8e737..00000000 --- a/launch/api_client/apis/paths/readyz.py +++ /dev/null @@ -1,7 +0,0 @@ -from launch.api_client.paths.readyz.get import ApiForget - - -class Readyz( - ApiForget, -): - pass diff --git a/launch/api_client/apis/paths/v1_async_tasks.py b/launch/api_client/apis/paths/v1_async_tasks.py deleted file mode 100644 index 74bb2cba..00000000 --- a/launch/api_client/apis/paths/v1_async_tasks.py +++ /dev/null @@ -1,7 +0,0 @@ -from launch.api_client.paths.v1_async_tasks.post import ApiForpost - - -class V1AsyncTasks( - ApiForpost, -): - pass diff --git a/launch/api_client/apis/paths/v1_async_tasks_task_id.py b/launch/api_client/apis/paths/v1_async_tasks_task_id.py deleted file mode 100644 index 473e4bb4..00000000 --- a/launch/api_client/apis/paths/v1_async_tasks_task_id.py +++ /dev/null @@ -1,7 +0,0 @@ -from launch.api_client.paths.v1_async_tasks_task_id.get import ApiForget - - -class V1AsyncTasksTaskId( - ApiForget, -): - pass diff --git a/launch/api_client/apis/paths/v1_batch_jobs.py b/launch/api_client/apis/paths/v1_batch_jobs.py deleted file mode 100644 index 73b60ebc..00000000 --- a/launch/api_client/apis/paths/v1_batch_jobs.py +++ /dev/null @@ -1,7 +0,0 @@ -from launch.api_client.paths.v1_batch_jobs.post import ApiForpost - - -class V1BatchJobs( - ApiForpost, -): - pass diff --git a/launch/api_client/apis/paths/v1_batch_jobs_batch_job_id.py b/launch/api_client/apis/paths/v1_batch_jobs_batch_job_id.py deleted file mode 100644 index 40bb9f1a..00000000 --- a/launch/api_client/apis/paths/v1_batch_jobs_batch_job_id.py +++ /dev/null @@ -1,9 +0,0 @@ -from launch.api_client.paths.v1_batch_jobs_batch_job_id.get import ApiForget -from launch.api_client.paths.v1_batch_jobs_batch_job_id.put import ApiForput - - -class V1BatchJobsBatchJobId( - ApiForget, - ApiForput, -): - pass diff --git a/launch/api_client/apis/paths/v1_model_bundles.py b/launch/api_client/apis/paths/v1_model_bundles.py deleted file mode 100644 index 205a87dc..00000000 --- a/launch/api_client/apis/paths/v1_model_bundles.py +++ /dev/null @@ -1,9 +0,0 @@ -from launch.api_client.paths.v1_model_bundles.get import ApiForget -from launch.api_client.paths.v1_model_bundles.post import ApiForpost - - -class V1ModelBundles( - ApiForget, - ApiForpost, -): - pass diff --git a/launch/api_client/apis/paths/v1_model_bundles_latest.py b/launch/api_client/apis/paths/v1_model_bundles_latest.py deleted file mode 100644 index e84f6419..00000000 --- a/launch/api_client/apis/paths/v1_model_bundles_latest.py +++ /dev/null @@ -1,7 +0,0 @@ -from launch.api_client.paths.v1_model_bundles_latest.get import ApiForget - - -class V1ModelBundlesLatest( - ApiForget, -): - pass diff --git a/launch/api_client/apis/paths/v1_model_endpoints.py b/launch/api_client/apis/paths/v1_model_endpoints.py deleted file mode 100644 index 070bb511..00000000 --- a/launch/api_client/apis/paths/v1_model_endpoints.py +++ /dev/null @@ -1,9 +0,0 @@ -from launch.api_client.paths.v1_model_endpoints.get import ApiForget -from launch.api_client.paths.v1_model_endpoints.post import ApiForpost - - -class V1ModelEndpoints( - ApiForget, - ApiForpost, -): - pass diff --git a/launch/api_client/apis/paths/v1_model_endpoints_api.py b/launch/api_client/apis/paths/v1_model_endpoints_api.py deleted file mode 100644 index 148b69f8..00000000 --- a/launch/api_client/apis/paths/v1_model_endpoints_api.py +++ /dev/null @@ -1,7 +0,0 @@ -from launch.api_client.paths.v1_model_endpoints_api.get import ApiForget - - -class V1ModelEndpointsApi( - ApiForget, -): - pass diff --git a/launch/api_client/apis/paths/v1_model_endpoints_model_endpoint_id.py b/launch/api_client/apis/paths/v1_model_endpoints_model_endpoint_id.py deleted file mode 100644 index 7f4d3e3a..00000000 --- a/launch/api_client/apis/paths/v1_model_endpoints_model_endpoint_id.py +++ /dev/null @@ -1,17 +0,0 @@ -from launch.api_client.paths.v1_model_endpoints_model_endpoint_id.delete import ( - ApiFordelete, -) -from launch.api_client.paths.v1_model_endpoints_model_endpoint_id.get import ( - ApiForget, -) -from launch.api_client.paths.v1_model_endpoints_model_endpoint_id.put import ( - ApiForput, -) - - -class V1ModelEndpointsModelEndpointId( - ApiForget, - ApiForput, - ApiFordelete, -): - pass diff --git a/launch/api_client/apis/paths/v1_sync_tasks.py b/launch/api_client/apis/paths/v1_sync_tasks.py deleted file mode 100644 index 6f4bfe5d..00000000 --- a/launch/api_client/apis/paths/v1_sync_tasks.py +++ /dev/null @@ -1,7 +0,0 @@ -from launch.api_client.paths.v1_sync_tasks.post import ApiForpost - - -class V1SyncTasks( - ApiForpost, -): - pass diff --git a/launch/api_client/exceptions.py b/launch/api_client/exceptions.py index 41d4f771..45437e1c 100644 --- a/launch/api_client/exceptions.py +++ b/launch/api_client/exceptions.py @@ -1,137 +1,46 @@ -# coding: utf-8 +import json +from typing import Any, Dict, Optional -""" - launch +from httpx import Headers, Response - No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator) # noqa: E501 +MAX_CONTENT = 200 - The version of the OpenAPI document: 1.0.0 - Generated by: https://openapi-generator.tech -""" +class ApiException(Exception): + """Base class""" -class OpenApiException(Exception): - """The base exception class for all OpenAPIExceptions""" +class UnexpectedResponse(ApiException): + def __init__(self, status_code: Optional[int], reason_phrase: str, content: bytes, headers: Headers) -> None: + self.status_code = status_code + self.reason_phrase = reason_phrase + self.content = content + self.headers = headers -class ApiTypeError(OpenApiException, TypeError): - def __init__(self, msg, path_to_item=None, valid_classes=None, key_type=None): - """Raises an exception for TypeErrors + @staticmethod + def for_response(response: Response) -> "ApiException": + return UnexpectedResponse( + status_code=response.status_code, + reason_phrase=response.reason_phrase, + content=response.content, + headers=response.headers, + ) - Args: - msg (str): the exception message - - Keyword Args: - path_to_item (list): a list of keys an indices to get to the - current_item - None if unset - valid_classes (tuple): the primitive classes that current item - should be an instance of - None if unset - key_type (bool): False if our value is a value in a dict - True if it is a key in a dict - False if our item is an item in a list - None if unset - """ - self.path_to_item = path_to_item - self.valid_classes = valid_classes - self.key_type = key_type - full_msg = msg - if path_to_item: - full_msg = "{0} at {1}".format(msg, render_path(path_to_item)) - super(ApiTypeError, self).__init__(full_msg) - - -class ApiValueError(OpenApiException, ValueError): - def __init__(self, msg, path_to_item=None): - """ - Args: - msg (str): the exception message - - Keyword Args: - path_to_item (list) the path to the exception in the - received_data dict. None if unset - """ - - self.path_to_item = path_to_item - full_msg = msg - if path_to_item: - full_msg = "{0} at {1}".format(msg, render_path(path_to_item)) - super(ApiValueError, self).__init__(full_msg) - - -class ApiAttributeError(OpenApiException, AttributeError): - def __init__(self, msg, path_to_item=None): - """ - Raised when an attribute reference or assignment fails. - - Args: - msg (str): the exception message - - Keyword Args: - path_to_item (None/list) the path to the exception in the - received_data dict - """ - self.path_to_item = path_to_item - full_msg = msg - if path_to_item: - full_msg = "{0} at {1}".format(msg, render_path(path_to_item)) - super(ApiAttributeError, self).__init__(full_msg) - - -class ApiKeyError(OpenApiException, KeyError): - def __init__(self, msg, path_to_item=None): - """ - Args: - msg (str): the exception message - - Keyword Args: - path_to_item (None/list) the path to the exception in the - received_data dict - """ - self.path_to_item = path_to_item - full_msg = msg - if path_to_item: - full_msg = "{0} at {1}".format(msg, render_path(path_to_item)) - super(ApiKeyError, self).__init__(full_msg) - - -class ApiException(OpenApiException): - def __init__( - self, - status=None, - reason=None, - api_response: "launch.api_client.api_client.ApiResponse" = None, - ): - if api_response: - self.status = api_response.response.status - self.reason = api_response.response.reason - self.body = api_response.response.data - self.headers = api_response.response.getheaders() + def __str__(self) -> str: + status_code_str = f"{self.status_code}" if self.status_code is not None else "" + if self.reason_phrase == "" and self.status_code is not None: + reason_phrase_str = "(Unrecognized Status Code)" else: - self.status = status - self.reason = reason - self.body = None - self.headers = None - - def __str__(self): - """Custom error messages for exception""" - error_message = "({0})\n" "Reason: {1}\n".format(self.status, self.reason) - if self.headers: - error_message += "HTTP response headers: {0}\n".format(self.headers) + reason_phrase_str = f"({self.reason_phrase})" + status_str = f"{status_code_str} {reason_phrase_str}".strip() + short_content = self.content if len(self.content) <= MAX_CONTENT else self.content[: MAX_CONTENT - 3] + b" ..." + raw_content_str = f"Raw response content:\n{short_content!r}" + return f"Unexpected Response: {status_str}\n{raw_content_str}" - if self.body: - error_message += "HTTP response body: {0}\n".format(self.body) + def structured(self) -> Dict[str, Any]: + return json.loads(self.content) - return error_message - -def render_path(path_to_item): - """Returns a string representation of a path""" - result = "" - for pth in path_to_item: - if isinstance(pth, int): - result += "[{0}]".format(pth) - else: - result += "['{0}']".format(pth) - return result +class ResponseHandlingException(ApiException): + def __init__(self, source: Exception): + self.source = source diff --git a/launch/api_client/models.py b/launch/api_client/models.py new file mode 100644 index 00000000..b59991f1 --- /dev/null +++ b/launch/api_client/models.py @@ -0,0 +1,285 @@ +from datetime import datetime +from enum import Enum +from typing import Any # noqa +from typing import Dict, List, Optional + +from pydantic import BaseModel, Field + + +class BatchJobSerializationFormat(str, Enum): + JSON = "JSON" + PICKLE = "PICKLE" + + +class BatchJobStatus(str, Enum): + PENDING = "PENDING" + CREATING_ENDPOINT = "CREATING_ENDPOINT" + STARTED = "STARTED" + SUCCESS = "SUCCESS" + FAILURE = "FAILURE" + CANCELLED = "CANCELLED" + UNDEFINED = "UNDEFINED" + + +class CloneModelBundleRequest(BaseModel): + new_app_config: "Optional[Any]" = Field(None, alias="new_app_config") + original_model_bundle_id: "str" = Field(..., alias="original_model_bundle_id") + + +class CreateAsyncTaskResponse(BaseModel): + task_id: "str" = Field(..., alias="task_id") + + +class CreateBatchJobRequest(BaseModel): + input_path: "str" = Field(..., alias="input_path") + labels: "Dict[str, str]" = Field(..., alias="labels") + model_bundle_id: "str" = Field(..., alias="model_bundle_id") + resource_requests: "CreateBatchJobResourceRequests" = Field(..., alias="resource_requests") + serialization_format: "BatchJobSerializationFormat" = Field(..., alias="serialization_format") + + +class CreateBatchJobResourceRequests(BaseModel): + cpus: "Optional[Any]" = Field(None, alias="cpus") + gpu_type: "Optional[GpuType]" = Field(None, alias="gpu_type") + gpus: "Optional[int]" = Field(None, alias="gpus") + max_workers: "Optional[int]" = Field(None, alias="max_workers") + memory: "Optional[Any]" = Field(None, alias="memory") + per_worker: "Optional[int]" = Field(None, alias="per_worker") + storage: "Optional[Any]" = Field(None, alias="storage") + + +class CreateBatchJobResponse(BaseModel): + job_id: "str" = Field(..., alias="job_id") + + +class CreateModelBundleRequest(BaseModel): + app_config: "Optional[Any]" = Field(None, alias="app_config") + env_params: "ModelBundleEnvironmentParams" = Field(..., alias="env_params") + location: "str" = Field(..., alias="location") + metadata: "Optional[Any]" = Field(None, alias="metadata") + name: "str" = Field(..., alias="name") + packaging_type: "Optional[ModelBundlePackagingType]" = Field(None, alias="packaging_type") + requirements: "List[str]" = Field(..., alias="requirements") + schema_location: "Optional[str]" = Field(None, alias="schema_location") + + +class CreateModelBundleResponse(BaseModel): + model_bundle_id: "str" = Field(..., alias="model_bundle_id") + + +class CreateModelEndpointRequest(BaseModel): + billing_tags: "Optional[Any]" = Field(None, alias="billing_tags") + cpus: "Any" = Field(..., alias="cpus") + default_callback_url: "Optional[str]" = Field(None, alias="default_callback_url") + endpoint_type: "ModelEndpointType" = Field(..., alias="endpoint_type") + gpu_type: "Optional[GpuType]" = Field(None, alias="gpu_type") + gpus: "int" = Field(..., alias="gpus") + labels: "Dict[str, str]" = Field(..., alias="labels") + max_workers: "int" = Field(..., alias="max_workers") + memory: "Any" = Field(..., alias="memory") + metadata: "Any" = Field(..., alias="metadata") + min_workers: "int" = Field(..., alias="min_workers") + model_bundle_id: "str" = Field(..., alias="model_bundle_id") + name: "str" = Field(..., alias="name") + optimize_costs: "Optional[bool]" = Field(None, alias="optimize_costs") + per_worker: "int" = Field(..., alias="per_worker") + post_inference_hooks: "Optional[List[str]]" = Field(None, alias="post_inference_hooks") + prewarm: "Optional[bool]" = Field(None, alias="prewarm") + storage: "Optional[Any]" = Field(None, alias="storage") + + +class CreateModelEndpointResponse(BaseModel): + endpoint_creation_task_id: "str" = Field(..., alias="endpoint_creation_task_id") + + +class DeleteModelEndpointResponse(BaseModel): + deleted: "bool" = Field(..., alias="deleted") + + +class EndpointPredictRequest(BaseModel): + args: "Optional[Any]" = Field(None, alias="args") + callback_url: "Optional[str]" = Field(None, alias="callback_url") + cloudpickle: "Optional[str]" = Field(None, alias="cloudpickle") + return_pickled: "Optional[bool]" = Field(None, alias="return_pickled") + url: "Optional[str]" = Field(None, alias="url") + + +class GetAsyncTaskResponse(BaseModel): + result: "Optional[Any]" = Field(None, alias="result") + status: "TaskStatus" = Field(..., alias="status") + task_id: "str" = Field(..., alias="task_id") + traceback: "Optional[str]" = Field(None, alias="traceback") + + +class GetBatchJobResponse(BaseModel): + duration: "Optional[float]" = Field(None, alias="duration") + num_tasks_completed: "Optional[int]" = Field(None, alias="num_tasks_completed") + num_tasks_pending: "Optional[int]" = Field(None, alias="num_tasks_pending") + result: "Optional[str]" = Field(None, alias="result") + status: "BatchJobStatus" = Field(..., alias="status") + + +class GetModelEndpointResponse(BaseModel): + aws_role: "Optional[str]" = Field(None, alias="aws_role") + bundle_name: "str" = Field(..., alias="bundle_name") + created_at: "datetime" = Field(..., alias="created_at") + created_by: "str" = Field(..., alias="created_by") + default_callback_url: "Optional[str]" = Field(None, alias="default_callback_url") + deployment_name: "Optional[str]" = Field(None, alias="deployment_name") + deployment_state: "Optional[ModelEndpointDeploymentState]" = Field(None, alias="deployment_state") + destination: "str" = Field(..., alias="destination") + endpoint_type: "ModelEndpointType" = Field(..., alias="endpoint_type") + id: "str" = Field(..., alias="id") + labels: "Optional[Dict[str, str]]" = Field(None, alias="labels") + last_updated_at: "datetime" = Field(..., alias="last_updated_at") + metadata: "Optional[Any]" = Field(None, alias="metadata") + name: "str" = Field(..., alias="name") + post_inference_hooks: "Optional[List[str]]" = Field(None, alias="post_inference_hooks") + resource_state: "Optional[ModelEndpointResourceState]" = Field(None, alias="resource_state") + results_s3_bucket: "Optional[str]" = Field(None, alias="results_s3_bucket") + status: "ModelEndpointStatus" = Field(..., alias="status") + + +class GpuType(str, Enum): + TESLA_T4 = "nvidia-tesla-t4" + AMPERE_A10 = "nvidia-ampere-a10" + A100 = "nvidia-a100" + + +class HTTPValidationError(BaseModel): + detail: "Optional[List[ValidationError]]" = Field(None, alias="detail") + + +class ListModelBundlesResponse(BaseModel): + model_bundles: "List[ModelBundleResponse]" = Field(..., alias="model_bundles") + + +class ListModelEndpointsResponse(BaseModel): + model_endpoints: "List[GetModelEndpointResponse]" = Field(..., alias="model_endpoints") + + +class ModelBundleEnvironmentParams(BaseModel): + ecr_repo: "Optional[str]" = Field(None, alias="ecr_repo") + framework_type: "ModelBundleFramework" = Field(..., alias="framework_type") + image_tag: "Optional[str]" = Field(None, alias="image_tag") + pytorch_image_tag: "Optional[str]" = Field(None, alias="pytorch_image_tag") + tensorflow_version: "Optional[str]" = Field(None, alias="tensorflow_version") + + +class ModelBundleFramework(str, Enum): + PYTORCH = "pytorch" + TENSORFLOW = "tensorflow" + CUSTOM_BASE_IMAGE = "custom_base_image" + + +class ModelBundleOrderBy(str, Enum): + NEWEST = "newest" + OLDEST = "oldest" + + +class ModelBundlePackagingType(str, Enum): + CLOUDPICKLE = "cloudpickle" + ZIP = "zip" + + +class ModelBundleResponse(BaseModel): + app_config: "Optional[Any]" = Field(None, alias="app_config") + created_at: "datetime" = Field(..., alias="created_at") + env_params: "ModelBundleEnvironmentParams" = Field(..., alias="env_params") + id: "str" = Field(..., alias="id") + location: "str" = Field(..., alias="location") + metadata: "Any" = Field(..., alias="metadata") + model_artifact_ids: "List[str]" = Field(..., alias="model_artifact_ids") + name: "str" = Field(..., alias="name") + packaging_type: "ModelBundlePackagingType" = Field(..., alias="packaging_type") + requirements: "List[str]" = Field(..., alias="requirements") + schema_location: "Optional[str]" = Field(None, alias="schema_location") + + +class ModelEndpointDeploymentState(BaseModel): + available_workers: "Optional[int]" = Field(None, alias="available_workers") + max_workers: "int" = Field(..., alias="max_workers") + min_workers: "int" = Field(..., alias="min_workers") + per_worker: "int" = Field(..., alias="per_worker") + unavailable_workers: "Optional[int]" = Field(None, alias="unavailable_workers") + + +class ModelEndpointOrderBy(str, Enum): + NEWEST = "newest" + OLDEST = "oldest" + ALPHABETICAL = "alphabetical" + + +class ModelEndpointResourceState(BaseModel): + cpus: "Any" = Field(..., alias="cpus") + gpu_type: "Optional[GpuType]" = Field(None, alias="gpu_type") + gpus: "int" = Field(..., alias="gpus") + memory: "Any" = Field(..., alias="memory") + optimize_costs: "Optional[bool]" = Field(None, alias="optimize_costs") + storage: "Optional[Any]" = Field(None, alias="storage") + + +class ModelEndpointStatus(str, Enum): + READY = "READY" + UPDATE_PENDING = "UPDATE_PENDING" + UPDATE_IN_PROGRESS = "UPDATE_IN_PROGRESS" + UPDATE_FAILED = "UPDATE_FAILED" + DELETE_IN_PROGRESS = "DELETE_IN_PROGRESS" + + +class ModelEndpointType(str, Enum): + ASYNC = "async" + SYNC = "sync" + + +class SyncEndpointPredictResponse(BaseModel): + result: "Optional[Any]" = Field(None, alias="result") + status: "TaskStatus" = Field(..., alias="status") + traceback: "Optional[str]" = Field(None, alias="traceback") + + +class TaskStatus(str, Enum): + PENDING = "PENDING" + STARTED = "STARTED" + SUCCESS = "SUCCESS" + FAILURE = "FAILURE" + UNDEFINED = "UNDEFINED" + + +class UpdateBatchJobRequest(BaseModel): + cancel: "bool" = Field(..., alias="cancel") + + +class UpdateBatchJobResponse(BaseModel): + success: "bool" = Field(..., alias="success") + + +class UpdateModelEndpointRequest(BaseModel): + aws_role: "Optional[str]" = Field(None, alias="aws_role") + billing_tags: "Optional[Any]" = Field(None, alias="billing_tags") + cpus: "Optional[Any]" = Field(None, alias="cpus") + default_callback_url: "Optional[str]" = Field(None, alias="default_callback_url") + gpu_type: "Optional[GpuType]" = Field(None, alias="gpu_type") + gpus: "Optional[int]" = Field(None, alias="gpus") + labels: "Optional[Dict[str, str]]" = Field(None, alias="labels") + max_workers: "Optional[int]" = Field(None, alias="max_workers") + memory: "Optional[Any]" = Field(None, alias="memory") + metadata: "Optional[Any]" = Field(None, alias="metadata") + min_workers: "Optional[int]" = Field(None, alias="min_workers") + model_bundle_id: "Optional[str]" = Field(None, alias="model_bundle_id") + optimize_costs: "Optional[bool]" = Field(None, alias="optimize_costs") + per_worker: "Optional[int]" = Field(None, alias="per_worker") + post_inference_hooks: "Optional[List[str]]" = Field(None, alias="post_inference_hooks") + prewarm: "Optional[bool]" = Field(None, alias="prewarm") + results_s3_bucket: "Optional[str]" = Field(None, alias="results_s3_bucket") + storage: "Optional[Any]" = Field(None, alias="storage") + + +class UpdateModelEndpointResponse(BaseModel): + endpoint_creation_task_id: "str" = Field(..., alias="endpoint_creation_task_id") + + +class ValidationError(BaseModel): + loc: "List[Any]" = Field(..., alias="loc") + msg: "str" = Field(..., alias="msg") + type: "str" = Field(..., alias="type") diff --git a/launch/api_client/models/__init__.py b/launch/api_client/models/__init__.py deleted file mode 100644 index 29cf94aa..00000000 --- a/launch/api_client/models/__init__.py +++ /dev/null @@ -1,104 +0,0 @@ -# coding: utf-8 - -# flake8: noqa - -# import all models into this package -# if you have many models here with many references from one model to another this may -# raise a RecursionError -# to avoid this, import only the models that you directly need like: -# from from launch.api_client.model.pet import Pet -# or import this package, but before doing it, use: -# import sys -# sys.setrecursionlimit(n) - -from launch.api_client.model.batch_job_serialization_format import ( - BatchJobSerializationFormat, -) -from launch.api_client.model.batch_job_status import BatchJobStatus -from launch.api_client.model.clone_model_bundle_request import ( - CloneModelBundleRequest, -) -from launch.api_client.model.create_async_task_response import ( - CreateAsyncTaskResponse, -) -from launch.api_client.model.create_batch_job_request import ( - CreateBatchJobRequest, -) -from launch.api_client.model.create_batch_job_resource_requests import ( - CreateBatchJobResourceRequests, -) -from launch.api_client.model.create_batch_job_response import ( - CreateBatchJobResponse, -) -from launch.api_client.model.create_model_bundle_request import ( - CreateModelBundleRequest, -) -from launch.api_client.model.create_model_bundle_response import ( - CreateModelBundleResponse, -) -from launch.api_client.model.create_model_endpoint_request import ( - CreateModelEndpointRequest, -) -from launch.api_client.model.create_model_endpoint_response import ( - CreateModelEndpointResponse, -) -from launch.api_client.model.delete_model_endpoint_response import ( - DeleteModelEndpointResponse, -) -from launch.api_client.model.endpoint_predict_request import ( - EndpointPredictRequest, -) -from launch.api_client.model.get_async_task_response import ( - GetAsyncTaskResponse, -) -from launch.api_client.model.get_batch_job_response import GetBatchJobResponse -from launch.api_client.model.get_model_endpoint_response import ( - GetModelEndpointResponse, -) -from launch.api_client.model.gpu_type import GpuType -from launch.api_client.model.http_validation_error import HTTPValidationError -from launch.api_client.model.list_model_bundles_response import ( - ListModelBundlesResponse, -) -from launch.api_client.model.list_model_endpoints_response import ( - ListModelEndpointsResponse, -) -from launch.api_client.model.model_bundle_environment_params import ( - ModelBundleEnvironmentParams, -) -from launch.api_client.model.model_bundle_framework import ModelBundleFramework -from launch.api_client.model.model_bundle_order_by import ModelBundleOrderBy -from launch.api_client.model.model_bundle_packaging_type import ( - ModelBundlePackagingType, -) -from launch.api_client.model.model_bundle_response import ModelBundleResponse -from launch.api_client.model.model_endpoint_deployment_state import ( - ModelEndpointDeploymentState, -) -from launch.api_client.model.model_endpoint_order_by import ( - ModelEndpointOrderBy, -) -from launch.api_client.model.model_endpoint_resource_state import ( - ModelEndpointResourceState, -) -from launch.api_client.model.model_endpoint_status import ModelEndpointStatus -from launch.api_client.model.model_endpoint_type import ModelEndpointType -from launch.api_client.model.request_schema import RequestSchema -from launch.api_client.model.response_schema import ResponseSchema -from launch.api_client.model.sync_endpoint_predict_response import ( - SyncEndpointPredictResponse, -) -from launch.api_client.model.task_status import TaskStatus -from launch.api_client.model.update_batch_job_request import ( - UpdateBatchJobRequest, -) -from launch.api_client.model.update_batch_job_response import ( - UpdateBatchJobResponse, -) -from launch.api_client.model.update_model_endpoint_request import ( - UpdateModelEndpointRequest, -) -from launch.api_client.model.update_model_endpoint_response import ( - UpdateModelEndpointResponse, -) -from launch.api_client.model.validation_error import ValidationError diff --git a/launch/client.py b/launch/client.py index 252f302c..ddd64cd2 100644 --- a/launch/client.py +++ b/launch/client.py @@ -14,35 +14,6 @@ from frozendict import frozendict from pydantic import BaseModel -from launch.api_client import ApiClient, Configuration -from launch.api_client.apis.tags.default_api import DefaultApi -from launch.api_client.model.clone_model_bundle_request import ( - CloneModelBundleRequest, -) -from launch.api_client.model.create_batch_job_request import ( - CreateBatchJobRequest, -) -from launch.api_client.model.create_model_bundle_request import ( - CreateModelBundleRequest, -) -from launch.api_client.model.create_model_endpoint_request import ( - CreateModelEndpointRequest, -) -from launch.api_client.model.endpoint_predict_request import ( - EndpointPredictRequest, -) -from launch.api_client.model.gpu_type import GpuType -from launch.api_client.model.model_bundle_environment_params import ( - ModelBundleEnvironmentParams, -) -from launch.api_client.model.model_bundle_framework import ModelBundleFramework -from launch.api_client.model.model_bundle_packaging_type import ( - ModelBundlePackagingType, -) -from launch.api_client.model.model_endpoint_type import ModelEndpointType -from launch.api_client.model.update_model_endpoint_request import ( - UpdateModelEndpointRequest, -) from launch.connection import Connection from launch.constants import ( BATCH_TASK_INPUT_SIGNED_URL_PATH, @@ -64,6 +35,37 @@ ModelEndpoint, SyncEndpoint, ) +from launch.openapi_client import ApiClient, Configuration +from launch.openapi_client.apis.tags.default_api import DefaultApi +from launch.openapi_client.model.clone_model_bundle_request import ( + CloneModelBundleRequest, +) +from launch.openapi_client.model.create_batch_job_request import ( + CreateBatchJobRequest, +) +from launch.openapi_client.model.create_model_bundle_request import ( + CreateModelBundleRequest, +) +from launch.openapi_client.model.create_model_endpoint_request import ( + CreateModelEndpointRequest, +) +from launch.openapi_client.model.endpoint_predict_request import ( + EndpointPredictRequest, +) +from launch.openapi_client.model.gpu_type import GpuType +from launch.openapi_client.model.model_bundle_environment_params import ( + ModelBundleEnvironmentParams, +) +from launch.openapi_client.model.model_bundle_framework import ( + ModelBundleFramework, +) +from launch.openapi_client.model.model_bundle_packaging_type import ( + ModelBundlePackagingType, +) +from launch.openapi_client.model.model_endpoint_type import ModelEndpointType +from launch.openapi_client.model.update_model_endpoint_request import ( + UpdateModelEndpointRequest, +) from launch.pydantic_schemas import get_model_definitions from launch.request_validation import validate_task_request diff --git a/launch/model_endpoint.py b/launch/model_endpoint.py index 8bf220e3..847ffa34 100644 --- a/launch/model_endpoint.py +++ b/launch/model_endpoint.py @@ -10,8 +10,8 @@ from dataclasses_json import Undefined, dataclass_json from deprecation import deprecated -from launch.api_client import ApiClient -from launch.api_client.apis.tags.default_api import DefaultApi +from launch.openapi_client import ApiClient +from launch.openapi_client.apis.tags.default_api import DefaultApi from launch.request_validation import validate_task_request TASK_PENDING_STATE = "PENDING" diff --git a/launch/openapi_client/__init__.py b/launch/openapi_client/__init__.py new file mode 100644 index 00000000..be53ca86 --- /dev/null +++ b/launch/openapi_client/__init__.py @@ -0,0 +1,30 @@ +# coding: utf-8 + +# flake8: noqa + +""" + launch + + No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator) # noqa: E501 + + The version of the OpenAPI document: 1.0.0 + Generated by: https://openapi-generator.tech +""" + +__version__ = "1.0.5" + +# import ApiClient +from launch.openapi_client.api_client import ApiClient + +# import Configuration +from launch.openapi_client.configuration import Configuration + +# import exceptions +from launch.openapi_client.exceptions import ( + ApiAttributeError, + ApiException, + ApiKeyError, + ApiTypeError, + ApiValueError, + OpenApiException, +) diff --git a/launch/openapi_client/api_client.py b/launch/openapi_client/api_client.py new file mode 100644 index 00000000..8ac49af2 --- /dev/null +++ b/launch/openapi_client/api_client.py @@ -0,0 +1,1543 @@ +# coding: utf-8 +""" + launch + + No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator) # noqa: E501 + + The version of the OpenAPI document: 1.0.0 + Generated by: https://openapi-generator.tech +""" + +import atexit +import email +import enum +import io +import json +import os +import re +import tempfile +import typing +from dataclasses import dataclass +from decimal import Decimal +from multiprocessing.pool import ThreadPool +from urllib.parse import quote, urlparse + +import frozendict +import typing_extensions +import urllib3 +from urllib3._collections import HTTPHeaderDict +from urllib3.fields import RequestField as RequestFieldBase + +from launch.openapi_client import rest +from launch.openapi_client.configuration import Configuration +from launch.openapi_client.exceptions import ApiTypeError, ApiValueError +from launch.openapi_client.schemas import ( + BinarySchema, + BoolClass, + FileIO, + NoneClass, + Schema, + Unset, + date, + datetime, + none_type, + unset, +) + + +class RequestField(RequestFieldBase): + def __eq__(self, other): + if not isinstance(other, RequestField): + return False + return self.__dict__ == other.__dict__ + + +class JSONEncoder(json.JSONEncoder): + compact_separators = (",", ":") + + def default(self, obj): + if isinstance(obj, str): + return str(obj) + elif isinstance(obj, float): + return float(obj) + elif isinstance(obj, int): + return int(obj) + elif isinstance(obj, Decimal): + if obj.as_tuple().exponent >= 0: + return int(obj) + return float(obj) + elif isinstance(obj, NoneClass): + return None + elif isinstance(obj, BoolClass): + return bool(obj) + elif isinstance(obj, (dict, frozendict.frozendict)): + return {key: self.default(val) for key, val in obj.items()} + elif isinstance(obj, (list, tuple)): + return [self.default(item) for item in obj] + raise ApiValueError("Unable to prepare type {} for serialization".format(obj.__class__.__name__)) + + +class ParameterInType(enum.Enum): + QUERY = "query" + HEADER = "header" + PATH = "path" + COOKIE = "cookie" + + +class ParameterStyle(enum.Enum): + MATRIX = "matrix" + LABEL = "label" + FORM = "form" + SIMPLE = "simple" + SPACE_DELIMITED = "spaceDelimited" + PIPE_DELIMITED = "pipeDelimited" + DEEP_OBJECT = "deepObject" + + +class PrefixSeparatorIterator: + # A class to store prefixes and separators for rfc6570 expansions + + def __init__(self, prefix: str, separator: str): + self.prefix = prefix + self.separator = separator + self.first = True + if separator in {".", "|", "%20"}: + item_separator = separator + else: + item_separator = "," + self.item_separator = item_separator + + def __iter__(self): + return self + + def __next__(self): + if self.first: + self.first = False + return self.prefix + return self.separator + + +class ParameterSerializerBase: + @classmethod + def _get_default_explode(cls, style: ParameterStyle) -> bool: + return False + + @staticmethod + def __ref6570_item_value(in_data: typing.Any, percent_encode: bool): + """ + Get representation if str/float/int/None/items in list/ values in dict + None is returned if an item is undefined, use cases are value= + - None + - [] + - {} + - [None, None None] + - {'a': None, 'b': None} + """ + if type(in_data) in {str, float, int}: + if percent_encode: + return quote(str(in_data)) + return str(in_data) + elif isinstance(in_data, none_type): + # ignored by the expansion process https://datatracker.ietf.org/doc/html/rfc6570#section-3.2.1 + return None + elif isinstance(in_data, list) and not in_data: + # ignored by the expansion process https://datatracker.ietf.org/doc/html/rfc6570#section-3.2.1 + return None + elif isinstance(in_data, dict) and not in_data: + # ignored by the expansion process https://datatracker.ietf.org/doc/html/rfc6570#section-3.2.1 + return None + raise ApiValueError("Unable to generate a ref6570 item representation of {}".format(in_data)) + + @staticmethod + def _to_dict(name: str, value: str): + return {name: value} + + @classmethod + def __ref6570_str_float_int_expansion( + cls, + variable_name: str, + in_data: typing.Any, + explode: bool, + percent_encode: bool, + prefix_separator_iterator: PrefixSeparatorIterator, + var_name_piece: str, + named_parameter_expansion: bool, + ) -> str: + item_value = cls.__ref6570_item_value(in_data, percent_encode) + if item_value is None or (item_value == "" and prefix_separator_iterator.separator == ";"): + return next(prefix_separator_iterator) + var_name_piece + value_pair_equals = "=" if named_parameter_expansion else "" + return next(prefix_separator_iterator) + var_name_piece + value_pair_equals + item_value + + @classmethod + def __ref6570_list_expansion( + cls, + variable_name: str, + in_data: typing.Any, + explode: bool, + percent_encode: bool, + prefix_separator_iterator: PrefixSeparatorIterator, + var_name_piece: str, + named_parameter_expansion: bool, + ) -> str: + item_values = [cls.__ref6570_item_value(v, percent_encode) for v in in_data] + item_values = [v for v in item_values if v is not None] + if not item_values: + # ignored by the expansion process https://datatracker.ietf.org/doc/html/rfc6570#section-3.2.1 + return "" + value_pair_equals = "=" if named_parameter_expansion else "" + if not explode: + return ( + next(prefix_separator_iterator) + + var_name_piece + + value_pair_equals + + prefix_separator_iterator.item_separator.join(item_values) + ) + # exploded + return next(prefix_separator_iterator) + next(prefix_separator_iterator).join( + [var_name_piece + value_pair_equals + val for val in item_values] + ) + + @classmethod + def __ref6570_dict_expansion( + cls, + variable_name: str, + in_data: typing.Any, + explode: bool, + percent_encode: bool, + prefix_separator_iterator: PrefixSeparatorIterator, + var_name_piece: str, + named_parameter_expansion: bool, + ) -> str: + in_data_transformed = {key: cls.__ref6570_item_value(val, percent_encode) for key, val in in_data.items()} + in_data_transformed = {key: val for key, val in in_data_transformed.items() if val is not None} + if not in_data_transformed: + # ignored by the expansion process https://datatracker.ietf.org/doc/html/rfc6570#section-3.2.1 + return "" + value_pair_equals = "=" if named_parameter_expansion else "" + if not explode: + return ( + next(prefix_separator_iterator) + + var_name_piece + + value_pair_equals + + prefix_separator_iterator.item_separator.join( + prefix_separator_iterator.item_separator.join(item_pair) + for item_pair in in_data_transformed.items() + ) + ) + # exploded + return next(prefix_separator_iterator) + next(prefix_separator_iterator).join( + [key + "=" + val for key, val in in_data_transformed.items()] + ) + + @classmethod + def _ref6570_expansion( + cls, + variable_name: str, + in_data: typing.Any, + explode: bool, + percent_encode: bool, + prefix_separator_iterator: PrefixSeparatorIterator, + ) -> str: + """ + Separator is for separate variables like dict with explode true, not for array item separation + """ + named_parameter_expansion = prefix_separator_iterator.separator in { + "&", + ";", + } + var_name_piece = variable_name if named_parameter_expansion else "" + if type(in_data) in {str, float, int}: + return cls.__ref6570_str_float_int_expansion( + variable_name, + in_data, + explode, + percent_encode, + prefix_separator_iterator, + var_name_piece, + named_parameter_expansion, + ) + elif isinstance(in_data, none_type): + # ignored by the expansion process https://datatracker.ietf.org/doc/html/rfc6570#section-3.2.1 + return "" + elif isinstance(in_data, list): + return cls.__ref6570_list_expansion( + variable_name, + in_data, + explode, + percent_encode, + prefix_separator_iterator, + var_name_piece, + named_parameter_expansion, + ) + elif isinstance(in_data, dict): + return cls.__ref6570_dict_expansion( + variable_name, + in_data, + explode, + percent_encode, + prefix_separator_iterator, + var_name_piece, + named_parameter_expansion, + ) + # bool, bytes, etc + raise ApiValueError("Unable to generate a ref6570 representation of {}".format(in_data)) + + +class StyleFormSerializer(ParameterSerializerBase): + @classmethod + def _get_default_explode(cls, style: ParameterStyle) -> bool: + if style is ParameterStyle.FORM: + return True + return super()._get_default_explode(style) + + def _serialize_form( + self, + in_data: typing.Union[None, int, float, str, bool, dict, list], + name: str, + explode: bool, + percent_encode: bool, + prefix_separator_iterator: typing.Optional[PrefixSeparatorIterator] = None, + ) -> str: + if prefix_separator_iterator is None: + prefix_separator_iterator = PrefixSeparatorIterator("", "&") + return self._ref6570_expansion( + variable_name=name, + in_data=in_data, + explode=explode, + percent_encode=percent_encode, + prefix_separator_iterator=prefix_separator_iterator, + ) + + +class StyleSimpleSerializer(ParameterSerializerBase): + def _serialize_simple( + self, + in_data: typing.Union[None, int, float, str, bool, dict, list], + name: str, + explode: bool, + percent_encode: bool, + ) -> str: + prefix_separator_iterator = PrefixSeparatorIterator("", ",") + return self._ref6570_expansion( + variable_name=name, + in_data=in_data, + explode=explode, + percent_encode=percent_encode, + prefix_separator_iterator=prefix_separator_iterator, + ) + + +class JSONDetector: + """ + Works for: + application/json + application/json; charset=UTF-8 + application/json-patch+json + application/geo+json + """ + + __json_content_type_pattern = re.compile("application/[^+]*[+]?(json);?.*") + + @classmethod + def _content_type_is_json(cls, content_type: str) -> bool: + if cls.__json_content_type_pattern.match(content_type): + return True + return False + + +@dataclass +class ParameterBase(JSONDetector): + name: str + in_type: ParameterInType + required: bool + style: typing.Optional[ParameterStyle] + explode: typing.Optional[bool] + allow_reserved: typing.Optional[bool] + schema: typing.Optional[typing.Type[Schema]] + content: typing.Optional[typing.Dict[str, typing.Type[Schema]]] + + __style_to_in_type = { + ParameterStyle.MATRIX: {ParameterInType.PATH}, + ParameterStyle.LABEL: {ParameterInType.PATH}, + ParameterStyle.FORM: {ParameterInType.QUERY, ParameterInType.COOKIE}, + ParameterStyle.SIMPLE: {ParameterInType.PATH, ParameterInType.HEADER}, + ParameterStyle.SPACE_DELIMITED: {ParameterInType.QUERY}, + ParameterStyle.PIPE_DELIMITED: {ParameterInType.QUERY}, + ParameterStyle.DEEP_OBJECT: {ParameterInType.QUERY}, + } + __in_type_to_default_style = { + ParameterInType.QUERY: ParameterStyle.FORM, + ParameterInType.PATH: ParameterStyle.SIMPLE, + ParameterInType.HEADER: ParameterStyle.SIMPLE, + ParameterInType.COOKIE: ParameterStyle.FORM, + } + __disallowed_header_names = {"Accept", "Content-Type", "Authorization"} + _json_encoder = JSONEncoder() + + @classmethod + def __verify_style_to_in_type(cls, style: typing.Optional[ParameterStyle], in_type: ParameterInType): + if style is None: + return + in_type_set = cls.__style_to_in_type[style] + if in_type not in in_type_set: + raise ValueError( + "Invalid style and in_type combination. For style={} only in_type={} are allowed".format( + style, in_type_set + ) + ) + + def __init__( + self, + name: str, + in_type: ParameterInType, + required: bool = False, + style: typing.Optional[ParameterStyle] = None, + explode: bool = False, + allow_reserved: typing.Optional[bool] = None, + schema: typing.Optional[typing.Type[Schema]] = None, + content: typing.Optional[typing.Dict[str, typing.Type[Schema]]] = None, + ): + if schema is None and content is None: + raise ValueError("Value missing; Pass in either schema or content") + if schema and content: + raise ValueError("Too many values provided. Both schema and content were provided. Only one may be input") + if name in self.__disallowed_header_names and in_type is ParameterInType.HEADER: + raise ValueError("Invalid name, name may not be one of {}".format(self.__disallowed_header_names)) + self.__verify_style_to_in_type(style, in_type) + if content is None and style is None: + style = self.__in_type_to_default_style[in_type] + if content is not None and in_type in self.__in_type_to_default_style and len(content) != 1: + raise ValueError("Invalid content length, content length must equal 1") + self.in_type = in_type + self.name = name + self.required = required + self.style = style + self.explode = explode + self.allow_reserved = allow_reserved + self.schema = schema + self.content = content + + def _serialize_json( + self, + in_data: typing.Union[None, int, float, str, bool, dict, list], + eliminate_whitespace: bool = False, + ) -> str: + if eliminate_whitespace: + return json.dumps(in_data, separators=self._json_encoder.compact_separators) + return json.dumps(in_data) + + +class PathParameter(ParameterBase, StyleSimpleSerializer): + def __init__( + self, + name: str, + required: bool = False, + style: typing.Optional[ParameterStyle] = None, + explode: bool = False, + allow_reserved: typing.Optional[bool] = None, + schema: typing.Optional[typing.Type[Schema]] = None, + content: typing.Optional[typing.Dict[str, typing.Type[Schema]]] = None, + ): + super().__init__( + name, + in_type=ParameterInType.PATH, + required=required, + style=style, + explode=explode, + allow_reserved=allow_reserved, + schema=schema, + content=content, + ) + + def __serialize_label( + self, in_data: typing.Union[None, int, float, str, bool, dict, list] + ) -> typing.Dict[str, str]: + prefix_separator_iterator = PrefixSeparatorIterator(".", ".") + value = self._ref6570_expansion( + variable_name=self.name, + in_data=in_data, + explode=self.explode, + percent_encode=True, + prefix_separator_iterator=prefix_separator_iterator, + ) + return self._to_dict(self.name, value) + + def __serialize_matrix( + self, in_data: typing.Union[None, int, float, str, bool, dict, list] + ) -> typing.Dict[str, str]: + prefix_separator_iterator = PrefixSeparatorIterator(";", ";") + value = self._ref6570_expansion( + variable_name=self.name, + in_data=in_data, + explode=self.explode, + percent_encode=True, + prefix_separator_iterator=prefix_separator_iterator, + ) + return self._to_dict(self.name, value) + + def __serialize_simple( + self, + in_data: typing.Union[None, int, float, str, bool, dict, list], + ) -> typing.Dict[str, str]: + value = self._serialize_simple( + in_data=in_data, + name=self.name, + explode=self.explode, + percent_encode=True, + ) + return self._to_dict(self.name, value) + + def serialize( + self, + in_data: typing.Union[ + Schema, + Decimal, + int, + float, + str, + date, + datetime, + None, + bool, + list, + tuple, + dict, + frozendict.frozendict, + ], + ) -> typing.Dict[str, str]: + if self.schema: + cast_in_data = self.schema(in_data) + cast_in_data = self._json_encoder.default(cast_in_data) + """ + simple -> path + path: + returns path_params: dict + label -> path + returns path_params + matrix -> path + returns path_params + """ + if self.style: + if self.style is ParameterStyle.SIMPLE: + return self.__serialize_simple(cast_in_data) + elif self.style is ParameterStyle.LABEL: + return self.__serialize_label(cast_in_data) + elif self.style is ParameterStyle.MATRIX: + return self.__serialize_matrix(cast_in_data) + # self.content will be length one + for content_type, schema in self.content.items(): + cast_in_data = schema(in_data) + cast_in_data = self._json_encoder.default(cast_in_data) + if self._content_type_is_json(content_type): + value = self._serialize_json(cast_in_data) + return self._to_dict(self.name, value) + raise NotImplementedError("Serialization of {} has not yet been implemented".format(content_type)) + + +class QueryParameter(ParameterBase, StyleFormSerializer): + def __init__( + self, + name: str, + required: bool = False, + style: typing.Optional[ParameterStyle] = None, + explode: typing.Optional[bool] = None, + allow_reserved: typing.Optional[bool] = None, + schema: typing.Optional[typing.Type[Schema]] = None, + content: typing.Optional[typing.Dict[str, typing.Type[Schema]]] = None, + ): + used_style = ParameterStyle.FORM if style is None else style + used_explode = self._get_default_explode(used_style) if explode is None else explode + + super().__init__( + name, + in_type=ParameterInType.QUERY, + required=required, + style=used_style, + explode=used_explode, + allow_reserved=allow_reserved, + schema=schema, + content=content, + ) + + def __serialize_space_delimited( + self, + in_data: typing.Union[None, int, float, str, bool, dict, list], + prefix_separator_iterator: typing.Optional[PrefixSeparatorIterator], + ) -> typing.Dict[str, str]: + if prefix_separator_iterator is None: + prefix_separator_iterator = self.get_prefix_separator_iterator() + value = self._ref6570_expansion( + variable_name=self.name, + in_data=in_data, + explode=self.explode, + percent_encode=True, + prefix_separator_iterator=prefix_separator_iterator, + ) + return self._to_dict(self.name, value) + + def __serialize_pipe_delimited( + self, + in_data: typing.Union[None, int, float, str, bool, dict, list], + prefix_separator_iterator: typing.Optional[PrefixSeparatorIterator], + ) -> typing.Dict[str, str]: + if prefix_separator_iterator is None: + prefix_separator_iterator = self.get_prefix_separator_iterator() + value = self._ref6570_expansion( + variable_name=self.name, + in_data=in_data, + explode=self.explode, + percent_encode=True, + prefix_separator_iterator=prefix_separator_iterator, + ) + return self._to_dict(self.name, value) + + def __serialize_form( + self, + in_data: typing.Union[None, int, float, str, bool, dict, list], + prefix_separator_iterator: typing.Optional[PrefixSeparatorIterator], + ) -> typing.Dict[str, str]: + if prefix_separator_iterator is None: + prefix_separator_iterator = self.get_prefix_separator_iterator() + value = self._serialize_form( + in_data, + name=self.name, + explode=self.explode, + percent_encode=True, + prefix_separator_iterator=prefix_separator_iterator, + ) + return self._to_dict(self.name, value) + + def get_prefix_separator_iterator( + self, + ) -> typing.Optional[PrefixSeparatorIterator]: + if self.style is ParameterStyle.FORM: + return PrefixSeparatorIterator("?", "&") + elif self.style is ParameterStyle.SPACE_DELIMITED: + return PrefixSeparatorIterator("", "%20") + elif self.style is ParameterStyle.PIPE_DELIMITED: + return PrefixSeparatorIterator("", "|") + + def serialize( + self, + in_data: typing.Union[ + Schema, + Decimal, + int, + float, + str, + date, + datetime, + None, + bool, + list, + tuple, + dict, + frozendict.frozendict, + ], + prefix_separator_iterator: typing.Optional[PrefixSeparatorIterator] = None, + ) -> typing.Dict[str, str]: + if self.schema: + cast_in_data = self.schema(in_data) + cast_in_data = self._json_encoder.default(cast_in_data) + """ + form -> query + query: + - GET/HEAD/DELETE: could use fields + - PUT/POST: must use urlencode to send parameters + returns fields: tuple + spaceDelimited -> query + returns fields + pipeDelimited -> query + returns fields + deepObject -> query, https://github.com/OAI/OpenAPI-Specification/issues/1706 + returns fields + """ + if self.style: + # TODO update query ones to omit setting values when [] {} or None is input + if self.style is ParameterStyle.FORM: + return self.__serialize_form(cast_in_data, prefix_separator_iterator) + elif self.style is ParameterStyle.SPACE_DELIMITED: + return self.__serialize_space_delimited(cast_in_data, prefix_separator_iterator) + elif self.style is ParameterStyle.PIPE_DELIMITED: + return self.__serialize_pipe_delimited(cast_in_data, prefix_separator_iterator) + # self.content will be length one + if prefix_separator_iterator is None: + prefix_separator_iterator = self.get_prefix_separator_iterator() + for content_type, schema in self.content.items(): + cast_in_data = schema(in_data) + cast_in_data = self._json_encoder.default(cast_in_data) + if self._content_type_is_json(content_type): + value = self._serialize_json(cast_in_data, eliminate_whitespace=True) + return self._to_dict( + self.name, + next(prefix_separator_iterator) + self.name + "=" + quote(value), + ) + raise NotImplementedError("Serialization of {} has not yet been implemented".format(content_type)) + + +class CookieParameter(ParameterBase, StyleFormSerializer): + def __init__( + self, + name: str, + required: bool = False, + style: typing.Optional[ParameterStyle] = None, + explode: typing.Optional[bool] = None, + allow_reserved: typing.Optional[bool] = None, + schema: typing.Optional[typing.Type[Schema]] = None, + content: typing.Optional[typing.Dict[str, typing.Type[Schema]]] = None, + ): + used_style = ParameterStyle.FORM if style is None and content is None and schema else style + used_explode = self._get_default_explode(used_style) if explode is None else explode + + super().__init__( + name, + in_type=ParameterInType.COOKIE, + required=required, + style=used_style, + explode=used_explode, + allow_reserved=allow_reserved, + schema=schema, + content=content, + ) + + def serialize( + self, + in_data: typing.Union[ + Schema, + Decimal, + int, + float, + str, + date, + datetime, + None, + bool, + list, + tuple, + dict, + frozendict.frozendict, + ], + ) -> typing.Dict[str, str]: + if self.schema: + cast_in_data = self.schema(in_data) + cast_in_data = self._json_encoder.default(cast_in_data) + """ + form -> cookie + returns fields: tuple + """ + if self.style: + """ + TODO add escaping of comma, space, equals + or turn encoding on + """ + value = self._serialize_form( + cast_in_data, + explode=self.explode, + name=self.name, + percent_encode=False, + prefix_separator_iterator=PrefixSeparatorIterator("", "&"), + ) + return self._to_dict(self.name, value) + # self.content will be length one + for content_type, schema in self.content.items(): + cast_in_data = schema(in_data) + cast_in_data = self._json_encoder.default(cast_in_data) + if self._content_type_is_json(content_type): + value = self._serialize_json(cast_in_data) + return self._to_dict(self.name, value) + raise NotImplementedError("Serialization of {} has not yet been implemented".format(content_type)) + + +class HeaderParameter(ParameterBase, StyleSimpleSerializer): + def __init__( + self, + name: str, + required: bool = False, + style: typing.Optional[ParameterStyle] = None, + explode: bool = False, + allow_reserved: typing.Optional[bool] = None, + schema: typing.Optional[typing.Type[Schema]] = None, + content: typing.Optional[typing.Dict[str, typing.Type[Schema]]] = None, + ): + super().__init__( + name, + in_type=ParameterInType.HEADER, + required=required, + style=style, + explode=explode, + allow_reserved=allow_reserved, + schema=schema, + content=content, + ) + + @staticmethod + def __to_headers(in_data: typing.Tuple[typing.Tuple[str, str], ...]) -> HTTPHeaderDict: + data = tuple(t for t in in_data if t) + headers = HTTPHeaderDict() + if not data: + return headers + headers.extend(data) + return headers + + def serialize( + self, + in_data: typing.Union[ + Schema, + Decimal, + int, + float, + str, + date, + datetime, + None, + bool, + list, + tuple, + dict, + frozendict.frozendict, + ], + ) -> HTTPHeaderDict: + if self.schema: + cast_in_data = self.schema(in_data) + cast_in_data = self._json_encoder.default(cast_in_data) + """ + simple -> header + headers: PoolManager needs a mapping, tuple is close + returns headers: dict + """ + if self.style: + value = self._serialize_simple(cast_in_data, self.name, self.explode, False) + return self.__to_headers(((self.name, value),)) + # self.content will be length one + for content_type, schema in self.content.items(): + cast_in_data = schema(in_data) + cast_in_data = self._json_encoder.default(cast_in_data) + if self._content_type_is_json(content_type): + value = self._serialize_json(cast_in_data) + return self.__to_headers(((self.name, value),)) + raise NotImplementedError("Serialization of {} has not yet been implemented".format(content_type)) + + +class Encoding: + def __init__( + self, + content_type: str, + headers: typing.Optional[typing.Dict[str, HeaderParameter]] = None, + style: typing.Optional[ParameterStyle] = None, + explode: bool = False, + allow_reserved: bool = False, + ): + self.content_type = content_type + self.headers = headers + self.style = style + self.explode = explode + self.allow_reserved = allow_reserved + + +@dataclass +class MediaType: + """ + Used to store request and response body schema information + encoding: + A map between a property name and its encoding information. + The key, being the property name, MUST exist in the schema as a property. + The encoding object SHALL only apply to requestBody objects when the media type is + multipart or application/x-www-form-urlencoded. + """ + + schema: typing.Optional[typing.Type[Schema]] = None + encoding: typing.Optional[typing.Dict[str, Encoding]] = None + + +@dataclass +class ApiResponse: + response: urllib3.HTTPResponse + body: typing.Union[Unset, Schema] + headers: typing.Union[Unset, typing.List[HeaderParameter]] + + def __init__( + self, + response: urllib3.HTTPResponse, + body: typing.Union[Unset, typing.Type[Schema]], + headers: typing.Union[Unset, typing.List[HeaderParameter]], + ): + """ + pycharm needs this to prevent 'Unexpected argument' warnings + """ + self.response = response + self.body = body + self.headers = headers + + +@dataclass +class ApiResponseWithoutDeserialization(ApiResponse): + response: urllib3.HTTPResponse + body: typing.Union[Unset, typing.Type[Schema]] = unset + headers: typing.Union[Unset, typing.List[HeaderParameter]] = unset + + +class OpenApiResponse(JSONDetector): + __filename_content_disposition_pattern = re.compile('filename="(.+?)"') + + def __init__( + self, + response_cls: typing.Type[ApiResponse] = ApiResponse, + content: typing.Optional[typing.Dict[str, MediaType]] = None, + headers: typing.Optional[typing.List[HeaderParameter]] = None, + ): + self.headers = headers + if content is not None and len(content) == 0: + raise ValueError("Invalid value for content, the content dict must have >= 1 entry") + self.content = content + self.response_cls = response_cls + + @staticmethod + def __deserialize_json(response: urllib3.HTTPResponse) -> typing.Any: + # python must be >= 3.9 so we can pass in bytes into json.loads + return json.loads(response.data) + + @staticmethod + def __file_name_from_response_url( + response_url: typing.Optional[str], + ) -> typing.Optional[str]: + if response_url is None: + return None + url_path = urlparse(response_url).path + if url_path: + path_basename = os.path.basename(url_path) + if path_basename: + _filename, ext = os.path.splitext(path_basename) + if ext: + return path_basename + return None + + @classmethod + def __file_name_from_content_disposition(cls, content_disposition: typing.Optional[str]) -> typing.Optional[str]: + if content_disposition is None: + return None + match = cls.__filename_content_disposition_pattern.search(content_disposition) + if not match: + return None + return match.group(1) + + def __deserialize_application_octet_stream( + self, response: urllib3.HTTPResponse + ) -> typing.Union[bytes, io.BufferedReader]: + """ + urllib3 use cases: + 1. when preload_content=True (stream=False) then supports_chunked_reads is False and bytes are returned + 2. when preload_content=False (stream=True) then supports_chunked_reads is True and + a file will be written and returned + """ + if response.supports_chunked_reads(): + file_name = self.__file_name_from_content_disposition( + response.headers.get("content-disposition") + ) or self.__file_name_from_response_url(response.geturl()) + + if file_name is None: + _fd, path = tempfile.mkstemp() + else: + path = os.path.join(tempfile.gettempdir(), file_name) + + with open(path, "wb") as new_file: + chunk_size = 1024 + while True: + data = response.read(chunk_size) + if not data: + break + new_file.write(data) + # release_conn is needed for streaming connections only + response.release_conn() + new_file = open(path, "rb") + return new_file + else: + return response.data + + @staticmethod + def __deserialize_multipart_form_data( + response: urllib3.HTTPResponse, + ) -> typing.Dict[str, typing.Any]: + msg = email.message_from_bytes(response.data) + return { + part.get_param("name", header="Content-Disposition"): part.get_payload(decode=True).decode( + part.get_content_charset() + ) + if part.get_content_charset() + else part.get_payload() + for part in msg.get_payload() + } + + def deserialize(self, response: urllib3.HTTPResponse, configuration: Configuration) -> ApiResponse: + content_type = response.getheader("content-type") + deserialized_body = unset + streamed = response.supports_chunked_reads() + + deserialized_headers = unset + if self.headers is not None: + # TODO add header deserialiation here + pass + + if self.content is not None: + if content_type not in self.content: + raise ApiValueError( + f"Invalid content_type returned. Content_type='{content_type}' was returned " + f"when only {str(set(self.content))} are defined for status_code={str(response.status)}" + ) + body_schema = self.content[content_type].schema + if body_schema is None: + # some specs do not define response content media type schemas + return self.response_cls(response=response, headers=deserialized_headers, body=unset) + + if self._content_type_is_json(content_type): + body_data = self.__deserialize_json(response) + elif content_type == "application/octet-stream": + body_data = self.__deserialize_application_octet_stream(response) + elif content_type.startswith("multipart/form-data"): + body_data = self.__deserialize_multipart_form_data(response) + content_type = "multipart/form-data" + else: + raise NotImplementedError("Deserialization of {} has not yet been implemented".format(content_type)) + deserialized_body = body_schema.from_openapi_data_oapg(body_data, _configuration=configuration) + elif streamed: + response.release_conn() + + return self.response_cls( + response=response, + headers=deserialized_headers, + body=deserialized_body, + ) + + +class ApiClient: + """Generic API client for OpenAPI client library builds. + + OpenAPI generic API client. This client handles the client- + server communication, and is invariant across implementations. Specifics of + the methods and models for each application are generated from the OpenAPI + templates. + + NOTE: This class is auto generated by OpenAPI Generator. + Ref: https://openapi-generator.tech + Do not edit the class manually. + + :param configuration: .Configuration object for this client + :param header_name: a header to pass when making calls to the API. + :param header_value: a header value to pass when making calls to + the API. + :param cookie: a cookie to include in the header when making calls + to the API + :param pool_threads: The number of threads to use for async requests + to the API. More threads means more concurrent API requests. + """ + + _pool = None + + def __init__( + self, + configuration: typing.Optional[Configuration] = None, + header_name: typing.Optional[str] = None, + header_value: typing.Optional[str] = None, + cookie: typing.Optional[str] = None, + pool_threads: int = 1, + ): + if configuration is None: + configuration = Configuration() + self.configuration = configuration + self.pool_threads = pool_threads + + self.rest_client = rest.RESTClientObject(configuration) + self.default_headers = HTTPHeaderDict() + if header_name is not None: + self.default_headers[header_name] = header_value + self.cookie = cookie + # Set default User-Agent. + self.user_agent = "OpenAPI-Generator/1.0.5/python" + + def __enter__(self): + return self + + def __exit__(self, exc_type, exc_value, traceback): + self.close() + + def close(self): + if self._pool: + self._pool.close() + self._pool.join() + self._pool = None + if hasattr(atexit, "unregister"): + atexit.unregister(self.close) + + @property + def pool(self): + """Create thread pool on first request + avoids instantiating unused threadpool for blocking clients. + """ + if self._pool is None: + atexit.register(self.close) + self._pool = ThreadPool(self.pool_threads) + return self._pool + + @property + def user_agent(self): + """User agent for this API client""" + return self.default_headers["User-Agent"] + + @user_agent.setter + def user_agent(self, value): + self.default_headers["User-Agent"] = value + + def set_default_header(self, header_name, header_value): + self.default_headers[header_name] = header_value + + def __call_api( + self, + resource_path: str, + method: str, + headers: typing.Optional[HTTPHeaderDict] = None, + body: typing.Optional[typing.Union[str, bytes]] = None, + fields: typing.Optional[typing.Tuple[typing.Tuple[str, str], ...]] = None, + auth_settings: typing.Optional[typing.List[str]] = None, + stream: bool = False, + timeout: typing.Optional[typing.Union[int, typing.Tuple]] = None, + host: typing.Optional[str] = None, + ) -> urllib3.HTTPResponse: + # header parameters + used_headers = HTTPHeaderDict(self.default_headers) + if self.cookie: + headers["Cookie"] = self.cookie + + # auth setting + self.update_params_for_auth(used_headers, auth_settings, resource_path, method, body) + + # must happen after cookie setting and auth setting in case user is overriding those + if headers: + used_headers.update(headers) + + # request url + if host is None: + url = self.configuration.host + resource_path + else: + # use server/host defined in path or operation instead + url = host + resource_path + + # perform request and return response + response = self.request( + method, + url, + headers=used_headers, + fields=fields, + body=body, + stream=stream, + timeout=timeout, + ) + return response + + def call_api( + self, + resource_path: str, + method: str, + headers: typing.Optional[HTTPHeaderDict] = None, + body: typing.Optional[typing.Union[str, bytes]] = None, + fields: typing.Optional[typing.Tuple[typing.Tuple[str, str], ...]] = None, + auth_settings: typing.Optional[typing.List[str]] = None, + async_req: typing.Optional[bool] = None, + stream: bool = False, + timeout: typing.Optional[typing.Union[int, typing.Tuple]] = None, + host: typing.Optional[str] = None, + ) -> urllib3.HTTPResponse: + """Makes the HTTP request (synchronous) and returns deserialized data. + + To make an async_req request, set the async_req parameter. + + :param resource_path: Path to method endpoint. + :param method: Method to call. + :param headers: Header parameters to be + placed in the request header. + :param body: Request body. + :param fields: Request post form parameters, + for `application/x-www-form-urlencoded`, `multipart/form-data`. + :param auth_settings: Auth Settings names for the request. + :param async_req: execute request asynchronously + :type async_req: bool, optional TODO remove, unused + :param stream: if True, the urllib3.HTTPResponse object will + be returned without reading/decoding response + data. Also when True, if the openapi spec describes a file download, + the data will be written to a local filesystme file and the BinarySchema + instance will also inherit from FileSchema and FileIO + Default is False. + :type stream: bool, optional + :param 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. + :param host: api endpoint host + :return: + If async_req parameter is True, + the request will be called asynchronously. + The method will return the request thread. + If parameter async_req is False or missing, + then the method will return the response directly. + """ + + if not async_req: + return self.__call_api( + resource_path, + method, + headers, + body, + fields, + auth_settings, + stream, + timeout, + host, + ) + + return self.pool.apply_async( + self.__call_api, + ( + resource_path, + method, + headers, + body, + json, + fields, + auth_settings, + stream, + timeout, + host, + ), + ) + + def request( + self, + method: str, + url: str, + headers: typing.Optional[HTTPHeaderDict] = None, + fields: typing.Optional[typing.Tuple[typing.Tuple[str, str], ...]] = None, + body: typing.Optional[typing.Union[str, bytes]] = None, + stream: bool = False, + timeout: typing.Optional[typing.Union[int, typing.Tuple]] = None, + ) -> urllib3.HTTPResponse: + """Makes the HTTP request using RESTClient.""" + if method == "GET": + return self.rest_client.GET(url, stream=stream, timeout=timeout, headers=headers) + elif method == "HEAD": + return self.rest_client.HEAD(url, stream=stream, timeout=timeout, headers=headers) + elif method == "OPTIONS": + return self.rest_client.OPTIONS( + url, + headers=headers, + fields=fields, + stream=stream, + timeout=timeout, + body=body, + ) + elif method == "POST": + return self.rest_client.POST( + url, + headers=headers, + fields=fields, + stream=stream, + timeout=timeout, + body=body, + ) + elif method == "PUT": + return self.rest_client.PUT( + url, + headers=headers, + fields=fields, + stream=stream, + timeout=timeout, + body=body, + ) + elif method == "PATCH": + return self.rest_client.PATCH( + url, + headers=headers, + fields=fields, + stream=stream, + timeout=timeout, + body=body, + ) + elif method == "DELETE": + return self.rest_client.DELETE(url, headers=headers, stream=stream, timeout=timeout, body=body) + else: + raise ApiValueError("http method must be `GET`, `HEAD`, `OPTIONS`," " `POST`, `PATCH`, `PUT` or `DELETE`.") + + def update_params_for_auth(self, headers, auth_settings, resource_path, method, body): + """Updates header and query params based on authentication setting. + + :param headers: Header parameters dict to be updated. + :param auth_settings: Authentication setting identifiers list. + :param resource_path: A string representation of the HTTP request resource path. + :param method: A string representation of the HTTP request method. + :param body: A object representing the body of the HTTP request. + The object type is the return value of _encoder.default(). + """ + if not auth_settings: + return + + for auth in auth_settings: + auth_setting = self.configuration.auth_settings().get(auth) + if not auth_setting: + continue + if auth_setting["in"] == "cookie": + headers.add("Cookie", auth_setting["value"]) + elif auth_setting["in"] == "header": + if auth_setting["type"] != "http-signature": + headers.add(auth_setting["key"], auth_setting["value"]) + elif auth_setting["in"] == "query": + """TODO implement auth in query + need to pass in prefix_separator_iterator + and need to output resource_path with query params added + """ + raise ApiValueError("Auth in query not yet implemented") + else: + raise ApiValueError("Authentication token must be in `query` or `header`") + + +class Api: + """NOTE: This class is auto generated by OpenAPI Generator + Ref: https://openapi-generator.tech + + Do not edit the class manually. + """ + + def __init__(self, api_client: typing.Optional[ApiClient] = None): + if api_client is None: + api_client = ApiClient() + self.api_client = api_client + + @staticmethod + def _verify_typed_dict_inputs_oapg( + cls: typing.Type[typing_extensions.TypedDict], + data: typing.Dict[str, typing.Any], + ): + """ + Ensures that: + - required keys are present + - additional properties are not input + - value stored under required keys do not have the value unset + Note: detailed value checking is done in schema classes + """ + missing_required_keys = [] + required_keys_with_unset_values = [] + for required_key in cls.__required_keys__: + if required_key not in data: + missing_required_keys.append(required_key) + continue + value = data[required_key] + if value is unset: + required_keys_with_unset_values.append(required_key) + if missing_required_keys: + raise ApiTypeError( + "{} missing {} required arguments: {}".format( + cls.__name__, + len(missing_required_keys), + missing_required_keys, + ) + ) + if required_keys_with_unset_values: + raise ApiValueError( + "{} contains invalid unset values for {} required keys: {}".format( + cls.__name__, + len(required_keys_with_unset_values), + required_keys_with_unset_values, + ) + ) + + disallowed_additional_keys = [] + for key in data: + if key in cls.__required_keys__ or key in cls.__optional_keys__: + continue + disallowed_additional_keys.append(key) + if disallowed_additional_keys: + raise ApiTypeError( + "{} got {} unexpected keyword arguments: {}".format( + cls.__name__, + len(disallowed_additional_keys), + disallowed_additional_keys, + ) + ) + + def _get_host_oapg( + self, + operation_id: str, + servers: typing.Tuple[typing.Dict[str, str], ...] = tuple(), + host_index: typing.Optional[int] = None, + ) -> typing.Optional[str]: + configuration = self.api_client.configuration + try: + if host_index is None: + index = configuration.server_operation_index.get(operation_id, configuration.server_index) + else: + index = host_index + server_variables = configuration.server_operation_variables.get( + operation_id, configuration.server_variables + ) + host = configuration.get_host_from_settings(index, variables=server_variables, servers=servers) + except IndexError: + if servers: + raise ApiValueError("Invalid host index. Must be 0 <= index < %s" % len(servers)) + host = None + return host + + +class SerializedRequestBody(typing_extensions.TypedDict, total=False): + body: typing.Union[str, bytes] + fields: typing.Tuple[typing.Union[RequestField, typing.Tuple[str, str]], ...] + + +class RequestBody(StyleFormSerializer, JSONDetector): + """ + A request body parameter + content: content_type to MediaType Schema info + """ + + __json_encoder = JSONEncoder() + + def __init__( + self, + content: typing.Dict[str, MediaType], + required: bool = False, + ): + self.required = required + if len(content) == 0: + raise ValueError("Invalid value for content, the content dict must have >= 1 entry") + self.content = content + + def __serialize_json(self, in_data: typing.Any) -> typing.Dict[str, bytes]: + in_data = self.__json_encoder.default(in_data) + json_str = json.dumps(in_data, separators=(",", ":"), ensure_ascii=False).encode("utf-8") + return dict(body=json_str) + + @staticmethod + def __serialize_text_plain(in_data: typing.Any) -> typing.Dict[str, str]: + if isinstance(in_data, frozendict.frozendict): + raise ValueError("Unable to serialize type frozendict.frozendict to text/plain") + elif isinstance(in_data, tuple): + raise ValueError("Unable to serialize type tuple to text/plain") + elif isinstance(in_data, NoneClass): + raise ValueError("Unable to serialize type NoneClass to text/plain") + elif isinstance(in_data, BoolClass): + raise ValueError("Unable to serialize type BoolClass to text/plain") + return dict(body=str(in_data)) + + def __multipart_json_item(self, key: str, value: Schema) -> RequestField: + json_value = self.__json_encoder.default(value) + return RequestField( + name=key, + data=json.dumps(json_value), + headers={"Content-Type": "application/json"}, + ) + + def __multipart_form_item(self, key: str, value: Schema) -> RequestField: + if isinstance(value, str): + return RequestField( + name=key, + data=str(value), + headers={"Content-Type": "text/plain"}, + ) + elif isinstance(value, bytes): + return RequestField( + name=key, + data=value, + headers={"Content-Type": "application/octet-stream"}, + ) + elif isinstance(value, FileIO): + request_field = RequestField( + name=key, + data=value.read(), + filename=os.path.basename(value.name), + headers={"Content-Type": "application/octet-stream"}, + ) + value.close() + return request_field + else: + return self.__multipart_json_item(key=key, value=value) + + def __serialize_multipart_form_data(self, in_data: Schema) -> typing.Dict[str, typing.Tuple[RequestField, ...]]: + if not isinstance(in_data, frozendict.frozendict): + raise ValueError(f"Unable to serialize {in_data} to multipart/form-data because it is not a dict of data") + """ + In a multipart/form-data request body, each schema property, or each element of a schema array property, + takes a section in the payload with an internal header as defined by RFC7578. The serialization strategy + for each property of a multipart/form-data request body can be specified in an associated Encoding Object. + + When passing in multipart types, boundaries MAY be used to separate sections of the content being + transferred – thus, the following default Content-Types are defined for multipart: + + If the (object) property is a primitive, or an array of primitive values, the default Content-Type is text/plain + If the property is complex, or an array of complex values, the default Content-Type is application/json + Question: how is the array of primitives encoded? + If the property is a type: string with a contentEncoding, the default Content-Type is application/octet-stream + """ + fields = [] + for key, value in in_data.items(): + if isinstance(value, tuple): + if value: + # values use explode = True, so the code makes a RequestField for each item with name=key + for item in value: + request_field = self.__multipart_form_item(key=key, value=item) + fields.append(request_field) + else: + # send an empty array as json because exploding will not send it + request_field = self.__multipart_json_item(key=key, value=value) + fields.append(request_field) + else: + request_field = self.__multipart_form_item(key=key, value=value) + fields.append(request_field) + + return dict(fields=tuple(fields)) + + def __serialize_application_octet_stream(self, in_data: BinarySchema) -> typing.Dict[str, bytes]: + if isinstance(in_data, bytes): + return dict(body=in_data) + # FileIO type + result = dict(body=in_data.read()) + in_data.close() + return result + + def __serialize_application_x_www_form_data(self, in_data: typing.Any) -> SerializedRequestBody: + """ + POST submission of form data in body + """ + if not isinstance(in_data, frozendict.frozendict): + raise ValueError( + f"Unable to serialize {in_data} to application/x-www-form-urlencoded because it is not a dict of data" + ) + cast_in_data = self.__json_encoder.default(in_data) + value = self._serialize_form(cast_in_data, name="", explode=True, percent_encode=True) + return dict(body=value) + + def serialize(self, in_data: typing.Any, content_type: str) -> SerializedRequestBody: + """ + If a str is returned then the result will be assigned to data when making the request + If a tuple is returned then the result will be used as fields input in encode_multipart_formdata + Return a tuple of + + The key of the return dict is + - body for application/json + - encode_multipart and fields for multipart/form-data + """ + media_type = self.content[content_type] + if isinstance(in_data, media_type.schema): + cast_in_data = in_data + elif isinstance(in_data, (dict, frozendict.frozendict)) and in_data: + cast_in_data = media_type.schema(**in_data) + else: + cast_in_data = media_type.schema(in_data) + # TODO check for and use encoding if it exists + # and content_type is multipart or application/x-www-form-urlencoded + if self._content_type_is_json(content_type): + return self.__serialize_json(cast_in_data) + elif content_type == "text/plain": + return self.__serialize_text_plain(cast_in_data) + elif content_type == "multipart/form-data": + return self.__serialize_multipart_form_data(cast_in_data) + elif content_type == "application/x-www-form-urlencoded": + return self.__serialize_application_x_www_form_data(cast_in_data) + elif content_type == "application/octet-stream": + return self.__serialize_application_octet_stream(cast_in_data) + raise NotImplementedError("Serialization has not yet been implemented for {}".format(content_type)) diff --git a/launch/api_client/apis/__init__.py b/launch/openapi_client/apis/__init__.py similarity index 100% rename from launch/api_client/apis/__init__.py rename to launch/openapi_client/apis/__init__.py diff --git a/launch/api_client/apis/path_to_api.py b/launch/openapi_client/apis/path_to_api.py similarity index 65% rename from launch/api_client/apis/path_to_api.py rename to launch/openapi_client/apis/path_to_api.py index 89fa65fe..25a69640 100644 --- a/launch/api_client/apis/path_to_api.py +++ b/launch/openapi_client/apis/path_to_api.py @@ -1,38 +1,40 @@ import typing_extensions -from launch.api_client.apis.paths.healthcheck import Healthcheck -from launch.api_client.apis.paths.healthz import Healthz -from launch.api_client.apis.paths.readyz import Readyz -from launch.api_client.apis.paths.v1_async_tasks import V1AsyncTasks -from launch.api_client.apis.paths.v1_async_tasks_task_id import ( +from launch.openapi_client.apis.paths.healthcheck import Healthcheck +from launch.openapi_client.apis.paths.healthz import Healthz +from launch.openapi_client.apis.paths.readyz import Readyz +from launch.openapi_client.apis.paths.v1_async_tasks import V1AsyncTasks +from launch.openapi_client.apis.paths.v1_async_tasks_task_id import ( V1AsyncTasksTaskId, ) -from launch.api_client.apis.paths.v1_batch_jobs import V1BatchJobs -from launch.api_client.apis.paths.v1_batch_jobs_batch_job_id import ( +from launch.openapi_client.apis.paths.v1_batch_jobs import V1BatchJobs +from launch.openapi_client.apis.paths.v1_batch_jobs_batch_job_id import ( V1BatchJobsBatchJobId, ) -from launch.api_client.apis.paths.v1_model_bundles import V1ModelBundles -from launch.api_client.apis.paths.v1_model_bundles_clone_with_changes import ( +from launch.openapi_client.apis.paths.v1_model_bundles import V1ModelBundles +from launch.openapi_client.apis.paths.v1_model_bundles_clone_with_changes import ( V1ModelBundlesCloneWithChanges, ) -from launch.api_client.apis.paths.v1_model_bundles_latest import ( +from launch.openapi_client.apis.paths.v1_model_bundles_latest import ( V1ModelBundlesLatest, ) -from launch.api_client.apis.paths.v1_model_bundles_model_bundle_id import ( +from launch.openapi_client.apis.paths.v1_model_bundles_model_bundle_id import ( V1ModelBundlesModelBundleId, ) -from launch.api_client.apis.paths.v1_model_endpoints import V1ModelEndpoints -from launch.api_client.apis.paths.v1_model_endpoints_api import ( +from launch.openapi_client.apis.paths.v1_model_endpoints import ( + V1ModelEndpoints, +) +from launch.openapi_client.apis.paths.v1_model_endpoints_api import ( V1ModelEndpointsApi, ) -from launch.api_client.apis.paths.v1_model_endpoints_model_endpoint_id import ( +from launch.openapi_client.apis.paths.v1_model_endpoints_model_endpoint_id import ( V1ModelEndpointsModelEndpointId, ) -from launch.api_client.apis.paths.v1_model_endpoints_schema_json import ( +from launch.openapi_client.apis.paths.v1_model_endpoints_schema_json import ( V1ModelEndpointsSchemaJson, ) -from launch.api_client.apis.paths.v1_sync_tasks import V1SyncTasks -from launch.api_client.paths import PathValues +from launch.openapi_client.apis.paths.v1_sync_tasks import V1SyncTasks +from launch.openapi_client.paths import PathValues PathToApi = typing_extensions.TypedDict( "PathToApi", diff --git a/launch/api_client/apis/paths/__init__.py b/launch/openapi_client/apis/paths/__init__.py similarity index 73% rename from launch/api_client/apis/paths/__init__.py rename to launch/openapi_client/apis/paths/__init__.py index d856e7a6..14b48c6b 100644 --- a/launch/api_client/apis/paths/__init__.py +++ b/launch/openapi_client/apis/paths/__init__.py @@ -1,3 +1,3 @@ # do not import all endpoints into this module because that uses a lot of memory and stack frames # if you need the ability to import all endpoints from this module, import them with -# from launch.api_client.apis.path_to_api import path_to_api +# from launch.openapi_client.apis.path_to_api import path_to_api diff --git a/launch/openapi_client/apis/paths/healthcheck.py b/launch/openapi_client/apis/paths/healthcheck.py new file mode 100644 index 00000000..0f230792 --- /dev/null +++ b/launch/openapi_client/apis/paths/healthcheck.py @@ -0,0 +1,7 @@ +from launch.openapi_client.paths.healthcheck.get import ApiForget + + +class Healthcheck( + ApiForget, +): + pass diff --git a/launch/openapi_client/apis/paths/healthz.py b/launch/openapi_client/apis/paths/healthz.py new file mode 100644 index 00000000..fdc9e14b --- /dev/null +++ b/launch/openapi_client/apis/paths/healthz.py @@ -0,0 +1,7 @@ +from launch.openapi_client.paths.healthz.get import ApiForget + + +class Healthz( + ApiForget, +): + pass diff --git a/launch/openapi_client/apis/paths/readyz.py b/launch/openapi_client/apis/paths/readyz.py new file mode 100644 index 00000000..01bac0ee --- /dev/null +++ b/launch/openapi_client/apis/paths/readyz.py @@ -0,0 +1,7 @@ +from launch.openapi_client.paths.readyz.get import ApiForget + + +class Readyz( + ApiForget, +): + pass diff --git a/launch/openapi_client/apis/paths/v1_async_tasks.py b/launch/openapi_client/apis/paths/v1_async_tasks.py new file mode 100644 index 00000000..b24d7301 --- /dev/null +++ b/launch/openapi_client/apis/paths/v1_async_tasks.py @@ -0,0 +1,7 @@ +from launch.openapi_client.paths.v1_async_tasks.post import ApiForpost + + +class V1AsyncTasks( + ApiForpost, +): + pass diff --git a/launch/openapi_client/apis/paths/v1_async_tasks_task_id.py b/launch/openapi_client/apis/paths/v1_async_tasks_task_id.py new file mode 100644 index 00000000..c9f731d4 --- /dev/null +++ b/launch/openapi_client/apis/paths/v1_async_tasks_task_id.py @@ -0,0 +1,7 @@ +from launch.openapi_client.paths.v1_async_tasks_task_id.get import ApiForget + + +class V1AsyncTasksTaskId( + ApiForget, +): + pass diff --git a/launch/openapi_client/apis/paths/v1_batch_jobs.py b/launch/openapi_client/apis/paths/v1_batch_jobs.py new file mode 100644 index 00000000..85128b11 --- /dev/null +++ b/launch/openapi_client/apis/paths/v1_batch_jobs.py @@ -0,0 +1,7 @@ +from launch.openapi_client.paths.v1_batch_jobs.post import ApiForpost + + +class V1BatchJobs( + ApiForpost, +): + pass diff --git a/launch/openapi_client/apis/paths/v1_batch_jobs_batch_job_id.py b/launch/openapi_client/apis/paths/v1_batch_jobs_batch_job_id.py new file mode 100644 index 00000000..9f868c85 --- /dev/null +++ b/launch/openapi_client/apis/paths/v1_batch_jobs_batch_job_id.py @@ -0,0 +1,13 @@ +from launch.openapi_client.paths.v1_batch_jobs_batch_job_id.get import ( + ApiForget, +) +from launch.openapi_client.paths.v1_batch_jobs_batch_job_id.put import ( + ApiForput, +) + + +class V1BatchJobsBatchJobId( + ApiForget, + ApiForput, +): + pass diff --git a/launch/openapi_client/apis/paths/v1_model_bundles.py b/launch/openapi_client/apis/paths/v1_model_bundles.py new file mode 100644 index 00000000..9c951e27 --- /dev/null +++ b/launch/openapi_client/apis/paths/v1_model_bundles.py @@ -0,0 +1,9 @@ +from launch.openapi_client.paths.v1_model_bundles.get import ApiForget +from launch.openapi_client.paths.v1_model_bundles.post import ApiForpost + + +class V1ModelBundles( + ApiForget, + ApiForpost, +): + pass diff --git a/launch/api_client/apis/paths/v1_model_bundles_clone_with_changes.py b/launch/openapi_client/apis/paths/v1_model_bundles_clone_with_changes.py similarity index 50% rename from launch/api_client/apis/paths/v1_model_bundles_clone_with_changes.py rename to launch/openapi_client/apis/paths/v1_model_bundles_clone_with_changes.py index df0644e3..4e895d48 100644 --- a/launch/api_client/apis/paths/v1_model_bundles_clone_with_changes.py +++ b/launch/openapi_client/apis/paths/v1_model_bundles_clone_with_changes.py @@ -1,4 +1,4 @@ -from launch.api_client.paths.v1_model_bundles_clone_with_changes.post import ( +from launch.openapi_client.paths.v1_model_bundles_clone_with_changes.post import ( ApiForpost, ) diff --git a/launch/openapi_client/apis/paths/v1_model_bundles_latest.py b/launch/openapi_client/apis/paths/v1_model_bundles_latest.py new file mode 100644 index 00000000..a04c91a8 --- /dev/null +++ b/launch/openapi_client/apis/paths/v1_model_bundles_latest.py @@ -0,0 +1,7 @@ +from launch.openapi_client.paths.v1_model_bundles_latest.get import ApiForget + + +class V1ModelBundlesLatest( + ApiForget, +): + pass diff --git a/launch/api_client/apis/paths/v1_model_bundles_model_bundle_id.py b/launch/openapi_client/apis/paths/v1_model_bundles_model_bundle_id.py similarity index 50% rename from launch/api_client/apis/paths/v1_model_bundles_model_bundle_id.py rename to launch/openapi_client/apis/paths/v1_model_bundles_model_bundle_id.py index 0e646257..a525a59d 100644 --- a/launch/api_client/apis/paths/v1_model_bundles_model_bundle_id.py +++ b/launch/openapi_client/apis/paths/v1_model_bundles_model_bundle_id.py @@ -1,4 +1,4 @@ -from launch.api_client.paths.v1_model_bundles_model_bundle_id.get import ( +from launch.openapi_client.paths.v1_model_bundles_model_bundle_id.get import ( ApiForget, ) diff --git a/launch/openapi_client/apis/paths/v1_model_endpoints.py b/launch/openapi_client/apis/paths/v1_model_endpoints.py new file mode 100644 index 00000000..44f88bf6 --- /dev/null +++ b/launch/openapi_client/apis/paths/v1_model_endpoints.py @@ -0,0 +1,9 @@ +from launch.openapi_client.paths.v1_model_endpoints.get import ApiForget +from launch.openapi_client.paths.v1_model_endpoints.post import ApiForpost + + +class V1ModelEndpoints( + ApiForget, + ApiForpost, +): + pass diff --git a/launch/openapi_client/apis/paths/v1_model_endpoints_api.py b/launch/openapi_client/apis/paths/v1_model_endpoints_api.py new file mode 100644 index 00000000..d4f3d056 --- /dev/null +++ b/launch/openapi_client/apis/paths/v1_model_endpoints_api.py @@ -0,0 +1,7 @@ +from launch.openapi_client.paths.v1_model_endpoints_api.get import ApiForget + + +class V1ModelEndpointsApi( + ApiForget, +): + pass diff --git a/launch/openapi_client/apis/paths/v1_model_endpoints_model_endpoint_id.py b/launch/openapi_client/apis/paths/v1_model_endpoints_model_endpoint_id.py new file mode 100644 index 00000000..66f72f0a --- /dev/null +++ b/launch/openapi_client/apis/paths/v1_model_endpoints_model_endpoint_id.py @@ -0,0 +1,17 @@ +from launch.openapi_client.paths.v1_model_endpoints_model_endpoint_id.delete import ( + ApiFordelete, +) +from launch.openapi_client.paths.v1_model_endpoints_model_endpoint_id.get import ( + ApiForget, +) +from launch.openapi_client.paths.v1_model_endpoints_model_endpoint_id.put import ( + ApiForput, +) + + +class V1ModelEndpointsModelEndpointId( + ApiForget, + ApiForput, + ApiFordelete, +): + pass diff --git a/launch/api_client/apis/paths/v1_model_endpoints_schema_json.py b/launch/openapi_client/apis/paths/v1_model_endpoints_schema_json.py similarity index 50% rename from launch/api_client/apis/paths/v1_model_endpoints_schema_json.py rename to launch/openapi_client/apis/paths/v1_model_endpoints_schema_json.py index 7c62f95b..d0db7c76 100644 --- a/launch/api_client/apis/paths/v1_model_endpoints_schema_json.py +++ b/launch/openapi_client/apis/paths/v1_model_endpoints_schema_json.py @@ -1,4 +1,4 @@ -from launch.api_client.paths.v1_model_endpoints_schema_json.get import ( +from launch.openapi_client.paths.v1_model_endpoints_schema_json.get import ( ApiForget, ) diff --git a/launch/openapi_client/apis/paths/v1_sync_tasks.py b/launch/openapi_client/apis/paths/v1_sync_tasks.py new file mode 100644 index 00000000..c2d68d32 --- /dev/null +++ b/launch/openapi_client/apis/paths/v1_sync_tasks.py @@ -0,0 +1,7 @@ +from launch.openapi_client.paths.v1_sync_tasks.post import ApiForpost + + +class V1SyncTasks( + ApiForpost, +): + pass diff --git a/launch/api_client/apis/tag_to_api.py b/launch/openapi_client/apis/tag_to_api.py similarity index 63% rename from launch/api_client/apis/tag_to_api.py rename to launch/openapi_client/apis/tag_to_api.py index 8b3c0d6d..944183ec 100644 --- a/launch/api_client/apis/tag_to_api.py +++ b/launch/openapi_client/apis/tag_to_api.py @@ -1,7 +1,7 @@ import typing_extensions -from launch.api_client.apis.tags import TagValues -from launch.api_client.apis.tags.default_api import DefaultApi +from launch.openapi_client.apis.tags import TagValues +from launch.openapi_client.apis.tags.default_api import DefaultApi TagToApi = typing_extensions.TypedDict( "TagToApi", diff --git a/launch/api_client/apis/tags/__init__.py b/launch/openapi_client/apis/tags/__init__.py similarity index 80% rename from launch/api_client/apis/tags/__init__.py rename to launch/openapi_client/apis/tags/__init__.py index 5595c045..0fd096c1 100644 --- a/launch/api_client/apis/tags/__init__.py +++ b/launch/openapi_client/apis/tags/__init__.py @@ -1,6 +1,6 @@ # do not import all endpoints into this module because that uses a lot of memory and stack frames # if you need the ability to import all endpoints from this module, import them with -# from launch.api_client.apis.tag_to_api import tag_to_api +# from launch.openapi_client.apis.tag_to_api import tag_to_api import enum diff --git a/launch/api_client/apis/tags/default_api.py b/launch/openapi_client/apis/tags/default_api.py similarity index 61% rename from launch/api_client/apis/tags/default_api.py rename to launch/openapi_client/apis/tags/default_api.py index 6a2bb4aa..9c63633a 100644 --- a/launch/api_client/apis/tags/default_api.py +++ b/launch/openapi_client/apis/tags/default_api.py @@ -9,61 +9,63 @@ Generated by: https://openapi-generator.tech """ -from launch.api_client.paths.healthcheck.get import HealthcheckHealthcheckGet -from launch.api_client.paths.healthz.get import HealthcheckHealthzGet -from launch.api_client.paths.readyz.get import HealthcheckReadyzGet -from launch.api_client.paths.v1_async_tasks.post import ( +from launch.openapi_client.paths.healthcheck.get import ( + HealthcheckHealthcheckGet, +) +from launch.openapi_client.paths.healthz.get import HealthcheckHealthzGet +from launch.openapi_client.paths.readyz.get import HealthcheckReadyzGet +from launch.openapi_client.paths.v1_async_tasks.post import ( CreateAsyncInferenceTaskV1AsyncTasksPost, ) -from launch.api_client.paths.v1_async_tasks_task_id.get import ( +from launch.openapi_client.paths.v1_async_tasks_task_id.get import ( GetAsyncInferenceTaskV1AsyncTasksTaskIdGet, ) -from launch.api_client.paths.v1_batch_jobs.post import ( +from launch.openapi_client.paths.v1_batch_jobs.post import ( CreateBatchJobV1BatchJobsPost, ) -from launch.api_client.paths.v1_batch_jobs_batch_job_id.get import ( +from launch.openapi_client.paths.v1_batch_jobs_batch_job_id.get import ( GetBatchJobV1BatchJobsBatchJobIdGet, ) -from launch.api_client.paths.v1_batch_jobs_batch_job_id.put import ( +from launch.openapi_client.paths.v1_batch_jobs_batch_job_id.put import ( UpdateBatchJobV1BatchJobsBatchJobIdPut, ) -from launch.api_client.paths.v1_model_bundles.get import ( +from launch.openapi_client.paths.v1_model_bundles.get import ( ListModelBundlesV1ModelBundlesGet, ) -from launch.api_client.paths.v1_model_bundles.post import ( +from launch.openapi_client.paths.v1_model_bundles.post import ( CreateModelBundleV1ModelBundlesPost, ) -from launch.api_client.paths.v1_model_bundles_clone_with_changes.post import ( +from launch.openapi_client.paths.v1_model_bundles_clone_with_changes.post import ( CloneModelBundleWithChangesV1ModelBundlesCloneWithChangesPost, ) -from launch.api_client.paths.v1_model_bundles_latest.get import ( +from launch.openapi_client.paths.v1_model_bundles_latest.get import ( GetLatestModelBundleV1ModelBundlesLatestGet, ) -from launch.api_client.paths.v1_model_bundles_model_bundle_id.get import ( +from launch.openapi_client.paths.v1_model_bundles_model_bundle_id.get import ( GetModelBundleV1ModelBundlesModelBundleIdGet, ) -from launch.api_client.paths.v1_model_endpoints.get import ( +from launch.openapi_client.paths.v1_model_endpoints.get import ( ListModelEndpointsV1ModelEndpointsGet, ) -from launch.api_client.paths.v1_model_endpoints.post import ( +from launch.openapi_client.paths.v1_model_endpoints.post import ( CreateModelEndpointV1ModelEndpointsPost, ) -from launch.api_client.paths.v1_model_endpoints_api.get import ( +from launch.openapi_client.paths.v1_model_endpoints_api.get import ( GetModelEndpointsApiV1ModelEndpointsApiGet, ) -from launch.api_client.paths.v1_model_endpoints_model_endpoint_id.delete import ( +from launch.openapi_client.paths.v1_model_endpoints_model_endpoint_id.delete import ( DeleteModelEndpointV1ModelEndpointsModelEndpointIdDelete, ) -from launch.api_client.paths.v1_model_endpoints_model_endpoint_id.get import ( +from launch.openapi_client.paths.v1_model_endpoints_model_endpoint_id.get import ( GetModelEndpointV1ModelEndpointsModelEndpointIdGet, ) -from launch.api_client.paths.v1_model_endpoints_model_endpoint_id.put import ( +from launch.openapi_client.paths.v1_model_endpoints_model_endpoint_id.put import ( UpdateModelEndpointV1ModelEndpointsModelEndpointIdPut, ) -from launch.api_client.paths.v1_model_endpoints_schema_json.get import ( +from launch.openapi_client.paths.v1_model_endpoints_schema_json.get import ( GetModelEndpointsSchemaV1ModelEndpointsSchemaJsonGet, ) -from launch.api_client.paths.v1_sync_tasks.post import ( +from launch.openapi_client.paths.v1_sync_tasks.post import ( CreateSyncInferenceTaskV1SyncTasksPost, ) diff --git a/launch/api_client/configuration.py b/launch/openapi_client/configuration.py similarity index 99% rename from launch/api_client/configuration.py rename to launch/openapi_client/configuration.py index 7c17d0c3..b8536513 100644 --- a/launch/api_client/configuration.py +++ b/launch/openapi_client/configuration.py @@ -17,7 +17,7 @@ import urllib3 -from launch.api_client.exceptions import ApiValueError +from launch.openapi_client.exceptions import ApiValueError JSON_SCHEMA_VALIDATION_KEYWORDS = { "multipleOf", @@ -98,7 +98,7 @@ class Configuration(object): Configure API client with HTTP basic authentication: - conf = launch.api_client.Configuration( + conf = launch.openapi_client.Configuration( username='the-user', password='the-password', ) @@ -161,7 +161,7 @@ def __init__( self.logger = {} """Logging Settings """ - self.logger["package_logger"] = logging.getLogger("launch.api_client") + self.logger["package_logger"] = logging.getLogger("launch.openapi_client") self.logger["urllib3_logger"] = logging.getLogger("urllib3") self.logger_format = "%(asctime)s %(levelname)s %(message)s" """Log format diff --git a/launch/openapi_client/exceptions.py b/launch/openapi_client/exceptions.py new file mode 100644 index 00000000..2751e056 --- /dev/null +++ b/launch/openapi_client/exceptions.py @@ -0,0 +1,137 @@ +# coding: utf-8 + +""" + launch + + No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator) # noqa: E501 + + The version of the OpenAPI document: 1.0.0 + Generated by: https://openapi-generator.tech +""" + + +class OpenApiException(Exception): + """The base exception class for all OpenAPIExceptions""" + + +class ApiTypeError(OpenApiException, TypeError): + def __init__(self, msg, path_to_item=None, valid_classes=None, key_type=None): + """Raises an exception for TypeErrors + + Args: + msg (str): the exception message + + Keyword Args: + path_to_item (list): a list of keys an indices to get to the + current_item + None if unset + valid_classes (tuple): the primitive classes that current item + should be an instance of + None if unset + key_type (bool): False if our value is a value in a dict + True if it is a key in a dict + False if our item is an item in a list + None if unset + """ + self.path_to_item = path_to_item + self.valid_classes = valid_classes + self.key_type = key_type + full_msg = msg + if path_to_item: + full_msg = "{0} at {1}".format(msg, render_path(path_to_item)) + super(ApiTypeError, self).__init__(full_msg) + + +class ApiValueError(OpenApiException, ValueError): + def __init__(self, msg, path_to_item=None): + """ + Args: + msg (str): the exception message + + Keyword Args: + path_to_item (list) the path to the exception in the + received_data dict. None if unset + """ + + self.path_to_item = path_to_item + full_msg = msg + if path_to_item: + full_msg = "{0} at {1}".format(msg, render_path(path_to_item)) + super(ApiValueError, self).__init__(full_msg) + + +class ApiAttributeError(OpenApiException, AttributeError): + def __init__(self, msg, path_to_item=None): + """ + Raised when an attribute reference or assignment fails. + + Args: + msg (str): the exception message + + Keyword Args: + path_to_item (None/list) the path to the exception in the + received_data dict + """ + self.path_to_item = path_to_item + full_msg = msg + if path_to_item: + full_msg = "{0} at {1}".format(msg, render_path(path_to_item)) + super(ApiAttributeError, self).__init__(full_msg) + + +class ApiKeyError(OpenApiException, KeyError): + def __init__(self, msg, path_to_item=None): + """ + Args: + msg (str): the exception message + + Keyword Args: + path_to_item (None/list) the path to the exception in the + received_data dict + """ + self.path_to_item = path_to_item + full_msg = msg + if path_to_item: + full_msg = "{0} at {1}".format(msg, render_path(path_to_item)) + super(ApiKeyError, self).__init__(full_msg) + + +class ApiException(OpenApiException): + def __init__( + self, + status=None, + reason=None, + api_response: "launch.openapi_client.api_client.ApiResponse" = None, + ): + if api_response: + self.status = api_response.response.status + self.reason = api_response.response.reason + self.body = api_response.response.data + self.headers = api_response.response.getheaders() + else: + self.status = status + self.reason = reason + self.body = None + self.headers = None + + def __str__(self): + """Custom error messages for exception""" + error_message = "({0})\n" "Reason: {1}\n".format(self.status, self.reason) + if self.headers: + error_message += "HTTP response headers: {0}\n".format(self.headers) + + if self.body: + error_message += "HTTP response body: {0}\n".format(self.body) + + return error_message + + +def render_path(path_to_item): + """Returns a string representation of a path""" + result = "" + for pth in path_to_item: + if isinstance(pth, int): + result += "[{0}]".format(pth) + else: + result += "['{0}']".format(pth) + return result diff --git a/launch/api_client/model/__init__.py b/launch/openapi_client/model/__init__.py similarity index 83% rename from launch/api_client/model/__init__.py rename to launch/openapi_client/model/__init__.py index 4a66ef07..9439f5e1 100644 --- a/launch/api_client/model/__init__.py +++ b/launch/openapi_client/model/__init__.py @@ -2,4 +2,4 @@ # reference which would not work in python2 # do not import all models into this module because that uses a lot of memory and stack frames # if you need the ability to import all models from one package, import them with -# from launch.api_client.models import ModelA, ModelB +# from launch.openapi_client.models import ModelA, ModelB diff --git a/launch/api_client/model/batch_job_serialization_format.py b/launch/openapi_client/model/batch_job_serialization_format.py similarity index 94% rename from launch/api_client/model/batch_job_serialization_format.py rename to launch/openapi_client/model/batch_job_serialization_format.py index 7bfc3103..fbc04cd6 100644 --- a/launch/api_client/model/batch_job_serialization_format.py +++ b/launch/openapi_client/model/batch_job_serialization_format.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class BatchJobSerializationFormat(schemas.EnumBase, schemas.StrSchema): diff --git a/launch/api_client/model/batch_job_serialization_format.pyi b/launch/openapi_client/model/batch_job_serialization_format.pyi similarity index 100% rename from launch/api_client/model/batch_job_serialization_format.pyi rename to launch/openapi_client/model/batch_job_serialization_format.pyi diff --git a/launch/api_client/model/batch_job_status.py b/launch/openapi_client/model/batch_job_status.py similarity index 96% rename from launch/api_client/model/batch_job_status.py rename to launch/openapi_client/model/batch_job_status.py index 052dada8..aa2c3796 100644 --- a/launch/api_client/model/batch_job_status.py +++ b/launch/openapi_client/model/batch_job_status.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class BatchJobStatus(schemas.EnumBase, schemas.StrSchema): diff --git a/launch/api_client/model/batch_job_status.pyi b/launch/openapi_client/model/batch_job_status.pyi similarity index 100% rename from launch/api_client/model/batch_job_status.pyi rename to launch/openapi_client/model/batch_job_status.pyi diff --git a/launch/api_client/model/clone_model_bundle_request.py b/launch/openapi_client/model/clone_model_bundle_request.py similarity index 98% rename from launch/api_client/model/clone_model_bundle_request.py rename to launch/openapi_client/model/clone_model_bundle_request.py index 75df3b51..deafa0ab 100644 --- a/launch/api_client/model/clone_model_bundle_request.py +++ b/launch/openapi_client/model/clone_model_bundle_request.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class CloneModelBundleRequest(schemas.DictSchema): diff --git a/launch/api_client/model/clone_model_bundle_request.pyi b/launch/openapi_client/model/clone_model_bundle_request.pyi similarity index 100% rename from launch/api_client/model/clone_model_bundle_request.pyi rename to launch/openapi_client/model/clone_model_bundle_request.pyi diff --git a/launch/api_client/model/create_async_task_response.py b/launch/openapi_client/model/create_async_task_response.py similarity index 98% rename from launch/api_client/model/create_async_task_response.py rename to launch/openapi_client/model/create_async_task_response.py index 54f85f97..0911cf8d 100644 --- a/launch/api_client/model/create_async_task_response.py +++ b/launch/openapi_client/model/create_async_task_response.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class CreateAsyncTaskResponse(schemas.DictSchema): diff --git a/launch/api_client/model/create_async_task_response.pyi b/launch/openapi_client/model/create_async_task_response.pyi similarity index 100% rename from launch/api_client/model/create_async_task_response.pyi rename to launch/openapi_client/model/create_async_task_response.pyi diff --git a/launch/api_client/model/create_batch_job_request.py b/launch/openapi_client/model/create_batch_job_request.py similarity index 97% rename from launch/api_client/model/create_batch_job_request.py rename to launch/openapi_client/model/create_batch_job_request.py index c90c8903..3b405905 100644 --- a/launch/api_client/model/create_batch_job_request.py +++ b/launch/openapi_client/model/create_batch_job_request.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class CreateBatchJobRequest(schemas.DictSchema): @@ -233,9 +233,9 @@ def __new__( ) -from launch.api_client.model.batch_job_serialization_format import ( +from launch.openapi_client.model.batch_job_serialization_format import ( BatchJobSerializationFormat, ) -from launch.api_client.model.create_batch_job_resource_requests import ( +from launch.openapi_client.model.create_batch_job_resource_requests import ( CreateBatchJobResourceRequests, ) diff --git a/launch/api_client/model/create_batch_job_request.pyi b/launch/openapi_client/model/create_batch_job_request.pyi similarity index 100% rename from launch/api_client/model/create_batch_job_request.pyi rename to launch/openapi_client/model/create_batch_job_request.pyi diff --git a/launch/api_client/model/create_batch_job_resource_requests.py b/launch/openapi_client/model/create_batch_job_resource_requests.py similarity index 99% rename from launch/api_client/model/create_batch_job_resource_requests.py rename to launch/openapi_client/model/create_batch_job_resource_requests.py index ed223c61..e2b08c0f 100644 --- a/launch/api_client/model/create_batch_job_resource_requests.py +++ b/launch/openapi_client/model/create_batch_job_resource_requests.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class CreateBatchJobResourceRequests(schemas.DictSchema): @@ -477,4 +477,4 @@ def __new__( ) -from launch.api_client.model.gpu_type import GpuType +from launch.openapi_client.model.gpu_type import GpuType diff --git a/launch/api_client/model/create_batch_job_resource_requests.pyi b/launch/openapi_client/model/create_batch_job_resource_requests.pyi similarity index 100% rename from launch/api_client/model/create_batch_job_resource_requests.pyi rename to launch/openapi_client/model/create_batch_job_resource_requests.pyi diff --git a/launch/api_client/model/create_batch_job_response.py b/launch/openapi_client/model/create_batch_job_response.py similarity index 98% rename from launch/api_client/model/create_batch_job_response.py rename to launch/openapi_client/model/create_batch_job_response.py index c160b679..d4c7475f 100644 --- a/launch/api_client/model/create_batch_job_response.py +++ b/launch/openapi_client/model/create_batch_job_response.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class CreateBatchJobResponse(schemas.DictSchema): diff --git a/launch/api_client/model/create_batch_job_response.pyi b/launch/openapi_client/model/create_batch_job_response.pyi similarity index 100% rename from launch/api_client/model/create_batch_job_response.pyi rename to launch/openapi_client/model/create_batch_job_response.pyi diff --git a/launch/api_client/model/create_model_bundle_request.py b/launch/openapi_client/model/create_model_bundle_request.py similarity index 97% rename from launch/api_client/model/create_model_bundle_request.py rename to launch/openapi_client/model/create_model_bundle_request.py index 255a190b..db57caa1 100644 --- a/launch/api_client/model/create_model_bundle_request.py +++ b/launch/openapi_client/model/create_model_bundle_request.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class CreateModelBundleRequest(schemas.DictSchema): @@ -288,9 +288,9 @@ def __new__( ) -from launch.api_client.model.model_bundle_environment_params import ( +from launch.openapi_client.model.model_bundle_environment_params import ( ModelBundleEnvironmentParams, ) -from launch.api_client.model.model_bundle_packaging_type import ( +from launch.openapi_client.model.model_bundle_packaging_type import ( ModelBundlePackagingType, ) diff --git a/launch/api_client/model/create_model_bundle_request.pyi b/launch/openapi_client/model/create_model_bundle_request.pyi similarity index 100% rename from launch/api_client/model/create_model_bundle_request.pyi rename to launch/openapi_client/model/create_model_bundle_request.pyi diff --git a/launch/api_client/model/create_model_bundle_response.py b/launch/openapi_client/model/create_model_bundle_response.py similarity index 98% rename from launch/api_client/model/create_model_bundle_response.py rename to launch/openapi_client/model/create_model_bundle_response.py index 23a7a0c0..4bb60b8e 100644 --- a/launch/api_client/model/create_model_bundle_response.py +++ b/launch/openapi_client/model/create_model_bundle_response.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class CreateModelBundleResponse(schemas.DictSchema): diff --git a/launch/api_client/model/create_model_bundle_response.pyi b/launch/openapi_client/model/create_model_bundle_response.pyi similarity index 100% rename from launch/api_client/model/create_model_bundle_response.pyi rename to launch/openapi_client/model/create_model_bundle_response.pyi diff --git a/launch/api_client/model/create_model_endpoint_request.py b/launch/openapi_client/model/create_model_endpoint_request.py similarity index 99% rename from launch/api_client/model/create_model_endpoint_request.py rename to launch/openapi_client/model/create_model_endpoint_request.py index 570250b6..675b8038 100644 --- a/launch/api_client/model/create_model_endpoint_request.py +++ b/launch/openapi_client/model/create_model_endpoint_request.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class CreateModelEndpointRequest(schemas.DictSchema): @@ -780,5 +780,5 @@ def __new__( ) -from launch.api_client.model.gpu_type import GpuType -from launch.api_client.model.model_endpoint_type import ModelEndpointType +from launch.openapi_client.model.gpu_type import GpuType +from launch.openapi_client.model.model_endpoint_type import ModelEndpointType diff --git a/launch/api_client/model/create_model_endpoint_request.pyi b/launch/openapi_client/model/create_model_endpoint_request.pyi similarity index 100% rename from launch/api_client/model/create_model_endpoint_request.pyi rename to launch/openapi_client/model/create_model_endpoint_request.pyi diff --git a/launch/api_client/model/create_model_endpoint_response.py b/launch/openapi_client/model/create_model_endpoint_response.py similarity index 98% rename from launch/api_client/model/create_model_endpoint_response.py rename to launch/openapi_client/model/create_model_endpoint_response.py index d198be40..4ac8ac49 100644 --- a/launch/api_client/model/create_model_endpoint_response.py +++ b/launch/openapi_client/model/create_model_endpoint_response.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class CreateModelEndpointResponse(schemas.DictSchema): diff --git a/launch/api_client/model/create_model_endpoint_response.pyi b/launch/openapi_client/model/create_model_endpoint_response.pyi similarity index 100% rename from launch/api_client/model/create_model_endpoint_response.pyi rename to launch/openapi_client/model/create_model_endpoint_response.pyi diff --git a/launch/api_client/model/delete_model_endpoint_response.py b/launch/openapi_client/model/delete_model_endpoint_response.py similarity index 98% rename from launch/api_client/model/delete_model_endpoint_response.py rename to launch/openapi_client/model/delete_model_endpoint_response.py index 01024336..77bf0dc5 100644 --- a/launch/api_client/model/delete_model_endpoint_response.py +++ b/launch/openapi_client/model/delete_model_endpoint_response.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class DeleteModelEndpointResponse(schemas.DictSchema): diff --git a/launch/api_client/model/delete_model_endpoint_response.pyi b/launch/openapi_client/model/delete_model_endpoint_response.pyi similarity index 100% rename from launch/api_client/model/delete_model_endpoint_response.pyi rename to launch/openapi_client/model/delete_model_endpoint_response.pyi diff --git a/launch/api_client/model/endpoint_predict_request.py b/launch/openapi_client/model/endpoint_predict_request.py similarity index 99% rename from launch/api_client/model/endpoint_predict_request.py rename to launch/openapi_client/model/endpoint_predict_request.py index 9814f08b..c5efff9d 100644 --- a/launch/api_client/model/endpoint_predict_request.py +++ b/launch/openapi_client/model/endpoint_predict_request.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class EndpointPredictRequest(schemas.DictSchema): diff --git a/launch/api_client/model/endpoint_predict_request.pyi b/launch/openapi_client/model/endpoint_predict_request.pyi similarity index 100% rename from launch/api_client/model/endpoint_predict_request.pyi rename to launch/openapi_client/model/endpoint_predict_request.pyi diff --git a/launch/api_client/model/get_async_task_response.py b/launch/openapi_client/model/get_async_task_response.py similarity index 97% rename from launch/api_client/model/get_async_task_response.py rename to launch/openapi_client/model/get_async_task_response.py index 02dc4e20..20df78e8 100644 --- a/launch/api_client/model/get_async_task_response.py +++ b/launch/openapi_client/model/get_async_task_response.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class GetAsyncTaskResponse(schemas.DictSchema): @@ -189,4 +189,4 @@ def __new__( ) -from launch.api_client.model.task_status import TaskStatus +from launch.openapi_client.model.task_status import TaskStatus diff --git a/launch/api_client/model/get_async_task_response.pyi b/launch/openapi_client/model/get_async_task_response.pyi similarity index 100% rename from launch/api_client/model/get_async_task_response.pyi rename to launch/openapi_client/model/get_async_task_response.pyi diff --git a/launch/api_client/model/get_batch_job_response.py b/launch/openapi_client/model/get_batch_job_response.py similarity index 97% rename from launch/api_client/model/get_batch_job_response.py rename to launch/openapi_client/model/get_batch_job_response.py index 9b0b0894..279a3d32 100644 --- a/launch/api_client/model/get_batch_job_response.py +++ b/launch/openapi_client/model/get_batch_job_response.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class GetBatchJobResponse(schemas.DictSchema): @@ -203,4 +203,4 @@ def __new__( ) -from launch.api_client.model.batch_job_status import BatchJobStatus +from launch.openapi_client.model.batch_job_status import BatchJobStatus diff --git a/launch/api_client/model/get_batch_job_response.pyi b/launch/openapi_client/model/get_batch_job_response.pyi similarity index 100% rename from launch/api_client/model/get_batch_job_response.pyi rename to launch/openapi_client/model/get_batch_job_response.pyi diff --git a/launch/api_client/model/get_model_endpoint_response.py b/launch/openapi_client/model/get_model_endpoint_response.py similarity index 97% rename from launch/api_client/model/get_model_endpoint_response.py rename to launch/openapi_client/model/get_model_endpoint_response.py index ea40d0d9..4e57ca51 100644 --- a/launch/api_client/model/get_model_endpoint_response.py +++ b/launch/openapi_client/model/get_model_endpoint_response.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class GetModelEndpointResponse(schemas.DictSchema): @@ -521,11 +521,13 @@ def __new__( ) -from launch.api_client.model.model_endpoint_deployment_state import ( +from launch.openapi_client.model.model_endpoint_deployment_state import ( ModelEndpointDeploymentState, ) -from launch.api_client.model.model_endpoint_resource_state import ( +from launch.openapi_client.model.model_endpoint_resource_state import ( ModelEndpointResourceState, ) -from launch.api_client.model.model_endpoint_status import ModelEndpointStatus -from launch.api_client.model.model_endpoint_type import ModelEndpointType +from launch.openapi_client.model.model_endpoint_status import ( + ModelEndpointStatus, +) +from launch.openapi_client.model.model_endpoint_type import ModelEndpointType diff --git a/launch/api_client/model/get_model_endpoint_response.pyi b/launch/openapi_client/model/get_model_endpoint_response.pyi similarity index 100% rename from launch/api_client/model/get_model_endpoint_response.pyi rename to launch/openapi_client/model/get_model_endpoint_response.pyi diff --git a/launch/api_client/model/gpu_type.py b/launch/openapi_client/model/gpu_type.py similarity index 95% rename from launch/api_client/model/gpu_type.py rename to launch/openapi_client/model/gpu_type.py index 3d1ee053..2de6e59f 100644 --- a/launch/api_client/model/gpu_type.py +++ b/launch/openapi_client/model/gpu_type.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class GpuType(schemas.EnumBase, schemas.StrSchema): diff --git a/launch/api_client/model/gpu_type.pyi b/launch/openapi_client/model/gpu_type.pyi similarity index 100% rename from launch/api_client/model/gpu_type.pyi rename to launch/openapi_client/model/gpu_type.pyi diff --git a/launch/api_client/model/http_validation_error.py b/launch/openapi_client/model/http_validation_error.py similarity index 96% rename from launch/api_client/model/http_validation_error.py rename to launch/openapi_client/model/http_validation_error.py index 5ba37eb2..53dfd1d9 100644 --- a/launch/api_client/model/http_validation_error.py +++ b/launch/openapi_client/model/http_validation_error.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class HTTPValidationError(schemas.DictSchema): @@ -130,4 +130,4 @@ def __new__( ) -from launch.api_client.model.validation_error import ValidationError +from launch.openapi_client.model.validation_error import ValidationError diff --git a/launch/api_client/model/http_validation_error.pyi b/launch/openapi_client/model/http_validation_error.pyi similarity index 100% rename from launch/api_client/model/http_validation_error.pyi rename to launch/openapi_client/model/http_validation_error.pyi diff --git a/launch/api_client/model/list_model_bundles_response.py b/launch/openapi_client/model/list_model_bundles_response.py similarity index 96% rename from launch/api_client/model/list_model_bundles_response.py rename to launch/openapi_client/model/list_model_bundles_response.py index a0b7e25b..76e8d603 100644 --- a/launch/api_client/model/list_model_bundles_response.py +++ b/launch/openapi_client/model/list_model_bundles_response.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class ListModelBundlesResponse(schemas.DictSchema): @@ -140,4 +140,6 @@ def __new__( ) -from launch.api_client.model.model_bundle_response import ModelBundleResponse +from launch.openapi_client.model.model_bundle_response import ( + ModelBundleResponse, +) diff --git a/launch/api_client/model/list_model_bundles_response.pyi b/launch/openapi_client/model/list_model_bundles_response.pyi similarity index 100% rename from launch/api_client/model/list_model_bundles_response.pyi rename to launch/openapi_client/model/list_model_bundles_response.pyi diff --git a/launch/api_client/model/list_model_endpoints_response.py b/launch/openapi_client/model/list_model_endpoints_response.py similarity index 96% rename from launch/api_client/model/list_model_endpoints_response.py rename to launch/openapi_client/model/list_model_endpoints_response.py index 64bea072..b4f3d30d 100644 --- a/launch/api_client/model/list_model_endpoints_response.py +++ b/launch/openapi_client/model/list_model_endpoints_response.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class ListModelEndpointsResponse(schemas.DictSchema): @@ -138,6 +138,6 @@ def __new__( ) -from launch.api_client.model.get_model_endpoint_response import ( +from launch.openapi_client.model.get_model_endpoint_response import ( GetModelEndpointResponse, ) diff --git a/launch/api_client/model/list_model_endpoints_response.pyi b/launch/openapi_client/model/list_model_endpoints_response.pyi similarity index 100% rename from launch/api_client/model/list_model_endpoints_response.pyi rename to launch/openapi_client/model/list_model_endpoints_response.pyi diff --git a/launch/api_client/model/model_bundle_environment_params.py b/launch/openapi_client/model/model_bundle_environment_params.py similarity index 97% rename from launch/api_client/model/model_bundle_environment_params.py rename to launch/openapi_client/model/model_bundle_environment_params.py index ec1a7b2b..5b39dbda 100644 --- a/launch/api_client/model/model_bundle_environment_params.py +++ b/launch/openapi_client/model/model_bundle_environment_params.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class ModelBundleEnvironmentParams(schemas.DictSchema): @@ -190,4 +190,6 @@ def __new__( ) -from launch.api_client.model.model_bundle_framework import ModelBundleFramework +from launch.openapi_client.model.model_bundle_framework import ( + ModelBundleFramework, +) diff --git a/launch/api_client/model/model_bundle_environment_params.pyi b/launch/openapi_client/model/model_bundle_environment_params.pyi similarity index 100% rename from launch/api_client/model/model_bundle_environment_params.pyi rename to launch/openapi_client/model/model_bundle_environment_params.pyi diff --git a/launch/api_client/model/model_bundle_framework.py b/launch/openapi_client/model/model_bundle_framework.py similarity index 95% rename from launch/api_client/model/model_bundle_framework.py rename to launch/openapi_client/model/model_bundle_framework.py index 66597ae1..5868a7a3 100644 --- a/launch/api_client/model/model_bundle_framework.py +++ b/launch/openapi_client/model/model_bundle_framework.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class ModelBundleFramework(schemas.EnumBase, schemas.StrSchema): diff --git a/launch/api_client/model/model_bundle_framework.pyi b/launch/openapi_client/model/model_bundle_framework.pyi similarity index 100% rename from launch/api_client/model/model_bundle_framework.pyi rename to launch/openapi_client/model/model_bundle_framework.pyi diff --git a/launch/api_client/model/model_bundle_order_by.py b/launch/openapi_client/model/model_bundle_order_by.py similarity index 95% rename from launch/api_client/model/model_bundle_order_by.py rename to launch/openapi_client/model/model_bundle_order_by.py index 987734ac..58a5683c 100644 --- a/launch/api_client/model/model_bundle_order_by.py +++ b/launch/openapi_client/model/model_bundle_order_by.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class ModelBundleOrderBy(schemas.EnumBase, schemas.StrSchema): diff --git a/launch/api_client/model/model_bundle_order_by.pyi b/launch/openapi_client/model/model_bundle_order_by.pyi similarity index 100% rename from launch/api_client/model/model_bundle_order_by.pyi rename to launch/openapi_client/model/model_bundle_order_by.pyi diff --git a/launch/api_client/model/model_bundle_packaging_type.py b/launch/openapi_client/model/model_bundle_packaging_type.py similarity index 95% rename from launch/api_client/model/model_bundle_packaging_type.py rename to launch/openapi_client/model/model_bundle_packaging_type.py index a36eab11..d4fba457 100644 --- a/launch/api_client/model/model_bundle_packaging_type.py +++ b/launch/openapi_client/model/model_bundle_packaging_type.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class ModelBundlePackagingType(schemas.EnumBase, schemas.StrSchema): diff --git a/launch/api_client/model/model_bundle_packaging_type.pyi b/launch/openapi_client/model/model_bundle_packaging_type.pyi similarity index 100% rename from launch/api_client/model/model_bundle_packaging_type.pyi rename to launch/openapi_client/model/model_bundle_packaging_type.pyi diff --git a/launch/api_client/model/model_bundle_response.py b/launch/openapi_client/model/model_bundle_response.py similarity index 98% rename from launch/api_client/model/model_bundle_response.py rename to launch/openapi_client/model/model_bundle_response.py index bd33d7f1..3027a84d 100644 --- a/launch/api_client/model/model_bundle_response.py +++ b/launch/openapi_client/model/model_bundle_response.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class ModelBundleResponse(schemas.DictSchema): @@ -381,9 +381,9 @@ def __new__( ) -from launch.api_client.model.model_bundle_environment_params import ( +from launch.openapi_client.model.model_bundle_environment_params import ( ModelBundleEnvironmentParams, ) -from launch.api_client.model.model_bundle_packaging_type import ( +from launch.openapi_client.model.model_bundle_packaging_type import ( ModelBundlePackagingType, ) diff --git a/launch/api_client/model/model_bundle_response.pyi b/launch/openapi_client/model/model_bundle_response.pyi similarity index 100% rename from launch/api_client/model/model_bundle_response.pyi rename to launch/openapi_client/model/model_bundle_response.pyi diff --git a/launch/api_client/model/model_endpoint_deployment_state.py b/launch/openapi_client/model/model_endpoint_deployment_state.py similarity index 99% rename from launch/api_client/model/model_endpoint_deployment_state.py rename to launch/openapi_client/model/model_endpoint_deployment_state.py index 076e6dd5..99ec97f3 100644 --- a/launch/api_client/model/model_endpoint_deployment_state.py +++ b/launch/openapi_client/model/model_endpoint_deployment_state.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class ModelEndpointDeploymentState(schemas.DictSchema): diff --git a/launch/api_client/model/model_endpoint_deployment_state.pyi b/launch/openapi_client/model/model_endpoint_deployment_state.pyi similarity index 100% rename from launch/api_client/model/model_endpoint_deployment_state.pyi rename to launch/openapi_client/model/model_endpoint_deployment_state.pyi diff --git a/launch/api_client/model/model_endpoint_order_by.py b/launch/openapi_client/model/model_endpoint_order_by.py similarity index 95% rename from launch/api_client/model/model_endpoint_order_by.py rename to launch/openapi_client/model/model_endpoint_order_by.py index e2eb86a1..07a304f2 100644 --- a/launch/api_client/model/model_endpoint_order_by.py +++ b/launch/openapi_client/model/model_endpoint_order_by.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class ModelEndpointOrderBy(schemas.EnumBase, schemas.StrSchema): diff --git a/launch/api_client/model/model_endpoint_order_by.pyi b/launch/openapi_client/model/model_endpoint_order_by.pyi similarity index 100% rename from launch/api_client/model/model_endpoint_order_by.pyi rename to launch/openapi_client/model/model_endpoint_order_by.pyi diff --git a/launch/api_client/model/model_endpoint_resource_state.py b/launch/openapi_client/model/model_endpoint_resource_state.py similarity index 99% rename from launch/api_client/model/model_endpoint_resource_state.py rename to launch/openapi_client/model/model_endpoint_resource_state.py index 32843450..4c9ba848 100644 --- a/launch/api_client/model/model_endpoint_resource_state.py +++ b/launch/openapi_client/model/model_endpoint_resource_state.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class ModelEndpointResourceState(schemas.DictSchema): @@ -466,4 +466,4 @@ def __new__( ) -from launch.api_client.model.gpu_type import GpuType +from launch.openapi_client.model.gpu_type import GpuType diff --git a/launch/api_client/model/model_endpoint_resource_state.pyi b/launch/openapi_client/model/model_endpoint_resource_state.pyi similarity index 100% rename from launch/api_client/model/model_endpoint_resource_state.pyi rename to launch/openapi_client/model/model_endpoint_resource_state.pyi diff --git a/launch/api_client/model/model_endpoint_status.py b/launch/openapi_client/model/model_endpoint_status.py similarity index 96% rename from launch/api_client/model/model_endpoint_status.py rename to launch/openapi_client/model/model_endpoint_status.py index 8703e65a..96943f08 100644 --- a/launch/api_client/model/model_endpoint_status.py +++ b/launch/openapi_client/model/model_endpoint_status.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class ModelEndpointStatus(schemas.EnumBase, schemas.StrSchema): diff --git a/launch/api_client/model/model_endpoint_status.pyi b/launch/openapi_client/model/model_endpoint_status.pyi similarity index 100% rename from launch/api_client/model/model_endpoint_status.pyi rename to launch/openapi_client/model/model_endpoint_status.pyi diff --git a/launch/api_client/model/model_endpoint_type.py b/launch/openapi_client/model/model_endpoint_type.py similarity index 94% rename from launch/api_client/model/model_endpoint_type.py rename to launch/openapi_client/model/model_endpoint_type.py index 57c909ed..a24ce4be 100644 --- a/launch/api_client/model/model_endpoint_type.py +++ b/launch/openapi_client/model/model_endpoint_type.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class ModelEndpointType(schemas.EnumBase, schemas.StrSchema): diff --git a/launch/api_client/model/model_endpoint_type.pyi b/launch/openapi_client/model/model_endpoint_type.pyi similarity index 100% rename from launch/api_client/model/model_endpoint_type.pyi rename to launch/openapi_client/model/model_endpoint_type.pyi diff --git a/launch/api_client/model/request_schema.py b/launch/openapi_client/model/request_schema.py similarity index 91% rename from launch/api_client/model/request_schema.py rename to launch/openapi_client/model/request_schema.py index 48c12333..b067de32 100644 --- a/launch/api_client/model/request_schema.py +++ b/launch/openapi_client/model/request_schema.py @@ -20,6 +20,6 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 RequestSchema = schemas.AnyTypeSchema diff --git a/launch/api_client/model/request_schema.pyi b/launch/openapi_client/model/request_schema.pyi similarity index 100% rename from launch/api_client/model/request_schema.pyi rename to launch/openapi_client/model/request_schema.pyi diff --git a/launch/api_client/model/response_schema.py b/launch/openapi_client/model/response_schema.py similarity index 91% rename from launch/api_client/model/response_schema.py rename to launch/openapi_client/model/response_schema.py index ac2a8608..8ff8d999 100644 --- a/launch/api_client/model/response_schema.py +++ b/launch/openapi_client/model/response_schema.py @@ -20,6 +20,6 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 ResponseSchema = schemas.AnyTypeSchema diff --git a/launch/api_client/model/response_schema.pyi b/launch/openapi_client/model/response_schema.pyi similarity index 100% rename from launch/api_client/model/response_schema.pyi rename to launch/openapi_client/model/response_schema.pyi diff --git a/launch/api_client/model/sync_endpoint_predict_response.py b/launch/openapi_client/model/sync_endpoint_predict_response.py similarity index 97% rename from launch/api_client/model/sync_endpoint_predict_response.py rename to launch/openapi_client/model/sync_endpoint_predict_response.py index 90dd7e21..2e214848 100644 --- a/launch/api_client/model/sync_endpoint_predict_response.py +++ b/launch/openapi_client/model/sync_endpoint_predict_response.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class SyncEndpointPredictResponse(schemas.DictSchema): @@ -170,4 +170,4 @@ def __new__( ) -from launch.api_client.model.task_status import TaskStatus +from launch.openapi_client.model.task_status import TaskStatus diff --git a/launch/api_client/model/sync_endpoint_predict_response.pyi b/launch/openapi_client/model/sync_endpoint_predict_response.pyi similarity index 100% rename from launch/api_client/model/sync_endpoint_predict_response.pyi rename to launch/openapi_client/model/sync_endpoint_predict_response.pyi diff --git a/launch/api_client/model/task_status.py b/launch/openapi_client/model/task_status.py similarity index 96% rename from launch/api_client/model/task_status.py rename to launch/openapi_client/model/task_status.py index 9693fad1..a6c86a4f 100644 --- a/launch/api_client/model/task_status.py +++ b/launch/openapi_client/model/task_status.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class TaskStatus(schemas.EnumBase, schemas.StrSchema): diff --git a/launch/api_client/model/task_status.pyi b/launch/openapi_client/model/task_status.pyi similarity index 100% rename from launch/api_client/model/task_status.pyi rename to launch/openapi_client/model/task_status.pyi diff --git a/launch/api_client/model/update_batch_job_request.py b/launch/openapi_client/model/update_batch_job_request.py similarity index 98% rename from launch/api_client/model/update_batch_job_request.py rename to launch/openapi_client/model/update_batch_job_request.py index ecf257dd..13b4300f 100644 --- a/launch/api_client/model/update_batch_job_request.py +++ b/launch/openapi_client/model/update_batch_job_request.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class UpdateBatchJobRequest(schemas.DictSchema): diff --git a/launch/api_client/model/update_batch_job_request.pyi b/launch/openapi_client/model/update_batch_job_request.pyi similarity index 100% rename from launch/api_client/model/update_batch_job_request.pyi rename to launch/openapi_client/model/update_batch_job_request.pyi diff --git a/launch/api_client/model/update_batch_job_response.py b/launch/openapi_client/model/update_batch_job_response.py similarity index 98% rename from launch/api_client/model/update_batch_job_response.py rename to launch/openapi_client/model/update_batch_job_response.py index 7230339e..0bf84145 100644 --- a/launch/api_client/model/update_batch_job_response.py +++ b/launch/openapi_client/model/update_batch_job_response.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class UpdateBatchJobResponse(schemas.DictSchema): diff --git a/launch/api_client/model/update_batch_job_response.pyi b/launch/openapi_client/model/update_batch_job_response.pyi similarity index 100% rename from launch/api_client/model/update_batch_job_response.pyi rename to launch/openapi_client/model/update_batch_job_response.pyi diff --git a/launch/api_client/model/update_model_endpoint_request.py b/launch/openapi_client/model/update_model_endpoint_request.py similarity index 99% rename from launch/api_client/model/update_model_endpoint_request.py rename to launch/openapi_client/model/update_model_endpoint_request.py index 484c135e..8440d694 100644 --- a/launch/api_client/model/update_model_endpoint_request.py +++ b/launch/openapi_client/model/update_model_endpoint_request.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class UpdateModelEndpointRequest(schemas.DictSchema): @@ -767,4 +767,4 @@ def __new__( ) -from launch.api_client.model.gpu_type import GpuType +from launch.openapi_client.model.gpu_type import GpuType diff --git a/launch/api_client/model/update_model_endpoint_request.pyi b/launch/openapi_client/model/update_model_endpoint_request.pyi similarity index 100% rename from launch/api_client/model/update_model_endpoint_request.pyi rename to launch/openapi_client/model/update_model_endpoint_request.pyi diff --git a/launch/api_client/model/update_model_endpoint_response.py b/launch/openapi_client/model/update_model_endpoint_response.py similarity index 98% rename from launch/api_client/model/update_model_endpoint_response.py rename to launch/openapi_client/model/update_model_endpoint_response.py index 162a8cc6..36e121c0 100644 --- a/launch/api_client/model/update_model_endpoint_response.py +++ b/launch/openapi_client/model/update_model_endpoint_response.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class UpdateModelEndpointResponse(schemas.DictSchema): diff --git a/launch/api_client/model/update_model_endpoint_response.pyi b/launch/openapi_client/model/update_model_endpoint_response.pyi similarity index 100% rename from launch/api_client/model/update_model_endpoint_response.pyi rename to launch/openapi_client/model/update_model_endpoint_response.pyi diff --git a/launch/api_client/model/validation_error.py b/launch/openapi_client/model/validation_error.py similarity index 99% rename from launch/api_client/model/validation_error.py rename to launch/openapi_client/model/validation_error.py index 3b7ad840..4d8f61fb 100644 --- a/launch/api_client/model/validation_error.py +++ b/launch/openapi_client/model/validation_error.py @@ -20,7 +20,7 @@ import frozendict # noqa: F401 import typing_extensions # noqa: F401 -from launch.api_client import schemas # noqa: F401 +from launch.openapi_client import schemas # noqa: F401 class ValidationError(schemas.DictSchema): diff --git a/launch/api_client/model/validation_error.pyi b/launch/openapi_client/model/validation_error.pyi similarity index 100% rename from launch/api_client/model/validation_error.pyi rename to launch/openapi_client/model/validation_error.pyi diff --git a/launch/openapi_client/models/__init__.py b/launch/openapi_client/models/__init__.py new file mode 100644 index 00000000..65f4e7cd --- /dev/null +++ b/launch/openapi_client/models/__init__.py @@ -0,0 +1,116 @@ +# coding: utf-8 + +# flake8: noqa + +# import all models into this package +# if you have many models here with many references from one model to another this may +# raise a RecursionError +# to avoid this, import only the models that you directly need like: +# from from launch.openapi_client.model.pet import Pet +# or import this package, but before doing it, use: +# import sys +# sys.setrecursionlimit(n) + +from launch.openapi_client.model.batch_job_serialization_format import ( + BatchJobSerializationFormat, +) +from launch.openapi_client.model.batch_job_status import BatchJobStatus +from launch.openapi_client.model.clone_model_bundle_request import ( + CloneModelBundleRequest, +) +from launch.openapi_client.model.create_async_task_response import ( + CreateAsyncTaskResponse, +) +from launch.openapi_client.model.create_batch_job_request import ( + CreateBatchJobRequest, +) +from launch.openapi_client.model.create_batch_job_resource_requests import ( + CreateBatchJobResourceRequests, +) +from launch.openapi_client.model.create_batch_job_response import ( + CreateBatchJobResponse, +) +from launch.openapi_client.model.create_model_bundle_request import ( + CreateModelBundleRequest, +) +from launch.openapi_client.model.create_model_bundle_response import ( + CreateModelBundleResponse, +) +from launch.openapi_client.model.create_model_endpoint_request import ( + CreateModelEndpointRequest, +) +from launch.openapi_client.model.create_model_endpoint_response import ( + CreateModelEndpointResponse, +) +from launch.openapi_client.model.delete_model_endpoint_response import ( + DeleteModelEndpointResponse, +) +from launch.openapi_client.model.endpoint_predict_request import ( + EndpointPredictRequest, +) +from launch.openapi_client.model.get_async_task_response import ( + GetAsyncTaskResponse, +) +from launch.openapi_client.model.get_batch_job_response import ( + GetBatchJobResponse, +) +from launch.openapi_client.model.get_model_endpoint_response import ( + GetModelEndpointResponse, +) +from launch.openapi_client.model.gpu_type import GpuType +from launch.openapi_client.model.http_validation_error import ( + HTTPValidationError, +) +from launch.openapi_client.model.list_model_bundles_response import ( + ListModelBundlesResponse, +) +from launch.openapi_client.model.list_model_endpoints_response import ( + ListModelEndpointsResponse, +) +from launch.openapi_client.model.model_bundle_environment_params import ( + ModelBundleEnvironmentParams, +) +from launch.openapi_client.model.model_bundle_framework import ( + ModelBundleFramework, +) +from launch.openapi_client.model.model_bundle_order_by import ( + ModelBundleOrderBy, +) +from launch.openapi_client.model.model_bundle_packaging_type import ( + ModelBundlePackagingType, +) +from launch.openapi_client.model.model_bundle_response import ( + ModelBundleResponse, +) +from launch.openapi_client.model.model_endpoint_deployment_state import ( + ModelEndpointDeploymentState, +) +from launch.openapi_client.model.model_endpoint_order_by import ( + ModelEndpointOrderBy, +) +from launch.openapi_client.model.model_endpoint_resource_state import ( + ModelEndpointResourceState, +) +from launch.openapi_client.model.model_endpoint_status import ( + ModelEndpointStatus, +) +from launch.openapi_client.model.model_endpoint_type import ModelEndpointType +from launch.openapi_client.model.request_schema import RequestSchema +from launch.openapi_client.model.response_schema import ResponseSchema +from launch.openapi_client.model.sync_endpoint_predict_response import ( + SyncEndpointPredictResponse, +) +from launch.openapi_client.model.task_status import TaskStatus +from launch.openapi_client.model.update_batch_job_request import ( + UpdateBatchJobRequest, +) +from launch.openapi_client.model.update_batch_job_response import ( + UpdateBatchJobResponse, +) +from launch.openapi_client.model.update_model_endpoint_request import ( + UpdateModelEndpointRequest, +) +from launch.openapi_client.model.update_model_endpoint_response import ( + UpdateModelEndpointResponse, +) +from launch.openapi_client.model.validation_error import ValidationError diff --git a/launch/api_client/paths/__init__.py b/launch/openapi_client/paths/__init__.py similarity index 94% rename from launch/api_client/paths/__init__.py rename to launch/openapi_client/paths/__init__.py index d7e772a6..2fa6bba2 100644 --- a/launch/api_client/paths/__init__.py +++ b/launch/openapi_client/paths/__init__.py @@ -1,6 +1,6 @@ # do not import all endpoints into this module because that uses a lot of memory and stack frames # if you need the ability to import all endpoints from this module, import them with -# from launch.api_client.apis.path_to_api import path_to_api +# from launch.openapi_client.apis.path_to_api import path_to_api import enum diff --git a/launch/api_client/paths/healthcheck/__init__.py b/launch/openapi_client/paths/healthcheck/__init__.py similarity index 66% rename from launch/api_client/paths/healthcheck/__init__.py rename to launch/openapi_client/paths/healthcheck/__init__.py index ae6dcb46..b6b3e4d8 100644 --- a/launch/api_client/paths/healthcheck/__init__.py +++ b/launch/openapi_client/paths/healthcheck/__init__.py @@ -1,7 +1,7 @@ # do not import all endpoints into this module because that uses a lot of memory and stack frames # if you need the ability to import all endpoints from this module, import them with -# from launch.api_client.paths.healthcheck import Api +# from launch.openapi_client.paths.healthcheck import Api -from launch.api_client.paths import PathValues +from launch.openapi_client.paths import PathValues path = PathValues.HEALTHCHECK diff --git a/launch/api_client/paths/healthcheck/get.py b/launch/openapi_client/paths/healthcheck/get.py similarity index 98% rename from launch/api_client/paths/healthcheck/get.py rename to launch/openapi_client/paths/healthcheck/get.py index 8b959525..2beb39c1 100644 --- a/launch/api_client/paths/healthcheck/get.py +++ b/launch/openapi_client/paths/healthcheck/get.py @@ -20,8 +20,8 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions from . import path diff --git a/launch/api_client/paths/healthcheck/get.pyi b/launch/openapi_client/paths/healthcheck/get.pyi similarity index 100% rename from launch/api_client/paths/healthcheck/get.pyi rename to launch/openapi_client/paths/healthcheck/get.pyi diff --git a/launch/api_client/paths/healthz/__init__.py b/launch/openapi_client/paths/healthz/__init__.py similarity index 66% rename from launch/api_client/paths/healthz/__init__.py rename to launch/openapi_client/paths/healthz/__init__.py index 3253e712..a55b8737 100644 --- a/launch/api_client/paths/healthz/__init__.py +++ b/launch/openapi_client/paths/healthz/__init__.py @@ -1,7 +1,7 @@ # do not import all endpoints into this module because that uses a lot of memory and stack frames # if you need the ability to import all endpoints from this module, import them with -# from launch.api_client.paths.healthz import Api +# from launch.openapi_client.paths.healthz import Api -from launch.api_client.paths import PathValues +from launch.openapi_client.paths import PathValues path = PathValues.HEALTHZ diff --git a/launch/api_client/paths/healthz/get.py b/launch/openapi_client/paths/healthz/get.py similarity index 98% rename from launch/api_client/paths/healthz/get.py rename to launch/openapi_client/paths/healthz/get.py index 0faead57..1209b307 100644 --- a/launch/api_client/paths/healthz/get.py +++ b/launch/openapi_client/paths/healthz/get.py @@ -20,8 +20,8 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions from . import path diff --git a/launch/api_client/paths/healthz/get.pyi b/launch/openapi_client/paths/healthz/get.pyi similarity index 100% rename from launch/api_client/paths/healthz/get.pyi rename to launch/openapi_client/paths/healthz/get.pyi diff --git a/launch/api_client/paths/readyz/__init__.py b/launch/openapi_client/paths/readyz/__init__.py similarity index 66% rename from launch/api_client/paths/readyz/__init__.py rename to launch/openapi_client/paths/readyz/__init__.py index 9b49ccf0..3d2d432b 100644 --- a/launch/api_client/paths/readyz/__init__.py +++ b/launch/openapi_client/paths/readyz/__init__.py @@ -1,7 +1,7 @@ # do not import all endpoints into this module because that uses a lot of memory and stack frames # if you need the ability to import all endpoints from this module, import them with -# from launch.api_client.paths.readyz import Api +# from launch.openapi_client.paths.readyz import Api -from launch.api_client.paths import PathValues +from launch.openapi_client.paths import PathValues path = PathValues.READYZ diff --git a/launch/api_client/paths/readyz/get.py b/launch/openapi_client/paths/readyz/get.py similarity index 98% rename from launch/api_client/paths/readyz/get.py rename to launch/openapi_client/paths/readyz/get.py index 548ecabd..3a897590 100644 --- a/launch/api_client/paths/readyz/get.py +++ b/launch/openapi_client/paths/readyz/get.py @@ -20,8 +20,8 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions from . import path diff --git a/launch/api_client/paths/readyz/get.pyi b/launch/openapi_client/paths/readyz/get.pyi similarity index 100% rename from launch/api_client/paths/readyz/get.pyi rename to launch/openapi_client/paths/readyz/get.pyi diff --git a/launch/api_client/paths/v1_async_tasks/__init__.py b/launch/openapi_client/paths/v1_async_tasks/__init__.py similarity index 65% rename from launch/api_client/paths/v1_async_tasks/__init__.py rename to launch/openapi_client/paths/v1_async_tasks/__init__.py index aafe0cc4..f03bddc9 100644 --- a/launch/api_client/paths/v1_async_tasks/__init__.py +++ b/launch/openapi_client/paths/v1_async_tasks/__init__.py @@ -1,7 +1,7 @@ # do not import all endpoints into this module because that uses a lot of memory and stack frames # if you need the ability to import all endpoints from this module, import them with -# from launch.api_client.paths.v1_async_tasks import Api +# from launch.openapi_client.paths.v1_async_tasks import Api -from launch.api_client.paths import PathValues +from launch.openapi_client.paths import PathValues path = PathValues.V1_ASYNCTASKS diff --git a/launch/api_client/paths/v1_async_tasks/post.py b/launch/openapi_client/paths/v1_async_tasks/post.py similarity index 97% rename from launch/api_client/paths/v1_async_tasks/post.py rename to launch/openapi_client/paths/v1_async_tasks/post.py index 7a4d2a1c..051630a5 100644 --- a/launch/api_client/paths/v1_async_tasks/post.py +++ b/launch/openapi_client/paths/v1_async_tasks/post.py @@ -20,15 +20,17 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions -from launch.api_client.model.create_async_task_response import ( +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions +from launch.openapi_client.model.create_async_task_response import ( CreateAsyncTaskResponse, ) -from launch.api_client.model.endpoint_predict_request import ( +from launch.openapi_client.model.endpoint_predict_request import ( EndpointPredictRequest, ) -from launch.api_client.model.http_validation_error import HTTPValidationError +from launch.openapi_client.model.http_validation_error import ( + HTTPValidationError, +) from . import path diff --git a/launch/api_client/paths/v1_async_tasks/post.pyi b/launch/openapi_client/paths/v1_async_tasks/post.pyi similarity index 100% rename from launch/api_client/paths/v1_async_tasks/post.pyi rename to launch/openapi_client/paths/v1_async_tasks/post.pyi diff --git a/launch/api_client/paths/v1_async_tasks_task_id/__init__.py b/launch/openapi_client/paths/v1_async_tasks_task_id/__init__.py similarity index 65% rename from launch/api_client/paths/v1_async_tasks_task_id/__init__.py rename to launch/openapi_client/paths/v1_async_tasks_task_id/__init__.py index 21c8a0fc..d0d72b88 100644 --- a/launch/api_client/paths/v1_async_tasks_task_id/__init__.py +++ b/launch/openapi_client/paths/v1_async_tasks_task_id/__init__.py @@ -1,7 +1,7 @@ # do not import all endpoints into this module because that uses a lot of memory and stack frames # if you need the ability to import all endpoints from this module, import them with -# from launch.api_client.paths.v1_async_tasks_task_id import Api +# from launch.openapi_client.paths.v1_async_tasks_task_id import Api -from launch.api_client.paths import PathValues +from launch.openapi_client.paths import PathValues path = PathValues.V1_ASYNCTASKS_TASK_ID diff --git a/launch/api_client/paths/v1_async_tasks_task_id/get.py b/launch/openapi_client/paths/v1_async_tasks_task_id/get.py similarity index 97% rename from launch/api_client/paths/v1_async_tasks_task_id/get.py rename to launch/openapi_client/paths/v1_async_tasks_task_id/get.py index 1faa81d9..97eb47c9 100644 --- a/launch/api_client/paths/v1_async_tasks_task_id/get.py +++ b/launch/openapi_client/paths/v1_async_tasks_task_id/get.py @@ -20,12 +20,14 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions -from launch.api_client.model.get_async_task_response import ( +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions +from launch.openapi_client.model.get_async_task_response import ( GetAsyncTaskResponse, ) -from launch.api_client.model.http_validation_error import HTTPValidationError +from launch.openapi_client.model.http_validation_error import ( + HTTPValidationError, +) from . import path diff --git a/launch/api_client/paths/v1_async_tasks_task_id/get.pyi b/launch/openapi_client/paths/v1_async_tasks_task_id/get.pyi similarity index 100% rename from launch/api_client/paths/v1_async_tasks_task_id/get.pyi rename to launch/openapi_client/paths/v1_async_tasks_task_id/get.pyi diff --git a/launch/api_client/paths/v1_batch_jobs/__init__.py b/launch/openapi_client/paths/v1_batch_jobs/__init__.py similarity index 66% rename from launch/api_client/paths/v1_batch_jobs/__init__.py rename to launch/openapi_client/paths/v1_batch_jobs/__init__.py index 3d5cef58..09ea71dd 100644 --- a/launch/api_client/paths/v1_batch_jobs/__init__.py +++ b/launch/openapi_client/paths/v1_batch_jobs/__init__.py @@ -1,7 +1,7 @@ # do not import all endpoints into this module because that uses a lot of memory and stack frames # if you need the ability to import all endpoints from this module, import them with -# from launch.api_client.paths.v1_batch_jobs import Api +# from launch.openapi_client.paths.v1_batch_jobs import Api -from launch.api_client.paths import PathValues +from launch.openapi_client.paths import PathValues path = PathValues.V1_BATCHJOBS diff --git a/launch/api_client/paths/v1_batch_jobs/post.py b/launch/openapi_client/paths/v1_batch_jobs/post.py similarity index 97% rename from launch/api_client/paths/v1_batch_jobs/post.py rename to launch/openapi_client/paths/v1_batch_jobs/post.py index e23cb7c8..914abb78 100644 --- a/launch/api_client/paths/v1_batch_jobs/post.py +++ b/launch/openapi_client/paths/v1_batch_jobs/post.py @@ -20,15 +20,17 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions -from launch.api_client.model.create_batch_job_request import ( +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions +from launch.openapi_client.model.create_batch_job_request import ( CreateBatchJobRequest, ) -from launch.api_client.model.create_batch_job_response import ( +from launch.openapi_client.model.create_batch_job_response import ( CreateBatchJobResponse, ) -from launch.api_client.model.http_validation_error import HTTPValidationError +from launch.openapi_client.model.http_validation_error import ( + HTTPValidationError, +) from . import path diff --git a/launch/api_client/paths/v1_batch_jobs/post.pyi b/launch/openapi_client/paths/v1_batch_jobs/post.pyi similarity index 100% rename from launch/api_client/paths/v1_batch_jobs/post.pyi rename to launch/openapi_client/paths/v1_batch_jobs/post.pyi diff --git a/launch/api_client/paths/v1_batch_jobs_batch_job_id/__init__.py b/launch/openapi_client/paths/v1_batch_jobs_batch_job_id/__init__.py similarity index 64% rename from launch/api_client/paths/v1_batch_jobs_batch_job_id/__init__.py rename to launch/openapi_client/paths/v1_batch_jobs_batch_job_id/__init__.py index 3fa32eda..236a4a73 100644 --- a/launch/api_client/paths/v1_batch_jobs_batch_job_id/__init__.py +++ b/launch/openapi_client/paths/v1_batch_jobs_batch_job_id/__init__.py @@ -1,7 +1,7 @@ # do not import all endpoints into this module because that uses a lot of memory and stack frames # if you need the ability to import all endpoints from this module, import them with -# from launch.api_client.paths.v1_batch_jobs_batch_job_id import Api +# from launch.openapi_client.paths.v1_batch_jobs_batch_job_id import Api -from launch.api_client.paths import PathValues +from launch.openapi_client.paths import PathValues path = PathValues.V1_BATCHJOBS_BATCH_JOB_ID diff --git a/launch/api_client/paths/v1_batch_jobs_batch_job_id/get.py b/launch/openapi_client/paths/v1_batch_jobs_batch_job_id/get.py similarity index 97% rename from launch/api_client/paths/v1_batch_jobs_batch_job_id/get.py rename to launch/openapi_client/paths/v1_batch_jobs_batch_job_id/get.py index 716c6706..0cd2f9c4 100644 --- a/launch/api_client/paths/v1_batch_jobs_batch_job_id/get.py +++ b/launch/openapi_client/paths/v1_batch_jobs_batch_job_id/get.py @@ -20,10 +20,14 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions -from launch.api_client.model.get_batch_job_response import GetBatchJobResponse -from launch.api_client.model.http_validation_error import HTTPValidationError +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions +from launch.openapi_client.model.get_batch_job_response import ( + GetBatchJobResponse, +) +from launch.openapi_client.model.http_validation_error import ( + HTTPValidationError, +) from . import path diff --git a/launch/api_client/paths/v1_batch_jobs_batch_job_id/get.pyi b/launch/openapi_client/paths/v1_batch_jobs_batch_job_id/get.pyi similarity index 100% rename from launch/api_client/paths/v1_batch_jobs_batch_job_id/get.pyi rename to launch/openapi_client/paths/v1_batch_jobs_batch_job_id/get.pyi diff --git a/launch/api_client/paths/v1_batch_jobs_batch_job_id/put.py b/launch/openapi_client/paths/v1_batch_jobs_batch_job_id/put.py similarity index 97% rename from launch/api_client/paths/v1_batch_jobs_batch_job_id/put.py rename to launch/openapi_client/paths/v1_batch_jobs_batch_job_id/put.py index 5c33fbb0..02d44177 100644 --- a/launch/api_client/paths/v1_batch_jobs_batch_job_id/put.py +++ b/launch/openapi_client/paths/v1_batch_jobs_batch_job_id/put.py @@ -20,13 +20,15 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions -from launch.api_client.model.http_validation_error import HTTPValidationError -from launch.api_client.model.update_batch_job_request import ( +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions +from launch.openapi_client.model.http_validation_error import ( + HTTPValidationError, +) +from launch.openapi_client.model.update_batch_job_request import ( UpdateBatchJobRequest, ) -from launch.api_client.model.update_batch_job_response import ( +from launch.openapi_client.model.update_batch_job_response import ( UpdateBatchJobResponse, ) diff --git a/launch/api_client/paths/v1_batch_jobs_batch_job_id/put.pyi b/launch/openapi_client/paths/v1_batch_jobs_batch_job_id/put.pyi similarity index 100% rename from launch/api_client/paths/v1_batch_jobs_batch_job_id/put.pyi rename to launch/openapi_client/paths/v1_batch_jobs_batch_job_id/put.pyi diff --git a/launch/api_client/paths/v1_model_bundles/__init__.py b/launch/openapi_client/paths/v1_model_bundles/__init__.py similarity index 65% rename from launch/api_client/paths/v1_model_bundles/__init__.py rename to launch/openapi_client/paths/v1_model_bundles/__init__.py index 46bf5ae2..bdf1ad22 100644 --- a/launch/api_client/paths/v1_model_bundles/__init__.py +++ b/launch/openapi_client/paths/v1_model_bundles/__init__.py @@ -1,7 +1,7 @@ # do not import all endpoints into this module because that uses a lot of memory and stack frames # if you need the ability to import all endpoints from this module, import them with -# from launch.api_client.paths.v1_model_bundles import Api +# from launch.openapi_client.paths.v1_model_bundles import Api -from launch.api_client.paths import PathValues +from launch.openapi_client.paths import PathValues path = PathValues.V1_MODELBUNDLES diff --git a/launch/api_client/paths/v1_model_bundles/get.py b/launch/openapi_client/paths/v1_model_bundles/get.py similarity index 96% rename from launch/api_client/paths/v1_model_bundles/get.py rename to launch/openapi_client/paths/v1_model_bundles/get.py index e917842b..5ec1a307 100644 --- a/launch/api_client/paths/v1_model_bundles/get.py +++ b/launch/openapi_client/paths/v1_model_bundles/get.py @@ -20,13 +20,17 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions -from launch.api_client.model.http_validation_error import HTTPValidationError -from launch.api_client.model.list_model_bundles_response import ( +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions +from launch.openapi_client.model.http_validation_error import ( + HTTPValidationError, +) +from launch.openapi_client.model.list_model_bundles_response import ( ListModelBundlesResponse, ) -from launch.api_client.model.model_bundle_order_by import ModelBundleOrderBy +from launch.openapi_client.model.model_bundle_order_by import ( + ModelBundleOrderBy, +) from . import path diff --git a/launch/api_client/paths/v1_model_bundles/get.pyi b/launch/openapi_client/paths/v1_model_bundles/get.pyi similarity index 100% rename from launch/api_client/paths/v1_model_bundles/get.pyi rename to launch/openapi_client/paths/v1_model_bundles/get.pyi diff --git a/launch/api_client/paths/v1_model_bundles/post.py b/launch/openapi_client/paths/v1_model_bundles/post.py similarity index 97% rename from launch/api_client/paths/v1_model_bundles/post.py rename to launch/openapi_client/paths/v1_model_bundles/post.py index 4ff01f10..4b755863 100644 --- a/launch/api_client/paths/v1_model_bundles/post.py +++ b/launch/openapi_client/paths/v1_model_bundles/post.py @@ -20,15 +20,17 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions -from launch.api_client.model.create_model_bundle_request import ( +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions +from launch.openapi_client.model.create_model_bundle_request import ( CreateModelBundleRequest, ) -from launch.api_client.model.create_model_bundle_response import ( +from launch.openapi_client.model.create_model_bundle_response import ( CreateModelBundleResponse, ) -from launch.api_client.model.http_validation_error import HTTPValidationError +from launch.openapi_client.model.http_validation_error import ( + HTTPValidationError, +) from . import path diff --git a/launch/api_client/paths/v1_model_bundles/post.pyi b/launch/openapi_client/paths/v1_model_bundles/post.pyi similarity index 100% rename from launch/api_client/paths/v1_model_bundles/post.pyi rename to launch/openapi_client/paths/v1_model_bundles/post.pyi diff --git a/launch/api_client/paths/v1_model_bundles_clone_with_changes/__init__.py b/launch/openapi_client/paths/v1_model_bundles_clone_with_changes/__init__.py similarity index 63% rename from launch/api_client/paths/v1_model_bundles_clone_with_changes/__init__.py rename to launch/openapi_client/paths/v1_model_bundles_clone_with_changes/__init__.py index 35511bd5..86365dfd 100644 --- a/launch/api_client/paths/v1_model_bundles_clone_with_changes/__init__.py +++ b/launch/openapi_client/paths/v1_model_bundles_clone_with_changes/__init__.py @@ -1,7 +1,7 @@ # do not import all endpoints into this module because that uses a lot of memory and stack frames # if you need the ability to import all endpoints from this module, import them with -# from launch.api_client.paths.v1_model_bundles_clone_with_changes import Api +# from launch.openapi_client.paths.v1_model_bundles_clone_with_changes import Api -from launch.api_client.paths import PathValues +from launch.openapi_client.paths import PathValues path = PathValues.V1_MODELBUNDLES_CLONEWITHCHANGES diff --git a/launch/api_client/paths/v1_model_bundles_clone_with_changes/post.py b/launch/openapi_client/paths/v1_model_bundles_clone_with_changes/post.py similarity index 97% rename from launch/api_client/paths/v1_model_bundles_clone_with_changes/post.py rename to launch/openapi_client/paths/v1_model_bundles_clone_with_changes/post.py index 0108f19d..383103ef 100644 --- a/launch/api_client/paths/v1_model_bundles_clone_with_changes/post.py +++ b/launch/openapi_client/paths/v1_model_bundles_clone_with_changes/post.py @@ -20,15 +20,17 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions -from launch.api_client.model.clone_model_bundle_request import ( +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions +from launch.openapi_client.model.clone_model_bundle_request import ( CloneModelBundleRequest, ) -from launch.api_client.model.create_model_bundle_response import ( +from launch.openapi_client.model.create_model_bundle_response import ( CreateModelBundleResponse, ) -from launch.api_client.model.http_validation_error import HTTPValidationError +from launch.openapi_client.model.http_validation_error import ( + HTTPValidationError, +) from . import path diff --git a/launch/api_client/paths/v1_model_bundles_clone_with_changes/post.pyi b/launch/openapi_client/paths/v1_model_bundles_clone_with_changes/post.pyi similarity index 100% rename from launch/api_client/paths/v1_model_bundles_clone_with_changes/post.pyi rename to launch/openapi_client/paths/v1_model_bundles_clone_with_changes/post.pyi diff --git a/launch/api_client/paths/v1_model_bundles_latest/__init__.py b/launch/openapi_client/paths/v1_model_bundles_latest/__init__.py similarity index 65% rename from launch/api_client/paths/v1_model_bundles_latest/__init__.py rename to launch/openapi_client/paths/v1_model_bundles_latest/__init__.py index b5e54b32..074d2cc6 100644 --- a/launch/api_client/paths/v1_model_bundles_latest/__init__.py +++ b/launch/openapi_client/paths/v1_model_bundles_latest/__init__.py @@ -1,7 +1,7 @@ # do not import all endpoints into this module because that uses a lot of memory and stack frames # if you need the ability to import all endpoints from this module, import them with -# from launch.api_client.paths.v1_model_bundles_latest import Api +# from launch.openapi_client.paths.v1_model_bundles_latest import Api -from launch.api_client.paths import PathValues +from launch.openapi_client.paths import PathValues path = PathValues.V1_MODELBUNDLES_LATEST diff --git a/launch/api_client/paths/v1_model_bundles_latest/get.py b/launch/openapi_client/paths/v1_model_bundles_latest/get.py similarity index 97% rename from launch/api_client/paths/v1_model_bundles_latest/get.py rename to launch/openapi_client/paths/v1_model_bundles_latest/get.py index 0660279f..29c9a499 100644 --- a/launch/api_client/paths/v1_model_bundles_latest/get.py +++ b/launch/openapi_client/paths/v1_model_bundles_latest/get.py @@ -20,10 +20,14 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions -from launch.api_client.model.http_validation_error import HTTPValidationError -from launch.api_client.model.model_bundle_response import ModelBundleResponse +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions +from launch.openapi_client.model.http_validation_error import ( + HTTPValidationError, +) +from launch.openapi_client.model.model_bundle_response import ( + ModelBundleResponse, +) from . import path diff --git a/launch/api_client/paths/v1_model_bundles_latest/get.pyi b/launch/openapi_client/paths/v1_model_bundles_latest/get.pyi similarity index 100% rename from launch/api_client/paths/v1_model_bundles_latest/get.pyi rename to launch/openapi_client/paths/v1_model_bundles_latest/get.pyi diff --git a/launch/api_client/paths/v1_model_bundles_model_bundle_id/__init__.py b/launch/openapi_client/paths/v1_model_bundles_model_bundle_id/__init__.py similarity index 64% rename from launch/api_client/paths/v1_model_bundles_model_bundle_id/__init__.py rename to launch/openapi_client/paths/v1_model_bundles_model_bundle_id/__init__.py index d3532d3a..99e031a3 100644 --- a/launch/api_client/paths/v1_model_bundles_model_bundle_id/__init__.py +++ b/launch/openapi_client/paths/v1_model_bundles_model_bundle_id/__init__.py @@ -1,7 +1,7 @@ # do not import all endpoints into this module because that uses a lot of memory and stack frames # if you need the ability to import all endpoints from this module, import them with -# from launch.api_client.paths.v1_model_bundles_model_bundle_id import Api +# from launch.openapi_client.paths.v1_model_bundles_model_bundle_id import Api -from launch.api_client.paths import PathValues +from launch.openapi_client.paths import PathValues path = PathValues.V1_MODELBUNDLES_MODEL_BUNDLE_ID diff --git a/launch/api_client/paths/v1_model_bundles_model_bundle_id/get.py b/launch/openapi_client/paths/v1_model_bundles_model_bundle_id/get.py similarity index 97% rename from launch/api_client/paths/v1_model_bundles_model_bundle_id/get.py rename to launch/openapi_client/paths/v1_model_bundles_model_bundle_id/get.py index 11174fab..2103a00b 100644 --- a/launch/api_client/paths/v1_model_bundles_model_bundle_id/get.py +++ b/launch/openapi_client/paths/v1_model_bundles_model_bundle_id/get.py @@ -20,10 +20,14 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions -from launch.api_client.model.http_validation_error import HTTPValidationError -from launch.api_client.model.model_bundle_response import ModelBundleResponse +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions +from launch.openapi_client.model.http_validation_error import ( + HTTPValidationError, +) +from launch.openapi_client.model.model_bundle_response import ( + ModelBundleResponse, +) from . import path diff --git a/launch/api_client/paths/v1_model_bundles_model_bundle_id/get.pyi b/launch/openapi_client/paths/v1_model_bundles_model_bundle_id/get.pyi similarity index 100% rename from launch/api_client/paths/v1_model_bundles_model_bundle_id/get.pyi rename to launch/openapi_client/paths/v1_model_bundles_model_bundle_id/get.pyi diff --git a/launch/api_client/paths/v1_model_endpoints/__init__.py b/launch/openapi_client/paths/v1_model_endpoints/__init__.py similarity index 65% rename from launch/api_client/paths/v1_model_endpoints/__init__.py rename to launch/openapi_client/paths/v1_model_endpoints/__init__.py index f58f6045..8a241c32 100644 --- a/launch/api_client/paths/v1_model_endpoints/__init__.py +++ b/launch/openapi_client/paths/v1_model_endpoints/__init__.py @@ -1,7 +1,7 @@ # do not import all endpoints into this module because that uses a lot of memory and stack frames # if you need the ability to import all endpoints from this module, import them with -# from launch.api_client.paths.v1_model_endpoints import Api +# from launch.openapi_client.paths.v1_model_endpoints import Api -from launch.api_client.paths import PathValues +from launch.openapi_client.paths import PathValues path = PathValues.V1_MODELENDPOINTS diff --git a/launch/api_client/paths/v1_model_endpoints/get.py b/launch/openapi_client/paths/v1_model_endpoints/get.py similarity index 96% rename from launch/api_client/paths/v1_model_endpoints/get.py rename to launch/openapi_client/paths/v1_model_endpoints/get.py index 03cf6126..c0a2051f 100644 --- a/launch/api_client/paths/v1_model_endpoints/get.py +++ b/launch/openapi_client/paths/v1_model_endpoints/get.py @@ -20,13 +20,15 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions -from launch.api_client.model.http_validation_error import HTTPValidationError -from launch.api_client.model.list_model_endpoints_response import ( +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions +from launch.openapi_client.model.http_validation_error import ( + HTTPValidationError, +) +from launch.openapi_client.model.list_model_endpoints_response import ( ListModelEndpointsResponse, ) -from launch.api_client.model.model_endpoint_order_by import ( +from launch.openapi_client.model.model_endpoint_order_by import ( ModelEndpointOrderBy, ) diff --git a/launch/api_client/paths/v1_model_endpoints/get.pyi b/launch/openapi_client/paths/v1_model_endpoints/get.pyi similarity index 100% rename from launch/api_client/paths/v1_model_endpoints/get.pyi rename to launch/openapi_client/paths/v1_model_endpoints/get.pyi diff --git a/launch/api_client/paths/v1_model_endpoints/post.py b/launch/openapi_client/paths/v1_model_endpoints/post.py similarity index 97% rename from launch/api_client/paths/v1_model_endpoints/post.py rename to launch/openapi_client/paths/v1_model_endpoints/post.py index 43f29056..40d5604e 100644 --- a/launch/api_client/paths/v1_model_endpoints/post.py +++ b/launch/openapi_client/paths/v1_model_endpoints/post.py @@ -20,15 +20,17 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions -from launch.api_client.model.create_model_endpoint_request import ( +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions +from launch.openapi_client.model.create_model_endpoint_request import ( CreateModelEndpointRequest, ) -from launch.api_client.model.create_model_endpoint_response import ( +from launch.openapi_client.model.create_model_endpoint_response import ( CreateModelEndpointResponse, ) -from launch.api_client.model.http_validation_error import HTTPValidationError +from launch.openapi_client.model.http_validation_error import ( + HTTPValidationError, +) from . import path diff --git a/launch/api_client/paths/v1_model_endpoints/post.pyi b/launch/openapi_client/paths/v1_model_endpoints/post.pyi similarity index 100% rename from launch/api_client/paths/v1_model_endpoints/post.pyi rename to launch/openapi_client/paths/v1_model_endpoints/post.pyi diff --git a/launch/api_client/paths/v1_model_endpoints_api/__init__.py b/launch/openapi_client/paths/v1_model_endpoints_api/__init__.py similarity index 65% rename from launch/api_client/paths/v1_model_endpoints_api/__init__.py rename to launch/openapi_client/paths/v1_model_endpoints_api/__init__.py index c4f1910d..7f145b43 100644 --- a/launch/api_client/paths/v1_model_endpoints_api/__init__.py +++ b/launch/openapi_client/paths/v1_model_endpoints_api/__init__.py @@ -1,7 +1,7 @@ # do not import all endpoints into this module because that uses a lot of memory and stack frames # if you need the ability to import all endpoints from this module, import them with -# from launch.api_client.paths.v1_model_endpoints_api import Api +# from launch.openapi_client.paths.v1_model_endpoints_api import Api -from launch.api_client.paths import PathValues +from launch.openapi_client.paths import PathValues path = PathValues.V1_MODELENDPOINTSAPI diff --git a/launch/api_client/paths/v1_model_endpoints_api/get.py b/launch/openapi_client/paths/v1_model_endpoints_api/get.py similarity index 98% rename from launch/api_client/paths/v1_model_endpoints_api/get.py rename to launch/openapi_client/paths/v1_model_endpoints_api/get.py index 24323282..088b43bd 100644 --- a/launch/api_client/paths/v1_model_endpoints_api/get.py +++ b/launch/openapi_client/paths/v1_model_endpoints_api/get.py @@ -20,8 +20,8 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions from . import path diff --git a/launch/api_client/paths/v1_model_endpoints_api/get.pyi b/launch/openapi_client/paths/v1_model_endpoints_api/get.pyi similarity index 100% rename from launch/api_client/paths/v1_model_endpoints_api/get.pyi rename to launch/openapi_client/paths/v1_model_endpoints_api/get.pyi diff --git a/launch/api_client/paths/v1_model_endpoints_model_endpoint_id/__init__.py b/launch/openapi_client/paths/v1_model_endpoints_model_endpoint_id/__init__.py similarity index 64% rename from launch/api_client/paths/v1_model_endpoints_model_endpoint_id/__init__.py rename to launch/openapi_client/paths/v1_model_endpoints_model_endpoint_id/__init__.py index fb6cdf7c..8051b60b 100644 --- a/launch/api_client/paths/v1_model_endpoints_model_endpoint_id/__init__.py +++ b/launch/openapi_client/paths/v1_model_endpoints_model_endpoint_id/__init__.py @@ -1,7 +1,7 @@ # do not import all endpoints into this module because that uses a lot of memory and stack frames # if you need the ability to import all endpoints from this module, import them with -# from launch.api_client.paths.v1_model_endpoints_model_endpoint_id import Api +# from launch.openapi_client.paths.v1_model_endpoints_model_endpoint_id import Api -from launch.api_client.paths import PathValues +from launch.openapi_client.paths import PathValues path = PathValues.V1_MODELENDPOINTS_MODEL_ENDPOINT_ID diff --git a/launch/api_client/paths/v1_model_endpoints_model_endpoint_id/delete.py b/launch/openapi_client/paths/v1_model_endpoints_model_endpoint_id/delete.py similarity index 97% rename from launch/api_client/paths/v1_model_endpoints_model_endpoint_id/delete.py rename to launch/openapi_client/paths/v1_model_endpoints_model_endpoint_id/delete.py index fb76c011..5f2d885f 100644 --- a/launch/api_client/paths/v1_model_endpoints_model_endpoint_id/delete.py +++ b/launch/openapi_client/paths/v1_model_endpoints_model_endpoint_id/delete.py @@ -20,12 +20,14 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions -from launch.api_client.model.delete_model_endpoint_response import ( +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions +from launch.openapi_client.model.delete_model_endpoint_response import ( DeleteModelEndpointResponse, ) -from launch.api_client.model.http_validation_error import HTTPValidationError +from launch.openapi_client.model.http_validation_error import ( + HTTPValidationError, +) from . import path diff --git a/launch/api_client/paths/v1_model_endpoints_model_endpoint_id/delete.pyi b/launch/openapi_client/paths/v1_model_endpoints_model_endpoint_id/delete.pyi similarity index 100% rename from launch/api_client/paths/v1_model_endpoints_model_endpoint_id/delete.pyi rename to launch/openapi_client/paths/v1_model_endpoints_model_endpoint_id/delete.pyi diff --git a/launch/api_client/paths/v1_model_endpoints_model_endpoint_id/get.py b/launch/openapi_client/paths/v1_model_endpoints_model_endpoint_id/get.py similarity index 97% rename from launch/api_client/paths/v1_model_endpoints_model_endpoint_id/get.py rename to launch/openapi_client/paths/v1_model_endpoints_model_endpoint_id/get.py index 70ec5816..055716d1 100644 --- a/launch/api_client/paths/v1_model_endpoints_model_endpoint_id/get.py +++ b/launch/openapi_client/paths/v1_model_endpoints_model_endpoint_id/get.py @@ -20,12 +20,14 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions -from launch.api_client.model.get_model_endpoint_response import ( +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions +from launch.openapi_client.model.get_model_endpoint_response import ( GetModelEndpointResponse, ) -from launch.api_client.model.http_validation_error import HTTPValidationError +from launch.openapi_client.model.http_validation_error import ( + HTTPValidationError, +) from . import path diff --git a/launch/api_client/paths/v1_model_endpoints_model_endpoint_id/get.pyi b/launch/openapi_client/paths/v1_model_endpoints_model_endpoint_id/get.pyi similarity index 100% rename from launch/api_client/paths/v1_model_endpoints_model_endpoint_id/get.pyi rename to launch/openapi_client/paths/v1_model_endpoints_model_endpoint_id/get.pyi diff --git a/launch/api_client/paths/v1_model_endpoints_model_endpoint_id/put.py b/launch/openapi_client/paths/v1_model_endpoints_model_endpoint_id/put.py similarity index 97% rename from launch/api_client/paths/v1_model_endpoints_model_endpoint_id/put.py rename to launch/openapi_client/paths/v1_model_endpoints_model_endpoint_id/put.py index ea63b0cb..a248b7b9 100644 --- a/launch/api_client/paths/v1_model_endpoints_model_endpoint_id/put.py +++ b/launch/openapi_client/paths/v1_model_endpoints_model_endpoint_id/put.py @@ -20,13 +20,15 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions -from launch.api_client.model.http_validation_error import HTTPValidationError -from launch.api_client.model.update_model_endpoint_request import ( +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions +from launch.openapi_client.model.http_validation_error import ( + HTTPValidationError, +) +from launch.openapi_client.model.update_model_endpoint_request import ( UpdateModelEndpointRequest, ) -from launch.api_client.model.update_model_endpoint_response import ( +from launch.openapi_client.model.update_model_endpoint_response import ( UpdateModelEndpointResponse, ) diff --git a/launch/api_client/paths/v1_model_endpoints_model_endpoint_id/put.pyi b/launch/openapi_client/paths/v1_model_endpoints_model_endpoint_id/put.pyi similarity index 100% rename from launch/api_client/paths/v1_model_endpoints_model_endpoint_id/put.pyi rename to launch/openapi_client/paths/v1_model_endpoints_model_endpoint_id/put.pyi diff --git a/launch/api_client/paths/v1_model_endpoints_schema_json/__init__.py b/launch/openapi_client/paths/v1_model_endpoints_schema_json/__init__.py similarity index 64% rename from launch/api_client/paths/v1_model_endpoints_schema_json/__init__.py rename to launch/openapi_client/paths/v1_model_endpoints_schema_json/__init__.py index 4dd4f68d..eaf987f0 100644 --- a/launch/api_client/paths/v1_model_endpoints_schema_json/__init__.py +++ b/launch/openapi_client/paths/v1_model_endpoints_schema_json/__init__.py @@ -1,7 +1,7 @@ # do not import all endpoints into this module because that uses a lot of memory and stack frames # if you need the ability to import all endpoints from this module, import them with -# from launch.api_client.paths.v1_model_endpoints_schema_json import Api +# from launch.openapi_client.paths.v1_model_endpoints_schema_json import Api -from launch.api_client.paths import PathValues +from launch.openapi_client.paths import PathValues path = PathValues.V1_MODELENDPOINTSSCHEMA_JSON diff --git a/launch/api_client/paths/v1_model_endpoints_schema_json/get.py b/launch/openapi_client/paths/v1_model_endpoints_schema_json/get.py similarity index 98% rename from launch/api_client/paths/v1_model_endpoints_schema_json/get.py rename to launch/openapi_client/paths/v1_model_endpoints_schema_json/get.py index d11fddbc..7bae63cf 100644 --- a/launch/api_client/paths/v1_model_endpoints_schema_json/get.py +++ b/launch/openapi_client/paths/v1_model_endpoints_schema_json/get.py @@ -20,8 +20,8 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions from . import path diff --git a/launch/api_client/paths/v1_model_endpoints_schema_json/get.pyi b/launch/openapi_client/paths/v1_model_endpoints_schema_json/get.pyi similarity index 100% rename from launch/api_client/paths/v1_model_endpoints_schema_json/get.pyi rename to launch/openapi_client/paths/v1_model_endpoints_schema_json/get.pyi diff --git a/launch/api_client/paths/v1_sync_tasks/__init__.py b/launch/openapi_client/paths/v1_sync_tasks/__init__.py similarity index 66% rename from launch/api_client/paths/v1_sync_tasks/__init__.py rename to launch/openapi_client/paths/v1_sync_tasks/__init__.py index e05b06c3..bc95fbd6 100644 --- a/launch/api_client/paths/v1_sync_tasks/__init__.py +++ b/launch/openapi_client/paths/v1_sync_tasks/__init__.py @@ -1,7 +1,7 @@ # do not import all endpoints into this module because that uses a lot of memory and stack frames # if you need the ability to import all endpoints from this module, import them with -# from launch.api_client.paths.v1_sync_tasks import Api +# from launch.openapi_client.paths.v1_sync_tasks import Api -from launch.api_client.paths import PathValues +from launch.openapi_client.paths import PathValues path = PathValues.V1_SYNCTASKS diff --git a/launch/api_client/paths/v1_sync_tasks/post.py b/launch/openapi_client/paths/v1_sync_tasks/post.py similarity index 97% rename from launch/api_client/paths/v1_sync_tasks/post.py rename to launch/openapi_client/paths/v1_sync_tasks/post.py index 0134b5e3..e2f5b667 100644 --- a/launch/api_client/paths/v1_sync_tasks/post.py +++ b/launch/openapi_client/paths/v1_sync_tasks/post.py @@ -20,13 +20,15 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client import schemas # noqa: F401 -from launch.api_client import api_client, exceptions -from launch.api_client.model.endpoint_predict_request import ( +from launch.openapi_client import schemas # noqa: F401 +from launch.openapi_client import api_client, exceptions +from launch.openapi_client.model.endpoint_predict_request import ( EndpointPredictRequest, ) -from launch.api_client.model.http_validation_error import HTTPValidationError -from launch.api_client.model.sync_endpoint_predict_response import ( +from launch.openapi_client.model.http_validation_error import ( + HTTPValidationError, +) +from launch.openapi_client.model.sync_endpoint_predict_response import ( SyncEndpointPredictResponse, ) diff --git a/launch/api_client/paths/v1_sync_tasks/post.pyi b/launch/openapi_client/paths/v1_sync_tasks/post.pyi similarity index 100% rename from launch/api_client/paths/v1_sync_tasks/post.pyi rename to launch/openapi_client/paths/v1_sync_tasks/post.pyi diff --git a/launch/api_client/rest.py b/launch/openapi_client/rest.py similarity index 99% rename from launch/api_client/rest.py rename to launch/openapi_client/rest.py index 96d2a3ae..41a3239a 100644 --- a/launch/api_client/rest.py +++ b/launch/openapi_client/rest.py @@ -18,7 +18,7 @@ import urllib3 from urllib3._collections import HTTPHeaderDict -from launch.api_client.exceptions import ApiException, ApiValueError +from launch.openapi_client.exceptions import ApiException, ApiValueError logger = logging.getLogger(__name__) diff --git a/launch/api_client/schemas.py b/launch/openapi_client/schemas.py similarity index 99% rename from launch/api_client/schemas.py rename to launch/openapi_client/schemas.py index d04c1959..4fbf12d8 100644 --- a/launch/api_client/schemas.py +++ b/launch/openapi_client/schemas.py @@ -22,8 +22,8 @@ import frozendict from dateutil.parser.isoparser import _takes_ascii, isoparser -from launch.api_client.configuration import Configuration -from launch.api_client.exceptions import ApiTypeError, ApiValueError +from launch.openapi_client.configuration import Configuration +from launch.openapi_client.exceptions import ApiTypeError, ApiValueError class Unset(object): diff --git a/pyproject.toml b/pyproject.toml index 2a335750..14e824c5 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -22,7 +22,8 @@ exclude = ''' [tool.mypy] exclude = [ '^launch/clientlib/', - '^launch/api_client/' + '^launch/api_client/', + '^launch/openapi_client/', ] [tool.poetry] @@ -88,4 +89,5 @@ extend-select = ['Q'] flake8-quotes = {inline-quotes = 'double', multiline-quotes = 'double'} exclude = [ "launch/api_client", + "launch/openapi_client", ] diff --git a/tests/test_client.py b/tests/test_client.py index a82f8db3..d6c6ed74 100644 --- a/tests/test_client.py +++ b/tests/test_client.py @@ -12,8 +12,8 @@ from urllib3 import HTTPResponse import launch -from launch.api_client.api_client import ApiResponseWithoutDeserialization -from launch.api_client.model.list_model_endpoints_response import ( +from launch.openapi_client.api_client import ApiResponseWithoutDeserialization +from launch.openapi_client.model.list_model_endpoints_response import ( ListModelEndpointsResponse, ) diff --git a/tests/test_model_endpoint.py b/tests/test_model_endpoint.py index 809e7cd9..7945c369 100644 --- a/tests/test_model_endpoint.py +++ b/tests/test_model_endpoint.py @@ -7,7 +7,7 @@ from urllib3 import HTTPResponse import launch -from launch.api_client.api_client import ApiResponseWithoutDeserialization +from launch.openapi_client.api_client import ApiResponseWithoutDeserialization def _get_mock_client():