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

support bitwarden jinja style (#1706)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-02-03 23:18:41 +08:00
committed by GitHub
parent eb0d4db7ee
commit 1ba225002b
3 changed files with 369 additions and 247 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -339,7 +339,7 @@ async def run_observer_task_helper(
observer_cruise_id=observer_cruise_id, organization_id=organization_id, status=ObserverTaskStatus.running
)
await app.WORKFLOW_SERVICE.mark_workflow_run_as_running(workflow_run_id=workflow_run.workflow_run_id)
await _set_up_workflow_context(workflow_id, workflow_run_id)
await _set_up_workflow_context(workflow_id, workflow_run_id, organization)
url = str(observer_task.url)
user_prompt = observer_task.prompt
@@ -717,18 +717,20 @@ async def handle_block_result(
)
async def _set_up_workflow_context(workflow_id: str, workflow_run_id: str) -> None:
async def _set_up_workflow_context(workflow_id: str, workflow_run_id: str, organization: Organization) -> None:
"""
TODO: see if we could remove this function as we can just set an empty workflow context
"""
# Get all <workflow parameter, workflow run parameter> tuples
wp_wps_tuples = await app.WORKFLOW_SERVICE.get_workflow_run_parameter_tuples(workflow_run_id=workflow_run_id)
workflow_output_parameters = await app.WORKFLOW_SERVICE.get_workflow_output_parameters(workflow_id=workflow_id)
app.WORKFLOW_CONTEXT_MANAGER.initialize_workflow_run_context(
await app.WORKFLOW_CONTEXT_MANAGER.initialize_workflow_run_context(
organization,
workflow_run_id,
wp_wps_tuples,
workflow_output_parameters,
[],
[],
)