diff --git a/skyvern/forge/sdk/schemas/task_v2.py b/skyvern/forge/sdk/schemas/task_v2.py index a10e48d7..ee70d3ad 100644 --- a/skyvern/forge/sdk/schemas/task_v2.py +++ b/skyvern/forge/sdk/schemas/task_v2.py @@ -78,8 +78,8 @@ class TaskV2(BaseModel): @field_validator("url", "webhook_callback_url", "totp_verification_url") @classmethod def validate_urls(cls, url: str | None) -> str | None: - if url is None: - return None + if not url: + return url return validate_url(url) @@ -137,8 +137,8 @@ class TaskV2Metadata(BaseModel): @field_validator("url") @classmethod def validate_urls(cls, v: str | None) -> str | None: - if v is None: - return None + if not v: + return v return validate_url(v) @@ -162,7 +162,7 @@ class TaskV2Request(BaseModel): @field_validator("url", "webhook_callback_url", "totp_verification_url") @classmethod def validate_urls(cls, url: str | None) -> str | None: - if url is None: - return None + if not url: + return url return validate_url(url) diff --git a/skyvern/forge/sdk/schemas/tasks.py b/skyvern/forge/sdk/schemas/tasks.py index 83706674..021b6c89 100644 --- a/skyvern/forge/sdk/schemas/tasks.py +++ b/skyvern/forge/sdk/schemas/tasks.py @@ -154,8 +154,8 @@ class TaskRequest(TaskBase): @field_validator("webhook_callback_url", "totp_verification_url") @classmethod def validate_optional_urls(cls, url: str | None) -> str | None: - if url is None: - return None + if not url: + return url return validate_url(url) diff --git a/skyvern/schemas/runs.py b/skyvern/schemas/runs.py index 08e6d586..c040ab70 100644 --- a/skyvern/schemas/runs.py +++ b/skyvern/schemas/runs.py @@ -353,8 +353,8 @@ class TaskRunRequest(BaseModel): Returns: The validated URL or None if no URL was provided """ - if url is None: - return None + if not url: + return url return validate_url(url) @@ -416,8 +416,8 @@ class WorkflowRunRequest(BaseModel): @field_validator("webhook_url", "totp_url") @classmethod def validate_urls(cls, url: str | None) -> str | None: - if url is None: - return None + if not url: + return url return validate_url(url) @model_validator(mode="after")