
    Oj                    ~    d dl mZ d dlmZ d dlmZ d dlmZ d dl	m
Z
 d dlmZ d dlmZ dd	gZdd
Z G d de      Zy)    )annotations)override)	parse_uri)version)SecurityError)ClientConnection)connect_WebSocketConnect_recv_bytesc                    t        j                  d      s| j                  d      S | j                         }t        |t              r|j                  d      S |S )Nz13.F)decodezutf-8)websockets_version
startswithrecv
isinstancestrencode)
connectionmessages     l/var/www/origus_pro_usr/data/www/origus.pro/core/.venv/lib/python3.12/site-packages/openai/lib/_websocket.pyr   r      sL    ((/e,, ooG&0#&>7>>'"KGK    c                  2     e Zd ZdZdZed fd       Z xZS )r
   z5Keep WebSocket authentication on the original origin.z0Cross-origin WebSocket redirects are not allowedc                >   t         |   |      }t        |t              r|t	        | j
                        }t	        |      }|j                  |j                  |j                  f|j                  |j                  |j                  fk7  rt        | j                        S |S )N)superprocess_redirectr   r   r   urisecurehostportr   _redirect_error_message)selfexc
uri_or_exccurrenttarget	__class__s        r   r   z"_WebSocketConnect.process_redirect   s|    W-c2
j#&)Gz*Fgll;v{{\b\g\g?hh$T%A%ABBr   )r"   	ExceptionreturnzException | str)__name__
__module____qualname____doc__r    r   r   __classcell__)r&   s   @r   r
   r
      s    ?P r   N)r   SyncWebSocketConnectionr(   bytes)
__future__r   typing_extensionsr   websockets.urir   websockets.versionr   r   websockets.exceptionsr   websockets.sync.clientr   r.   websockets.asyncio.clientr	   __all__r   r
    r   r   <module>r9      s6    " & $ < / N -
.L r   