3fi=dZddlmZddlmZddlmZmZmZm Z m Z m Z ddl m Z mZddlmZddlmZddlmZerdd lmZdd Zd Zd Zd ZdZGddeZGdde ZGddZGddZGddeZ y)z*Callback Handler that prints to streamlit.) annotations)Enum) TYPE_CHECKINGAnyDictList NamedTupleOptional) AgentAction AgentFinish)BaseCallbackHandler) LLMResult)MutableExpander)DeltaGeneratorc&|jddS)zZConvert newline characters to markdown newline sequences (space, space, newline).  z )replace)texts /var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_community/callbacks/streamlit/streamlit_callback_handler.py_convert_newlinesrs <<f %%u✅z:thinking_face:z:books:u⚠️ceZdZdZdZdZdZy)LLMThoughtStatez#Enumerator of the LLMThought state.THINKING RUNNING_TOOLCOMPLETEN)__name__ __module__ __qualname____doc__rrrrrrrs-H!LHrrc&eZdZUdZded<ded<y) ToolRecordzTool record as a NamedTuple.strname input_strN)rrrr __annotations__r!rrr#r#*s& INrr#cXeZdZdZeddZeddZeddZeddZy) LLMThoughtLabelerz Generates markdown labels for LLMThought containers. Pass a custom subclass of this to StreamlitCallbackHandler to override its default labeling logic. ctdS)ziReturn the markdown label for a new LLMThought that doesn't have an associated tool yet. z **Thinking...**)THINKING_EMOJIr!rrget_initial_labelz#LLMThoughtLabeler.get_initial_label8s !!!122rc|j}|j}|rtnt}|dk(rt}d}t dt |g}|d|}t |j|kDr|dz}|jdd}|d|d |}|S) a}Return the label for an LLMThought that has an associated tool. Parameters ---------- tool The tool's ToolRecord is_complete True if the thought is complete; False if the thought is still receiving input. Returns ------- The markdown label for the thought's container. _Exceptionz Parsing error<rz...r z **z:** )r&r%CHECKMARK_EMOJIr+EXCEPTION_EMOJIminlenr)tool is_completeinputr%emojiidxlabels rget_tool_labelz LLMThoughtLabeler.get_tool_label?s&yy#.N < #E"D2s5z"#a  t~~  $EME dC('TF$ug. rctdS)zmReturn a markdown label for the special 'history' container that contains overflow thoughts. z **History**) HISTORY_EMOJIr!rrget_history_labelz#LLMThoughtLabeler.get_history_label`s   --rctdS)zReturn the markdown label for the agent's final thought - the "Now I have the answer" thought, that doesn't involve a tool. z **Complete!**)r1r!rrget_final_agent_thought_labelz/LLMThoughtLabeler.get_final_agent_thought_labelgs "".11rN)returnr$)r5r#r6boolrAr$) rrrr staticmethodr,r;r>r@r!rrr)r)1sX 33 @.. 22rr)ceZdZdZ ddZeddZeddZddZddZ ddZ ddZ dd Z dd Z d dd Zdd Z d ddZdddZddZy ) LLMThoughtz&A thought in the LLM's thought stream.ct||j||_tj|_d|_d|_d|_||_ ||_ y)a@Initialize the LLMThought. Args: parent_container: The container we're writing into. labeler: The labeler to use for this thought. expanded: Whether the thought should be expanded by default. collapse_on_complete: Whether the thought should be collapsed. )parent_containerr:expandedN) rr, _containerrr_state_llm_token_stream_llm_token_writer_idx _last_tool_collapse_on_complete_labeler)selfrGlabelerrHcollapse_on_completes r__init__zLLMThought.__init__ssW*-++-  &.. !#48"04%9" rc|jS)z!The container we're writing into.)rJrQs r containerzLLMThought.containerrc|jS)z&The last tool executed by this thought)rNrVs r last_toolzLLMThought.last_toolrXrc d|_d|_yNrI)rLrMrVs r_reset_llm_token_streamz"LLMThought._reset_llm_token_streams!#%)"rc$|jyNr])rQ serializedpromptss r on_llm_startzLLMThought.on_llm_starts $$&rc |xjt|z c_|jj|j|j|_y)N)index)rLrrJmarkdownrMrQtokenkwargss ron_llm_new_tokenzLLMThought.on_llm_new_tokensF "3E"::%)__%=%=  " "$*D*D&>& "rc $|jyr_r`rQresponseris r on_llm_endzLLMThought.on_llm_ends $$&rc p|jjd|jj|y)Nz**LLM encountered an error...**rJrf exceptionrQerrorris r on_llm_errorzLLMThought.on_llm_errors'   !BC !!%(rc tj|_|d}t|||_|j j |jj|jdy)Nr%)r%r&Fr6 new_label) rrrKr#rNrJupdaterPr;)rQrar&ri tool_names r on_tool_startzLLMThought.on_tool_starts\ &22 v& $)yI mm224??PU2V  rNc T|jjdt|dy)Nz**)rJrfr$rQoutputcolorobservation_prefix llm_prefixris r on_tool_endzLLMThought.on_tool_ends#   2c&k]"!56rc p|jjd|jj|y)Nz **Tool encountered an error...**rprrs r on_tool_errorzLLMThought.on_tool_errors'   !CD !!%(rc yr_r!rQactionrris ron_agent_actionzLLMThought.on_agent_action rcl|W|jtjk(r:|jJd|jj |jd}tj |_|jr|jj|dy|jj|y)zFinish the thought.Nz;_last_tool should never be null when _state == RUNNING_TOOLTrvF)rx new_expandedrw) rKrrrNrPr;rrOrJry)rQ final_labels rcompletezLLMThought.completes  4;;/2N2N#N??. M .--66T7K&..  % % OO " "[u " M OO " "[ " 9rc8|jjy)zFRemove the thought from the screen. A cleared thought can't be reused.N)rJclearrVs rrzLLMThought.clears r)rGrrRr)rHrBrSrB)rAr)rAzOptional[ToolRecord]rANone)raDict[str, Any]rb List[str]rArrhr$rirrArrmrrirrArrs BaseExceptionrirrArrarr&r$rirrArNNN r~rr Optional[str]rrrrrirrArr_rr rrrirrArrrrAr)rrrr rTpropertyrWrZr]rcrjrnrtr{rrrrrr!rrrErEps0 ( #   #  6*' ' )  (  58  DG     $,0$( 777* 7 " 7  7 7) ;? ! *7 JM   : rrEc\eZdZdZddddd ddZddZddZedd Zddd Z dd Z d d Z d!d Z d"dZ d#dZ d$dZ d% d&dZd#dZ d' d(dZ d)dZd*dZd#dZ d d+dZ d d,dZy)-StreamlitCallbackHandlerz0Callback handler that writes to a Streamlit app.TN)max_thought_containersexpand_new_thoughtscollapse_completed_thoughtsthought_labelerc||_|j|_d|_d|_g|_t |d|_||_||_ |xs t|_ y)aCreate a StreamlitCallbackHandler instance. Parameters ---------- parent_container The `st.container` that will contain all the Streamlit elements that the Handler creates. max_thought_containers The max number of completed LLM thought containers to show at once. When this threshold is reached, a new thought will cause the oldest thoughts to be collapsed into a "History" expander. Defaults to 4. expand_new_thoughts Each LLM "thought" gets its own `st.expander`. This param controls whether that expander is expanded by default. Defaults to True. collapse_completed_thoughts If True, LLM thought expanders will be collapsed when completed. Defaults to True. thought_labeler An optional custom LLMThoughtLabeler instance. If unspecified, the handler will use the default thought labeling logic. Defaults to None. N) _parent_containerrW_history_parent_history_container_current_thought_completed_thoughtsmax_max_thought_containers_expand_new_thoughts_collapse_completed_thoughtsr)_thought_labeler)rQrGrrrrs rrTz!StreamlitCallbackHandler.__init__sf<"2/99;=A6:57 '*+A1'E$$7!,G) / F3D3FrcH|j td|jS)z]Return our current LLMThought. Raise an error if we have no current thought. z(Current LLMThought is unexpectedly None!)r RuntimeErrorrVs r_require_current_thoughtz1StreamlitCallbackHandler._require_current_thoughts(  (IJ J$$$rc~t|jdkDr%|jt|jdz Sy)zJReturn our most recent completed LLMThought, or None if we don't have one.rrN)r4rrVs r_get_last_completed_thoughtz4StreamlitCallbackHandler._get_last_completed_thoughts9 t'' (1 ,++C0H0H,IA,MN Nrctt|j}|j|dz }|j|dz }|S)zThe number of 'thought containers' we're currently showing: the number of completed thought containers, the history container (if it exists), and the current thought container (if it exists). r)r4rrr)rQcounts r_num_thought_containersz0StreamlitCallbackHandler._num_thought_containers sC D,,-  " " . QJE  , QJE rc|j}|j||jj|d|_y)z{Complete the current thought, optionally assigning it a new label. Add it to our _completed_thoughts list. N)rrrappendr)rQrthoughts r_complete_current_thoughtz2StreamlitCallbackHandler._complete_current_thought-s<//1%   ''0 $rc|j|jkDr*t|jdkDr|jD|jdkDr5t |j |jjd|_|jjd}|jT|jj|jj|jj|j|j|j|jkDrt|jdkDryyyy)zgIf we have too many thoughts onscreen, move older thoughts to the 'history container.' rNrF)r:rH)rrr4rrrrrr>poprfrWr: append_copyr)rQoldest_thoughts r_prune_old_thought_containersz6StreamlitCallbackHandler._prune_old_thought_containers6s  ( (4+G+G GD,,-1 &&.43O3ORS3S*9((//AAC"+' "5599!rs0"GG:8,T8&" d<2<2~t t n|)2|)r