diff --git a/src/backend/base/langflow/api/v1/schemas.py b/src/backend/base/langflow/api/v1/schemas.py index f5d0b97d1..0cc6274c2 100644 --- a/src/backend/base/langflow/api/v1/schemas.py +++ b/src/backend/base/langflow/api/v1/schemas.py @@ -26,7 +26,7 @@ class BuildStatus(Enum): class TweaksRequest(BaseModel): - tweaks: Optional[Dict[str, Dict[str, str]]] = Field(default_factory=dict) + tweaks: Optional[Dict[str, Dict[str, Any]]] = Field(default_factory=dict) class UpdateTemplateRequest(BaseModel): diff --git a/src/backend/base/langflow/schema/graph.py b/src/backend/base/langflow/schema/graph.py index a515b3dd2..6b6cca66a 100644 --- a/src/backend/base/langflow/schema/graph.py +++ b/src/backend/base/langflow/schema/graph.py @@ -1,7 +1,8 @@ -from typing import List, Optional, Union +from typing import Any, List, Optional, Union + +from pydantic import BaseModel, Field, RootModel from langflow.schema.schema import InputType -from pydantic import BaseModel, Field, RootModel class InputValue(BaseModel): @@ -14,7 +15,7 @@ class InputValue(BaseModel): class Tweaks(RootModel): - root: dict[str, Union[str, dict[str, str]]] = Field( + root: dict[str, Union[str, dict[str, Any]]] = Field( description="A dictionary of tweaks to adjust the flow's execution. Allows customizing flow behavior dynamically. All tweaks are overridden by the input values.", ) model_config = {