阺hydZddlmZddlZddlZddlmZddlmZGdde ej Z Gd d e Z Gd d e Z Gd dejZdS)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 Zdd Zd Zdd Zej dd Z xZ S)_ParserBaseExceptionz7Base class for parser exceptions including normal exit.msgs Sequence[str]ctjd|g|Ri|||_d|_dS)N)super__init__join_msgs _ipython_env)selfrargskwargs __class__s t/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/google/generativeai/notebook/argument_parser.pyr z_ParserBaseException.__init__'sI8888888 ;?envipython_env.IPythonEnvreturnNonec||_dSN)rrrs rset_ipython_envz$_ParserBaseException.set_ipython_env,src:||jdSr)displayrrs r_ipython_display_z&_ParserBaseException._ipython_display_/s T&'''''rc|jSr)rr!s rrz_ParserBaseException.msgs2s zripython_env.IPythonEnv | NonecdS)z-Display this exception on an IPython console.Nrs rr z_ParserBaseException.display5sr)rr )rrrr)rr rr$rr) __name__ __module__ __qualname____doc__r rr"rabcabstractmethodr __classcell__rs@rrr$sAA@@@@@@     ((( <<<<<<<s2 rr2ceZdZdZddZdS) ParserErrorz(Exception thrown when there is an error.rr$rrcj|jD]}t|||ddSdS)Nz,ERROR)rr4 display_htmlr5s rr zParserError.displayLsG:  C #JJJJ ?   K L L L L L ?rNr'r7r&rrr9r9Is422MMMMMMrr9c2eZdZdZfdZddZddZxZS) 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. cHtj|i|g|_dSr)r r _messages)rrrrs rr zArgumentParser.__init__]s*$)&)))$&rNc<~|j|dS)z2Override ArgumentParser's _print_message() method.N)r?append)rmessagefiles r_print_messagezArgumentParser._print_messageas!  g&&&&&rrc|r|||j}g|_|dkrt|t|)z(Override ArgumentParser's exit() method.r)r)rDr?r2r9)rstatusrBrs rexitzArgumentParser.exitfsW  )    ( ( (~ Q;;"--- -4((( (rr)rN)r(r)r*r+r rDrGr.r/s@rr=r=Tsj''''''''' ) ) ) ) ) ) ) )rr=)r+ __future__rr,argparsetypingrgoogle.generativeai.notebookr RuntimeErrorABCMetarr2r9r=r&rrrNs!  #""""" 444444<<<<<<3;<<<<4+MMMMM&MMM)))))X,)))))r