}h RdZddlZddlmZmZmZmZddlmZgdZ GddeZ y)z4Callback Handler streams to stdout on new llm token.N)AnyDictListOptional)StreamingStdOutCallbackHandler)FinalAnswer:c eZdZdZdeddfdZdefdZdddd d ee ed ed eddffd Z de ee fde ede ddfdZ dede ddfdZxZS)#FinalStreamingStdOutCallbackHandlerzCallback handler for streaming in agents. Only works with agents using LLMs that support streaming. Only the final output of the agent will be streamed. tokenreturnNcP|jj||jj|jt |jt |j kDr7|jj d|jj dyy)Nr) last_tokensappendlast_tokens_strippedstriplenanswer_prefix_tokenspop)selfr s i/var/www/html/engine/venv/lib/python3.12/site-packages/langchain/callbacks/streaming_stdout_final_only.pyappend_to_last_tokensz9FinalStreamingStdOutCallbackHandler.append_to_last_tokenss| & !!((7 t 3t'@'@#A A     #  % % ) )! , Bc~|jr|j|jk(S|j|jk(S)N) strip_tokensranswer_prefix_tokens_strippedrr)rs rcheck_if_answer_reachedz;FinalStreamingStdOutCallbackHandler.check_if_answer_reacheds9   ,,0R0RR R##t'@'@@ @rTF)rr stream_prefixrrrczt|| t|_n||_|r.|jDcgc]}|j c}|_n|j|_dgt |jz|_dgt |jz|_||_ ||_ d|_ ycc}w)aInstantiate 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? NF) super__init__DEFAULT_ANSWER_PREFIX_TOKENSrrrrrrrranswer_reached)rrrrr __class__s rr#z,FinalStreamingStdOutCallbackHandler.__init__s"   '(DD %(t j j|t j jyy)z?Run on new LLM token. Only available when streaming is enabled.TN) rrr%rrsysstdoutwriteflush)rr r)ts ron_llm_new_tokenz4FinalStreamingStdOutCallbackHandler.on_llm_new_tokenGs ""5)  ' ' )"&D !!))(AJJ$$Q'(   "     JJ  U # JJ    r)__name__ __module__ __qualname____doc__strrboolrrrr#rrr+r2 __classcell__)r&s@rr r s -3-4-AA59!# $'tCy1 $ $  $  $D$sCx.$379$HK$ $ cSTrr ) r6r-typingrrrrlangchain_core.callbacksrr$r rrr=s): ,,C7N*HNr