rhdZddlmZddlZddlmZmZmZeZd dZ d dZ Gddee efej Z Gd d e Zy)zLLMFnOutputRow.) annotationsN)AnyIteratorMappingcHt|dr |jSt|S)N__name__)hasattrrstr)xs /var/www/html/ai-insurance-compliance-backend/venv/lib/python3.12/site-packages/google/generativeai/notebook/lib/llmfn_output_row.py_get_name_of_typer sq*zz q6Mc |tk(ryt||s6tdjt |t t |y)Nz2Value of last entry must be of type "{}", got "{}")r isinstance ValueErrorformatr type)value result_types r _validate_is_result_typer sKc e[ ) @ G G!+.!$u+.    *rceZdZdZej ddZej d dZej d dZej d dZ ej d dZ y) LLMFnOutputRowViewz!Immutable view of LLMFnOutputRow.cy)z)For expressions like: x in this_instance.Nselfks r __contains__zLLMFnOutputRowView.__contains__0rcy)z)For expressions like: str(this_instance).Nrrs r __str__zLLMFnOutputRowView.__str__4rrcy)z.Returns the type enforced for the result cell.Nrr!s r rzLLMFnOutputRowView.result_type9rrcy)z!Get the value of the result cell.Nrr!s r result_valuezLLMFnOutputRowView.result_value=rrcy)zGet the key of the result cell.Nrr!s r result_keyzLLMFnOutputRowView.result_keyArrNrr returnboolr)r r) type[Any]r)r) r __module__ __qualname____doc__abcabstractmethodrr"rr%r'rrr rr,s+ 88 88 == 00 ..rr) metaclasscjeZdZdZddZddZddZddZddZddZ ddZ dd Z dd Z ddd Z dd Zy )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. ct||_|js td||_t |jj d}t ||jy)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)rdatarr%s r __init__zLLMFnOutputRow.__init__ZsV15T  zz:; ;'DJJ--/04  t/@/@Arc6|jjSN)r:__iter__r!s r rBzLLMFnOutputRow.__iter__lszz""$$rc6|jjSrA)r:__len__r!s r rDzLLMFnOutputRow.__len__oszz!!##rc8|jj|SrA)r: __getitem__rs r rFzLLMFnOutputRow.__getitem__rszz%%a((rc8|jj|SrA)r:rrs r rzLLMFnOutputRow.__contains__vszz&&q))rcTdj|jjS)NzLLMFnOutputRow: {})rr:r"r!s r r"zLLMFnOutputRow.__str__ys #**4::+=+=+?@@rc|jSrA)r;r!s r rzLLMFnOutputRow.result_type|s   rc<|j|jSrA)r:r'r!s r r%zLLMFnOutputRow.result_valueszz$//+,,rcNt|jjdS)Nr8)r<r:keysr!s r r'zLLMFnOutputRow.result_keysDJJOO%&r**rNct||j|j}|||k(r||j|<y|j|=||j|<y)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:)rrkey current_keys r set_result_valuezLLMFnOutputRow.set_result_valuesV !(9(9:oo' ;#,&+DJJ{ #  JJ{ # 3rc&||jvr:d}|}||jvr&dj||}|dz}||jvr&|}|j}|jj|}||j|<||j|<y)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)rrNridx candidate_keyr'r%s r addzLLMFnOutputRow.adds $** CM4::- 'sC 8 Ag 4::- C__& zz~~j1  3!- :r)r>zMapping[str, _CELLVALUETYPE]rr-)r)z Iterator[str])r)int)rr r)_CELLVALUETYPEr(r+r,r.rA)rrrNz str | Noner)None)rNr rrXr)rY)rr/r0r1r?rBrDrFrr"rr%r'rPrVrrr r6r6Fs@&B$%$)*A!-+  &.rr6)r r-r)r )rrrr-r)rY)r1 __future__rr2typingrrrrXr rr ABCMetarr6rrr r]sR" ))  .n!45.4k.'k.r