}h6dZddlmZddlZddlmZmZmZmZm 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/m0Z0m1Z1m2Z2ddl3m4Z4Gddee#Z5dddd d"dZ6d#dZ7d$dZ8de9ee9e"ddd d%dZ:edddde9ee9e"ddf d&dZ;de9ee9e"dddd  d'd!Z!*C"%wh;.fX6F#'x #'>>&2B#Cx ::g& 7D)"$*LLNHDAqa<6G1a4H H ' &B1#F  s0BB6 B0%B0*B60B66 C?CCNc`|r dfd }ntj}||S)z Create a structured query output parser from components. Args: allowed_comparators: allowed comparators allowed_operators: allowed operators Returns: a structured query output parser cttttj |}t |}|S)Nallowed_comparatorsallowed_operatorsallowed_attributes)r rrrr?fix_filter_directive) raw_filterr,fixedrErCrDs r>r)z>StructuredQueryOutputParser.from_components..ast_parseZs>h79K9KJ9WX,(;&7'9   rB)r))rGstrreturnOptional[FilterDirective])rr?)clsrCrDrE fix_invalidr)s ``` r>from_componentsz+StructuredQueryOutputParser.from_componentsEs>&   #$7"3#5e  Y''rI)r7rJrKr)NNNF) rCOptional[Sequence[Comparator]]rDOptional[Sequence[Operator]]rEOptional[Sequence[str]]rNboolrKr()__name__ __module__ __qualname____doc____annotations__r? classmethodrOr0rIr>r(r(*sf7S*?C:>6:! $(;$(8$(4 $(  $( % $($(rIr(rBc |s|s|r|s|St|tr$|r|j|vry|r|j|vry|St|tr|r|j |vry|j Dcgc]!}|ttt||||#}}|syt|dk(r1|j tjtjfvr|dSt |j |S|Scc}w)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. NrBr)operator arguments) isinstancer comparator attributerr\r]r rrFr1rANDOR)r,rCrDrEargargss r>rFrFms&!$59K  FJ ' 6#4#4_format_attribute_inforvs_J 0a)/ 6::f%&0 ::j + 3 3C > F FsD QQrIcg}t|D]Y\}\}}tj|djddjdd}|dz||d}|j |[|S) zConstruct examples from input-output pairs. Args: input_output_pairs: Sequence of input-output pairs. Returns: List of examples. rgrhrjrkrlrmr[)rt user_querystructured_request) enumeraterorprqappend)input_output_pairsexamplesrt_inputoutputryexamples r>construct_examplesrsH();< ! FF JJva ( 0 0d ; C CC N Q "4    ! OrIFr}rCrD enable_limit schema_promptc >|rtnt}|xs|}t|} |jdj |dj |} |r_t |dt rLt|}t} tj| || } tjt|dz} nR|xs|rtnt}t} tj| } t!jt|dz|| } t#d 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 | )rCrDr)schemacontent attributesr[)rt)r)rtrrr+)r}example_promptinput_variablessuffixprefixr0)r#rrvformatjoinr^tuplerr%r"r$r1r!rr rrrlist)document_contentsattribute_infor}rCrDrrkwargsdefault_schema_prompt attribute_strrrrrs r>get_query_constructor_promptrs-:%1 6K":%:M*>:M  ! !!JJ':;**%67"FJx{E2%h/6(//#4 ,22S]Q5FG #/ 5E (&&f5&&(ma):}  ! h%      rIz0.2.13load_query_constructor_runnablez1.0)since alternativeremovalc t|||||||} g} |D]2} | jt| tr | jn| d4t j ||| } | | _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. rrfrB)llmprompt output_parserr0) rr{r^r&rfr(rOrr) rrrr}rCrDrrrrrEainfors r>load_query_constructor_chainrsD*/+!#F !!$UM:EJJf   0??/+-@M )F  RF- R6 RRrI)r}rCrDrrrNc t||f|||||d| } g} |D]2} | jt| tr | jn| d4t j ||| |} | |z| zS)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. rrf)rCrDrErN)rr{r^r&rfr(rO)rrrr}rCrDrrrNrrrErrs r>rr?sD* /+!#   F !!$UM:EJJf   0??/+- @M C<- ''rI) r,rLrCrPrDrQrErRrKrL)rr$Sequence[Union[AttributeInfo, dict]]rKrJ)r|zSequence[Tuple[str, dict]]rKz List[dict])rrJrrr}Optional[Sequence]rCSequence[Comparator]rDSequence[Operator]rrSrOptional[BasePromptTemplate]rrrKr)rrrrJrrr}zOptional[List]rCrrDrrrSrrrrrKr)rrrrJrrr}rrCrrDrrrSrrrNrSrrrKr)=rW __future__rrotypingrrrrrr r r langchain_core._apir langchain_core.exceptionsr langchain_core.language_modelsrlangchain_core.output_parsersr"langchain_core.output_parsers.jsonrlangchain_core.promptsrlangchain_core.prompts.few_shotrlangchain_core.runnablesrlangchain_core.structured_queryrrrrrrlangchain.chains.llmr)langchain.chains.query_constructor.parserr)langchain.chains.query_constructor.promptrrrrr r!r"r#r$r%)langchain.chains.query_constructor.schemar&r(rFrvrrrrrr0rIr>rsbF" NNN*;<:L5A-*@   D@("2?"C@(L;?6:26 7 %7874 7 0 7  7tR6$(05j0A,1(O26<<8<! < . < * <<0<<<~ 1  $05j0A,1(O262S 2S2S92S 2S . 2S * 2S2S02S2S2S  2St$(05j0A,1(O267( 7(7(97( ! 7( . 7(*7(7(07(7(7(7(rI