; i~CdZddlmZddlZddlZddlZddlZddlmZm Z m Z m Z m Z ddl mZddl mZe eeefZe eeje ee eegeffe eejejfZe egefZd)d Zd*d ZejdGddejZ d+d,dZGddejZ GddejZ!Gdd ejZ"ejdGd!d"eZ#Gd#d$e#Z$Gd%d&eZ%ejdGd'd(eZ&dS)-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_outputsx type[Any]returnstrcP |jS#t$rt|cYSwxYwN)__name__AttributeErrorrr s }/var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/google/generativeai/notebook/flag_def.py_get_type_namer?s9z 1vv s  %%namec^|std|ddkrtd|S)z.Validation for long and short names for flags.zCannot be emptyr-zCannot start with dash) ValueError)rs r_validate_flag_namerFs: ,*+++ Aw#~~1222 KT)frozenceZdZUdZded<dZded<dZded <eZd ed <dZ d ed <dZ ded<dZ ded<dZ ded<e jddZe jddZd dZd!dZdZdS)"FlagDefaAbstract 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_msgparserargparse.ArgumentParserr NonecdS)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)selfr(s radd_argument_to_parserzFlagDef.add_argument_to_parserqrcdS)z.For child classes to do additional validation.Nr,r-s r_do_additional_validationz!FlagDef._do_additional_validation|r/rtype[_DESTTYPES]c,|j|jn|jS)z!Returns the final converted type.)r$r#r1s r_get_dest_typezFlagDef._get_dest_types"&."8tdnLr_PARSEFNcf|j|jS|}||jkrdS|S)z;Returns a function to convert from parse_type to dest_type.Nc|Srr,rs rz4FlagDef._get_parse_to_dest_type_fn..sQr)r%r5r#)r-r$s r_get_parse_to_dest_type_fnz"FlagDef._get_parse_to_dest_type_fnsB  % 1- -''))  ' ';  rct|j|jt|j|dSr)rrr"r2r1s r __post_init__zFlagDef.__post_init__sBDI&&& ? &  0 0 0 &&(((((rr(r)r r*r r*)r r3)r r6)r __module__ __qualname____doc____annotations__r r"rr#r$r%r&r'abcabstractmethodr.r2r5r:r<r,rrrrOs$(IIIH!J!!!!$'J'''')-I-----11111(,G,,,,H     ====MMMM    )))))rrF namespaceargparse.Namespacedest has_defaultr default_valuercVt||sdS|sdSt|||kS)a Returns 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)rErGrHrIs r_has_non_default_valuerMs?$ 9d # #u t 9d # #} 44rc0eZdZdZdfd Z dddZxZS)_SingleValueStoreActionzCustom Action for storing a value in an argparse.Namespace. This action checks that the flag is specified at-most once. r$r r%r6c Ztj||fi|||_||_dSrsuper__init__ _dest_type_parse_to_dest_type_fnr-option_stringsrGr$r%kwargs __class__s rrSz _SingleValueStoreAction.__init__; 88888#&;###rNr(r)rErFvaluesstr | Sequence[Any] | None option_stringr!c t|tst|trJt||jt |dt |dr(tj|d | | |d}n\#t$rO}tj|d |dtt||d}~wwxYwt||jsOtd t|jtt|t!||j|dS)NdefaultrHrICannot set {} more than oncer!Error with value "{}", got {}: {}+Converted to wrong type, expected {} got {}) isinstancerbytesrMrGrKrLargparse ArgumentErrorformatrU ExceptionrtyperT RuntimeErrorsetattr)r-r(rEr[r]converted_valuees r__call__z _SingleValueStoreAction.__call__sfc**L:fe3L3LLLL !  Ii00!$ 22     e (/M/T/TUb/c/cdd d "99&)DDOO   (3::6!9nUYZ[U\U\F]F]_`aa   /4?;; =DD"4?33"4#8#899   49o66666s B&& C?0A C::C?r$r r%r6rr(r)rErFr[r\r]r!rr?r@rArSro __classcell__rYs@rrOrOse < < < < < <"%) "7"7"7"7"7"7"7"7"7rrOc0eZdZdZdfd Z dddZxZS)_MultiValuesAppendActionzCustom Action for appending values in an argparse.Namespace. This action checks that the flag is specified at-most once. r$r r%r6c Ztj||fi|||_||_dSrrQrVs rrSz!_MultiValuesAppendAction.__init__rZrNr(r)rErFr[r\r]r!c t|tst|trJt||j}|r(t j|d||D]} ||}n\#t$rO}t j|d|dtt||d}~wwxYwt||j s5td|j t|||vr(t j|d|||dS)NrarbrrczDuplicate values "{}")rdrrerLrGrfrgrhrUrirrjrTrkappend) r-r(rEr[r] curr_valuevaluermrns rroz!_MultiValuesAppendAction.__call__sfc**L:fe3L3LLLLY 22  e(/M/T/TUb/c/cdd d / /E "&"="=e"D"D   ,7>>q >$q''#:#:A ot?? "AHHo)>)> *,,,T3J3Q3QRW3X3XYYY   o . . . .) / /s2B C!A CC!rprrqrrrts@rrvrvse < < < < < <"%) #/#/#/#/#/#/#/#/#/rrvc.eZdZdZfdZ d dd ZxZS)_BooleanValueStoreActionzCustom 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 >tj||fi|dSr)rRrS)r-rWrGrXrYs rrSz!_BooleanValueStoreAction.__init__-s, 8888888rNr(r)rErFr[r\r]r!ct||jddr(tj|d|t ||jddS)NTFr`ra)rMrGrfrgrhrl)r-r(rEr[r]s rroz!_BooleanValueStoreAction.__call__5si "  I     e (/M/T/TUb/c/cdd d 49d+++++rrrqrrrts@rr}r}%s`99999%) ,,,,,,,,,rr}cjeZdZUdZGddejZejZde d<ddZ dd Z dd Z dS)SingleValueFlagDefaoDefinition 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. ceZdZdZdZdS) SingleValueFlagDef._DefaultValuezSpecial 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". N)rr?r@rANOT_SETr,rr _DefaultValuerUs  rrz!_DESTTYPES | _DefaultValue | NonerIr rc6|jtjjkS)z2Returns whether `default_value` has been provided.)rIrrrr1s r_has_default_valuez%SingleValueFlagDef._has_default_valueas!%7%E%MMMrr(r)r*c Xd|jzg}|j|d|jzgz }i}|r |j|d<|j |j|d<|j |j|d<|j|t|j| | |j dd|dS)N--rr_r&help)actionrjr$r%r nargs) rr"rrIr&r' add_argumentrOr#r5r:r r-r(argsrXs rr.z)SingleValueFlagDef.add_argument_to_parseresty ! ? & S4?*+ +D  " " $ $ 3 $ 2F9  < # $ F9  = $!]F6N *))++"&"A"A"C"C]  rcL|jr$|rtdn#|std|r=|j8t |j|stddSdSdS)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)r rrrIrdr5r1s rr2z,SingleValueFlagDef._do_additional_validation}s = M&&(( M !KLLL M**,, M !KLLL  " " $ $ c);)Gd0$2E2E2G2GHH c !abbb c c)G)G c crN)r rr=r>) rr?r@rAenumEnumrrrIrBrr.r2r,rrrrGs   8E7LMLLLLNNNN    0 c c c c c crrc$eZdZdZdfd ZxZS) EnumFlagDefaTDefinition 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) enum_typetype[enum.Enum]ct|tjstdd|vrt dt |d<d|vrt d||d<d|vrI|dD]?} ||#t$r$t d|dwxYwnd|D|d<tj|i|dS) 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": "{}"cg|] }|j Sr,)r{).0r s r z(EnumFlagDef.__init__..s < < ?? ? 6 ! !_`` `"| & ^__ _'{   I& ] ]]IaLLLL!]]]$%G%N%Nq%Q%QRRX\\] ] != <) < < )rr?r@rAr.r2r,rrrrsF    .      rrc"eZdZdZd dZd dZdS) BooleanFlagDefzDefinition 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. r r*c|jtd|jtd|jtddS)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&r1s rr2z(BooleanFlagDef._do_additional_validationsP > %IJJ J  % 1UVV V < #GHH H $ #rr(r)c d|jzg}|j|d|jzgz }i}|j |j|d<|j|tt dddd|dS)NrrrFr)rrjr r_r)rr"r'rr}rrs rr.z%BooleanFlagDef.add_argument_to_parsersty ! ? & S4?*+ +D = $!]F6N +        rNr>r=)rr?r@rAr2r.r,rrrrsJIIII      rr)r r r r)rrr r)FN) rErFrGrrHrrIrr r)'rA __future__rrCrf dataclassesrtypingrrrrr google.generativeai.notebook.libr r rintfloat _PARSETYPESrLLMFnInputsSourceLLMFnOutputsSink _DESTTYPESr6rr dataclassABCrrMActionrOrvr}rrrrr,rrrs"#"""""  88888888888888@@@@@@::::::CeO$ I #xc C( () SM)" $  [M:- .d###F)F)F)F)F)cgF)F)$#F)X 55555:4747474747ho474747n5/5/5/5/5/x5/5/5/p,,,,,x,,,Dd###?c?c?c?c?c?c?c$#?cD"*"*"*"*"*$"*"*"*J# # # # # # # # Ld###! ! ! ! ! W! ! $#! ! ! r