iH$ddlZddlZddlmZmZmZmZmZmZddl m Z ddl m Z ddlmZmZddlmZddlmZerddlmZGdd ZdS) N) TYPE_CHECKINGDictListOptionalSequenceUnion)Block)ChunkMarkdownTextChunk)Metadata) SlackResponse) WebClientceZdZdZdddddddedejded ed eed eed eefd Z ddddeedee e e e fdeefdZddddddeedee e e e fdee ee e e effdee e efdef dZddee e e e fdefdZdS) ChatStreamaA helper class for streaming markdown text into a conversation using the chat streaming APIs. This class provides a convenient interface for the chat.startStream, chat.appendStream, and chat.stopStream API methods, with automatic buffering and state management. N)recipient_team_idrecipient_user_idtask_display_modeclientrchannellogger thread_ts buffer_sizerrrc ||_||_| dd|_|||||d| |_d|_d|_d|_||_dS)aInitialize a new ChatStream instance. The __init__ method creates a unique ChatStream instance that keeps track of one chat stream. Args: client: The WebClient instance to use for API calls. channel: An encoded ID that represents a channel, private group, or DM. logger: A logging channel for outputs. thread_ts: Provide another message's ts value to reply to. Streamed messages should always be replies to a user request. recipient_team_id: The encoded ID of the team the user receiving the streaming text belongs to. Required when streaming to channels. recipient_user_id: The encoded ID of the user to receive the streaming text. Required when streaming to channels. task_display_mode: Specifies how tasks are displayed in the message. A "timeline" displays individual tasks with text and "plan" displays all tasks together. buffer_size: The length of markdown_text to buffer in-memory before calling a method. Increasing this value decreases the number of method calls made for the same amount of text, which is useful to avoid rate limits. **kwargs: Additional arguments passed to the underlying API calls. tokenN)rrrrrstarting) _client_loggerpop_token _stream_args_buffer_state _stream_ts _buffer_size) selfrrrrrrrrkwargss T/root/projects/butler/venv/lib/python3.11/site-packages/slack_sdk/web/chat_stream.py__init__zChatStream.__init__su@  %+ZZ%>%> "!2!2!2        )-') markdown_textchunksr+r,returnc |jdkrtjd|j|dr|d|_||xj|z c_t|j|jks||j d d|i|St|j|j|j d|j d|j d|j d d }|j d tj|dS) aAppend to the stream. The "append" method appends to the chat stream being used. This method can be called multiple times. After the stream is stopped this method cannot be called. Args: chunks: An array of streaming chunks. Chunks can be markdown text, plan, or task update chunks. markdown_text: Accepts message text formatted in markdown. Limit this field to 12,000 characters. This text is what will be appended to the message received so far. **kwargs: Additional arguments passed to the underlying API calls. Returns: SlackResponse if the buffer was flushed, None if buffering. Raises: SlackRequestError: If the stream is already completed. Example: ```python streamer = client.chat_stream( channel="C0123456789", thread_ts="1700000001.123456", recipient_team_id="T0123456789", recipient_user_id="U0123456789", ) streamer.append(markdown_text="**hello wo") streamer.append(markdown_text="rld!**") streamer.stop() ``` completedz)Cannot append to stream: stream state is rNr,rrrr) buffer_lengthrrrrrzChatStream appended to buffer: )r#eSlackRequestErrorgetrr r"lenr% _flush_bufferr!rdebugjsondumps)r&r+r,r'detailss r(appendzChatStream.appendFsIJ ;+ % %%&_RVR]&_&_`` ` ::g   . **W--DK  $ LLM )LL t|   1 1 1V5G%4%>>V>v>> > ..,(,,Y77!%!2!6!67J!K!K!%!2!6!67J!K!K*..{;;    RTZ=P=PRRSSStr*)r+r,blocksmetadatar<r=c |jdkrtjd|j|dr|d|_|r|xj|z c_|jsj|jj d i|j d|ji}|dstjdt|d|_d|_g}t|jdkr(| t|j||||jjd |j|j d |j|||d |}d|_|S) aNStop the stream and finalize the message. Args: blocks: A list of blocks that will be rendered at the bottom of the finalized message. chunks: An array of streaming chunks. Chunks can be markdown text, plan, or task update chunks. markdown_text: Accepts message text formatted in markdown. Limit this field to 12,000 characters. This text is what will be appended to the message received so far. metadata: JSON object with event_type and event_payload fields, presented as a URL-encoded string. Metadata you post to Slack is accessible to any app or user who is a member of that workspace. **kwargs: Additional arguments passed to the underlying API calls. Returns: SlackResponse from the chat.stopStream API call. Raises: SlackRequestError: If the stream is already completed. Example: ```python streamer = client.chat_stream( channel="C0123456789", thread_ts="1700000001.123456", recipient_team_id="T0123456789", recipient_user_id="U0123456789", ) streamer.append(markdown_text="**hello wo") streamer.append(markdown_text="rld!**") streamer.stop() ``` r/z$Cannot stop stream: stream state is rtsz)Failed to stop stream: stream not started in_progressrtextNr)rrr?r<r,r=r1)r#r2r3r4rr r"r$rchat_startStreamr!strr5r;r extendchat_stopStream)r&r+r,r<r=r'response flushingss r(stopzChatStream.stop~sN ;+ % %%&ZT[&Z&Z[[ [ ::g   . **W--DK  * LLM )LL (4t|4#kH<<%% W)*UVVV!(4.11DO'DK.0 t|   ! !   .DLAAA B B B     V $ $ $/4</ +%i0     " r*c g}t|jdkr(|t|j||||jsH|jjd i|jd|j i|d|i}| d|_d|_ n/|jj d |j |jd|jd |d|i}d |_|S) zFFlush the internal buffer with chunks by making appropriate API calls.rrANrr,r?r@r)rrr?rr1) r5r"r;r rEr$rrCr!r r4r#chat_appendStream)r&r,r'chunks_to_flushrGs r(r6zChatStream._flush_buffers946 t|   ! !  " "#4$,#G#G#G H H H    " "6 * * * 4t|4#k' H 'll400DO'DKK5t|5k))4?  ' H r*)N)__name__ __module__ __qualname____doc__rDloggingLoggerintrr)rrrr r r;r r rIr6r1r*r(rrs,0+/+/.(.(.(.( .(  .(  .(.($C=.($C=.($C=.(.(.(.(f(,9= 666 }6%e "456 6 - 6666v(,9=EI48 EEE }E%e "456 E sHU4;-?$@@AB E 5x01 E EEEENHXeD%K6H-I$J`mr*r)r8rQtypingrrrrrrslack_sdk.errorserrorsr2slack_sdk.models.blocks.blocksr slack_sdk.models.messages.chunkr r slack_sdk.models.metadatar slack_sdk.web.slack_responser slack_sdkrrr1r*r(r\s GGGGGGGGGGGGGGGG000000DDDDDDDD......666666$######OOOOOOOOOOr*