f3fidZddlmZddlmZmZddlmZddlm Z m Z ddl m Z e rddl mZGdd eZdd ZGd d e ZGd deeZGddeeZGddeeZGddeZGddeZGddeZy)z7Internal representation of a structured query language.) annotations)ABCabstractmethod)Enum) TYPE_CHECKINGAny) BaseModel)SequencecpeZdZUdZdZded< dZded< d dZed dZ ed d Z edd Z y)Visitorz=Defines interface for IR translation using a visitor pattern.NzSequence[Comparator] | Noneallowed_comparatorszSequence[Operator] | Noneallowed_operatorsc$t|tr7|j+||jvrd|d|j}t|t|tr9|j ,||j vrd|d|j }t|yyy)NzReceived disallowed operator z. Allowed comparators are zReceived disallowed comparator ) isinstanceOperatorr ValueError Comparatorr )selffuncmsgs ]/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_core/structured_query.py_validate_funczVisitor._validate_funcs tX &&&2D2220v6##'#9#9":< S/ ! tZ (((4D4442$8##'#;#;"<> S/ ! 55 )cy)z^Translate an Operation. Args: operation: Operation to translate. N)r operations rvisit_operationzVisitor.visit_operation-rcy)z`Translate a Comparison. Args: comparison: Comparison to translate. Nr)r comparisons rvisit_comparisonzVisitor.visit_comparison5rrcy)zpTranslate a StructuredQuery. Args: structured_query: StructuredQuery to translate. Nr)rstructured_querys rvisit_structured_queryzVisitor.visit_structured_query=rr)rzOperator | ComparatorreturnNone)r Operationr%r)r Comparisonr%r)r#StructuredQueryr%r) __name__ __module__ __qualname____doc__r __annotations__rrrrr!r$rrrr r sdG7;4;.3707,",      rr cd}t|D]D\}}|jr|dk7r|d|jzz }2||jz }F|S)zConvert a name into snake_case.r_) enumerateisupperlower)name snake_caseichars r_to_snake_caser9FsZJT?'4 <<>a1f # , ,J $**, &J ' rceZdZdZddZy)ExprzBase class for all expressions.cft|dt|jj|S)zAccept a visitor. Args: visitor: visitor to accept. Returns: result of visiting. visit_)getattrr9 __class__r*)rvisitors racceptz Expr.acceptTs5Tww&8O8O)P(Q RS   rN)r@r r%r)r*r+r,r-rArrrr;r;Qs )  rr;ceZdZdZdZdZdZy)rzEnumerator of the operations.andornotN)r*r+r,r-ANDORNOTrrrrrbs' C B Crrc8eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z y ) rz'Enumerator of the comparison operators.eqnegtgteltltecontainlikeinninN)r*r+r,r-EQNEGTGTELTLTECONTAINLIKEINNINrrrrrjs71 B B B C B CG D B CrrceZdZdZy)FilterDirectivezFiltering expression.N)r*r+r,r-rrrr_r_ysrr_c^eZdZUdZded< ded< ded< d fd ZxZS) r(zComparison to a value.r comparatorstr attributervaluec ,t|d|||d|y)zCreate a Comparison. Args: comparator: The comparator to use. attribute: The attribute to compare. value: The value to compare to. )rarcrdNrsuper__init__)rrarcrdkwargsr?s rrhzComparison.__init__s&  !Ye GM r) rarrcrbrdrrirr%r&r*r+r,r-r.rh __classcell__r?s@rr(r(}sQ  N# J"  $  14  =@  LO     rr(cNeZdZUdZded< ded< dfd ZxZS)r'z(Logical operation over other directives.roperatorlist[FilterDirective] argumentsc *t|d||d|y)zCreate an Operation. Args: operator: The operator to use. arguments: The arguments to the operator. )rnrpNrrf)rrnrprir?s rrhzOperation.__init__s J(iJ6Jr)rnrrprorirr%r&rjrls@rr'r'sH2$$( K  K-B KNQ K  K Krr'c`eZdZUdZded< ded< ded< d d fd ZxZS) r)zStructured query.rbqueryFilterDirective | Nonefilter int | Nonelimitc ,t|d|||d|y)zCreate a StructuredQuery. Args: query: The query string. filter: The filtering expression. limit: The limit on the number of results. )rsrurwNrrf)rrsrurwrir?s rrhzStructuredQuery.__init__s KuV5KFKr)N) rsrbrurtrwrvrirr%r&rjrls@rr)r)sf J "" ) ! LL'L L  L  LLrr)N)r5rbr%rb)r- __future__rabcrrenumrtypingrrpydanticr collections.abcr r r9r;rbrrr_r(r'r)rrrrs="#%(4 c4 n 9 "sD d  dC   4KK*LdLr