Ë
    ümxi|  ã                  óÜ   — d Z ddlmZ ddlmZmZmZmZ ddlm	Z	 ddl
mZ ddl
mZ ddlmZ ddlmZ dd	„Zdd
„Zdd„Zdd„Z G d„ dej(                  «      Z G d„ dej,                  «      Zy)zSheetsInputs.é    )Úannotations)ÚAnyÚCallableÚMappingÚSequence)Úparse)Úgspread_client)Ú	sheets_id)Úllmfn_inputs_source)Úllmfn_outputsc                ó  — 	 t        j                  | «      }|j                  rNt	        j
                  t	        j                  | «      ¬«      }t        j                  «       j                  |«       |S y# t        $ r Y yw xY w)z4Try to open a Sheets document with `value` as a URL.N)Úurl)
r   ÚurlparseÚ
ValueErrorÚschemer
   ÚSheetsIdentifierÚ	SheetsURLr	   Ú
get_clientÚvalidate)ÚvalueÚparse_resultÚsids      ú`/home/htdocs/ttos/venv/lib/python3.12/site-packages/google/generativeai/notebook/sheets_utils.pyÚ_try_sheet_id_as_urlr      sp   € ðÜ—~‘~ eÓ,ˆð
 ×Òô ×(Ñ(¬Y×-@Ñ-@ÀÓ-GÔHˆÜ×!Ñ!Ó#×,Ñ,¨SÔ1Øˆ
àøô ò áðús   ‚A2 Á2	A>Á=A>c                óò   — 	 t        j                  t        j                  | «      ¬«      }	 t	        j
                  «       j                  |«       |S # t        $ r Y yw xY w# t        j                  $ r Y yw xY w)z4Try to open a Sheets document with `value` as a key.)ÚkeyN)r
   r   Ú	SheetsKeyr   r	   r   r   ÚSpreadsheetNotFoundError©r   r   s     r   Ú_try_sheet_id_as_keyr    -   sr   € ðÜ×(Ñ(¬Y×-@Ñ-@ÀÓ-GÔHˆð
Ü×!Ñ!Ó#×,Ñ,¨SÔ1ð €Jøô ò áðûô ×2Ñ2ò Ùðús"   ‚)A ¬#A  Á	AÁAÁ A6Á5A6c                ó¬   — t        j                  | ¬«      }	 t        j                  «       j	                  |«       |S # t        j
                  $ r Y yw xY w)z5Try to open a Sheets document with `value` as a name.)ÚnameN)r
   r   r	   r   r   r   r   s     r   Ú_try_sheet_id_as_namer#   <   sO   € ä
×
$Ñ
$¨%Ô
0€CðÜ×!Ñ!Ó#×,Ñ,¨SÔ1ð €Jøô ×2Ñ2ò Ùðús   ˜#= ½AÁAc                ó�   — t        | «      x}r|S t        | «      x}r|S t        | «      x}r|S t        dj	                  | «      «      ‚)Nz-No Sheets found with "{}" as URL, key or name)r   r    r#   ÚRuntimeErrorÚformatr   s     r   Úget_sheets_id_from_strr'   F   sX   € Ü" 5Ó)Ð)€sÐ)Øˆ
Ü" 5Ó)Ð)€sÐ)Øˆ
Ü# EÓ*Ð*€sÐ*Øˆ
Ü
ÐF×MÑMÈeÓTÓ
UÐUó    c                  ó2   ‡ — e Zd ZdZddˆ fd„Z	 	 dd„Zˆ xZS )ÚSheetsInputsz,Inputs to an LLMFunction from Google Sheets.c                ó>   •— t         ‰| �  «        || _        || _        y ©N)ÚsuperÚ__init__Ú_sidÚ_worksheet_id)Úselfr   Úworksheet_idÚ	__class__s      €r   r.   zSheetsInputs.__init__S   s   ø€ Ü‰ÑÔØˆŒ	Ø)ˆÕr(   c                ót   — t        j                  «       j                  | j                  | j                  ¬«      S )N)r   r2   )r	   r   Úget_all_recordsr/   r0   )r1   s    r   Ú_to_normalized_inputs_implz'SheetsInputs._to_normalized_inputs_implX   s4   € ô ×(Ñ(Ó*×:Ñ:Ø—	‘	¨×(:Ñ(:ð ;ó 
ð 	
r(   )r   )r   úsheets_id.SheetsIdentifierr2   Úint)Úreturnz6tuple[Sequence[Mapping[str, str]], Callable[[], None]])Ú__name__Ú
__module__Ú__qualname__Ú__doc__r.   r6   Ú__classcell__)r3   s   @r   r*   r*   P   s   ø„ Ù6ö*ð

à	?÷
r(   r*   c                  ó    — e Zd ZdZdd„Zdd„Zy)ÚSheetsOutputsz4Writes outputs from an LLMFunction to Google Sheets.c                ó   — || _         y r,   )r/   )r1   r   s     r   r.   zSheetsOutputs.__init__c   s	   € Øˆ�	r(   c                ó8  — |j                  «       }t        |j                  «       «      g}|j                  t	        |j                  «       Ž D �cg c]  }t        |«      ‘Œ c}«       t        j                  «       j                  | j                  |¬«       y c c}w )N)r   Úrows)
Úas_dictÚlistÚkeysÚextendÚzipÚvaluesr	   r   Úwrite_recordsr/   )r1   ÚoutputsÚoutputs_dictÚoutputs_rowsÚxs        r   Úwrite_outputszSheetsOutputs.write_outputsf   s{   € à—‘Ó(ˆÜ-1°,×2CÑ2CÓ2EÓ-FÐ,GˆØ×Ñ¬c°<×3FÑ3FÓ3HÐ.IÖJ¨œT !�WÒJÔKä×!Ñ!Ó#×1Ñ1Ø—	‘	Øð 	2õ 	
ùò Ks   ÁBN)r   r7   )rK   zllmfn_outputs.LLMFnOutputsBaser9   ÚNone)r:   r;   r<   r=   r.   rO   © r(   r   r@   r@   `   s   „ Ù>óô	
r(   r@   N)r   Ústrr9   z!sheets_id.SheetsIdentifier | None)r   rR   r9   r7   )r=   Ú
__future__r   Útypingr   r   r   r   Úurllibr   Úgoogle.generativeai.notebookr	   r
   Ú google.generativeai.notebook.libr   r   r   r    r#   r'   ÚLLMFnInputsSourcer*   ÚLLMFnOutputsSinkr@   rQ   r(   r   ú<module>rZ      s\   ðñ Ý "ç 3Ó 3Ý Ý 7Ý 2Ý @Ý :óó&óóVô
Ð&×8Ñ8ô 
ô 
�M×2Ñ2õ 
r(   