o Jjgz@spdZddlZddlmZddlmZmZmZmZm Z ddl m Z ddl m Z Gddde ZGd d d e ZdS) zEvaluators for parsing strings.N)eq)AnyCallableOptionalUnioncast)parse_json_markdown)StringEvaluatorc seZdZdZdeddffdd ZedefddZedefd d Z ede fd d Z  dd e de e de e dede f ddZZS)JsonValidityEvaluatoraEvaluate whether the prediction is valid JSON. This evaluator checks if the prediction is a valid JSON string. It does not require any input or reference. Attributes: requires_input (bool): Whether this evaluator requires an input string. Always False. requires_reference (bool): Whether this evaluator requires a reference string. Always False. evaluation_name (str): The name of the evaluation metric. Always "json". Examples: >>> evaluator = JsonValidityEvaluator() >>> prediction = '{"name": "John", "age": 30, "city": "New York"}' >>> evaluator.evaluate(prediction) {'score': 1} >>> prediction = '{"name": "John", "age": 30, "city": "New York",}' >>> evaluator.evaluate(prediction) {'score': 0, 'reasoning': 'Expecting property name enclosed in double quotes'} kwargsreturnNc stdSN)super__init__)selfr  __class__Y/var/www/html/zoom/venv/lib/python3.10/site-packages/langchain/evaluation/parsing/base.pyr%szJsonValidityEvaluator.__init__cCdSNFrrrrrrequires_input(z$JsonValidityEvaluator.requires_inputcCrrrrrrrrequires_reference,rz(JsonValidityEvaluator.requires_referencecCr)N json_validityrrrrrevaluation_name0rz%JsonValidityEvaluator.evaluation_name predictioninput referencec KsLz t|tjdddiWSty%}z dt|dWYd}~Sd}~ww)aLEvaluate the prediction string. Args: prediction (str): The prediction string to evaluate. input (str, optional): Not used in this evaluator. Defaults to None. reference (str, optional): Not used in this evaluator. Defaults to None. Returns: dict: A dictionary containing the evaluation score. The score is 1 if the prediction is valid JSON, and 0 otherwise. If the prediction is not valid JSON, the dictionary also contains a "reasoning" field with the error message. )parserscorer)r! reasoningN)rjsonloads Exceptionstr)rrrrr errr_evaluate_strings4s z'JsonValidityEvaluator._evaluate_stringsNN)__name__ __module__ __qualname____doc__rrpropertyboolrrr'rrdictr) __classcell__rrrrr s,r c seZdZdZddeededdffdd Zede fdd Z ede fd d Z ede fd d Z dedeeedee ee ffddZ  dde dee dee dedef ddZZS)JsonEqualityEvaluatoraEvaluate whether the prediction is equal to the reference after parsing both as JSON. This evaluator checks if the prediction, after parsing as JSON, is equal to the reference, which is also parsed as JSON. It does not require an input string. Attributes: requires_input (bool): Whether this evaluator requires an input string. Always False. requires_reference (bool): Whether this evaluator requires a reference string. Always True. evaluation_name (str): The name of the evaluation metric. Always "parsed_equality". Examples: >>> evaluator = JsonEqualityEvaluator() >>> evaluator.evaluate_strings('{"a": 1}', reference='{"a": 1}') {'score': True} >>> evaluator.evaluate_strings('{"a": 1}', reference='{"a": 2}') {'score': False} >>> evaluator = JsonEqualityEvaluator(operator=lambda x, y: x['a'] == y['a']) >>> evaluator.evaluate_strings('{"a": 1}', reference='{"a": 1}') {'score': True} >>> evaluator.evaluate_strings('{"a": 1}', reference='{"a": 2}') {'score': False} Noperatorr r c st|pt|_dSr )rrrr4)rr4r rrrros zJsonEqualityEvaluator.__init__cCrrrrrrrrsrz$JsonEqualityEvaluator.requires_inputcCr)NTrrrrrrwrz(JsonEqualityEvaluator.requires_referencecCr)N json_equalityrrrrrr{rz%JsonEqualityEvaluator.evaluation_namestringcCst|tr t|S|Sr ) isinstancer'r)rr6rrr _parse_jsons z!JsonEqualityEvaluator._parse_jsonrrrcKsf||}|tt|}t|tr+t|tsddiSt|ddd}t|ddd}d|||iS)aOEvaluate the prediction string. Args: prediction (str): The prediction string to evaluate. input (str, optional): Not used in this evaluator. reference (str): The reference string to compare against. Returns: dict: A dictionary containing the evaluation score. r!rcSt|Sr r'xrrrz9JsonEqualityEvaluator._evaluate_strings..)keycSr9r r:r;rrrr=r>)r8rr'r7listsortedr4)rrrrr parsedlabelrrrr)s   z'JsonEqualityEvaluator._evaluate_stringsr r*)r+r,r-r.rrrrr/r0rrr'rrr1r@floatintr8r)r2rrrrr3Ps6   r3)r.r$r4rtypingrrrrrlangchain_core.utils.jsonrlangchain.evaluation.schemar r r3rrrrs   D