
    rh%                        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y)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y)_PyVarOutputsSinkz.Sink that writes results to a Python variable.c                    || _         y N)	_var_name)selfvar_names     |/var/www/html/ai-insurance-compliance-backend/venv/lib/python3.12/site-packages/google/generativeai/notebook/output_utils.py__init__z_PyVarOutputsSink.__init__   s	    !    c                j    t        j                  | j                  t        j                  |             y r
   )r   
set_py_varr   copydeepcopy)r   outputss     r   write_outputsz_PyVarOutputsSink.write_outputs   s    DNNDMM',BCr   N)r   str)r   zllmfn_outputs.LLMFnOutputsBasereturnNone)__name__
__module____qualname____doc__r   r    r   r   r   r      s    8"Dr   r   c                    t        j                  |       r1t        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%   $   sB    
 8$""8,dM::;KX&&r   c                    |j                   D ]  }| j                  |        |j                  D ]  }| j                  |        y)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_outputsr,   0   sF     ## t// tr   )r   r   r   zllmfn_outputs.LLMFnOutputsSink)r)   zllmfn_outputs.LLMFnOutputsr*   zparsed_args_lib.ParsedArgsr   r   )r   
__future__r   r   google.generativeai.notebookr   r    google.generativeai.notebook.libr   r#   r   r%   r,   r   r   r   <module>r0      sa    9 "  8 1 :D66 D	'	'#	''+ 
r   