diff --git a/skyvern/forge/sdk/schemas/task_v2.py b/skyvern/forge/sdk/schemas/task_v2.py index c2f4d24f..62a08cd2 100644 --- a/skyvern/forge/sdk/schemas/task_v2.py +++ b/skyvern/forge/sdk/schemas/task_v2.py @@ -4,8 +4,8 @@ from typing import Any from pydantic import BaseModel, ConfigDict, Field, field_validator -from skyvern.forge.sdk.core.validators import validate_url from skyvern.schemas.runs import ProxyLocation +from skyvern.utils.url_validators import validate_url DEFAULT_WORKFLOW_TITLE = "New Workflow" diff --git a/skyvern/forge/sdk/schemas/tasks.py b/skyvern/forge/sdk/schemas/tasks.py index ad5324a7..cca48dd9 100644 --- a/skyvern/forge/sdk/schemas/tasks.py +++ b/skyvern/forge/sdk/schemas/tasks.py @@ -7,10 +7,10 @@ from typing import Any from pydantic import BaseModel, Field, field_validator from skyvern.exceptions import InvalidTaskStatusTransition, TaskAlreadyCanceled, TaskAlreadyTimeout -from skyvern.forge.sdk.core.validators import validate_url from skyvern.forge.sdk.db.enums import TaskType from skyvern.forge.sdk.schemas.files import FileInfo from skyvern.schemas.runs import ProxyLocation +from skyvern.utils.url_validators import validate_url class TaskBase(BaseModel): diff --git a/skyvern/forge/sdk/workflow/models/block.py b/skyvern/forge/sdk/workflow/models/block.py index 6e0aead2..ac6f2515 100644 --- a/skyvern/forge/sdk/workflow/models/block.py +++ b/skyvern/forge/sdk/workflow/models/block.py @@ -49,7 +49,6 @@ from skyvern.forge.sdk.api.files import ( ) from skyvern.forge.sdk.api.llm.api_handler_factory import LLMAPIHandlerFactory from skyvern.forge.sdk.artifact.models import ArtifactType -from skyvern.forge.sdk.core.validators import prepend_scheme_and_validate_url from skyvern.forge.sdk.db.enums import TaskType from skyvern.forge.sdk.schemas.files import FileInfo from skyvern.forge.sdk.schemas.task_v2 import TaskV2Status @@ -72,6 +71,7 @@ from skyvern.forge.sdk.workflow.models.parameter import ( OutputParameter, WorkflowParameter, ) +from skyvern.utils.url_validators import prepend_scheme_and_validate_url from skyvern.webeye.browser_factory import BrowserState from skyvern.webeye.utils.page import SkyvernFrame diff --git a/skyvern/forge/sdk/workflow/models/workflow.py b/skyvern/forge/sdk/workflow/models/workflow.py index 80afc2f2..4c3a4ee3 100644 --- a/skyvern/forge/sdk/workflow/models/workflow.py +++ b/skyvern/forge/sdk/workflow/models/workflow.py @@ -4,13 +4,13 @@ from typing import Any, List from pydantic import BaseModel, field_validator -from skyvern.forge.sdk.core.validators import validate_url from skyvern.forge.sdk.schemas.files import FileInfo from skyvern.forge.sdk.schemas.task_v2 import TaskV2 from skyvern.forge.sdk.workflow.exceptions import WorkflowDefinitionHasDuplicateBlockLabels from skyvern.forge.sdk.workflow.models.block import BlockTypeVar from skyvern.forge.sdk.workflow.models.parameter import PARAMETER_TYPE from skyvern.schemas.runs import ProxyLocation +from skyvern.utils.url_validators import validate_url class WorkflowRequestBody(BaseModel): diff --git a/skyvern/schemas/runs.py b/skyvern/schemas/runs.py index 17899a4b..bd0671e2 100644 --- a/skyvern/schemas/runs.py +++ b/skyvern/schemas/runs.py @@ -4,7 +4,7 @@ from zoneinfo import ZoneInfo from pydantic import BaseModel, field_validator -from skyvern.forge.sdk.core.validators import validate_url +from skyvern.utils.url_validators import validate_url class ProxyLocation(StrEnum): diff --git a/skyvern/forge/sdk/core/validators.py b/skyvern/utils/url_validators.py similarity index 100% rename from skyvern/forge/sdk/core/validators.py rename to skyvern/utils/url_validators.py