?h*dZddlZddlmZmZddlmZmZmZm Z m Z m Z ddl m Z ddlmZmZddlmZee ee fZGdd eZGd d eZGd d eZy)z! BatchRequest class definitions. N)ABCabstractmethod)ListSequenceOptionalDictAnyUnion)uuid4)get_valid_uuid get_vector)UUIDc eZdZdZddZdefdZdefdZddZ ddede fd Z e d Z e deeeeefeeefffd Ze d ed eeedeeedefdZedeeefd eeedeeedefdZy) BatchRequestzM BatchRequest abstract class used as a interface for batch requests. returnNcg|_yN_itemsselfs U/home/chris/cleankitchens-env/lib/python3.12/site-packages/weaviate/batch/requests.py__init__zBatchRequest.__init__s ,. c,t|jSrlenrrs r__len__zBatchRequest.__len__s4;;rc2t|jdk(S)z Check if BatchRequest is empty. Returns ------- bool Whether the BatchRequest is empty. rrrs ris_emptyzBatchRequest.is_emptys4;;1$$rcg|_y)z= Remove all the items from the BatchRequest. Nrrs remptyzBatchRequest.empty&s  rindexc8|jj|S)a Remove and return item at index (default last). Parameters ---------- index : int, optional The index of the item to pop, by default -1 (last item). Returns ------- dict The popped item. Raises ------- IndexError If batch is empty or index is out of range. )rpop)rr#s rr%zBatchRequest.pop-s({{u%%rcy)zAdd objects to BatchRequest.N)rargskwargss raddzBatchRequest.addCrcy)zQReturn the request body to be digested by weaviate that contains all batch items.Nr'rs rget_request_bodyzBatchRequest.get_request_bodyGr+r response_itemerrors_to_excludeerrors_to_includecy)aAdd failed items from a weaviate response. Parameters ---------- response_item : BatchResponse Weaviate response that contains the status for all objects. errors_to_exclude : Optional[List[str]] Which errors should NOT be retried. errors_to_include : Optional[List[str]] Which errors should be retried. Returns ------ BatchResponse: Contains responses form all successful object, eg. those that have not been added to this batch. Nr')rr.r/r0s r add_failed_objects_from_responsez-BatchRequest.add_failed_objects_from_responseKr+rentryct|ddk(s(d|dvs!d|ddvst|ddddk(ry|'|dddD]tfd|Dsyy|'|dddD]tfd|Dsyyy) NresultrerrorserrorTc3,K|] }|dv ywmessageNr').0exclerrs r z3BatchRequest._skip_objects_retry..sL$ts9~-LFc3,K|] }|dv ywr9r')r;inclr=s rr>z3BatchRequest._skip_objects_retry..xr?r@)rany)r3r/r0r=s @r_skip_objects_retryz BatchRequest._skip_objects_retrybs h A %uX.eHoh775?8,W56!;  (Xx09 L:KLL   *Xx09 !L:KLL  !r)rN))__name__ __module__ __qualname____doc__rintrboolr r"dictr%rr*r rrstrr r- BatchResponserr2 staticmethodrDr'rrrrs+/   %$ %&&d&,++`%T#s(^(= 1.14.0. It will be required in future versions of Weaviate Server and Clients. Use None value ONLY for Weaviate < v1.14.0, by default None Raises ------ TypeError If arguments are not of type str. ValueError If 'uuid' is not valid or cannot be extracted. z5'from_object_class_name' argument must be of type strz1'from_property_name' argument must be of type strNz3'to_object_class_name' argument must be of type strzweaviate://localhost//fromtorW) isinstancerM TypeErrorr rappend) rrRrSrTrUrVrW to_beaconitems rr*zReferenceBatchRequest.addsN0#6ST T,c2OP P  +J?SUX4YQR R'7)*:;  +/0D/EQ~FVWI//?@I,$%  ! !    #DN 4 rc|jS)z Get request body as a list of dictionaries, where each dictionary is a Weaviate-object reference. Returns ------- List[dict] A list of Weaviate-objects references as dictionaries. rrs rr-z&ReferenceBatchRequest.get_request_bodys{{rresponser/r0cg}|D]K}|j|||r|j|(|jj|d|ddM|S)Nr[r\rZ)rDr_r)rrcr/r0successful_responsesrefs rr2z6ReferenceBatchRequest.add_failed_objects_from_responsesg " GC''->@QR$++C0 KK  F 3t9E F  G $#r)NN)rFrGrHrIrMrrr*rrr r-rNr2r'rrrQrQ~s/3 $E! #E!E! E!  E! 'sm E! E! E!N $tCH~"6  $ $$DI. $$DI. $  $rrQc eZdZdZ ddededeedeedeedef d Z de ee ffd Z d e d eeed eeede fdZy)ObjectsBatchRequestz{ Collect objects for one batch request to weaviate. Caution this batch will not be validated through weaviate. N data_object class_nameuuidvectorrWrcRt|ts tdt|ts td|t j |d}| t |}nt t}||d<|t||d<|||d<|jj||S)aK Add one object to this batch. Does NOT validate the consistency of the object against the client's schema. Checks the arguments' type and UUIDs' format. Parameters ---------- class_name : str The name of the class this object belongs to. data_object : dict Object to be added as a dict datatype. uuid : str or None, optional UUID of the object as a string, by default None vector: Sequence or None, optional The embedding of the object that should be validated. Can be used when: - a class does not have a vectorization module. - The given vector was generated using the _identical_ vectorization module that is configured for the class. In this case this vector takes precedence. Supported types are `list`, 'numpy.ndarray`, `torch.Tensor` and `tf.Tensor`, by default None. tenant: str, optional Tenant of the object Returns ------- str The UUID of the added object. If one was not provided a UUIDv3 will be generated. Raises ------ TypeError If an argument passed is not of an appropriate type. ValueError If 'uuid' is not of a proper form. zObject must be of type dictzClass name must be of type str)class propertiesidrlrW) r]rLr^rMcopydeepcopyr r r rr_)rrirjrkrlrW batch_item valid_uuids rr*zObjectsBatchRequest.addsZ+t,9: :*c*<= =){9ST  '-J'0J% 4  #-f#5Jx  #)Jx  :&rc"dg|jdS)z Get the request body as it is needed for the Weaviate server. Returns ------- dict The request body as a dict. ALL)fieldsobjectsrrs rr-z$ObjectsBatchRequest.get_request_body0s!'dkk::rrcr/r0c g}|D]f}|j|||r|j|(|j|d|d|d|jdd|jddh|S)NrornrprlrW)rirjrkrlrW)rDr_r*get)rrcr/r0reobjs rr2z4ObjectsBatchRequest.add_failed_objects_from_response<s " C''->@QR$++C0 HH -w<Ywwx.wwx.   $#r)NNN)rFrGrHrIrLrMrrrr*rr r-rNrr2r'rrrhrhs $%) $ @@@tn @ " @  @ @D ;$sCx. ;$$$DI.$$DI. $  $rrh)rIrqabcrrtypingrrrrr r rkr weaviate.utilr r weaviate.typesrrMrNrrQrhr'rrrs_ #==4T#s(^$ l3l^g$Lg$Tg$,g$r