; iT!*dZddlmZddlZddlZddlmZmZmZm Z m Z m Z m Z ddl mZddl mZddlZGddZejGd d Zdd ZGdde eZGddejZGddeZdS)zOutput of LLMFunction.) annotationsN)overloadAnyCallableIterableIteratorMappingSequence)llmfn_output_row)modelc*eZdZdZdZdZdZdZdZdZ dS) ColumnNamesz3Names of columns that are used to represent output.z Prompt Numz Input Numz Result NumPromptz Prompt vars text_resultN) __name__ __module__ __qualname____doc__ PROMPT_NUM INPUT_NUM RESULT_NUM MODEL_INPUT PROMPT_VARS TEXT_RESULT/var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/google/generativeai/notebook/lib/llmfn_outputs.pyrr#s4==JIJ KKKKKrrcfeZdZUdZded<ded<ded<ded<d Zd ed <d Zd ed <d Zd ed<d S)LLMFnOutputEntryaThe output of a single model input from LLMFunction. A model input is a prompt where the keyword placeholders have been substituted (by `prompt_vars`). E.g. If we have: prompt: "the opposite of {word} is" prompt_vars: {"word", "hot"} Then we will have the following model input: model_input: "the opposite of hot is" Note: The model may produce one-or-more results for a given model_input. This is represented by the sequence `output_rows`. int prompt_num input_numzMapping[str, str] prompt_varsz)Sequence[llmfn_output_row.LLMFnOutputRow] output_rowsNz str | Noneprompt model_inputzmodel_lib.ModelResults | None model_results)rrrr__annotations__r%r&r'rrrrr2s{  OOONNN""""::::F"K""""37M777777rroutputsIterable[LLMFnOutputEntry]c$|D] }|jdS dS)NTF)r&)r)entrys r_has_model_input_fieldr-Ms+   (44 ) 5rceZdZdZ dddZddZdd Zed d Zed!dZd"dZd#dZ d$dZ d%dZ d&dZ dS)'LLMFnOutputsBasezParent class for LLMFnOutputs. This class exists mainly to avoid a circular dependency between LLMFnOutputs and LLMFnOutputsSink. Most users should use LLMFnOutputs directly instead. Nr)!Iterable[LLMFnOutputEntry] | Nonec6|t|ng|_dS)zcConstructor. Args: outputs: The contents of this LLMFnOutputs instance. N)list_outputsselfr)s r__init__zLLMFnOutputsBase.__init__[s!BIATW Z\ rreturnIterator[LLMFnOutputEntry]c4|jSN)r3__iter__r5s rr;zLLMFnOutputsBase.__iter__gs}%%'''rr c4|jSr:)r3__len__r<s rr>zLLMFnOutputsBase.__len__ks}$$&&&rxrcdSr:rr5r?s r __getitem__zLLMFnOutputsBase.__getitem__os7:srsliceSequence[LLMFnOutputEntry]cdSr:rrAs rrBzLLMFnOutputsBase.__getitem__rsCF3r int | slice-LLMFnOutputEntry | Sequence[LLMFnOutputEntry]c6|j|Sr:)r3rBrAs rrBzLLMFnOutputsBase.__getitem__us}((+++rboolc*t|jSr:)rIr3r<s r__bool__zLLMFnOutputsBase.__bool__ysDM"""rstrcN|Sr:)as_pandas_dataframe__str__r<s rrOzLLMFnOutputsBase.__str__|s ''))11333rMapping[str, Sequence[Any]]c>tjgtjgtjgi}t |jrg|tj<|js|Sd|D}|jD]=}|jD]3}t|ddD]}d||<4>|jD]6}|jD],}d|t|d<-7|D]}g||<d}|jD]}|jD]}|tj |j |tj |j tj|vr*|tj |j |D] \}}|| |!|D]*}t!||kr| d+|dz } |S)z'Formats returned results as dictionary.ci|]}|dSr:r).0ks r z,LLMFnOutputsBase.as_dict..s*H*H*Hq1d*H*H*HrN)rrrrr-r3rkeysr$r2appendr!r"r&itemsvalueslen) r5datatotal_keys_setoutputresultkey next_num_rowsvaluecolumns ras_dictzLLMFnOutputsBase.as_dicts[  "B  !2  "B   "$- 0 0 /,.D( )} K+I*HDIIKK*H*H*Hm / /F , / / ..ss3//C*.N3''/ /m ? ?F , ? ?:>tFKKMM222677 ? "  CDII m # #F , # #[+,33F4EFFF[*+2263CDDD*d2201889KLLL"(,,..,,JCI$$U++++#kkmm,,F6{{]22 d+++"  #  rpandas.DataFramecNtj|Sr:)pandas DataFramerer<s rrNz$LLMFnOutputsBase.as_pandas_dataframes ///rr:)r)r0)r7r8)r7r )r?r r7r)r?rCr7rD)r?rFr7rG)r7rI)r7rL)r7rP)r7rf) rrrrr6r;r>rrBrKrOrerNrrrr/r/Ts6: ] ] ] ] ](((('''':::X: FFFXF,,,,####4444::::x000000rr/ceZdZdZddZdS) LLMFnOutputsSinkzAbstract class representing an exporter for the output of LLMFunction. This class could be extended to write to external documents, such as Google Sheets. r)r/r7NonecdS)z%Writes `outputs` to some destination.Nrr4s r write_outputszLLMFnOutputsSink.write_outputssrN)r)r/r7rl)rrrrrnrrrrkrks2 444444rrkc8eZdZdZ ddfd ZdZdd ZxZS) LLMFnOutputsaA sequence of LLMFnOutputEntry instances. Notes: - Each LLMFnOutputEntry represents the results of running one model input (see documentation for LLMFnOutputEntry for what "model input" means.) - A single model input may produce more-than-one text results. Nr)r0ipython_display_fn%Callable[[LLMFnOutputs], None] | Nonect||r(||_t|dt |ddSdS)aConstructor. Args: outputs: The contents of this LLMFnOutputs instance. ipython_display_fn: An optional function for pretty-printing this instance when it is the output of a cell in a notebook. If this argument is not None, the _ipython_display_ method will be defined which will in turn invoke this function. )r)_ipython_display__ipython_display_implN)superr6_ipython_display_fnsetattrgetattr)r5r)rq __class__s rr6zLLMFnOutputs.__init__sl )))  '9D $ #566       rc0||dS)zActual implementation of _ipython_display_. This method should only be used invoked if self._ipython_display_fn is set. N)rwr<s rruz"LLMFnOutputs._ipython_display_impls   &&&&&rsinkrkr7rlc0||dS)zExport contents to `sink`.N)rn)r5r|s rexportzLLMFnOutputs.exports 4     r)NN)r)r0rqrr)r|rkr7rl)rrrrr6rur~ __classcell__)rzs@rrprpsw6:DH8'''!!!!!!!!rrp)r)r*)r __future__rabc dataclassestypingrrrrrr r google.generativeai.notebook.libr r model_librhr dataclassrr-r/ABCrkrprrrrs"""""" >=====??????           888888884i0i0i0i0i0x 01i0i0i0X44444sw444/!/!/!/!/!#/!/!/!/!/!r