mxiydZddlmZddlZddlZddlmZddlmZGdde ejZ Gd d e Z Gd d e Z Gd dejZy)aCustomized ArgumentParser. The default behvaior of argparse.ArgumentParser's parse_args() method is to exit with a SystemExit exception in the following cases: 1. When the user requests a help message (with the --help or -h flags), or 2. When there's a parsing error (e.g. missing required flags or mistyped flags) To make the errors more user-friendly, this class customizes argparse.ArgumentParser and raises either ParserNormalExit for (1) or ParserError for (2); this way the caller has control over how to display them to the user. ) annotationsN)Sequence) ipython_envc`eZdZdZdfd ZddZdZd dZejd dZ xZ S) _ParserBaseExceptionz7Base class for parser exceptions including normal exit.cft|dj|g|i|||_d|_y)N)super__init__join_msgs _ipython_env)selfmsgsargskwargs __class__s c/home/htdocs/ttos/venv/lib/python3.12/site-packages/google/generativeai/notebook/argument_parser.pyr z_ParserBaseException.__init__'s1 888 ;?c||_yN)rrenvs rset_ipython_envz$_ParserBaseException.set_ipython_env,s rc:|j|jyr)displayrrs r_ipython_display_z&_ParserBaseException._ipython_display_/s T&&'rc|jSr)r rs rrz_ParserBaseException.msgs2s zzrcy)z-Display this exception on an IPython console.Nrs rrz_ParserBaseException.display5sr)r Sequence[str])rzipython_env.IPythonEnvreturnNone)r#r"rzipython_env.IPythonEnv | Noner#r$) __name__ __module__ __qualname____doc__r rrrabcabstractmethodr __classcell__rs@rrr$s5A@  ( <s  rr0ceZdZdZddZy) ParserErrorz(Exception thrown when there is an error.cd|jD] }t|||jdyy)Nz,ERROR)r r2 display_htmlr3s rrzParserError.displayLs5:: C #J  ?   K L rNr%r5r!rrr7r7Is 2Mrr7c2eZdZdZfdZddZddZxZS)ArgumentParsera!Customized ArgumentParser for LLM Magics. This class overrides the parent argparse.ArgumentParser's error-handling methods to avoid side-effects like printing to stderr. The messages are accumulated and passed into the raised exceptions for the caller to handle them. c2t||i|g|_yr)r r _messages)rrrrs rr zArgumentParser.__init__]s $)&)$&rc<~|jj|y)z2Override ArgumentParser's _print_message() method.N)r=append)rmessagefiles r_print_messagezArgumentParser._print_messageas  g&rc|r|j||j}g|_|dk(r t|t|)z(Override ArgumentParser's exit() method.r)r)rBr=r0r7)rstatusr@rs rexitzArgumentParser.exitfs@     (~~ Q;"- -4( (rr)rN)r&r'r(r)r rBrEr,r-s@rr;r;Ts'' )rr;)r) __future__rr*argparsetypingrgoogle.generativeai.notebookr RuntimeErrorABCMetarr0r7r;r!rrrLs[ # 4<<3;;<4+M&M)X,,)r