o Tjh@sUddlmZddlZddlZddlZddlZddlZddlmZddl 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+eddGdddZ,Gddde-Z.Gddde-Z/e Z0de1d<e2Z3dBd$d%Z4dBd&d'Z5dCd,d-Z6dDd2d3Z7Gd4d5d5Z8Gd6d7d7Z9Gd8d9d9Z:Gd:d;d;Z;Gdd?d?Z=Gd@dAdAZ>dS)E) annotationsN) TracebackType)Anycast)WeakSet)deque) dataclass)Self)omit)maybe_transform) model_copy) BaseModel)WebSocketConnectionClosedError)ResponseResponseStreamEvent)ResponseStreamState)ResponsesConnectionAsyncResponsesConnection)ResponsesClientEvent)ResponsesServerEvent)ResponsesClientEventParamT)frozenc@sPeZdZUdZded<ded<ded<ded<ded<dZd ed <dd dZdS)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_bytesreturnNonecCsPt|D]\}}|dkr|durqt|tr!t|ts!|dkr%tdqdS)Nr"rz2WebSocket session limits must be positive integers)varsitems isinstancerbool ValueError)selfnamevaluer-j/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/openai/lib/responses_websocket/_session.py __post_init__.sz&ResponsesWebSocketLimits.__post_init__r#r$)__name__ __module__ __qualname____doc____annotations__r"r/r-r-r-r.rs   rc@seZdZdZdS)ResponsesWebSocketBufferErrorzDA session or response exceeded an application-selected buffer limit.N)r1r2r3r4r-r-r-r.r66sr6cs"eZdZdZdfdd ZZS) ResponsesWebSocketErrorzNA request-scoped protocol error. The original event is available as ``event``.eventrr#r$cstd||_dS)Nz7The Responses WebSocket request returned an error event)super__init__r8)r*r8 __class__r-r.r:=s  z ResponsesWebSocketError.__init__)r8rr#r$)r1r2r3r4r: __classcell__r-r-r;r.r7:sr7zWeakSet[object]_claimed connectionobjectr#r$cCsDt|tvr tdt|WddS1swYdS)Nz/This connection already has a WebSocket session) _claim_lockr> RuntimeErroraddr?r-r-r._claimFs  "rEcCs4tt|WddS1swYdSN)rAr>discardrDr-r-r._releaseMs "rHr8r+strrcCs&t|tr td||St||dS)Nzdict[str, Any])r'dictrgetgetattr)r8r+r-r-r._fieldRs&rM0ResponsesClientEvent | ResponsesClientEventParam stream_id str | Nonetuple[str, bool]cCstt|tr |jddnt|}t|ddk}|r0d|vr(|d|kr(td|dur0||d<tt |t |fS)NT) exclude_unsettypezresponse.createrOz-The command stream_id does not match its lane) r'rto_dictrJr(rKr)jsondumpsr r)r8rOdata is_creater-r-r._commandVsrYc@s eZdZdddZdd d Zd S) _Accumulatorlimitr!r#r$cCs"||_d|_d|_d|_d|_dS)Nr)r[sizestatefinalerror)r*r[r-r-r.r:bs  z_Accumulator.__init__r8rr\rcst|d}|dkrd_d_d_tttd_|dkr-d_d_d_t|_dSjdur4dS|dv}|dv}|rIt|ddurItd |sM|rij|7_j durijj krid_d_t d jdur}|r}j t t t|d d |rt tt|d_|d krjdurjj_n#jdurtjdddurtj_fddtjjDj_d_d_dSdS)NrSresponse.createdr) input_tools text_formatr_>response.output_item.doner`response.completed>response.failedrdresponse.incompleteresponsez,Terminal WebSocket event is missing responsez1Response accumulation exceeded max_response_bytesT)deeprdoutputcsg|]}jj|qSr-)r]_completed_output).0indexr*r-r. s z$_Accumulator.add..)rMr\r^r_rr r]r7r)r[r6 handle_eventrrr r_completed_responserLsortedrjri)r*r8r\kind compatibleterminalr-rmr.rCisL         z_Accumulator.addN)r[r!r#r$r8rr\rr#r$)r1r2r3r:rCr-r-r-r.rZas rZc@seZdZd ddZd S) _LaneStaterOrPr[r!r#r$cCsJ||_t|_d|_d|_d|_d|_d|_d|_t |_ t ||_ dSNrF)rOrqueuer\closed in_flightclosed_connectionreceived_generationconsumed_generationasyncioEventwakerZ accumulator)r*rOr[r-r-r.r:s z_LaneState.__init__N)rOrPr[r!r#r$)r1r2r3r:r-r-r-r.rvsrvc@s^eZdZd%ddZd&d d Zd'ddZd(ddZd)ddZd*d+ddZd,dd Z d-d#d$Z dS)._Routerlimitsrr?.ResponsesConnection | AsyncResponsesConnectionr#r$cCsD||_||_t|_i|_d|_d|_d|_d|_ | d|_ dSrw) r?r threading Condition conditionlaneseventsr\endedr_registerdefault)r*rr?r-r-r.r:s z_Router.__init__rOr@rvcCs|durt|trtd|durtd|j:|jr td||jvr)tdt |j|j j kr6t dt ||j j}||j|<|durI||_|WdS1sUwYdS)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'rIre fullmatchr)rrrBrlenrrr6rvr"rr*rOlaner-r-r.rs$  $z_Router.registerrcCs|jH|jr WddSd|_|jj|_|jt|j8_|j|j8_|j d|_t |j j |_ |j|jWddS1sNwYdS)NTr)rryr? _connectionr{rrrxr\clearrZrr"rrset notify_allr*rr-r-r.detachs    "z_Router.detachcCs\|j!t|jD]\}}|jr|j|ur|j|=q WddS1s'wYdSrF)rlistrr&ryr{)r*r?rOrr-r-r. reconnecteds"z_Router.reconnectedr8rr\rcCsN|j|jr WddSt|d}|j||j}|jr#|j}|jr/ WddSt|j|j j ksQ|j |j j ksQ|j ||j jksQ|j ||j jkrUtd|j||ft|d|jkrvt|ddkrv|jd7_d|_|j |7_ |j |7_ |j d7_ |j|jWddS1swYdS)NrOz.The session exceeded its buffered event limitsrSr`T)rrrMrrKrryrrxrrrrr\rr r6appendrOr|rzrrr)r*r8r\rOrr-r-r.dispatchs4    "z_Router.dispatchNr_BaseException | NonecCsf|j&d|_|jdur||_|jD]}|jq|jWddS1s,wYdS)NT)rrr_rvaluesrrr)r*r_rr-r-r.finishs   "z_Router.finishResponsesServerEvent | Nonec Cs|jrtd|jrw|j\}}|j|8_|j|8_|jd8_t|d|jkr0|Sz |j ||Wnt yS}z||j_ d|j_ WYd}~nd}~wwt|ddkrg|j d7_ d|_|St|ddvru|j|j k|_|S|jr|j dur|j tddS) NzThe WebSocket lane is detachedrrOrSr`T>r_rerdrfzThe WebSocket connection closed)ryEOFErrorrxpopleftr\rrMrOrrC Exceptionr_r]r}rzr|r)r*rr8r\excr-r-r.pops8 z _Router.poprXr(cCsv|j.|js |jrtd|r)|jrtdt|jj|_d|_WddSWddS1s4wYdS)N+The WebSocket session or lane is not activezGConsume the current response before sending another create on this laneT) rrryrBrzrZrr"r)r*rrXr-r-r.prepare"s "z_Router.prepare)rrr?rr#r$)rOr@r#rvrrvr#r$r?r@r#r$rurF)r_rr#r$)rrvr#r)rrvrXr(r#r$) r1r2r3r:rrrrrrrr-r-r-r.rs      rc@sNeZdZdZddd Zdd d ZdddddZdddddZdddZdS)ResponsesWebSocketLaneaA routed view of one session. ``close`` detaches it without closing the socket. Use a single consumer per lane. Cancel a synchronous wait with ``recv(timeout=...)``; a timeout never consumes an event. Different lanes may be consumed concurrently. sessionResponsesWebSocketSessionr]rvr#r$cC||_||_dSrF_session_stater*rr]r-r-r.r:4 zResponsesWebSocketLane.__init__r8rNcCs|jjj}|j|jt||jj\}}|jjj%|jjjr$t d||jjjur/t d|jj |j|Wdn1sBwY| |dS)NFThe WebSocket connection is reconnecting; restore state before sendingz>The WebSocket connection changed; restore state before sending) rr?r _check_sendrrYrO_connection_lock_is_reconnectingrB_routerrsendr*r8r?rWrXr-r-r.r8s    zResponsesWebSocketLane.sendNtimeoutr float | NonercCs|jjdur td|durdnt|}|jj}|j4 ||j}|dur2|WdS|dur8dn|t}|durJ|dkrJt d|j |q1sTwYdS)N3Enter the WebSocket session before receiving eventsTrz1Timed out waiting for a Responses WebSocket event) r_threadrBtime monotonicrrrr TimeoutErrorwait)r*rdeadlinerouterr8 remainingr-r-r.recvIs   zResponsesWebSocketLane.recvrcCs|durdnt|}|jj}|jjjdurK|jj|krK|jj|krK|dur)dntd|t}|j|d|jjjdurK|jj|krK|jj|ks#|jjjdurz|jjj dur^|jjj |durddntd|t}|j|d|jjjdusR|jjjS)zNConsume this response's remaining events; failed/incomplete are final results.Ngr) rrrr}rr^r|maxrr_)r*rr generationrr-r-r.get_final_responseXs"        z)ResponsesWebSocketLane.get_final_responsecC|jj|jdSrFrrrrrmr-r-r.closejzResponsesWebSocketLane.close)rrr]rvr#r$r8rNr#r$)rrr#r)rrr#rr0 r1r2r3r4r:rrrrr-r-r-r.r-s  rc@sbeZdZdZd%dd Zd&d d Zd'd dZd(ddZd)d*ddZd+dd Z d'd!d"Z d'd#d$Z dS),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. r?rrrr#r$cCs8||_t|||_d|_d|_d|_t||jj|_dSNF)r?rrr_closed_owns_connectionrrr*r?rr-r-r.r:~  z"ResponsesWebSocketSession.__init__r cCsl|jdus|jr tdt|jd|_tj|jddd|_z|j W|St y5| d|_w)N,A WebSocket session can only be entered onceTresponses-websocket)targetr+daemon) rrrBrEr?rrThread_readstartr_release_connectionrmr-r-r. __enter__s   z#ResponsesWebSocketSession.__enter__cCsRt|jrt|jd|_WddSWddS1s"wYdSr)rArr>rGr?rmr-r-r.rs "z-ResponsesWebSocketSession._release_connectionexc_typetype[BaseException] | Nonerr tracebackTracebackType | NonecCs.z|WdSty|durYdSwrFrrr*rrrr-r-r.__exit__s z"ResponsesWebSocketSession.__exit__NrOrPrcC$t||j|}|dur||_|SrF)rrrrrr-r-r.rzResponsesWebSocketSession.lanerrvcC:|jdus|js|jjs|jrtd|jjrtddSNrr)rrrrryrBr?rrr-r-r.r z%ResponsesWebSocketSession._check_sendc CsHddlm}m}zzR z|j}Wn8|yYn?|yJ}z$|j|r6|j|jjWYd}~q |jj }|rEt d|d|d}~ww|j |j |t|~q Wn)ty}z|j|z|jWn tyyYnwWYd}~nd}~wwW|j|jr|dSdS|j|jr|wwNr)ConnectionClosedOKConnectionClosedErrorTz0WebSocket connection closed with unsent messages)unsent_messages)websockets.exceptionsrrr? recv_bytes _reconnectrrr _send_queuedrainrr parse_eventrrrrrrr*rrrWrunsentr-r-r.rsV           zResponsesWebSocketSession._readc Cs|jr|jdus |jsdSd|_|j|jdurdSz%|jW|jturB|jj dd|jrr5LockrArErHrMrYrZrvrrrrrr-r-r-r.sR                   9As.