uPj BUdZddlmZddlmZddlmZmZddlZddl m Z ddl m Z ddl mZGd d eZGd d eZGd dZeZded<edGddZeZded<ed ddZddZGddZ ddZy) zCThin Python wrapper for the native Rust Responses WebSocket bridge.) annotations) dataclass)FinalProtocolN)ConnectionClosedOK)get_native_bridge)timeout_to_secondsc$eZdZddZddZddZy)RustResponsesWebSocketc KywNselftexts l/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/rust_bridge/responses_websocket.py send_textz RustResponsesWebSocket.send_textc Kywr rrs r recv_textz RustResponsesWebSocket.recv_textrrc Kywr rrs rclosezRustResponsesWebSocket.closerrNrstrreturnNone)rz str | Nonerr)__name__ __module__ __qualname__rrrrrr r s 50&r#r c.eZdZe ddZy) RustResponsesWebSocketConnectionc Kywr r)clsurlheaderstimeout_secondss rconnectz(RustResponsesWebSocketConnection.connects "%rN)r(rr)dict[str, str]r*z float | Nonerr )r r!r" classmethodr+rr#rr%r%s:% % %& % %%r#r%c eZdZy)_UnsetN)r r!r"rr#rr/r/!sr#r/z Final[_Unset]_UNSETT)slotsceZdZUdZded<y)_RustResponsesWebSocketStateN'RustResponsesWebSocketConnection | None connection)r r!r"r5__annotations__rr#rr3r3(s :>J7>r#r3z#Final[_RustResponsesWebSocketState]_STATEr5c<t|ts |t_yyr ) isinstancer/r7r5r8s rset_rust_responses_websocketr;0s j& )& *r#cztjtjSt}|yt|dd}|S)NResponsesWebSocketConnection)r7r5rgetattr) native_bridgeconnection_types rload_rust_responses_websocketrA8sF $   ,.MFM5tGO r#c,eZdZddZddZddZd dZy) _ConnectionAdapterc||_yr ) _connection)rr5s r__init__z_ConnectionAdapter.__init__Es :Dr#cVK|jj|d{y7wr )rErrs rsendz_ConnectionAdapter.sendHs ((...s )')crK|jjd{}| tdd|S7wr )rErr)rmessages rrecvz_ConnectionAdapter.recvKs9#//99;; ?$T40 0rXsI"!" 48;'X'%x%    ???/K.L+L DJ'@' ' ''"* *** *  *r#