laminar integration (#2887)

skyvern/forge/sdk/workflow/models/block.py

@@ -54,6 +54,7 @@ from skyvern.forge.sdk.db.enums import TaskType
 from skyvern.forge.sdk.schemas.files import FileInfo
 from skyvern.forge.sdk.schemas.task_v2 import TaskV2Status
 from skyvern.forge.sdk.schemas.tasks import Task, TaskOutput, TaskStatus
+from skyvern.forge.sdk.trace import TraceManager
 from skyvern.forge.sdk.workflow.context_manager import BlockMetadata, WorkflowRunContext
 from skyvern.forge.sdk.workflow.exceptions import (
     CustomizedCodeException,
@@ -292,6 +293,7 @@ class Block(BaseModel, abc.ABC):
                 organization_id=organization_id,
             )
 
+    @TraceManager.traced_async(ignore_inputs=["kwargs"])
     async def execute_safe(
         self,
         workflow_run_id: str,

skyvern/forge/sdk/workflow/service.py

@@ -30,6 +30,7 @@ from skyvern.forge.sdk.schemas.files import FileInfo
 from skyvern.forge.sdk.schemas.organizations import Organization
 from skyvern.forge.sdk.schemas.tasks import Task
 from skyvern.forge.sdk.schemas.workflow_runs import WorkflowRunBlock, WorkflowRunTimeline, WorkflowRunTimelineType
+from skyvern.forge.sdk.trace import TraceManager
 from skyvern.forge.sdk.workflow.exceptions import (
     ContextParameterSourceNotDefined,
     InvalidWaitBlockTime,
@@ -248,6 +249,7 @@ class WorkflowService:
 
         return workflow_run
 
+    @TraceManager.traced_async(ignore_inputs=["organization", "api_key"])
     async def execute_workflow(
         self,
         workflow_run_id: str,