?h5 dZddlZddlmZmZmZmZddlmZ ddl m Z ddl m Z ddlmZmZerdd lmZGd d Zy) z! ConfigBuilder class definition. N)DictAnycast TYPE_CHECKING)ConnectionError) Connection)UnexpectedStatusCodeException)_capitalize_first_letter_decode_json_response_dict)ClassificationceZdZdZdeddfdZdeddfdZd eddfd Z d eddfd Z d e ddfdZ de ddfdZ deddfdZdeddfdZdeddfdZddZdeddfdZddZdefdZdefdZy) ConfigBuilderzO ConfigBuild class that is used to configure a classification process. connectionclassificationr c<||_||_i|_d|_y)a Initialize a ConfigBuilder class instance. Parameters ---------- connection : weaviate.connect.Connection Connection object to an active and running weaviate instance. classification : weaviate.classification.Classification Classification object to be configured using this ConfigBuilder instance. FN) _connection_classification_config_wait_for_completion)selfrrs d/home/chris/cleankitchens-env/lib/python3.12/site-packages/weaviate/classification/config_builder.py__init__zConfigBuilder.__init__s#&-') $)!classification_typereturnc$||jd<|S)z Set classification type. Parameters ---------- classification_type : str Type of the desired classification. Returns ------- ConfigBuilder Updated ConfigBuilder. typer)rrs r with_typezConfigBuilder.with_type(s 3 V rkcld|jvrd|i|jd<|S||jdd<|S)a Set k number for the kNN. Parameters ---------- k : int Number of objects to use to make a classification guess. (For kNN) Returns ------- ConfigBuilder Updated ConfigBuilder. settingsr!r)rr!s rwith_kzConfigBuilder.with_k:sC T\\ )(+QxDLL $ -.DLL $S ) r class_namec6t||jd<|S)z What Object type to classify. Parameters ---------- class_name : str Name of the class to be classified. Returns ------- ConfigBuilder Updated ConfigBuilder. class)r r)rr%s rwith_class_namezConfigBuilder.with_class_namePs!9 D W rclassify_propertiesc$||jd<|S)z Set the classify properties. Parameters ---------- classify_properties: list A list of properties to classify. Returns ------- ConfigBuilder Updated ConfigBuilder. classifyPropertiesr)rr)s rwith_classify_propertiesz&ConfigBuilder.with_classify_propertiesbs.A )* rbased_on_propertiesc$||jd<|S)a Set properties to build the classification on. Parameters ---------- based_on_properties: list A list of properties to classify on. Returns ------- ConfigBuilder Updated ConfigBuilder. basedOnPropertiesr)rr-s rwith_based_on_propertiesz&ConfigBuilder.with_based_on_propertiests-@ () r where_filtercdd|jvri|jd<||jdd<|S)z Set Source 'where' Filter. Parameters ---------- where_filter : dict Filter to use, as a dict. Returns ------- ConfigBuilder Updated ConfigBuilder. filters sourceWhererrr1s rwith_source_where_filterz&ConfigBuilder.with_source_where_filter6 DLL (&(DLL #1= Y . rcdd|jvri|jd<||jdd<|S)z Set Training set 'where' Filter. Parameters ---------- where_filter : dict Filter to use, as a dict. Returns ------- ConfigBuilder Updated ConfigBuilder. r3trainingSetWhererr5s rwith_training_set_where_filterz,ConfigBuilder.with_training_set_where_filters7 DLL (&(DLL #6B Y 23 rcdd|jvri|jd<||jdd<|S)z Set Target 'where' Filter. Parameters ---------- where_filter : dict Filter to use, as a dict. Returns ------- ConfigBuilder Updated ConfigBuilder. r3 targetWhererr5s rwith_target_where_filterz&ConfigBuilder.with_target_where_filterr7rcd|_|S)z Wait for completion. Returns ------- ConfigBuilder Updated ConfigBuilder. T)r)rs rwith_wait_for_completionz&ConfigBuilder.with_wait_for_completions%)! rr#c|d|jvr||jd<|S|D]}|||jd|<|S)a Set settings for the classification. NOTE if you are using 'kNN' the value 'k' can be set by this method or by 'with_k'. This method keeps previously set 'settings'. Parameters ---------- settings: dict Additional settings to be set/overwritten. Returns ------- ConfigBuilder Updated ConfigBuilder. r#r)rr#keys r with_settingszConfigBuilder.with_settingssR" T\\ )'/DLL $   >08  Z(- > rNc6gd}|D]}||jvst|dd|jvr(t|jdts t d|jddk(r*d|jj dgvr tdy y ) z Validate the current classification configuration. Raises ------ ValueError If a mandatory field is not set. )rr'r/r+z# is not set for this classificationr#z!"settings" should be of type dictrknnr!z$k is not set for this classificationN)r ValueError isinstancedict TypeErrorget)rrequired_fieldsfields r_validate_configzConfigBuilder._validate_configsW$ PEDLL( E7*M!NOO P  %dll:6= CDD << 5 ($,,**:r:: !GHH; )rc |jjd|j}|jdk(rt |d}|J|St d|#t$r}td|d}~wwxYw)aa Start the classification based on the configuration set. Returns ------- dict Classification result. Raises ------ requests.ConnectionError If the network connection to weaviate fails. weaviate.UnexpectedStatusCodeException Unexpected error. z/classifications)pathweaviate_objectzClassification may not started.NzStart classification)rpostrRequestsConnectionError status_coder r )rresponseconn_errress r_startzConfigBuilder._starts" ['',,2DVZVbVb,cH   3 &,X7MNC? "?J+,BHMM ' [)*KLRZ Z [s'A A0 A++A0cR|j|j}|js|S|d}|jj |r1t j d|jj |r1tt|jj|S)z| Start the classification. Returns ------- dict Classification result. idg@) rLrWrr is_runningtimesleeprrGrI)rrTclassification_uuids rdozConfigBuilder.dos ;;=((O'tn""--.AB JJsO""--.ABD$..223FGHHr)rr)rN)__name__ __module__ __qualname____doc__rrstrr intr$r(listr,r0rGr6r:r=r?rBrLrWr^rrrrs*:*?O*$S_$,#/$D_$D_$To(4O(To( d0I.NN6IDIrr)rbr[typingrrrrrequests.exceptionsrrRweaviate.connectrweaviate.exceptionsr weaviate.utilr r rr rrfrrrls5 11J'=N.]I]Ir