o RWj @s:ddlmZddlZddlZddlmZGdddZdS)) annotationsN)WebSocketQueueFullErrorc@sXeZdZdZddddZdd d Zd ddZd!ddZd"ddZd#ddZ d$ddZ dS)% SendQueuezBounded byte-size queue for outgoing WebSocket messages. Messages are stored as pre-serialized strings. The queue enforces a maximum byte budget so that unbounded buffering cannot occur during reconnection windows.  max_bytesintreturnNonecCs g|_d|_||_t|_dSNr)_queue_bytes _max_bytes threadingLock_lock)selfrra/home/esfera/Documents/content_generation/venv/lib/python3.10/site-packages/openai/_send_queue.py__init__szSendQueue.__init__datastrcCspt|d}|j$|j||jkrtd|j||f|j|7_WddS1s1wYdS)zAppend *data* to the queue. Raises :class:`WebSocketQueueFullError` if the message would exceed the byte-size limit. zutf-8z%send queue is full, message discardedN)lenencoderr rrr append)rr byte_lengthrrrenqueues"zSendQueue.enqueuesendtyping.Callable[[str], object]c Cs|jt|j}|jd|_Wdn1swYt|D]>\}\}}z||Wq$tyb|j||d}||j|_tdd|jD|_Wd1s\wYwdS)zSend every queued message via *send*. If *send* raises, the failing message and all subsequent messages are re-queued and the error is re-raised. rNcs|]\}}|VqdSNr.0_blrrr 8z'SendQueue.flush_sync..rlistr clearr enumerate Exceptionsumrrpendingir _byte_length remainingrrr flush_sync&s&       zSendQueue.flush_sync0typing.Callable[[str], typing.Awaitable[object]]c s|jt|j}|jd|_Wdn1swYt|D]A\}\}}z ||IdHWq%tyf|j||d}||j|_tdd|jD|_Wd1s`wYwdS)z$Async variant of :meth:`flush_sync`.rNcsrr rr!rrrr%Ir&z(SendQueue.flush_async..r'r-rrr flush_async;s(      zSendQueue.flush_async list[str]cCsN|jdd|jD}|jd|_|WdS1s wYdS)z&Remove and return all queued messages.cSsg|]\}}|qSrr)r"rr#rrr Osz#SendQueue.drain..rN)rr r)r )ritemsrrrdrainLs  $zSendQueue.draincCs4|j t|jWdS1swYdSr rrr rrrr__len__Ts$zSendQueue.__len__boolcCs8|jt|jdkWdS1swYdSr r9r:rrr__bool__Xs $zSendQueue.__bool__N)r)rrr r )rrr r )rrr r )rr3r r )r r5)r r)r r<) __name__ __module__ __qualname____doc__rrr2r4r8r;r=rrrrr s     r) __future__rtypingr _exceptionsrrrrrrs