a aŠÝg ã@sLdZddlZddlmZmZmZmZddlmZgd¢Z Gdd„deƒZ dS)z4Callback Handler streams to stdout on new llm token.éN)ÚAnyÚDictÚListÚOptional)ÚStreamingStdOutCallbackHandler)ÚFinalZAnswerú:csŽeZdZdZeddœdd„Zedœdd„Zdd d d œee eeedd œ‡fd d„Z e ee fe ee ddœdd„Z ee ddœdd„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. N)ÚtokenÚreturncCsL|j |¡|j | ¡¡t|jƒt|jƒkrH|j d¡|j d¡dS)Nr)Ú last_tokensÚappendÚlast_tokens_strippedÚstripÚlenÚanswer_prefix_tokensÚpop)Úselfr ©rú}/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain/callbacks/streaming_stdout_final_only.pyÚappend_to_last_tokenss   z9FinalStreamingStdOutCallbackHandler.append_to_last_tokens)r cCs"|jr|j|jkS|j|jkSdS)N)Ú strip_tokensrÚanswer_prefix_tokens_strippedr r)rrrrÚcheck_if_answer_reacheds z;FinalStreamingStdOutCallbackHandler.check_if_answer_reachedTF)rrÚ stream_prefix)rrrr csztƒ ¡|durt|_n||_|r8dd„|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Ú 6sz@FinalStreamingStdOutCallbackHandler.__init__..ÚF) ÚsuperÚ__init__ÚDEFAULT_ANSWER_PREFIX_TOKENSrrrr rrrÚanswer_reached)rrrr©Ú __class__rrrs ÿ z,FinalStreamingStdOutCallbackHandler.__init__)Ú serializedÚpromptsÚkwargsr cKs d|_dS)zRun when LLM starts running.FN)r!)rr$r%r&rrrÚ on_llm_startAsz0FinalStreamingStdOutCallbackHandler.on_llm_start)r r&r cKsd| |¡| ¡rDd|_|jr@|jD]}tj |¡q$tj ¡dS|jr`tj |¡tj ¡dS)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 s  û ú#þ r ) r1r(ÚtypingrrrrZlangchain_core.callbacksrr r rrrrÚs