o Jjg6@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 NO_FILTERr-csi|] \}}|vr||qSr0).0kv allowed_keysr0_/var/www/html/zoom/venv/lib/python3.10/site-packages/langchain/chains/query_constructor/base.py >sz5StructuredQueryOutputParser.parse..z Parsing text z raised following error: r0) rlenr'getpopritems Exceptionr )selfr( expected_keysparseder0r4r6parse0s(    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}|SNrBrDrF)r rrrrAfix_filter_directive)rJr,fixedrFrBrDr0r6r'Zsz>StructuredQueryOutputParser.from_components..ast_parserM)r'N)rJr)r*rK)rrA)clsrBrDrFrHr'r0rPr6from_componentsEs  z+StructuredQueryOutputParser.from_components)r(r)r*r)NNNF) rBrCrDrErFrGrHrIr*r&)__name__ __module__ __qualname____doc____annotations__rA classmethodrRr0r0r0r6r&*s  r&rMr,rKrBrCrDrErFrGr*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|dqSrL)r rrN)r1argrPr0r6 s  z(fix_filter_directive..r)operator arguments) isinstancer comparator attributerr\r]r8rANDOR)r,rBrDrFargsr0rPr6rNms<    rNinfo$Sequence[Union[AttributeInfo, dict]]r)cCsBi}|D] }t|}|||d<qtj|ddddddS)Nnameindent{{{}}})dictr:jsondumpsreplace)rd info_dictsii_dictr0r0r6_format_attribute_infos ruinput_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. rgrhrjrkrlrmr[)rs user_querystructured_request) enumeraterorprqappend)rvexamplesrs_inputoutputrzexampler0r0r6construct_exampless  rFr}rBrD enable_limit schema_promptdocument_contentsattribute_infor}Optional[Sequence]Sequence[Comparator]Sequence[Operator]rrIrOptional[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 | )rBrDr)schemacontent attributesr[)rs)r)rsrrr+)r}example_promptinput_variablessuffixprefixNr0)r"rruformatjoinr^tuplerr$r!r#r8r rrrrrlist)rrr}rBrDrrrdefault_schema_prompt attribute_strrrrrr0r0r6get_query_constructor_prompts@   rz0.2.13load_query_constructor_runnablez1.0)since alternativeremovalllmr 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. rrfrM)rprompt output_parserNr0) rr|r^r%rfr&rRrr) rrrr}rBrDrrrrrFainforr0r0r6load_query_constructor_chains*" r)r}rBrDrrrHrHrcKsht||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. rrf)rBrDrFrH)rr|r^r%rfr&rR)rrrr}rBrDrrrHrrrFrrr0r0r6r?s0"  ) r,rKrBrCrDrErFrGr*rK)rdrer*r))rvrwr*rx)rr)rrer}rrBrrDrrrIrrrrr*r)rr rr)rrer}rrBrrDrrrIrrrrr*r)rr rr)rrer}rrBrrDrrrIrrrHrIrrr*r)=rV __future__rrotypingrrrrrrr r langchain_core._apir langchain_core.exceptionsr langchain_core.language_modelsr langchain_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.promptrrrrrr r!r"r#r$)langchain.chains.query_constructor.schemar%r&rNrurrrrrr0r0r0r6s` (          0 F : ? :