
    ; i%                    x    d Z ddlmZ ddlZddlmZ ddlmZ ddlmZ  G d dej	                  Z
ddZddZdS )z2Utilities for exporting outputs from LLMFunctions.    )annotationsN)parsed_args_lib)py_utils)llmfn_outputsc                  "    e Zd ZdZddZdd	Zd
S )_PyVarOutputsSinkz.Sink that writes results to a Python variable.var_namestrc                    || _         d S N)	_var_name)selfr	   s     /var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/google/generativeai/notebook/output_utils.py__init__z_PyVarOutputsSink.__init__   s    !    outputsllmfn_outputs.LLMFnOutputsBasereturnNonec                ^    t          j        | j        t          j        |                     d S r   )r   
set_py_varr   copydeepcopy)r   r   s     r   write_outputsz_PyVarOutputsSink.write_outputs   s'    DNDM',B,BCCCCCr   N)r	   r
   )r   r   r   r   )__name__
__module____qualname____doc__r   r    r   r   r   r      sH        88" " " "D D D D D Dr   r   r	   r
   r   llmfn_outputs.LLMFnOutputsSinkc                    t          j        |           r0t          j        |           }t          |t          j                  r|S t          |           S r   )r   
has_py_var
get_py_var
isinstancer   LLMFnOutputsSinkr   )r	   datas     r   get_outputs_sink_from_py_varr'   $   sO    
 8$$ "8,,dM:;; 	KX&&&r   resultsllmfn_outputs.LLMFnOutputsparsed_argsparsed_args_lib.ParsedArgsr   c                    |j         D ]}|                     |           |j        D ]}|                     |           dS )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)r(   r*   sinks      r   write_to_outputsr0   0   s\     #  t/  t r   )r	   r
   r   r    )r(   r)   r*   r+   r   r   )r   
__future__r   r   google.generativeai.notebookr   r    google.generativeai.notebook.libr   r%   r   r'   r0   r   r   r   <module>r4      s    9 8 " " " " " "  8 8 8 8 8 8 1 1 1 1 1 1 : : : : : :D D D D D6 D D D	' 	' 	' 	'     r   