`gjmddZddlmZddlmZmZmZmZddlm Z d dZ d dZ d dZ Gdd Z y )uShared FAL.ai SDK plumbing. Holds the stateless atoms that every FAL-backed tool needs: * :func:`import_fal_client` — lazy import + ``lazy_deps`` integration so ``fal_client`` isn't pulled at cold start (it added ~64 ms per CLI invocation when imported eagerly). * :class:`_ManagedFalSyncClient` — wrapper that drives a Nous-managed fal-queue gateway through the standard ``fal_client.SyncClient`` primitives. * :func:`_normalize_fal_queue_url_format`, :func:`_extract_http_status` — small helpers used by both the managed client wrapper and ``_submit_fal_request``. Stateful pieces (cache globals, ``_managed_fal_client*`` selectors, ``_submit_fal_request``) intentionally stay on :mod:`tools.image_generation_tool`. That module is the patch target for existing test suites (``tests/tools/test_image_generation.py``, ``tests/tools/test_managed_media_gateways.py``) and for the ``plugins/image_gen/fal/`` plugin's ``_it`` indirection — moving the caches here would silently defeat ``monkeypatch.setattr(image_tool, "_managed_fal_client", None)`` because the lookups would go against ``fal_common``'s namespace instead. See the per-rule walkthrough at issue #26241 for details. ) annotations)AnyDictOptionalUnion) urlencodec ddlm}|ddddl}|S#t$rYt$r}tt |d}~wwxYw)uImport ``fal_client`` (via ``lazy_deps`` when available) and return the module reference. Callers are responsible for caching the result on their own module global — keeping per-module globals lets tests monkey-patch the target module's ``fal_client`` attribute and have the patched value stick for that module's call sites. Raises :class:`ImportError` if the package is genuinely unavailable. r)ensurez image.falF)promptN)tools.lazy_depsr ImportError Exceptionstr fal_client) _lazy_ensureexcrs C/root/.hermes/venv/lib/python3.12/site-packages/tools/fal_common.pyimport_fal_clientr!sK$:[/     $#c(##$s AA?Ac~t|xsdjjd}|s td|dS)N/z$Managed FAL queue origin is required)rstriprstrip ValueError)queue_run_originnormalized_origins r_normalize_fal_queue_url_formatr7sD,2399;BB3G ?@@  ""ct|dd}|t|dd}t|tr|St|dd}t|tr|Sy)uReturn an HTTP status code from httpx/fal exceptions, else None. Defensive across exception shapes — httpx.HTTPStatusError exposes ``.response.status_code`` while fal_client wrappers may expose ``.status_code`` directly. responseN status_code)getattr isinstanceint)rr statuss r_extract_http_statusr&>sUsJ-H=$7 fc "M S- .F&# rcNeZdZdZddZddddddd ddZy) _ManagedFalSyncClientasSmall per-instance wrapper around ``fal_client.SyncClient`` for managed queue hosts. The wrapper carries its own ``fal_client`` module reference instead of reaching into a module global, so callers stay in control of which module's ``fal_client`` is in scope (matters for the test patches that swap the legacy module's ``fal_client`` attribute). cZt|dd}| tdt|dd}| tdt||_|||_t|jdd|_t|dd|_t|dd|_t|d d|_t|d d|_ t|d d|_ t|d d|_ |j td |j |j td|j tdy)N SyncClientz>fal_client.SyncClient is required for managed FAL gateway modeclientz:fal_client.client is required for managed FAL gateway mode)key_client_maybe_retry_request_raise_for_statusSyncRequestHandleadd_hint_headeradd_priority_headeradd_timeout_headerzFfal_client.SyncClient._client is required for managed FAL gateway modezKfal_client.client request helpers are required for managed FAL gateway modezLfal_client.client.SyncRequestHandle is required for managed FAL gateway mode) r" RuntimeErrorr_queue_url_format _sync_client _http_clientr.r/_request_handle_class_add_hint_header_add_priority_header_add_timeout_header)selfrr,rsync_client_class client_modules r__init__z_ManagedFalSyncClient.__init__Zs=#J dC  $_` ` Hd;  [\ \!@AQ!R-#6#D$5$5y$G$+M;QSW$X!!(8KT!R%,]"%),059/ / "/  /  / #/ / */ 3/ rr(N)returnr)rrr_r)r BaseExceptionr_z Optional[int])r] __future__rtypingrrrr urllib.parserrrr&r(r^rrrds34#--",#$S S r