o Zh6@sdZddlmZddlZddlmZmZmZmZm Z m Z m Z m Z ddl mZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZmZm Z m!Z!m"Z"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2ddl3m4Z4Gdddee#Z5dddddIddZ6dJd#d$Z7dKd(d)Z8de9ee9e"d*dd+dLd9d:Z:ed;dde9ee9e"d*dfdMdCdDZ;de9ee9e"d*dd*dEdNdHd<ZeZdZUdZded< ddd Ze ddddZd S)StructuredQueryOutputParserz-Output parser that parses a structured query.r ast_parsetextstrreturnrc szSddg}gdt||}|ddust|ddkr d|d<|ddks*|ds/d|d<n ||d|d<|dsC|ddtd ifdd |DWStyi}z td |d |d}~ww) Nqueryfilter)r+r,limitr Z NO_FILTERr-csi|] \}}|vr||qSr/).0kvZ allowed_keysr/^/var/www/html/lang_env/lib/python3.10/site-packages/langchain/chains/query_constructor/base.py >sz5StructuredQueryOutputParser.parse..z Parsing text z raised following error: r/) rlenr'getpopritems Exceptionr )selfr(Z expected_keysparseder/r3r4parse0s(    z!StructuredQueryOutputParser.parseNFallowed_comparatorsOptional[Sequence[Comparator]]allowed_operatorsOptional[Sequence[Operator]]allowed_attributesOptional[Sequence[str]] fix_invalidboolcs2|r d fdd }ntdj}||dS) z Create a structured query output parser from components. Args: allowed_comparators: allowed comparators allowed_operators: allowed operators Returns: a structured query output parser raw_filterr)r*Optional[FilterDirective]cs*tttt|}t|d}|SNr?rArC)r rrrr>fix_filter_directive)rGr,fixedrCr?rAr/r4r'Zsz>StructuredQueryOutputParser.from_components..ast_parserJ)r'N)rGr)r*rH)rr>)clsr?rArCrEr'r/rMr4from_componentsEs  z+StructuredQueryOutputParser.from_components)r(r)r*r)NNNF) r?r@rArBrCrDrErFr*r&)__name__ __module__ __qualname____doc____annotations__r> classmethodrOr/r/r/r4r&*s  r&rJr,rHr?r@rArBrCrDr*csssr|s |St|tr#r|jvrdSr!|jvr!dS|St|tr[r1|jvr1dSfdd|jD}|sAdSt|dkrT|jtj tj fvrT|dSt|j|dS|S)ahFix invalid filter directive. Args: filter: Filter directive to fix. allowed_comparators: allowed comparators. Defaults to all comparators. allowed_operators: allowed operators. Defaults to all operators. allowed_attributes: allowed attributes. Defaults to all attributes. Returns: Fixed filter directive. Nc s*g|]}|durttt|dqSrI)r rrK)r0argrMr/r4 s  z(fix_filter_directive..r)operator arguments) isinstancerZ comparator attributerrYrZr6rANDOR)r,r?rArCargsr/rMr4rKms<    rKinfo$Sequence[Union[AttributeInfo, dict]]r)cCsBi}|D] }t|}|||d<qtj|ddddddS)Nnameindent{{{}}})dictr8jsondumpsreplace)r`Z info_dictsiZi_dictr/r/r4_format_attribute_infos roinput_output_pairsSequence[Tuple[str, dict]] List[dict]cCsVg}t|D]"\}\}}tj|dddddd}|d||d}||q|S) zConstruct examples from input-output pairs. Args: input_output_pairs: Sequence of input-output pairs. Returns: List of examples. rcrdrfrgrhrirX)rnZ user_querystructured_request) enumeraterkrlrmappend)rpexamplesrn_inputoutputrsZexampler/r/r4construct_exampless  ryFrvr?rA enable_limit schema_promptdocument_contentsattribute_inforvOptional[Sequence]Sequence[Comparator]Sequence[Operator]r{rFr|Optional[BasePromptTemplate]kwargsrrcKs|rtnt}|p |}t|} |jd|d|d} |r=t|dtr=t|}t} t j| || d} t jt |dd} n|pD|rCt nt }t} tj| d} tjt |d|| d} td t|| d g| | d |S) a#Create query construction prompt. Args: document_contents: The contents of the document to be queried. attribute_info: A list of AttributeInfo objects describing the attributes of the document. examples: Optional list of examples to use for the chain. allowed_comparators: Sequence of allowed comparators. allowed_operators: Sequence of allowed operators. enable_limit: Whether to enable the limit operator. Defaults to False. schema_prompt: Prompt for describing query schema. Should have string input variables allowed_comparators and allowed_operators. kwargs: Additional named params to pass to FewShotPromptTemplate init. Returns: A prompt template that can be used to construct queries. z | )r?rAr)schemacontent attributesrX)rn)r)rnrrr+)rvexample_promptZinput_variablessuffixprefixNr/)r"rroformatjoinr[tupleryr$r!r#r6r rrrrrlist)r}r~rvr?rAr{r|rZdefault_schema_promptZ attribute_strrrrrr/r/r4get_query_constructor_prompts@   rz0.2.13load_query_constructor_runnablez1.0)Zsince alternativeZremovalllmr Optional[List]rc Ksnt|||||||d} g} |D]} | t| tr| jn| dqtj||| d} | | _td|| | d|S)aSLoad a query constructor chain. Args: llm: BaseLanguageModel to use for the chain. document_contents: The contents of the document to be queried. attribute_info: Sequence of attributes in the document. examples: Optional list of examples to use for the chain. allowed_comparators: Sequence of allowed comparators. Defaults to all Comparators. allowed_operators: Sequence of allowed operators. Defaults to all Operators. enable_limit: Whether to enable the limit operator. Defaults to False. schema_prompt: Prompt for describing query schema. Should have string input variables allowed_comparators and allowed_operators. **kwargs: Arbitrary named params to pass to LLMChain. Returns: A LLMChain that can be used to construct queries. rzrbrJ)rprompt output_parserNr/) rrur[r%rbr&rOrr) rr}r~rvr?rAr{r|rrrCainforr/r/r4load_query_constructor_chains*" r)rvr?rAr{r|rErErcKsht||f|||||d| } g} |D]} | t| tr| jn| dqtj||| |d} | |B| BS)aLoad a query constructor runnable chain. Args: llm: BaseLanguageModel to use for the chain. document_contents: Description of the page contents of the document to be queried. attribute_info: Sequence of attributes in the document. examples: Optional list of examples to use for the chain. allowed_comparators: Sequence of allowed comparators. Defaults to all Comparators. allowed_operators: Sequence of allowed operators. Defaults to all Operators. enable_limit: Whether to enable the limit operator. Defaults to False. schema_prompt: Prompt for describing query schema. Should have string input variables allowed_comparators and allowed_operators. fix_invalid: Whether to fix invalid filter directives by ignoring invalid operators, comparators and attributes. kwargs: Additional named params to pass to FewShotPromptTemplate init. Returns: A Runnable that can be used to construct queries. rzrb)r?rArCrE)rrur[r%rbr&rO)rr}r~rvr?rAr{r|rErrrCrrr/r/r4r?s0"  ) r,rHr?r@rArBrCrDr*rH)r`rar*r))rprqr*rr)r}r)r~rarvrr?rrArr{rFr|rrrr*r)rr r}r)r~rarvrr?rrArr{rFr|rrrr*r)rr r}r)r~rarvrr?rrArr{rFr|rrErFrrr*r)=rS __future__rrktypingrrrrrrr r Zlangchain_core._apir Zlangchain_core.exceptionsr Zlangchain_core.language_modelsr Zlangchain_core.output_parsersrZ"langchain_core.output_parsers.jsonrZlangchain_core.promptsrZlangchain_core.prompts.few_shotrZlangchain_core.runnablesrZlangchain_core.structured_queryrrrrrrZlangchain.chains.llmrZ)langchain.chains.query_constructor.parserrZ)langchain.chains.query_constructor.promptrrrrrr r!r"r#r$Z)langchain.chains.query_constructor.schemar%r&rKroryrrrrr/r/r/r4s` (          0 F : ? :