vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-09-15 01:39:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

add observer cruise to get workflow_run api response (#1470)

Browse source
This commit is contained in:
Shuchang Zheng 2025-01-02 12:54:18 -08:00 • committed by GitHub
parent b9d3228f77
commit c85d5428f3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 10 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

9
skyvern/forge/sdk/routes/agent_protocol.py
View file

@ -720,12 +720,19 @@ async def get_workflow_run(
current_org: Organization = Depends(org_auth_service.get_current_org), current_org: Organization = Depends(org_auth_service.get_current_org),
) -> WorkflowRunStatusResponse: ) -> WorkflowRunStatusResponse:
analytics.capture("skyvern-oss-agent-workflow-run-get") analytics.capture("skyvern-oss-agent-workflow-run-get")
return await app.WORKFLOW_SERVICE.build_workflow_run_status_response( workflow_run_status_response = await app.WORKFLOW_SERVICE.build_workflow_run_status_response(
workflow_permanent_id=workflow_id, workflow_permanent_id=workflow_id,
workflow_run_id=workflow_run_id, workflow_run_id=workflow_run_id,
organization_id=current_org.organization_id, organization_id=current_org.organization_id,
include_cost=True, include_cost=True,
) )
observer_cruise = await app.DATABASE.get_observer_cruise_by_workflow_run_id(
workflow_run_id=workflow_run_id,
organization_id=current_org.organization_id,
)
if observer_cruise:
workflow_run_status_response.observer_cruise = observer_cruise
return workflow_run_status_response
@base_router.get( @base_router.get(

2
skyvern/forge/sdk/workflow/models/workflow.py
View file

@ -5,6 +5,7 @@ from typing import Any, List
from pydantic import BaseModel, field_validator from pydantic import BaseModel, field_validator
from skyvern.forge.sdk.core.validators import validate_url from skyvern.forge.sdk.core.validators import validate_url
from skyvern.forge.sdk.schemas.observers import ObserverCruise
from skyvern.forge.sdk.schemas.tasks import ProxyLocation from skyvern.forge.sdk.schemas.tasks import ProxyLocation
from skyvern.forge.sdk.workflow.exceptions import WorkflowDefinitionHasDuplicateBlockLabels from skyvern.forge.sdk.workflow.exceptions import WorkflowDefinitionHasDuplicateBlockLabels
from skyvern.forge.sdk.workflow.models.block import BlockTypeVar from skyvern.forge.sdk.workflow.models.block import BlockTypeVar
@ -136,3 +137,4 @@ class WorkflowRunStatusResponse(BaseModel):
outputs: dict[str, Any] | None = None outputs: dict[str, Any] | None = None
total_steps: int | None = None total_steps: int | None = None
total_cost: float | None = None total_cost: float | None = None
observer_cruise: ObserverCruise | None = None