o j6h\@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 ModelRegistryr CommandNameRUN_CMDr RunCommand COMPILE_CMDr CompileCommand COMPARE_CMDrCompareCommandEVAL_CMDr EvalCommand _cmd_handlers)selfrrrr$y/var/www/html/chefvision.cloud.itp360.com/venv/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)r CmdLineParser parse_line)r#r'r)r$r$r%r.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.cmdr"parse_post_processing_tokens)r#r'r) parsed_argspost_processing_tokenscmd_namehandlerpost_processing_fnsr$r$r% _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_cellrget_placeholdersr7executerParserNormalExitrset_ipython_env ParserErrordisplaymsgsformat SystemExit) r#r'r8cellr)r5r2r6e error_msgr$r$r% execute_cellWs      zMagicsEngine.execute_cell)NN)rrrr)r'r(r)r*r+r,)r'r(r)r*r+r/)r'r(r8r()__name__ __module__ __qualname____doc__r&r.r7rHr$r$r$r%r"s  rr8r(r+cCs|}|dr |dd}|S)N r9)endswith)r8rEr$r$r%r;ts  r;N)r8r(r+r()rL __future__rtypingrrgoogle.generativeai.notebookrrrrr r r r r rr google.generativeai.notebook.librrr;r$r$r$r%s"             R