U h@sdZddlmZddlZddlmZmZmZeZddddd Z d dd d d dZ Gdddee efej dZ Gddde ZdS)zLLMFnOutputRow.) annotationsN)AnyIteratorMapping type[Any]str)xreturncCst|dr|jSt|S)N__name__)hasattrr r)rr U/tmp/pip-unpacked-wheel-doshhd5e/google/generativeai/notebook/lib/llmfn_output_row.py_get_name_of_types rrNone)value result_typer cCs6|tkr dSt||s2tdt|tt|dS)Nz2Value of last entry must be of type "{}", got "{}")r isinstance ValueErrorformatrtype)rrr r r _validate_is_result_type s  rc@sveZdZdZejdddddZejdddd Zejd dd d Zejd dddZ ejddddZ dS)LLMFnOutputRowViewz!Immutable view of LLMFnOutputRow.rboolkr cCsdS)z)For expressions like: x in this_instance.Nr selfrr r r __contains__0szLLMFnOutputRowView.__contains__r cCsdS)z)For expressions like: str(this_instance).Nr rr r r __str__4szLLMFnOutputRowView.__str__rcCsdS)z.Returns the type enforced for the result cell.Nr rr r r r9szLLMFnOutputRowView.result_typercCsdS)z!Get the value of the result cell.Nr rr r r result_value=szLLMFnOutputRowView.result_valuecCsdS)zGet the key of the result cell.Nr rr r r result_keyAszLLMFnOutputRowView.result_keyN) r __module__ __qualname____doc__abcabstractmethodrr rr!r"r r r r r,sr) metaclassc@seZdZdZdddddZddd d Zd dd d ZdddddZdddddZddddZ ddddZ ddddZ ddddZ d(dd d!d"d#d$Z ddd!d%d&d'ZdS))LLMFnOutputRowaContainer that represents a single row in a table of outputs. We represent outputs as a table. This class represents a single row in the table like a dictionary, where the key is the column name and the value is the cell value. A single cell is designated the "result". This contains the output of the LLM model after running any post-processing functions specified by the user. In addition to behaving like a dictionary, this class provides additional methods, including: - Getting the value of the "result" cell - Setting the value (and optionally the key) of the "result" cell. - Add a new non-result cell Notes: As an implementation detail, the result-cell is always kept as the rightmost cell. zMapping[str, _CELLVALUETYPE]r)datarcCs@t||_|jstd||_t|jd}t||jdS)aConstructor. Args: data: The initial value of the row. The last entry will be treated as the result. Cannot be empty. The value of the last entry must be `str`. result_type: The type of the result cell. This will be enforced at runtime. zMust provide non-empty dataN)dict_datar _result_typelistvaluesr)rr*rr!r r r __init__Zs zLLMFnOutputRow.__init__z Iterator[str]rcCs |jSN)r-__iter__rr r r r3nszLLMFnOutputRow.__iter__intcCs |jSr2)r-__len__rr r r r5qszLLMFnOutputRow.__len__r_CELLVALUETYPErcCs |j|Sr2)r- __getitem__rr r r r7tszLLMFnOutputRow.__getitem__rcCs |j|Sr2)r-rrr r r rxszLLMFnOutputRow.__contains__cCsd|jS)NzLLMFnOutputRow: {})rr-r rr r r r {szLLMFnOutputRow.__str__cCs|jSr2)r.rr r r r~szLLMFnOutputRow.result_typercCs|j|Sr2)r-r"rr r r r!szLLMFnOutputRow.result_valuecCst|jdS)Nr+)r/r-keysrr r r r"szLLMFnOutputRow.result_keyNz str | Noner)rkeyr cCsHt||j|}|dks$||kr2||j|<dS|j|=||j|<dS)zSet the value of the result cell. Sets the value (and optionally the key) of the result cell. Args: value: The value to set the result cell today. key: Optionally change the key as well. N)rr.r"r-)rrr9 current_keyr r r set_result_values  zLLMFnOutputRow.set_result_value)r9rr cCsb||jkr6d}|}||jkr2d||}|d}q|}|}|j|}||j|<||j|<dS)zAdd a non-result cell. Adds a new non-result cell. This does not affect the result cell. Args: key: The key of the new cell to add. value: The value of the new cell to add. z{}_{}N)r-rr"pop)rr9ridxZ candidate_keyr"r!r r r adds      zLLMFnOutputRow.add)N)r r#r$r%r1r3r5r7rr rr!r"r;r?r r r r r)Fsr))r% __future__rr&typingrrrr6rrrABCMetarr)r r r r s