EieUddlmZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z ddl m Z mZddlmZmZmZmZmZddlmZmZmZmZddlmZddlmZdd lmZm Z dd l!m"Z"dd l#m$Z$ee%Z&d e'd <dZ(d e'd<dZ)d e'd<dddddddddddddd Z*d e'd<gdZ+d e'd<d Z,d!Z-d"Z.d e'd#<d$Z/d e'd%<d&Z0d e'd'<d(Z1d e'd)<dDd*Z2dEd+Z3e d,-dFd.Z4dGd/Z5e d,-dHd0Z6dId1Z7dId2Z8Gd3d4Z9dJd5Z:dKd6Z;dLd7Z<e d8-dMd9Z=dNd:Z> dOd;Z?dPd<Z@ed=e d>ef?ZAe dQd@ZBe dR dSdAZBdRdTdBZB dR dUdCZCy)V) annotationsN)CallableSized) lru_cachewraps)AnyFinalTypeVarcastoverload)config file_util type_utilutil) get_logger) ForwardMsg)ArgumentCommand)RerunException)get_script_run_ctxr _LOGGER_MAX_TRACKED_COMMANDS_MAX_TRACKED_PER_COMMANDDG DataFrame PandasIndex PandasSeries PlotlyFigure BokehFigureMatplotlibFigure PandasStylerSnowparkConnection SQLConnection) z(streamlit.delta_generator.DeltaGeneratorzpandas.core.frame.DataFramezpandas.core.indexes.base.Indexzpandas.core.series.Serieszpandas.DataFramez pandas.Indexz pandas.Seriesz plotly.graph_objs._figure.Figurezbokeh.plotting.figure.Figurezmatplotlib.figure.Figurezpandas.io.formats.style.Stylerz>"'',,w7 8N) W   c*tt|S)a Detect Streamlit-shipped agent skills in well-known locations. Returns a sorted, deduplicated list of ``"::"`` tokens. ``location`` is ``home``, ``app``, or ``repo``; ``harness`` is one of ``agents``, ``claude``, ``codex``, ``cortex``, ``cursor``, ``gemini``, or ``opencode``; ``skill`` is one of ``_STREAMLIT_SKILL_NAMES``. Never raises: filesystem errors are swallowed and produce an empty list. The result is cached per ``app_dir`` for the lifetime of the process. )list_detect_installed_skills_cached)app_dirs r_detect_installed_skillsr"s /8 99r)maxsizec  tjjd}|rtjj|ntj}t |}||d}|Dtjj |tjj |k7r||d<t}|jD]\}}tD]\}} } } |dk(r>tjjtjj|| sK|dk(r| n| } tD]a} tjj|| | t}tjj|sI|j|d|d| ct!t#|S#t$$r!}t&j)d|Yd}~yd}~wwxYw) N~)homeapprepor:z+Failed to detect installed Streamlit skillsexc_info)rr expanduserrgetcwdrnormcasesetitemsrisdirrrrisfileaddtuplesorted Exceptionrdebug)rrrrrootstokenslocationrootharness project_dirhome_skills_diragent_home_dir harness_dirskillmarkerexs rrr0s ww!!#&*1bggoog&ryy{c")-c :   0 0 6"'':J:J3:O O E&M5#kkm DNHdIS DEo~v%bggmmGGLL~6/19V1Co 3DEWW\\k52HFww~~f- hZq 5'#BC D D D VF^$$  Cb QsE2F(52F(( G1G  Gc(ttS)aDetect agent harnesses installed under the user's home directory. Returns a sorted, deduplicated list of harness name tokens (``agents``, ``claude``, ``codex``, ``cortex``, ``cursor``, ``gemini``, ``opencode``) for each harness whose home-level config directory exists. Independent of whether Streamlit-specific skills are installed for that harness. The result is cached for the lifetime of the process. Never raises: filesystem errors are swallowed and produce an empty list. )r_detect_installed_agents_cachedrrr_detect_installed_agentsrUs /1 22rc tjjd}t}tD]V\}}}}tjj tjj ||sF|j|Xtt|S#t$r!}tjd|Yd}~yd}~wwxYw)Nrz!Failed to detect installed agentsrr) rrrrrrrrrrrrr )rr r _project_dir_home_skills_dirrrs rrrcs ww!!#&5GQ $ CG\#3^ww}}RWW\\$?@ 7# $VF^$$  9B GsA4B7&B C'CCctjjtr,t td5}|j }ddd|Stjjt r,t t d5}|j }ddd|Sttj}|S#1swYSxYw#1swYSxYw)aDGet the machine ID. This is a unique identifier for a user for tracking metrics, that is broken in different ways in some Linux distros and Docker images. - at times just a hash of '', which means many machines map to the same ID - at times a hash of the same string, when running in a Docker container zutf-8)encodingN) rrr_ETC_MACHINE_ID_PATHopenread_DBUS_MACHINE_ID_PATHstruuidgetnode)f machine_ids r_get_machine_id_v3r'qs ww~~*+ & 9 "QJ "  - . '' : "aJ " (  "   " sB>C >C Cctjdsytjd}d}tj j |r.tj|5}|j}ddd|sNttj}tj|5}|j|ddd|S|S#1swY[xYw#1swY|SxYw)aGet a random ID that is stable for each machine, generating if needed. This is a unique identifier for a user for tracking metrics. Instead of relying on a hardware address in the container or host we'll generate a UUID and store it in the ~/.streamlit hidden folder. zbrowser.gatherUsageStatszno-machine-id-v4 machine_id_v4N)r get_optionrget_streamlit_file_pathrrrstreamlit_readr r"r#uuid4streamlit_writewrite)filepath stable_idfileoutputs r_get_machine_id_v4r4s   7 8"00AHI ww~~h  % %h / $4 I $  %  & &x 0 $F LL # $ 9 $ $  $ s"C /C CC#creZdZUejZdZded<eddZ d dZ d dZ e d dZ y) InstallationNzInstallation | None _instancec|j<|j5|jt|_ddd|jS|jS#1swY|jSxYw)z#Returns the singleton Installation.N)r7_instance_lockr6)clss rinstancezInstallation.instances[ == ## 3==($0NCM 3}}s}}  3}}s AA)cttjtjt |_t |_yN)r"r#uuid5 NAMESPACE_DNSr'installation_id_v3r4installation_id_v4selfs r__init__zInstallation.__init__s4"% JJt))+=+? @# #5"6rc,tj|Sr=)rrepr_rBs r__repr__zInstallation.__repr__szz$rc|jSr=)r@rBs rinstallation_idzInstallation.installation_ids&&&r)returnr6)rJNonerJr")__name__ __module__ __qualname__ threadingLockr9r7__annotations__ classmethodr;rDrGpropertyrIrrrr6r6sK#Y^^%N%)I")  7 ''rr6cptjt5tj|r|n t |}d}t |dr |j}nt |dr |j}|jdk7r|jd|}|tvr t|}|cdddS#1swYyxYw)z7Get a simplified name for the type of the given object.unknownrOrMbuiltins.Nfailed) contextlibsuppressrinspectisclasstypehasattrrOrMrNr&)objobj_type type_names r_get_type_namercs   Y '!//#.3DI 8^ , --I Xz * ))I   * ,#../q z+_get_arg_keywords_cached..sI 66    - -w/@/@/V/V W X sAA)rget_func_parametersr)rgparamss r_get_arg_keywords_cachedr}s. * *4 0F  rctj|rtt|jStt|S)aZReturn argument names from a function's signature. This returns argument names matching the behavior of getfullargspec().args: - Both POSITIONAL_ONLY and POSITIONAL_OR_KEYWORD parameters (not keyword-only) - Includes 'self' for bound methods Uses caching to avoid repeated expensive inspect.signature() calls. Note: The underlying LRU cache holds strong references to function objects. This is fine for typical Streamlit usage with module-level functions, but dynamically created callables (e.g., closures, partials) may be retained until evicted from the cache. )r\ismethodrr}__func__)rgs r_get_arg_keywordsrs8",T]];<< (. //rct|}d}g}tj|}|}t|D]h\} } | } |r| dz } t || kDr|| n| } | dk(r| }-t | t | | } t| }|r|| _|j| j|jD]@\}}t |t |} t|}|r|| _|j| Bt|}|dk7rd|d|}|d k(r)|r't|d r|jrd |j}|d k(r+t |dkDrt|dtr |d}d |}t!||S)zCGet telemetry information for the given callable and its arguments.NrrC)ktry)rr streamlitz external:rcreate_instancerwz component:_bidi_componentz component_v2:)rwargs)rr\r enumeratermrrcromappendrrir_rwrkr"r) _command_func _command_namerkwargs arg_keywordsself_arg arguments is_methodrwirnposkeywordargument arg_metadatakwarg kwarg_valuetop_level_modulecomponent_names r_get_command_telemetryrs%]3LH "I  /I DD/#3  FA%(%6%:,q/1# f H g)<D(- %HJ"# %lln#{e~k'BC(5 %HJ" #-];;&+,AdV4 !!  Hf % MMHMM?+   SY]z$q'37Oa~./ 9 --rct|dzS)z"Convert seconds into microseconds.i@B)int)secondss rto_microsecondsrTs w" ##rF.)boundcyr=rrwrgs rgather_metricsr\s rcyr=rrs rrrcsrcvstjdd|dfd }|S|tdfd }tjt t 5|jjjtj|_ dddtd|S#1swYxYw) aFunction decorator to add telemetry tracking to commands. Parameters ---------- name : str The name to use for telemetry tracking. func : callable or None The function to track for telemetry. If ``None`` (default), returns a decorator that can be applied to a function. Examples -------- >>> @st.gather_metrics("my_command") ... def my_command(url): ... return url zgather_metrics: name is empty undefinedNrct|S)Nr)r)r%rws rwrapperzgather_metrics..wrappers! rc$ddlm}|}td}|duxr9|jxr+|j xrt |j tk}d}d}|r|r t g|i|}|j|jvs |j|jtkr|j j||jj|jgd|_d}  |i|} |r |rd|_ |r|rt%||z |_| S#t$r!}tj!d|Yd}~Yd}~wwxYw#t"$r|r|rt%||z |_wxYw#|r |rd|_wwwxYw)Nr) default_timerT)suppress_warningFz#Failed to collect command telemetryr)timeitrrgather_usage_statscommand_tracking_deactivatedrmtracked_commandsrrrwtracked_commands_counterrrupdaterrr rrtime) rrtimer exec_startctxtracking_activatedcommand_telemetry$has_set_command_tracking_deactivatedrresultrwnon_optional_funcs r wrapped_funcz$gather_metrics..wrapped_funcs1W  $7 tO $&& $444 $C(()#$ -1 05, % R$:%t%.2%6<%! &**#2N2NN334E4J4JK./((//0AB,,335F5K5K4LM3708<4 9&77F;380 "3%4UWz5I%J  " / R Cb QQ R  "&7)8:9M)N!&   ;380r)t*j@jC| j>} |jDjtG| |jHjtK|Scc} w)z2Create and return the full PageProfile ForwardMsg.zserver.headlessz:defaultN)&r page_profilecommandsextend exec_time prep_timer r*headless _server_mode server_moder_config_optionsis_manually_setr is_defaultrsysmodulesr"platformrrtznametimezone attributionsuncaught_exceptionrrlfragment_ids_this_runis_fragment_runmain_script_pathrrinstalled_skillsrinstalled_agentsr) rrrrmsgrconfig_options option_name config_option attributionrrrs rcreate_page_profile_messagers ,C##L  *&L&L"--.?@L#)#6#6   #uN !11 K))+6"22;?M   ,9,D,D;-x(+   ~. 2  #++ % L #,,'LO ,L$$\2*< 'G ""s"'+C,E,E'F $   ggooc&:&:;G!!(()A')JK!!(()A)CD J-sH?)rr"rJ str | None)rrrJ list[str])rrrJtuple[str, ...])rJr)rJrrL)r`objectrJr")rgCallable[..., Any]rJr")rnrrJr)rgrrJr)rgrrJr) rrrr"rrrrrJr)rfloatrJr)rwr"rgrrJrr=)rwr"rgrKrJzCallable[[F], F])rwr"rgzF | NonerJzCallable[[F], F] | F) rz list[Command]rrrrrrrJr)D __future__rrZr\rrrPrr#collections.abcrr functoolsrrtypingrr r r r rr rrrstreamlit.loggerrstreamlit.proto.ForwardMsg_pb2rstreamlit.proto.PageProfile_pb2rr/streamlit.runtime.scriptrunner_utils.exceptionsr7streamlit.runtime.scriptrunner_utils.script_run_contextrrMrrRrrr&rrr!rrrrrrrrrr'r4r6rcriror}rrrrrrrrrrsf#  +&6688'5=JVH%% #u"#%%$15#.&3!/#!#(6$1 2&4DX:Ie&r!rh)2!!+* E$&5%* : 1!!H 3 1  .<''@()  3,0,6.%6.696.BE6.QT6. 6.r$  CxS)*            k#d&* ::::# :  :r