Ë
    ü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  G d„ dej                  «      Zy)zThe compile command.é    )Úannotations)ÚSequence)Úcommand)Úcommand_utils)Úipython_env)Úmodel_registry)Úparsed_args_lib)Úpost_process_utils)Úpy_utilsc                  óV   ‡ — 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.c                ó>   •— t         ‰| �  «        || _        || _        y)zyConstructor.

        Args:
          models: ModelRegistry instance.
          env: The IPythonEnv environment.
        N)ÚsuperÚ__init__Ú_modelsÚ_ipython_env)ÚselfÚmodelsÚenvÚ	__class__s      €ú_/home/htdocs/ttos/venv/lib/python3.12/site-packages/google/generativeai/notebook/compile_cmd.pyr   zCompileCommand.__init__   s   ø€ ô 	‰ÑÔØˆŒØˆÕó    c                óÔ   — t        j                  | j                  | j                  |||¬«      }t	        j
                  |j                  |«       dj                  |j                  «      S )N)r   r   Úparsed_argsÚcell_contentÚpost_processing_fnsz%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Ñ2Ø—<‘<Ø×!Ñ!Ø#Ø%Ø 3ô
ˆô 	×Ñ˜K×9Ñ9¸6ÔBØ6×=Ñ=¸k×>[Ñ>[Ó\Ð\r   c                ó,   — t        j                  |«      S ©N)r
   Úresolve_post_processing_tokens)r   Útokenss     r   Úparse_post_processing_tokensz+CompileCommand.parse_post_processing_tokens?   s   € ô "×@Ñ@ÀÓHÐHr   r$   )r   zmodel_registry.ModelRegistryr   zipython_env.IPythonEnv | None)r   zparsed_args_lib.ParsedArgsr   Ústrr   ú2Sequence[post_process_utils.ParsedPostProcessExpr]Úreturnr(   )r&   zSequence[Sequence[str]]r*   r)   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r"   r'   Ú__classcell__)r   s   @r   r   r      sk   ø„ Ù2ð
 .2ð à,ð ð +õ ð]à/ð]ð ð]ð Pð	]ð
 
ó]ð"IØ-ðIà	;÷Ir   r   N)r.   Ú
__future__r   Útypingr   Úgoogle.generativeai.notebookr   r   r   r   r	   r
   r   ÚCommandr   © r   r   ú<module>r5      s7   ðñ Ý "å Ý 0Ý 6Ý 4Ý 7Ý 8Ý ;Ý 1ô&I�W—_‘_õ &Ir   