Ë
    ü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Z G d„ dej                  «      Zy)zThe compare command.é    )Úannotations)ÚSequence)Úcommand)Úcommand_utils)Úinput_utils)Úipython_env)Úoutput_utils)Úparsed_args_lib)Úpost_process_utilsNc                  óR   ‡ — e Zd ZdZ	 d	 dˆ fd„Z	 	 	 	 	 	 	 	 dd„Z	 	 	 	 dd„Zˆ xZS )	ÚCompareCommandz$Implementation of "compare" command.c                ó0   •— t         ‰| �  «        || _        y)zOConstructor.

        Args:
          env: The IPythonEnv environment.
        N)ÚsuperÚ__init__Ú_ipython_env)ÚselfÚenvÚ	__class__s     €ú_/home/htdocs/ttos/venv/lib/python3.12/site-packages/google/generativeai/notebook/compare_cmd.pyr   zCompareCommand.__init__!   s   ø€ ô 	‰ÑÔØˆÕó    c                óÔ   — t        j                  |d¬«      }t        j                  | j                  ||¬«      } ||¬«      }t        j                  ||¬«       |j                  «       S )NT)Úsuppress_status_msgs)r   Úparsed_argsÚpost_processing_fns)Úinputs)Úresultsr   )r   Újoin_inputs_sourcesr   Úcreate_llm_compare_functionr   r	   Úwrite_to_outputsÚas_pandas_dataframe)r   r   Úcell_contentr   r   Ú
llm_cmp_fnr   s          r   ÚexecutezCompareCommand.execute-   sa   € ô ×0Ñ0°ÐSWÔXˆä"×>Ñ>Ø×!Ñ!Ø#Ø 3ô
ˆ
ñ  FÔ+ˆÜ×%Ñ%¨gÀ;ÕOØ×*Ñ*Ó,Ð,r   c                ó    — |rt        d«      ‚g S )Nz-Post-processing is not supported by "compare")ÚRuntimeError)r   Útokenss     r   Úparse_post_processing_tokensz+CompareCommand.parse_post_processing_tokensB   s   € ñ ÜÐNÓOÐOØˆ	r   )N)r   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      sY   ø„ Ù.ð .2ð
 à*õ
 ð-à/ð-ð ð-ð Pð	-ð
 
ó-ð*Ø-ðà	;÷r   r   )r.   Ú
__future__r   Útypingr   Úgoogle.generativeai.notebookr   r   r   r   r	   r
   r   ÚpandasÚCommandr   © r   r   ú<module>r6      s8   ðñ Ý "å å 0Ý 6Ý 4Ý 4Ý 5Ý 8Ý ;Û ô)�W—_‘_õ )r   