ó
    
û!i
  ã                  ó¦   • S r SSKJr  SSKJr  SSKJr  SSKJr  SSKJr  SSKJ	r	  SSKJ
r
  SS	KJr  SS
KJr  SSKr " S S\R                  5      rg)zThe compare command.é    )Úannotations)ÚSequence)Úcommand)Úcommand_utils)Úinput_utils)Úipython_env)Úoutput_utils)Úparsed_args_lib)Úpost_process_utilsNc                  ód   ^ • \ rS rSrSr S SU 4S jjjr        S	S jr    S
S jrSrU =r	$ )ÚCompareCommandé   z$Implementation of "compare" command.c                ó.   >• [         TU ]  5         Xl        g)z7Constructor.

Args:
  env: The IPythonEnv environment.
N)ÚsuperÚ__init__Ú_ipython_env)ÚselfÚenvÚ	__class__s     €Úb/home/james-whalen/.local/lib/python3.13/site-packages/google/generativeai/notebook/compare_cmd.pyr   ÚCompareCommand.__init__!   s   ø€ ô 	‰ÑÔØÕó    c                óÀ   • [         R                  " USS9n[        R                  " U R                  UUS9nU" US9n[
        R                  " XaS9  UR                  5       $ )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   ÚexecuteÚCompareCommand.execute-   s_   € ô ×0Ò0°ÐSWÑXˆä"×>Ò>Ø×!Ñ!Ø#Ø 3ñ
ˆ
ñ  FÑ+ˆÜ×%Ò%¨gÒOØ×*Ñ*Ó,Ð,r   c                ó*   • U(       a  [        S5      e/ $ )Nz-Post-processing is not supported by "compare")ÚRuntimeError)r   Útokenss     r   Úparse_post_processing_tokensÚ+CompareCommand.parse_post_processing_tokensB   s   € ö ÜÐNÓOÐOØˆ	r   )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__Ú__firstlineno__Ú__doc__r   r%   r*   Ú__static_attributes__Ú__classcell__)r   s   @r   r   r      sc   ø† Ù.ð .2ð
 à*÷
 ð 
 ð-à/ð-ð ð-ð Pð	-ð
 
ô-ð*Ø-ðà	;÷ò r   r   )r3   Ú
__future__r   Útypingr   Úgoogle.generativeai.notebookr   r   r   r   r	   r
   r   ÚpandasÚCommandr   © r   r   Ú<module>r<      s8   ðñ Ý "å å 0Ý 6Ý 4Ý 4Ý 5Ý 8Ý ;Û ô)W—_‘_õ )r   