o i@sdZddlZddlZddlmZmZmZmZddlm Z m Z ddl m Z ddl mZded ed dfd d ZGd dde ZGddde ZdS)a Custom client entrypoints that extend the generated BaseClient/AsyncBaseClient. Adds support for: - `access_token` as an alternative to `api_key` with the following rules: - If `access_token` is provided, it takes precedence and sets `Authorization: bearer ` - When `access_token` is used, `api_key` is forced to "token" to satisfy the generator, but the Authorization header is overridden for all HTTP and WebSocket requests. - `session_id` as a header sent with every request and websocket connection: - If `session_id` is provided, it will be used; otherwise, a UUID is auto-generated - The session_id is sent as the `x-deepgram-session-id` header - `transport_factory` to replace the default `websockets` transport with a custom one: - A callable: ``factory(url, headers) -> transport`` returning an object with ``send()``, ``recv()``, iteration, and ``close()`` support. N)AnyCallableDictOptional)AsyncBaseClient BaseClient)install_transport)BaseClientWrapperclient_wrapper bearer_tokenreturncs`|jdtdtttfffdd }t|||_t|dr,t|jdr.|j|j_dSdSdS)zOverride header providers to always use a Bearer authorization token. This updates both: - client_wrapper.get_headers() used by WebSocket clients - client_wrapper.httpx_client.base_headers used by HTTP clients _selfr cs}d|d<|S)Nzbearer Authorization)rheadersr original_get_headersrC/home/ubuntu/.local/lib/python3.10/site-packages/deepgram/client.py_get_headers_with_bearer$szF_apply_bearer_authorization_override.._get_headers_with_bearer httpx_client base_headersN) get_headersrrstrtypes MethodTypehasattrrr)r r rrrr$_apply_bearer_authorization_overrides  rc"eZdZdZdfdd ZZS)DeepgramClienta9 Custom Deepgram client that extends the generated BaseClient. Supports: - `session_id`: Optional session identifier. If not provided, a UUID is auto-generated. Sent as `x-deepgram-session-id` header in all requests and websocket connections. - `access_token`: Alternative to `api_key`. If provided, uses Bearer token authentication. - `transport_factory`: Custom sync WebSocket transport factory. A callable ``factory(url, headers) -> transport`` whose return value must support ``send()``, ``recv()``, iteration, and ``close()``. - `telemetry_opt_out`: Telemetry opt-out flag (maintained for backwards compatibility, no-op). - `telemetry_handler`: Telemetry handler (maintained for backwards compatibility, no-op). r Nc |dd}|dd}|dd}t|dd}|dd}|dur&|ntt}|d} | dur;i} | |d<|| d<|durN|d durNd |d <tj|i|||_|durdt |j ||durmt |d d|_ dS) N access_token session_idtransport_factorytelemetry_opt_outTtelemetry_handlerrx-deepgram-session-idapi_keytoken) sync_factory popboolruuiduuid4getsuper__init__r"r_client_wrapperr _telemetry_handler selfargskwargsr!r"r#r$r%final_session_idr __class__rrr1@*        zDeepgramClient.__init__r N__name__ __module__ __qualname____doc__r1 __classcell__rrr9rr1rcr)AsyncDeepgramClientaK Custom async Deepgram client that extends the generated AsyncBaseClient. Supports: - `session_id`: Optional session identifier. If not provided, a UUID is auto-generated. Sent as `x-deepgram-session-id` header in all requests and websocket connections. - `access_token`: Alternative to `api_key`. If provided, uses Bearer token authentication. - `transport_factory`: Custom async WebSocket transport factory. A callable ``factory(url, headers) -> transport`` whose return value must support ``send()``, ``recv()``, async iteration, and ``close()``. - `telemetry_opt_out`: Telemetry opt-out flag (maintained for backwards compatibility, no-op). - `telemetry_handler`: Telemetry handler (maintained for backwards compatibility, no-op). r Nc r ) Nr!r"r#r$Tr%rr&r'r() async_factoryr*r4r9rrr1yr;zAsyncDeepgramClient.__init__r<r=rrr9rrDjrCrD)rArr-typingrrrr base_clientrr transportr deepgram.core.client_wrapperr rrrrDrrrrs  9