Ë
    ümxi¸
  ã                  ó²   — d 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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lZ G d„ dej                  «      Zy)zThe run command.é    )Úannotations)ÚSequence)Úcommand)Úcommand_utils)Úinput_utils)Úipython_env)Úmodel_registry)Úoutput_utils)Úparsed_args_lib)Úpost_process_utilsNc                  óV   ‡ — e Zd ZdZ	 d	 	 	 dˆ fd„Z	 	 	 	 	 	 	 	 dd„Z	 	 	 	 dd„Zˆ xZS )	Ú
RunCommandz$Implementation of the "run" command.c                ó>   •— t         ‰| �  «        || _        || _        y)zyConstructor.

        Args:
          models: ModelRegistry instance.
          env: The IPythonEnv environment.
        N)ÚsuperÚ__init__Ú_modelsÚ_ipython_env)ÚselfÚmodelsÚenvÚ	__class__s      €ú[/home/htdocs/ttos/venv/lib/python3.12/site-packages/google/generativeai/notebook/run_cmd.pyr   zRunCommand.__init__"   s   ø€ ô 	‰ÑÔØˆŒØˆÕó    c                óì   — t        j                  |d¬«      }t        j                  | j                  | j
                  |||¬«      } ||¬«      }t        j                  ||¬«       |j                  «       S )NT)Úsuppress_status_msgs)r   r   Úparsed_argsÚcell_contentÚpost_processing_fns)Úinputs)Úresultsr   )	r   Újoin_inputs_sourcesr   Úcreate_llm_functionr   r   r
   Úwrite_to_outputsÚas_pandas_dataframe)r   r   r   r   r   Úllm_fnr    s          r   ÚexecutezRunCommand.execute1   sk   € ô ×0Ñ0°ÐSWÔXˆä×2Ñ2Ø—<‘<Ø×!Ñ!Ø#Ø%Ø 3ô
ˆñ  Ô'ˆÜ×%Ñ%¨gÀ;ÕOØ×*Ñ*Ó,Ð,r   c                ó,   — t        j                  |«      S ©N)r   Úresolve_post_processing_tokens)r   Útokenss     r   Úparse_post_processing_tokensz'RunCommand.parse_post_processing_tokensH   s   € ô "×@Ñ@ÀÓHÐHr   r(   )r   zmodel_registry.ModelRegistryr   zipython_env.IPythonEnv | None)r   zparsed_args_lib.ParsedArgsr   Ústrr   ú2Sequence[post_process_utils.ParsedPostProcessExpr]Úreturnzpandas.DataFrame)r*   zSequence[Sequence[str]]r.   r-   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r&   r+   Ú__classcell__)r   s   @r   r   r      sf   ø„ Ù.ð
 .2ð à,ð ð +õ ð-à/ð-ð ð-ð Pð	-ð
 
ó-ð.IØ-ðIà	;÷Ir   r   )r2   Ú
__future__r   Útypingr   Úgoogle.generativeai.notebookr   r   r   r   r	   r
   r   r   ÚpandasÚCommandr   © r   r   ú<module>r:      s=   ðñ Ý "å å 0Ý 6Ý 4Ý 4Ý 7Ý 5Ý 8Ý ;Û ô,I�—‘õ ,Ir   