o wi} @sxddlmZddlZddlZddlZddlmZmZddlm Z m Z ddl m Z ddl mZdd dZGdddZdS)) annotationsN) ApiRequest ApiResponse)wandb_settings wandb_setup)ServiceConnection) MailboxHandle connectionServiceConnection | Noneapi_idstrreturnNonecCsF|dur!tt||WddS1swYdSdS)z6Clean up the api resources associated with the api id.N) contextlibsuppress Exceptionapi_cleanup_request)r r rQ/home/ubuntu/.local/lib/python3.10/site-packages/wandb/apis/public/service_api.py_cleanup s   "rc@s<eZdZdZdddZddd Z ddddZdddZd S) ServiceApizNA lazy initialized handle to the wandb-core service for handling API requests.settingswandb_settings.SettingscCs||_d|_tt|_dSN) _settings_service_connectionr uuiduuid4_api_id)selfrrrr__init__szServiceApi.__init__r rcCsL|jdur#t|_|j|j}|j|_t |t |j|j|jS)zLConnects to the service and initializes resources for handling API requests.N) rr singletonensure_serviceapi_init_requestrto_protoidrweakreffinalizer)rresponserrr_get_service_connections z"ServiceApi._get_service_connectionNrequestrtimeout float | NonercCs|}|j|_|j||dS)zSend an API request to the backend service. Creates the backend service connection if it has not been created yet. )r+)r)rr% api_request)rr*r+connrrrsend_api_request1s zServiceApi.send_api_requestMailboxHandle[ApiResponse]cs"|}|j|_||IdHS)zSend an API request to the backend service asynchronously. Args: request: The Api request to send. timeout: The timeout for the request. N)r)rr%api_request_async)rr*r.rrrsend_api_request_async>s z!ServiceApi.send_api_request_async)rr)r rr)r*rr+r,r r)r*rr r0)__name__ __module__ __qualname____doc__r r)r/r2rrrrrs    r)r r r r r r) __future__rrrr&wandb.proto.wandb_api_pb2rr wandb.sdkrr(wandb.sdk.lib.service.service_connectionr wandb.sdk.mailbox.mailbox_handlerrrrrrrs