o j6h~C@sdZddlmZddlZddlZddlZddlZddlmZm Z m Z m Z m Z ddl mZddl mZe eeefZe eeje ee eegeffe eejejfZe egefZd,d d Zd-ddZejddGdddejZ  d.d/ddZGdddejZ Gd d!d!ejZ!Gd"d#d#ejZ"ejddGd$d%d%eZ#Gd&d'd'e#Z$Gd(d)d)eZ%ejddGd*d+d+eZ&dS)0aYClasses 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]returnstrcCs$z|jWStyt|YSwN)__name__AttributeErrorr r rt/var/www/html/chefvision.cloud.itp360.com/venv/lib/python3.10/site-packages/google/generativeai/notebook/flag_def.py_get_type_name?s   rnamecCs$|std|ddkrtd|S)z.Validation for long and short names for flags.zCannot be emptyr-zCannot start with dash) ValueError)rrrr_validate_flag_nameFs  rT)frozenc@seZdZUdZded<dZded<dZded <eZd ed <dZ d ed <dZ ded<dZ ded<dZ ded<e jd#ddZe jd$ddZd%ddZd&dd Zd!d"ZdS)'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. r rFboolrequiredN str | None short_nameztype[_PARSETYPES] parse_typeztype[_DESTTYPES] | None dest_typez_PARSEFN | Noneparse_to_dest_type_fnzlist[_PARSETYPES] | Nonechoiceshelp_msgparserargparse.ArgumentParserr NonecCdS)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. Nr)selfr$rrradd_argument_to_parserqzFlagDef.add_argument_to_parsercCr')z.For child classes to do additional validation.Nrr(rrr_do_additional_validation|r*z!FlagDef._do_additional_validationtype[_DESTTYPES]cCs|jdur|jS|jS)z!Returns the final converted type.N)r rr+rrr_get_dest_typeszFlagDef._get_dest_type_PARSEFNcCs.|jdur|jS|}||jkrddS|S)z;Returns a function to convert from parse_type to dest_type.NcSs|Srrrrrrr*z4FlagDef._get_parse_to_dest_type_fn..)r!r.r)r(r rrr_get_parse_to_dest_type_fns  z"FlagDef._get_parse_to_dest_type_fncCs*t|j|jdurt|j|dSr)rrrr,r+rrr __post_init__s    zFlagDef.__post_init__r$r%r r&r r&)r r-)r r/)r __module__ __qualname____doc____annotations__rrr rr r!r"r#abcabstractmethodr)r,r.r1r2rrrrrOs"             rF namespaceargparse.Namespacedest has_defaultr default_valuercCs$t||sdS|s dSt|||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)r;r=r>r?rrr_has_non_default_values rBc0eZdZdZdfdd Z ddddZZS)_SingleValueStoreActionzCustom Action for storing a value in an argparse.Namespace. This action checks that the flag is specified at-most once. r r r!r/c &tj||fi|||_||_dSrsuper__init__ _dest_type_parse_to_dest_type_fnr(option_stringsr=r r!kwargs __class__rrrH z _SingleValueStoreAction.__init__Nr$r%r;r<valuesstr | Sequence[Any] | None option_stringrc Cst|ts t|tr Jt||jt|dt|ddr$t|d |z | |d}Wnt yJ}zt|d |dt t ||d}~wwt||js`td t |jt t |t||j|dS)Ndefaultr>r?Cannot set {} more than oncer!Error with value "{}", got {}: {}+Converted to wrong type, expected {} got {}) isinstancer bytesrBr=r@rAargparse ArgumentErrorformatrJ ExceptionrtyperI RuntimeErrorsetattr)r(r$r;rQrSconverted_valueerrr__call__s2   z _SingleValueStoreAction.__call__r r r!r/rr$r%r;r<rQrRrSrrr5r6r7rHrd __classcell__rrrNrrD rDcrC)_MultiValuesAppendActionzCustom Action for appending values in an argparse.Namespace. This action checks that the flag is specified at-most once. r r r!r/c rErrFrKrNrrrHrPz!_MultiValuesAppendAction.__init__Nr$r%r;r<rQrRrSrc Cst|ts t|tr Jt||j}|rt|d||D]J}z||}Wnt yE}zt|d|dt t ||d}~wwt||j sWt d|j t |||vrdt|d|||qdS)NrVrWrrXzDuplicate values "{}")rYr rZrAr=r[r\r]rJr^rr_rIr`append) r(r$r;rQrS curr_valuevaluerbrcrrrrds4    z!_MultiValuesAppendAction.__call__rerrfrgrrrNrrjrirjcs.eZdZdZfddZ ddd dZZS)_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 stj||fi|dSr)rGrH)r(rLr=rMrNrrrH-sz!_BooleanValueStoreAction.__init__Nr$r%r;r<rQrRrSrcCs6t||jdddrt|d|t||jddS)NTFrUrV)rBr=r[r\r]ra)r(r$r;rQrSrrrrd5sz!_BooleanValueStoreAction.__call__rrfrgrrrNrrn%s   rnc@sPeZdZUdZGdddejZejZde d<ddd Z dd dZ dddZ 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. c@seZdZdZdZdS)z 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)rr5r6r7NOT_SETrrrr _DefaultValueUsrqz!_DESTTYPES | _DefaultValue | Noner?r rcCs|jtjjkS)z2Returns whether `default_value` has been provided.)r?rorqrpr+rrr_has_default_valueasz%SingleValueFlagDef._has_default_valuer$r%r&c Csd|jg}|jdur|d|jg7}i}|r|j|d<|jdur(|j|d<|jdur2|j|d<|j|t|j| | |j dd|dS)N--rrTr"help)actionr_r r!rnargs) rrrrr?r"r# add_argumentrDrr.r1rr(r$argsrMrrrr)es*        z)SingleValueFlagDef.add_argument_to_parsercCs^|jr |r tdn|std|r)|jdur+t|j|s-tddSdSdS)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)rrrrr?rYr.r+rrrr,}sz,SingleValueFlagDef._do_additional_validationN)r rr3r4) rr5r6r7enumEnumrqrpr?r8rrr)r,rrrrroGs    rocs"eZdZdZdfdd ZZS) 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]c st|tjs tdd|vrtdt|d<d|vrtd||d<d|vrC|dD]}z||Wq*tyAtd|dwn dd |D|d<tj|i|dS) Nz "enum_type" must be of type Enumrz@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": "{}"cSsg|]}|jqSr)rm).0r rrr sz(EnumFlagDef.__init__..) issubclassr{r| TypeErrorrr r]rGrH)r(r~rzrMr rNrrrHs$    zEnumFlagDef.__init__)r~r)rr5r6r7rHrhrrrNrr}s r}c@s$eZdZdZd ddZd dd Zd S) MultiValuesFlagDefa6Definition for a flag that takes multiple values. Sample usage: # This defines a flag that can be specified on the command line as: # --colors=red green blue flag = MultiValuesFlagDef(name="colors", parse_type=str, required=True) flag.add_argument_to_parser(argument_parser) r$r%r r&c Csd|jg}|jdur|d|jg7}i}|jdur|j|d<|jdur)|j|d<|j|t|j|||j gdd|dS)Nrsrr"rt+)rvr_r r!rrTrw) rrr"r#rxrjrr.r1rryrrrr)s(       z)MultiValuesFlagDef.add_argument_to_parsercCsdSrrr+rrrr,sz,MultiValuesFlagDef._do_additional_validationNr3r4)rr5r6r7r)r,rrrrrs rc@s$eZdZdZd ddZd dd Zd S) 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&cCs:|jdur td|jdurtd|jdurtddS)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+rrrr,s   z(BooleanFlagDef._do_additional_validationr$r%cCs^d|jg}|jdur|d|jg7}i}|jdur|j|d<|j|ttdddd|dS)NrsrrtFr)rvr_rrTrw)rrr#rxrnrryrrrr)s     z%BooleanFlagDef.add_argument_to_parserNr4r3)rr5r6r7r,r)rrrrrs r)r r r r )rr r r )FN) r;r<r=r r>rr?rr r)'r7 __future__rr9r[ dataclassesr{typingrrrrr google.generativeai.notebook.librr r intfloat _PARSETYPESr|LLMFnInputsSourceLLMFnOutputsSink _DESTTYPESr/rr dataclassABCrrBActionrDrjrnror}rrrrrrsH      L 78 "B% &