vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

add observer_cruise_id to SkyvernContext and logs (#1469)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-01-02 12:49:48 -08:00
committed by GitHub
parent 3835f3ce69
commit b9d3228f77
3 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
skyvern/forge/sdk/core/skyvern_context.py
View File

@@ -10,6 +10,7 @@ class SkyvernContext:
task_id: str | None = None task_id: str | None = None
workflow_id: str | None = None workflow_id: str | None = None
workflow_run_id: str | None = None workflow_run_id: str | None = None
observer_cruise_id: str | None = None
max_steps_override: int | None = None max_steps_override: int | None = None
tz_info: ZoneInfo | None = None tz_info: ZoneInfo | None = None
totp_codes: dict[str, str | None] = field(default_factory=dict) totp_codes: dict[str, str | None] = field(default_factory=dict)

2
skyvern/forge/sdk/forge_log.py
View File

@@ -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 event_dict["workflow_id"] = context.workflow_id
if context.workflow_run_id: if context.workflow_run_id:
event_dict["workflow_run_id"] = 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 # Add env to the log
event_dict["env"] = settings.ENV event_dict["env"] = settings.ENV

4
skyvern/forge/sdk/services/observer_service.py
View File

@@ -84,6 +84,10 @@ async def initialize_observer_cruise(
prompt=user_prompt, prompt=user_prompt,
organization_id=organization.organization_id, 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_thought = await app.DATABASE.create_observer_thought(
observer_cruise_id=observer_cruise.observer_cruise_id, observer_cruise_id=observer_cruise.observer_cruise_id,