OjhPUddlmZddlZddlZddlZddlZddlZddlmZddl m Z m Z ddl m Z ddlmZddlmZddlmZd d lmZd d lmZd d lmZd d lmZd dlmZd dlmZm Z ddl!m"Z"d dl#m$Z$m%Z%d dl&m'Z'd dl(m)Z)d dl*m+Z+edGddZ,Gdde-Z.Gdde-Z/e Z0de1d<ejdZ3d2d Z4d2d!Z5d3d"Z6d4d#Z7Gd$d%Z8Gd&d'Z9Gd(d)Z:Gd*d+Z;Gd,d-Z<Gd.d/Z=Gd0d1Z>y)5) annotationsN) TracebackType)Anycast)WeakSet)deque) dataclass)Self)omit)maybe_transform) model_copy) BaseModel)WebSocketConnectionClosedError)ResponseResponseStreamEvent)ResponseStreamState)ResponsesConnectionAsyncResponsesConnection)ResponsesClientEvent)ResponsesServerEvent)ResponsesClientEventParamT)frozencZeZdZUdZded<ded<ded<ded<ded<dZd ed <d d Zy) ResponsesWebSocketLimitsaApplication-selected limits for the opt-in session, not server limits. Byte counts use received message bytes, before parsing. They bound buffered event data, not transport framing, decoding, or Python object overhead. ``max_response_bytes`` optionally bounds the total event data fed into one accumulator; its default of ``None`` leaves response accumulation uncapped. Choose values appropriate to your output, including images and tool results. int max_lanesmax_events_per_lane max_eventsmax_bytes_per_lane max_bytesN int | Nonemax_response_bytesct|jD]=\}}|dk(r|t|trt|ts|dks4t dy)Nr$rz2WebSocket session limits must be positive integers)varsitems isinstancerbool ValueError)selfnamevalues ~/var/www/origus_pro_usr/data/www/origus.pro/core/.venv/lib/python3.12/site-packages/openai/lib/responses_websocket/_session.py __post_init__z&ResponsesWebSocketLimits.__post_init__.sZ:++- WKD%++ eS)Zt-DQR  !UVV  WreturnNone)__name__ __module__ __qualname____doc____annotations__r$r/r0r.rrs5NON%) )Wr0rceZdZdZy)ResponsesWebSocketBufferErrorzDA session or response exceeded an application-selected buffer limit.N)r4r5r6r7r9r0r.r;r;6sNr0r;c$eZdZdZdfd ZxZS)ResponsesWebSocketErrorzNA request-scoped protocol error. The original event is available as ``event``.c2t|d||_y)Nz7The Responses WebSocket request returned an error event)super__init__event)r+rA __class__s r.r@z ResponsesWebSocketError.__init__=s RS r0)rArr2r3)r4r5r6r7r@ __classcell__)rBs@r.r=r=:sXr0r=zWeakSet[object]_claimedct5|tvr tdtj|dddy#1swYyxYw)Nz/This connection already has a WebSocket session) _claim_lockrD RuntimeErroradd connections r._claimrKFs: !  !PQ Q Z !!!s )9Acdt5tj|dddy#1swYyxYwN)rFrDdiscardrIs r._releaserOMs( %$%%%s&/crt|trtd|j|St ||dS)Nzdict[str, Any])r(dictrgetgetattr)rAr,s r._fieldrTRs66@6M4 % ( , ,T 2mSZ[`bfhlSmmr0c t|tr|jdn t|}t |j ddk(}|rd|vr|d|k7r t d|||d<tjt|t|fS)NT) exclude_unsettypezresponse.create stream_idz-The command stream_id does not match its lane) r(rto_dictrQr)rRr*jsondumpsr r)rArXdata is_creates r._commandr^Vs0:5)0L5==t= ,RVW\R]DTXXf%)::;I $ 4 #4 #ALM M  )D  ::od,EF G RRr0ceZdZddZddZy) _AccumulatorcJ||_d|_d|_d|_d|_y)Nr)limitsizestatefinalerror)r+rbs r.r@z_Accumulator.__init__bs%  9= &* '+ r0c Lt|d}|dk(r/d|_d|_d|_t t t |_|dk(r&d|_d|_d|_t||_y|jy|dv}|dv}|rt|d td |s|rS|xj|z c_|j2|j|jkDrd|_d|_td |j 6|r4|j jttt|d |rttt|d|_|d k(r(|j |j j |_n|j t#|jddmt|j|_t%|j j&Dcgc]}|j j&|c}|j_d|_d|_yycc}w)NrWresponse.createdr) input_tools text_formatrf>response.output_item.donerhresponse.completed>response.failedrlresponse.incompleteresponsez,Terminal WebSocket event is missing responsez1Response accumulation exceeded max_response_bytesT)deeprloutput)rTrcrerfrr rdr=r*rbr; handle_eventrrrr_completed_responserSsorted_completed_outputrq)r+rArckind compatibleterminalindexs r.rHz_Accumulator.addiseV$ % %DIDJDJ,4PDJ 7?DJDIDJ07DJ  :: !    [[ uj19KL L  II Izz%$))djj*@! ! 34ghh :: !j JJ # #D)Z$[ \ huj(ABDJ++ 0F!ZZ;; 'GDJJ$,O,W' 3 EKDJJLhLhEi%Cg   04#$ #$ MMO '.r0N)rX str | Nonerbr#r2r3)r4r5r6r@r9r0r.r|r|s /r0r|cZeZdZ d dZd dZd dZd dZddZdddZddZ dd Z y)_Routerc||_||_tj|_i|_d|_d|_d|_d|_ |jd|_ yr~) rJlimits threading Condition conditionlaneseventsrcendedrfregisterdefault)r+rrJs r.r@z_Router.__init__sU% ",,.35    +/ }}T* r0c|1t|trtjd| t d|j 5|j r td||jvr t dt|j|jjk\r tdt||jj}||j|<|||_|cdddS#1swYyxYw)Nz[A-Za-z0-9_.-]{1,256}uXstream_id must be None or 1–256 ASCII letters, digits, underscores, hyphens or periodszThe WebSocket session has endedz0A lane with this stream_id is already registeredzThe session exceeded max_lanes)r(strre fullmatchr*rrrGrlenrrr;r|r$rr+rXlanes r.rz_Router.registers  9c*bll;SU^._.gwx x ^^ zz"#DEEDJJ& !STT4::$++"7"7734TUUi)G)GHD$(DJJy ! #    s B!C++C4c6|j5|jr dddyd|_|jj|_|xj t |jzc_|xj|jzc_|jjd|_t|jj|_ |jj|jj!dddy#1swYyxYw)NTr)rrrJ _connectionrrrrrcclearr`rr$rrset notify_allr+rs r.detachz_Router.detachs ^^ ({{ ( (DK%)__%@%@D " KK3tzz? *K II "I JJ   DI+DKK,J,JKD  IIMMO NN % % ' ( ( (sDC"DDc|j5t|jjD].\}}|js|j |us"|j|=0 dddy#1swYyxYwrM)rlistrr'rr)r+rJrXrs r. reconnectedz_Router.reconnectedse ^^ .#' (8(8(:#; . 4;;4#9#9#K 9-  . . . .s6A,A,A,,A5c&|j5|jr dddyt|d}|jj ||j }|j r |j }|j r dddyt|j|jjk\so|j|jjk\sL|j|z|jjkDs&|j|z|jjkDr t!d|jj#||ft|d|j$k(r+t|ddk(r|xj&dz c_d|_|xj|z c_ |xj|z c_ |xjdz c_ |j*j-|jj/dddy#1swYyxYw)NrXz.The session exceeded its buffered event limitsrWrhT)rrrTrrRrrrrrrrr rcr!r"r;appendrXrrrrr)r+rArcrXrs r.dispatchz_Router.dispatchs ^^ (zz ( (uk2I::>>)T\\:D{{||{{ ( (DJJ4;;#B#BB;;$++"8"8899t#dkk&D&DD99t#dkk&;&;;34dee JJ  udm ,e[)T^^;uf@UYk@k((A-(!% II I II I KK1 K IIMMO NN % % '1 ( ( (sHAHE:HHNc |j5d|_|j||_|jj D]}|j j |jjdddy#1swYyxYw)NT)rrrfrvaluesrrr)r+rfrs r.finishz_Router.finishso ^^ (DJzz!"  ))+   NN % % '  ( ( (s A.BB c|jr td|jr|jj\}}|xj|zc_|xj|zc_|xj dzc_t |d|jk7r|S |jj||t |ddk(r|xjdz c_ d|_|St |ddvr|j|jkD|_|S|j r#|j |jtdy#t$r,}||j_ d|j_ Yd}~d}~wwxYw) NzThe WebSocket lane is detachedrrXrWrhT>rfrmrlrnzThe WebSocket connection closed)rEOFErrorrpopleftrcrrTrXrrH Exceptionrfrdrrrr)r+rrArcexcs r.popz _Router.popsL ;;;< < ::**,,.KE4 II I II I KK1 Ke[)T^^;  .  $$UD1eV$(::((A-(!%Lv&*ss!%!9!9D>&'pqq#/ 0N0N#O !% & & &s A(A>>B)rrrJz.ResponsesConnection | AsyncResponsesConnectionr2r3)rXobjectr2r|rr|r2r3rJrr2r3rzrM)rfBaseException | Noner2r3)rr|r2zResponsesServerEvent | None)rr|r]r)r2r3) r4r5r6r@rrrrrrrr9r0r.rrsB +. +The WebSocket connection changed; restore state before sending) rrJr _check_sendrr^rX_connection_lock_is_reconnectingrG_routerrsendr+rArJr\r]s r.rzResponsesWebSocketLane.send8s]]--99  !!$++."5$++*?*?@i ]] % % 6 6 B}}''88"#kll!9!9!E!EE"#cdd MM ! ! ) )$++y A  B  B Bs B D,,D5Ntimeoutc|jj td|dntj|z}|jj }|j 5 |j|j}| |cdddS|dn|tjz }||dkr td|j j|q#1swYyxYw)N3Enter the WebSocket session before receiving eventsrz1Timed out waiting for a Responses WebSocket event) r_threadrGtime monotonicrrrr TimeoutErrorwait)r+rdeadlinerouterrA remainings r.recvzResponsesWebSocketLane.recvIs == (TU U"?40@70J&&    1 4;;/$  1 1 %-$4D(T^^EU:U (Y!^&'Z[[  %%i0 1 1s C A CCc|dntj|z}|jj}|jjj |jj|k(r|jj |kDr|dn td|tjz }|j||jjj 3|jj|k(r|jj |kDr|jjj |jjj |jjj|dn td|tjz }|j||jjj |jjj S)zNConsume this response's remaining events; failed/incomplete are final results.Ngr) rrrrrrermaxrrf)r+rr generationrs r.get_final_responsez)ResponsesWebSocketLane.get_final_responseXsl"?40@70J[[44 KK # # ) ) 5 //:= //*< ( 0c#x$..JZ?Z6[I IIiI ( KK # # ) ) 5 //:= //*<kk%%++3{{&&,,8kk--333 ( 0c#x$..JZ?Z6[I IIiI ( kk%%++3 {{&&,,,r0cb|jjj|jyrMrrrrr+s r.closezResponsesWebSocketLane.closej $$T[[1r0)rResponsesWebSocketSessionrdr|r2r3rA0ResponsesClientEvent | ResponsesClientEventParamr2r3)r float | Noner2r)rrr2rr1 r4r5r6r7r@rrrrr9r0r.rr-s+ "/3 1=A-$2r0rcbeZdZdZd dZd dZd dZ ddZdddZddZ d d Z d d Z y)ra`Own one existing connection and route its events with one background reader. Enter the session before sending. While active, consume events only through its lanes, not the connection's recv/iterator/dispatch_events. Closing the session closes its connection; closing a lane does not. Register lanes before sending requests. Events for unregistered lanes go to the default lane for inspection, without changing its response state. Every registered lane, including the default and closed lanes, reserves its ID and max_lanes slot until reconnect. Keep a lane open for sequential responses. Reconnect releases closed lanes; open lanes remain registered on the new socket. Recovery remains the existing connection's opt-in policy. Reconnecting a transport does not restore server-side response state or replay lane sends. c||_t|||_d|_d|_d|_t ||jj|_yNF)rJrrr_closed_owns_connectionrrr+rJrs r.r@z"ResponsesWebSocketSession.__init__~sE$vz2 04  %-dDLL4H4HI r0cR|j |jr tdt|jd|_t j|jdd|_ |jj|S#t$r|jd|_wxYw)N,A WebSocket session can only be entered onceTresponses-websocket)targetr,daemon) rrrGrKrJrrThread_readstartr_release_connectionrs r. __enter__z#ResponsesWebSocketSession.__enter__s << #t||MN Nt $ ''tzz@U^bc   LL       $ $ &DL  s (B"B&ct5|jr&tj|jd|_dddy#1swYyxYwr)rFrrDrNrJrs r.rz-ResponsesWebSocketSession._release_connections=  .$$  1(-% . . .s 3AA cJ |jy#t$r|YywxYwrMrrr+exc_typer tracebacks r.__exit__z"ResponsesWebSocketSession.__exit__s,  JJL { s  ""Ncbt||jj|}|||_|SrM)rrrrrs r.rzResponsesWebSocketSession.lanes/%dDLL,A,A),LM  DL r0c|j.|js"|jjs |jr t d|j jr t dyNrr)rrrrrrGrJrrs r.rz%ResponsesWebSocketSession._check_sendsN << 4<<4<<3E3ELM M ?? + +gh h ,r0cddlm}m} |jj }|j j|jj|t|~[#|$rYn|$r}|jj |r4|j j|jjYd}~|jjj}|rtd||d}~wwxYwn\#t$rP}|j j!| |jj#n#t$rYnwxYwYd}~nd}~wwxYw|j j!|j$r|j'yy#|j j!|j$r|j'wwxYwNr)ConnectionClosedOKConnectionClosedErrorz0WebSocket connection closed with unsent messages)unsent_messages)websockets.exceptionsrrrJ recv_bytes _reconnectrrr _send_queuedrainrr parse_eventrrrrrrr+rrr\runsents r.rzResponsesWebSocketSession._readsS + ??557D %%doo&A&A$&GTS!*, 11#6 001L1LM !__88>>@F<N`f"#   LL   $ %%'     LL   !||((* LL   !||((*sC>A&AC>&C:+C>-C:1A C5;C>5C55C::C>=F> EE#D>=E> E E E  E FEF9G c|jr'|j|jjsyd|_|jj |jy |j j |jtjurR|jjd|jjr td|jyy#|jtjurR|jjd|jjr td|jwwxYw)NTrz7The Responses WebSocket reader did not stop after close) rris_aliverrrJrrcurrent_threadjoinrGrrs r.rzResponsesWebSocketSession.closes  <t||9#;#;#== !!!!,<<((*&'`aa((* >s #C00A4E$)rJrrrr2r3r2r r1rztype[BaseException] | NonerrrzTracebackType | Noner2r3rM)rXrr2rr) r4r5r6r7r@rrrrrrrr9r0r.rrnsR J .29MZn  i +@+r0rc8eZdZdZddZd dZd dZd dZd dZy) AsyncResponsesWebSocketLanezKAn async routed lane. Canceling a recv/final-response wait never closes it.c ||_||_yrMrrs r.r@z$AsyncResponsesWebSocketLane.__init__rr0crK|jjj}|jj|jt ||jj \}}|jjj|j||j|d{y7wrM) rrJrrrr^rXrrrrs r.rz AsyncResponsesWebSocketLane.sends{]]--99  !!$++."5$++*?*?@i %%dkk9=ood###sB-B7/B50B7cK|jj td|jj} |j5|j |j }| |cdddS|j jjddd|j jjd{#1swY6xYw7w)Nr) r_taskrGrrrrrrr)r+rrAs r.rz AsyncResponsesWebSocketLane.recvs ==   &TU U&&!! ) 4;;/$  ) )   &&(  ) ++""'') ) )  ) ) *s6ACC & C0$C /CCC CCc,K|jj}|jjj|jj|k(r|jj|kDrk|j d{|jjj3|jj|k(r|jj|kDrk|jjjy|jjj |jjj |j d{|jjjy|jjjS77HwrM)rrrrerrrf)r+rs r.rz.AsyncResponsesWebSocketLane.get_final_responses([[44 KK # # ) ) 5 //:= //*<))+   KK # # ) ) 5 //:= //*<kk%%++3{{&&,,8kk--333))+  kk%%++3{{&&,,,  s2AF?AFA3F F $F/!FFcb|jjj|jyrMrrs r.rz!AsyncResponsesWebSocketLane.close rr0N)rAsyncResponsesWebSocketSessionrdr|r2r3r)r2r)r2rr1rr9r0r.rrsU$ * -2r0rcbeZdZdZd dZd dZ d dZdddZddZddZ dd Z dd Z y)rzOAsync counterpart of ResponsesWebSocketSession, owning the connection lifetime.c||_t|||_d|_d|_d|_t ||jj|_yr)rJrrr _close_taskrrrrs r.r@z'AsyncResponsesWebSocketSession.__init__sE$vz2 04 6: 249M9MN r0cK|j |jr tdt|jt j |jd|_|Sw)Nrrr,)rrrGrKrJr create_taskrrs r. __aenter__z)AsyncResponsesWebSocketSession.__aenter__sN :: !T\\MN Nt((>@F<N`f"# %%    LL   $ oo++---     LL   !DLL   !s GDA2A0A2AD0A22D7D9D=D B4D D5D  DDF&F/F E/(E+)E/.F/ E;8F:E;;F>F&FF& G&GGc<K|jcd|_|jj|jyt j |j|jd|_t j|jd{y7w)NTzresponses-websocket-closer") r rrrrrr#_closeshieldrs r.rz$AsyncResponsesWebSocketSession.closeWsq    #DL LL   !zz!&224;;tzz3JQlmD nnT--...sBBBBc K |j |d{|jj d{t |jy7<#tj$rYOwxYw75#t |jwxYwwrM)cancelrr*rJrrO)r+readers r.r,z%AsyncResponsesWebSocketSession._closeass & MMO  //'') ) ) T__ % ))   * T__ %sXBA4AAAA4A2A4BAA/,A4.A//A44B  B)rJrrrr2r3rrrM)rXrr2rrr1)r0zasyncio.Task[None]r2r3) r4r5r6r7r@r$r&rrrrr,r9r0r.rrsPYO29MZn  i "B/ &r0rr)rArr,rr2r)rArrXrr2ztuple[str, bool])? __future__rrrZrrrtypesrtypingrrweakrefr collectionsr dataclassesr typing_extensionsr _typesr _utilsr _compatr_modelsr _exceptionsrtypes.responsesrrstreaming.responsesrresources.responses.responsesrr&types.responses.responses_client_eventr&types.responses.responses_server_eventr,types.responses.responses_client_event_paramrrrGr;r=rDr8LockrFrKrOrTr^r`r|rrrrrr9r0r.rDs" !"%! 9<5ZJJU $WWW2OLOl$I/%inn !% nS66r / /B&B&J>2>2Bp+p+f+2+2\\&\&r0