yjmldZddlmZddlmZmZmZmZddlm Z ddZ dd Z ddZ GddZ dS)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) urlencodereturnrc ddlm}|ddn9#t$rYn-t$r!}tt |d}~wwxYwddl}|S)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 6/home/ragecks/.hermes/hermes-agent/tools/fal_common.pyimport_fal_clientr!s$:::::: [/////      $$$#c((###$ s A  A AA queue_run_originrct|pdd}|std|dS)N/z$Managed FAL queue origin is required)rstriprstrip ValueError)rnormalized_origins r_normalize_fal_queue_url_formatr7sW,23399;;BB3GG A?@@@ " " ""r BaseException Optional[int]ct|dd}|(t|dd}t|tr|St|dd}t|tr|SdS)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)>spsJ--H=$77 fc " " M S- . .F&# 4rc2eZdZdZddZdd d d d d d ddZd S)_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). rrkeyrrct|dd}|tdt|dd}|tdt||_|||_t|jdd|_t|dd|_t|dd|_t|d d|_t|d d|_ t|d d|_ t|d d|_ |jtd |j|jtd|jtddS)N SyncClientz>fal_client.SyncClient is required for managed FAL gateway modeclientz:fal_client.client is required for managed FAL gateway mode)r,_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_clientr1r2_request_handle_class_add_hint_header_add_priority_header_add_timeout_header)selfrr,rsync_client_class client_modules r__init__z_ManagedFalSyncClient.__init__Zsg#J dCC  $_`` ` Hd;;  [\\ \!@AQ!R!R--#666#D$5y$GG$+M;QSW$X$X!!(8KT!R!R%,]r1r:r%r9r2rTr;) r?rIrJrCrDrErFrGrHurlrequest_headersr#datas rsubmitz_ManagedFalSyncClient.submitts${2  * 3S))) )C  " 3M;#?@@@ @Cw}"--   5 A  ! !$ 8 8 8  (0"#lmmm  % %h @ @ @  $'/"#jkkk  $ $]O D D D,,    D-/@%HH# -   x(((}}))L)n-L)L)$ *   r)rrr,rrr)rIrrJrKrCrrDrLrErLrFrrGrMrHrN)__name__ __module__ __qualname____doc__rBr_rrr+r+Pshoooo>"%),059/ / / / / / / / rr+N)r r)rrr r)rr r r!)rc __future__rtypingrrrr urllib.parserrrr)r+rdrrrhs4#"""""------------"""""",####$S S S S S S S S S S r