!iBSrSSKJr SSKrSSKJrJrJr SSKJ r SSK J r SSK J r SSK J r "S S \5r"S S \R 5r"S S\\ R$5r"SS\\ R(5rSSjrSSjrSSjrSSjrSSjrg)z2Utilities for working with post-processing tokens.) annotationsN)AnyCallableSequence)py_utils) llm_function)llmfn_output_row)llmfn_post_processc\rSrSrSrSrg)PostProcessParseErrorz,An error parsing the post-processing tokens.N)__name__ __module__ __qualname____firstlineno____doc____static_attributes__ri/home/james-whalen/.local/lib/python3.13/site-packages/google/generativeai/notebook/post_process_utils.pyr r s6rr ch\rSrSrSr\R SSj5r\R SSj5rSr g) ParsedPostProcessExprz:A post-processing expression parsed from the command line.cg)z$Returns the name of this expression.Nrselfs rnameParsedPostProcessExpr.name"rcg)zEAdds this parsed expression to `llm_fn` as a post-processing command.Nrrllm_fns radd_to_llm_function)ParsedPostProcessExpr.add_to_llm_function&rrrNreturnstrr"llm_function.LLMFunctionr&r)) rrrrrabcabstractmethodrr#rrrrrrs9D33 TTrrc@\rSrSrSrS SjrS SjrS SjrS SjrSr g) _ParsedPostProcessAddExpr+zEAn expression that returns the value of a new column to add to a row.cXlX lg)zConstructor. Args: name: The name of the expression. The name of the new column will be derived from this. fn: A function that takes the result of a row and returns a new value to add as a new column in the row. N_name_fnrrfns r__init__"_ParsedPostProcessAddExpr.__init__0s rcUR$Nr1rs rr_ParsedPostProcessAddExpr.name< zzrcjUVs/sH!o RUR55PM# sn$s snfr8r2 result_valuerrowsrows r__call__"_ParsedPostProcessAddExpr.__call__?*8<=))+,===(0c6URURUS9$Nrr4)add_post_process_add_fnr1r!s rr#-_ParsedPostProcessAddExpr.add_to_llm_functionBs--4::$-GGrr2r1N)rr'r4Callable[[str], Any]r%)r@-Sequence[llmfn_output_row.LLMFnOutputRowView]r&z Sequence[Any]r( rrrrrr5rrBr#rrrrr-r-+sP >Hrr-c@\rSrSrSrS SjrS SjrS SjrS SjrSr g) _ParsedPostProcessReplaceExprFz:An expression that returns the new result value for a row.cXlX lg)zConstructor. Args: name: The name of the expression. fn: A function that takes the result of a row and returns the new result. Nr0r3s rr5&_ParsedPostProcessReplaceExpr.__init__Ks rcUR$r8r9rs rr"_ParsedPostProcessReplaceExpr.nameUr;rcjUVs/sH!o RUR55PM# sn$s snfr8r=r?s rrB&_ParsedPostProcessReplaceExpr.__call__XrDrEc6URURUS9$rG)add_post_process_replace_fnr1r!s rr#1_ParsedPostProcessReplaceExpr.add_to_llm_function[s11tzzd1KKrrKN)rr'r4Callable[[str], str]r%)r@rMr& Sequence[str]r(rNrrrrPrPFsE>LrrPc*[URUS9$rG)r-rr4s rpost_process_add_fnr_`s $"++" ==rc*[URUS9$rG)rPrr^s rpost_process_replace_fnrads (bkkb AArc\U(d [S5e[U5S:a [S5eg)Nz,Cannot have empty post-processing expressionz3Post-processing expression should be a single token)r len)tokenss r'validate_one_post_processing_expressionrfhs/ #$RSS 6{Q#$YZZrc >[U5 USRS5n[R"5n[ U5HHup4[ U5nXE;a/[ SRSRUSUS-555eXTnMJ SRU5U4$)z)Returns name and the resolved expression.r.zUnable to resolve "{}"Nrc ) rfsplitrget_main_module enumeratevarsr formatjoin)re token_partscurrent_modulepart_numpartcurrent_module_varss r'_resolve_one_post_processing_expressionruqs,F3)//#&K--/N#K0">2  *'(//^xRS|9T0UV -21 HHV n --rc/nUHn[U5up4[U[5(aURU5 M8[U[5(aUR[ X4S95 Mg[ SRU55e U$)aResolves post-processing tokens into ParsedPostProcessExprs. E.g. Given [["add_length"], ["to_upper"]] as input, this function will return a sequence of ParsedPostProcessExprs that will execute add_length() and to_upper() on each entry of the LLM output as post-processing operations. Raises: PostProcessParseError: An error parsing or resolving the tokens. Args: tokens: A sequence of post-processing tokens after splitting. Returns: A sequence of ParsedPostProcessExprs. rHz{} is not callable)ru isinstancerappendrr-r rn)reresults expression expr_name expr_values rresolve_post_processing_tokensr}sw$,.G G S j"7 8 8 NN: &  H - - NN4)S T'(<(C(CI(NO O Nr)r4rL)r4r[)rer\r&None)rer\r&ztuple[str, Any])rezSequence[Sequence[str]]r&zSequence[ParsedPostProcessExpr])r __future__rr*typingrrrgoogle.generativeai.notebookr google.generativeai.notebook.librr r RuntimeErrorr ABCrLLMFnPostProcessBatchAddFnr-LLMFnPostProcessBatchReplaceFnrPr_rarfrur}rrrrs9" **19=?7L7 TCGG TH-HHH6L-LLL4>B[ [ [. ..* #$r