o bi@svdZddlmZddlZddlmZmZmZeZdd d Z dddZ Gdddee efej dZ Gddde ZdS)zLLMFnOutputRow.) annotationsN)AnyIteratorMappingx type[Any]returnstrcCst|dr|jSt|S)N__name__)hasattrr r )rr k/home/kim/smarthome/.venv/lib/python3.10/site-packages/google/generativeai/notebook/lib/llmfn_output_row.py_get_name_of_types rvaluer result_typeNonecCs6|tkrdSt||stdt|tt|dS)Nz2Value of last entry must be of type "{}", got "{}")r isinstance ValueErrorformatrtype)rrr r r _validate_is_result_type s  rc@s`eZdZdZejdddZejddd Zejdd d ZejdddZ ejdddZ dS)LLMFnOutputRowViewz!Immutable view of LLMFnOutputRow.kr rboolcCdS)z)For expressions like: x in this_instance.Nr selfrr r r __contains__0zLLMFnOutputRowView.__contains__cCr)z)For expressions like: str(this_instance).Nr rr r r __str__4rzLLMFnOutputRowView.__str__rcCr)z.Returns the type enforced for the result cell.Nr rr r r r9rzLLMFnOutputRowView.result_typercCr)z!Get the value of the result cell.Nr rr r r result_value=rzLLMFnOutputRowView.result_valuecCr)zGet the key of the result cell.Nr rr r r result_keyArzLLMFnOutputRowView.result_keyNrr rrrr rrrr) r __module__ __qualname____doc__abcabstractmethodrr rr!r"r r r r r,s    r) metaclassc@seZdZdZd)ddZd*d d Zd+d dZd,ddZd-ddZd.ddZ d/ddZ d0ddZ d.ddZ d1d2d%d&Z d3d'd(Zd S)4LLMFnOutputRowaContainer 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. dataMapping[str, _CELLVALUETYPE]rrcCs@t||_|js td||_t|jd}t||jdS)a*Constructor. 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__r Iterator[str]cC |jSN)r2__iter__rr r r r:l zLLMFnOutputRow.__iter__intcCr8r9)r2__len__rr r r r=or;zLLMFnOutputRow.__len__rr _CELLVALUETYPEcC |j|Sr9)r2 __getitem__rr r r r@r zLLMFnOutputRow.__getitem__rcCr?r9)r2rrr r r rvrAzLLMFnOutputRow.__contains__cCsd|jS)NzLLMFnOutputRow: {})rr2r rr r r r yszLLMFnOutputRow.__str__cCs|jSr9)r3rr r r r|szLLMFnOutputRow.result_typercCs|j|Sr9)r2r"rr r r r!szLLMFnOutputRow.result_valuecCst|jdS)Nr0)r4r2keysrr r r r"szLLMFnOutputRow.result_keyNrkey str | NonercCsHt||j|}|dus||kr||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)rr3r"r2)rrrC current_keyr r r set_result_values  zLLMFnOutputRow.set_result_valuecCsj||jvrd}|}||jvrd||}|d}||jvs|}|}|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)r2rr"pop)rrCridxZ candidate_keyr"r!r r r adds      zLLMFnOutputRow.add)r.r/rr)rr7)rr<)rr rr>r#r$r%r&r9)rrrCrDrr)rCr rr>rr)r r'r(r)r6r:r=r@rr rr!r"rFrJr r r r r-Fs          r-)rrrr )rrrrrr)r) __future__rr*typingrrrr>rrr ABCMetarr-r r r r s