33
44from fastapi import Depends , FastAPI
55from pydantic import NonNegativeInt
6- from servicelib .aiohttp .application_setup import ApplicationSetupError
76from servicelib .fastapi .dependencies import get_app
87from servicelib .rabbitmq import RabbitMQClient
98from servicelib .rabbitmq ._client_rpc import RabbitMQRPCClient
109from tenacity import before_sleep_log , retry , stop_after_delay , wait_fixed
1110
11+ from ...exceptions .custom_errors import ApplicationSetupError
1212from ...services_http .log_streaming import LogDistributor
1313
1414_MAX_WAIT_FOR_LOG_DISTRIBUTOR_SECONDS : Final [int ] = 10
1717
1818
1919def get_rabbitmq_rpc_client (
20- app : Annotated [FastAPI , Depends (get_app )]
20+ app : Annotated [FastAPI , Depends (get_app )],
2121) -> RabbitMQRPCClient :
2222 assert app .state .rabbitmq_rpc_client # nosec
2323 return cast (RabbitMQRPCClient , app .state .rabbitmq_rpc_client )
@@ -42,7 +42,7 @@ def get_log_distributor(app: Annotated[FastAPI, Depends(get_app)]) -> LogDistrib
4242async def wait_till_log_distributor_ready (app ) -> None :
4343 if not hasattr (app .state , "log_distributor" ):
4444 msg = f"Api server's log_distributor was not ready within { _MAX_WAIT_FOR_LOG_DISTRIBUTOR_SECONDS = } seconds"
45- raise ApplicationSetupError (msg )
45+ raise ApplicationSetupError (tip = msg )
4646
4747
4848def get_log_check_timeout (app : Annotated [FastAPI , Depends (get_app )]) -> NonNegativeInt :
0 commit comments