o
    Zh                     @  s   d Z ddlmZ ddlZddlZddlmZmZmZ ddl	m
Z
 ddlmZ ddlmZ ddlmZ dd	lmZ eee  Zeeegef ZG d
d dejZejddG dd dZdS )zResults from parsing the commandline.

This module separates the results from commandline parsing from the parser
itself so that classes that operate on the results (e.g. the subclasses of
Command) do not have to depend on the commandline parser as well.
    )annotationsN)AnyCallableSequence)model_registry)llm_function)llmfn_inputs_source)llmfn_outputs)modelc                   @  s   e Zd ZdZdZdZdZdS )CommandNameruncompilecompareevalN)__name__
__module____qualname__ZRUN_CMDZCOMPILE_CMDZCOMPARE_CMDZEVAL_CMD r   r   c/var/www/html/lang_env/lib/python3.10/site-packages/google/generativeai/notebook/parsed_args_lib.pyr   +   s
    r   T)frozenc                   @  s   e Zd ZU dZded< ded< dZded< d	Zd
ed< eje	dZ
ded< eje	dZded< eje	dZded< eje	dZded< dZded< dZded< dZded< eje	dZded< eje	dZded< dS )
ParsedArgsz(The results of parsing the command line.r   cmdzmodel_lib.ModelArgumentsZ
model_argsNzmodel_registry.ModelName | None
model_typeFboolunique)default_factoryz/Sequence[llmfn_inputs_source.LLMFnInputsSource]inputssheets_input_namesz(Sequence[llmfn_outputs.LLMFnOutputsSink]outputssheets_output_namesz
str | Nonecompile_save_namez+tuple[str, llm_function.LLMFunction] | Nonelhs_name_and_fnrhs_name_and_fnz)Sequence[tuple[str, TextResultCompareFn]]
compare_fnzSequence[str]ground_truth)r   r   r   __doc____annotations__r   r   dataclassesfieldlistr   r   r   r   r    r!   r"   r#   r$   r   r   r   r   r   2   s*   
 r   )r%   
__future__r   r'   enumtypingr   r   r   Zgoogle.generativeai.notebookr   Z google.generativeai.notebook.libr   r   r	   r
   Z	model_libstrZPostProcessingTokensZTextResultCompareFnEnumr   	dataclassr   r   r   r   r   <module>   s   
