; i\dZddlmZddlmZmZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lm Z dd lmZdd lmZdd lmZddlmZddlmZGddZddZdS)zMagicsEngine class.) annotations) AbstractSetSequence)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_utilsc8eZdZdZ dddZddZddZddZdS) 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. Nregistry#model_registry.ModelRegistry | Noneenvipython_env.IPythonEnv | Nonec r||_|ptj}tjjt j||tjjtj ||tjj tj |tjjtj||i|_dS)N)modelsr)r) _ipython_envr ModelRegistryr CommandNameRUN_CMDr RunCommand COMPILE_CMDr CompileCommand COMPARE_CMDr CompareCommandEVAL_CMDr EvalCommand _cmd_handlers)selfrrrs /var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/google/generativeai/notebook/magics_engine.py__init__zMagicsEngine.__init__)s  ;^9;;  ' /1C6WZ1[1[1[  ' 3[5O3666  ' 3[5OTW5X5X5X  ' 0(2FfZ]2^2^2^ R linestr placeholdersAbstractSet[str]returnGtuple[parsed_args_lib.ParsedArgs, parsed_args_lib.PostProcessingTokens]cPtj||S)N)r CmdLineParser parse_line)r&r*r,s r'r2zMagicsEngine.parse_line9s# ,..99$ MMMr)ftuple[command.Command, parsed_args_lib.ParsedArgs, Sequence[post_process_utils.ParsedPostProcessExpr]]c|||\}}|j}|j|}||}|||fS)a}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 )r2cmdr%parse_post_processing_tokens)r&r*r, parsed_argspost_processing_tokenscmd_namehandlerpost_processing_fnss r' _get_handlerzMagicsEngine._get_handler@sT"/3oodL.Q.Q+ +?$X.%BBCYZZ %888r) cell_contentc(t|}tj|} |||\}}}||||S#t j$r-}|j||j|cYd}~Sd}~wt j $rr}| |j| r-d | dnd} t| |d}~wwxYw)z2Executes the supplied magic line and cell payload.NzGot parser error: {}) _clean_cellrget_placeholdersr<executerParserNormalExitrset_ipython_env ParserErrordisplaymsgsformat SystemExit) r&r*r=cellr,r:r7r;e error_msgs r' execute_cellzMagicsEngine.execute_cellWs!<((#4T::  /8<8I8I$P\8]8] 5G["5??;6IJJ J/    ,!!$"3444HHHHHH* / / / IId' ( ( (HIvvxxW.55affhhrlCCCUWIY''Q . /s*0AD%"B D DA-D  D)NN)rrrr)r*r+r,r-r.r/)r*r+r,r-r.r3)r*r+r=r+)__name__ __module__ __qualname____doc__r(r2r<rNr)r'rr"s9=-1     NNNN9999.//////r)rr=r+r.cH|}|dr |dd}|S)N r?)endswith)r=rKs r'rArAts/ D }}TCRCy Kr)N)r=r+r.r+)rR __future__rtypingrrgoogle.generativeai.notebookrrrr r r r r rrr google.generativeai.notebook.librrrArSr)r'r[s^""""""((((((((888888888888000000444444444444111111444444777777888888;;;;;;000000999999O/O/O/O/O/O/O/O/dr)