From 25d423021a43309b01b8c55072ca8625345a60a8 Mon Sep 17 00:00:00 2001 From: ruiciroT Date: Mon, 24 Jun 2024 07:45:53 -0600 Subject: [PATCH] Fix: Adding status setting to span creation --- guardrails/utils/hub_telemetry_utils.py | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/guardrails/utils/hub_telemetry_utils.py b/guardrails/utils/hub_telemetry_utils.py index 6d9517130..41dcf01a4 100644 --- a/guardrails/utils/hub_telemetry_utils.py +++ b/guardrails/utils/hub_telemetry_utils.py @@ -118,9 +118,14 @@ def create_new_span( span_name, # type: ignore (Fails in Python 3.8 for invalid reason) context=self.extract_current_context() if has_parent else None, ) as span: - if is_parent: - # Inject the current context - self.inject_current_context() - - for attribute in attributes: - span.set_attribute(attribute[0], attribute[1]) + try: + if is_parent: + # Inject the current context + self.inject_current_context() + for attribute in attributes: + span.set_attribute(attribute[0], attribute[1]) + + span.set_status(trace.Status(trace.StatusCode.OK)) + except Exception as e: + logging.error("An error occured during sopan creation: " + str(e)) + span.set_status(trace.Status(trace.StatusCode.ERROR))