From b9d3228f77af3fe26fcef5898ad50aa8268c04d0 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Thu, 2 Jan 2025 12:49:48 -0800 Subject: [PATCH] add observer_cruise_id to SkyvernContext and logs (#1469) --- skyvern/forge/sdk/core/skyvern_context.py | 1 + skyvern/forge/sdk/forge_log.py | 2 ++ skyvern/forge/sdk/services/observer_service.py | 4 ++++ 3 files changed, 7 insertions(+) diff --git a/skyvern/forge/sdk/core/skyvern_context.py b/skyvern/forge/sdk/core/skyvern_context.py index 8497e079..534bd4ba 100644 --- a/skyvern/forge/sdk/core/skyvern_context.py +++ b/skyvern/forge/sdk/core/skyvern_context.py @@ -10,6 +10,7 @@ class SkyvernContext: task_id: str | None = None workflow_id: str | None = None workflow_run_id: str | None = None + observer_cruise_id: str | None = None max_steps_override: int | None = None tz_info: ZoneInfo | None = None totp_codes: dict[str, str | None] = field(default_factory=dict) diff --git a/skyvern/forge/sdk/forge_log.py b/skyvern/forge/sdk/forge_log.py index a6862cc6..ac6e72a2 100644 --- a/skyvern/forge/sdk/forge_log.py +++ b/skyvern/forge/sdk/forge_log.py @@ -32,6 +32,8 @@ def add_kv_pairs_to_msg(logger: logging.Logger, method_name: str, event_dict: Ev event_dict["workflow_id"] = context.workflow_id if context.workflow_run_id: event_dict["workflow_run_id"] = context.workflow_run_id + if context.observer_cruise_id: + event_dict["observer_cruise_id"] = context.observer_cruise_id # Add env to the log event_dict["env"] = settings.ENV diff --git a/skyvern/forge/sdk/services/observer_service.py b/skyvern/forge/sdk/services/observer_service.py index 83e01c84..d57cc6d1 100644 --- a/skyvern/forge/sdk/services/observer_service.py +++ b/skyvern/forge/sdk/services/observer_service.py @@ -84,6 +84,10 @@ async def initialize_observer_cruise( prompt=user_prompt, organization_id=organization.organization_id, ) + # set observer cruise id in context + context = skyvern_context.current() + if context: + context.observer_cruise_id = observer_cruise.observer_cruise_id observer_thought = await app.DATABASE.create_observer_thought( observer_cruise_id=observer_cruise.observer_cruise_id,