
    ; i
                        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dS )zThe compare command.    )annotations)Sequence)command)command_utils)input_utils)ipython_env)output_utils)parsed_args_lib)post_process_utilsNc                  8     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.Nenvipython_env.IPythonEnv | Nonec                V    t                                                       || _        dS )zOConstructor.

        Args:
          env: The IPythonEnv environment.
        N)super__init___ipython_env)selfr   	__class__s     /var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/google/generativeai/notebook/compare_cmd.pyr   zCompareCommand.__init__!   s)     	    parsed_argsparsed_args_lib.ParsedArgscell_contentstrpost_processing_fns2Sequence[post_process_utils.ParsedPostProcessExpr]returnpandas.DataFramec                    t          j        |d          }t          j        | j        ||          } ||          }t          j        ||           |                                S )NT)suppress_status_msgs)r   r   r   )inputs)resultsr   )r   join_inputs_sourcesr   create_llm_compare_functionr   r	   write_to_outputsas_pandas_dataframe)r   r   r   r   r"   
llm_cmp_fnr#   s          r   executezCompareCommand.execute-   sy     0SWXXX">!# 3
 
 

 *F+++%g;OOOO**,,,r   tokensSequence[Sequence[str]]c                (    |rt          d          g S )Nz-Post-processing is not supported by "compare")RuntimeError)r   r*   s     r   parse_post_processing_tokensz+CompareCommand.parse_post_processing_tokensB   s!      	PNOOO	r   )N)r   r   )r   r   r   r   r   r   r   r   )r*   r+   r   r   )__name__
__module____qualname____doc__r   r)   r.   __classcell__)r   s   @r   r   r      su        .. .2
  
  
  
  
  
  
 - - - -*       r   r   )r2   
__future__r   typingr   google.generativeai.notebookr   r   r   r   r	   r
   r   pandasCommandr    r   r   <module>r:      s      " " " " " "       0 0 0 0 0 0 6 6 6 6 6 6 4 4 4 4 4 4 4 4 4 4 4 4 5 5 5 5 5 5 8 8 8 8 8 8 ; ; ; ; ; ; ) ) ) ) )W_ ) ) ) ) )r   