Changed to use with instead of explicit try-catch.

Author: mfateev

langchain/langchain_interceptor.py

@@ -142,23 +142,14 @@ def start_activity(
         self, input: worker.StartActivityInput
     ) -> workflow.ActivityHandle:
         with workflow.unsafe.sandbox_unrestricted():
-            t = trace(name=f"start_activity:{input.activity}", run_id=workflow.uuid4())
-            t.__enter__()
-        try:
-            set_header_from_context(input, workflow.payload_converter())
-            return self.next.start_activity(input)
-        finally:
-            with workflow.unsafe.sandbox_unrestricted():
-                t.__exit__()
+            with trace(name=f"start_activity:{input.activity}", run_id=workflow.uuid4()):
+                set_header_from_context(input, workflow.payload_converter())
+                return self.next.start_activity(input)
 
     async def start_child_workflow(
         self, input: worker.StartChildWorkflowInput
     ) -> workflow.ChildWorkflowHandle:
         with workflow.unsafe.sandbox_unrestricted():
-            t = trace(name=f"start_child_workflow:{input.workflow}", run_id=workflow.uuid4())
-        try:
-            set_header_from_context(input, workflow.payload_converter())
-            return await self.next.start_child_workflow(input)
-        finally:
-            with workflow.unsafe.sandbox_unrestricted():
-                t.__exit__()
+            with trace(name=f"start_child_workflow:{input.workflow}", run_id=workflow.uuid4()):
+                set_header_from_context(input, workflow.payload_converter())
+                return await self.next.start_child_workflow(input)