mxiT!dZddlmZddlZddlZddlmZmZmZm Z m Z m Z m Z ddl mZddl mZddlZGddZej&Gd d Zdd ZGd d e eZGddej.ZGddeZy)zOutput of LLMFunction.) annotationsN)overloadAnyCallableIterableIteratorMappingSequence)llmfn_output_row)modelc(eZdZdZdZdZdZdZdZdZ y) 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_RESULTe/home/htdocs/ttos/venv/lib/python3.12/site-packages/google/generativeai/notebook/lib/llmfn_outputs.pyrr#s%=JIJ KKKrrcdeZdZUdZded<ded<ded<ded<d Zd ed <d Zd ed <d Zd ed<y )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> ON""::FJ"K"37M07rrc.|D]}|jyy)NTF)r&)outputsentrys r_has_model_input_fieldr,Ms%    ( rczeZdZdZ d ddZddZddZeddZeddZddZdd Z dd Z dd Z dd Z y)LLMFnOutputsBasezParent class for LLMFnOutputs. This class exists mainly to avoid a circular dependency between LLMFnOutputs and LLMFnOutputsSink. Most users should use LLMFnOutputs directly instead. Nc8|t||_yg|_y)zcConstructor. Args: outputs: The contents of this LLMFnOutputs instance. N)list_outputsselfr*s r__init__zLLMFnOutputsBase.__init__[sBIATW  Z\ rc6|jjSN)r1__iter__r3s rr7zLLMFnOutputsBase.__iter__gs}}%%''rc6|jjSr6)r1__len__r8s rr:zLLMFnOutputsBase.__len__ks}}$$&&rcyr6rr3xs r __getitem__zLLMFnOutputsBase.__getitem__os7:rcyr6rr<s rr>zLLMFnOutputsBase.__getitem__rsCFrc8|jj|Sr6)r1r>r<s rr>zLLMFnOutputsBase.__getitem__us}}((++rc,t|jSr6)boolr1r8s r__bool__zLLMFnOutputsBase.__bool__ysDMM""rc>|jjSr6)as_pandas_dataframe__str__r8s rrFzLLMFnOutputsBase.__str__|s'')1133rctjgtjgtjgi}t |j rg|tj <|j s|S|jDcic]}|d}}|j D]9}|jD](}t|jddD]}d||< *;|j D]2}|jD]!}d|t|jd<#4|D]}g||< d}|j D]}|jD]}|tjj|j|tjj|jtj |vr,|tj j|j|jD]\}}||j||jD]"} t!| |ks| jd$|dz }|Scc}w)z'Formats returned results as dictionary.N)rrrrr,r1rkeysr$r0appendr!r"r&itemsvalueslen) r3dataktotal_keys_setoutputresultkey next_num_rowsvaluecolumns ras_dictzLLMFnOutputsBase.as_dicts7  " "B  ! !2  " "B   "$-- 0,.D(( )}}K=AIIK*Hq1d7*H*Hmm /F ,, / .s3/C*.N3'/ / /mm ?F ,, ?:>tFKKM2267 ? ? " CDI  mm #F ,, #[++,33F4E4EF[**+2263C3CD**d2001889K9KL"(,,.,JCI$$U+,#kkm,F6{]2 d+,"  # #" A+Is; IcHtj|jSr6)pandas DataFramerXr8s rrEz$LLMFnOutputsBase.as_pandas_dataframes //rr6)r*!Iterable[LLMFnOutputEntry] | None)returnzIterator[LLMFnOutputEntry])r]r )r=r r]r)r=slicer]zSequence[LLMFnOutputEntry])r=z int | slicer]z-LLMFnOutputEntry | Sequence[LLMFnOutputEntry])r]rB)r]str)r]zMapping[str, Sequence[Any]])r]zpandas.DataFrame) rrrrr4r7r:rr>rCrFrXrErrrr.r.Ts]6: ]2 ](':: FF,#4:x0rr.ceZdZdZddZy)LLMFnOutputsSinkzAbstract class representing an exporter for the output of LLMFunction. This class could be extended to write to external documents, such as Google Sheets. cy)z%Writes `outputs` to some destination.Nrr2s r write_outputszLLMFnOutputsSink.write_outputssrN)r*r.r]None)rrrrrcrrrraras  4rrac>eZdZdZ d dfd ZdZddZxZS) 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. cht|||r||_t|dt |dyy)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)superr4_ipython_display_fnsetattrgetattr)r3r*ipython_display_fn __class__s rr4zLLMFnOutputs.__init__s@ ) '9D $ #56  rc&|j|y)zActual implementation of _ipython_display_. This method should only be used invoked if self._ipython_display_fn is set. N)rkr8s rriz"LLMFnOutputs._ipython_display_impls   &rc&|j|y)zExport contents to `sink`.N)rc)r3sinks rexportzLLMFnOutputs.exports 4 r)NN)r*r\rnz%Callable[[LLMFnOutputs], None] | None)rrrar]rd)rrrrr4rirs __classcell__)ros@rrfrfs26:DH2B8'!rrf)r*zIterable[LLMFnOutputEntry])r __future__rabc dataclassestypingrrrrrr r google.generativeai.notebook.libr r model_librZr dataclassrr,r.ABCrarfrrrr}s" >?     8884i0x 01i0X4sww4/!#/!r