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

auto prepend scheme to url (#1335)

Browse Source
This commit is contained in:
LawyZheng
2024-12-06 11:56:12 +08:00
committed by GitHub
parent db5b9d1dbd
commit 873b484ffa
5 changed files with 72 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
skyvern/forge/agent.py
View File

@@ -48,7 +48,7 @@ from skyvern.forge.sdk.api.files import get_path_for_workflow_download_directory
from skyvern.forge.sdk.artifact.models import ArtifactType
from skyvern.forge.sdk.core import skyvern_context
from skyvern.forge.sdk.core.security import generate_skyvern_signature
from skyvern.forge.sdk.core.validators import validate_url
from skyvern.forge.sdk.core.validators import prepend_scheme_and_validate_url
from skyvern.forge.sdk.db.enums import TaskType
from skyvern.forge.sdk.models import Organization, Step, StepStatus
from skyvern.forge.sdk.schemas.tasks import Task, TaskRequest, TaskResponse, TaskStatus
@@ -139,7 +139,11 @@ class ForgeAgent:
task_url = working_page.url
task_url = validate_url(task_url)
task_url = prepend_scheme_and_validate_url(task_url)
totp_verification_url = task_block.totp_verification_url
if totp_verification_url:
totp_verification_url = prepend_scheme_and_validate_url(totp_verification_url)
task = await app.DATABASE.create_task(
url=task_url,
task_type=task_block.task_type,
@@ -147,7 +151,7 @@ class ForgeAgent:
terminate_criterion=task_block.terminate_criterion,
title=task_block.title or task_block.label,
webhook_callback_url=None,
totp_verification_url=task_block.totp_verification_url,
totp_verification_url=totp_verification_url,
totp_identifier=task_block.totp_identifier,
navigation_goal=task_block.navigation_goal,
data_extraction_goal=task_block.data_extraction_goal,