o Zh\@sdZddlmZddlmZmZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lm Z dd lmZdd lmZdd lmZddlmZddlmZGdddZdddZdS)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_utilsc@s>eZdZdZ  ddddZdddZdddZdddZdS) 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 Csb||_|pt}tjjtj||dtjjt j ||dtjj t j |dtjjtj||di|_dS)N)modelsr)r) _ipython_envr Z ModelRegistryr Z CommandNameZRUN_CMDrZ RunCommandZ COMPILE_CMDr ZCompileCommandZ COMPARE_CMDrZCompareCommandZEVAL_CMDr Z EvalCommand _cmd_handlers)selfrrrra/var/www/html/lang_env/lib/python3.10/site-packages/google/generativeai/notebook/magics_engine.py__init__)s   zMagicsEngine.__init__linestr placeholdersAbstractSet[str]returnGtuple[parsed_args_lib.ParsedArgs, parsed_args_lib.PostProcessingTokens]cCst||S)N)rZ CmdLineParser parse_line)rrrrrrr#9szMagicsEngine.parse_lineftuple[command.Command, parsed_args_lib.ParsedArgs, Sequence[post_process_utils.ParsedPostProcessExpr]]cCs4|||\}}|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 )r#cmdrZparse_post_processing_tokens)rrr parsed_argsZpost_processing_tokenscmd_namehandlerpost_processing_fnsrrr _get_handler@s    zMagicsEngine._get_handler cell_contentc Cst|}t|}z|||\}}}||||WStjy:}z|jdur.||j|WYd}~Sd}~wtj y`}z| |j| rUd | dnd} t | |d}~ww)z2Executes the supplied magic line and cell payload.NzGot parser error: {}) _clean_cellrZget_placeholdersr*executerZParserNormalExitrZset_ipython_envZ ParserErrordisplayZmsgsformat SystemExit) rrr+cellrr(r&r)e error_msgrrr execute_cellWs      zMagicsEngine.execute_cell)NN)rrrr)rrrr r!r")rrrr r!r$)rrr+r)__name__ __module__ __qualname____doc__rr#r*r6rrrrr"s  rr+rr!cCs|}|dr |dd}|S)N r,)endswith)r+r3rrrr.ts  r.N)r+rr!r)r: __future__rtypingrrZgoogle.generativeai.notebookrrrrr r r r r rrZ google.generativeai.notebook.librrr.rrrrs"             R