g+ddlmZddlZddlZddlmZddlmZmZm Z m Z ddl Z ddl m Z mZddlmZddlmZddlmZer$dd lmZdd lmZdd lmZdd lmZdd lmZddlm Z edGddZ!edGddZ"edGddZ#e e"e#fZ$eGddZ% GddejLZ' ddZ( ddZ)y)) annotationsN) dataclass) TYPE_CHECKINGAnyLiteralUnion)runtimeutilshow_deprecation_warning)CacheReplayClosureError)in_cached_function)Iterator) FunctionType)MessageDeltaGenerator)Block) CacheTypeT)frozenc,eZdZUded<ded<ded<y) MediaMsgData bytes | strmediastrmimetypemedia_idN__name__ __module__ __qualname____annotations__d/var/www/openai/venv/lib/python3.12/site-packages/streamlit/runtime/caching/cached_message_replay.pyrr)s MMr$rcHeZdZUdZded<ded<ded<ded<dZd ed <y) ElementMsgDatazAn element's message and related metadata for replaying that element's function call. media_data is filled in iff this is a media element (image, audio, video). r delta_typermessageid_of_dg_called_onreturned_dgs_idNzlist[MediaMsgData] | None media_data)rr r!__doc__r"r,r#r$r%r'r'0s+ O ,0J)0r$r'c,eZdZUded<ded<ded<y) BlockMsgDatarr)rr*r+Nrr#r$r%r/r/?s Nr$r/c:eZdZUdZded<ded<ded<ded<y ) CachedResultz}The full results of calling a cache-decorated function, enough to replay the st functions called while executing it. rvaluez list[MsgData]messagesrmain_id sidebar_idN)rr r!r-r"r#r$r%r1r1Is J LOr$r1ceZdZdZd dZd dZejd dZ d dZ ddZ ddZ ddZ y )CachedMessageReplayContextzA utility for storing messages generated by `st` commands called inside a cached function. Data is stored in a thread-local object, so it's safe to use an instance of this class across multiple threads. cJg|_g|_g|_g|_||_yN)_cached_message_stack_seen_dg_stack_most_recent_messages _media_data _cache_type)self cache_types r%__init__z#CachedMessageReplayContext.__init__|s*:<".046"/1%r$c,tj|Sr9)r repr_)r?s r%__repr__z#CachedMessageReplayContext.__repr__szz$r$c#:K|jjg|jjtd}t j rd}t jd d|jj |_|jj |st jdyy#|jj |_|jj |st jdwwxYww)zContext manager that should wrap the invocation of a cached function. It allows us to track any `st.foo` messages that are generated from inside the function for playback during cache retrieval. FTN)r:appendr;setrgetpopr<)r?func nested_calls r%calling_cached_functionz2CachedMessageReplayContext.calling_cached_functions ""))"- ""35)  ! ! #K t$ . )-)C)C)G)G)ID &    # # %#&&u-*.)C)C)G)G)ID &    # # %#&&u-s A,D/C3ADADDcNtjsyt|jdk\rO|j ||}|j }t |||||}|jD]} | j|g|_|jD]} | j|y)zRecord the element protobuf as having been produced during any currently executing cached functions, so they can be replayed any time the function's execution is skipped because they're in the cache. N) r existslenr:select_dg_to_saver=r'rFr;add) r?r( element_proto invoked_dg_id used_dg_idreturned_dg_id id_to_saver,element_msg_datamsgsss r%save_element_messagez/CachedMessageReplayContext.save_element_messages~~  t)) *a /// zJJ))J-   22 ,-3 $$A EE. !%r$c|j||}|jD]}|jt||| |jD]}|j |yr9)rQr:rFr/r;rR)r? block_protorTrUrVrWrYrZs r%save_block_messagez-CachedMessageReplayContext.save_block_messagesW++M:F ..D KK [*nM N/$$A EE. !%r$c\t|jdkDr||jdvr|S|S)aSelect the id of the DG that this message should be invoked on during message replay. See Note [DeltaGenerator method invocation] invoked_id is the DG the st function was called on, usually `st._main`. acting_on_id is the DG the st function ultimately runs on, which may be different if the invoked DG delegated to another one because it was in a `with` block. r)rPr;)r? invoked_id acting_on_ids r%rQz,CachedMessageReplayContext.select_dg_to_saves6 t"" #a 'LD>- #,,.==AA JJ t}}!/"#"8"89;;s~~s{{Ch78@L!4!45C.!#"8"893;;/4: S001# G%j+>BFGs$C E( E(+>Fc"td|dy)NzThe cached widget replay feature was removed in 1.38. The `experimental_allow_widgets` parameter will also be removed in a future release. Please remove the `experimental_allow_widgets` parameter from the `@st.a` decorator and move all widget commands outside of cached functions. To speed up your app, we recommend moving your widgets into fragments. Find out more about fragments in [our docs](https://docs.streamlit.io/develop/api-reference/execution-flow/st.fragment). If you have a specific use-case that requires the `experimental_allow_widgets` functionality, please tell us via an [issue on Github](https://github.com/streamlit/streamlit/issues).r ) decorators r%show_widget_replay_deprecationrs$ # $L L r$)rvr1r@rrwrrgrh)rz'Literal['cache_data', 'cache_resource']rgrh)* __future__rri threading dataclassesrtypingrrrr streamlitrmr r streamlit.deprecation_utilr &streamlit.runtime.caching.cache_errorsr 7streamlit.runtime.scriptrunner_utils.script_run_contextrcollections.abcrtypesrgoogle.protobuf.messagerrlrstreamlit.proto.Block_pb2r$streamlit.runtime.caching.cache_typerrr'r/MsgDatar1localr7rrr#r$r%rs#!55#?J("/8/> $  $ 1 1 1 $  , -  >oNoNd+G +G&/+G>J+G +G\6 r$