+
    i\                        R t ^ RIHt ^ RIHtHt ^ RIHt ^ RIHt ^ RIH	t	 ^ RIH
t
 ^ RIHt ^ RIHt ^ R	IHt ^ R
IHt ^ RIHt ^ RIHt ^ RIHt ^ RIHt  ! R R4      tR R ltR# )zMagicsEngine class.)annotations)AbstractSetSequence)argument_parser)cmd_line_parser)command)compare_cmd)compile_cmd)eval_cmd)ipython_env)model_registry)parsed_args_lib)post_process_utils)run_cmd)prompt_utilsc                  N    ] tR t^"tRtRR R lltR R ltR R ltR	 R
 ltRt	R# )MagicsEnginezImplementation of functionality used by Magics.

This class provides the implementation for Magics, decoupled from the
details of integrating with Colab Magics such as registration.
Nc                    V ^8  d   QhRRRR/# )   registryz#model_registry.ModelRegistry | Noneenvzipython_env.IPythonEnv | None )formats   "}/Users/igloo/.openclaw/workspace/scratch/fb_ad_env/lib/python3.14/site-packages/google/generativeai/notebook/magics_engine.py__annotate__MagicsEngine.__annotate__)   s     
 
5
 +
    c                	   W n         T;'       g    \        P                  ! 4       p\        P                  P
                  \        P                  ! W2R 7      \        P                  P                  \        P                  ! W2R 7      \        P                  P                  \        P                  ! VR7      \        P                  P                  \        P                   ! W2R 7      /V n        R# ))modelsr   )r   N)_ipython_envr   ModelRegistryr   CommandNameRUN_CMDr   
RunCommandCOMPILE_CMDr	   CompileCommandCOMPARE_CMDr   CompareCommandEVAL_CMDr
   EvalCommand_cmd_handlers)selfr   r   r   s   &&& r   __init__MagicsEngine.__init__)   s    
  ;;^99;''//1C1C61[''33[5O5O6 ''33[5O5OTW5X''00(2F2Ff2^R
r   c               $    V ^8  d   QhRRRRRR/# )r   linestrplaceholdersAbstractSet[str]returnzGtuple[parsed_args_lib.ParsedArgs, parsed_args_lib.PostProcessingTokens]r   )r   s   "r   r   r   9   s.     N NN 'N 
Q	Nr   c                	J    \         P                  ! 4       P                  W4      # )N)r   CmdLineParser
parse_line)r+   r/   r1   s   &&&r   r6   MagicsEngine.parse_line9   s    
 ,,.99$MMr   c               $    V ^8  d   QhRRRRRR/# )r   r/   r0   r1   r2   r3   zftuple[command.Command, parsed_args_lib.ParsedArgs, Sequence[post_process_utils.ParsedPostProcessExpr]]r   )r   s   "r   r   r   @   s&     9 9 94D 9 J 9r   c                    V P                  W4      w  r4VP                  pV P                  V,          pVP                  V4      pWcV3# )a5  Given the command line, parse and return all components.

Args:
  line: The LLM Magics command line.
  placeholders: Placeholders from prompts in the cell contents.

Returns:
  A three-tuple containing:
  - The command (e.g. "run")
  - Parsed arguments for the command,
  - Parsed post-processing expressions
)r6   cmdr*   parse_post_processing_tokens)r+   r/   r1   parsed_argspost_processing_tokenscmd_namehandlerpost_processing_fnss   &&&     r   _get_handlerMagicsEngine._get_handler@   sL    " /3ood.Q+??$$X.%BBCYZ%888r   c                    V ^8  d   QhRRRR/# )r   r/   r0   cell_contentr   )r   s   "r   r   r   W   s     / / /C /r   c                &   \        V4      p\        P                  ! V4      p V P                  W4      w  rVpVP	                  WcV4      #   \
        P                   d6   pT P                  e   TP                  T P                  4       Tu Rp?# Rp?i\
        P                   dk   pTP                  T P                  4       TP                  4       '       d'   RP                  TP                  4       R,          4      MRp	\        T	4      ThRp?ii ; i)z2Executes the supplied magic line and cell payload.NzGot parser error: {} )_clean_cellr   get_placeholdersrA   executer   ParserNormalExitr   set_ipython_envParserErrordisplaymsgsr   
SystemExit)
r+   r/   rD   cellr1   r?   r<   r@   e	error_msgs
   &&&       r   execute_cellMagicsEngine.execute_cellW   s    <(#44T:	/8<8I8I$8]5G"5??;6IJJ// 	  ,!!$"3"34 H** 	/IId''( HIvvxx.55affhrlCUWIY'Q.	/s0   %A	 	D*BDD%D&A%DD)r*   r   )NN)
__name__
__module____qualname____firstlineno____doc__r,   r6   rA   rT   __static_attributes__r   r   r   r   r   "   s!    
 N9./ /r   r   c                    V ^8  d   QhRRRR/# )r   rD   r0   r3   r   )r   s   "r   r   r   t   s      c c r   c                B    T pVP                  R 4      '       d   VRR pV# )
NrG   )endswith)rD   rQ   s   & r   rH   rH   t   s(    
 D}}TCRyKr   N)rZ   
__future__r   typingr   r   google.generativeai.notebookr   r   r   r   r	   r
   r   r   r   r   r    google.generativeai.notebook.libr   r   rH   r   r   r   <module>rd      sF     " ( 8 8 0 4 4 1 4 7 8 ; 0 9O/ O/dr   