U DAf@sddlZddlZddlmZmZmZmZmZddlZ ddl m Z m Z ddl m Z ddlmZddlmZmZmZddlmZmZmZmZGdd d eZeGd d d ZdS) N)AnyCallableDictOptionalSet)TemplateSyntaxErrormeta)SandboxedEnvironment) TypeAlias) componentdefault_from_dictdefault_to_dict)deserialize_callabledeserialize_typeserialize_callableserialize_typec@seZdZdZdS)OutputAdaptationExceptionzAException raised when there is an error during output adaptation.N)__name__ __module__ __qualname____doc__rrQ/tmp/pip-unpacked-wheel-z752163x/haystack/components/converters/output_adapter.pyrsrc@s~eZdZdZdeeeeeefdddZ ddZ eee fdd d Z e eee fdd d d ZeeedddZdS) OutputAdaptera Adapts output of a Component using Jinja templates. Usage example: ```python from haystack import Document from haystack.components.converters import OutputAdapter adapter = OutputAdapter(template="{{ documents[0].content }}", output_type=str) documents = [Document(content="Test content"] result = adapter.run(documents=documents) assert result["output"] == "Test content" ``` Ntemplate output_typecustom_filtersc Cs|pi|_t}ttjjd|_z|j|||_Wn8t k rp}zt d|d||W5d}~XYnX|j D]\}}||jj |<q|| |j}||tj|fdd|Dtj|fd|i||_dS)aK Create an OutputAdapter component. :param template: A Jinja template that defines how to adapt the input data. The variables in the template define the input of this instance. e.g. With this template: ``` {{ documents[0].content }} ``` The Component input will be `documents`. :param output_type: The type of output this instance will return. :param custom_filters: A dictionary of custom Jinja filters used in the template. )Z undefinedzInvalid Jinja template 'z': NcSsi|] }|tqSr)r).0varrrr Nsz*OutputAdapter.__init__..output)rsetr jinja2runtimeZStrictUndefined_envparserr ValueErroritemsfilters_extract_variablesupdater Zset_input_typesZset_output_typesr) selfrrrZ input_typesename filter_funcZroute_input_namesrrr__init__(s   (  zOutputAdapter.__init__c Ks|s td|jD]\}}||jj|<qi}zj|j|j}|jf|}t|t j j rpt d|jd|t tt|}W5QRX||d<Wn@tk r}z"t d|jd|d||W5d}~XYnX|S) ab Renders the Jinja template with the provided inputs. :param kwargs: Must contain all variables used in the `template` string. :returns: A dictionary with the following keys: - `output`: Rendered Jinja template. :raises OutputAdaptationException: If template rendering fails. z,No input data provided for output adaptationz#Undefined variable in the template z ; kwargs: r!zError adapting z with z: N)r'rr(r%r)Z from_stringrrender isinstancer#r$ Undefinedr contextlibsuppress Exceptionast literal_eval)r,kwargsr.r/Zadapted_outputsZadapted_output_templateZ output_resultr-rrrrunRs    0zOutputAdapter.run)returncCs,dd|jD}t||jt|j|dS)z{ Serializes the component to a dictionary. :returns: Dictionary with serialized data. cSsi|]\}}|t|qSr)r)rr.r/rrrr }sz)OutputAdapter.to_dict..r)rr(r rrr)r,Z se_filtersrrrto_dictvszOutputAdapter.to_dict)datar;cCsX|di}t|d|d<|diD] \}}|r@t|nd|d|<q,t||S)z Deserializes the component from a dictionary. :param data: The dictionary to deserialize from. :returns: The deserialized component. Zinit_parametersrrN)getrr(rr )clsr=Z init_paramsr.r/rrr from_dicts zOutputAdapter.from_dict)envr;cCs||j}t|S)z Extracts all variables from a list of Jinja template strings. :param env: A Jinja native environment. :return: A set of variable names extracted from the template strings. )r&rrZfind_undeclared_variables)r,rAr7rrrr*s z OutputAdapter._extract_variables)N)rrrrstrr rrrr0r:rr< classmethodr@r rr*rrrrrs *$ r)r7r4typingrrrrrZjinja2.runtimer#rrZjinja2.sandboxr Ztyping_extensionsr Zhaystackr r r Zhaystack.utilsrrrrr6rrrrrrs