
    ; 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  G d dej                  ZdS )zThe compile command.    )annotations)Sequence)command)command_utils)ipython_env)model_registry)parsed_args_lib)post_process_utils)py_utilsc                  8     e Zd ZdZ	 dd fdZddZddZ xZS )CompileCommandz(Implementation of the "compile" command.Nmodelsmodel_registry.ModelRegistryenvipython_env.IPythonEnv | Nonec                d    t                                                       || _        || _        dS )zyConstructor.

        Args:
          models: ModelRegistry instance.
          env: The IPythonEnv environment.
        N)super__init___models_ipython_env)selfr   r   	__class__s      /var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/google/generativeai/notebook/compile_cmd.pyr   zCompileCommand.__init__   s0     	    parsed_argsparsed_args_lib.ParsedArgscell_contentstrpost_processing_fns2Sequence[post_process_utils.ParsedPostProcessExpr]returnc                    t          j        | j        | j        |||          }t	          j        |j        |           d                    |j                  S )N)r   r   r   r   r   z%Saved function to Python variable: {})r   create_llm_functionr   r   r   
set_py_varcompile_save_nameformat)r   r   r   r   llm_fns        r   executezCompileCommand.execute.   s^     2<!#% 3
 
 
 	K96BBB6==k>[\\\r   tokensSequence[Sequence[str]]c                *    t          j        |          S N)r
   resolve_post_processing_tokens)r   r)   s     r   parse_post_processing_tokensz+CompileCommand.parse_post_processing_tokens?   s     "@HHHr   r,   )r   r   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      s        22
 .2             ] ] ] ]"I I I I I I I Ir   r   N)r2   
__future__r   typingr   google.generativeai.notebookr   r   r   r   r	   r
   r   Commandr    r   r   <module>r9      s      " " " " " "       0 0 0 0 0 0 6 6 6 6 6 6 4 4 4 4 4 4 7 7 7 7 7 7 8 8 8 8 8 8 ; ; ; ; ; ; 1 1 1 1 1 1&I &I &I &I &IW_ &I &I &I &I &Ir   