Forbid whitespaces in parameters (#3672)
This commit is contained in:
@@ -3,7 +3,7 @@ from dataclasses import dataclass
|
||||
from enum import StrEnum
|
||||
from typing import Annotated, Any, Literal
|
||||
|
||||
from pydantic import BaseModel, Field
|
||||
from pydantic import BaseModel, Field, field_validator
|
||||
|
||||
from skyvern.config import settings
|
||||
from skyvern.forge.sdk.workflow.models.parameter import OutputParameter, ParameterType, WorkflowParameterType
|
||||
@@ -74,6 +74,13 @@ class ParameterYAML(BaseModel, abc.ABC):
|
||||
key: str
|
||||
description: str | None = None
|
||||
|
||||
@field_validator("key")
|
||||
@classmethod
|
||||
def validate_no_whitespace(cls, v: str) -> str:
|
||||
if any(char in v for char in [" ", "\t", "\n", "\r"]):
|
||||
raise ValueError("Key cannot contain whitespaces")
|
||||
return v
|
||||
|
||||
|
||||
class AWSSecretParameterYAML(ParameterYAML):
|
||||
# There is a mypy bug with Literal. Without the type: ignore, mypy will raise an error:
|
||||
|
||||
Reference in New Issue
Block a user