o «©Zh ã@sLdZddlZddlmZmZmZmZddlmZgd¢Z Gdd„deƒZ dS)z4Callback Handler streams to stdout on new llm token.éN)ÚAnyÚDictÚListÚOptional)ÚStreamingStdOutCallbackHandler)ÚFinalZAnswerú:c s eZdZdZdeddfdd„Zdefdd„Zdd d d œd ee ed ededdf‡fdd„Z de ee fde ede ddfdd„Z dede ddfdd„Z‡ZS)Ú#FinalStreamingStdOutCallbackHandlerz¦Callback handler for streaming in agents. Only works with agents using LLMs that support streaming. Only the final output of the agent will be streamed. ÚtokenÚreturnNcCsP|j |¡|j | ¡¡t|jƒt|jƒkr&|j d¡|j d¡dSdS)Nr)Ú last_tokensÚappendÚlast_tokens_strippedÚstripÚlenÚanswer_prefix_tokensÚpop)Úselfr ©rúf/var/www/html/lang_env/lib/python3.10/site-packages/langchain/callbacks/streaming_stdout_final_only.pyÚappend_to_last_tokenss  þz9FinalStreamingStdOutCallbackHandler.append_to_last_tokenscCs|jr |j|jkS|j|jkS)N)Ú strip_tokensrÚanswer_prefix_tokens_strippedr r)rrrrÚcheck_if_answer_reacheds  z;FinalStreamingStdOutCallbackHandler.check_if_answer_reachedTF)rrÚ stream_prefixrrrcsztƒ ¡|dur t|_n||_|rdd„|jDƒ|_n|j|_dgt|jƒ|_dgt|jƒ|_||_||_ d|_ dS)aÊInstantiate FinalStreamingStdOutCallbackHandler. Args: answer_prefix_tokens: Token sequence that prefixes the answer. Default is ["Final", "Answer", ":"] strip_tokens: Ignore white spaces and new lines when comparing answer_prefix_tokens to last tokens? (to determine if answer has been reached) stream_prefix: Should answer prefix itself also be streamed? NcSsg|]}| ¡‘qSr)r)Ú.0r rrrÚ 6sÿz@FinalStreamingStdOutCallbackHandler.__init__..ÚF) ÚsuperÚ__init__ÚDEFAULT_ANSWER_PREFIX_TOKENSrrrr rrrÚanswer_reached)rrrr©Ú __class__rrrs  ÿ z,FinalStreamingStdOutCallbackHandler.__init__Ú serializedÚpromptsÚkwargscKs d|_dS)zRun when LLM starts running.FN)r!)rr$r%r&rrrÚ on_llm_startAs z0FinalStreamingStdOutCallbackHandler.on_llm_startcKsh| |¡| ¡r"d|_|jr |jD]}tj |¡qtj ¡dS|jr2tj |¡tj ¡dSdS)z?Run on new LLM token. Only available when streaming is enabled.TN) rrr!rr ÚsysÚstdoutÚwriteÚflush)rr r&ÚtrrrÚon_llm_new_tokenGs    þz4FinalStreamingStdOutCallbackHandler.on_llm_new_token)Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚstrrÚboolrrrrrrr'r-Ú __classcell__rrr"rr s4 û ýüûú" ÿÿÿ þr ) r1r(ÚtypingrrrrZlangchain_core.callbacksrr r rrrrÚs