diff --git a/src/langtrace_python_sdk/instrumentation/cleanlab/patch.py b/src/langtrace_python_sdk/instrumentation/cleanlab/patch.py index abb8cf5f..8a2732bf 100644 --- a/src/langtrace_python_sdk/instrumentation/cleanlab/patch.py +++ b/src/langtrace_python_sdk/instrumentation/cleanlab/patch.py @@ -63,6 +63,10 @@ def traced_method( result_json = str(result) span.set_attribute("tlm.result", str(result_json)) + trustworthiness_score = result_json["trustworthiness_score"] + log = result_json["log"] + span.set_attribute("tlm.trustworthiness_score", str(trustworthiness_score)) + span.set_attribute("tlm.explanation", str(log.get("explanation", ""))) span.set_status(Status(StatusCode.OK)) return result