
    
!i%                        S r SSKJr  SSKrSSKJr  SSKJr  SSKJr   " S S\R                  5      r
    SS	 jr      SS
 jrg)z2Utilities for exporting outputs from LLMFunctions.    )annotationsN)parsed_args_lib)py_utils)llmfn_outputsc                  ,    \ rS rSrSrSS jrSS jrSrg)	_PyVarOutputsSink   z.Sink that writes results to a Python variable.c                    Xl         g N	_var_name)selfvar_names     c/home/james-whalen/.local/lib/python3.13/site-packages/google/generativeai/notebook/output_utils.py__init___PyVarOutputsSink.__init__   s    !    c                n    [         R                  " U R                  [        R                  " U5      5        g r   )r   
set_py_varr   copydeepcopy)r   outputss     r   write_outputs_PyVarOutputsSink.write_outputs   s    DNNDMM',BCr   r   N)r   str)r   zllmfn_outputs.LLMFnOutputsBasereturnNone)__name__
__module____qualname____firstlineno____doc__r   r   __static_attributes__ r   r   r   r      s    8"Dr   r   c                    [         R                  " U 5      (       a7  [         R                  " U 5      n[        U[        R
                  5      (       a  U$ [        U 5      $ r   )r   
has_py_var
get_py_var
isinstancer   LLMFnOutputsSinkr   )r   datas     r   get_outputs_sink_from_py_varr+   $   sH    
 8$$""8,dM::;;KX&&r   c                    UR                    H  nU R                  U5        M     UR                   H  nU R                  U5        M     g)zWrites `results` to the sinks provided.

Args:
  results: The results to export.
  parsed_args: Arguments parsed from the command line.
N)r   exportsheets_output_names)resultsparsed_argssinks      r   write_to_outputsr2   0   s<     ##t $//t 0r   )r   r   r   zllmfn_outputs.LLMFnOutputsSink)r/   zllmfn_outputs.LLMFnOutputsr0   zparsed_args_lib.ParsedArgsr   r   )r"   
__future__r   r   google.generativeai.notebookr   r    google.generativeai.notebook.libr   r)   r   r+   r2   r$   r   r   <module>r6      sa    9 "  8 1 :D66 D	'	'#	''+ 
r   