!i~CSrSSKJr SSKrSSKrSSKrSSKrSSKJrJ r J r J r J r SSK Jr SSK Jr \ \\\4r\ \\R(\ \\ \\/\44\ \\R*\R,4r\ \/\4rSSjrSSjr\R6"S S 9"S S \R855rSSS jjr"SS\R>5r "SS\R>5r!"SS\R>5r"\R6"S S 9"SS\55r#"SS\#5r$"SS\5r%\R6"S S 9"SS\55r&g) aYClasses that define arguments for populating ArgumentParser. The argparse module's ArgumentParser.add_argument() takes several parameters and is quite customizable. However this can lead to bugs where arguments do not behave as expected. For better ease-of-use and better testability, define a set of classes for the types of flags used by LLM Magics. Sample usage: str_flag = SingleValueFlagDef(name="title", required=True) enum_flag = EnumFlagDef(name="colors", required=True, enum_type=ColorsEnum) str_flag.add_argument_to_parser(my_parser) enum_flag.add_argument_to_parser(my_parser) ) annotationsN)AnyCallableSequenceTupleUnion)llmfn_inputs_source) llmfn_outputscRUR$![a [U5s$f=fN)__name__AttributeErrorstrxs _/home/james-whalen/.local/lib/python3.13/site-packages/google/generativeai/notebook/flag_def.py_get_type_namer?s)zz 1v s &&cRU(d [S5eUSS:Xa [S5eU$)z.Validation for long and short names for flags.zCannot be emptyr-zCannot start with dash) ValueError)names r_validate_flag_namerFs- *++ Aw#~122 KT)frozenc\rSrSr%SrS\S'SrS\S'SrS \S '\r S \S 'Sr S \S'Sr S\S'Sr S\S'Sr S \S'\RSSj5r\RSSj5rSSjrSSjrSrSrg)FlagDefOaAbstract base class for flag definitions. Attributes: name: Long name, e.g. "colors" will define the flag "--colors". required: Whether the flag must be provided on the command line. short_name: Optional short name. parse_type: The type that ArgumentParser should parse the command line argument to. dest_type: The type that the parsed value is converted to. This is used when we want ArgumentParser to parse as one type, then convert to a different type. E.g. for enums we parse as "str" then convert to the desired enum type in order to provide cleaner help messages. parse_to_dest_type_fn: If provided, this function will be used to convert the value from `parse_type` to `dest_type`. This can be used for validation as well. choices: If provided, limit the set of acceptable values to these choices. help_msg: If provided, adds help message when -h is used in the command line. rrFboolrequiredN str | None short_nameztype[_PARSETYPES] parse_typeztype[_DESTTYPES] | None dest_typez_PARSEFN | Noneparse_to_dest_type_fnzlist[_PARSETYPES] | Nonechoiceshelp_msgcg)zAdds this flag as an argument to `parser`. Child classes should implement this as a call to parser.add_argument() with the appropriate parameters. Args: parser: The parser to which this argument will be added. N)selfparsers radd_argument_to_parserFlagDef.add_argument_to_parserqrcg)z.For child classes to do additional validation.Nr(r)s r_do_additional_validation!FlagDef._do_additional_validation|r-rcLURc UR$UR$)z!Returns the final converted type.)r#r"r/s r_get_dest_typeFlagDef._get_dest_types"&.."8tLdnnLrc|URb UR$UR5nXR:XaS$U$)z;Returns a function to convert from parse_type to dest_type.cU$r r(rs r4FlagDef._get_parse_to_dest_type_fn..sQr)r$r3r")r)r#s r_get_parse_to_dest_type_fn"FlagDef._get_parse_to_dest_type_fns@  % % 1-- -'')  '  rc[UR5 URb[UR5 UR5 gr )rrr!r0r/s r __post_init__FlagDef.__post_init__s0DII& ?? &  0 &&(rr(r*argparse.ArgumentParserreturnNoner@rA)r@ztype[_DESTTYPES])r@_PARSEFN)r __module__ __qualname____firstlineno____doc____annotations__rr!rr"r#r$r%r&abcabstractmethodr+r0r3r9r<__static_attributes__r(rrrrOs( IHd!J !$'J!')-I&--1?1(,G %,Hj   ==M   )rrcP[X5(dgU(dg[X5U:g$)aReturns true if `namespace.dest` is set to a non-default value. Args: namespace: The Namespace that is populated by ArgumentParser. dest: The attribute in the Namespace to be populated. has_default: "None" is a valid default value so we use an additional `has_default` boolean to indicate that `default_value` is present. default_value: The default value to use when `has_default` is True. Returns: Whether namespace.dest is set to something other than the default value. FT)hasattrgetattr) namespacedest has_default default_values r_has_non_default_valuerSs*$ 9 # #  9 #} 44rcV^\rSrSrSrSU4SjjrSSSjjrSrU=r$) _SingleValueStoreActionzyCustom Action for storing a value in an argparse.Namespace. This action checks that the flag is specified at-most once. c @>[TU]"X40UD6 X0lX@lgr super__init__ _dest_type_parse_to_dest_type_fnr)option_stringsrPr#r$kwargs __class__s rrZ _SingleValueStoreAction.__init__" 88#&;#rc [U[5(d[U[5(ae[UUR[ US5[ US5S9(a&[R"USRU55eURUS5n[XPR5(d@[SR[UR5[[U5555e[!X RU5 g![aBn[R"USRUS[[U55U55eSnAff=f)NdefaultrQrRCannot set {} more than oncer!Error with value "{}", got {}: {}+Converted to wrong type, expected {} got {}) isinstancerbytesrSrPrMrNargparse ArgumentErrorformatr\ Exceptionrtyper[ RuntimeErrorsetattr)r)r*rOvalues option_stringconverted_valuees r__call__ _SingleValueStoreAction.__call__s$fc**:fe3L3LLL !  IIi0!$ 2  ((/M/T/TUb/cd d "99&)DO/??;;=DD"4??3"4#89   99o6 ((3::6!9nUYZ[U\F]_`a  sD E=EEr[r\r# type[Any]r$rCr r*r?rOargparse.Namespacerrzstr | Sequence[Any] | Nonersr r rDrErFrGrZrvrK __classcell__r`s@rrUrUsV < < ( <"%) "7'"7&"7+ "7 " "7"7rrUcV^\rSrSrSrSU4SjjrSSSjjrSrU=r$) _MultiValuesAppendActionzzCustom Action for appending values in an argparse.Namespace. This action checks that the flag is specified at-most once. c @>[TU]"X40UD6 X0lX@lgr rXr]s rrZ!_MultiValuesAppendAction.__init__rbrc [U[5(d[U[5(ae[X R5nU(a&[ R "USRU55eUHnURU5n[XpR5(d.[SRUR[U555eXu;a&[ R "USRU55eURU5 M g![aBn[ R "USRUS[[U55U55eSnAff=f)NrfrgrrhzDuplicate values "{}")rirrjrNrPrkrlrmr\rnrror[rpappend) r)r*rOrrrs curr_valuevaluertrus rrv!_MultiValuesAppendAction.__call__s-fc**:fe3L3LLLY 2 ((/M/T/TUb/cd dE "&"="=e"Do??"AHHo)> ,,,T3J3Q3QRW3XYY   o .) ,,7>>q >$q'#:A s5D E=EErxryr r{r}rs@rrrsV < < ( <"%) #/'#/&#/+ #/ " #/#/rrcJ^\rSrSrSrU4SjrSSSjjrSrU=r$)_BooleanValueStoreActioni%zCustom Action for setting a boolean value in argparse.Namespace. The boolean flag expects the default to be False and will set the value to True. This action checks that the flag is specified at-most once. c (>[TU]"X40UD6 gr )rYrZ)r)r^rPr_r`s rrZ!_BooleanValueStoreAction.__init__-s 88rc[UURSSS9(a&[R"USR U55e[ X RS5 g)NTFrerf)rSrPrkrlrmrq)r)r*rOrrrss rrv!_BooleanValueStoreAction.__call__5sL "  II  ((/M/T/TUb/cd d 99d+rr(r r{r}rs@rrr%sB9%) ,',&,+ , " ,,rrc\rSrSr%Sr"SS\R 5r\Rr S\ S'S Sjr S Sjr SS jr S rg )SingleValueFlagDefiGaODefinition for a flag that takes a single value. Sample usage: # This defines a flag that can be specified on the command line as: # --count=10 flag = SingleValueFlagDef(name="count", parse_type=int, required=True) flag.add_argument_to_parser(argument_parser) Attributes: default_value: Default value for optional flags. c\rSrSrSrSrSrg) SingleValueFlagDef._DefaultValueiUzSpecial value to represent "no value provided". "None" can be used as a default value, so in order to differentiate between "None" and "no value provided", create a special value for "no value provided". Nr()r rDrErFrGNOT_SETrKr(rr _DefaultValuerUs rrz!_DESTTYPES | _DefaultValue | NonerRcPUR[RR:g$)z2Returns whether `default_value` has been provided.)rRrrrr/s r_has_default_value%SingleValueFlagDef._has_default_valueas !!%7%E%E%M%MMMrc SUR-/nURbUSUR-/- n0nUR5(aURUS'URbURUS'UR bUR US'UR "U[URUR5UR5URSS.UD6 g)N--rrdr%help)actionror#r$rnargs) rr!rrRr%r& add_argumentrUr"r3r9rr)r*argsr_s rr+)SingleValueFlagDef.add_argument_to_parserestyy ! ?? & S4??*+ +D  " " $ $ $ 2 2F9  << # $ F9  == $!]]F6N *))+"&"A"A"C]]  rcXUR(a!UR5(a [S5eO UR5(d [S5eUR5(aCURb5[ URUR 55(d [S5eggg)Nz(Required flags cannot have default valuez(Optional flags must have a default valuez>Default value must be of the same type as the destination type)rrrrRrir3r/s rr0,SingleValueFlagDef._do_additional_validation}s ==&&(( !KLL)**,, !KLL  " " $ $););)Gd00$2E2E2GHH !abbI*H $rr(N)r@rr>rB)r rDrErFrGenumEnumrrrRrHrr+r0rKr(rrrrGs<  8E7L7LM4LN 0 crrc0^\rSrSrSrSU4SjjrSrU=r$) EnumFlagDefia8Definition for a flag that takes a value from an Enum. Sample usage: # This defines a flag that can be specified on the command line as: # --color=red flag = SingleValueFlagDef(name="color", enum_type=ColorsEnum, required=True) flag.add_argument_to_parser(argument_parser) c>[U[R5(d [S5eSU;a [ S5e[ US'SU;a [ S5eXS'SU;aUSH nU"U5 M OUVs/sHoDRPM snUS'[TU]$"U0UD6 g![a [ SR U55Sef=fs snf)Nz "enum_type" must be of type Enumr"z@Cannot set "parse_type" for EnumFlagDef; set "enum_type" insteadr#z?Cannot set "dest_type" for EnumFlagDef; set "enum_type" insteadr%z Invalid value in "choices": "{}") issubclassrr TypeErrorrrrmrrYrZ)r) enum_typerr_rr`s rrZEnumFlagDef.__init__s)TYY//>? ? 6 !_` `"| & ^_ _'{  I&]aL' 3< <)Q) rB)r rDrErFrGr+r0rKr(rrrrs . rrc,\rSrSrSrSSjrSSjrSrg) BooleanFlagDefizDefinition for a Boolean flag. A boolean flag is always optional with a default value of False. The flag does not take any values. Specifying the flag on the commandline will set it to True. cURb [S5eURb [S5eURb [S5eg)Nz*dest_type cannot be set for BooleanFlagDefz6parse_to_dest_type_fn cannot be set for BooleanFlagDefz(choices cannot be set for BooleanFlagDef)r#rr$r%r/s rr0(BooleanFlagDef._do_additional_validationsK >> %IJ J  % % 1UV V << #GH H $rc SUR-/nURbUSUR-/- n0nURbURUS'UR"U[[ SSSS.UD6 g)NrrrFr)rrorrdr)rr!r&rrrrs rr+%BooleanFlagDef.add_argument_to_parsersytyy ! ?? & S4??*+ +D == $!]]F6N +   rr(NrBr>)r rDrErFrGr0r+rKr(rrrrsI rr)rrzr@r)rrr@r)FN) rOr|rPrrQrrRrr@r)'rG __future__rrIrk dataclassesrtypingrrrrr google.generativeai.notebook.libr r rintfloat _PARSETYPESrLLMFnInputsSourceLLMFnOutputsSink _DESTTYPESrCrr dataclassABCrrSActionrUrrrrrrr(rrrs"#  88@:CeO$ II #xc C( () SM))"" $  [M:- . d#F)cggF)$F)X 5!5 55 5  5:47hoo47n5/x5/p,x,D d#?c?c$?cD"*$"*J# # L d#! W! $! r