; idZddlmZddlZddlmZmZmZeZdd Z ddZ Gddee efej Z Gdde ZdS)zLLMFnOutputRow.) annotationsN)AnyIteratorMappingx type[Any]returnstrcNt|dr|jSt|S)N__name__)hasattrr r )rs /var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/google/generativeai/notebook/lib/llmfn_output_row.py_get_name_of_typers'q*z q66Mvaluer result_typeNonec |tkrdSt||sJtdt |t t |dS)Nz2Value of last entry must be of type "{}", got "{}")r isinstance ValueErrorformatrtype)rrs r_validate_is_result_typer soc e[ ) )  @ G G!+..!$u++..       rceZdZdZejddZejddZejdd Zejdd Z ejdd Z d S)LLMFnOutputRowViewz!Immutable view of LLMFnOutputRow.kr r boolcdS)z)For expressions like: x in this_instance.Nselfrs r __contains__zLLMFnOutputRowView.__contains__0rcdS)z)For expressions like: str(this_instance).Nrr!s r__str__zLLMFnOutputRowView.__str__4r#rrcdS)z.Returns the type enforced for the result cell.Nrr%s rrzLLMFnOutputRowView.result_type9r#rrcdS)z!Get the value of the result cell.Nrr%s r result_valuezLLMFnOutputRowView.result_value=r#rcdS)zGet the key of the result cell.Nrr%s r result_keyzLLMFnOutputRowView.result_keyAr#rNrr r rr r r rr r) r __module__ __qualname____doc__abcabstractmethodr"r&rr)r+rrrrr,s++ 8888 8888 ==== 0000 ......rr) metaclasscleZdZdZddZdd Zd d Zd!dZd"dZd#dZ d$dZ d%dZ d#dZ d&d'dZ d(dZdS))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. dataMapping[str, _CELLVALUETYPE]rrct||_|jstd||_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)r!r8rr)s r__init__zLLMFnOutputRow.__init__Zsk15T  z <:;; ;'DJ--//004  t/@AAAAArr Iterator[str]c4|jSN)r=__iter__r%s rrEzLLMFnOutputRow.__iter__lsz""$$$rintc4|jSrD)r=__len__r%s rrHzLLMFnOutputRow.__len__osz!!###rrr _CELLVALUETYPEc6|j|SrD)r= __getitem__r s rrKzLLMFnOutputRow.__getitem__rsz%%a(((rrc6|j|SrD)r=r"r s rr"zLLMFnOutputRow.__contains__vsz&&q)))rcZd|jS)NzLLMFnOutputRow: {})rr=r&r%s rr&zLLMFnOutputRow.__str__ys$#**4:+=+=+?+?@@@rc|jSrD)r>r%s rrzLLMFnOutputRow.result_type|s   rrc@|j|SrD)r=r+r%s rr)zLLMFnOutputRow.result_valuesz$//++,,rcZt|jdS)Nr;)r?r=keysr%s rr+zLLMFnOutputRow.result_keys!DJOO%%&&r**rNrkey str | Nonerct||j|}|||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)rr>r+r=)r!rrR current_keys rset_result_valuezLLMFnOutputRow.set_result_values_ !(9:::oo'' ;#,,&+DJ{ # F J{ # 3rc||jvr3d}|}||jvr$d||}|dz}||jv$|}|}|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)r!rRridx candidate_keyr+r)s raddzLLMFnOutputRow.adds $*  CM4:-- 'sC 8 8 Ag 4:-- C__&& z~~j11  3!- :r)r8r9rr)r rB)r rF)rr r rIr,r-r.r/rD)rrrRrSr r)rRr rrIr r)r r0r1r2rArErHrKr"r&rr)r+rVr\rrrr7r7Fs&BBBB$%%%%$$$$))))****AAAA!!!!----++++      &......rr7)rrr r )rrrrr r)r2 __future__rr3typingrrrrIrrr ABCMetarr7rrrr`s"""""" ))))))))))     .....n!45....4k.k.k.k.k.'k.k.k.k.k.r