?hv VdZddlZddlmZmZddlmZddlmZddl m Z ddl m Z m Z mZddlmZdd lmZdd lmZdd lmZmZmZhd Zhd ZhdZej;ej;eZej;eZgdZ GddeZ!GddeZ"GddeZ#Gdde#Z$Gdde#Z%Gdde#Z&Gdde#Z'Gdde#Z(Gd d!e(Z)Gd"d#e(Z*Gd$d%e(Z+Gd&d'e(Z,Gd(d)e(Z-Gd*d+e(Z.Gd,d-e#Z/Gd.d/e#Z0d0e1d1e1fd2Z2d3e3d1e1fd4Z4d3e d0e1d1dfd5Z5d3e d0e1d1dfd6Z6d3e7d1e1fd7Z8d3e9d1e1fd8Z:d9e7d1dfd:Z;d;e7d1dfd<Ze1d3e d?ee e>e>fe>fd1dfd@Z?d;e7d1e1fdAZ@dBe3d1e1fdCZAy)Dzr GraphQL filters for `Get` and `Aggregate` commands. GraphQL abstract class for GraphQL commands to inherit from. N)ABCabstractmethod)deepcopy)Enumdumps)AnyTupleUnion)ConnectionError) Connection)FILTER_BEACON_V14_CLS_NS_W) get_vector _sanitize_str_decode_json_response_dict> valueIntList valueTextListvalueNumberListvalueStringListvalueBooleanList> valueIntArrayvalueTextArrayvalueNumberArrayvalueStringArrayvalueBooleanArray>valueInt valueDate valueText valueNumber valueString valueBoolean valueGeoRange) And ContainsAll ContainsAnyEqual GreaterThanGreaterThanEqualIsNullLessThan LessThanEqualLikeNotEqualOrWithinGeoRangec$eZdZdZdZdZdZdZdZy) MediaTypeimageaudiovideothermaldepthimuN) __name__ __module__ __qualname__IMAGEAUDIOVIDEOTHERMALDEPTHIMUQ/home/chris/cleankitchens-env/lib/python3.12/site-packages/weaviate/gql/filter.pyr1r1@s E E EG E CrBr1c>eZdZdZdefdZedefdZde fdZ y)GraphQLzM A base abstract class for GraphQL commands, such as Get, Aggregate. connectionc||_y)z Initialize a GraphQL abstract class instance. Parameters ---------- connection : weaviate.connect.Connection Connection object to an active and running weaviate instance. N) _connection)selfrFs rC__init__zGraphQL.__init__Ns &rBreturncy)z Build method to be overloaded by the child classes. It should return the GraphQL query as a str. Returns ------- str The query. NrArIs rCbuildz GraphQL.buildZrBc|j} |jjdd|i}t |d}|J|S#t$r}td|d}~wwxYw)a[ Builds and runs the query. Returns ------- dict The response of the query. Raises ------ requests.ConnectionError If the network connection to weaviate fails. weaviate.UnexpectedStatusCodeException If weaviate reports a none OK status. z/graphqlquery)pathweaviate_objectzQuery was not successful.NzQuery was not successful)rNrHpostRequestsConnectionErrorr)rIrQresponseconn_errress rCdoz GraphQL.dofsv   U'',,*wX]N^,_H)3MN ' U)*EFH T UsA A AAN) r8r9r:__doc__r rJrstrrNdictrYrArBrCrErEIs: &: &  s    DrBrEcHeZdZdZdefdZedefdZe defdZ y)Filterz0 A base abstract class for all filters. contentct|ts-t|jjdt |t ||_y)z Initialize a Filter class instance. Parameters ---------- content : dict The content of the `Filter` clause. z+ filter is expected to be type dict but is N) isinstancer\ TypeError __class__r8typer_contentrIr_s rCrJzFilter.__init__sL'4(>>**+,''+G}o7 !) rBrKcy)zA Should be implemented in each inheriting class. NrArMs rC__str__zFilter.__str__rOrBc|jS)N)rerMs rCr_zFilter.contents }}rBN) r8r9r:rZr\rJrr[rhpropertyr_rArBrCr^r^sI**"   rBr^c4eZdZdZdeffd ZdefdZxZS)NearTextz NearText class used to filter weaviate objects. Can be used with text models only (text2vec). E.g.: text2vec-contextionary, text2vec-transformers. r_c(t||t|jd|jvr8d|jvr t dt d|jdt d|jvrt d|jdt d|jvrt|jdd|jvrt|jdd|jvr t d|jdtyy) an Initialize a NearText class instance. Parameters ---------- content : dict The content of the `nearText` clause. Raises ------ TypeError If 'content' is not of type dict. ValueError If 'content' has key "certainty"/"distance" but the value is not float. certaintydistanceSCannot have both 'certainty' and 'distance' at the same time. Only one is accepted.var_namevaluedtypemoveTo moveAwayFrom autocorrectN) superrJ_check_conceptre ValueError _check_typefloat_check_direction_clauseboolrIr_rcs rCrJzNearText.__init__s" !t}}% $-- 'T]]* , DMM+4NV[ \  & 4==3LTY Z t}} $ #DMM($; < T]] * #DMM.$A B DMM ) dmmM6RZ^ _ *rBrKcdt|jd}d|jvr|d|jdz }d|jvr|d|jdz }d|jvrL|jd}|d|d z }d|vr|d t|dz }d |vr|t|d z }|d z }d |jvrL|jd }|d|d z }d|vr|d t|dz }d |vr|t|d z }|d z }d|jvr|dt|jdz }|dzS)NznearText: {concepts: conceptsrn certainty: ro distance: ruz moveTo: {force: forcez concepts: objects}rvz moveAwayFrom: {force: rw autocorrect: } )rre_move_clause_objects_to_str _bool_to_str)rI near_textmove_tomove_away_froms rCrhzNearText.__str__s,U4==3L-M,NO $-- ' < k(B'CD DI  & ;t}}Z'@&AB BI t}} $mmH-G -gg.>-?@ @IW${51D+E*FGG G#89KLL  I T]] *!]]>:N 3N74K3LM MI^+{5 1K+L*MNN N*8 9RSS  I DMM ) >,t}}]7S*T)UV VI4rB r8r9r:rZr\rJr[rh __classcell__rcs@rCrlrls# &`&`P  rBrlc4eZdZdZdeffd ZdefdZxZS) NearVectorz; NearVector class used to filter weaviate objects. r_ct||d|jvr tdd|jvr8d|jvr t dt d|jdt d|jvrt d|jdt t|jd|jd<y)aL Initialize a NearVector class instance. Parameters ---------- content : list The content of the `nearVector` clause. Raises ------ TypeError If 'content' is not of type dict. KeyError If 'content' does not contain "vector". TypeError If 'content["vector"]' is not of type list. AttributeError If invalid 'content' keys are provided. ValueError If 'content' has key "certainty"/"distance" but the value is not float. vectorz&No 'vector' key in `content` argument.rnrorprqN)rxrJreKeyErrorrzr{r|rrs rCrJzNearVector.__init__s. ! 4== (CD D $-- 'T]]* , DMM+4NV[ \  & 4==3LTY Z",T]]8-D"E hrBrKcdt|jd}d|jvr|d|jdz }d|jvr|d|jdz }|dzS)NznearVector: {vector: rrnrrorr)rre)rI near_vectors rCrhzNearVector.__str__sv.uT]]85L/M.NO $-- ' \$-- *D)EF FK  & [z)B(CD DKT!!rBrrs@rCrrs#'F'FR""rBrc8eZdZdZdedeffd ZdefdZxZ S) NearObjectz; NearObject class used to filter weaviate objects. r_is_server_version_14ct||d|jvrd|jvr tdd|jvrd|_nbd|_|rYt |jdj djddk(r tjttdt|j|j|jtd |jvr8d |jvr td td |jd td |jvr td |jd ty y ) a  Initialize a NearVector class instance. Parameters ---------- content : list The content of the `nearVector` clause. is_server_version_14 : bool Whether the Server version is >= 1.14.0. Raises ------ TypeError If 'content' is not of type dict. ValueError If 'content' has key "certainty"/"distance" but the value is not float. TypeError If 'id'/'beacon' key does not have a value of type str! idbeaconz>The 'content' argument should contain EITHER `id` OR `beacon`!/)messagecategory stacklevelrqrnrorpN)rxrJrerzobj_idlenstripsplitwarningswarnrDeprecationWarningr{r[r|)rIr_rrcs rCrJzNearObject.__init__'s* ! DMM !DMM(A]^ ^ 4== DK"DK#DMM(,C,I,I#,N,T,TUX,Y(Z^_(_ 6/  T[[ dkk0JRUV $-- 'T]]* , DMM+4NV[ \  & 4==3LTY Z 'rBrKcd|jd|j|jd}d|jvr|d|jdz }d|jvr|d|jdz }|dzS) Nz nearObject: {: ""rnrrorr)rre)rI near_objects rCrhzNearObject.__str__Xs&t{{m3t}}T[[7Q6RRST $-- ' \$-- *D)EF FK  & [z)B(CD DKT!!rB) r8r9r:rZr\r~rJr[rhrrs@rCrr"s+/[/[D/[b""rBrc4eZdZdZdeffd ZdefdZxZS)AskzI Ask class used to filter weaviate objects by asking a question. r_c t||d|jvr tdt d|jdt d|jvr8d|jvr tdt d|jdt d|jvrt d|jdt d|jvrt d|jdtd|jvrt d|jdtd |jvrat d |jd tt ft|jd t r|jd g|jd <y y y ) a Initialize a Ask class instance. Parameters ---------- content : list The content of the `ask` clause. Raises ------ TypeError If 'content' is not of type dict. ValueError If 'content' has key "certainty"/"distance" but the value is not float. TypeError If 'content' has key "properties" but the type is not list or str. questionz6Mandatory "question" key not present in the "content"!rqrnrorprwrerank propertiesN) rxrJrerzr{r[r|r~listrars rCrJz Ask.__init__fsG& ! T]] *UV VZt}}Z/HPST $-- 'T]]* , DMM+4NV[ \  & 4==3LTY Z DMM ) dmmM6RZ^ _ t}} $ x1HPT U 4== ( T]]<5PY]_bXc d$-- 5s;/3}}\/J.K l+< )rBrKcdt|jd}d|jvr|d|jdz }d|jvr|d|jdz }d|jvr|dt|jdz }d |jvr|d t|jd z }d |jvr|d t|jd z }|d zS)Nzask: {question: rrnrrorrz properties: rwrrz rerank: r)rrer)rIasks rCrhz Ask.__str__s!% j(A"B!CD $-- ' \$-- " >C  & [z!: ;< ` clause. Raises ------ TypeError If 'content' is not of type dict. TypeError If 'content[""]' is not of type str. ValueError If 'content' has key "certainty"/"distance" but the value is not float. z("content" is missing the mandatory key "z"!rqrnrorpN) rxrJ _media_typersrerzr{r[r|)rIr_rrcs rCrJzNearMedia.__init__s. !%    ! ! 6GHXHXH^H^G__abc c%%++4==AQAQAWAW3X`c  $-- 'T]]* , DMM+4NV[ \  & 4==3LTY Z 'rBrKc|jjj}|jtjk(r$|jjj }d|d|jjd|j |jjd}d|j vr|d|j dz }d|j vr|d|j dz }|d zS) Nnearz: {rrrnrrorr)rrs capitalizer1r@upperre)rImedia near_medias rCrhzNearMedia.__str__s  &&113   y}} ,$$**002E5'd..445StGWGWG]G]9^8__` a  $-- ' L{)C(DE EJ  & K j(A'BC CJD  rB) r8r9r:r\r1rJr[rhrrs@rCrrs*)[)[)[V ! !rBrc(eZdZdZdeffd ZxZS) NearImagez: NearImage class used to filter weaviate objects. r_cBt||tjy)a Initialize a NearImage class instance. Parameters ---------- content : list The content of the `nearImage` clause. Raises ------ TypeError If 'content' is not of type dict. TypeError If 'content["image"]' is not of type str. ValueError If 'content' has key "certainty"/"distance" but the value is not float. N)rxrJr1r;rs rCrJzNearImage.__init__* )//2rBr8r9r:rZr\rJrrs@rCrr333rBrc(eZdZdZdeffd ZxZS) NearVideoz: NearVideo class used to filter weaviate objects. r_cBt||tjy)a Initialize a NearVideo class instance. Parameters ---------- content : list The content of the `nearVideo` clause. Raises ------ TypeError If 'content' is not of type dict. TypeError If 'content["video"]' is not of type str. ValueError If 'content' has key "certainty"/"distance" but the value is not float. N)rxrJr1r=rs rCrJzNearVideo.__init__rrBrrs@rCrrrrBrc(eZdZdZdeffd ZxZS) NearAudioz: NearAudio class used to filter weaviate objects. r_cBt||tjy)a Initialize a NearAudio class instance. Parameters ---------- content : list The content of the `nearAudio` clause. Raises ------ TypeError If 'content' is not of type dict. TypeError If 'content["audio"]' is not of type str. ValueError If 'content' has key "certainty"/"distance" but the value is not float. N)rxrJr1r<rs rCrJzNearAudio.__init__rrBrrs@rCrrrrBrc(eZdZdZdeffd ZxZS) NearDepthz: NearDepth class used to filter weaviate objects. r_cBt||tjy)a Initialize a NearDepth class instance. Parameters ---------- content : list The content of the `nearDepth` clause. Raises ------ TypeError If 'content' is not of type dict. TypeError If 'content["depth"]' is not of type str. ValueError If 'content' has key "certainty"/"distance" but the value is not float. N)rxrJr1r?rs rCrJzNearDepth.__init__9rrBrrs@rCrr4rrBrc(eZdZdZdeffd ZxZS) NearThermalz< NearThermal class used to filter weaviate objects. r_cBt||tjy)a Initialize a NearThermal class instance. Parameters ---------- content : list The content of the `nearThermal` clause. Raises ------ TypeError If 'content' is not of type dict. TypeError If 'content["thermal"]' is not of type str. ValueError If 'content' has key "certainty"/"distance" but the value is not float. N)rxrJr1r>rs rCrJzNearThermal.__init__Vs* )"3"34rBrrs@rCrrQs555rBrc(eZdZdZdeffd ZxZS)NearIMUz8 NearIMU class used to filter weaviate objects. r_cBt||tjy)a Initialize a NearIMU class instance. Parameters ---------- content : list The content of the `nearIMU` clause. Raises ------ TypeError If 'content' is not of type dict. TypeError If 'content["imu"]' is not of type str. ValueError If 'content' has key "certainty"/"distance" but the value is not float. N)rxrJr1r@rs rCrJzNearIMU.__init__ss* )--0rBrrs@rCrrns111rBrcXeZdZdZdeeefffd ZdeeefddfdZde fdZ xZ S)Sortz: Sort filter class used to sort weaviate objects. r_cNt|dgi|j|y)a| Initialize a Where filter class instance. Parameters ---------- content : list or dict The content of the `sort` filter clause or a single clause. Raises ------ TypeError If 'content' is not of type dict. ValueError If a mandatory key is missing in the filter content. sortr_N)rxrJaddrs rCrJz Sort.__init__s'& &". !rBrKNct|tr|g}t|tstdt |dt |dk(r t d|D]f}d|vsd|vr t dtd|dttd|dt|jd j|d|dd hy ) a Add more sort clauses to the already existing sort clauses. Parameters ---------- content : list or dict The content of the `sort` filter clause or a single clause to be added to the already existing ones. Raises ------ TypeError If 'content' is not of type dict. ValueError If a mandatory key is missing in the filter content. z4'content' must be of type dict or list. Given type: .rz"'content' cannot be an empty list.rRorderzIOne of the sort clause is missing required fields: 'path' and/or 'order'.rqr)rRrN) rar\rrbrdrrzr{r[reappend)rIr_clauses rCrzSort.adds$ gt $iG'4(RSWX_S`Raabcd d w<1 AB B FV#wf'< _ Vn   Wo  MM& ! ( ("6N#G_ # rBcrd}|jdD]}|dt|dd|ddz }|dz }|S) Nzsort: [rz{ path: rRz order: rz } ])rer)rIrrs rCrhz Sort.__str__sYmmF+ UF ifVn 56hvg>OtT TD U   rB) r8r9r:rZr r\rrJrr[rhrrs@rCrrsF"dDj 1".15t,11frBrcTeZdZdZdeffd ZdeddfdZdeddfdZdefdZ xZ S) Wherez= Where filter class used to filter weaviate objects. r_ct||d|jvr#d|_|j |jyd|jvr#d|_|j |jyt d|j)ab Initialize a Where filter class instance. Parameters ---------- content : dict The content of the `where` filter clause. Raises ------ TypeError If 'content' is not of type dict. ValueError If a mandatory key is missing in the filter content. rRToperandsFz?Filter is missing required fields `path` or `operands`. Given: N)rxrJre is_filter _parse_filter_parse_operatorrzrs rCrJzWhere.__init__s{" ! T]] "!DN   t}} - 4== ("DN   /TUYUbUbTcd rBrKNcd|vrtd||dtvr'td|dddjtt|d|_|d|_t ||_||j|_|j dk(r5|jdk7r%td|j d |jy y ) a Set filter fields for the Where filter. Parameters ---------- content : dict The content of the `where` filter clause. Raises ------ ValueError If 'content' is missing required fields. operator4Filter is missing required field `operator`. Given: Operator ( is not allowed. Allowed operators are: , rRr/r"z; requires a value of type valueGeoRange. Given value type: N) rzWHERE_OPERATORSjoinrrRr_find_value_type value_typersrfs rCrzWhere._parse_filters W $VW^V_`a a : o 5GJ/01**.))O*D)EG '&/*  + *73T__- ==, ,O1SDMM?+%%)__$57 2T ,rBcd|vrtd||dtvrtd|ddtt|}|d|_g|_|dD]&}|jj t |(y)a Set operator fields for the Where filter. Parameters ---------- content : dict The content of the `where` filter clause. Raises ------ ValueError If 'content' is missing required fields. rrrrrN)rzrrrrrr)rIr_reoperands rCrzWhere._parse_operator's W $VW^V_`a a : o 5GJ/01**9):< G$ ,   + 1G MM w 0 1rBc0|jr1d|jd|jdt|jd}|jdvrE|jdvr t |j |j||j dz }|dzS|jdvr|jd vr t |j |jt|j tr8|j Dcgc] }t|}}|t|dz }|dzS|t|j dz }|dzS|jd vr|jd vr t |j |jt|j tr |t|j dz }|dzS|t|j dz }|dzS|jd k(r@t|j |j|t|j dz }|dzS|d |j dz }|dzSg}|jD]}|jt!|dd!dj#|}d|jd|dScc}w)Nzwhere: {path: z operator:  : )rrrrrr)rrrrr)rr rrrr)rrrr)r!rr)rrr"rz"}rzwhere: {operator: z operands: [z]} )rrRr_convert_value_typer_check_is_listrsrarr _render_listr_check_is_not_list_geo_range_to_strrrr[r)rIgqlvval operands_strrrs rCrhz Where.__str__Cs >>#DII;k$--J]^b^m^mJnIooqrC#??' #4::t?$**R((B9 A%??' #4::t?djj$/59ZZ@=+@C@l3/033C9 mDJJ78;;C9 $]]??&OO"4::t?djj$/l4::67r::C9  l4::67r::C 9  O3"4::t?+DJJ78;;9 4::,c**9  }} 4G   G Qr 2 3 499\*$T]]O<zNN/As8J) r8r9r:rZr\rJrrr[rhrrs@rCrrsI<TdB1t118:O:OrBr_typerKc|dk(s|dk(ry|dk(s|dk(ry|dk(s|dk(ry |d k(s|d k(ry |d k(s|dk(ry|dk(s|dk(ry|S)aConvert the value type to match `json` formatting required by the Weaviate-defined GraphQL endpoints. NOTE: This is crucially different to the Batch REST endpoints wherein the where filter is also used. Parameters ---------- _type : str The Python-defined type to be converted. Returns ------- str The string interpretation of the type in Weaviate-defined `json` format. rrrrrr rrrrrrrrr!valueDateArray valueDateListrrA)rs rCrrs}   E_$< $ $1B(B / !Un%< $ $1B(B % %2D)D " "e&> rBrsc,ddj|dS)aConvert a list of values to string (lowercased) to match `json` formatting. Parameters ---------- value : list The value to be converted Returns ------- str The string interpretation of the value in `json` format. [,r)rrss rCrrssxxq !!rBcFt|tstd|d|y)a#Checks whether the provided value is a list to match the given `value_type`. Parameters ---------- value : list The value to be checked. _type : str The type to be checked against. Raises ------ TypeError If the value is not a list. z7Must provide a list when constructing where filter for  with Nrarrbrsrs rCrrs2 eT "EeWFSXRY Z   #rBcFt|trtd|d|y)aChecks whether the provided value is a list to match the given `value_type`. Parameters ---------- value : list The value to be checked. _type : str The type to be checked against. Raises ------ TypeError If the value is a list. z9Cannot provide a list when constructing where filter for rNrrs rCrrs2%GwfUZT[ \  rBcJ|dd}|dd}|dd}d|d|d|d S) z Convert the valueGeoRange object to match `json` formatting. Parameters ---------- value : dict The value to be converted. Returns ------- str The string interpretation of the value in `json` format. geoCoordinateslatitude longituderomaxz{ geoCoordinates: { latitude: z longitude: z } distance: { max: z }}rA)rsrrros rCrrsR%&z2H&' 4IZ 'H -hZ|I;Ndemdnns ttrBc|duryy)a Convert a bool value to string (lowercased) to match `json` formatting. Parameters ---------- value : bool The value to be converted Returns ------- str The string interpretation of the value in `json` format. TtruefalserAr s rCrrs } rB directionctd|td|vrd|vr tdd|vr t|d|vr t |d|vr tdtd|dt y) aX Validate the direction sub clause. Parameters ---------- direction : dict A sub clause of the Explore filter. Raises ------ TypeError If 'direction' is not a dict. TypeError If the value of the "force" key is not float. ValueError If no "force" key in the 'direction'. moveXXXrqrrz=The 'move' clause should contain `concepts` OR/AND `objects`!rz&'move' clause needs to state a 'force'N)r{r\rzry_check_objectsr|)rs rCr}r}sp&)4@)#)9*DXYYYy!Iy!iABB '(:%HrBr_cd|vr tdtd|dttft |dtr |dg|d<yy)a7 Validate the concept sub clause. Parameters ---------- content : dict An Explore (sub) clause to check for 'concepts'. Raises ------ ValueError If no "concepts" key in the 'content' dict. TypeError If the value of the "concepts" is of wrong type. rzNo concepts in contentrqN)rzr{rr[rars rCryry"sZ" 122j!Sk '*%s+&z23 ,rBctd|dttft|dtr |dg|d<t |ddk(r t d|dD]$}t |dk7s d|vsd|vst dy ) aM Validate the `objects` sub clause of the `move` clause. Parameters ---------- content : dict An Explore (sub) clause to check for 'objects'. Raises ------ ValueError If no "concepts" key in the 'content' dict. TypeError If the value of the "concepts" is of wrong type. rrqrz;'moveXXX' clause specifies 'objects' but no value provided.rrrzEEach object from the `move` clause should have ONLY `id` OR `beacon`!N)r{rr\rarrz)r_objs rCrr?s"')*' field is either missing or incorrect: z. Valid values are: rrz1Multiple fields 'value' are not supported: )ALL_VALUE_TYPESsetkeysrrz VALUE_TYPESpop)r_rs rCrrwsv(!3w||~#66J :!B7)K_`k_llm n   :!LWIVWW >> rBrc `d}|D]#}d|vrd}nd}|d|dt||dz }%|dzS)z _summary_ Parameters ---------- objects : list _description_ Returns ------- str _description_ z objects: [rr{rrrr)r to_returnr! id_beacons rCrrsZIB 3;I Ir)BuS^'<&=SAA B s?rB)BrZrabcrrcopyrenumrjsonrtypingr r r requests.exceptionsr rUweaviate.connectr weaviate.error_msgsr weaviate.utilrrrVALUE_LIST_TYPESVALUE_ARRAY_TYPESVALUE_PRIMITIVE_TYPESunionr&r)rr1rEr^rlrrrrrrrrrrrrr[rrrrrr\rr~rr}ryrrdr{rrrArBrCr<sl#$$J':OO#(():;AABWX%%&;< "5c5pSBF vF R4"4"n<"<"~?&?D7!7!t3 3:3 3:3 3:3 3:5)5:1i1:U6UpZOFZOzss> " " "  # c d * c # $ *uTucu((ItIIB4D4T4:DT> # c %dDj8I48O2P UY 2ds@#rB