fix small bugs

Author: dinmukhamedm

src/lmnr/opentelemetry_lib/decorators/__init__.py

@@ -79,6 +79,8 @@ def _process_input(
     try:
         if input_formatter is not None:
             inp = input_formatter(*args, **kwargs)
+            if not isinstance(inp, str):
+                inp = json_dumps(inp)
         else:
             inp = get_input_from_func_args(
                 fn,
@@ -115,6 +117,8 @@ def _process_output(
     try:
         if output_formatter is not None:
             output = output_formatter(result)
+            if not isinstance(output, str):
+                output = json_dumps(output)
         else:
             output = result
 

src/lmnr/sdk/laminar.py

@@ -880,11 +880,13 @@ def use_span(
             raise
 
         finally:
-            context_api.detach(context_token)
-            detach_context(isolated_context_token)
-            wrapper.pop_span_context()
-            if end_on_exit:
-                span.end()
+            try:
+                context_api.detach(context_token)
+                detach_context(isolated_context_token)
+                wrapper.pop_span_context()
+            finally:
+                if end_on_exit:
+                    span.end()
 
     @classmethod
     def start_active_span(