
    rh
                        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     {/var/www/html/ai-insurance-compliance-backend/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   