@'h VddlmZddlmZmZddlmZddlmZgdZ GddeZ y) ) annotations)AnyOptional) LLMResult)AsyncIteratorCallbackHandler)FinalAnswer:cteZdZdZd dZd dZdddd dfdZ dd Zdd Zdd Z xZ S)!AsyncFinalIteratorCallbackHandlerznCallback handler that returns an async iterator. Only the final output of the agent will be iterated. cP|jj||jj|jt |jt |j kDr7|jj d|jj dyy)Nr) last_tokensappendlast_tokens_strippedstriplenanswer_prefix_tokenspop)selftokens u/home/kushmeetdev/Regenta/Chatbot/venv/lib/python3.12/site-packages/langchain/callbacks/streaming_aiter_final_only.pyappend_to_last_tokensz7AsyncFinalIteratorCallbackHandler.append_to_last_tokenss| & !!((7 t 3t'@'@#A A     #  % % ) )! , Bc~|jr|j|jk(S|j|jk(SN) strip_tokensranswer_prefix_tokens_strippedrr)rs rcheck_if_answer_reachedz9AsyncFinalIteratorCallbackHandler.check_if_answer_reacheds9   ,,0R0RR R##t'@'@@ @rNTF)rr stream_prefixczt|| t|_n||_|r.|jDcgc]}|j c}|_n|j|_dgt |jz|_dgt |jz|_||_ ||_ d|_ ycc}w)aInstantiate AsyncFinalIteratorCallbackHandler. 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*AsyncFinalIteratorCallbackHandler.__init__s"   '(DD %(rEs&" ,L7M)(DM)r