o Zh6@sdZddlmZddlmZmZmZmZmZddl m Z m Z m Z ddl mZddlmZddlmZddlmZmZdd lmZd efd d ZGd ddeeZGdddeZGdddeeZGdddeeZdS)z:String distance evaluators based on the RapidFuzz library.)Enum)AnyCallableDictListOptional)AsyncCallbackManagerForChainRunCallbackManagerForChainRun Callbacks)pre_init)Field)Chain)PairwiseStringEvaluatorStringEvaluator)RUN_KEYreturncCs(zddl}W|jStytdw)z Load the RapidFuzz library. Raises: ImportError: If the rapidfuzz library is not installed. Returns: Any: The rapidfuzz.distance module. rNzyPlease install the rapidfuzz library to use the FuzzyMatchStringEvaluator.Please install it with `pip install rapidfuzz`.) rapidfuzz ImportErrordistance)rr`/var/www/html/lang_env/lib/python3.10/site-packages/langchain/evaluation/string_distance/base.py_load_rapidfuzzs  rc@s(eZdZdZdZdZdZdZdZdZ dS) StringDistancea7Distance metric to use. Attributes: DAMERAU_LEVENSHTEIN: The Damerau-Levenshtein distance. LEVENSHTEIN: The Levenshtein distance. JARO: The Jaro distance. JARO_WINKLER: The Jaro-Winkler distance. HAMMING: The Hamming distance. INDEL: The Indel distance. Zdamerau_levenshteinZ levenshteinZjaroZ jaro_winklerZhammingZindelN) __name__ __module__ __qualname____doc__DAMERAU_LEVENSHTEIN LEVENSHTEINJARO JARO_WINKLERHAMMINGINDELrrrrr's rc@seZdZUdZeejdZeed<eddZ e ed< e de e efde e effdd Zedee fd d Zd e e efde e effd dZedde de defddZedefddZde de defddZdS)_RapidFuzzChainMixinz>> from langchain.evaluation import StringDistanceEvalChain >>> evaluator = StringDistanceEvalChain() >>> evaluator.evaluate_strings( prediction="Mindy is the CTO", reference="Mindy is the CEO", ) Using the `load_evaluator` function: >>> from langchain.evaluation import load_evaluator >>> evaluator = load_evaluator("string_distance") >>> evaluator.evaluate_strings( prediction="The answer is three", reference="three", ) rcCdS)z8 This evaluator does not require input. Frr*rrrrequires_inputz&StringDistanceEvalChain.requires_inputcCr?)z> This evaluator does not require a reference. Trr*rrrrequires_referencerAz*StringDistanceEvalChain.requires_referencecCddgS)^ Get the input keys. Returns: List[str]: The input keys. reference predictionrr*rrr input_keysz"StringDistanceEvalChain.input_keyscCs|jjdS)b Get the evaluation name. Returns: str: The evaluation name. _distancervaluer*rrrevaluation_namesz'StringDistanceEvalChain.evaluation_nameNinputs run_managercCd||d|diS)a^ Compute the string distance between the prediction and the reference. Args: inputs (Dict[str, Any]): The input values. run_manager (Optional[CallbackManagerForChainRun]): The callback manager. Returns: Dict[str, Any]: The evaluation results containing the score. r)rErFr7r+rNrOrrr_callszStringDistanceEvalChain._callcd||d|diS)a} Asynchronously compute the string distance between the prediction and the reference. Args: inputs (Dict[str, Any]): The input values. run_manager (Optional[AsyncCallbackManagerForChainRun]: The callback manager. Returns: Dict[str, Any]: The evaluation results containing the score. r)rErFrQrRrrr_acallszStringDistanceEvalChain._acallF)rEinput callbackstagsmetadatainclude_run_inforFrErVrWrXrYrZkwargsc Ks"|||d||||d} || S)a Evaluate the string distance between the prediction and the reference. Args: prediction (str): The prediction string. reference (Optional[str], optional): The reference string. input (Optional[str], optional): The input string. callbacks (Callbacks, optional): The callbacks to use. kwargs: Additional keyword arguments. Returns: dict: The evaluation results containing the score. rFrErNrWrXrYrZr/ r+rFrErVrWrXrYrZr[r-rrr_evaluate_stringss z)StringDistanceEvalChain._evaluate_stringsc s,|j||d||||dIdH} || S)a Asynchronously evaluate the string distance between the prediction and the reference. Args: prediction (str): The prediction string. reference (Optional[str], optional): The reference string. input (Optional[str], optional): The input string. callbacks (Callbacks, optional): The callbacks to use. kwargs: Additional keyword arguments. Returns: dict: The evaluation results containing the score. r\r]NZacallr/r_rrr_aevaluate_strings(s z*StringDistanceEvalChain._aevaluate_stringsN)rrrrr;r9r@rBrr:rGrMrrrr rSrrUr r.r`rbrrrrr>s            '    r>c@s(eZdZdZedeefddZedefddZ dde ee fd e e de ee ffd d Z dde ee fd e ede ee ffd d Zddddddededede eede e ee fdede defddZddddddededede eede e ee fdede defddZdS)PairwiseStringDistanceEvalChainz6Compute string edit distances between two predictions.rcCrC)rDrF prediction_brr*rrrrGOrHz*PairwiseStringDistanceEvalChain.input_keyscCsd|jjdS)rIZ pairwise_rJrKr*rrrrMYsz/PairwiseStringDistanceEvalChain.evaluation_nameNrNrOcCrP)aN Compute the string distance between two predictions. Args: inputs (Dict[str, Any]): The input values. run_manager (CallbackManagerForChainRun , optional): The callback manager. Returns: Dict[str, Any]: The evaluation results containing the score. r)rFrerQrRrrrrScsz%PairwiseStringDistanceEvalChain._callcrT)ab Asynchronously compute the string distance between two predictions. Args: inputs (Dict[str, Any]): The input values. run_manager (AsyncCallbackManagerForChainRun , optional): The callback manager. Returns: Dict[str, Any]: The evaluation results containing the score. r)rFrerQrRrrrrUwsz&PairwiseStringDistanceEvalChain._acallF)rWrXrYrZrFrerWrXrYrZr[c Ks"|||d||||d}||S)a$ Evaluate the string distance between two predictions. Args: prediction (str): The first prediction string. prediction_b (str): The second prediction string. callbacks (Callbacks, optional): The callbacks to use. tags (List[str], optional): Tags to apply to traces. metadata (Dict[str, Any], optional): Metadata to apply to traces. kwargs: Additional keyword arguments. Returns: dict: The evaluation results containing the score. rFrer]r^ r+rFrerWrXrYrZr[r-rrr_evaluate_string_pairss z6PairwiseStringDistanceEvalChain._evaluate_string_pairsc s,|j||d||||dIdH}||S)a3 Asynchronously evaluate the string distance between two predictions. Args: prediction (str): The first prediction string. prediction_b (str): The second prediction string. callbacks (Callbacks, optional): The callbacks to use. tags (List[str], optional): Tags to apply to traces. metadata (Dict[str, Any], optional): Metadata to apply to traces. kwargs: Additional keyword arguments. Returns: dict: The evaluation results containing the score. rfr]Nrargrrr_aevaluate_string_pairss z7PairwiseStringDistanceEvalChain._aevaluate_string_pairsrc)rrrrr;rr:rGrMrrrr rSrrUr r9r.rhrirrrrrdLs           '   rdN)renumrtypingrrrrrZ langchain_core.callbacks.managerrr r Zlangchain_core.utilsr Zpydanticr Zlangchain.chains.baser Zlangchain.evaluation.schemarrZlangchain.schemarrr:rr#r>rdrrrrs     l&