?hKY dZddlZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z mZmZmZmZmZmZmZddlZddlZddlmZddlmZmZmZddlmZd Zd Z d Z!Gd d e Z"Gddee"Z#de e$e fde$fdZ%de e$e fde$fdZ&de$de'fdZ(de$de'fdZ) dLde e$ejTfdee$de+fdZ,de e$e+fde+fdZ-de$de.fd Z/de$de.fd!Z0d"e e$ejTfde$fd#Z1d$ede2fd%Z3de$de$fd&Z4d'e+d(e+de.fd)Z5d*e2d+e2de.fd,Z6d*e2d+e2de.fd-Z7dMd.ed/ede$fd0Z8d1e$de$fd2Z9d3eeee$efddfd4Z: dNd5ed6e$d7e;d8e.ddf d9Zd5e$de$fd=Z?d>e$de@fd?ZAd@e$de.fdAZBd@e$dBe$de.fdCZCdDe eeefedfdeeeffdEZDdFedeee$effdGZEdHejdIe$deee$effdJZGdHejdIe$deeee$effdKZHy)Oz Helper functions! N)EnumEnumMeta)BufferedReader)UnionSequenceAnyOptionalListDictTuplecast)JSONDecodeError)SchemaValidationExceptionUnexpectedStatusCodeException ResponseCannotBeDecodedException)NUMBERSz*https://pypi.org/pypi/weaviate-client/jsonzv1.16.0ceZdZdedefdZy)MetaEnumitemreturnc |j|jjvS#t$r||jjvcYSwxYwN)name __members__keysAttributeError)clsrs K/home/chris/cleankitchens-env/lib/python3.12/site-packages/weaviate/util.py __contains__zMetaEnum.__contains__$sJ 299 4 4 66 6 23??//11 1 2s%(%AAN)__name__ __module__ __qualname__rboolr rrr#s222r&rc eZdZy)BaseEnumN)r!r"r#r%r&rr(r(-sr&r() metaclassimage_or_image_pathrct|trStjj |st d|zt |d5}|j}dddn,t|tr|j}n tdtjjdS#1swY-xYw)a Encode a image in a Weaviate understandable format from a binary read file or by providing the image path. Parameters ---------- image_or_image_path : str, io.BufferedReader The binary read file or the path to the file. Returns ------- str Encoded image. Raises ------ ValueError If the argument is str and does not point to an existing file. TypeError If the argument is of a wrong data type. No file found at location brNzV"image_or_image_path" should be a image path or a binary read file (io.BufferedReader)utf-8 isinstancestrospathisfile ValueErroropenreadr TypeErrorbase64 b64encodedecode)r*filecontents rimage_encoder_b64r>1s.%s+ww~~129= 1.14.0). Parameters ---------- to_uuid : str or uuid.UUID The UUID for which to create a local beacon. class_name : Optional[str], optional The class name of the `to_uuid` object. Used with Weaviate >= 1.14.0. For Weaviate < 1.14.0 use None value. Returns ------- dict The local beacon. Raises ------ TypeError If 'to_uuid' is not of type str. ValueError If the 'to_uuid' is not valid. z"Uuid does not have the proper formNz0Expected to_object_uuid of type str or uuid.UUIDbeaconzweaviate://localhost//)r0r1uuid_libUUIDr5r8_capitalize_first_letter)rLrMuuids rgenerate_local_beaconrUs8'3 Mx}}W-.D GX]] +7|JKK1$899)*B:*N)OqQUPVW  MAB L Ms A>>Bobject_c8| tdt|tr|St|trt j |rPt j|}|jdk(rtt|jStd|ztjj|std|zt|d5}tttj |cdddStd#1swYtdxYw)a  Takes an object that should describe a dict e.g. a schema or an object and tries to retrieve the dict. Parameters ---------- object_ : str or dict The object from which to retrieve the dict. Can be a python dict, or the path to a json file or a url of a json file. Returns ------- dict The object as a dict. Raises ------ TypeError If 'object_' is neither a string nor a dict. ValueError If no dict can be retrieved from object. Nzargument is NonezCould not download file r,rziArgument is not of the supported types. Supported types are url or file path as string or schema as dict.)r8r0dictr1 validatorsurlrequestsget status_coder jsonr5r2r3r4r6load)rVresponser<s r_get_dict_from_objectrcs2*++'4 '3 >>' "||G,H##s*D(--/227'AB Bww~~g&9GCD D '3  /4diio. / /  8  /  8 s #DDr\c$t|tsy|jdsy|dd}|jd}t |dvry|ddk7rt j |dsy tj|d y #t$rYywxYw) aR Checks if the input follows a normal Weaviate 'beacon' like this: 'weaviate://localhost/ClassName/28f3f61b-b524-45e0-9bbe-2c1550bf73d2' Parameters ---------- url : str The URL to be validated. Returns ------- bool True if the 'url' is a Weaviate object URL. False otherwise. Fz weaviate:// NrPrr localhostT) r0r1 startswithsplitlenr[domainrQrRr5)r\rks ris_weaviate_object_urlrns" c3  >>- ( bc(C IIcNE 5z Qx;  q* eBi   s*B BBc|jd}t|dk7ry|djd}t|dvry tj|d|dd k(syy #t$rYywxYw) a Validates an url like 'http://localhost:8080/v1/objects/1c9cd584-88fe-5010-83d0-017cb3fcb446' or '/v1/objects/1c9cd584-88fe-5010-83d0-017cb3fcb446' references a object. It only validates the path format and UUID, not the host or the protocol. Parameters ---------- url : str The URL to be validated. Returns ------- bool True if the 'url' is a valid path to an object. False otherwise. z/v1/rgFrPrfrirobjectsT)rkrlrQrRr5)r\v1_splitrks r is_object_urlrs's$yy H 8} QK  c "E 5z eBi  8y   sA&& A21A2rTct|tjr t|St|ts t dtt |zt |}t|}|}|s|r|jdd} ttj|}|S#t$r tddwxYw)a Validate and extract the UUID. Parameters ---------- uuid : str or uuid.UUID The UUID to be validated and extracted. Should be in the form of an UUID or in form of an URL (weaviate 'beacon' or 'href'). E.g. 'http://localhost:8080/v1/objects/fc7eb129-f138-457f-b727-1b29db191a67' or 'weaviate://localhost/28f3f61b-b524-45e0-9bbe-2c1550bf73d2' or 'fc7eb129-f138-457f-b727-1b29db191a67' Returns ------- str The extracted UUID. Raises ------ TypeError If 'uuid' is not of type str. ValueError If 'uuid' is not valid or cannot be extracted. z2'uuid' must be of type str or uuid.UUID, but was: rPriz:Not valid 'uuid' or 'uuid' can not be extracted from valueN) r0rQrRr1r8typernrsrkr5)rT_is_weaviate_url_is_object_url_uuids rget_valid_uuidryLs:$ &4y dC LsSWX\S]^__-d3"4(N E> 3#aHMM%() L aUV\``as B''B=vectorct|tr|S |jjS#t$rI |j jjcYS#t$r t ddwxYwwxYw)a Get weaviate compatible format of the embedding vector. Parameters ---------- vector: Sequence The embedding of an object. Used only for class objects that do not have a vectorization module. Supported types are `list`, `numpy.ndarray`, `torch.Tensor` and `tf.Tensor`. Returns ------- list The embedding as a list. Raises ------ TypeError If 'vector' is not of a supported type. zThe type of the 'vector' argument is not supported! Supported types are `list`, 'numpy.ndarray`, `torch.Tensor` and `tf.Tensor`N)r0listsqueezetolistrnumpyr8)rzs r get_vectorr{s*&$  ~~&&((  <<>))+224 4 ^   s 2 B+A*'B*BBc0|ddjddS)a Get the domain from a weaviate URL. Parameters ---------- url : str The weaviate URL. Of this form: 'weaviate://localhost/objects/28f3f61b-b524-45e0-9bbe-2c1550bf73d2' Returns ------- str The domain. reNrPr)rkr\s rget_domain_from_weaviate_urlrs rs8>># q !!r& sub_schemaschemacX|jdg}d|vr|d}n|g}t||S)a Check for a subset in a schema. Parameters ---------- sub_schema : dict The smaller schema that should be contained in the 'schema'. schema : dict The schema for which to check if 'sub_schema' is a part of. Must have the 'classes' key. Returns ------- bool True is 'sub_schema' is a subset of the 'schema'. False otherwise. classes)r^_compare_class_sets)rrschema_classessub_schema_classess r_is_sub_schemars=$ZZ 2.NJ' 2(\ 1> BBr&sub_setset_c|D]R}d}|D]E}d|vr tdt|dt|dk(s0t|d|dsCd}n|rRyy)a_ Check for a subset in a set of classes. Parameters ---------- sub_set : list The smaller set that should be contained in the 'set'. set_ : list The set for which to check if 'sub_set' is a part of. Returns ------- bool True is 'sub_set' is a subset of the 'set'. False otherwise. Fclassz9The sub schema class/es MUST have a 'class' keyword each! propertiesT)rrS_compare_properties)rr sub_set_classfound set_classs rrrs$!  Im+/O( g(>?C['"D'}\'BIlD[\ E  r&cJ|D]}d}|D]}|d|dk(sd}n|ryy)ab Check for a subset in a set of properties. Parameters ---------- sub_set : list The smaller set that should be contained in the 'set'. set_ : list The set for which to check if 'sub_set' is a part of. Returns ------- bool True is 'sub_set' is a subset of the 'set'. False otherwise. FrTr%)rrsub_set_propertyr set_propertys rrrsO$$  L'<+??  r& identifier namespacec ttjtjt|t|zS)a Generate an UUIDv5, may be used to consistently generate the same UUID for a specific identifier and namespace. Parameters ---------- identifier : Any The identifier/object that should be used as basis for the UUID. namespace : Any, optional Allows to namespace the identifier, by default "" Returns ------- str The UUID as a string. )r1rQuuid5 NAMESPACE_DNS)rrs rgenerate_uuid5rs-$ x~~h44c)ns:6VW XXr&stringcpt|dk(r|jS|dj|ddzS)z Capitalize only the first letter of the `string`. Parameters ---------- string : str The string to be capitalized. Returns ------- str The capitalized string. rprN)rl capitalize)rs rrSrS$s= 6{a  "" !9   !F12J ..r&resultscj|y|D]+}d|vsd|dvsd|ddvst|dd-y)z Check batch results for errors. Parameters ---------- results : dict The Weaviate batch creation return value. Nresulterrorserror)print)rrs rcheck_batch_resultr8sS2 v (fX.>">&*844fX&x012r&valuearg_name data_type include_zeroct||rt|trtd|d|d|r|dkrtd|dy|dkrtd|dy)a6 Check if the `value` of the `arg_name` is a positive number. Parameters ---------- value : Union[int, float] The value to check. arg_name : str The name of the variable from the original function call. Used for error message. data_type : type The data type to check for. include_zero : bool Wether zero counts as positive or not. By default False. Raises ------ TypeError If the `value` is not of type `data_type`. ValueError If the `value` has a non positive value. 'z' must be of type .rz8' must be positive, i.e. greater or equal to zero (>=0).z0' must be positive, i.e. greater that zero (>0).N)r0r$r8r5)rrrrs r_check_positive_numrLsy2 eY ':eT+B!H:%7 {!DEE 19q *bcd d  A:q *Z[\ \ r&cvd|jvxs&d|jvxsd|jvS)Nz weaviate.iozsemi.technologyzweaviate.cloud)lowerrs ris_weaviate_domainros;$ +  + + syy{ *r&sc&|jddS)N  )replace)rs rstrip_newlinesrws 99T3 r&cRt|}tjdd|}d|dS)z Ensures string is sanitized for GraphQL. Parameters ---------- value : str The value to be converted. Returns ------- str The sanitized string. z(?. Is it in the format '(v)x.y.z' (e.g. 'v1.18.2' or '1.18.0')?)countrmatchtuplemapintgroupsr5)rpatternrver_tups rparse_version_stringrss}}SQD.G HHWg &E C01?yIG H  r&current_version_strc@t|}tt}||kDS)aZ Check if the user should be gently nudged to upgrade their Weaviate server version. Parameters ---------- current_version_str : str The version of the Weaviate server that the client is connected to. (e.g. "v1.18.2" or "1.18.0") Returns ------- bool : True if the user should be nudged to upgrade. )rMINIMUM_NO_WARNING_VERSION)rcurrent_versionminimum_versions ris_weaviate_too_oldrs% ++>?O*+EFO _ ,,r&latest_version_strc t|}t|\}}t|tz d}||f}||kDS#t$rYywxYw)a! Check if the user should be gently nudged to upgrade their Weaviate client version. Parameters ---------- current_version_str : str The version of the Weaviate client that is being used (e.g. "v1.18.2" or "1.18.0") latest_version_str : str The latest version of the Weaviate client to compare against (e.g. "v1.18.2" or "1.18.0") Returns ------- bool : True if the user should be nudged to upgrade. False if the user is using a valid version or if the version could not be parsed. rF)rmaxMAXIMUM_MINOR_VERSION_DELTAr5)rrr latest_major latest_minor minimum_minorrs ris_weaviate_client_too_oldrs[&./BC%9:L%M" lL+FFJ '700 s47 AAtimeout_configcdtttttfdfdtfd}t |t st |t r(t |ts|J|dkr td||fSt |ts tdt|dk7r td ||d r1||d r&t |d trt |d tr td |d dks|d dkr td|S) a Validate and return TimeOut configuration. Parameters ---------- timeout_config : tuple(NUMBERS, NUMBERS) or NUMBERS or None, optional Set the timeout configuration for all requests to the Weaviate server. It can be a number or, a tuple of two numbers: (connect timeout, read timeout). If only one number is passed then both connect and read timeout will be set to that value. Raises ------ TypeError If arguments are of a wrong data type. ValueError If 'timeout_config' is not a tuple of 2. ValueError If 'timeout_config' is/contains negative number/s. numNrcFt|txst|tSr)r0floatr)rs r check_numberz/_get_valid_timeout_config..check_numbers#u%=C)==r&gz1'timeout_config' cannot be non-positive number/s!z='timeout_config' should be a (or tuple of) positive number/s!rgz%'timeout_config' must be of length 2!rrpz)'timeout_config' must be tuple of numbers) rrr r$r0rrr5rr8rl)rrs r_get_valid_timeout_configrs0>%w/?)@$ FG>D> >5)Z-LV`W))) S PQ Q~-- ne ,WXX >a@AA * + ^A=N0O>!$d+ >!;Ld0SCDDaC>!#4#;LMM r& json_responsec0|yt|tsJ|Sr)r0rZ)rs r_type_request_responsers! mT ** * r&rblocationc|yd|jcxkrdkr$nn! tt|j}|St ||#t$r t ||wxYwNrXi,)r_r rZr`rrrrbrrs r_decode_json_response_dictrsm h""(S( G x}}7M  (( ;; G28XF F G A Ac|yd|jcxkrdkr$nn! |j}tt|St ||#t$r t ||wxYwr)r_r`r r|rrrrs r_decode_json_response_listr&sm h""(S( G$MMOMm, , (( ;; G28XF F Grr))F)I__doc__r9r`r2rrTrQenumrriortypingrrrr r r r r r]r[requests.exceptionsrweaviate.exceptionsrrrweaviate.typesrPYPI_PACKAGE_URLrrrr(r1r>rBbytesrHrKrRrZrUrcr$rnrsryr|rrrrrrrSrrurrrrrrrrrrResponserrr%r&rrs JJJ/ #? 2x2 tx $55n1D+E$5#$5N#5c>.A(B#5s#5L;S;U;$:3:5:,!%* 3 % &* * *Z05d#300f   F"s"t"J,sHMM12,s,^#x#D#L"c"c"&CtCTCdC4!!T!d!HTd:YsYsYCY*/S/S/(2 d4S>* +2 2*FK ]  ] ]*. ]>B ]  ]FCD c c * # % :-S-T-*CSUY:-% 017D@A- 7G -`#(4S>2J < <+. < d38n < < <+. < d4S>"#