?hPdZddlmZddlmZddlmZmZddl m Z GddZ y ) z" Classification class definition. )ConnectionError) Connection)get_valid_uuid_decode_json_response_dict) ConfigBuilderc|eZdZdZdefdZdefdZdede fdZ dede fdZ dede fd Z dede fd Zded ede fd Zy )Classificationz| Classification class used to schedule and/or check the status of a classification process of Weaviate objects. connectionc||_y)z Initialize a Classification class instance. Parameters ---------- connection : weaviate.connect.Connection Connection object to an active and running Weaviate instance. N) _connection)selfr s d/home/chris/cleankitchens-env/lib/python3.12/site-packages/weaviate/classification/classification.py__init__zClassification.__init__s &returnc.t|j|S)a Schedule a Classification of the Objects within Weaviate. Returns ------- weaviate.classification.config_builder.ConfigBuilder A ConfigBuilder that should be configured to the desired classification task )rr )rs rschedulezClassification.schedulesT--t44rclassification_uuidct|} |jjd|z}t |d}|J|S#t$r}td|d}~wwxYw)a- Polls the current state of the given classification. Parameters ---------- classification_uuid : str Identifier of the classification. Returns ------- dict A dict containing the Weaviate answer. Raises ------ ValueError If not a proper uuid. requests.ConnectionError If the network connection to weaviate fails. weaviate.UnexpectedStatusCodeException If weaviate reports a none OK status. z/classifications/)pathz-Classification status could not be retrieved.NzGet classification status)rr getRequestsConnectionErrorr)rrresponseconn_errress rrzClassification.get*s|0--@A ''++(+>>,H)3NO ' )?  s> A AAc&|j|dS)a4 Checks if a started classification job has completed. Parameters ---------- classification_uuid : str Identifier of the classification. Returns ------- bool True if given classification has finished, False otherwise. completed _check_statusrrs r is_completezClassification.is_completeQs!!"5{CCrc&|j|dS)a) Checks if a started classification job has failed. Parameters ---------- classification_uuid : str Identifier of the classification. Returns ------- bool True if the classification failed, False otherwise. failedrr!s r is_failedzClassification.is_failedbs!!"5x@@rc&|j|dS)a- Checks if a started classification job is running. Parameters ---------- classification_uuid : str Identifier of the classification. Returns ------- bool True if the classification is running, False otherwise. runningrr!s r is_runningzClassification.is_runningss!!"5yAArstatuscX |j|}|d|k(ryy#t$rYywxYw)aP Check for a status of a classification. Parameters ---------- classification_uuid : str Identifier of the classification. status : str Status to check for. Returns ------- bool True if 'status' is satisfied, False otherwise. Fr)T)rr)rrr)rs rr zClassification._check_statuss?" xx 34H H  ' '  s  ))N)__name__ __module__ __qualname____doc__rrrrstrdictrboolr"r%r(r rrr r s &: & 5- 5%s%t%NDsDtD"ASATA"BcBdB"cdrr N) r.requests.exceptionsrrweaviate.connectr weaviate.utilrrconfig_builderrr r2rrr7s%K'D)PPr