~j*BdZddlmZddlZddlZddlZddlZddlmZddl m Z m Z m Z ddl Z ddlmZejeZGddeZGd d eZGd d eZGd deZed/dZd0dZd1d"Zd2d%Zd&d'd(d3d-Zgd.ZdS)4a#Shared HTTP + JSON helpers for the provider clients. Keeps the four AI provider files small and uniformly secured: - secrets read from settings only; never inline. - one httpx.AsyncClient per provider (process-lifetime). - all providers accept the configured *_MAX_CONCURRENCY bound. ) annotationsN)asynccontextmanager)Any AsyncIteratorOptional)redact_secretsceZdZdZdS)JsonExtractionErrorz HTTP )rrrrsuper__init__)selfrrr __class__s rrzProviderHttpError.__init__ sU   "4(($/  H77v7788888r)rrrrrrrr)r r rrr __classcell__)r s@rrrsCAA9999999999rrceZdZdZdS)RetryableProviderErroru<5xx / 429 / timeout — retried by the pipeline retry layer.Nr rrrr#r#'sFFDrr#ceZdZdZdS)NonRetryableProviderErroru14xx (other than 429) — fail-fast, do not retry.Nr rrrr%r%,s;;Drr% semaphoreasyncio.SemaphorerAsyncIterator[None]cnK|4d{VdWVdddd{VdS#1d{VswxYwYdS)Nr)r&s rboundedr*1s s $ ..limitrcFtjtd|S)N)asyncio Semaphoremax)r+s r_build_semaphorer17s  SE]] + ++rclienthttpx.AsyncClienturlrheadersdict[str, str]payloaddict[str, Any]timeoutfloat provider_namecK |||||d{V}nS#tjtjf$r5}td||t |dd|d}~wwxYw|j}|j} |dks|dkrt ||| |dkrt|||  | S#t$r}t|d |d |d}~wwxYw) zPOST JSON; on success return the decoded JSON body. Raises: RetryableProviderError on network error, 429, or any 5xx. NonRetryableProviderError on any other non-2xx status. )jsonr5r9Nz%s HTTP error: %srz network: iiiz returned non-JSON body (HTTP )) posthttpxTimeoutException NetworkErrorloggerwarningr# status_codetextr%r= ValueErrorr ) r2r4r5r7r9r;responseexcrrs rhttp_post_jsonrJ;sDJSwQXYYYYYYYY  "E$6 7JJJ*M3???$]A7H37H7HIIIJ !F =D }}# $]FDAAA }}' vtDDD}} ! E EF E E E   s,$A40A//A4:C C3C..C3rFrcf|pd}|std tj|S#t$rYnwxYwt jd|t jt jz}|r8 tj| dS#t$rYnwxYwdD]m\}}| |}| |}|dkr8||kr2 tj|||dzcS#t$rYiwxYwntd)a:Extract a JSON value from a provider response that may contain prose. Tries (in order): 1. strict JSON parse of the whole text (when the model was called with response_format=json_schema / json_object). 2. The first ``` fenced JSON block. 3. The first {...} or [...] substring. zempty responsez```(?:json)?\s*(.*?)\s*```r-)){})[]zno JSON found in response) stripr r=loadsrGresearchDOTALL IGNORECASEgroupfindrfind)rFfenceopenercloserstartends r extract_jsonr`]sg JB    D 4!"2333 z$      I3T29r};T U UE  :ekk!nn-- -    D 2 &!!jj   B;;3;; z$uS1W}"566666     9 : ::s3= A  A =&B$$ B10B12D D D )choicesrmessagecontentT) choices_pathdedent_responserdtuple[str, ...]reboolcKt||||||d{V}|} |D]} t| trEt| tr| t | krt |d| | | } \t| t r| | vrt |d| | | } t| tst |d| S)zRun a chat-completions-style call and return the assistant text. `choices_path` is the dotted path into the response body for the assistant text. Adjust it for providers that nest it differently. )r2r4r5r7r9r;Nz: choices path missing at z#: assistant content is not a string)rJ isinstancerlistlenr dictr) r2r4r5r7r9r;rdredatanodekeys r complete_chatrp~s1   #       DD c3   dD)) ]SCII-=-=)]*[*[VY*[*[\\\9DDdD)) _S__)]*]*]VY*]*]^^^9DD dC Y!]"W"W"WXXX Kr) r*rpr`rJr r%rr#r1)r&r'rr()r+rrr')r2r3r4rr5r6r7r8r9r:r;rrr8)rFrrr)r2r3r4rr5r6r7r8r9r:r;rrdrfrergrr)r __future__rr.r=loggingrT contextlibrtypingrrrr@ app.loggingr getLoggerr rCrGr RuntimeErrorrr#r%r*r1rJr`rp__all__rrrrys#"""""  ******////////// &&&&&&  8 $ $GGGGG*GGG99999 999     .        1    ,,,,D;;;;R%I $$$$$$R   r